blockTools.ts 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559
  1. import { DiscardBlock } from 'babylonjs/Materials/Node/Blocks/Fragment/discardBlock';
  2. import { BonesBlock } from 'babylonjs/Materials/Node/Blocks/Vertex/bonesBlock';
  3. import { InstancesBlock } from 'babylonjs/Materials/Node/Blocks/Vertex/instancesBlock';
  4. import { MorphTargetsBlock } from 'babylonjs/Materials/Node/Blocks/Vertex/morphTargetsBlock';
  5. import { ImageProcessingBlock } from 'babylonjs/Materials/Node/Blocks/Fragment/imageProcessingBlock';
  6. import { ColorMergerBlock } from 'babylonjs/Materials/Node/Blocks/colorMergerBlock';
  7. import { VectorMergerBlock } from 'babylonjs/Materials/Node/Blocks/vectorMergerBlock';
  8. import { ColorSplitterBlock } from 'babylonjs/Materials/Node/Blocks/colorSplitterBlock';
  9. import { VectorSplitterBlock } from 'babylonjs/Materials/Node/Blocks/vectorSplitterBlock';
  10. import { RemapBlock } from 'babylonjs/Materials/Node/Blocks/remapBlock';
  11. import { TextureBlock } from 'babylonjs/Materials/Node/Blocks/Dual/textureBlock';
  12. import { ReflectionTextureBlock } from 'babylonjs/Materials/Node/Blocks/Dual/reflectionTextureBlock';
  13. import { LightBlock } from 'babylonjs/Materials/Node/Blocks/Dual/lightBlock';
  14. import { FogBlock } from 'babylonjs/Materials/Node/Blocks/Dual/fogBlock';
  15. import { VertexOutputBlock } from 'babylonjs/Materials/Node/Blocks/Vertex/vertexOutputBlock';
  16. import { FragmentOutputBlock } from 'babylonjs/Materials/Node/Blocks/Fragment/fragmentOutputBlock';
  17. import { NormalizeBlock } from 'babylonjs/Materials/Node/Blocks/normalizeBlock';
  18. import { AddBlock } from 'babylonjs/Materials/Node/Blocks/addBlock';
  19. import { ScaleBlock } from 'babylonjs/Materials/Node/Blocks/scaleBlock';
  20. import { TrigonometryBlock, TrigonometryBlockOperations } from 'babylonjs/Materials/Node/Blocks/trigonometryBlock';
  21. import { ClampBlock } from 'babylonjs/Materials/Node/Blocks/clampBlock';
  22. import { CrossBlock } from 'babylonjs/Materials/Node/Blocks/crossBlock';
  23. import { DotBlock } from 'babylonjs/Materials/Node/Blocks/dotBlock';
  24. import { MultiplyBlock } from 'babylonjs/Materials/Node/Blocks/multiplyBlock';
  25. import { TransformBlock } from 'babylonjs/Materials/Node/Blocks/transformBlock';
  26. import { NodeMaterialBlockConnectionPointTypes } from 'babylonjs/Materials/Node/Enums/nodeMaterialBlockConnectionPointTypes';
  27. import { FresnelBlock } from 'babylonjs/Materials/Node/Blocks/fresnelBlock';
  28. import { LerpBlock } from 'babylonjs/Materials/Node/Blocks/lerpBlock';
  29. import { NLerpBlock } from 'babylonjs/Materials/Node/Blocks/nLerpBlock';
  30. import { DivideBlock } from 'babylonjs/Materials/Node/Blocks/divideBlock';
  31. import { SubtractBlock } from 'babylonjs/Materials/Node/Blocks/subtractBlock';
  32. import { StepBlock } from 'babylonjs/Materials/Node/Blocks/stepBlock';
  33. import { SmoothStepBlock } from 'babylonjs/Materials/Node/Blocks/smoothStepBlock';
  34. import { InputBlock } from 'babylonjs/Materials/Node/Blocks/Input/inputBlock';
  35. import { NodeMaterialSystemValues } from 'babylonjs/Materials/Node/Enums/nodeMaterialSystemValues';
  36. import { AnimatedInputBlockTypes } from 'babylonjs/Materials/Node/Blocks/Input/animatedInputBlockTypes';
  37. import { OneMinusBlock } from 'babylonjs/Materials/Node/Blocks/oneMinusBlock';
  38. import { ViewDirectionBlock } from 'babylonjs/Materials/Node/Blocks/viewDirectionBlock';
  39. import { LightInformationBlock } from 'babylonjs/Materials/Node/Blocks/Vertex/lightInformationBlock';
  40. import { MaxBlock } from 'babylonjs/Materials/Node/Blocks/maxBlock';
  41. import { MinBlock } from 'babylonjs/Materials/Node/Blocks/minBlock';
  42. import { PerturbNormalBlock } from 'babylonjs/Materials/Node/Blocks/Fragment/perturbNormalBlock';
  43. import { LengthBlock } from 'babylonjs/Materials/Node/Blocks/lengthBlock';
  44. import { DistanceBlock } from 'babylonjs/Materials/Node/Blocks/distanceBlock';
  45. import { FrontFacingBlock } from 'babylonjs/Materials/Node/Blocks/Fragment/frontFacingBlock';
  46. import { NegateBlock } from 'babylonjs/Materials/Node/Blocks/negateBlock';
  47. import { PowBlock } from 'babylonjs/Materials/Node/Blocks/powBlock';
  48. import { Scene } from 'babylonjs/scene';
  49. import { RandomNumberBlock } from 'babylonjs/Materials/Node/Blocks/randomNumberBlock';
  50. import { ReplaceColorBlock } from 'babylonjs/Materials/Node/Blocks/replaceColorBlock';
  51. import { PosterizeBlock } from 'babylonjs/Materials/Node/Blocks/posterizeBlock';
  52. import { ArcTan2Block } from 'babylonjs/Materials/Node/Blocks/arcTan2Block';
  53. import { ReciprocalBlock } from 'babylonjs/Materials/Node/Blocks/reciprocalBlock';
  54. import { GradientBlock } from 'babylonjs/Materials/Node/Blocks/gradientBlock';
  55. import { WaveBlock, WaveBlockKind } from 'babylonjs/Materials/Node/Blocks/waveBlock';
  56. import { NodeMaterial } from 'babylonjs/Materials/Node/nodeMaterial';
  57. import { WorleyNoise3DBlock } from 'babylonjs/Materials/Node/Blocks/worleyNoise3DBlock';
  58. import { SimplexPerlin3DBlock } from 'babylonjs/Materials/Node/Blocks/simplexPerlin3DBlock';
  59. import { NormalBlendBlock } from 'babylonjs/Materials/Node/Blocks/normalBlendBlock';
  60. import { Rotate2dBlock } from 'babylonjs/Materials/Node/Blocks/rotate2dBlock';
  61. import { DerivativeBlock } from 'babylonjs/Materials/Node/Blocks/Fragment/derivativeBlock';
  62. import { RefractBlock } from 'babylonjs/Materials/Node/Blocks/refractBlock';
  63. import { ReflectBlock } from 'babylonjs/Materials/Node/Blocks/reflectBlock';
  64. import { DesaturateBlock } from 'babylonjs/Materials/Node/Blocks/desaturateBlock';
  65. import { PBRMetallicRoughnessBlock } from 'babylonjs/Materials/Node/Blocks/PBR/pbrMetallicRoughnessBlock';
  66. import { SheenBlock } from 'babylonjs/Materials/Node/Blocks/PBR/sheenBlock';
  67. import { AmbientOcclusionBlock } from 'babylonjs/Materials/Node/Blocks/PBR/ambientOcclusionBlock';
  68. import { ReflectivityBlock } from 'babylonjs/Materials/Node/Blocks/PBR/reflectivityBlock';
  69. import { AnisotropyBlock } from 'babylonjs/Materials/Node/Blocks/PBR/anisotropyBlock';
  70. import { ReflectionBlock } from 'babylonjs/Materials/Node/Blocks/PBR/reflectionBlock';
  71. import { ClearCoatBlock } from 'babylonjs/Materials/Node/Blocks/PBR/clearCoatBlock';
  72. import { RefractionBlock } from 'babylonjs/Materials/Node/Blocks/PBR/refractionBlock';
  73. import { SubSurfaceBlock } from 'babylonjs/Materials/Node/Blocks/PBR/subSurfaceBlock';
  74. import { CurrentScreenBlock } from 'babylonjs/Materials/Node/Blocks/Dual/currentScreenBlock';
  75. import { ParticleTextureBlock } from 'babylonjs/Materials/Node/Blocks/Particle/particleTextureBlock';
  76. export class BlockTools {
  77. public static GetBlockFromString(data: string, scene: Scene, nodeMaterial: NodeMaterial) {
  78. switch (data) {
  79. case "DesaturateBlock":
  80. return new DesaturateBlock("Desaturate");
  81. case "RefractBlock":
  82. return new RefractBlock("Refract");
  83. case "ReflectBlock":
  84. return new ReflectBlock("Reflect");
  85. case "DerivativeBlock":
  86. return new DerivativeBlock("Derivative");
  87. case "Rotate2dBlock":
  88. return new Rotate2dBlock("Rotate2d");
  89. case "NormalBlendBlock":
  90. return new NormalBlendBlock("NormalBlend");
  91. case "WorleyNoise3DBlock":
  92. return new WorleyNoise3DBlock("WorleyNoise3D");
  93. case "SimplexPerlin3DBlock":
  94. return new SimplexPerlin3DBlock("SimplexPerlin3D");
  95. case "BonesBlock":
  96. return new BonesBlock("Bones");
  97. case "InstancesBlock":
  98. return new InstancesBlock("Instances");
  99. case "MorphTargetsBlock":
  100. return new MorphTargetsBlock("MorphTargets");
  101. case "DiscardBlock":
  102. return new DiscardBlock("Discard");
  103. case "ImageProcessingBlock":
  104. return new ImageProcessingBlock("ImageProcessing");
  105. case "ColorMergerBlock":
  106. return new ColorMergerBlock("ColorMerger");
  107. case "VectorMergerBlock":
  108. return new VectorMergerBlock("VectorMerger");
  109. case "ColorSplitterBlock":
  110. return new ColorSplitterBlock("ColorSplitter");
  111. case "VectorSplitterBlock":
  112. return new VectorSplitterBlock("VectorSplitter");
  113. case "TextureBlock":
  114. return new TextureBlock("Texture");
  115. case "ReflectionTextureBlock":
  116. return new ReflectionTextureBlock("Reflection texture");
  117. case "LightBlock":
  118. return new LightBlock("Lights");
  119. case "FogBlock":
  120. return new FogBlock("Fog");
  121. case "VertexOutputBlock":
  122. return new VertexOutputBlock("VertexOutput");
  123. case "FragmentOutputBlock":
  124. return new FragmentOutputBlock("FragmentOutput");
  125. case "AddBlock":
  126. return new AddBlock("Add");
  127. case "ClampBlock":
  128. return new ClampBlock("Clamp");
  129. case "ScaleBlock":
  130. return new ScaleBlock("Scale");
  131. case "CrossBlock":
  132. return new CrossBlock("Cross");
  133. case "DotBlock":
  134. return new DotBlock("Dot");
  135. case "PowBlock":
  136. return new PowBlock("Pow");
  137. case "MultiplyBlock":
  138. return new MultiplyBlock("Multiply");
  139. case "TransformBlock":
  140. return new TransformBlock("Transform");
  141. case "TrigonometryBlock":
  142. return new TrigonometryBlock("Trigonometry");
  143. case "RemapBlock":
  144. return new RemapBlock("Remap");
  145. case "NormalizeBlock":
  146. return new NormalizeBlock("Normalize");
  147. case "FresnelBlock":
  148. return new FresnelBlock("Fresnel");
  149. case "LerpBlock":
  150. return new LerpBlock("Lerp");
  151. case "NLerpBlock":
  152. return new NLerpBlock("NLerp");
  153. case "DivideBlock":
  154. return new DivideBlock("Divide");
  155. case "SubtractBlock":
  156. return new SubtractBlock("Subtract");
  157. case "StepBlock":
  158. return new StepBlock("Step");
  159. case "SmoothStepBlock":
  160. return new SmoothStepBlock("Smooth step");
  161. case "OneMinusBlock":
  162. return new OneMinusBlock("One minus");
  163. case "ReciprocalBlock":
  164. return new ReciprocalBlock("Reciprocal");
  165. case "ViewDirectionBlock":
  166. return new ViewDirectionBlock("View direction");
  167. case "LightInformationBlock":
  168. let lightInformationBlock = new LightInformationBlock("Light information");
  169. lightInformationBlock.light = scene.lights.length ? scene.lights[0] : null;
  170. return lightInformationBlock;
  171. case "MaxBlock":
  172. return new MaxBlock("Max");
  173. case "MinBlock":
  174. return new MinBlock("Min");
  175. case "LengthBlock":
  176. return new LengthBlock("Length");
  177. case "DistanceBlock":
  178. return new DistanceBlock("Distance");
  179. case "NegateBlock":
  180. return new NegateBlock("Negate");
  181. case "PerturbNormalBlock":
  182. return new PerturbNormalBlock("Perturb normal");
  183. case "RandomNumberBlock":
  184. return new RandomNumberBlock("Random number");
  185. case "ReplaceColorBlock":
  186. return new ReplaceColorBlock("Replace color");
  187. case "PosterizeBlock":
  188. return new PosterizeBlock("Posterize");
  189. case "ArcTan2Block":
  190. return new ArcTan2Block("ArcTan2");
  191. case "GradientBlock":
  192. return new GradientBlock("Gradient");
  193. case "FrontFacingBlock":
  194. return new FrontFacingBlock("Front facing");
  195. case "CosBlock": {
  196. let cosBlock = new TrigonometryBlock("Cos");
  197. cosBlock.operation = TrigonometryBlockOperations.Cos;
  198. return cosBlock;
  199. }
  200. case "SinBlock": {
  201. let sinBlock = new TrigonometryBlock("Sin");
  202. sinBlock.operation = TrigonometryBlockOperations.Sin;
  203. return sinBlock;
  204. }
  205. case "AbsBlock": {
  206. let absBlock = new TrigonometryBlock("Abs");
  207. absBlock.operation = TrigonometryBlockOperations.Abs;
  208. return absBlock;
  209. }
  210. case "SqrtBlock": {
  211. let sqrtBlock = new TrigonometryBlock("Sqrt");
  212. sqrtBlock.operation = TrigonometryBlockOperations.Sqrt;
  213. return sqrtBlock;
  214. }
  215. case "ArcCosBlock": {
  216. let acosBlock = new TrigonometryBlock("ArcCos");
  217. acosBlock.operation = TrigonometryBlockOperations.ArcCos;
  218. return acosBlock;
  219. }
  220. case "ArcSinBlock": {
  221. let asinBlock = new TrigonometryBlock("ArcSin");
  222. asinBlock.operation = TrigonometryBlockOperations.ArcSin;
  223. return asinBlock;
  224. }
  225. case "TanBlock": {
  226. let tanBlock = new TrigonometryBlock("Tan");
  227. tanBlock.operation = TrigonometryBlockOperations.Tan;
  228. return tanBlock;
  229. }
  230. case "ArcTanBlock": {
  231. let atanBlock = new TrigonometryBlock("ArcTan");
  232. atanBlock.operation = TrigonometryBlockOperations.ArcTan;
  233. return atanBlock;
  234. }
  235. case "FractBlock": {
  236. let fractBlock = new TrigonometryBlock("Fract");
  237. fractBlock.operation = TrigonometryBlockOperations.Fract;
  238. return fractBlock;
  239. }
  240. case "SignBlock": {
  241. let signBlock = new TrigonometryBlock("Sign");
  242. signBlock.operation = TrigonometryBlockOperations.Sign;
  243. return signBlock;
  244. }
  245. case "LogBlock": {
  246. let logBlock = new TrigonometryBlock("Log");
  247. logBlock.operation = TrigonometryBlockOperations.Log;
  248. return logBlock;
  249. }
  250. case "ExpBlock": {
  251. let expBlock = new TrigonometryBlock("Exp");
  252. expBlock.operation = TrigonometryBlockOperations.Exp;
  253. return expBlock;
  254. }
  255. case "Exp2Block": {
  256. let exp2Block = new TrigonometryBlock("Exp2");
  257. exp2Block.operation = TrigonometryBlockOperations.Exp2;
  258. return exp2Block;
  259. }
  260. case "DegreesToRadiansBlock": {
  261. let degreesToRadiansBlock = new TrigonometryBlock("Degrees to radians");
  262. degreesToRadiansBlock.operation = TrigonometryBlockOperations.Radians;
  263. return degreesToRadiansBlock;
  264. }
  265. case "RadiansToDegreesBlock": {
  266. let radiansToDegreesBlock = new TrigonometryBlock("Radians to degrees");
  267. radiansToDegreesBlock.operation = TrigonometryBlockOperations.Degrees;
  268. return radiansToDegreesBlock;
  269. }
  270. case "RoundBlock": {
  271. let roundBlock = new TrigonometryBlock("Round");
  272. roundBlock.operation = TrigonometryBlockOperations.Round;
  273. return roundBlock;
  274. }
  275. case "CeilingBlock": {
  276. let ceilingBlock = new TrigonometryBlock("Ceiling");
  277. ceilingBlock.operation = TrigonometryBlockOperations.Ceiling;
  278. return ceilingBlock;
  279. }
  280. case "FloorBlock": {
  281. let floorBlock = new TrigonometryBlock("Floor");
  282. floorBlock.operation = TrigonometryBlockOperations.Floor;
  283. return floorBlock;
  284. }
  285. case "SawToothWaveBlock": {
  286. let sawToothWaveBlock = new WaveBlock("SawTooth wave");
  287. sawToothWaveBlock.kind = WaveBlockKind.SawTooth;
  288. return sawToothWaveBlock;
  289. }
  290. case "SquareWaveBlock": {
  291. let squareWaveBlock = new WaveBlock("Square wave");
  292. squareWaveBlock.kind = WaveBlockKind.Square;
  293. return squareWaveBlock;
  294. }
  295. case "TriangleWaveBlock": {
  296. let triangleWaveBlock = new WaveBlock("Triangle wave");
  297. triangleWaveBlock.kind = WaveBlockKind.Triangle;
  298. return triangleWaveBlock;
  299. }
  300. case "WorldMatrixBlock": {
  301. let worldMatrixBlock = new InputBlock("World");
  302. worldMatrixBlock.setAsSystemValue(NodeMaterialSystemValues.World);
  303. return worldMatrixBlock;
  304. }
  305. case "WorldViewMatrixBlock": {
  306. let worldViewMatrixBlock = new InputBlock("World x View");
  307. worldViewMatrixBlock.setAsSystemValue(NodeMaterialSystemValues.WorldView);
  308. return worldViewMatrixBlock;
  309. }
  310. case "WorldViewProjectionMatrixBlock": {
  311. let worldViewProjectionMatrixBlock = new InputBlock("World x View x Projection");
  312. worldViewProjectionMatrixBlock.setAsSystemValue(NodeMaterialSystemValues.WorldViewProjection);
  313. return worldViewProjectionMatrixBlock;
  314. }
  315. case "ViewMatrixBlock": {
  316. let viewMatrixBlock = new InputBlock("View");
  317. viewMatrixBlock.setAsSystemValue(NodeMaterialSystemValues.View);
  318. return viewMatrixBlock;
  319. }
  320. case "ViewProjectionMatrixBlock": {
  321. let viewProjectionMatrixBlock = new InputBlock("View x Projection");
  322. viewProjectionMatrixBlock.setAsSystemValue(NodeMaterialSystemValues.ViewProjection);
  323. return viewProjectionMatrixBlock;
  324. }
  325. case "ProjectionMatrixBlock": {
  326. let projectionMatrixBlock = new InputBlock("Projection");
  327. projectionMatrixBlock.setAsSystemValue(NodeMaterialSystemValues.Projection);
  328. return projectionMatrixBlock;
  329. }
  330. case "CameraPositionBlock": {
  331. let cameraPosition = new InputBlock("Camera position");
  332. cameraPosition.setAsSystemValue(NodeMaterialSystemValues.CameraPosition);
  333. return cameraPosition;
  334. }
  335. case "FogColorBlock": {
  336. let FogColor = new InputBlock("Fog color");
  337. FogColor.setAsSystemValue(NodeMaterialSystemValues.FogColor);
  338. return FogColor;
  339. }
  340. case "PositionBlock": {
  341. let meshPosition = new InputBlock("position");
  342. meshPosition.setAsAttribute("position");
  343. return meshPosition;
  344. }
  345. case "Position2DBlock": {
  346. let meshPosition = new InputBlock("position");
  347. meshPosition.setAsAttribute("position2d");
  348. return meshPosition;
  349. }
  350. case "UVBlock": {
  351. let meshUV = new InputBlock("uv");
  352. meshUV.setAsAttribute("uv");
  353. return meshUV;
  354. }
  355. case "ColorBlock": {
  356. let meshColor = new InputBlock("color");
  357. meshColor.setAsAttribute("color");
  358. return meshColor;
  359. }
  360. case "NormalBlock": {
  361. let meshNormal = new InputBlock("normal");
  362. meshNormal.setAsAttribute("normal");
  363. return meshNormal;
  364. }
  365. case "TangentBlock": {
  366. let meshTangent = new InputBlock("tangent");
  367. meshTangent.setAsAttribute("tangent");
  368. return meshTangent;
  369. }
  370. case "MatrixIndicesBlock": {
  371. let meshMatrixIndices = new InputBlock("matricesIndices");
  372. meshMatrixIndices.setAsAttribute("matricesIndices");
  373. return meshMatrixIndices;
  374. }
  375. case "MatrixWeightsBlock": {
  376. let meshMatrixWeights = new InputBlock("matricesWeights");
  377. meshMatrixWeights.setAsAttribute("matricesWeights");
  378. return meshMatrixWeights;
  379. }
  380. case "TimeBlock": {
  381. let timeBlock = new InputBlock("Time", undefined, NodeMaterialBlockConnectionPointTypes.Float);
  382. timeBlock.animationType = AnimatedInputBlockTypes.Time;
  383. return timeBlock;
  384. }
  385. case "DeltaTimeBlock": {
  386. let deltaTimeBlock = new InputBlock("Delta time");
  387. deltaTimeBlock.setAsSystemValue(NodeMaterialSystemValues.DeltaTime);
  388. return deltaTimeBlock;
  389. }
  390. case "WorldPositionBlock": {
  391. let worldPositionBlock = nodeMaterial.getInputBlockByPredicate((b) => b.isAttribute && b.name === "position");
  392. if (!worldPositionBlock) {
  393. worldPositionBlock = new InputBlock("position");
  394. worldPositionBlock.setAsAttribute("position");
  395. }
  396. let worldMatrixBlock = nodeMaterial.getInputBlockByPredicate((b) => b.isSystemValue && b.systemValue === NodeMaterialSystemValues.World);
  397. if (!worldMatrixBlock) {
  398. worldMatrixBlock = new InputBlock("World");
  399. worldMatrixBlock.setAsSystemValue(NodeMaterialSystemValues.World);
  400. }
  401. let transformBlock = new TransformBlock("World position");
  402. worldPositionBlock.connectTo(transformBlock);
  403. worldMatrixBlock.connectTo(transformBlock);
  404. return transformBlock;
  405. }
  406. case "WorldNormalBlock": {
  407. let worldNormalBlock = nodeMaterial.getInputBlockByPredicate((b) => b.isAttribute && b.name === "normal");
  408. if (!worldNormalBlock) {
  409. worldNormalBlock = new InputBlock("normal");
  410. worldNormalBlock.setAsAttribute("normal");
  411. }
  412. let worldMatrixBlock = nodeMaterial.getInputBlockByPredicate((b) => b.isSystemValue && b.systemValue === NodeMaterialSystemValues.World);
  413. if (!worldMatrixBlock) {
  414. worldMatrixBlock = new InputBlock("World");
  415. worldMatrixBlock.setAsSystemValue(NodeMaterialSystemValues.World);
  416. }
  417. let transformBlock = new TransformBlock("World normal");
  418. worldNormalBlock.connectTo(transformBlock);
  419. worldMatrixBlock.connectTo(transformBlock);
  420. return transformBlock;
  421. }
  422. case "WorldTangentBlock": {
  423. let worldTangentBlock = nodeMaterial.getInputBlockByPredicate((b) => b.isAttribute && b.name === "tangent");
  424. if (!worldTangentBlock) {
  425. worldTangentBlock = new InputBlock("tangent");
  426. worldTangentBlock.setAsAttribute("tangent");
  427. }
  428. let worldMatrixBlock = nodeMaterial.getInputBlockByPredicate((b) => b.isSystemValue && b.systemValue === NodeMaterialSystemValues.World);
  429. if (!worldMatrixBlock) {
  430. worldMatrixBlock = new InputBlock("World");
  431. worldMatrixBlock.setAsSystemValue(NodeMaterialSystemValues.World);
  432. }
  433. let transformBlock = new TransformBlock("World tangent");
  434. worldTangentBlock.connectTo(transformBlock);
  435. worldMatrixBlock.connectTo(transformBlock);
  436. return transformBlock;
  437. }
  438. case "PBRMetallicRoughnessBlock":
  439. return new PBRMetallicRoughnessBlock("PBRMetallicRoughness");
  440. case "SheenBlock":
  441. return new SheenBlock("Sheen");
  442. case "AmbientOcclusionBlock":
  443. return new AmbientOcclusionBlock("AmbientOcclusion");
  444. case "ReflectivityBlock":
  445. return new ReflectivityBlock("Reflectivity");
  446. case "AnisotropyBlock":
  447. return new AnisotropyBlock("Anisotropy");
  448. case "ReflectionBlock":
  449. return new ReflectionBlock("Reflection");
  450. case "ClearCoatBlock":
  451. return new ClearCoatBlock("ClearCoat");
  452. case "RefractionBlock":
  453. return new RefractionBlock("Refraction");
  454. case "SubSurfaceBlock":
  455. return new SubSurfaceBlock("SubSurface");
  456. case "CurrentScreenBlock":
  457. return new CurrentScreenBlock("CurrentScreen");
  458. case "ParticleUVBlock": {
  459. let uv = new InputBlock("uv");
  460. uv.setAsAttribute("particle_uv");
  461. return uv;
  462. }
  463. case "ParticleTextureBlock":
  464. return new ParticleTextureBlock("ParticleTexture");
  465. case "ParticleColorBlock": {
  466. let color = new InputBlock("Color");
  467. color.setAsAttribute("particle_color");
  468. return color;
  469. }
  470. case "ParticleTextureMaskBlock": {
  471. let u = new InputBlock("TextureMask");
  472. u.setAsAttribute("particle_texturemask");
  473. return u;
  474. }
  475. }
  476. return null;
  477. }
  478. public static GetColorFromConnectionNodeType(type: NodeMaterialBlockConnectionPointTypes) {
  479. let color = "#880000";
  480. switch (type) {
  481. case NodeMaterialBlockConnectionPointTypes.Float:
  482. color = "#cb9e27";
  483. break;
  484. case NodeMaterialBlockConnectionPointTypes.Vector2:
  485. color = "#16bcb1";
  486. break;
  487. case NodeMaterialBlockConnectionPointTypes.Vector3:
  488. case NodeMaterialBlockConnectionPointTypes.Color3:
  489. color = "#b786cb";
  490. break;
  491. case NodeMaterialBlockConnectionPointTypes.Vector4:
  492. case NodeMaterialBlockConnectionPointTypes.Color4:
  493. color = "#be5126";
  494. break;
  495. case NodeMaterialBlockConnectionPointTypes.Matrix:
  496. color = "#591990";
  497. break;
  498. }
  499. return color;
  500. }
  501. public static GetConnectionNodeTypeFromString(type: string) {
  502. switch (type) {
  503. case "Float":
  504. return NodeMaterialBlockConnectionPointTypes.Float;
  505. case "Vector2":
  506. return NodeMaterialBlockConnectionPointTypes.Vector2;
  507. case "Vector3":
  508. return NodeMaterialBlockConnectionPointTypes.Vector3;
  509. case "Vector4":
  510. return NodeMaterialBlockConnectionPointTypes.Vector4;
  511. case "Matrix":
  512. return NodeMaterialBlockConnectionPointTypes.Matrix;
  513. case "Color3":
  514. return NodeMaterialBlockConnectionPointTypes.Color3;
  515. case "Color4":
  516. return NodeMaterialBlockConnectionPointTypes.Color4;
  517. }
  518. return NodeMaterialBlockConnectionPointTypes.AutoDetect;
  519. }
  520. public static GetStringFromConnectionNodeType(type: NodeMaterialBlockConnectionPointTypes) {
  521. switch (type){
  522. case NodeMaterialBlockConnectionPointTypes.Float:
  523. return "Float";
  524. case NodeMaterialBlockConnectionPointTypes.Vector2:
  525. return "Vector2";
  526. case NodeMaterialBlockConnectionPointTypes.Vector3:
  527. return "Vector3";
  528. case NodeMaterialBlockConnectionPointTypes.Vector4:
  529. return "Vector4";
  530. case NodeMaterialBlockConnectionPointTypes.Color3:
  531. return "Color3";
  532. case NodeMaterialBlockConnectionPointTypes.Color4:
  533. return "Color4";
  534. case NodeMaterialBlockConnectionPointTypes.Matrix:
  535. return "Matrix";
  536. }
  537. return "";
  538. }
  539. }