blockTools.ts 28 KB

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