3MFLoader.js 36 KB

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