IFCLoader.js 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177
  1. //Example: https://github.com/tomvandig/web-ifc-three/tree/main/examples/jsm
  2. import { IfcAPI } from './ifc/web-ifc-api.js';
  3. import {
  4. FileLoader,
  5. Loader,
  6. Object3D,
  7. Mesh,
  8. Color,
  9. MeshPhongMaterial,
  10. DoubleSide,
  11. Matrix4,
  12. BufferGeometry,
  13. InterleavedBuffer,
  14. InterleavedBufferAttribute,
  15. BufferAttribute,
  16. } from '../../../build/three.module.js';
  17. var ifcAPI = new IfcAPI();
  18. function IFCLoader( manager ) {
  19. Loader.call( this, manager );
  20. }
  21. IFCLoader.prototype = Object.assign( Object.create( Loader.prototype ), {
  22. constructor: IFCLoader,
  23. load: function ( url, onLoad, onProgress, onError ) {
  24. var scope = this;
  25. var loader = new FileLoader( scope.manager );
  26. loader.setPath( scope.path );
  27. loader.setResponseType( 'arraybuffer' );
  28. loader.setRequestHeader( scope.requestHeader );
  29. loader.setWithCredentials( scope.withCredentials );
  30. loader.load(
  31. url,
  32. async function ( buffer ) {
  33. try {
  34. onLoad( await scope.parse( buffer ) );
  35. } catch ( e ) {
  36. if ( onError ) {
  37. onError( e );
  38. } else {
  39. console.error( e );
  40. }
  41. scope.manager.itemError( url );
  42. }
  43. },
  44. onProgress,
  45. onError
  46. );
  47. },
  48. parse: async function ( buffer ) {
  49. if ( ifcAPI.wasmModule === undefined ) {
  50. await ifcAPI.Init();
  51. }
  52. var data = new Uint8Array( buffer );
  53. var modelID = ifcAPI.OpenModel( 'example.ifc', data );
  54. return loadAllGeometry( modelID );
  55. function loadAllGeometry( modelID ) {
  56. var flatMeshes = getFlatMeshes( modelID );
  57. var mainObject = new Object3D();
  58. for ( var i = 0; i < flatMeshes.size(); i ++ ) {
  59. var placedGeometries = flatMeshes.get( i ).geometries;
  60. for ( var j = 0; j < placedGeometries.size(); j ++ )
  61. mainObject.add( getPlacedGeometry( modelID, placedGeometries.get( j ) ) );
  62. }
  63. return mainObject;
  64. }
  65. function getFlatMeshes( modelID ) {
  66. var flatMeshes = ifcAPI.LoadAllGeometry( modelID );
  67. return flatMeshes;
  68. }
  69. function getPlacedGeometry( modelID, placedGeometry ) {
  70. var geometry = getBufferGeometry( modelID, placedGeometry );
  71. var material = getMeshMaterial( placedGeometry.color );
  72. var mesh = new Mesh( geometry, material );
  73. mesh.matrix = getMeshMatrix( placedGeometry.flatTransformation );
  74. mesh.matrixAutoUpdate = false;
  75. return mesh;
  76. }
  77. function getBufferGeometry( modelID, placedGeometry ) {
  78. var geometry = ifcAPI.GetGeometry(
  79. modelID,
  80. placedGeometry.geometryExpressID
  81. );
  82. var verts = ifcAPI.GetVertexArray(
  83. geometry.GetVertexData(),
  84. geometry.GetVertexDataSize()
  85. );
  86. var indices = ifcAPI.GetIndexArray(
  87. geometry.GetIndexData(),
  88. geometry.GetIndexDataSize()
  89. );
  90. var bufferGeometry = ifcGeometryToBuffer( verts, indices );
  91. return bufferGeometry;
  92. }
  93. function getMeshMaterial( color ) {
  94. var col = new Color( color.x, color.y, color.z );
  95. var material = new MeshPhongMaterial( { color: col, side: DoubleSide } );
  96. material.transparent = color.w !== 1;
  97. if ( material.transparent ) material.opacity = color.w;
  98. return material;
  99. }
  100. function getMeshMatrix( matrix ) {
  101. var mat = new Matrix4();
  102. mat.fromArray( matrix );
  103. // mat.elements[15 - 3] *= 0.001;
  104. // mat.elements[15 - 2] *= 0.001;
  105. // mat.elements[15 - 1] *= 0.001;
  106. return mat;
  107. }
  108. function ifcGeometryToBuffer( vertexData, indexData ) {
  109. var geometry = new BufferGeometry();
  110. var buffer32 = new InterleavedBuffer( vertexData, 6 );
  111. geometry.setAttribute(
  112. 'position',
  113. new InterleavedBufferAttribute( buffer32, 3, 0 )
  114. );
  115. geometry.setAttribute(
  116. 'normal',
  117. new InterleavedBufferAttribute( buffer32, 3, 3 )
  118. );
  119. geometry.setIndex( new BufferAttribute( indexData, 1 ) );
  120. return geometry;
  121. }
  122. }
  123. } );
  124. export { IFCLoader };