SVGRenderer.js 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505
  1. /**
  2. * @author mrdoob / http://mrdoob.com/
  3. */
  4. THREE.SVGObject = function ( node ) {
  5. THREE.Object3D.call( this );
  6. this.node = node;
  7. };
  8. THREE.SVGObject.prototype = Object.create( THREE.Object3D.prototype );
  9. THREE.SVGObject.prototype.constructor = THREE.SVGObject;
  10. THREE.SVGRenderer = function () {
  11. console.log( 'THREE.SVGRenderer', THREE.REVISION );
  12. var _this = this,
  13. _renderData, _elements, _lights,
  14. _projector = new THREE.Projector(),
  15. _svg = document.createElementNS( 'http://www.w3.org/2000/svg', 'svg' ),
  16. _svgWidth, _svgHeight, _svgWidthHalf, _svgHeightHalf,
  17. _v1, _v2, _v3,
  18. _clipBox = new THREE.Box2(),
  19. _elemBox = new THREE.Box2(),
  20. _color = new THREE.Color(),
  21. _diffuseColor = new THREE.Color(),
  22. _ambientLight = new THREE.Color(),
  23. _directionalLights = new THREE.Color(),
  24. _pointLights = new THREE.Color(),
  25. _clearColor = new THREE.Color(),
  26. _clearAlpha = 1,
  27. _vector3 = new THREE.Vector3(), // Needed for PointLight
  28. _centroid = new THREE.Vector3(),
  29. _normal = new THREE.Vector3(),
  30. _normalViewMatrix = new THREE.Matrix3(),
  31. _viewMatrix = new THREE.Matrix4(),
  32. _viewProjectionMatrix = new THREE.Matrix4(),
  33. _svgPathPool = [],
  34. _svgNode, _pathCount = 0,
  35. _currentPath, _currentStyle,
  36. _quality = 1, _precision = null;
  37. this.domElement = _svg;
  38. this.autoClear = true;
  39. this.sortObjects = true;
  40. this.sortElements = true;
  41. this.info = {
  42. render: {
  43. vertices: 0,
  44. faces: 0
  45. }
  46. };
  47. this.setQuality = function ( quality ) {
  48. switch ( quality ) {
  49. case "high": _quality = 1; break;
  50. case "low": _quality = 0; break;
  51. }
  52. };
  53. this.setClearColor = function ( color, alpha ) {
  54. _clearColor.set( color );
  55. _clearAlpha = alpha !== undefined ? alpha : 1;
  56. };
  57. this.setPixelRatio = function () {};
  58. this.setSize = function ( width, height ) {
  59. _svgWidth = width; _svgHeight = height;
  60. _svgWidthHalf = _svgWidth / 2; _svgHeightHalf = _svgHeight / 2;
  61. _svg.setAttribute( 'viewBox', ( - _svgWidthHalf ) + ' ' + ( - _svgHeightHalf ) + ' ' + _svgWidth + ' ' + _svgHeight );
  62. _svg.setAttribute( 'width', _svgWidth );
  63. _svg.setAttribute( 'height', _svgHeight );
  64. _clipBox.min.set( - _svgWidthHalf, - _svgHeightHalf );
  65. _clipBox.max.set( _svgWidthHalf, _svgHeightHalf );
  66. };
  67. this.setPrecision = function ( precision ) {
  68. _precision = precision;
  69. };
  70. function removeChildNodes() {
  71. _pathCount = 0;
  72. while ( _svg.childNodes.length > 0 ) {
  73. _svg.removeChild( _svg.childNodes[ 0 ] );
  74. }
  75. }
  76. function getSvgColor( color, opacity ) {
  77. var arg = Math.floor( color.r * 255 ) + ',' + Math.floor( color.g * 255 ) + ',' + Math.floor( color.b * 255 );
  78. if ( opacity === undefined || opacity === 1 ) return 'rgb(' + arg + ')';
  79. return 'rgb(' + arg + '); fill-opacity: ' + opacity;
  80. }
  81. function convert( c ) {
  82. return _precision !== null ? c.toFixed( _precision ) : c;
  83. }
  84. this.clear = function () {
  85. removeChildNodes();
  86. _svg.style.backgroundColor = getSvgColor( _clearColor, _clearAlpha );
  87. };
  88. this.render = function ( scene, camera ) {
  89. if ( camera instanceof THREE.Camera === false ) {
  90. console.error( 'THREE.SVGRenderer.render: camera is not an instance of THREE.Camera.' );
  91. return;
  92. }
  93. var background = scene.background;
  94. if ( background && background.isColor ) {
  95. removeChildNodes();
  96. _svg.style.backgroundColor = getSvgColor( background );
  97. } else if ( this.autoClear === true ) {
  98. this.clear();
  99. }
  100. _this.info.render.vertices = 0;
  101. _this.info.render.faces = 0;
  102. _viewMatrix.copy( camera.matrixWorldInverse );
  103. _viewProjectionMatrix.multiplyMatrices( camera.projectionMatrix, _viewMatrix );
  104. _renderData = _projector.projectScene( scene, camera, this.sortObjects, this.sortElements );
  105. _elements = _renderData.elements;
  106. _lights = _renderData.lights;
  107. _normalViewMatrix.getNormalMatrix( camera.matrixWorldInverse );
  108. calculateLights( _lights );
  109. // reset accumulated path
  110. _currentPath = '';
  111. _currentStyle = '';
  112. for ( var e = 0, el = _elements.length; e < el; e ++ ) {
  113. var element = _elements[ e ];
  114. var material = element.material;
  115. if ( material === undefined || material.opacity === 0 ) continue;
  116. _elemBox.makeEmpty();
  117. if ( element instanceof THREE.RenderableSprite ) {
  118. _v1 = element;
  119. _v1.x *= _svgWidthHalf; _v1.y *= - _svgHeightHalf;
  120. renderSprite( _v1, element, material );
  121. } else if ( element instanceof THREE.RenderableLine ) {
  122. _v1 = element.v1; _v2 = element.v2;
  123. _v1.positionScreen.x *= _svgWidthHalf; _v1.positionScreen.y *= - _svgHeightHalf;
  124. _v2.positionScreen.x *= _svgWidthHalf; _v2.positionScreen.y *= - _svgHeightHalf;
  125. _elemBox.setFromPoints( [ _v1.positionScreen, _v2.positionScreen ] );
  126. if ( _clipBox.intersectsBox( _elemBox ) === true ) {
  127. renderLine( _v1, _v2, element, material );
  128. }
  129. } else if ( element instanceof THREE.RenderableFace ) {
  130. _v1 = element.v1; _v2 = element.v2; _v3 = element.v3;
  131. if ( _v1.positionScreen.z < - 1 || _v1.positionScreen.z > 1 ) continue;
  132. if ( _v2.positionScreen.z < - 1 || _v2.positionScreen.z > 1 ) continue;
  133. if ( _v3.positionScreen.z < - 1 || _v3.positionScreen.z > 1 ) continue;
  134. _v1.positionScreen.x *= _svgWidthHalf; _v1.positionScreen.y *= - _svgHeightHalf;
  135. _v2.positionScreen.x *= _svgWidthHalf; _v2.positionScreen.y *= - _svgHeightHalf;
  136. _v3.positionScreen.x *= _svgWidthHalf; _v3.positionScreen.y *= - _svgHeightHalf;
  137. _elemBox.setFromPoints( [
  138. _v1.positionScreen,
  139. _v2.positionScreen,
  140. _v3.positionScreen
  141. ] );
  142. if ( _clipBox.intersectsBox( _elemBox ) === true ) {
  143. renderFace3( _v1, _v2, _v3, element, material );
  144. }
  145. }
  146. }
  147. flushPath(); // just to flush last svg:path
  148. scene.traverseVisible( function ( object ) {
  149. if ( object instanceof THREE.SVGObject ) {
  150. _vector3.setFromMatrixPosition( object.matrixWorld );
  151. _vector3.applyMatrix4( _viewProjectionMatrix );
  152. var x = _vector3.x * _svgWidthHalf;
  153. var y = - _vector3.y * _svgHeightHalf;
  154. var node = object.node;
  155. node.setAttribute( 'transform', 'translate(' + x + ',' + y + ')' );
  156. _svg.appendChild( node );
  157. }
  158. } );
  159. };
  160. function calculateLights( lights ) {
  161. _ambientLight.setRGB( 0, 0, 0 );
  162. _directionalLights.setRGB( 0, 0, 0 );
  163. _pointLights.setRGB( 0, 0, 0 );
  164. for ( var l = 0, ll = lights.length; l < ll; l ++ ) {
  165. var light = lights[ l ];
  166. var lightColor = light.color;
  167. if ( light.isAmbientLight ) {
  168. _ambientLight.r += lightColor.r;
  169. _ambientLight.g += lightColor.g;
  170. _ambientLight.b += lightColor.b;
  171. } else if ( light.isDirectionalLight ) {
  172. _directionalLights.r += lightColor.r;
  173. _directionalLights.g += lightColor.g;
  174. _directionalLights.b += lightColor.b;
  175. } else if ( light.isPointLight ) {
  176. _pointLights.r += lightColor.r;
  177. _pointLights.g += lightColor.g;
  178. _pointLights.b += lightColor.b;
  179. }
  180. }
  181. }
  182. function calculateLight( lights, position, normal, color ) {
  183. for ( var l = 0, ll = lights.length; l < ll; l ++ ) {
  184. var light = lights[ l ];
  185. var lightColor = light.color;
  186. if ( light.isDirectionalLight ) {
  187. var lightPosition = _vector3.setFromMatrixPosition( light.matrixWorld ).normalize();
  188. var amount = normal.dot( lightPosition );
  189. if ( amount <= 0 ) continue;
  190. amount *= light.intensity;
  191. color.r += lightColor.r * amount;
  192. color.g += lightColor.g * amount;
  193. color.b += lightColor.b * amount;
  194. } else if ( light.isPointLight ) {
  195. var lightPosition = _vector3.setFromMatrixPosition( light.matrixWorld );
  196. var amount = normal.dot( _vector3.subVectors( lightPosition, position ).normalize() );
  197. if ( amount <= 0 ) continue;
  198. amount *= light.distance == 0 ? 1 : 1 - Math.min( position.distanceTo( lightPosition ) / light.distance, 1 );
  199. if ( amount == 0 ) continue;
  200. amount *= light.intensity;
  201. color.r += lightColor.r * amount;
  202. color.g += lightColor.g * amount;
  203. color.b += lightColor.b * amount;
  204. }
  205. }
  206. }
  207. function renderSprite( v1, element, material ) {
  208. var scaleX = element.scale.x * _svgWidthHalf;
  209. var scaleY = element.scale.y * _svgHeightHalf;
  210. if ( material.isPointsMaterial ) {
  211. scaleX *= material.size;
  212. scaleY *= material.size;
  213. }
  214. var path = 'M' + convert( v1.x - scaleX * 0.5 ) + ',' + convert( v1.y - scaleY * 0.5 ) + 'h' + convert( scaleX ) + 'v' + convert( scaleY ) + 'h' + convert( - scaleX ) + 'z';
  215. var style = "";
  216. if ( material.isSpriteMaterial || material.isPointsMaterial ) {
  217. style = 'fill:' + getSvgColor( material.color, material.opacity );
  218. }
  219. addPath( style, path );
  220. }
  221. function renderLine( v1, v2, element, material ) {
  222. var path = 'M' + convert( v1.positionScreen.x ) + ',' + convert( v1.positionScreen.y ) + 'L' + convert( v2.positionScreen.x ) + ',' + convert( v2.positionScreen.y );
  223. if ( material.isLineBasicMaterial ) {
  224. var style = 'fill:none;stroke:' + getSvgColor( material.color, material.opacity ) + ';stroke-width:' + material.linewidth + ';stroke-linecap:' + material.linecap;
  225. if ( material.isLineDashedMaterial ) {
  226. style = style + ';stroke-dasharray:' + material.dashSize + "," + material.gapSize;
  227. }
  228. addPath( style, path );
  229. }
  230. }
  231. function renderFace3( v1, v2, v3, element, material ) {
  232. _this.info.render.vertices += 3;
  233. _this.info.render.faces ++;
  234. var path = 'M' + convert( v1.positionScreen.x ) + ',' + convert( v1.positionScreen.y ) + 'L' + convert( v2.positionScreen.x ) + ',' + convert( v2.positionScreen.y ) + 'L' + convert( v3.positionScreen.x ) + ',' + convert( v3.positionScreen.y ) + 'z';
  235. var style = '';
  236. if ( material.isMeshBasicMaterial ) {
  237. _color.copy( material.color );
  238. if ( material.vertexColors === THREE.FaceColors || material.vertexColors === THREE.VertexColors ) {
  239. _color.multiply( element.color );
  240. }
  241. } else if ( material.isMeshLambertMaterial || material.isMeshPhongMaterial || material.isMeshStandardMaterial ) {
  242. _diffuseColor.copy( material.color );
  243. if ( material.vertexColors === THREE.FaceColors || material.vertexColors === THREE.VertexColors ) {
  244. _diffuseColor.multiply( element.color );
  245. }
  246. _color.copy( _ambientLight );
  247. _centroid.copy( v1.positionWorld ).add( v2.positionWorld ).add( v3.positionWorld ).divideScalar( 3 );
  248. calculateLight( _lights, _centroid, element.normalModel, _color );
  249. _color.multiply( _diffuseColor ).add( material.emissive );
  250. } else if ( material.isMeshNormalMaterial ) {
  251. _normal.copy( element.normalModel ).applyMatrix3( _normalViewMatrix );
  252. _color.setRGB( _normal.x, _normal.y, _normal.z ).multiplyScalar( 0.5 ).addScalar( 0.5 );
  253. }
  254. if ( material.wireframe ) {
  255. style = 'fill:none;stroke:' + getSvgColor( _color, material.opacity ) + ';stroke-width:' + material.wireframeLinewidth + ';stroke-linecap:' + material.wireframeLinecap + ';stroke-linejoin:' + material.wireframeLinejoin;
  256. } else {
  257. style = 'fill:' + getSvgColor( _color, material.opacity );
  258. }
  259. addPath( style, path );
  260. }
  261. function addPath( style, path ) {
  262. if ( _currentStyle === style ) {
  263. _currentPath += path;
  264. } else {
  265. flushPath();
  266. _currentStyle = style;
  267. _currentPath = path;
  268. }
  269. }
  270. function flushPath() {
  271. if ( _currentPath ) {
  272. _svgNode = getPathNode( _pathCount ++ );
  273. _svgNode.setAttribute( 'd', _currentPath );
  274. _svgNode.setAttribute( 'style', _currentStyle );
  275. _svg.appendChild( _svgNode );
  276. }
  277. _currentPath = '';
  278. _currentStyle = '';
  279. }
  280. function getPathNode( id ) {
  281. if ( _svgPathPool[ id ] == null ) {
  282. _svgPathPool[ id ] = document.createElementNS( 'http://www.w3.org/2000/svg', 'path' );
  283. if ( _quality == 0 ) {
  284. _svgPathPool[ id ].setAttribute( 'shape-rendering', 'crispEdges' ); //optimizeSpeed
  285. }
  286. return _svgPathPool[ id ];
  287. }
  288. return _svgPathPool[ id ];
  289. }
  290. };