123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225 |
- /**
- * Export draco compressed files from threejs geometry objects.
- *
- * Draco files are compressed and usually are smaller than conventional 3D file formats.
- *
- * The exporter receives a options object containing
- * - decodeSpeed, indicates how to tune the encoder regarding decode speed (0 gives better speed but worst quality)
- * - encodeSpeed, indicates how to tune the encoder parameters (0 gives better speed but worst quality)
- * - encoderMethod
- * - quantization, indicates the presision of each type of data stored in the draco file in the order (POSITION, NORMAL, COLOR, TEX_COORD, GENERIC)
- * - exportUvs
- * - exportNormals
- */
- /* global DracoEncoderModule */
- class DRACOExporter {
- parse( object, options = {
- decodeSpeed: 5,
- encodeSpeed: 5,
- encoderMethod: DRACOExporter.MESH_EDGEBREAKER_ENCODING,
- quantization: [ 16, 8, 8, 8, 8 ],
- exportUvs: true,
- exportNormals: true,
- exportColor: false,
- } ) {
- if ( DracoEncoderModule === undefined ) {
- throw new Error( 'THREE.DRACOExporter: required the draco_encoder to work.' );
- }
- const geometry = object.geometry;
- const dracoEncoder = DracoEncoderModule();
- const encoder = new dracoEncoder.Encoder();
- let builder;
- let dracoObject;
- if ( object.isMesh === true ) {
- builder = new dracoEncoder.MeshBuilder();
- dracoObject = new dracoEncoder.Mesh();
- const vertices = geometry.getAttribute( 'position' );
- builder.AddFloatAttributeToMesh( dracoObject, dracoEncoder.POSITION, vertices.count, vertices.itemSize, vertices.array );
- const faces = geometry.getIndex();
- if ( faces !== null ) {
- builder.AddFacesToMesh( dracoObject, faces.count / 3, faces.array );
- } else {
- const faces = new ( vertices.count > 65535 ? Uint32Array : Uint16Array )( vertices.count );
- for ( let i = 0; i < faces.length; i ++ ) {
- faces[ i ] = i;
- }
- builder.AddFacesToMesh( dracoObject, vertices.count, faces );
- }
- if ( options.exportNormals === true ) {
- const normals = geometry.getAttribute( 'normal' );
- if ( normals !== undefined ) {
- builder.AddFloatAttributeToMesh( dracoObject, dracoEncoder.NORMAL, normals.count, normals.itemSize, normals.array );
- }
- }
- if ( options.exportUvs === true ) {
- const uvs = geometry.getAttribute( 'uv' );
- if ( uvs !== undefined ) {
- builder.AddFloatAttributeToMesh( dracoObject, dracoEncoder.TEX_COORD, uvs.count, uvs.itemSize, uvs.array );
- }
- }
- if ( options.exportColor === true ) {
- const colors = geometry.getAttribute( 'color' );
- if ( colors !== undefined ) {
- builder.AddFloatAttributeToMesh( dracoObject, dracoEncoder.COLOR, colors.count, colors.itemSize, colors.array );
- }
- }
- } else if ( object.isPoints === true ) {
- builder = new dracoEncoder.PointCloudBuilder();
- dracoObject = new dracoEncoder.PointCloud();
- const vertices = geometry.getAttribute( 'position' );
- builder.AddFloatAttribute( dracoObject, dracoEncoder.POSITION, vertices.count, vertices.itemSize, vertices.array );
- if ( options.exportColor === true ) {
- const colors = geometry.getAttribute( 'color' );
- if ( colors !== undefined ) {
- builder.AddFloatAttribute( dracoObject, dracoEncoder.COLOR, colors.count, colors.itemSize, colors.array );
- }
- }
- } else {
- throw new Error( 'DRACOExporter: Unsupported object type.' );
- }
- //Compress using draco encoder
- const encodedData = new dracoEncoder.DracoInt8Array();
- //Sets the desired encoding and decoding speed for the given options from 0 (slowest speed, but the best compression) to 10 (fastest, but the worst compression).
- const encodeSpeed = ( options.encodeSpeed !== undefined ) ? options.encodeSpeed : 5;
- const decodeSpeed = ( options.decodeSpeed !== undefined ) ? options.decodeSpeed : 5;
- encoder.SetSpeedOptions( encodeSpeed, decodeSpeed );
- // Sets the desired encoding method for a given geometry.
- if ( options.encoderMethod !== undefined ) {
- encoder.SetEncodingMethod( options.encoderMethod );
- }
- // Sets the quantization (number of bits used to represent) compression options for a named attribute.
- // The attribute values will be quantized in a box defined by the maximum extent of the attribute values.
- if ( options.quantization !== undefined ) {
- for ( let i = 0; i < 5; i ++ ) {
- if ( options.quantization[ i ] !== undefined ) {
- encoder.SetAttributeQuantization( i, options.quantization[ i ] );
- }
- }
- }
- let length;
- if ( object.isMesh === true ) {
- length = encoder.EncodeMeshToDracoBuffer( dracoObject, encodedData );
- } else {
- length = encoder.EncodePointCloudToDracoBuffer( dracoObject, true, encodedData );
- }
- dracoEncoder.destroy( dracoObject );
- if ( length === 0 ) {
- throw new Error( 'THREE.DRACOExporter: Draco encoding failed.' );
- }
- //Copy encoded data to buffer.
- const outputData = new Int8Array( new ArrayBuffer( length ) );
- for ( let i = 0; i < length; i ++ ) {
- outputData[ i ] = encodedData.GetValue( i );
- }
- dracoEncoder.destroy( encodedData );
- dracoEncoder.destroy( encoder );
- dracoEncoder.destroy( builder );
- return outputData;
- }
- }
- // Encoder methods
- DRACOExporter.MESH_EDGEBREAKER_ENCODING = 1;
- DRACOExporter.MESH_SEQUENTIAL_ENCODING = 0;
- // Geometry type
- DRACOExporter.POINT_CLOUD = 0;
- DRACOExporter.TRIANGULAR_MESH = 1;
- // Attribute type
- DRACOExporter.INVALID = - 1;
- DRACOExporter.POSITION = 0;
- DRACOExporter.NORMAL = 1;
- DRACOExporter.COLOR = 2;
- DRACOExporter.TEX_COORD = 3;
- DRACOExporter.GENERIC = 4;
- export { DRACOExporter };
|