blockTools.ts 28 KB

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