threejs/editor/js/three/loaders/MaterialXLoader.js

( 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;
} )();