VOXLoader.js 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248
  1. ( function () {
  2. class VOXLoader extends THREE.Loader {
  3. load( url, onLoad, onProgress, onError ) {
  4. const scope = this;
  5. const loader = new THREE.FileLoader( scope.manager );
  6. loader.setPath( scope.path );
  7. loader.setResponseType( 'arraybuffer' );
  8. loader.setRequestHeader( scope.requestHeader );
  9. loader.load( url, function ( buffer ) {
  10. try {
  11. onLoad( scope.parse( buffer ) );
  12. } catch ( e ) {
  13. if ( onError ) {
  14. onError( e );
  15. } else {
  16. console.error( e );
  17. }
  18. scope.manager.itemError( url );
  19. }
  20. }, onProgress, onError );
  21. }
  22. parse( buffer ) {
  23. const data = new DataView( buffer );
  24. const id = data.getUint32( 0, true );
  25. const version = data.getUint32( 4, true );
  26. if ( id !== 542658390 || version !== 150 ) {
  27. console.error( 'Not a valid VOX file' );
  28. return;
  29. }
  30. const DEFAULT_PALETTE = [ 0x00000000, 0xffffffff, 0xffccffff, 0xff99ffff, 0xff66ffff, 0xff33ffff, 0xff00ffff, 0xffffccff, 0xffccccff, 0xff99ccff, 0xff66ccff, 0xff33ccff, 0xff00ccff, 0xffff99ff, 0xffcc99ff, 0xff9999ff, 0xff6699ff, 0xff3399ff, 0xff0099ff, 0xffff66ff, 0xffcc66ff, 0xff9966ff, 0xff6666ff, 0xff3366ff, 0xff0066ff, 0xffff33ff, 0xffcc33ff, 0xff9933ff, 0xff6633ff, 0xff3333ff, 0xff0033ff, 0xffff00ff, 0xffcc00ff, 0xff9900ff, 0xff6600ff, 0xff3300ff, 0xff0000ff, 0xffffffcc, 0xffccffcc, 0xff99ffcc, 0xff66ffcc, 0xff33ffcc, 0xff00ffcc, 0xffffcccc, 0xffcccccc, 0xff99cccc, 0xff66cccc, 0xff33cccc, 0xff00cccc, 0xffff99cc, 0xffcc99cc, 0xff9999cc, 0xff6699cc, 0xff3399cc, 0xff0099cc, 0xffff66cc, 0xffcc66cc, 0xff9966cc, 0xff6666cc, 0xff3366cc, 0xff0066cc, 0xffff33cc, 0xffcc33cc, 0xff9933cc, 0xff6633cc, 0xff3333cc, 0xff0033cc, 0xffff00cc, 0xffcc00cc, 0xff9900cc, 0xff6600cc, 0xff3300cc, 0xff0000cc, 0xffffff99, 0xffccff99, 0xff99ff99, 0xff66ff99, 0xff33ff99, 0xff00ff99, 0xffffcc99, 0xffcccc99, 0xff99cc99, 0xff66cc99, 0xff33cc99, 0xff00cc99, 0xffff9999, 0xffcc9999, 0xff999999, 0xff669999, 0xff339999, 0xff009999, 0xffff6699, 0xffcc6699, 0xff996699, 0xff666699, 0xff336699, 0xff006699, 0xffff3399, 0xffcc3399, 0xff993399, 0xff663399, 0xff333399, 0xff003399, 0xffff0099, 0xffcc0099, 0xff990099, 0xff660099, 0xff330099, 0xff000099, 0xffffff66, 0xffccff66, 0xff99ff66, 0xff66ff66, 0xff33ff66, 0xff00ff66, 0xffffcc66, 0xffcccc66, 0xff99cc66, 0xff66cc66, 0xff33cc66, 0xff00cc66, 0xffff9966, 0xffcc9966, 0xff999966, 0xff669966, 0xff339966, 0xff009966, 0xffff6666, 0xffcc6666, 0xff996666, 0xff666666, 0xff336666, 0xff006666, 0xffff3366, 0xffcc3366, 0xff993366, 0xff663366, 0xff333366, 0xff003366, 0xffff0066, 0xffcc0066, 0xff990066, 0xff660066, 0xff330066, 0xff000066, 0xffffff33, 0xffccff33, 0xff99ff33, 0xff66ff33, 0xff33ff33, 0xff00ff33, 0xffffcc33, 0xffcccc33, 0xff99cc33, 0xff66cc33, 0xff33cc33, 0xff00cc33, 0xffff9933, 0xffcc9933, 0xff999933, 0xff669933, 0xff339933, 0xff009933, 0xffff6633, 0xffcc6633, 0xff996633, 0xff666633, 0xff336633, 0xff006633, 0xffff3333, 0xffcc3333, 0xff993333, 0xff663333, 0xff333333, 0xff003333, 0xffff0033, 0xffcc0033, 0xff990033, 0xff660033, 0xff330033, 0xff000033, 0xffffff00, 0xffccff00, 0xff99ff00, 0xff66ff00, 0xff33ff00, 0xff00ff00, 0xffffcc00, 0xffcccc00, 0xff99cc00, 0xff66cc00, 0xff33cc00, 0xff00cc00, 0xffff9900, 0xffcc9900, 0xff999900, 0xff669900, 0xff339900, 0xff009900, 0xffff6600, 0xffcc6600, 0xff996600, 0xff666600, 0xff336600, 0xff006600, 0xffff3300, 0xffcc3300, 0xff993300, 0xff663300, 0xff333300, 0xff003300, 0xffff0000, 0xffcc0000, 0xff990000, 0xff660000, 0xff330000, 0xff0000ee, 0xff0000dd, 0xff0000bb, 0xff0000aa, 0xff000088, 0xff000077, 0xff000055, 0xff000044, 0xff000022, 0xff000011, 0xff00ee00, 0xff00dd00, 0xff00bb00, 0xff00aa00, 0xff008800, 0xff007700, 0xff005500, 0xff004400, 0xff002200, 0xff001100, 0xffee0000, 0xffdd0000, 0xffbb0000, 0xffaa0000, 0xff880000, 0xff770000, 0xff550000, 0xff440000, 0xff220000, 0xff110000, 0xffeeeeee, 0xffdddddd, 0xffbbbbbb, 0xffaaaaaa, 0xff888888, 0xff777777, 0xff555555, 0xff444444, 0xff222222, 0xff111111 ];
  31. let i = 8;
  32. let chunk;
  33. const chunks = [];
  34. while ( i < data.byteLength ) {
  35. let id = '';
  36. for ( let j = 0; j < 4; j ++ ) {
  37. id += String.fromCharCode( data.getUint8( i ++, true ) );
  38. }
  39. const chunkSize = data.getUint32( i, true );
  40. i += 4;
  41. data.getUint32( i, true );
  42. i += 4; // childChunks
  43. if ( id === 'SIZE' ) {
  44. const x = data.getUint32( i, true );
  45. i += 4;
  46. const y = data.getUint32( i, true );
  47. i += 4;
  48. const z = data.getUint32( i, true );
  49. i += 4;
  50. chunk = {
  51. palette: DEFAULT_PALETTE,
  52. size: {
  53. x: x,
  54. y: y,
  55. z: z
  56. }
  57. };
  58. chunks.push( chunk );
  59. i += chunkSize - 3 * 4;
  60. } else if ( id === 'XYZI' ) {
  61. const numVoxels = data.getUint32( i, true );
  62. i += 4;
  63. chunk.data = new Uint8Array( buffer, i, numVoxels * 4 );
  64. i += numVoxels * 4;
  65. } else if ( id === 'RGBA' ) {
  66. const palette = [ 0 ];
  67. for ( let j = 0; j < 256; j ++ ) {
  68. palette[ j + 1 ] = data.getUint32( i, true );
  69. i += 4;
  70. }
  71. chunk.palette = palette;
  72. } else {
  73. // console.log( id, chunkSize, childChunks );
  74. i += chunkSize;
  75. }
  76. }
  77. return chunks;
  78. }
  79. }
  80. class VOXMesh extends THREE.Mesh {
  81. constructor( chunk ) {
  82. const data = chunk.data;
  83. const size = chunk.size;
  84. const palette = chunk.palette; //
  85. const vertices = [];
  86. const colors = [];
  87. const nx = [ 0, 0, 0, 0, 0, 1, 0, 1, 0, 0, 1, 1, 0, 1, 0, 0, 0, 1 ];
  88. const px = [ 1, 0, 0, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0 ];
  89. const py = [ 0, 0, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1 ];
  90. const ny = [ 0, 0, 0, 0, 1, 0, 1, 0, 0, 1, 1, 0, 1, 0, 0, 0, 1, 0 ];
  91. const nz = [ 0, 0, 1, 0, 0, 0, 1, 0, 1, 1, 0, 0, 1, 0, 1, 0, 0, 0 ];
  92. const pz = [ 0, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 0, 0, 1, 0, 1, 1, 1 ];
  93. function add( tile, x, y, z, r, g, b ) {
  94. x -= size.x / 2;
  95. y -= size.z / 2;
  96. z += size.y / 2;
  97. for ( let i = 0; i < 18; i += 3 ) {
  98. vertices.push( tile[ i + 0 ] + x, tile[ i + 1 ] + y, tile[ i + 2 ] + z );
  99. colors.push( r, g, b );
  100. }
  101. } // Store data in a volume for sampling
  102. const offsety = size.x;
  103. const offsetz = size.x * size.y;
  104. const array = new Uint8Array( size.x * size.y * size.z );
  105. for ( let j = 0; j < data.length; j += 4 ) {
  106. const x = data[ j + 0 ];
  107. const y = data[ j + 1 ];
  108. const z = data[ j + 2 ];
  109. const index = x + y * offsety + z * offsetz;
  110. array[ index ] = 255;
  111. } // Construct geometry
  112. let hasColors = false;
  113. for ( let j = 0; j < data.length; j += 4 ) {
  114. const x = data[ j + 0 ];
  115. const y = data[ j + 1 ];
  116. const z = data[ j + 2 ];
  117. const c = data[ j + 3 ];
  118. const hex = palette[ c ];
  119. const r = ( hex >> 0 & 0xff ) / 0xff;
  120. const g = ( hex >> 8 & 0xff ) / 0xff;
  121. const b = ( hex >> 16 & 0xff ) / 0xff;
  122. if ( r > 0 || g > 0 || b > 0 ) hasColors = true;
  123. const index = x + y * offsety + z * offsetz;
  124. if ( array[ index + 1 ] === 0 || x === size.x - 1 ) add( px, x, z, - y, r, g, b );
  125. if ( array[ index - 1 ] === 0 || x === 0 ) add( nx, x, z, - y, r, g, b );
  126. if ( array[ index + offsety ] === 0 || y === size.y - 1 ) add( ny, x, z, - y, r, g, b );
  127. if ( array[ index - offsety ] === 0 || y === 0 ) add( py, x, z, - y, r, g, b );
  128. if ( array[ index + offsetz ] === 0 || z === size.z - 1 ) add( pz, x, z, - y, r, g, b );
  129. if ( array[ index - offsetz ] === 0 || z === 0 ) add( nz, x, z, - y, r, g, b );
  130. }
  131. const geometry = new THREE.BufferGeometry();
  132. geometry.setAttribute( 'position', new THREE.Float32BufferAttribute( vertices, 3 ) );
  133. geometry.computeVertexNormals();
  134. const material = new THREE.MeshStandardMaterial();
  135. if ( hasColors ) {
  136. geometry.setAttribute( 'color', new THREE.Float32BufferAttribute( colors, 3 ) );
  137. material.vertexColors = true;
  138. }
  139. super( geometry, material );
  140. }
  141. }
  142. class VOXDataTexture3D extends THREE.DataTexture3D {
  143. constructor( chunk ) {
  144. const data = chunk.data;
  145. const size = chunk.size;
  146. const offsety = size.x;
  147. const offsetz = size.x * size.y;
  148. const array = new Uint8Array( size.x * size.y * size.z );
  149. for ( let j = 0; j < data.length; j += 4 ) {
  150. const x = data[ j + 0 ];
  151. const y = data[ j + 1 ];
  152. const z = data[ j + 2 ];
  153. const index = x + y * offsety + z * offsetz;
  154. array[ index ] = 255;
  155. }
  156. super( array, size.x, size.y, size.z );
  157. this.format = THREE.RedFormat;
  158. this.minFilter = THREE.NearestFilter;
  159. this.magFilter = THREE.LinearFilter;
  160. this.unpackAlignment = 1;
  161. }
  162. }
  163. THREE.VOXDataTexture3D = VOXDataTexture3D;
  164. THREE.VOXLoader = VOXLoader;
  165. THREE.VOXMesh = VOXMesh;
  166. } )();