obj2gltf/lib/loadObj.js

753 lines
27 KiB
JavaScript
Raw Normal View History

2017-03-13 15:28:51 -04:00
'use strict';
2016-07-22 14:09:13 -04:00
var Cesium = require('cesium');
2015-10-16 17:32:23 -04:00
var path = require('path');
2017-03-13 15:28:51 -04:00
var Promise = require('bluebird');
2016-07-22 14:09:13 -04:00
2017-03-13 15:28:51 -04:00
var ArrayStorage = require('./ArrayStorage');
2017-04-12 16:55:03 -04:00
var loadImage = require('./loadImage');
var loadMtl = require('./loadMtl');
2017-03-13 15:28:51 -04:00
var readLines = require('./readLines');
2016-07-22 14:09:13 -04:00
2017-04-20 14:41:39 -04:00
var Axis = Cesium.Axis;
var Cartesian2 = Cesium.Cartesian2;
2017-04-20 14:41:39 -04:00
var Cartesian3 = Cesium.Cartesian3;
2017-03-13 15:28:51 -04:00
var ComponentDatatype = Cesium.ComponentDatatype;
var defaultValue = Cesium.defaultValue;
2016-06-09 13:33:08 -04:00
var defined = Cesium.defined;
var IntersectionTests = Cesium.IntersectionTests;
var Matrix3 = Cesium.Matrix3;
2017-04-20 14:41:39 -04:00
var Matrix4 = Cesium.Matrix4;
var OrientedBoundingBox = Cesium.OrientedBoundingBox;
var Plane = Cesium.Plane;
var PolygonPipeline = Cesium.PolygonPipeline;
var Ray = Cesium.Ray;
2017-03-13 15:28:51 -04:00
var RuntimeError = Cesium.RuntimeError;
var WindingOrder = Cesium.WindingOrder;
2015-10-16 17:32:23 -04:00
2017-03-13 15:28:51 -04:00
module.exports = loadObj;
2015-10-16 17:32:23 -04:00
2017-03-13 15:28:51 -04:00
// Object name (o) -> node
// Group name (g) -> mesh
// Material name (usemtl) -> primitive
2015-10-16 17:32:23 -04:00
2017-03-13 15:28:51 -04:00
function Node() {
this.name = undefined;
this.meshes = [];
2015-10-16 17:32:23 -04:00
}
2017-03-13 15:28:51 -04:00
function Mesh() {
this.name = undefined;
this.primitives = [];
this.positions = new ArrayStorage(ComponentDatatype.FLOAT);
this.normals = new ArrayStorage(ComponentDatatype.FLOAT);
this.uvs = new ArrayStorage(ComponentDatatype.FLOAT);
}
function Primitive() {
this.material = undefined;
this.indices = new ArrayStorage(ComponentDatatype.UNSIGNED_INT);
}
2017-03-13 15:28:51 -04:00
// OBJ regex patterns are modified from ThreeJS (https://github.com/mrdoob/three.js/blob/master/examples/js/loaders/OBJLoader.js)
var vertexPattern = /v( +[\d|\.|\+|\-|e|E]+)( +[\d|\.|\+|\-|e|E]+)( +[\d|\.|\+|\-|e|E]+)/; // v float float float
var normalPattern = /vn( +[\d|\.|\+|\-|e|E]+)( +[\d|\.|\+|\-|e|E]+)( +[\d|\.|\+|\-|e|E]+)/; // vn float float float
var uvPattern = /vt( +[\d|\.|\+|\-|e|E]+)( +[\d|\.|\+|\-|e|E]+)/; // vt float float
var facePattern1 = /f(\s+-?\d+\/?\/?){3,}/; // f vertex vertex vertex ...
var facePattern2 = /f(\s+-?\d+\/-?\d+){3,}/; // f vertex/uv vertex/uv vertex/uv ...
var facePattern3 = /f(\s+-?\d+\/-?\d+\/-?\d+){3,}/; // f vertex/uv/normal vertex/uv/normal vertex/uv/normal ...
var facePattern4 = /f(\s+-?\d+\/\/-?\d+){3,}/; // f vertex//normal vertex//normal vertex//normal ...
2017-03-13 15:28:51 -04:00
var faceSpacePattern = /(f?\s+)|(\s+\/)|(\s*\\)/g;
var faceSpaceOrSlashPattern = /(f?\s+)|(\/+\s*)|(\s+\/)|(\s*\\)/g;
2017-04-20 14:41:39 -04:00
var scratchCartesian = new Cartesian3();
2017-03-13 15:28:51 -04:00
/**
* Parse an obj file.
*
* @param {String} objPath Path to the obj file.
2017-04-10 17:57:56 -04:00
* @param {Object} options An object with the following properties:
* @param {Boolean} options.checkTransparency Do a more exhaustive check for texture transparency by looking at the alpha channel of each pixel.
* @param {Boolean} options.secure Prevent the converter from reading image or mtl files outside of the input obj directory.
2017-04-20 14:41:39 -04:00
* @param {String} options.inputUpAxis Up axis of the obj.
* @param {String} options.outputUpAxis Up axis of the converted glTF.
2017-04-10 17:57:56 -04:00
* @param {Boolean} options.logger A callback function for handling logged messages. Defaults to console.log.
2017-03-13 15:28:51 -04:00
* @returns {Promise} A promise resolving to the obj data.
* @exception {RuntimeError} The file does not have any geometry information in it.
*
* @private
*/
2017-03-17 16:05:51 -04:00
function loadObj(objPath, options) {
2017-04-20 14:41:39 -04:00
var axisTransform = getAxisTransform(options.inputUpAxis, options.outputUpAxis);
2017-03-13 15:28:51 -04:00
// Global store of vertex attributes listed in the obj file
var positions = new ArrayStorage(ComponentDatatype.FLOAT);
var normals = new ArrayStorage(ComponentDatatype.FLOAT);
var uvs = new ArrayStorage(ComponentDatatype.FLOAT);
// The current node, mesh, and primitive
var node;
var mesh;
var primitive;
// All nodes seen in the obj
var nodes = [];
// Used to build the indices. The vertex cache is unique to each mesh.
var vertexCache = {};
var vertexCacheLimit = 1000000;
var vertexCacheCount = 0;
var vertexCount = 0;
// All mtl paths seen in the obj
var mtlPaths = [];
// Buffers for face data that spans multiple lines
2017-06-26 12:23:34 -04:00
var lineBuffer = "";
2017-03-13 15:28:51 -04:00
function getName(name) {
return (name === '' ? undefined : name);
}
function addNode(name) {
node = new Node();
node.name = getName(name);
nodes.push(node);
addMesh();
}
function addMesh(name) {
mesh = new Mesh();
mesh.name = getName(name);
node.meshes.push(mesh);
addPrimitive();
// Clear the vertex cache for each new mesh
vertexCache = {};
vertexCacheCount = 0;
vertexCount = 0;
}
function addPrimitive() {
primitive = new Primitive();
mesh.primitives.push(primitive);
}
function useMaterial(name) {
// Look to see if this material has already been used by a primitive in the mesh
var material = getName(name);
var primitives = mesh.primitives;
var primitivesLength = primitives.length;
for (var i = 0; i < primitivesLength; ++i) {
2017-03-17 11:40:54 -04:00
if (primitives[i].material === material) {
primitive = primitives[i];
2017-03-13 15:28:51 -04:00
return;
2016-07-22 16:17:27 -04:00
}
}
2017-03-13 15:28:51 -04:00
// Add a new primitive with this material
addPrimitive();
primitive.material = getName(name);
}
2016-07-22 16:17:27 -04:00
var intPoint = new Cartesian3();
var xAxis = Cesium.Cartesian3.UNIT_X.clone();
var yAxis = Cesium.Cartesian3.UNIT_Y.clone();
var zAxis = Cesium.Cartesian3.UNIT_Z.clone();
var origin = new Cartesian3();
var normal = new Cartesian3();
var ray = new Ray();
var plane = new Plane(Cesium.Cartesian3.UNIT_X, 0);
function projectTo2D(positions) {
var positions2D = new Array(positions.length);
var obb = OrientedBoundingBox.fromPoints(positions);
var halfAxes = obb.halfAxes;
Matrix3.getColumn(halfAxes, 0, xAxis);
Matrix3.getColumn(halfAxes, 1, yAxis);
Matrix3.getColumn(halfAxes, 2, zAxis);
var xMag = Cartesian3.magnitude(xAxis);
var yMag = Cartesian3.magnitude(yAxis);
var zMag = Cartesian3.magnitude(zAxis);
var min = Math.min(xMag, yMag, zMag);
// If all the points are on a line, just remove one of the zero dimensions
if (xMag === 0 && (yMag === 0 || zMag === 0)) {
var i;
for (i = 0; i < positions.length; i++) {
positions2D[i] = new Cartesian2(positions[i].y, positions[i].z);
}
return positions2D;
} else if (yMag === 0 && zMag === 0) {
for (i = 0; i < positions.length; i++) {
positions2D[i] = new Cartesian2(positions[i].x, positions[i].y);
}
return positions2D;
}
var center = obb.center;
var planeXAxis;
var planeYAxis;
if (min === xMag) {
if (!xAxis.equals(Cartesian3.ZERO)) {
Cartesian3.add(center, xAxis, origin);
Cartesian3.normalize(xAxis, normal);
}
planeXAxis = Cartesian3.normalize(yAxis, yAxis);
planeYAxis = Cartesian3.normalize(zAxis, zAxis);
} else if (min === yMag) {
if (!yAxis.equals(Cartesian3.ZERO)) {
Cartesian3.add(center, yAxis, origin);
Cartesian3.normalize(yAxis, normal);
}
planeXAxis = Cartesian3.normalize(xAxis, xAxis);
planeYAxis = Cartesian3.normalize(zAxis, zAxis);
} else {
if (!zAxis.equals(Cartesian3.ZERO)) {
Cartesian3.add(center, zAxis, origin);
Cartesian3.normalize(zAxis, normal);
}
planeXAxis = Cartesian3.normalize(xAxis, xAxis);
planeYAxis = Cartesian3.normalize(yAxis, yAxis);
}
if (min === 0) {
normal = Cartesian3.cross(planeXAxis, planeYAxis, normal);
normal = Cartesian3.normalize(normal, normal);
}
Plane.fromPointNormal(origin, normal, plane);
ray.direction = normal;
for (i = 0; i < positions.length; i++) {
ray.origin = positions[i];
var intersectionPoint = IntersectionTests.rayPlane(ray, plane, intPoint);
if (!defined(intersectionPoint)) {
Cartesian3.negate(ray.direction, ray.direction);
intersectionPoint = IntersectionTests.rayPlane(ray, plane, intPoint);
}
var v = Cartesian3.subtract(intersectionPoint, origin, intersectionPoint);
var x = Cartesian3.dot(planeXAxis, v);
var y = Cartesian3.dot(planeYAxis, v);
positions2D[i] = new Cartesian2(x, y);
}
return positions2D;
}
2017-03-13 15:28:51 -04:00
function getOffset(a, attributeData, components) {
var i = parseInt(a);
if (i < 0) {
// Negative vertex indexes reference the vertices immediately above it
return (attributeData.length / components + i) * components;
2016-06-09 13:33:08 -04:00
}
2017-03-13 15:28:51 -04:00
return (i - 1) * components;
}
2015-10-16 17:32:23 -04:00
2017-03-13 15:28:51 -04:00
function createVertex(p, u, n) {
// Positions
if (defined(p)) {
var pi = getOffset(p, positions, 3);
var px = positions.get(pi + 0);
var py = positions.get(pi + 1);
var pz = positions.get(pi + 2);
mesh.positions.push(px);
mesh.positions.push(py);
mesh.positions.push(pz);
2016-06-09 13:33:08 -04:00
}
2016-07-22 16:17:27 -04:00
2017-03-13 15:28:51 -04:00
// Normals
if (defined(n)) {
var ni = getOffset(n, normals, 3);
var nx = normals.get(ni + 0);
var ny = normals.get(ni + 1);
var nz = normals.get(ni + 2);
mesh.normals.push(nx);
mesh.normals.push(ny);
mesh.normals.push(nz);
}
2016-07-22 16:17:27 -04:00
2017-03-13 15:28:51 -04:00
// UVs
if (defined(u)) {
var ui = getOffset(u, uvs, 2);
var ux = uvs.get(ui + 0);
var uy = uvs.get(ui + 1);
mesh.uvs.push(ux);
mesh.uvs.push(uy);
2016-06-09 13:33:08 -04:00
}
2017-03-13 15:28:51 -04:00
}
2015-10-16 17:32:23 -04:00
2017-03-13 15:28:51 -04:00
function addVertex(v, p, u, n) {
var index = vertexCache[v];
if (!defined(index)) {
index = vertexCount++;
vertexCache[v] = index;
createVertex(p, u, n);
// Prevent the vertex cache from growing too large. As a result of clearing the cache there
// may be some duplicate vertices.
vertexCacheCount++;
if (vertexCacheCount > vertexCacheLimit) {
vertexCacheCount = 0;
vertexCache = {};
2016-07-22 16:17:27 -04:00
}
2017-03-13 15:28:51 -04:00
}
return index;
}
function get3DPoint(index, result) {
var pi = getOffset(index, positions, 3);
var px = positions.get(pi + 0);
var py = positions.get(pi + 1);
var pz = positions.get(pi + 2);
return Cartesian3.fromElements(px, py, pz, result);
}
function get3DNormal(index, result) {
var ni = getOffset(index, normals, 3);
var nx = normals.get(ni + 0);
var ny = normals.get(ni + 1);
var nz = normals.get(ni + 2);
return Cartesian3.fromElements(nx, ny, nz, result);
}
// Given a sequence of three points A B C, determine whether vector BC
// "turns" clockwise (positive) or counter-clockwise (negative) from vector AB
var scratch1 = new Cartesian3();
var scratch2 = new Cartesian3();
function getTurnDirection(pointA, pointB, pointC) {
var vector1 = Cartesian2.subtract(pointA, pointB, scratch1);
var vector2 = Cartesian2.subtract(pointC, pointB, scratch2);
return vector1.x * vector2.y - vector1.y * vector2.x;
}
// Given the cartesian 2 vertices of a polygon, determine if convex
function isConvex(positions2D) {
var i;
var turnDirection = getTurnDirection(positions2D[0], positions2D[1], positions2D[2]);
for (i=1; i < positions2D.length-2; ++i) {
var currentTurnDirection = getTurnDirection(positions2D[i], positions2D[i+1], positions2D[i+2]);
if (turnDirection * currentTurnDirection < 0) {
return false;
}
}
return true;
}
var scratch3 = new Cartesian3();
// Checks if winding order matches the given normal.
2017-06-26 12:47:08 -04:00
function checkWindingCorrect(positionIndex1, positionIndex2, positionIndex3, normal) {
var A = get3DPoint(positionIndex1, scratch1);
var B = get3DPoint(positionIndex2, scratch2);
var C = get3DPoint(positionIndex3, scratch3);
2017-06-26 12:47:08 -04:00
Cartesian3.subtract(B, A, B);
Cartesian3.subtract(C, A, C);
var cross = Cartesian3.cross(A, C, scratch1);
return (Cartesian3.dot(normal, cross) >= 0);
}
function addTriangle(index1, index2, index3, correctWinding) {
if (correctWinding) {
primitive.indices.push(index1);
primitive.indices.push(index2);
primitive.indices.push(index3);
} else {
primitive.indices.push(index1);
primitive.indices.push(index3);
primitive.indices.push(index2);
}
}
var scratchNormal = new Cartesian3();
2017-06-13 20:38:38 -04:00
function addFace(vertices, positions, uvs, normals) {
var u1, u2, u3, n1, n2, n3;
2017-06-26 12:47:08 -04:00
var isWindingCorrect = true;
var faceNormal;
// If normals are defined, find a face normal to use in winding order sanitization.
// If no face normal, we have to assume the winding is correct.
if (normals) {
faceNormal = get3DNormal(normals[0], scratchNormal);
2017-06-26 12:47:08 -04:00
isWindingCorrect = checkWindingCorrect(positions[0], positions[1], positions[2], faceNormal);
}
2017-06-13 20:38:38 -04:00
if (vertices.length === 3) {
2017-06-13 20:38:38 -04:00
if (uvs) {
u1 = uvs[0];
u2 = uvs[1];
u3 = uvs[2];
}
2017-03-13 15:28:51 -04:00
if (normals) {
n1 = normals[0];
n2 = normals[1];
n3 = normals[2];
}
var index1 = addVertex(vertices[0], positions[0], u1, n1);
var index2 = addVertex(vertices[1], positions[1], u2, n2);
var index3 = addVertex(vertices[2], positions[2], u3, n3);
2017-03-13 15:28:51 -04:00
2017-06-26 12:47:08 -04:00
addTriangle(index1, index2, index3, isWindingCorrect);
} else { // Triangulate if the face is not a triangle
var positions3D = [];
var vertexIndices = [];
var i;
for (i = 0; i < vertices.length; ++i) {
var u = (defined(uvs)) ? uvs[i] : undefined;
var n = (defined(normals)) ? normals[i] : undefined;
var index = addVertex(vertices[i], positions[i], u, n);
vertexIndices.push(index);
// Collect the vertex positions as 3D points
positions3D.push(get3DPoint(positions[i], new Cartesian3()));
}
var positions2D = projectTo2D(positions3D);
if (isConvex(positions2D)) {
for (i=1; i < vertices.length-1; ++i) {
2017-06-26 12:47:08 -04:00
addTriangle(vertexIndices[0], vertexIndices[i], vertexIndices[i+1], isWindingCorrect);
}
} else {
// Since the projection doesn't preserve winding order, reverse the order of
// the vertices before triangulating to enforce counter clockwise.
var projectedWindingOrder = PolygonPipeline.computeWindingOrder2D(positions2D);
if (projectedWindingOrder === WindingOrder.CLOCKWISE) {
positions2D.reverse();
}
// Use an ear-clipping algorithm to triangulate
var positionIndices = PolygonPipeline.triangulate(positions2D);
for (i = 0; i < positionIndices.length-2; i += 3) {
2017-06-26 12:47:08 -04:00
addTriangle(vertexIndices[positionIndices[i]], vertexIndices[positionIndices[i+1]], vertexIndices[positionIndices[i+2]], isWindingCorrect);
}
}
2016-07-22 16:17:27 -04:00
}
}
2017-06-13 20:38:38 -04:00
function trimEmpty(entry) { return entry.trim() !== ''; }
// Parse a line of face data.
// Because face data can span multiple lines, attribute data is buffered in parallel lists
// faceVertices : names of a vertices for caching
// facePosition : indices into position array
// faceUvs : indices into uv array
// faceNormals : indices into normal array
function parseFaceLine(line, hasUvs, hasNormals) {
// get vertex data (attributes '/' separated)
2017-06-26 12:23:34 -04:00
var faceVertices = line.replace(faceSpacePattern, ' ').split(' ').filter(trimEmpty);
// get all vertex attributes for this line
var faceAttributes = line.replace(faceSpaceOrSlashPattern, ' ').split(' ').filter(trimEmpty);
2017-06-26 12:23:34 -04:00
var facePositions = [];
var faceUvs = [];
var faceNormals = [];
var i;
if (hasUvs && hasNormals) {
for (i=0; i <= faceAttributes.length - 3; i += 3)
{
facePositions.push(faceAttributes[i]);
faceUvs.push(faceAttributes[i+1]);
faceNormals.push(faceAttributes[i+2]);
}
} else if (hasUvs) {
for (i=0; i <= faceAttributes.length - 2; i += 2)
{
facePositions.push(faceAttributes[i]);
faceUvs.push(faceAttributes[i+1]);
}
faceNormals = undefined;
} else if (hasNormals) {
for (i=0; i <= faceAttributes.length - 2; i += 2)
{
facePositions.push(faceAttributes[i]);
faceNormals.push(faceAttributes[i+1]);
}
faceUvs = undefined;
} else {
for (i=0; i < faceAttributes.length; ++i)
{
facePositions.push(faceAttributes[i]);
}
faceUvs = undefined;
faceNormals = undefined;
}
addFace(faceVertices, facePositions, faceUvs, faceNormals);
2017-03-13 15:28:51 -04:00
}
2016-06-09 13:33:08 -04:00
2017-03-13 15:28:51 -04:00
function parseLine(line) {
line = line.trim();
var result;
if ((line.length === 0) || (line.charAt(0) === '#')) {
// Don't process empty lines or comments
} else if (/^o\s/i.test(line)) {
var objectName = line.substring(2).trim();
addNode(objectName);
} else if (/^g\s/i.test(line)) {
var groupName = line.substring(2).trim();
addMesh(groupName);
} else if (/^usemtl\s/i.test(line)) {
var materialName = line.substring(7).trim();
useMaterial(materialName);
} else if (/^mtllib/i.test(line)) {
var paths = line.substring(7).trim().split(' ');
mtlPaths = mtlPaths.concat(paths);
} else if ((result = vertexPattern.exec(line)) !== null) {
2017-04-20 14:41:39 -04:00
var position = scratchCartesian;
position.x = parseFloat(result[1]);
position.y = parseFloat(result[2]);
position.z = parseFloat(result[3]);
if (defined(axisTransform)) {
Matrix4.multiplyByPoint(axisTransform, position, position);
}
positions.push(position.x);
positions.push(position.y);
positions.push(position.z);
2017-03-13 15:28:51 -04:00
} else if ((result = normalPattern.exec(line) ) !== null) {
2017-04-20 14:41:39 -04:00
var normal = scratchCartesian;
normal.x = parseFloat(result[1]);
normal.y = parseFloat(result[2]);
normal.z = parseFloat(result[3]);
if (defined(axisTransform)) {
Matrix4.multiplyByPointAsVector(axisTransform, normal, normal);
}
normals.push(normal.x);
normals.push(normal.y);
normals.push(normal.z);
2017-03-13 15:28:51 -04:00
} else if ((result = uvPattern.exec(line)) !== null) {
uvs.push(parseFloat(result[1]));
uvs.push(1.0 - parseFloat(result[2])); // Flip y so 0.0 is the bottom of the image
} else { // face line or invalid line
2017-06-26 12:23:34 -04:00
// Because face lines can contain n vertices, we use a line buffer in case the face data spans multiple lines.
// If there's a line continuation don't create face yet
if (line.slice(-1) === '\\') {
lineBuffer += line.substring(0, -1);
return;
}
lineBuffer += line;
if (facePattern1.test(lineBuffer)) { // format "f v v v ..."
parseFaceLine(line, false, false);
2017-06-26 12:23:34 -04:00
} else if (facePattern2.test(lineBuffer)) { // format "f v/uv v/uv v/uv ..."
parseFaceLine(line, true, false);
2017-06-26 12:23:34 -04:00
} else if (facePattern3.test(lineBuffer)) { // format "v/uv/n v/uv/n v/uv/n ..."
parseFaceLine(line, true, true);
2017-06-26 12:23:34 -04:00
} else if (facePattern4.test(lineBuffer)) { // format "v//n v//n v//n ..."
parseFaceLine(line, false, true);
2017-06-13 20:38:38 -04:00
}
2017-06-26 12:23:34 -04:00
lineBuffer = "";
2016-06-09 13:33:08 -04:00
}
2017-03-13 15:28:51 -04:00
}
2015-10-16 17:32:23 -04:00
2017-03-13 15:28:51 -04:00
// Create a default node in case there are no o/g/usemtl lines in the obj
addNode();
// Parse the obj file
return readLines(objPath, parseLine)
.then(function() {
// Unload resources
positions = undefined;
normals = undefined;
uvs = undefined;
// Load materials and images
2017-03-17 16:05:51 -04:00
return finishLoading(nodes, mtlPaths, objPath, options);
2016-07-22 16:17:27 -04:00
});
2017-03-13 15:28:51 -04:00
}
2016-07-22 16:17:27 -04:00
2017-03-17 16:05:51 -04:00
function finishLoading(nodes, mtlPaths, objPath, options) {
2017-03-13 15:28:51 -04:00
nodes = cleanNodes(nodes);
if (nodes.length === 0) {
2017-04-10 17:57:56 -04:00
return Promise.reject(new RuntimeError(objPath + ' does not have any geometry data'));
2017-03-13 15:28:51 -04:00
}
2017-04-04 16:45:21 -04:00
return loadMaterials(mtlPaths, objPath, options)
2017-03-13 15:28:51 -04:00
.then(function(materials) {
var imagePaths = getImagePaths(materials);
2017-04-04 16:45:21 -04:00
return loadImages(imagePaths, objPath, options)
2017-03-13 15:28:51 -04:00
.then(function(images) {
return {
nodes : nodes,
materials : materials,
images : images
};
});
});
2017-03-13 15:28:51 -04:00
}
2017-04-04 16:45:21 -04:00
function outsideDirectory(filePath, objPath) {
return (path.relative(path.dirname(objPath), filePath).indexOf('..') === 0);
}
function loadMaterials(mtlPaths, objPath, options) {
2017-04-10 17:57:56 -04:00
var secure = options.secure;
var logger = options.logger;
var objDirectory = path.dirname(objPath);
2017-03-13 15:28:51 -04:00
var materials = {};
return Promise.map(mtlPaths, function(mtlPath) {
2017-04-10 17:57:56 -04:00
mtlPath = path.resolve(objDirectory, mtlPath);
if (secure && outsideDirectory(mtlPath, objPath)) {
logger('Could not read mtl file at ' + mtlPath + ' because it is outside of the obj directory and the secure flag is true. Using default material instead.');
2017-04-04 16:45:21 -04:00
return;
}
2017-03-13 15:28:51 -04:00
return loadMtl(mtlPath)
.then(function(materialsInMtl) {
2017-04-10 17:57:56 -04:00
materials = Object.assign(materials, materialsInMtl);
2017-04-04 16:45:21 -04:00
})
.catch(function() {
2017-04-10 17:57:56 -04:00
logger('Could not read mtl file at ' + mtlPath + '. Using default material instead.');
2017-03-13 15:28:51 -04:00
});
2017-04-10 17:57:56 -04:00
}, {concurrency : 10})
.thenReturn(materials);
2017-03-13 15:28:51 -04:00
}
2017-04-04 16:45:21 -04:00
function loadImages(imagePaths, objPath, options) {
2017-04-10 17:57:56 -04:00
var secure = options.secure;
var logger = options.logger;
2017-03-13 15:28:51 -04:00
var images = {};
return Promise.map(imagePaths, function(imagePath) {
2017-04-10 17:57:56 -04:00
if (secure && outsideDirectory(imagePath, objPath)) {
logger('Could not read image file at ' + imagePath + ' because it is outside of the obj directory and the secure flag is true. Material will ignore this image.');
2017-04-04 16:45:21 -04:00
return;
}
2017-03-17 16:05:51 -04:00
return loadImage(imagePath, options)
2017-03-13 15:28:51 -04:00
.then(function(image) {
2017-04-10 17:57:56 -04:00
images[imagePath] = image;
2017-04-04 16:45:21 -04:00
})
.catch(function() {
2017-04-10 17:57:56 -04:00
logger('Could not read image file at ' + imagePath + '. Material will ignore this image.');
2017-03-13 15:28:51 -04:00
});
2017-04-10 17:57:56 -04:00
}, {concurrency : 10})
.thenReturn(images);
2016-06-09 13:33:08 -04:00
}
2015-10-16 17:32:23 -04:00
2017-03-13 15:28:51 -04:00
function getImagePaths(materials) {
2017-03-17 11:40:54 -04:00
var imagePaths = {};
2016-06-09 13:33:08 -04:00
for (var name in materials) {
if (materials.hasOwnProperty(name)) {
var material = materials[name];
2017-04-10 17:57:56 -04:00
if (defined(material.ambientTexture)) {
imagePaths[material.ambientTexture] = true;
2015-10-16 17:32:23 -04:00
}
2017-04-10 17:57:56 -04:00
if (defined(material.diffuseTexture)) {
imagePaths[material.diffuseTexture] = true;
}
2017-04-10 17:57:56 -04:00
if (defined(material.emissionTexture)) {
imagePaths[material.emissionTexture] = true;
2016-06-09 13:33:08 -04:00
}
2017-04-10 17:57:56 -04:00
if (defined(material.specularTexture)) {
imagePaths[material.specularTexture] = true;
2016-06-09 13:33:08 -04:00
}
}
}
2017-03-17 11:40:54 -04:00
return Object.keys(imagePaths);
2017-03-13 15:28:51 -04:00
}
2017-03-13 15:28:51 -04:00
function removeEmptyMeshes(meshes) {
2017-04-10 17:57:56 -04:00
return meshes.filter(function(mesh) {
// Remove empty primitives
mesh.primitives = mesh.primitives.filter(function(primitive) {
return primitive.indices.length > 0;
});
// Valid meshes must have at least one primitive and contain positions
return (mesh.primitives.length > 0) && (mesh.positions.length > 0);
});
2017-03-13 15:28:51 -04:00
}
2016-08-08 11:35:21 -04:00
2017-03-13 15:28:51 -04:00
function meshesHaveNames(meshes) {
var meshesLength = meshes.length;
for (var i = 0; i < meshesLength; ++i) {
if (defined(meshes[i].name)) {
return true;
}
}
return false;
}
2017-03-13 15:28:51 -04:00
function removeEmptyNodes(nodes) {
var final = [];
var nodesLength = nodes.length;
for (var i = 0; i < nodesLength; ++i) {
var node = nodes[i];
var meshes = removeEmptyMeshes(node.meshes);
if (meshes.length === 0) {
continue;
}
node.meshes = meshes;
if (!defined(node.name) && meshesHaveNames(meshes)) {
// If the obj has groups (g) but not object groups (o) then convert meshes to nodes
var meshesLength = meshes.length;
for (var j = 0; j < meshesLength; ++j) {
var mesh = meshes[j];
var convertedNode = new Node();
convertedNode.name = mesh.name;
convertedNode.meshes = [mesh];
final.push(convertedNode);
2016-07-22 14:09:13 -04:00
}
2017-03-13 15:28:51 -04:00
} else {
final.push(node);
}
}
return final;
}
2016-06-09 13:33:08 -04:00
2017-03-13 15:28:51 -04:00
function setDefaultNames(items, defaultName, usedNames) {
var itemsLength = items.length;
for (var i = 0; i < itemsLength; ++i) {
var item = items[i];
var name = defaultValue(item.name, defaultName);
var occurrences = usedNames[name];
if (defined(occurrences)) {
usedNames[name]++;
name = name + '_' + occurrences;
} else {
usedNames[name] = 1;
}
item.name = name;
}
}
2016-07-22 14:09:13 -04:00
2017-03-13 15:28:51 -04:00
function setDefaults(nodes) {
var usedNames = {};
setDefaultNames(nodes, 'Node', usedNames);
var nodesLength = nodes.length;
for (var i = 0; i < nodesLength; ++i) {
var node = nodes[i];
setDefaultNames(node.meshes, node.name + '-Mesh', usedNames);
}
}
function cleanNodes(nodes) {
nodes = removeEmptyNodes(nodes);
setDefaults(nodes);
return nodes;
2016-06-09 13:33:08 -04:00
}
2017-04-20 14:41:39 -04:00
function getAxisTransform(inputUpAxis, outputUpAxis) {
if (inputUpAxis === 'X' && outputUpAxis === 'Y') {
return Axis.X_UP_TO_Y_UP;
} else if (inputUpAxis === 'X' && outputUpAxis === 'Z') {
return Axis.X_UP_TO_Z_UP;
} else if (inputUpAxis === 'Y' && outputUpAxis === 'X') {
return Axis.Y_UP_TO_X_UP;
} else if (inputUpAxis === 'Y' && outputUpAxis === 'Z') {
return Axis.Y_UP_TO_Z_UP;
} else if (inputUpAxis === 'Z' && outputUpAxis === 'X') {
return Axis.Z_UP_TO_X_UP;
} else if (inputUpAxis === 'Z' && outputUpAxis === 'Y') {
return Axis.Z_UP_TO_Y_UP;
}
}