( function () {
/**
* https://github.com/gkjohnson/collada-exporter-js
*
* Usage:
* const exporter = new ColladaExporter();
*
* const data = exporter.parse(mesh);
*
* Format Definition:
* https://www.khronos.org/collada/
*/
class ColladaExporter {
parse( object, onDone, options = {} ) {
options = Object.assign( {
version: '1.4.1',
author: null,
textureDirectory: '',
upAxis: 'Y_UP',
unitName: null,
unitMeter: null
}, options );
if ( options.upAxis.match( /^[XYZ]_UP$/ ) === null ) {
console.error( 'ColladaExporter: Invalid upAxis: valid values are X_UP, Y_UP or Z_UP.' );
return null;
}
if ( options.unitName !== null && options.unitMeter === null ) {
console.error( 'ColladaExporter: unitMeter needs to be specified if unitName is specified.' );
return null;
}
if ( options.unitMeter !== null && options.unitName === null ) {
console.error( 'ColladaExporter: unitName needs to be specified if unitMeter is specified.' );
return null;
}
if ( options.textureDirectory !== '' ) {
options.textureDirectory = `${options.textureDirectory}/`.replace( /\\/g, '/' ).replace( /\/+/g, '/' );
}
const version = options.version;
if ( version !== '1.4.1' && version !== '1.5.0' ) {
console.warn( `ColladaExporter : Version ${version} not supported for export. Only 1.4.1 and 1.5.0.` );
return null;
} // Convert the urdf xml into a well-formatted, indented format
function format( urdf ) {
const IS_END_TAG = /^<\//;
const IS_SELF_CLOSING = /(\?>$)|(\/>$)/;
const HAS_TEXT = /<[^>]+>[^<]*<\/[^<]+>/;
const pad = ( ch, num ) => num > 0 ? ch + pad( ch, num - 1 ) : '';
let tagnum = 0;
return urdf.match( /(<[^>]+>[^<]+<\/[^<]+>)|(<[^>]+>)/g ).map( tag => {
if ( ! HAS_TEXT.test( tag ) && ! IS_SELF_CLOSING.test( tag ) && IS_END_TAG.test( tag ) ) {
tagnum --;
}
const res = `${pad( ' ', tagnum )}${tag}`;
if ( ! HAS_TEXT.test( tag ) && ! IS_SELF_CLOSING.test( tag ) && ! IS_END_TAG.test( tag ) ) {
tagnum ++;
}
return res;
} ).join( '\n' );
} // Convert an image into a png format for saving
function base64ToBuffer( str ) {
const b = atob( str );
const buf = new Uint8Array( b.length );
for ( let i = 0, l = buf.length; i < l; i ++ ) {
buf[ i ] = b.charCodeAt( i );
}
return buf;
}
let canvas, ctx;
function imageToData( image, ext ) {
canvas = canvas || document.createElement( 'canvas' );
ctx = ctx || canvas.getContext( '2d' );
canvas.width = image.width;
canvas.height = image.height;
ctx.drawImage( image, 0, 0 ); // Get the base64 encoded data
const base64data = canvas.toDataURL( `image/${ext}`, 1 ).replace( /^data:image\/(png|jpg);base64,/, '' ); // Convert to a uint8 array
return base64ToBuffer( base64data );
} // gets the attribute array. Generate a new array if the attribute is interleaved
const getFuncs = [ 'getX', 'getY', 'getZ', 'getW' ];
function attrBufferToArray( attr ) {
if ( attr.isInterleavedBufferAttribute ) {
// use the typed array constructor to save on memory
const arr = new attr.array.constructor( attr.count * attr.itemSize );
const size = attr.itemSize;
for ( let i = 0, l = attr.count; i < l; i ++ ) {
for ( let j = 0; j < size; j ++ ) {
arr[ i * size + j ] = attr[ getFuncs[ j ] ]( i );
}
}
return arr;
} else {
return attr.array;
}
} // Returns an array of the same type starting at the `st` index,
// and `ct` length
function subArray( arr, st, ct ) {
if ( Array.isArray( arr ) ) return arr.slice( st, st + ct ); else return new arr.constructor( arr.buffer, st * arr.BYTES_PER_ELEMENT, ct );
} // Returns the string for a geometry's attribute
function getAttribute( attr, name, params, type ) {
const array = attrBufferToArray( attr );
const res = `';
return res;
} // Returns the string for a node's transform information
let transMat;
function getTransform( o ) {
// ensure the object's matrix is up to date
// before saving the transform
o.updateMatrix();
transMat = transMat || new THREE.Matrix4();
transMat.copy( o.matrix );
transMat.transpose();
return `${transMat.toArray().join( ' ' )}`;
} // Process the given piece of geometry into the geometry library
// Returns the mesh id
function processGeometry( g ) {
let info = geometryInfo.get( g );
if ( ! info ) {
// convert the geometry to bufferGeometry if it isn't already
const bufferGeometry = g;
if ( bufferGeometry.isBufferGeometry !== true ) {
throw new Error( 'THREE.ColladaExporter: Geometry is not of type THREE.BufferGeometry.' );
}
const meshid = `Mesh${libraryGeometries.length + 1}`;
const indexCount = bufferGeometry.index ? bufferGeometry.index.count * bufferGeometry.index.itemSize : bufferGeometry.attributes.position.count;
const groups = bufferGeometry.groups != null && bufferGeometry.groups.length !== 0 ? bufferGeometry.groups : [ {
start: 0,
count: indexCount,
materialIndex: 0
} ];
const gname = g.name ? ` name="${g.name}"` : '';
let gnode = ``; // define the geometry node and the vertices for the geometry
const posName = `${meshid}-position`;
const vertName = `${meshid}-vertices`;
gnode += getAttribute( bufferGeometry.attributes.position, posName, [ 'X', 'Y', 'Z' ], 'float' );
gnode += ``; // NOTE: We're not optimizing the attribute arrays here, so they're all the same length and
// can therefore share the same triangle indices. However, MeshLab seems to have trouble opening
// models with attributes that share an offset.
// MeshLab Bug#424: https://sourceforge.net/p/meshlab/bugs/424/
// serialize normals
let triangleInputs = ``;
if ( 'normal' in bufferGeometry.attributes ) {
const normName = `${meshid}-normal`;
gnode += getAttribute( bufferGeometry.attributes.normal, normName, [ 'X', 'Y', 'Z' ], 'float' );
triangleInputs += ``;
} // serialize uvs
if ( 'uv' in bufferGeometry.attributes ) {
const uvName = `${meshid}-texcoord`;
gnode += getAttribute( bufferGeometry.attributes.uv, uvName, [ 'S', 'T' ], 'float' );
triangleInputs += ``;
} // serialize lightmap uvs
if ( 'uv2' in bufferGeometry.attributes ) {
const uvName = `${meshid}-texcoord2`;
gnode += getAttribute( bufferGeometry.attributes.uv2, uvName, [ 'S', 'T' ], 'float' );
triangleInputs += ``;
} // serialize colors
if ( 'color' in bufferGeometry.attributes ) {
const colName = `${meshid}-color`;
gnode += getAttribute( bufferGeometry.attributes.color, colName, [ 'X', 'Y', 'Z' ], 'uint8' );
triangleInputs += ``;
}
let indexArray = null;
if ( bufferGeometry.index ) {
indexArray = attrBufferToArray( bufferGeometry.index );
} else {
indexArray = new Array( indexCount );
for ( let i = 0, l = indexArray.length; i < l; i ++ ) indexArray[ i ] = i;
}
for ( let i = 0, l = groups.length; i < l; i ++ ) {
const group = groups[ i ];
const subarr = subArray( indexArray, group.start, group.count );
const polycount = subarr.length / 3;
gnode += ``;
gnode += triangleInputs;
gnode += `${subarr.join( ' ' )}
`;
gnode += '';
}
gnode += '';
libraryGeometries.push( gnode );
info = {
meshid: meshid,
bufferGeometry: bufferGeometry
};
geometryInfo.set( g, info );
}
return info;
} // Process the given texture into the image library
// Returns the image library
function processTexture( tex ) {
let texid = imageMap.get( tex );
if ( texid == null ) {
texid = `image-${libraryImages.length + 1}`;
const ext = 'png';
const name = tex.name || texid;
let imageNode = ``;
if ( version === '1.5.0' ) {
imageNode += `[${options.textureDirectory}${name}.${ext}]`;
} else {
// version image node 1.4.1
imageNode += `${options.textureDirectory}${name}.${ext}`;
}
imageNode += '';
libraryImages.push( imageNode );
imageMap.set( tex, texid );
textures.push( {
directory: options.textureDirectory,
name,
ext,
data: imageToData( tex.image, ext ),
original: tex
} );
}
return texid;
} // Process the given material into the material and effect libraries
// Returns the material id
function processMaterial( m ) {
let matid = materialMap.get( m );
if ( matid == null ) {
matid = `Mat${libraryEffects.length + 1}`;
let type = 'phong';
if ( m.isMeshLambertMaterial === true ) {
type = 'lambert';
} else if ( m.isMeshBasicMaterial === true ) {
type = 'constant';
if ( m.map !== null ) {
// The Collada spec does not support diffuse texture maps with the
// constant shader type.
// mrdoob/three.js#15469
console.warn( 'ColladaExporter: Texture maps not supported with THREE.MeshBasicMaterial.' );
}
}
const emissive = m.emissive ? m.emissive : new THREE.Color( 0, 0, 0 );
const diffuse = m.color ? m.color : new THREE.Color( 0, 0, 0 );
const specular = m.specular ? m.specular : new THREE.Color( 1, 1, 1 );
const shininess = m.shininess || 0;
const reflectivity = m.reflectivity || 0; // Do not export and alpha map for the reasons mentioned in issue (#13792)
// in three.js alpha maps are black and white, but collada expects the alpha
// channel to specify the transparency
let transparencyNode = '';
if ( m.transparent === true ) {
transparencyNode += '' + ( m.map ? '' : '1' ) + '';
if ( m.opacity < 1 ) {
transparencyNode += `${m.opacity}`;
}
}
const techniqueNode = `<${type}>` + '' + ( m.emissiveMap ? '' : `${emissive.r} ${emissive.g} ${emissive.b} 1` ) + '' + ( type !== 'constant' ? '' + ( m.map ? '' : `${diffuse.r} ${diffuse.g} ${diffuse.b} 1` ) + '' : '' ) + ( type !== 'constant' ? '' + ( m.normalMap ? '' : '' ) + '' : '' ) + ( type === 'phong' ? `${specular.r} ${specular.g} ${specular.b} 1` + '' + ( m.specularMap ? '' : `${shininess}` ) + '' : '' ) + `${diffuse.r} ${diffuse.g} ${diffuse.b} 1` + `${reflectivity}` + transparencyNode + `${type}>`;
const effectnode = `` + '' + ( m.map ? '' + `${processTexture( m.map )}` + '' + '' : '' ) + ( m.specularMap ? '' + `${processTexture( m.specularMap )}` + '' + '' : '' ) + ( m.emissiveMap ? '' + `${processTexture( m.emissiveMap )}` + '' + '' : '' ) + ( m.normalMap ? '' + `${processTexture( m.normalMap )}` + '' + '' : '' ) + techniqueNode + ( m.side === THREE.DoubleSide ? '1' : '' ) + '' + '';
const materialName = m.name ? ` name="${m.name}"` : '';
const materialNode = ``;
libraryMaterials.push( materialNode );
libraryEffects.push( effectnode );
materialMap.set( m, matid );
}
return matid;
} // Recursively process the object into a scene
function processObject( o ) {
let node = ``;
node += getTransform( o );
if ( o.isMesh === true && o.geometry !== null ) {
// function returns the id associated with the mesh and a "BufferGeometry" version
// of the geometry in case it's not a geometry.
const geomInfo = processGeometry( o.geometry );
const meshid = geomInfo.meshid;
const geometry = geomInfo.bufferGeometry; // ids of the materials to bind to the geometry
let matids = null;
let matidsArray; // get a list of materials to bind to the sub groups of the geometry.
// If the amount of subgroups is greater than the materials, than reuse
// the materials.
const mat = o.material || new THREE.MeshBasicMaterial();
const materials = Array.isArray( mat ) ? mat : [ mat ];
if ( geometry.groups.length > materials.length ) {
matidsArray = new Array( geometry.groups.length );
} else {
matidsArray = new Array( materials.length );
}
matids = matidsArray.fill().map( ( v, i ) => processMaterial( materials[ i % materials.length ] ) );
node += `` + ( matids != null ? '' + matids.map( ( id, i ) => `` + '' + '' ).join( '' ) + '' : '' ) + '';
}
o.children.forEach( c => node += processObject( c ) );
node += '';
return node;
}
const geometryInfo = new WeakMap();
const materialMap = new WeakMap();
const imageMap = new WeakMap();
const textures = [];
const libraryImages = [];
const libraryGeometries = [];
const libraryEffects = [];
const libraryMaterials = [];
const libraryVisualScenes = processObject( object );
const specLink = version === '1.4.1' ? 'http://www.collada.org/2005/11/COLLADASchema' : 'https://www.khronos.org/collada/';
let dae = '' + `` + '' + ( '' + 'three.js Collada Exporter' + ( options.author !== null ? `${options.author}` : '' ) + '' + `${new Date().toISOString()}` + `${new Date().toISOString()}` + ( options.unitName !== null ? `` : '' ) + `${options.upAxis}` ) + '';
dae += `${libraryImages.join( '' )}`;
dae += `${libraryEffects.join( '' )}`;
dae += `${libraryMaterials.join( '' )}`;
dae += `${libraryGeometries.join( '' )}`;
dae += `${libraryVisualScenes}`;
dae += '';
dae += '';
const res = {
data: format( dae ),
textures
};
if ( typeof onDone === 'function' ) {
requestAnimationFrame( () => onDone( res ) );
}
return res;
}
}
THREE.ColladaExporter = ColladaExporter;
} )();