OBJExporter.js 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313
  1. import {
  2. Color,
  3. Matrix3,
  4. Vector2,
  5. Vector3
  6. } from '../../../build/three.module.js';
  7. var OBJExporter = function () {};
  8. OBJExporter.prototype = {
  9. constructor: OBJExporter,
  10. parse: function ( object ) {
  11. var output = '';
  12. var indexVertex = 0;
  13. var indexVertexUvs = 0;
  14. var indexNormals = 0;
  15. var vertex = new Vector3();
  16. var color = new Color();
  17. var normal = new Vector3();
  18. var uv = new Vector2();
  19. var i, j, k, l, m, face = [];
  20. var parseMesh = function ( mesh ) {
  21. var nbVertex = 0;
  22. var nbNormals = 0;
  23. var nbVertexUvs = 0;
  24. var geometry = mesh.geometry;
  25. var normalMatrixWorld = new Matrix3();
  26. if ( geometry.isBufferGeometry !== true ) {
  27. throw new Error( 'THREE.OBJExporter: Geometry is not of type THREE.BufferGeometry.' );
  28. }
  29. // shortcuts
  30. var vertices = geometry.getAttribute( 'position' );
  31. var normals = geometry.getAttribute( 'normal' );
  32. var uvs = geometry.getAttribute( 'uv' );
  33. var indices = geometry.getIndex();
  34. // name of the mesh object
  35. output += 'o ' + mesh.name + '\n';
  36. // name of the mesh material
  37. if ( mesh.material && mesh.material.name ) {
  38. output += 'usemtl ' + mesh.material.name + '\n';
  39. }
  40. // vertices
  41. if ( vertices !== undefined ) {
  42. for ( i = 0, l = vertices.count; i < l; i ++, nbVertex ++ ) {
  43. vertex.x = vertices.getX( i );
  44. vertex.y = vertices.getY( i );
  45. vertex.z = vertices.getZ( i );
  46. // transform the vertex to world space
  47. vertex.applyMatrix4( mesh.matrixWorld );
  48. // transform the vertex to export format
  49. output += 'v ' + vertex.x + ' ' + vertex.y + ' ' + vertex.z + '\n';
  50. }
  51. }
  52. // uvs
  53. if ( uvs !== undefined ) {
  54. for ( i = 0, l = uvs.count; i < l; i ++, nbVertexUvs ++ ) {
  55. uv.x = uvs.getX( i );
  56. uv.y = uvs.getY( i );
  57. // transform the uv to export format
  58. output += 'vt ' + uv.x + ' ' + uv.y + '\n';
  59. }
  60. }
  61. // normals
  62. if ( normals !== undefined ) {
  63. normalMatrixWorld.getNormalMatrix( mesh.matrixWorld );
  64. for ( i = 0, l = normals.count; i < l; i ++, nbNormals ++ ) {
  65. normal.x = normals.getX( i );
  66. normal.y = normals.getY( i );
  67. normal.z = normals.getZ( i );
  68. // transform the normal to world space
  69. normal.applyMatrix3( normalMatrixWorld ).normalize();
  70. // transform the normal to export format
  71. output += 'vn ' + normal.x + ' ' + normal.y + ' ' + normal.z + '\n';
  72. }
  73. }
  74. // faces
  75. if ( indices !== null ) {
  76. for ( i = 0, l = indices.count; i < l; i += 3 ) {
  77. for ( m = 0; m < 3; m ++ ) {
  78. j = indices.getX( i + m ) + 1;
  79. face[ m ] = ( indexVertex + j ) + ( normals || uvs ? '/' + ( uvs ? ( indexVertexUvs + j ) : '' ) + ( normals ? '/' + ( indexNormals + j ) : '' ) : '' );
  80. }
  81. // transform the face to export format
  82. output += 'f ' + face.join( ' ' ) + '\n';
  83. }
  84. } else {
  85. for ( i = 0, l = vertices.count; i < l; i += 3 ) {
  86. for ( m = 0; m < 3; m ++ ) {
  87. j = i + m + 1;
  88. face[ m ] = ( indexVertex + j ) + ( normals || uvs ? '/' + ( uvs ? ( indexVertexUvs + j ) : '' ) + ( normals ? '/' + ( indexNormals + j ) : '' ) : '' );
  89. }
  90. // transform the face to export format
  91. output += 'f ' + face.join( ' ' ) + '\n';
  92. }
  93. }
  94. // update index
  95. indexVertex += nbVertex;
  96. indexVertexUvs += nbVertexUvs;
  97. indexNormals += nbNormals;
  98. };
  99. var parseLine = function ( line ) {
  100. var nbVertex = 0;
  101. var geometry = line.geometry;
  102. var type = line.type;
  103. if ( geometry.isBufferGeometry !== true ) {
  104. throw new Error( 'THREE.OBJExporter: Geometry is not of type THREE.BufferGeometry.' );
  105. }
  106. // shortcuts
  107. var vertices = geometry.getAttribute( 'position' );
  108. // name of the line object
  109. output += 'o ' + line.name + '\n';
  110. if ( vertices !== undefined ) {
  111. for ( i = 0, l = vertices.count; i < l; i ++, nbVertex ++ ) {
  112. vertex.x = vertices.getX( i );
  113. vertex.y = vertices.getY( i );
  114. vertex.z = vertices.getZ( i );
  115. // transform the vertex to world space
  116. vertex.applyMatrix4( line.matrixWorld );
  117. // transform the vertex to export format
  118. output += 'v ' + vertex.x + ' ' + vertex.y + ' ' + vertex.z + '\n';
  119. }
  120. }
  121. if ( type === 'Line' ) {
  122. output += 'l ';
  123. for ( j = 1, l = vertices.count; j <= l; j ++ ) {
  124. output += ( indexVertex + j ) + ' ';
  125. }
  126. output += '\n';
  127. }
  128. if ( type === 'LineSegments' ) {
  129. for ( j = 1, k = j + 1, l = vertices.count; j < l; j += 2, k = j + 1 ) {
  130. output += 'l ' + ( indexVertex + j ) + ' ' + ( indexVertex + k ) + '\n';
  131. }
  132. }
  133. // update index
  134. indexVertex += nbVertex;
  135. };
  136. var parsePoints = function ( points ) {
  137. var nbVertex = 0;
  138. var geometry = points.geometry;
  139. if ( geometry.isBufferGeometry !== true ) {
  140. throw new Error( 'THREE.OBJExporter: Geometry is not of type THREE.BufferGeometry.' );
  141. }
  142. var vertices = geometry.getAttribute( 'position' );
  143. var colors = geometry.getAttribute( 'color' );
  144. output += 'o ' + points.name + '\n';
  145. if ( vertices !== undefined ) {
  146. for ( i = 0, l = vertices.count; i < l; i ++, nbVertex ++ ) {
  147. vertex.fromBufferAttribute( vertices, i );
  148. vertex.applyMatrix4( points.matrixWorld );
  149. output += 'v ' + vertex.x + ' ' + vertex.y + ' ' + vertex.z;
  150. if ( colors !== undefined ) {
  151. color.fromBufferAttribute( colors, i );
  152. output += ' ' + color.r + ' ' + color.g + ' ' + color.b;
  153. }
  154. output += '\n';
  155. }
  156. }
  157. output += 'p ';
  158. for ( j = 1, l = vertices.count; j <= l; j ++ ) {
  159. output += ( indexVertex + j ) + ' ';
  160. }
  161. output += '\n';
  162. // update index
  163. indexVertex += nbVertex;
  164. };
  165. object.traverse( function ( child ) {
  166. if ( child.isMesh === true ) {
  167. parseMesh( child );
  168. }
  169. if ( child.isLine === true ) {
  170. parseLine( child );
  171. }
  172. if ( child.isPoints === true ) {
  173. parsePoints( child );
  174. }
  175. } );
  176. return output;
  177. }
  178. };
  179. export { OBJExporter };