3MFLoader.js 37 KB

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