3MFLoader.js 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614
  1. /**
  2. * @author technohippy / https://github.com/technohippy
  3. */
  4. THREE.ThreeMFLoader = function ( manager ) {
  5. this.manager = ( manager !== undefined ) ? manager : THREE.DefaultLoadingManager;
  6. this.availableExtensions = [];
  7. };
  8. THREE.ThreeMFLoader.prototype = {
  9. constructor: THREE.ThreeMFLoader,
  10. load: function ( url, onLoad, onProgress, onError ) {
  11. var scope = this;
  12. var loader = new THREE.FileLoader( scope.manager );
  13. loader.setPath( scope.path );
  14. loader.setResponseType( 'arraybuffer' );
  15. loader.load( url, function ( buffer ) {
  16. onLoad( scope.parse( buffer ) );
  17. }, onProgress, onError );
  18. },
  19. setPath: function ( value ) {
  20. this.path = value;
  21. return this;
  22. },
  23. parse: function ( data ) {
  24. var scope = this;
  25. function loadDocument( data ) {
  26. var zip = null;
  27. var file = null;
  28. var relsName;
  29. var modelPartNames = [];
  30. var printTicketPartNames = [];
  31. var texturesPartNames = [];
  32. var otherPartNames = [];
  33. var rels;
  34. var modelParts = {};
  35. var printTicketParts = {};
  36. var texturesParts = {};
  37. var otherParts = {};
  38. try {
  39. zip = new JSZip( data ); // eslint-disable-line no-undef
  40. } catch ( e ) {
  41. if ( e instanceof ReferenceError ) {
  42. console.error( 'THREE.ThreeMFLoader: jszip missing and file is compressed.' );
  43. return null;
  44. }
  45. }
  46. for ( file in zip.files ) {
  47. if ( file.match( /\.rels$/ ) ) {
  48. relsName = file;
  49. } else if ( file.match( /^3D\/.*\.model$/ ) ) {
  50. modelPartNames.push( file );
  51. } else if ( file.match( /^3D\/Metadata\/.*\.xml$/ ) ) {
  52. printTicketPartNames.push( file );
  53. } else if ( file.match( /^3D\/Textures\/.*/ ) ) {
  54. texturesPartNames.push( file );
  55. } else if ( file.match( /^3D\/Other\/.*/ ) ) {
  56. otherPartNames.push( file );
  57. }
  58. }
  59. var relsView = new Uint8Array( zip.file( relsName ).asArrayBuffer() );
  60. var relsFileText = THREE.LoaderUtils.decodeText( relsView );
  61. rels = parseRelsXml( relsFileText );
  62. for ( var i = 0; i < modelPartNames.length; i ++ ) {
  63. var modelPart = modelPartNames[ i ];
  64. var view = new Uint8Array( zip.file( modelPart ).asArrayBuffer() );
  65. var fileText = THREE.LoaderUtils.decodeText( view );
  66. var xmlData = new DOMParser().parseFromString( fileText, 'application/xml' );
  67. if ( xmlData.documentElement.nodeName.toLowerCase() !== 'model' ) {
  68. console.error( 'THREE.ThreeMFLoader: Error loading 3MF - no 3MF document found: ', modelPart );
  69. }
  70. var modelNode = xmlData.querySelector( 'model' );
  71. var extensions = {};
  72. for ( var i = 0; i < modelNode.attributes.length; i ++ ) {
  73. var attr = modelNode.attributes[ i ];
  74. if ( attr.name.match( /^xmlns:(.+)$/ ) ) {
  75. extensions[ attr.value ] = RegExp.$1;
  76. }
  77. }
  78. var modelData = parseModelNode( modelNode );
  79. modelData[ 'xml' ] = modelNode;
  80. if ( 0 < Object.keys( extensions ).length ) {
  81. modelData[ 'extensions' ] = extensions;
  82. }
  83. modelParts[ modelPart ] = modelData;
  84. }
  85. for ( var i = 0; i < texturesPartNames.length; i ++ ) {
  86. var texturesPartName = texturesPartNames[ i ];
  87. texturesParts[ texturesPartName ] = zip.file( texturesPartName ).asBinary();
  88. }
  89. return {
  90. rels: rels,
  91. model: modelParts,
  92. printTicket: printTicketParts,
  93. texture: texturesParts,
  94. other: otherParts
  95. };
  96. }
  97. function parseRelsXml( relsFileText ) {
  98. var relsXmlData = new DOMParser().parseFromString( relsFileText, 'application/xml' );
  99. var relsNode = relsXmlData.querySelector( 'Relationship' );
  100. var target = relsNode.getAttribute( 'Target' );
  101. var id = relsNode.getAttribute( 'Id' );
  102. var type = relsNode.getAttribute( 'Type' );
  103. return {
  104. target: target,
  105. id: id,
  106. type: type
  107. };
  108. }
  109. function parseMetadataNodes( metadataNodes ) {
  110. var metadataData = {};
  111. for ( var i = 0; i < metadataNodes.length; i ++ ) {
  112. var metadataNode = metadataNodes[ i ];
  113. var name = metadataNode.getAttribute( 'name' );
  114. var validNames = [
  115. 'Title',
  116. 'Designer',
  117. 'Description',
  118. 'Copyright',
  119. 'LicenseTerms',
  120. 'Rating',
  121. 'CreationDate',
  122. 'ModificationDate'
  123. ];
  124. if ( 0 <= validNames.indexOf( name ) ) {
  125. metadataData[ name ] = metadataNode.textContent;
  126. }
  127. }
  128. return metadataData;
  129. }
  130. function parseBasematerialsNode( basematerialsNode ) {
  131. }
  132. function parseMeshNode( meshNode, extensions ) {
  133. var meshData = {};
  134. var vertices = [];
  135. var vertexNodes = meshNode.querySelectorAll( 'vertices vertex' );
  136. for ( var i = 0; i < vertexNodes.length; i ++ ) {
  137. var vertexNode = vertexNodes[ i ];
  138. var x = vertexNode.getAttribute( 'x' );
  139. var y = vertexNode.getAttribute( 'y' );
  140. var z = vertexNode.getAttribute( 'z' );
  141. vertices.push( parseFloat( x ), parseFloat( y ), parseFloat( z ) );
  142. }
  143. meshData[ 'vertices' ] = new Float32Array( vertices.length );
  144. for ( var i = 0; i < vertices.length; i ++ ) {
  145. meshData[ 'vertices' ][ i ] = vertices[ i ];
  146. }
  147. var triangleProperties = [];
  148. var triangles = [];
  149. var triangleNodes = meshNode.querySelectorAll( 'triangles triangle' );
  150. for ( var i = 0; i < triangleNodes.length; i ++ ) {
  151. var triangleNode = triangleNodes[ i ];
  152. var v1 = triangleNode.getAttribute( 'v1' );
  153. var v2 = triangleNode.getAttribute( 'v2' );
  154. var v3 = triangleNode.getAttribute( 'v3' );
  155. var p1 = triangleNode.getAttribute( 'p1' );
  156. var p2 = triangleNode.getAttribute( 'p2' );
  157. var p3 = triangleNode.getAttribute( 'p3' );
  158. var pid = triangleNode.getAttribute( 'pid' );
  159. triangles.push( parseInt( v1, 10 ), parseInt( v2, 10 ), parseInt( v3, 10 ) );
  160. var triangleProperty = {};
  161. if ( p1 ) {
  162. triangleProperty[ 'p1' ] = parseInt( p1, 10 );
  163. }
  164. if ( p2 ) {
  165. triangleProperty[ 'p2' ] = parseInt( p2, 10 );
  166. }
  167. if ( p3 ) {
  168. triangleProperty[ 'p3' ] = parseInt( p3, 10 );
  169. }
  170. if ( pid ) {
  171. triangleProperty[ 'pid' ] = pid;
  172. }
  173. if ( 0 < Object.keys( triangleProperty ).length ) {
  174. triangleProperties.push( triangleProperty );
  175. }
  176. }
  177. meshData[ 'triangleProperties' ] = triangleProperties;
  178. meshData[ 'triangles' ] = new Uint32Array( triangles.length );
  179. for ( var i = 0; i < triangles.length; i ++ ) {
  180. meshData[ 'triangles' ][ i ] = triangles[ i ];
  181. }
  182. return meshData;
  183. }
  184. function parseComponentsNode( componentsNode ) {
  185. }
  186. function parseObjectNode( objectNode ) {
  187. var objectData = {
  188. type: objectNode.getAttribute( 'type' )
  189. };
  190. var id = objectNode.getAttribute( 'id' );
  191. if ( id ) {
  192. objectData[ 'id' ] = id;
  193. }
  194. var pid = objectNode.getAttribute( 'pid' );
  195. if ( pid ) {
  196. objectData[ 'pid' ] = pid;
  197. }
  198. var pindex = objectNode.getAttribute( 'pindex' );
  199. if ( pindex ) {
  200. objectData[ 'pindex' ] = pindex;
  201. }
  202. var thumbnail = objectNode.getAttribute( 'thumbnail' );
  203. if ( thumbnail ) {
  204. objectData[ 'thumbnail' ] = thumbnail;
  205. }
  206. var partnumber = objectNode.getAttribute( 'partnumber' );
  207. if ( partnumber ) {
  208. objectData[ 'partnumber' ] = partnumber;
  209. }
  210. var name = objectNode.getAttribute( 'name' );
  211. if ( name ) {
  212. objectData[ 'name' ] = name;
  213. }
  214. var meshNode = objectNode.querySelector( 'mesh' );
  215. if ( meshNode ) {
  216. objectData[ 'mesh' ] = parseMeshNode( meshNode );
  217. }
  218. var componentsNode = objectNode.querySelector( 'components' );
  219. if ( componentsNode ) {
  220. objectData[ 'components' ] = parseComponentsNode( componentsNode );
  221. }
  222. return objectData;
  223. }
  224. function parseResourcesNode( resourcesNode ) {
  225. var resourcesData = {};
  226. var basematerialsNode = resourcesNode.querySelector( 'basematerials' );
  227. if ( basematerialsNode ) {
  228. resourcesData[ 'basematerial' ] = parseBasematerialsNode( basematerialsNode );
  229. }
  230. resourcesData[ 'object' ] = {};
  231. var objectNodes = resourcesNode.querySelectorAll( 'object' );
  232. for ( var i = 0; i < objectNodes.length; i ++ ) {
  233. var objectNode = objectNodes[ i ];
  234. var objectData = parseObjectNode( objectNode );
  235. resourcesData[ 'object' ][ objectData[ 'id' ] ] = objectData;
  236. }
  237. return resourcesData;
  238. }
  239. function parseBuildNode( buildNode ) {
  240. var buildData = [];
  241. var itemNodes = buildNode.querySelectorAll( 'item' );
  242. for ( var i = 0; i < itemNodes.length; i ++ ) {
  243. var itemNode = itemNodes[ i ];
  244. var buildItem = {
  245. objectid: itemNode.getAttribute( 'objectid' )
  246. };
  247. var transform = itemNode.getAttribute( 'transform' );
  248. if ( transform ) {
  249. var t = [];
  250. transform.split( ' ' ).forEach( function ( s ) {
  251. t.push( parseFloat( s ) );
  252. } );
  253. var mat4 = new THREE.Matrix4();
  254. buildItem[ 'transform' ] = mat4.set(
  255. t[ 0 ], t[ 3 ], t[ 6 ], t[ 9 ],
  256. t[ 1 ], t[ 4 ], t[ 7 ], t[ 10 ],
  257. t[ 2 ], t[ 5 ], t[ 8 ], t[ 11 ],
  258. 0.0, 0.0, 0.0, 1.0
  259. );
  260. }
  261. buildData.push( buildItem );
  262. }
  263. return buildData;
  264. }
  265. function parseModelNode( modelNode ) {
  266. var modelData = { unit: modelNode.getAttribute( 'unit' ) || 'millimeter' };
  267. var metadataNodes = modelNode.querySelectorAll( 'metadata' );
  268. if ( metadataNodes ) {
  269. modelData[ 'metadata' ] = parseMetadataNodes( metadataNodes );
  270. }
  271. var resourcesNode = modelNode.querySelector( 'resources' );
  272. if ( resourcesNode ) {
  273. modelData[ 'resources' ] = parseResourcesNode( resourcesNode );
  274. }
  275. var buildNode = modelNode.querySelector( 'build' );
  276. if ( buildNode ) {
  277. modelData[ 'build' ] = parseBuildNode( buildNode );
  278. }
  279. return modelData;
  280. }
  281. function buildMesh( meshData, data3mf ) {
  282. var geometry = new THREE.BufferGeometry();
  283. geometry.setIndex( new THREE.BufferAttribute( meshData[ 'triangles' ], 1 ) );
  284. geometry.addAttribute( 'position', new THREE.BufferAttribute( meshData[ 'vertices' ], 3 ) );
  285. if ( meshData[ 'colors' ] ) {
  286. geometry.addAttribute( 'color', new THREE.BufferAttribute( meshData[ 'colors' ], 3 ) );
  287. }
  288. geometry.computeBoundingSphere();
  289. var materialOpts = {
  290. flatShading: true
  291. };
  292. if ( meshData[ 'colors' ] && 0 < meshData[ 'colors' ].length ) {
  293. materialOpts[ 'vertexColors' ] = THREE.VertexColors;
  294. } else {
  295. materialOpts[ 'color' ] = 0xaaaaff;
  296. }
  297. var material = new THREE.MeshPhongMaterial( materialOpts );
  298. return new THREE.Mesh( geometry, material );
  299. }
  300. function applyExtensions( extensions, meshData, modelXml, data3mf ) {
  301. if ( ! extensions ) {
  302. return;
  303. }
  304. var availableExtensions = [];
  305. var keys = Object.keys( extensions );
  306. for ( var i = 0; i < keys.length; i ++ ) {
  307. var ns = keys[ i ];
  308. for ( var j = 0; j < scope.availableExtensions.length; j ++ ) {
  309. var extension = scope.availableExtensions[ j ];
  310. if ( extension.ns === ns ) {
  311. availableExtensions.push( extension );
  312. }
  313. }
  314. }
  315. for ( var i = 0; i < availableExtensions.length; i ++ ) {
  316. var extension = availableExtensions[ i ];
  317. extension.apply( modelXml, extensions[ extension[ 'ns' ] ], meshData );
  318. }
  319. }
  320. function buildMeshes( data3mf ) {
  321. var modelsData = data3mf.model;
  322. var meshes = {};
  323. var modelsKeys = Object.keys( modelsData );
  324. for ( var i = 0; i < modelsKeys.length; i ++ ) {
  325. var modelsKey = modelsKeys[ i ];
  326. var modelData = modelsData[ modelsKey ];
  327. var modelXml = modelData[ 'xml' ];
  328. var extensions = modelData[ 'extensions' ];
  329. var objectIds = Object.keys( modelData[ 'resources' ][ 'object' ] );
  330. for ( var j = 0; j < objectIds.length; j ++ ) {
  331. var objectId = objectIds[ j ];
  332. var objectData = modelData[ 'resources' ][ 'object' ][ objectId ];
  333. var meshData = objectData[ 'mesh' ];
  334. applyExtensions( extensions, meshData, modelXml, data3mf );
  335. meshes[ objectId ] = buildMesh( meshData, data3mf );
  336. }
  337. }
  338. return meshes;
  339. }
  340. function build( meshes, refs, data3mf ) {
  341. var group = new THREE.Group();
  342. var buildData = data3mf.model[ refs[ 'target' ].substring( 1 ) ][ 'build' ];
  343. for ( var i = 0; i < buildData.length; i ++ ) {
  344. var buildItem = buildData[ i ];
  345. var mesh = meshes[ buildItem[ 'objectid' ] ];
  346. if ( buildItem[ 'transform' ] ) {
  347. mesh.geometry.applyMatrix( buildItem[ 'transform' ] );
  348. }
  349. group.add( mesh );
  350. }
  351. return group;
  352. }
  353. var data3mf = loadDocument( data );
  354. var meshes = buildMeshes( data3mf );
  355. return build( meshes, data3mf[ 'rels' ], data3mf );
  356. },
  357. addExtension: function ( extension ) {
  358. this.availableExtensions.push( extension );
  359. }
  360. };