You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

392 lines
14 KiB

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