3MFLoader.js 34 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340
  1. ( function () {
  2. /**
  3. *
  4. * 3D Manufacturing Format (3MF) specification: https://3mf.io/specification/
  5. *
  6. * The following features from the core specification are supported:
  7. *
  8. * - 3D Models
  9. * - Object Resources (Meshes and Components)
  10. * - Material Resources (Base Materials)
  11. *
  12. * 3MF Materials and Properties Extension are only partially supported.
  13. *
  14. * - Texture 2D
  15. * - Texture 2D Groups
  16. * - THREE.Color Groups (Vertex Colors)
  17. * - Metallic Display Properties (PBR)
  18. */
  19. class ThreeMFLoader extends THREE.Loader {
  20. constructor( manager ) {
  21. super( manager );
  22. this.availableExtensions = [];
  23. }
  24. load( url, onLoad, onProgress, onError ) {
  25. const scope = this;
  26. const loader = new THREE.FileLoader( scope.manager );
  27. loader.setPath( scope.path );
  28. loader.setResponseType( 'arraybuffer' );
  29. loader.setRequestHeader( scope.requestHeader );
  30. loader.setWithCredentials( scope.withCredentials );
  31. loader.load( url, function ( buffer ) {
  32. try {
  33. onLoad( scope.parse( buffer ) );
  34. } catch ( e ) {
  35. if ( onError ) {
  36. onError( e );
  37. } else {
  38. console.error( e );
  39. }
  40. scope.manager.itemError( url );
  41. }
  42. }, onProgress, onError );
  43. }
  44. parse( data ) {
  45. const scope = this;
  46. const textureLoader = new THREE.TextureLoader( this.manager );
  47. function loadDocument( data ) {
  48. let zip = null;
  49. let file = null;
  50. let relsName;
  51. let modelRelsName;
  52. const modelPartNames = [];
  53. const texturesPartNames = [];
  54. let modelRels;
  55. const modelParts = {};
  56. const printTicketParts = {};
  57. const texturesParts = {};
  58. try {
  59. zip = fflate.unzipSync( new Uint8Array( data ) ); // eslint-disable-line no-undef
  60. } catch ( e ) {
  61. if ( e instanceof ReferenceError ) {
  62. console.error( 'THREE.3MFLoader: fflate missing and file is compressed.' );
  63. return null;
  64. }
  65. }
  66. for ( file in zip ) {
  67. if ( file.match( /\_rels\/.rels$/ ) ) {
  68. relsName = file;
  69. } else if ( file.match( /3D\/_rels\/.*\.model\.rels$/ ) ) {
  70. modelRelsName = file;
  71. } else if ( file.match( /^3D\/.*\.model$/ ) ) {
  72. modelPartNames.push( file );
  73. } else if ( file.match( /^3D\/Textures?\/.*/ ) ) {
  74. texturesPartNames.push( file );
  75. }
  76. } //
  77. const relsView = zip[ relsName ];
  78. const relsFileText = THREE.LoaderUtils.decodeText( relsView );
  79. const rels = parseRelsXml( relsFileText ); //
  80. if ( modelRelsName ) {
  81. const relsView = zip[ modelRelsName ];
  82. const relsFileText = THREE.LoaderUtils.decodeText( relsView );
  83. modelRels = parseRelsXml( relsFileText );
  84. } //
  85. for ( let i = 0; i < modelPartNames.length; i ++ ) {
  86. const modelPart = modelPartNames[ i ];
  87. const view = zip[ modelPart ];
  88. const fileText = THREE.LoaderUtils.decodeText( view );
  89. const xmlData = new DOMParser().parseFromString( fileText, 'application/xml' );
  90. if ( xmlData.documentElement.nodeName.toLowerCase() !== 'model' ) {
  91. console.error( 'THREE.3MFLoader: Error loading 3MF - no 3MF document found: ', modelPart );
  92. }
  93. const modelNode = xmlData.querySelector( 'model' );
  94. const extensions = {};
  95. for ( let i = 0; i < modelNode.attributes.length; i ++ ) {
  96. const attr = modelNode.attributes[ i ];
  97. if ( attr.name.match( /^xmlns:(.+)$/ ) ) {
  98. extensions[ attr.value ] = RegExp.$1;
  99. }
  100. }
  101. const modelData = parseModelNode( modelNode );
  102. modelData[ 'xml' ] = modelNode;
  103. if ( 0 < Object.keys( extensions ).length ) {
  104. modelData[ 'extensions' ] = extensions;
  105. }
  106. modelParts[ modelPart ] = modelData;
  107. } //
  108. for ( let i = 0; i < texturesPartNames.length; i ++ ) {
  109. const texturesPartName = texturesPartNames[ i ];
  110. texturesParts[ texturesPartName ] = zip[ texturesPartName ].buffer;
  111. }
  112. return {
  113. rels: rels,
  114. modelRels: modelRels,
  115. model: modelParts,
  116. printTicket: printTicketParts,
  117. texture: texturesParts
  118. };
  119. }
  120. function parseRelsXml( relsFileText ) {
  121. const relationships = [];
  122. const relsXmlData = new DOMParser().parseFromString( relsFileText, 'application/xml' );
  123. const relsNodes = relsXmlData.querySelectorAll( 'Relationship' );
  124. for ( let i = 0; i < relsNodes.length; i ++ ) {
  125. const relsNode = relsNodes[ i ];
  126. const relationship = {
  127. target: relsNode.getAttribute( 'Target' ),
  128. //required
  129. id: relsNode.getAttribute( 'Id' ),
  130. //required
  131. type: relsNode.getAttribute( 'Type' ) //required
  132. };
  133. relationships.push( relationship );
  134. }
  135. return relationships;
  136. }
  137. function parseMetadataNodes( metadataNodes ) {
  138. const metadataData = {};
  139. for ( let i = 0; i < metadataNodes.length; i ++ ) {
  140. const metadataNode = metadataNodes[ i ];
  141. const name = metadataNode.getAttribute( 'name' );
  142. const validNames = [ 'Title', 'Designer', 'Description', 'Copyright', 'LicenseTerms', 'Rating', 'CreationDate', 'ModificationDate' ];
  143. if ( 0 <= validNames.indexOf( name ) ) {
  144. metadataData[ name ] = metadataNode.textContent;
  145. }
  146. }
  147. return metadataData;
  148. }
  149. function parseBasematerialsNode( basematerialsNode ) {
  150. const basematerialsData = {
  151. id: basematerialsNode.getAttribute( 'id' ),
  152. // required
  153. basematerials: []
  154. };
  155. const basematerialNodes = basematerialsNode.querySelectorAll( 'base' );
  156. for ( let i = 0; i < basematerialNodes.length; i ++ ) {
  157. const basematerialNode = basematerialNodes[ i ];
  158. const basematerialData = parseBasematerialNode( basematerialNode );
  159. basematerialData.index = i; // the order and count of the material nodes form an implicit 0-based index
  160. basematerialsData.basematerials.push( basematerialData );
  161. }
  162. return basematerialsData;
  163. }
  164. function parseTexture2DNode( texture2DNode ) {
  165. const texture2dData = {
  166. id: texture2DNode.getAttribute( 'id' ),
  167. // required
  168. path: texture2DNode.getAttribute( 'path' ),
  169. // required
  170. contenttype: texture2DNode.getAttribute( 'contenttype' ),
  171. // required
  172. tilestyleu: texture2DNode.getAttribute( 'tilestyleu' ),
  173. tilestylev: texture2DNode.getAttribute( 'tilestylev' ),
  174. filter: texture2DNode.getAttribute( 'filter' )
  175. };
  176. return texture2dData;
  177. }
  178. function parseTextures2DGroupNode( texture2DGroupNode ) {
  179. const texture2DGroupData = {
  180. id: texture2DGroupNode.getAttribute( 'id' ),
  181. // required
  182. texid: texture2DGroupNode.getAttribute( 'texid' ),
  183. // required
  184. displaypropertiesid: texture2DGroupNode.getAttribute( 'displaypropertiesid' )
  185. };
  186. const tex2coordNodes = texture2DGroupNode.querySelectorAll( 'tex2coord' );
  187. const uvs = [];
  188. for ( let i = 0; i < tex2coordNodes.length; i ++ ) {
  189. const tex2coordNode = tex2coordNodes[ i ];
  190. const u = tex2coordNode.getAttribute( 'u' );
  191. const v = tex2coordNode.getAttribute( 'v' );
  192. uvs.push( parseFloat( u ), parseFloat( v ) );
  193. }
  194. texture2DGroupData[ 'uvs' ] = new Float32Array( uvs );
  195. return texture2DGroupData;
  196. }
  197. function parseColorGroupNode( colorGroupNode ) {
  198. const colorGroupData = {
  199. id: colorGroupNode.getAttribute( 'id' ),
  200. // required
  201. displaypropertiesid: colorGroupNode.getAttribute( 'displaypropertiesid' )
  202. };
  203. const colorNodes = colorGroupNode.querySelectorAll( 'color' );
  204. const colors = [];
  205. const colorObject = new THREE.Color();
  206. for ( let i = 0; i < colorNodes.length; i ++ ) {
  207. const colorNode = colorNodes[ i ];
  208. const color = colorNode.getAttribute( 'color' );
  209. colorObject.setStyle( color.substring( 0, 7 ) );
  210. colorObject.convertSRGBToLinear(); // color is in sRGB
  211. colors.push( colorObject.r, colorObject.g, colorObject.b );
  212. }
  213. colorGroupData[ 'colors' ] = new Float32Array( colors );
  214. return colorGroupData;
  215. }
  216. function parseMetallicDisplaypropertiesNode( metallicDisplaypropetiesNode ) {
  217. const metallicDisplaypropertiesData = {
  218. id: metallicDisplaypropetiesNode.getAttribute( 'id' ) // required
  219. };
  220. const metallicNodes = metallicDisplaypropetiesNode.querySelectorAll( 'pbmetallic' );
  221. const metallicData = [];
  222. for ( let i = 0; i < metallicNodes.length; i ++ ) {
  223. const metallicNode = metallicNodes[ i ];
  224. metallicData.push( {
  225. name: metallicNode.getAttribute( 'name' ),
  226. // required
  227. metallicness: parseFloat( metallicNode.getAttribute( 'metallicness' ) ),
  228. // required
  229. roughness: parseFloat( metallicNode.getAttribute( 'roughness' ) ) // required
  230. } );
  231. }
  232. metallicDisplaypropertiesData.data = metallicData;
  233. return metallicDisplaypropertiesData;
  234. }
  235. function parseBasematerialNode( basematerialNode ) {
  236. const basematerialData = {};
  237. basematerialData[ 'name' ] = basematerialNode.getAttribute( 'name' ); // required
  238. basematerialData[ 'displaycolor' ] = basematerialNode.getAttribute( 'displaycolor' ); // required
  239. basematerialData[ 'displaypropertiesid' ] = basematerialNode.getAttribute( 'displaypropertiesid' );
  240. return basematerialData;
  241. }
  242. function parseMeshNode( meshNode ) {
  243. const meshData = {};
  244. const vertices = [];
  245. const vertexNodes = meshNode.querySelectorAll( 'vertices vertex' );
  246. for ( let i = 0; i < vertexNodes.length; i ++ ) {
  247. const vertexNode = vertexNodes[ i ];
  248. const x = vertexNode.getAttribute( 'x' );
  249. const y = vertexNode.getAttribute( 'y' );
  250. const z = vertexNode.getAttribute( 'z' );
  251. vertices.push( parseFloat( x ), parseFloat( y ), parseFloat( z ) );
  252. }
  253. meshData[ 'vertices' ] = new Float32Array( vertices );
  254. const triangleProperties = [];
  255. const triangles = [];
  256. const triangleNodes = meshNode.querySelectorAll( 'triangles triangle' );
  257. for ( let i = 0; i < triangleNodes.length; i ++ ) {
  258. const triangleNode = triangleNodes[ i ];
  259. const v1 = triangleNode.getAttribute( 'v1' );
  260. const v2 = triangleNode.getAttribute( 'v2' );
  261. const v3 = triangleNode.getAttribute( 'v3' );
  262. const p1 = triangleNode.getAttribute( 'p1' );
  263. const p2 = triangleNode.getAttribute( 'p2' );
  264. const p3 = triangleNode.getAttribute( 'p3' );
  265. const pid = triangleNode.getAttribute( 'pid' );
  266. const triangleProperty = {};
  267. triangleProperty[ 'v1' ] = parseInt( v1, 10 );
  268. triangleProperty[ 'v2' ] = parseInt( v2, 10 );
  269. triangleProperty[ 'v3' ] = parseInt( v3, 10 );
  270. triangles.push( triangleProperty[ 'v1' ], triangleProperty[ 'v2' ], triangleProperty[ 'v3' ] ); // optional
  271. if ( p1 ) {
  272. triangleProperty[ 'p1' ] = parseInt( p1, 10 );
  273. }
  274. if ( p2 ) {
  275. triangleProperty[ 'p2' ] = parseInt( p2, 10 );
  276. }
  277. if ( p3 ) {
  278. triangleProperty[ 'p3' ] = parseInt( p3, 10 );
  279. }
  280. if ( pid ) {
  281. triangleProperty[ 'pid' ] = pid;
  282. }
  283. if ( 0 < Object.keys( triangleProperty ).length ) {
  284. triangleProperties.push( triangleProperty );
  285. }
  286. }
  287. meshData[ 'triangleProperties' ] = triangleProperties;
  288. meshData[ 'triangles' ] = new Uint32Array( triangles );
  289. return meshData;
  290. }
  291. function parseComponentsNode( componentsNode ) {
  292. const components = [];
  293. const componentNodes = componentsNode.querySelectorAll( 'component' );
  294. for ( let i = 0; i < componentNodes.length; i ++ ) {
  295. const componentNode = componentNodes[ i ];
  296. const componentData = parseComponentNode( componentNode );
  297. components.push( componentData );
  298. }
  299. return components;
  300. }
  301. function parseComponentNode( componentNode ) {
  302. const componentData = {};
  303. componentData[ 'objectId' ] = componentNode.getAttribute( 'objectid' ); // required
  304. const transform = componentNode.getAttribute( 'transform' );
  305. if ( transform ) {
  306. componentData[ 'transform' ] = parseTransform( transform );
  307. }
  308. return componentData;
  309. }
  310. function parseTransform( transform ) {
  311. const t = [];
  312. transform.split( ' ' ).forEach( function ( s ) {
  313. t.push( parseFloat( s ) );
  314. } );
  315. const matrix = new THREE.Matrix4();
  316. matrix.set( t[ 0 ], t[ 3 ], t[ 6 ], t[ 9 ], t[ 1 ], t[ 4 ], t[ 7 ], t[ 10 ], t[ 2 ], t[ 5 ], t[ 8 ], t[ 11 ], 0.0, 0.0, 0.0, 1.0 );
  317. return matrix;
  318. }
  319. function parseObjectNode( objectNode ) {
  320. const objectData = {
  321. type: objectNode.getAttribute( 'type' )
  322. };
  323. const id = objectNode.getAttribute( 'id' );
  324. if ( id ) {
  325. objectData[ 'id' ] = id;
  326. }
  327. const pid = objectNode.getAttribute( 'pid' );
  328. if ( pid ) {
  329. objectData[ 'pid' ] = pid;
  330. }
  331. const pindex = objectNode.getAttribute( 'pindex' );
  332. if ( pindex ) {
  333. objectData[ 'pindex' ] = pindex;
  334. }
  335. const thumbnail = objectNode.getAttribute( 'thumbnail' );
  336. if ( thumbnail ) {
  337. objectData[ 'thumbnail' ] = thumbnail;
  338. }
  339. const partnumber = objectNode.getAttribute( 'partnumber' );
  340. if ( partnumber ) {
  341. objectData[ 'partnumber' ] = partnumber;
  342. }
  343. const name = objectNode.getAttribute( 'name' );
  344. if ( name ) {
  345. objectData[ 'name' ] = name;
  346. }
  347. const meshNode = objectNode.querySelector( 'mesh' );
  348. if ( meshNode ) {
  349. objectData[ 'mesh' ] = parseMeshNode( meshNode );
  350. }
  351. const componentsNode = objectNode.querySelector( 'components' );
  352. if ( componentsNode ) {
  353. objectData[ 'components' ] = parseComponentsNode( componentsNode );
  354. }
  355. return objectData;
  356. }
  357. function parseResourcesNode( resourcesNode ) {
  358. const resourcesData = {};
  359. resourcesData[ 'basematerials' ] = {};
  360. const basematerialsNodes = resourcesNode.querySelectorAll( 'basematerials' );
  361. for ( let i = 0; i < basematerialsNodes.length; i ++ ) {
  362. const basematerialsNode = basematerialsNodes[ i ];
  363. const basematerialsData = parseBasematerialsNode( basematerialsNode );
  364. resourcesData[ 'basematerials' ][ basematerialsData[ 'id' ] ] = basematerialsData;
  365. } //
  366. resourcesData[ 'texture2d' ] = {};
  367. const textures2DNodes = resourcesNode.querySelectorAll( 'texture2d' );
  368. for ( let i = 0; i < textures2DNodes.length; i ++ ) {
  369. const textures2DNode = textures2DNodes[ i ];
  370. const texture2DData = parseTexture2DNode( textures2DNode );
  371. resourcesData[ 'texture2d' ][ texture2DData[ 'id' ] ] = texture2DData;
  372. } //
  373. resourcesData[ 'colorgroup' ] = {};
  374. const colorGroupNodes = resourcesNode.querySelectorAll( 'colorgroup' );
  375. for ( let i = 0; i < colorGroupNodes.length; i ++ ) {
  376. const colorGroupNode = colorGroupNodes[ i ];
  377. const colorGroupData = parseColorGroupNode( colorGroupNode );
  378. resourcesData[ 'colorgroup' ][ colorGroupData[ 'id' ] ] = colorGroupData;
  379. } //
  380. resourcesData[ 'pbmetallicdisplayproperties' ] = {};
  381. const pbmetallicdisplaypropertiesNodes = resourcesNode.querySelectorAll( 'pbmetallicdisplayproperties' );
  382. for ( let i = 0; i < pbmetallicdisplaypropertiesNodes.length; i ++ ) {
  383. const pbmetallicdisplaypropertiesNode = pbmetallicdisplaypropertiesNodes[ i ];
  384. const pbmetallicdisplaypropertiesData = parseMetallicDisplaypropertiesNode( pbmetallicdisplaypropertiesNode );
  385. resourcesData[ 'pbmetallicdisplayproperties' ][ pbmetallicdisplaypropertiesData[ 'id' ] ] = pbmetallicdisplaypropertiesData;
  386. } //
  387. resourcesData[ 'texture2dgroup' ] = {};
  388. const textures2DGroupNodes = resourcesNode.querySelectorAll( 'texture2dgroup' );
  389. for ( let i = 0; i < textures2DGroupNodes.length; i ++ ) {
  390. const textures2DGroupNode = textures2DGroupNodes[ i ];
  391. const textures2DGroupData = parseTextures2DGroupNode( textures2DGroupNode );
  392. resourcesData[ 'texture2dgroup' ][ textures2DGroupData[ 'id' ] ] = textures2DGroupData;
  393. } //
  394. resourcesData[ 'object' ] = {};
  395. const objectNodes = resourcesNode.querySelectorAll( 'object' );
  396. for ( let i = 0; i < objectNodes.length; i ++ ) {
  397. const objectNode = objectNodes[ i ];
  398. const objectData = parseObjectNode( objectNode );
  399. resourcesData[ 'object' ][ objectData[ 'id' ] ] = objectData;
  400. }
  401. return resourcesData;
  402. }
  403. function parseBuildNode( buildNode ) {
  404. const buildData = [];
  405. const itemNodes = buildNode.querySelectorAll( 'item' );
  406. for ( let i = 0; i < itemNodes.length; i ++ ) {
  407. const itemNode = itemNodes[ i ];
  408. const buildItem = {
  409. objectId: itemNode.getAttribute( 'objectid' )
  410. };
  411. const transform = itemNode.getAttribute( 'transform' );
  412. if ( transform ) {
  413. buildItem[ 'transform' ] = parseTransform( transform );
  414. }
  415. buildData.push( buildItem );
  416. }
  417. return buildData;
  418. }
  419. function parseModelNode( modelNode ) {
  420. const modelData = {
  421. unit: modelNode.getAttribute( 'unit' ) || 'millimeter'
  422. };
  423. const metadataNodes = modelNode.querySelectorAll( 'metadata' );
  424. if ( metadataNodes ) {
  425. modelData[ 'metadata' ] = parseMetadataNodes( metadataNodes );
  426. }
  427. const resourcesNode = modelNode.querySelector( 'resources' );
  428. if ( resourcesNode ) {
  429. modelData[ 'resources' ] = parseResourcesNode( resourcesNode );
  430. }
  431. const buildNode = modelNode.querySelector( 'build' );
  432. if ( buildNode ) {
  433. modelData[ 'build' ] = parseBuildNode( buildNode );
  434. }
  435. return modelData;
  436. }
  437. function buildTexture( texture2dgroup, objects, modelData, textureData ) {
  438. const texid = texture2dgroup.texid;
  439. const texture2ds = modelData.resources.texture2d;
  440. const texture2d = texture2ds[ texid ];
  441. if ( texture2d ) {
  442. const data = textureData[ texture2d.path ];
  443. const type = texture2d.contenttype;
  444. const blob = new Blob( [ data ], {
  445. type: type
  446. } );
  447. const sourceURI = URL.createObjectURL( blob );
  448. const texture = textureLoader.load( sourceURI, function () {
  449. URL.revokeObjectURL( sourceURI );
  450. } );
  451. texture.encoding = THREE.sRGBEncoding; // texture parameters
  452. switch ( texture2d.tilestyleu ) {
  453. case 'wrap':
  454. texture.wrapS = THREE.RepeatWrapping;
  455. break;
  456. case 'mirror':
  457. texture.wrapS = THREE.MirroredRepeatWrapping;
  458. break;
  459. case 'none':
  460. case 'clamp':
  461. texture.wrapS = THREE.ClampToEdgeWrapping;
  462. break;
  463. default:
  464. texture.wrapS = THREE.RepeatWrapping;
  465. }
  466. switch ( texture2d.tilestylev ) {
  467. case 'wrap':
  468. texture.wrapT = THREE.RepeatWrapping;
  469. break;
  470. case 'mirror':
  471. texture.wrapT = THREE.MirroredRepeatWrapping;
  472. break;
  473. case 'none':
  474. case 'clamp':
  475. texture.wrapT = THREE.ClampToEdgeWrapping;
  476. break;
  477. default:
  478. texture.wrapT = THREE.RepeatWrapping;
  479. }
  480. switch ( texture2d.filter ) {
  481. case 'auto':
  482. texture.magFilter = THREE.LinearFilter;
  483. texture.minFilter = THREE.LinearMipmapLinearFilter;
  484. break;
  485. case 'linear':
  486. texture.magFilter = THREE.LinearFilter;
  487. texture.minFilter = THREE.LinearFilter;
  488. break;
  489. case 'nearest':
  490. texture.magFilter = THREE.NearestFilter;
  491. texture.minFilter = THREE.NearestFilter;
  492. break;
  493. default:
  494. texture.magFilter = THREE.LinearFilter;
  495. texture.minFilter = THREE.LinearMipmapLinearFilter;
  496. }
  497. return texture;
  498. } else {
  499. return null;
  500. }
  501. }
  502. function buildBasematerialsMeshes( basematerials, triangleProperties, meshData, objects, modelData, textureData, objectData ) {
  503. const objectPindex = objectData.pindex;
  504. const materialMap = {};
  505. for ( let i = 0, l = triangleProperties.length; i < l; i ++ ) {
  506. const triangleProperty = triangleProperties[ i ];
  507. const pindex = triangleProperty.p1 !== undefined ? triangleProperty.p1 : objectPindex;
  508. if ( materialMap[ pindex ] === undefined ) materialMap[ pindex ] = [];
  509. materialMap[ pindex ].push( triangleProperty );
  510. } //
  511. const keys = Object.keys( materialMap );
  512. const meshes = [];
  513. for ( let i = 0, l = keys.length; i < l; i ++ ) {
  514. const materialIndex = keys[ i ];
  515. const trianglePropertiesProps = materialMap[ materialIndex ];
  516. const basematerialData = basematerials.basematerials[ materialIndex ];
  517. const material = getBuild( basematerialData, objects, modelData, textureData, objectData, buildBasematerial ); //
  518. const geometry = new THREE.BufferGeometry();
  519. const positionData = [];
  520. const vertices = meshData.vertices;
  521. for ( let j = 0, jl = trianglePropertiesProps.length; j < jl; j ++ ) {
  522. const triangleProperty = trianglePropertiesProps[ j ];
  523. positionData.push( vertices[ triangleProperty.v1 * 3 + 0 ] );
  524. positionData.push( vertices[ triangleProperty.v1 * 3 + 1 ] );
  525. positionData.push( vertices[ triangleProperty.v1 * 3 + 2 ] );
  526. positionData.push( vertices[ triangleProperty.v2 * 3 + 0 ] );
  527. positionData.push( vertices[ triangleProperty.v2 * 3 + 1 ] );
  528. positionData.push( vertices[ triangleProperty.v2 * 3 + 2 ] );
  529. positionData.push( vertices[ triangleProperty.v3 * 3 + 0 ] );
  530. positionData.push( vertices[ triangleProperty.v3 * 3 + 1 ] );
  531. positionData.push( vertices[ triangleProperty.v3 * 3 + 2 ] );
  532. }
  533. geometry.setAttribute( 'position', new THREE.Float32BufferAttribute( positionData, 3 ) ); //
  534. const mesh = new THREE.Mesh( geometry, material );
  535. meshes.push( mesh );
  536. }
  537. return meshes;
  538. }
  539. function buildTexturedMesh( texture2dgroup, triangleProperties, meshData, objects, modelData, textureData, objectData ) {
  540. // geometry
  541. const geometry = new THREE.BufferGeometry();
  542. const positionData = [];
  543. const uvData = [];
  544. const vertices = meshData.vertices;
  545. const uvs = texture2dgroup.uvs;
  546. for ( let i = 0, l = triangleProperties.length; i < l; i ++ ) {
  547. const triangleProperty = triangleProperties[ i ];
  548. positionData.push( vertices[ triangleProperty.v1 * 3 + 0 ] );
  549. positionData.push( vertices[ triangleProperty.v1 * 3 + 1 ] );
  550. positionData.push( vertices[ triangleProperty.v1 * 3 + 2 ] );
  551. positionData.push( vertices[ triangleProperty.v2 * 3 + 0 ] );
  552. positionData.push( vertices[ triangleProperty.v2 * 3 + 1 ] );
  553. positionData.push( vertices[ triangleProperty.v2 * 3 + 2 ] );
  554. positionData.push( vertices[ triangleProperty.v3 * 3 + 0 ] );
  555. positionData.push( vertices[ triangleProperty.v3 * 3 + 1 ] );
  556. positionData.push( vertices[ triangleProperty.v3 * 3 + 2 ] ); //
  557. uvData.push( uvs[ triangleProperty.p1 * 2 + 0 ] );
  558. uvData.push( uvs[ triangleProperty.p1 * 2 + 1 ] );
  559. uvData.push( uvs[ triangleProperty.p2 * 2 + 0 ] );
  560. uvData.push( uvs[ triangleProperty.p2 * 2 + 1 ] );
  561. uvData.push( uvs[ triangleProperty.p3 * 2 + 0 ] );
  562. uvData.push( uvs[ triangleProperty.p3 * 2 + 1 ] );
  563. }
  564. geometry.setAttribute( 'position', new THREE.Float32BufferAttribute( positionData, 3 ) );
  565. geometry.setAttribute( 'uv', new THREE.Float32BufferAttribute( uvData, 2 ) ); // material
  566. const texture = getBuild( texture2dgroup, objects, modelData, textureData, objectData, buildTexture );
  567. const material = new THREE.MeshPhongMaterial( {
  568. map: texture,
  569. flatShading: true
  570. } ); // mesh
  571. const mesh = new THREE.Mesh( geometry, material );
  572. return mesh;
  573. }
  574. function buildVertexColorMesh( colorgroup, triangleProperties, meshData, objectData ) {
  575. // geometry
  576. const geometry = new THREE.BufferGeometry();
  577. const positionData = [];
  578. const colorData = [];
  579. const vertices = meshData.vertices;
  580. const colors = colorgroup.colors;
  581. for ( let i = 0, l = triangleProperties.length; i < l; i ++ ) {
  582. const triangleProperty = triangleProperties[ i ];
  583. const v1 = triangleProperty.v1;
  584. const v2 = triangleProperty.v2;
  585. const v3 = triangleProperty.v3;
  586. positionData.push( vertices[ v1 * 3 + 0 ] );
  587. positionData.push( vertices[ v1 * 3 + 1 ] );
  588. positionData.push( vertices[ v1 * 3 + 2 ] );
  589. positionData.push( vertices[ v2 * 3 + 0 ] );
  590. positionData.push( vertices[ v2 * 3 + 1 ] );
  591. positionData.push( vertices[ v2 * 3 + 2 ] );
  592. positionData.push( vertices[ v3 * 3 + 0 ] );
  593. positionData.push( vertices[ v3 * 3 + 1 ] );
  594. positionData.push( vertices[ v3 * 3 + 2 ] ); //
  595. const p1 = triangleProperty.p1 !== undefined ? triangleProperty.p1 : objectData.pindex;
  596. const p2 = triangleProperty.p2 !== undefined ? triangleProperty.p2 : p1;
  597. const p3 = triangleProperty.p3 !== undefined ? triangleProperty.p3 : p1;
  598. colorData.push( colors[ p1 * 3 + 0 ] );
  599. colorData.push( colors[ p1 * 3 + 1 ] );
  600. colorData.push( colors[ p1 * 3 + 2 ] );
  601. colorData.push( colors[ p2 * 3 + 0 ] );
  602. colorData.push( colors[ p2 * 3 + 1 ] );
  603. colorData.push( colors[ p2 * 3 + 2 ] );
  604. colorData.push( colors[ p3 * 3 + 0 ] );
  605. colorData.push( colors[ p3 * 3 + 1 ] );
  606. colorData.push( colors[ p3 * 3 + 2 ] );
  607. }
  608. geometry.setAttribute( 'position', new THREE.Float32BufferAttribute( positionData, 3 ) );
  609. geometry.setAttribute( 'color', new THREE.Float32BufferAttribute( colorData, 3 ) ); // material
  610. const material = new THREE.MeshPhongMaterial( {
  611. vertexColors: true,
  612. flatShading: true
  613. } ); // mesh
  614. const mesh = new THREE.Mesh( geometry, material );
  615. return mesh;
  616. }
  617. function buildDefaultMesh( meshData ) {
  618. const geometry = new THREE.BufferGeometry();
  619. geometry.setIndex( new THREE.BufferAttribute( meshData[ 'triangles' ], 1 ) );
  620. geometry.setAttribute( 'position', new THREE.BufferAttribute( meshData[ 'vertices' ], 3 ) );
  621. const material = new THREE.MeshPhongMaterial( {
  622. color: 0xffffff,
  623. flatShading: true
  624. } );
  625. const mesh = new THREE.Mesh( geometry, material );
  626. return mesh;
  627. }
  628. function buildMeshes( resourceMap, meshData, objects, modelData, textureData, objectData ) {
  629. const keys = Object.keys( resourceMap );
  630. const meshes = [];
  631. for ( let i = 0, il = keys.length; i < il; i ++ ) {
  632. const resourceId = keys[ i ];
  633. const triangleProperties = resourceMap[ resourceId ];
  634. const resourceType = getResourceType( resourceId, modelData );
  635. switch ( resourceType ) {
  636. case 'material':
  637. const basematerials = modelData.resources.basematerials[ resourceId ];
  638. const newMeshes = buildBasematerialsMeshes( basematerials, triangleProperties, meshData, objects, modelData, textureData, objectData );
  639. for ( let j = 0, jl = newMeshes.length; j < jl; j ++ ) {
  640. meshes.push( newMeshes[ j ] );
  641. }
  642. break;
  643. case 'texture':
  644. const texture2dgroup = modelData.resources.texture2dgroup[ resourceId ];
  645. meshes.push( buildTexturedMesh( texture2dgroup, triangleProperties, meshData, objects, modelData, textureData, objectData ) );
  646. break;
  647. case 'vertexColors':
  648. const colorgroup = modelData.resources.colorgroup[ resourceId ];
  649. meshes.push( buildVertexColorMesh( colorgroup, triangleProperties, meshData, objectData ) );
  650. break;
  651. case 'default':
  652. meshes.push( buildDefaultMesh( meshData ) );
  653. break;
  654. default:
  655. console.error( 'THREE.3MFLoader: Unsupported resource type.' );
  656. }
  657. }
  658. if ( objectData.name ) {
  659. for ( let i = 0; i < meshes.length; i ++ ) {
  660. meshes[ i ].name = objectData.name;
  661. }
  662. }
  663. return meshes;
  664. }
  665. function getResourceType( pid, modelData ) {
  666. if ( modelData.resources.texture2dgroup[ pid ] !== undefined ) {
  667. return 'texture';
  668. } else if ( modelData.resources.basematerials[ pid ] !== undefined ) {
  669. return 'material';
  670. } else if ( modelData.resources.colorgroup[ pid ] !== undefined ) {
  671. return 'vertexColors';
  672. } else if ( pid === 'default' ) {
  673. return 'default';
  674. } else {
  675. return undefined;
  676. }
  677. }
  678. function analyzeObject( meshData, objectData ) {
  679. const resourceMap = {};
  680. const triangleProperties = meshData[ 'triangleProperties' ];
  681. const objectPid = objectData.pid;
  682. for ( let i = 0, l = triangleProperties.length; i < l; i ++ ) {
  683. const triangleProperty = triangleProperties[ i ];
  684. let pid = triangleProperty.pid !== undefined ? triangleProperty.pid : objectPid;
  685. if ( pid === undefined ) pid = 'default';
  686. if ( resourceMap[ pid ] === undefined ) resourceMap[ pid ] = [];
  687. resourceMap[ pid ].push( triangleProperty );
  688. }
  689. return resourceMap;
  690. }
  691. function buildGroup( meshData, objects, modelData, textureData, objectData ) {
  692. const group = new THREE.Group();
  693. const resourceMap = analyzeObject( meshData, objectData );
  694. const meshes = buildMeshes( resourceMap, meshData, objects, modelData, textureData, objectData );
  695. for ( let i = 0, l = meshes.length; i < l; i ++ ) {
  696. group.add( meshes[ i ] );
  697. }
  698. return group;
  699. }
  700. function applyExtensions( extensions, meshData, modelXml ) {
  701. if ( ! extensions ) {
  702. return;
  703. }
  704. const availableExtensions = [];
  705. const keys = Object.keys( extensions );
  706. for ( let i = 0; i < keys.length; i ++ ) {
  707. const ns = keys[ i ];
  708. for ( let j = 0; j < scope.availableExtensions.length; j ++ ) {
  709. const extension = scope.availableExtensions[ j ];
  710. if ( extension.ns === ns ) {
  711. availableExtensions.push( extension );
  712. }
  713. }
  714. }
  715. for ( let i = 0; i < availableExtensions.length; i ++ ) {
  716. const extension = availableExtensions[ i ];
  717. extension.apply( modelXml, extensions[ extension[ 'ns' ] ], meshData );
  718. }
  719. }
  720. function getBuild( data, objects, modelData, textureData, objectData, builder ) {
  721. if ( data.build !== undefined ) return data.build;
  722. data.build = builder( data, objects, modelData, textureData, objectData );
  723. return data.build;
  724. }
  725. function buildBasematerial( materialData, objects, modelData ) {
  726. let material;
  727. const displaypropertiesid = materialData.displaypropertiesid;
  728. const pbmetallicdisplayproperties = modelData.resources.pbmetallicdisplayproperties;
  729. if ( displaypropertiesid !== null && pbmetallicdisplayproperties[ displaypropertiesid ] !== undefined ) {
  730. // metallic display property, use StandardMaterial
  731. const pbmetallicdisplayproperty = pbmetallicdisplayproperties[ displaypropertiesid ];
  732. const metallicData = pbmetallicdisplayproperty.data[ materialData.index ];
  733. material = new THREE.MeshStandardMaterial( {
  734. flatShading: true,
  735. roughness: metallicData.roughness,
  736. metalness: metallicData.metallicness
  737. } );
  738. } else {
  739. // otherwise use PhongMaterial
  740. material = new THREE.MeshPhongMaterial( {
  741. flatShading: true
  742. } );
  743. }
  744. material.name = materialData.name; // displaycolor MUST be specified with a value of a 6 or 8 digit hexadecimal number, e.g. "#RRGGBB" or "#RRGGBBAA"
  745. const displaycolor = materialData.displaycolor;
  746. const color = displaycolor.substring( 0, 7 );
  747. material.color.setStyle( color );
  748. material.color.convertSRGBToLinear(); // displaycolor is in sRGB
  749. // process alpha if set
  750. if ( displaycolor.length === 9 ) {
  751. material.opacity = parseInt( displaycolor.charAt( 7 ) + displaycolor.charAt( 8 ), 16 ) / 255;
  752. }
  753. return material;
  754. }
  755. function buildComposite( compositeData, objects, modelData, textureData ) {
  756. const composite = new THREE.Group();
  757. for ( let j = 0; j < compositeData.length; j ++ ) {
  758. const component = compositeData[ j ];
  759. let build = objects[ component.objectId ];
  760. if ( build === undefined ) {
  761. buildObject( component.objectId, objects, modelData, textureData );
  762. build = objects[ component.objectId ];
  763. }
  764. const object3D = build.clone(); // apply component transform
  765. const transform = component.transform;
  766. if ( transform ) {
  767. object3D.applyMatrix4( transform );
  768. }
  769. composite.add( object3D );
  770. }
  771. return composite;
  772. }
  773. function buildObject( objectId, objects, modelData, textureData ) {
  774. const objectData = modelData[ 'resources' ][ 'object' ][ objectId ];
  775. if ( objectData[ 'mesh' ] ) {
  776. const meshData = objectData[ 'mesh' ];
  777. const extensions = modelData[ 'extensions' ];
  778. const modelXml = modelData[ 'xml' ];
  779. applyExtensions( extensions, meshData, modelXml );
  780. objects[ objectData.id ] = getBuild( meshData, objects, modelData, textureData, objectData, buildGroup );
  781. } else {
  782. const compositeData = objectData[ 'components' ];
  783. objects[ objectData.id ] = getBuild( compositeData, objects, modelData, textureData, objectData, buildComposite );
  784. }
  785. if ( objectData.name ) {
  786. objects[ objectData.id ].name = objectData.name;
  787. }
  788. }
  789. function buildObjects( data3mf ) {
  790. const modelsData = data3mf.model;
  791. const modelRels = data3mf.modelRels;
  792. const objects = {};
  793. const modelsKeys = Object.keys( modelsData );
  794. const textureData = {}; // evaluate model relationships to textures
  795. if ( modelRels ) {
  796. for ( let i = 0, l = modelRels.length; i < l; i ++ ) {
  797. const modelRel = modelRels[ i ];
  798. const textureKey = modelRel.target.substring( 1 );
  799. if ( data3mf.texture[ textureKey ] ) {
  800. textureData[ modelRel.target ] = data3mf.texture[ textureKey ];
  801. }
  802. }
  803. } // start build
  804. for ( let i = 0; i < modelsKeys.length; i ++ ) {
  805. const modelsKey = modelsKeys[ i ];
  806. const modelData = modelsData[ modelsKey ];
  807. const objectIds = Object.keys( modelData[ 'resources' ][ 'object' ] );
  808. for ( let j = 0; j < objectIds.length; j ++ ) {
  809. const objectId = objectIds[ j ];
  810. buildObject( objectId, objects, modelData, textureData );
  811. }
  812. }
  813. return objects;
  814. }
  815. function fetch3DModelPart( rels ) {
  816. for ( let i = 0; i < rels.length; i ++ ) {
  817. const rel = rels[ i ];
  818. const extension = rel.target.split( '.' ).pop();
  819. if ( extension.toLowerCase() === 'model' ) return rel;
  820. }
  821. }
  822. function build( objects, data3mf ) {
  823. const group = new THREE.Group();
  824. const relationship = fetch3DModelPart( data3mf[ 'rels' ] );
  825. const buildData = data3mf.model[ relationship[ 'target' ].substring( 1 ) ][ 'build' ];
  826. for ( let i = 0; i < buildData.length; i ++ ) {
  827. const buildItem = buildData[ i ];
  828. const object3D = objects[ buildItem[ 'objectId' ] ].clone(); // apply transform
  829. const transform = buildItem[ 'transform' ];
  830. if ( transform ) {
  831. object3D.applyMatrix4( transform );
  832. }
  833. group.add( object3D );
  834. }
  835. return group;
  836. }
  837. const data3mf = loadDocument( data );
  838. const objects = buildObjects( data3mf );
  839. return build( objects, data3mf );
  840. }
  841. addExtension( extension ) {
  842. this.availableExtensions.push( extension );
  843. }
  844. }
  845. THREE.ThreeMFLoader = ThreeMFLoader;
  846. } )();