@needle-tools/three 0.145.4 → 0.146.2

package/examples/js/loaders/MaterialXLoader.js

@@ -0,0 +1,392 @@
+( function () {
+const colorSpaceLib = {
+  THREE.mx_srgb_texture_to_lin_rec709
+};
+class MtlXElement {
+  constructor(name, nodeFunc, params = null) {
+    this.name = name;
+    this.nodeFunc = nodeFunc;
+    this.params = params;
+  }
+}
+
+// Ref: https://github.com/mrdoob/three.js/issues/24674
+
+const MtlXElements = [
+// << Math >>
+new MtlXElement('add', THREE.add, ['in1', 'in2']), new MtlXElement('subtract', THREE.sub, ['in1', 'in2']), new MtlXElement('multiply', THREE.mul, ['in1', 'in2']), new MtlXElement('divide', THREE.div, ['in1', 'in2']), new MtlXElement('modulo', THREE.mod, ['in1', 'in2']), new MtlXElement('absval', THREE.abs, ['in1', 'in2']), new MtlXElement('sign', THREE.sign, ['in1', 'in2']), new MtlXElement('floor', THREE.floor, ['in1', 'in2']), new MtlXElement('ceil', THREE.ceil, ['in1', 'in2']), new MtlXElement('round', THREE.round, ['in1', 'in2']), new MtlXElement('power', THREE.pow, ['in1', 'in2']), new MtlXElement('sin', THREE.sin, ['in']), new MtlXElement('cos', THREE.cos, ['in']), new MtlXElement('tan', THREE.tan, ['in']), new MtlXElement('asin', THREE.asin, ['in']), new MtlXElement('acos', THREE.acos, ['in']), new MtlXElement('atan2', THREE.atan2, ['in1', 'in2']), new MtlXElement('sqrt', THREE.sqrt, ['in']),
+//new MtlXElement( 'ln', ... ),
+new MtlXElement('exp', THREE.exp, ['in']), new MtlXElement('clamp', THREE.clamp, ['in', 'low', 'high']), new MtlXElement('min', THREE.min, ['in1', 'in2']), new MtlXElement('max', THREE.max, ['in1', 'in2']), new MtlXElement('normalize', THREE.normalize, ['in']), new MtlXElement('magnitude', THREE.length, ['in1', 'in2']), new MtlXElement('dotproduct', THREE.dot, ['in1', 'in2']), new MtlXElement('crossproduct', THREE.cross, ['in']),
+//new MtlXElement( 'transformpoint', ... ),
+//new MtlXElement( 'transformvector', ... ),
+//new MtlXElement( 'transformnormal', ... ),
+//new MtlXElement( 'transformmatrix', ... ),
+new MtlXElement('normalmap', THREE.normalMap, ['in', 'scale']),
+//new MtlXElement( 'transpose', ... ),
+//new MtlXElement( 'determinant', ... ),
+//new MtlXElement( 'invertmatrix', ... ),
+//new MtlXElement( 'rotate2d', rotateUV, [ 'in', radians( 'amount' )** ] ),
+//new MtlXElement( 'rotate3d', ... ),
+//new MtlXElement( 'arrayappend', ... ),
+//new MtlXElement( 'dot', ... ),
+
+// << Adjustment >>
+new MtlXElement('remap', THREE.remap, ['in', 'inlow', 'inhigh', 'outlow', 'outhigh']), new MtlXElement('smoothstep', THREE.smoothstep, ['in', 'low', 'high']),
+//new MtlXElement( 'curveadjust', ... ),
+//new MtlXElement( 'curvelookup', ... ),
+new MtlXElement('luminance', THREE.luminance, ['in', 'lumacoeffs']), new MtlXElement('rgbtohsv', THREE.mx_rgbtohsv, ['in']), new MtlXElement('hsvtorgb', THREE.mx_hsvtorgb, ['in']),
+// << Mix >>
+new MtlXElement('mix', THREE.mix, ['bg', 'fg', 'mix']),
+// << Channel >>
+new MtlXElement('combine2', THREE.vec2, ['in1', 'in2']), new MtlXElement('combine3', THREE.vec3, ['in1', 'in2', 'in3']), new MtlXElement('combine4', THREE.vec4, ['in1', 'in2', 'in3', 'in4']),
+// << Procedural >>
+new MtlXElement('ramplr', THREE.mx_ramplr, ['valuel', 'valuer', 'texcoord']), new MtlXElement('ramptb', THREE.mx_ramptb, ['valuet', 'valueb', 'texcoord']), new MtlXElement('splitlr', THREE.mx_splitlr, ['valuel', 'valuer', 'texcoord']), new MtlXElement('splittb', THREE.mx_splittb, ['valuet', 'valueb', 'texcoord']), new MtlXElement('noise2d', THREE.mx_noise_float, ['texcoord', 'amplitude', 'pivot']), new MtlXElement('noise3d', THREE.mx_noise_float, ['texcoord', 'amplitude', 'pivot']), new MtlXElement('fractal3d', THREE.mx_fractal_noise_float, ['position', 'octaves', 'lacunarity', 'diminish', 'amplitude']), new MtlXElement('cellnoise2d', THREE.mx_cell_noise_float, ['texcoord']), new MtlXElement('cellnoise3d', THREE.mx_cell_noise_float, ['texcoord']), new MtlXElement('worleynoise2d', THREE.mx_worley_noise_float, ['texcoord', 'jitter']), new MtlXElement('worleynoise3d', THREE.mx_worley_noise_float, ['texcoord', 'jitter']),
+// << Supplemental >>
+//new MtlXElement( 'tiledimage', ... ),
+//new MtlXElement( 'triplanarprojection', triplanarTextures, [ 'filex', 'filey', 'filez' ] ),
+//new MtlXElement( 'ramp4', ... ),
+//new MtlXElement( 'place2d', mx_place2d, [ 'texcoord', 'pivot', 'scale', 'rotate', 'offset' ] ),
+new MtlXElement('safepower', THREE.mx_safepower, ['in1', 'in2']), new MtlXElement('contrast', THREE.mx_contrast, ['in', 'amount', 'pivot']),
+//new MtlXElement( 'hsvadjust', ... ),
+new MtlXElement('saturate', THREE.saturation, ['in', 'amount'])
+//new MtlXElement( 'extract', ... ),
+//new MtlXElement( 'separate2', ... ),
+//new MtlXElement( 'separate3', ... ),
+//new MtlXElement( 'separate4', ... )
+];
+
+const MtlXLibrary = {};
+MtlXElements.forEach(element => MtlXLibrary[element.name] = element);
+class MaterialXLoader extends THREE.Loader {
+  constructor(manager) {
+    super(manager);
+  }
+  load(url, onLoad, onProgress, onError) {
+    new THREE.FileLoader(this.manager).setPath(this.path).load(url, async text => {
+      try {
+        onLoad(this.parse(text));
+      } catch (e) {
+        onError(e);
+      }
+    }, onProgress, onError);
+    return this;
+  }
+  parse(text) {
+    return new MaterialX(this.manager, this.path).parse(text);
+  }
+}
+class MaterialXNode {
+  constructor(materialX, nodeXML, nodePath = '') {
+    this.materialX = materialX;
+    this.nodeXML = nodeXML;
+    this.nodePath = nodePath ? nodePath + '/' + this.name : this.name;
+    this.parent = null;
+    this.node = null;
+    this.children = [];
+  }
+  get element() {
+    return this.nodeXML.nodeName;
+  }
+  get nodeGraph() {
+    return this.getAttribute('nodegraph');
+  }
+  get nodeName() {
+    return this.getAttribute('nodename');
+  }
+  get interfaceName() {
+    return this.getAttribute('interfacename');
+  }
+  get output() {
+    return this.getAttribute('output');
+  }
+  get name() {
+    return this.getAttribute('name');
+  }
+  get type() {
+    return this.getAttribute('type');
+  }
+  get value() {
+    return this.getAttribute('value');
+  }
+  getNodeGraph() {
+    let nodeX = this;
+    while (nodeX !== null) {
+      if (nodeX.element === 'nodegraph') {
+        break;
+      }
+      nodeX = nodeX.parent;
+    }
+    return nodeX;
+  }
+  getRoot() {
+    let nodeX = this;
+    while (nodeX.parent !== null) {
+      nodeX = nodeX.parent;
+    }
+    return nodeX;
+  }
+  get referencePath() {
+    let referencePath = null;
+    if (this.nodeGraph !== null && this.output !== null) {
+      referencePath = this.nodeGraph + '/' + this.output;
+    } else if (this.nodeName !== null || this.interfaceName !== null) {
+      referencePath = this.getNodeGraph().nodePath + '/' + (this.nodeName || this.interfaceName);
+    }
+    return referencePath;
+  }
+  get hasReference() {
+    return this.referencePath !== null;
+  }
+  get isConst() {
+    return this.element === 'input' && this.value !== null && this.type !== 'filename';
+  }
+  getColorSpaceNode() {
+    const csSource = this.getAttribute('colorspace');
+    const csTarget = this.getRoot().getAttribute('colorspace');
+    const nodeName = `mx_${csSource}_to_${csTarget}`;
+    return colorSpaceLib[nodeName];
+  }
+  getTexture() {
+    const filePrefix = this.getRecursiveAttribute('fileprefix') || '';
+    const THREE.texture = this.materialX.textureLoader.load(filePrefix + this.value);
+    THREE.texture.wrapS = THREE.texture.wrapT = THREE.RepeatWrapping;
+    THREE.texture.flipY = false;
+    return THREE.texture;
+  }
+  getClassFromType(type) {
+    let nodeClass = null;
+    if (type === 'integer') nodeClass = THREE.int;else if (type === 'float') nodeClass = THREE.float;else if (type === 'vector2') nodeClass = THREE.vec2;else if (type === 'vector3') nodeClass = THREE.vec3;else if (type === 'vector4' || type === 'color4') nodeClass = THREE.vec4;else if (type === 'color3') nodeClass = THREE.color;else if (type === 'boolean') nodeClass = THREE.bool;
+    return nodeClass;
+  }
+  getNode() {
+    let node = this.node;
+    if (node !== null) {
+      return node;
+    }
+
+    //
+
+    const type = this.type;
+    if (this.isConst) {
+      const nodeClass = this.getClassFromType(type);
+      node = nodeClass(...this.getVector());
+    } else if (this.hasReference) {
+      node = this.materialX.getMaterialXNode(this.referencePath).getNode();
+    } else {
+      const element = this.element;
+      if (element === 'convert') {
+        const nodeClass = this.getClassFromType(type);
+        node = nodeClass(this.getNodeByName('in'));
+      } else if (element === 'constant') {
+        node = this.getNodeByName('value');
+      } else if (element === 'position') {
+        node = THREE.positionLocal;
+      } else if (element === 'tiledimage') {
+        const file = this.getChildByName('file');
+        const textureFile = file.getTexture();
+        const uvTiling = THREE.mx_transform_uv(...this.getNodesByNames(['uvtiling', 'uvoffset']));
+        node = THREE.texture(textureFile, uvTiling);
+        const colorSpaceNode = file.getColorSpaceNode();
+        if (colorSpaceNode) {
+          node = colorSpaceNode(node);
+        }
+      } else if (element === 'image') {
+        const file = this.getChildByName('file');
+        const uvNode = this.getNodeByName('texcoord');
+        const textureFile = file.getTexture();
+        node = THREE.texture(textureFile, uvNode);
+        const colorSpaceNode = file.getColorSpaceNode();
+        if (colorSpaceNode) {
+          node = colorSpaceNode(node);
+        }
+      } else if (MtlXLibrary[element] !== undefined) {
+        const nodeElement = MtlXLibrary[element];
+        node = nodeElement.nodeFunc(...this.getNodesByNames(...nodeElement.params));
+      }
+    }
+
+    //
+
+    if (node === null) {
+      console.warn(`THREE.MaterialXLoader: Unexpected node ${new XMLSerializer().serializeToString(this.nodeXML)}.`);
+      node = THREE.float(0);
+    }
+
+    //
+
+    const nodeToTypeClass = this.getClassFromType(type);
+    if (nodeToTypeClass !== null) {
+      node = nodeToTypeClass(node);
+    }
+    node.name = this.name;
+    this.node = node;
+    return node;
+  }
+  getChildByName(name) {
+    for (const input of this.children) {
+      if (input.name === name) {
+        return input;
+      }
+    }
+  }
+  getNodes() {
+    const nodes = {};
+    for (const input of this.children) {
+      const node = input.getNode();
+      nodes[node.name] = node;
+    }
+    return nodes;
+  }
+  getNodeByName(name) {
+    return this.getChildByName(name)?.getNode();
+  }
+  getNodesByNames(...names) {
+    const nodes = [];
+    for (const name of names) {
+      const node = this.getNodeByName(name);
+      if (node) nodes.push(node);
+    }
+    return nodes;
+  }
+  getValue() {
+    return this.value.trim();
+  }
+  getVector() {
+    const vector = [];
+    for (const val of this.getValue().split(/[,|\s]/)) {
+      if (val !== '') {
+        vector.push(Number(val.trim()));
+      }
+    }
+    return vector;
+  }
+  getAttribute(name) {
+    return this.nodeXML.getAttribute(name);
+  }
+  getRecursiveAttribute(name) {
+    let attribute = this.nodeXML.getAttribute(name);
+    if (attribute === null && this.parent !== null) {
+      attribute = this.parent.getRecursiveAttribute(name);
+    }
+    return attribute;
+  }
+  setStandardSurfaceToGltfPBR(material) {
+    const inputs = this.getNodes();
+
+    //
+
+    let colorNode = null;
+    if (inputs.base && inputs.base_color) colorNode = THREE.mul(inputs.base, inputs.base_color);else if (inputs.base) colorNode = inputs.base;else if (inputs.base_color) colorNode = inputs.base_color;
+
+    //
+
+    let roughnessNode = null;
+    if (inputs.specular_roughness) roughnessNode = inputs.specular_roughness;
+
+    //
+
+    let metalnessNode = null;
+    if (inputs.metalness) metalnessNode = inputs.metalness;
+
+    //
+
+    let clearcoatNode = null;
+    let clearcoatRoughnessNode = null;
+    if (inputs.coat) clearcoatNode = inputs.coat;
+    if (inputs.coat_roughness) clearcoatRoughnessNode = inputs.coat_roughness;
+    if (inputs.coat_color) {
+      colorNode = colorNode ? THREE.mul(colorNode, inputs.coat_color) : colorNode;
+    }
+
+    //
+
+    material.colorNode = colorNode || THREE.color(0.8, 0.8, 0.8);
+    material.roughnessNode = roughnessNode || THREE.float(0.2);
+    material.metalnessNode = metalnessNode || THREE.float(0);
+    material.clearcoatNode = clearcoatNode || THREE.float(0);
+    material.clearcoatRoughnessNode = clearcoatRoughnessNode || THREE.float(0);
+  }
+
+  /*setGltfPBR( material ) {
+  		const inputs = this.getNodes();
+  		console.log( inputs );
+  	}*/
+
+  setMaterial(material) {
+    const element = this.element;
+    if (element === 'gltf_pbr') {
+
+      //this.setGltfPBR( material );
+    } else if (element === 'standard_surface') {
+      this.setStandardSurfaceToGltfPBR(material);
+    }
+  }
+  toMaterial() {
+    const material = new THREE.MeshPhysicalNodeMaterial();
+    material.name = this.name;
+    for (const nodeX of this.children) {
+      const shaderProperties = this.materialX.getMaterialXNode(nodeX.nodeName);
+      shaderProperties.setMaterial(material);
+    }
+    return material;
+  }
+  toMaterials() {
+    const materials = {};
+    for (const nodeX of this.children) {
+      if (nodeX.element === 'surfacematerial') {
+        const material = nodeX.toMaterial();
+        materials[material.name] = material;
+      }
+    }
+    return materials;
+  }
+  THREE.add(materialXNode) {
+    materialXNode.parent = this;
+    this.children.push(materialXNode);
+  }
+}
+class MaterialX {
+  constructor(manager, path) {
+    this.manager = manager;
+    this.path = path;
+    this.resourcePath = '';
+    this.nodesXLib = new Map();
+    //this.nodesXRefLib = new WeakMap();
+
+    this.textureLoader = new THREE.TextureLoader(manager);
+  }
+  addMaterialXNode(materialXNode) {
+    this.nodesXLib.set(materialXNode.nodePath, materialXNode);
+  }
+
+  /*getMaterialXNodeFromXML( xmlNode ) {
+       return this.nodesXRefLib.get( xmlNode );
+   }*/
+
+  getMaterialXNode(...names) {
+    return this.nodesXLib.get(names.join('/'));
+  }
+  parseNode(nodeXML, nodePath = '') {
+    const materialXNode = new MaterialXNode(this, nodeXML, nodePath);
+    if (materialXNode.nodePath) this.addMaterialXNode(materialXNode);
+    for (const childNodeXML of nodeXML.children) {
+      const childMXNode = this.parseNode(childNodeXML, materialXNode.nodePath);
+      materialXNode.add(childMXNode);
+    }
+    return materialXNode;
+  }
+  parse(text) {
+    const rootXML = new DOMParser().parseFromString(text, 'application/xml').documentElement;
+    this.textureLoader.setPath(this.path);
+
+    //
+
+    const materials = this.parseNode(rootXML).toMaterials();
+    return {
+      materials
+    };
+  }
+}
+
+THREE.MaterialXLoader = MaterialXLoader;
+} )();