TransformControlsGizmo.js 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587
  1. /**
  2. * @author arodic / https://github.com/arodic
  3. */
  4. THREE.TransformControlsGizmo = function () {
  5. 'use strict';
  6. THREE.Object3D.call( this );
  7. this.type = 'TransformControlsGizmo';
  8. // shared materials
  9. var gizmoMaterial = new THREE.MeshBasicMaterial({
  10. depthTest: false,
  11. depthWrite: false,
  12. transparent: true,
  13. side: THREE.DoubleSide,
  14. fog: false
  15. });
  16. var gizmoLineMaterial = new THREE.LineBasicMaterial({
  17. depthTest: false,
  18. depthWrite: false,
  19. transparent: true,
  20. linewidth: 1,
  21. fog: false
  22. });
  23. var matInvisible = gizmoMaterial.clone();
  24. matInvisible.opacity = 0.15;
  25. var matRed = gizmoMaterial.clone();
  26. matRed.color.set(0xff0000);
  27. var matGreen = gizmoMaterial.clone();
  28. matGreen.color.set(0x00ff00);
  29. var matBlue = gizmoMaterial.clone();
  30. matBlue.color.set(0x0000ff);
  31. var matWhiteTransperent = gizmoMaterial.clone();
  32. matWhiteTransperent.opacity = 0.25;
  33. var matYellowTransparent = matWhiteTransperent.clone();
  34. matYellowTransparent.color.set(0xffff00);
  35. var matCyanTransparent = matWhiteTransperent.clone();
  36. matCyanTransparent.color.set(0x00ffff);
  37. var matMagentaTransparent = matWhiteTransperent.clone();
  38. matMagentaTransparent.color.set(0xff00ff);
  39. var matYellow = gizmoMaterial.clone();
  40. matYellow.color.set(0xffff00);
  41. var matLineRed = gizmoLineMaterial.clone();
  42. matLineRed.color.set(0xff0000);
  43. var matLineGreen = gizmoLineMaterial.clone();
  44. matLineGreen.color.set(0x00ff00);
  45. var matLineBlue = gizmoLineMaterial.clone();
  46. matLineBlue.color.set(0x0000ff);
  47. var matLineCyan = gizmoLineMaterial.clone();
  48. matLineCyan.color.set(0x00ffff);
  49. var matLineMagenta = gizmoLineMaterial.clone();
  50. matLineMagenta.color.set(0xff00ff);
  51. var matLineBlue = gizmoLineMaterial.clone();
  52. matLineBlue.color.set(0x0000ff);
  53. var matLineYellow = gizmoLineMaterial.clone();
  54. matLineYellow.color.set(0xffff00);
  55. var matLineGray = gizmoLineMaterial.clone();
  56. matLineGray.color.set(0x787878);
  57. var matLineYellowTransparent = matLineYellow.clone();
  58. matLineYellowTransparent.opacity = 0.25;
  59. // shared objects
  60. var arrowGeometry = new THREE.CylinderGeometry( 0, 0.05, 0.2, 12, 1, false);
  61. var scaleHandleGeometry = new THREE.BoxGeometry( 0.125, 0.125, 0.125);
  62. var lineGeometry = new THREE.BufferGeometry( );
  63. lineGeometry.addAttribute('position', new THREE.Float32BufferAttribute([ 0, 0, 0, 1, 0, 0 ], 3));
  64. var CircleGeometry = function(radius, arc) {
  65. var geometry = new THREE.BufferGeometry( );
  66. var vertices = [];
  67. for (var i = 0; i <= 64 * arc; ++i) {
  68. vertices.push(0, Math.cos(i / 32 * Math.PI) * radius, Math.sin(i / 32 * Math.PI) * radius);
  69. }
  70. geometry.addAttribute('position', new THREE.Float32BufferAttribute(vertices, 3));
  71. return geometry;
  72. };
  73. // gizmos
  74. var gizmoTranslate = {
  75. X: [
  76. [ new THREE.Mesh( arrowGeometry, matRed ), [ 1, 0, 0 ], [ 0, 0, -Math.PI / 2 ], null, 'fwd' ],
  77. [ new THREE.Mesh( arrowGeometry, matRed ), [ 1, 0, 0 ], [ 0, 0, Math.PI / 2 ], null, 'bwd' ],
  78. [ new THREE.Line( lineGeometry, matLineRed ) ]
  79. ],
  80. Y: [
  81. [ new THREE.Mesh( arrowGeometry, matGreen ), [ 0, 1, 0 ], null, null, 'fwd' ],
  82. [ new THREE.Mesh( arrowGeometry, matGreen ), [ 0, 1, 0 ], [ Math.PI, 0, 0 ], null, 'bwd' ],
  83. [ new THREE.Line( lineGeometry, matLineGreen ), null, [ 0, 0, Math.PI / 2 ] ]
  84. ],
  85. Z: [
  86. [ new THREE.Mesh( arrowGeometry, matBlue ), [ 0, 0, 1 ], [ Math.PI / 2, 0, 0 ], null, 'fwd' ],
  87. [ new THREE.Mesh( arrowGeometry, matBlue ), [ 0, 0, 1 ], [ -Math.PI / 2, 0, 0 ], null, 'bwd' ],
  88. [ new THREE.Line( lineGeometry, matLineBlue ), null, [ 0, -Math.PI / 2, 0 ] ]
  89. ],
  90. XYZ: [
  91. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.1, 0 ), matWhiteTransperent ), [ 0, 0, 0 ], [ 0, 0, 0 ] ]
  92. ],
  93. XY: [
  94. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.295, 0.295 ), matYellowTransparent ), [ 0.15, 0.15, 0 ] ],
  95. [ new THREE.Line( lineGeometry, matLineYellow ), [ 0.18, 0.3, 0 ], null, [ 0.125, 1, 1 ] ],
  96. [ new THREE.Line( lineGeometry, matLineYellow ), [ 0.3, 0.18, 0 ], [ 0, 0, Math.PI / 2 ], [ 0.125, 1, 1 ] ]
  97. ],
  98. YZ: [
  99. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.295, 0.295 ), matCyanTransparent ), [ 0, 0.15, 0.15 ], [ 0, Math.PI / 2, 0 ] ],
  100. [ new THREE.Line( lineGeometry, matLineCyan ), [ 0, 0.18, 0.3 ], [ 0, 0, Math.PI / 2 ], [ 0.125, 1, 1 ] ],
  101. [ new THREE.Line( lineGeometry, matLineCyan ), [ 0, 0.3, 0.18 ], [ 0, -Math.PI / 2, 0 ], [ 0.125, 1, 1 ] ]
  102. ],
  103. XZ: [
  104. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.295, 0.295 ), matMagentaTransparent ), [ 0.15, 0, 0.15 ], [ -Math.PI / 2, 0, 0 ] ],
  105. [ new THREE.Line( lineGeometry, matLineMagenta ), [ 0.18, 0, 0.3 ], null, [ 0.125, 1, 1 ] ],
  106. [ new THREE.Line( lineGeometry, matLineMagenta ), [ 0.3, 0, 0.18 ], [ 0, -Math.PI / 2, 0 ], [ 0.125, 1, 1 ] ]
  107. ]
  108. };
  109. var pickerTranslate = {
  110. X: [
  111. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.2, 0, 1, 4, 1, false ), matInvisible ), [ 0.6, 0, 0 ], [ 0, 0, -Math.PI / 2 ] ]
  112. ],
  113. Y: [
  114. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.2, 0, 1, 4, 1, false ), matInvisible ), [ 0, 0.6, 0 ] ]
  115. ],
  116. Z: [
  117. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.2, 0, 1, 4, 1, false ), matInvisible ), [ 0, 0, 0.6 ], [ Math.PI / 2, 0, 0 ] ]
  118. ],
  119. XYZ: [
  120. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.2, 0 ), matInvisible ) ]
  121. ],
  122. XY: [
  123. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.4, 0.4 ), matInvisible ), [ 0.2, 0.2, 0 ] ]
  124. ],
  125. YZ: [
  126. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.4, 0.4 ), matInvisible ), [ 0, 0.2, 0.2 ], [ 0, Math.PI / 2, 0 ] ]
  127. ],
  128. XZ: [
  129. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.4, 0.4 ), matInvisible ), [ 0.2, 0, 0.2 ], [ -Math.PI / 2, 0, 0 ] ]
  130. ]
  131. };
  132. var gizmoRotate = {
  133. X: [
  134. [ new THREE.Line( new CircleGeometry( 1, 0.5 ), matLineRed ) ],
  135. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.04, 0 ), matRed ), [ 0, 0, 0.99 ], null, [ 1, 3, 1 ], 'linear' ],
  136. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.03, 0 ), matRed ), [ 0, 0, 1 ], null, [ 4, 1, 4 ], 'radial' ],
  137. ],
  138. Y: [
  139. [ new THREE.Line( new CircleGeometry( 1, 0.5 ), matLineGreen ), null, [ 0, 0, -Math.PI / 2 ] ],
  140. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.04, 0 ), matGreen ), [ 0, 0, 0.99 ], null, [ 3, 1, 1 ], 'linear' ],
  141. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.03, 0 ), matGreen ), [ 0, 0, 1 ], null, [ 1, 4, 4 ], 'radial' ],
  142. ],
  143. Z: [
  144. [ new THREE.Line( new CircleGeometry( 1, 0.5 ), matLineBlue ), null, [ 0, Math.PI / 2, 0 ] ],
  145. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.04, 0 ), matBlue ), [ 0.99, 0, 0 ], null, [ 1, 3, 1 ], 'linear' ],
  146. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.03, 0 ), matBlue ), [ 1, 0, 0 ], null, [ 4, 1, 4 ], 'radial' ],
  147. ],
  148. E: [
  149. [ new THREE.Line( new CircleGeometry( 1.25, 1 ), matLineYellowTransparent ), null, [ 0, Math.PI / 2, 0 ] ],
  150. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.03, 0, 0.15, 4, 1, false ), matLineYellowTransparent ), [ 1.17, 0, 0 ], [ 0, 0, -Math.PI / 2 ], [ 1, 1, 0.001 ]],
  151. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.03, 0, 0.15, 4, 1, false ), matLineYellowTransparent ), [ -1.17, 0, 0 ], [ 0, 0, Math.PI / 2 ], [ 1, 1, 0.001 ]],
  152. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.03, 0, 0.15, 4, 1, false ), matLineYellowTransparent ), [ 0, -1.17, 0 ], [ Math.PI, 0, 0 ], [ 1, 1, 0.001 ]],
  153. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.03, 0, 0.15, 4, 1, false ), matLineYellowTransparent ), [ 0, 1.17, 0 ], [ 0, 0, 0 ], [ 1, 1, 0.001 ]],
  154. ],
  155. XYZE: [
  156. [ new THREE.Line( new CircleGeometry( 0.99, 1 ), matLineGray ), null, [ 0, Math.PI / 2, 0 ] ]
  157. ]
  158. };
  159. var pickerRotate = {
  160. X: [
  161. [ new THREE.Mesh( new THREE.TorusGeometry( 1, 0.1, 4, 24 ), matInvisible ), [ 0, 0, 0 ], [ 0, -Math.PI / 2, -Math.PI / 2 ] ],
  162. ],
  163. Y: [
  164. [ new THREE.Mesh( new THREE.TorusGeometry( 1, 0.1, 4, 24 ), matInvisible ), [ 0, 0, 0 ], [ Math.PI / 2, 0, 0 ] ],
  165. ],
  166. Z: [
  167. [ new THREE.Mesh( new THREE.TorusGeometry( 1, 0.1, 4, 24 ), matInvisible ), [ 0, 0, 0 ], [ 0, 0, -Math.PI / 2 ] ],
  168. ],
  169. E: [
  170. [ new THREE.Mesh( new THREE.TorusGeometry( 1.25, 0.1, 2, 24 ), matInvisible ) ]
  171. ],
  172. XYZE: [
  173. [ new THREE.Mesh( new THREE.SphereGeometry( 0.7, 10, 8 ), matInvisible ) ]
  174. ]
  175. };
  176. var gizmoScale = {
  177. X: [
  178. [ new THREE.Mesh( scaleHandleGeometry, matRed ), [ 0.8, 0, 0 ], [ 0, 0, -Math.PI / 2 ] ],
  179. [ new THREE.Line( lineGeometry, matLineRed ), null, null, [ 0.8, 1, 1 ] ]
  180. ],
  181. Y: [
  182. [ new THREE.Mesh( scaleHandleGeometry, matGreen ), [ 0, 0.8, 0 ] ],
  183. [ new THREE.Line( lineGeometry, matLineGreen ), null, [ 0, 0, Math.PI / 2 ], [ 0.8, 1, 1 ] ]
  184. ],
  185. Z: [
  186. [ new THREE.Mesh( scaleHandleGeometry, matBlue ), [ 0, 0, 0.8 ], [ Math.PI / 2, 0, 0 ] ],
  187. [ new THREE.Line( lineGeometry, matLineBlue ), null, [ 0, -Math.PI / 2, 0 ], [ 0.8, 1, 1 ] ]
  188. ],
  189. XY: [
  190. [ new THREE.Mesh( scaleHandleGeometry, matYellowTransparent ), [ 0.85, 0.85, 0 ], null, [ 2, 2, 0.2 ] ],
  191. [ new THREE.Line( lineGeometry, matLineYellow ), [ 0.855, 0.98, 0 ], null, [ 0.125, 1, 1 ] ],
  192. [ new THREE.Line( lineGeometry, matLineYellow ), [ 0.98, 0.855, 0 ], [ 0, 0, Math.PI / 2 ], [ 0.125, 1, 1 ] ]
  193. ],
  194. YZ: [
  195. [ new THREE.Mesh( scaleHandleGeometry, matCyanTransparent ), [ 0, 0.85, 0.85 ], null, [ 0.2, 2, 2 ] ],
  196. [ new THREE.Line( lineGeometry, matLineCyan ), [ 0, 0.855, 0.98 ], [ 0, 0, Math.PI / 2 ], [ 0.125, 1, 1 ] ],
  197. [ new THREE.Line( lineGeometry, matLineCyan ), [ 0, 0.98, 0.855 ], [ 0, -Math.PI / 2, 0 ], [ 0.125, 1, 1 ] ]
  198. ],
  199. XZ: [
  200. [ new THREE.Mesh( scaleHandleGeometry, matMagentaTransparent ), [ 0.85, 0, 0.85 ], null, [ 2, 0.2, 2 ] ],
  201. [ new THREE.Line( lineGeometry, matLineMagenta ), [ 0.855, 0, 0.98 ], null, [ 0.125, 1, 1 ] ],
  202. [ new THREE.Line( lineGeometry, matLineMagenta ), [ 0.98, 0, 0.855 ], [ 0, -Math.PI / 2, 0 ], [ 0.125, 1, 1 ] ]
  203. ],
  204. XYZX: [
  205. [ new THREE.Mesh( new THREE.BoxGeometry( 0.125, 0.125, 0.125 ), matWhiteTransperent ), [ 1.1, 0, 0 ] ],
  206. ],
  207. XYZY: [
  208. [ new THREE.Mesh( new THREE.BoxGeometry( 0.125, 0.125, 0.125 ), matWhiteTransperent ), [ 0, 1.1, 0 ] ],
  209. ],
  210. XYZZ: [
  211. [ new THREE.Mesh( new THREE.BoxGeometry( 0.125, 0.125, 0.125 ), matWhiteTransperent ), [ 0, 0, 1.1 ] ],
  212. ]
  213. };
  214. var pickerScale = {
  215. X: [
  216. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.2, 0, 0.8, 4, 1, false ), matInvisible ), [ 0.5, 0, 0 ], [ 0, 0, -Math.PI / 2 ] ]
  217. ],
  218. Y: [
  219. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.2, 0, 0.8, 4, 1, false ), matInvisible ), [ 0, 0.5, 0 ] ]
  220. ],
  221. Z: [
  222. [ new THREE.Mesh( new THREE.CylinderGeometry( 0.2, 0, 0.8, 4, 1, false ), matInvisible ), [ 0, 0, 0.5 ], [ Math.PI / 2, 0, 0 ] ]
  223. ],
  224. XY: [
  225. [ new THREE.Mesh( scaleHandleGeometry, matInvisible ), [ 0.85, 0.85, 0 ], null, [ 3, 3, 0.2 ] ],
  226. ],
  227. YZ: [
  228. [ new THREE.Mesh( scaleHandleGeometry, matInvisible ), [ 0, 0.85, 0.85 ], null, [ 0.2, 3, 3 ] ],
  229. ],
  230. XZ: [
  231. [ new THREE.Mesh( scaleHandleGeometry, matInvisible ), [ 0.85, 0, 0.85 ], null, [ 3, 0.2, 3 ] ],
  232. ],
  233. XYZX: [
  234. [ new THREE.Mesh( new THREE.BoxGeometry( 0.2, 0.2, 0.2 ), matInvisible ), [ 1.1, 0, 0 ] ],
  235. ],
  236. XYZY: [
  237. [ new THREE.Mesh( new THREE.BoxGeometry( 0.2, 0.2, 0.2 ), matInvisible ), [ 0, 1.1, 0 ] ],
  238. ],
  239. XYZZ: [
  240. [ new THREE.Mesh( new THREE.BoxGeometry( 0.2, 0.2, 0.2 ), matInvisible ), [ 0, 0, 1.1 ] ],
  241. ]
  242. };
  243. var setupGizmo = function( gizmoMap ) {
  244. var gizmo = new THREE.Object3D();
  245. for ( var name in gizmoMap ) {
  246. for ( var i = gizmoMap[ name ].length; i --; ) {
  247. var object = gizmoMap[ name ][ i ][ 0 ].clone();
  248. var position = gizmoMap[ name ][ i ][ 1 ];
  249. var rotation = gizmoMap[ name ][ i ][ 2 ];
  250. var scale = gizmoMap[ name ][ i ][ 3 ];
  251. var tag = gizmoMap[ name ][ i ][ 4 ];
  252. object.name = name;
  253. object.tag = tag;
  254. if (position) {
  255. object.position.set(position[ 0 ], position[ 1 ], position[ 2 ]);
  256. }
  257. if (rotation) {
  258. object.rotation.set(rotation[ 0 ], rotation[ 1 ], rotation[ 2 ]);
  259. }
  260. if (scale) {
  261. object.scale.set(scale[ 0 ], scale[ 1 ], scale[ 2 ]);
  262. }
  263. object.updateMatrix();
  264. var tempGeometry = object.geometry.clone();
  265. tempGeometry.applyMatrix(object.matrix);
  266. object.geometry = tempGeometry;
  267. object.position.set(0, 0, 0);
  268. object.rotation.set(0, 0, 0);
  269. object.scale.set(1, 1, 1);
  270. gizmo.add(object);
  271. }
  272. }
  273. return gizmo;
  274. };
  275. var vec1 = new THREE.Vector3( 0, 0, 0 );
  276. var tempVector = new THREE.Vector3();
  277. var alignVector = new THREE.Vector3( 0, 1, 0 );
  278. var lookAtMatrix = new THREE.Matrix4();
  279. var tempQuaternion = new THREE.Quaternion();
  280. var tempQuaternion2 = new THREE.Quaternion();
  281. var identityEuler = new THREE.Euler();
  282. var unitX = new THREE.Vector3( 1, 0, 0 );
  283. var unitY = new THREE.Vector3( 0, 1, 0 );
  284. var unitZ = new THREE.Vector3( 0, 0, 1 );
  285. this.gizmo = {};
  286. this.picker = {};
  287. this.add( this.gizmo[ "translate" ] = setupGizmo( gizmoTranslate ) );
  288. this.add( this.gizmo[ "rotate" ] = setupGizmo( gizmoRotate ) );
  289. this.add( this.gizmo[ "scale" ] = setupGizmo( gizmoScale ) );
  290. this.add( this.picker[ "translate" ] = setupGizmo( pickerTranslate ) );
  291. this.add( this.picker[ "rotate" ] = setupGizmo( pickerRotate ) );
  292. this.add( this.picker[ "scale" ] = setupGizmo( pickerScale ) );
  293. this.picker[ "translate" ].visible = false;
  294. this.picker[ "rotate" ].visible = false;
  295. this.picker[ "scale" ].visible = false;
  296. this.updateMatrixWorld = function () {
  297. var rotation = this.parent.space === "local" ? this.parent._worldRotation : identityEuler;
  298. var eye = this.parent._eye
  299. var mode = this.parent.mode;
  300. var axis = this.parent.axis;
  301. this.gizmo[ "translate" ].visible = mode === "translate";
  302. this.gizmo[ "rotate" ].visible = mode === "rotate";
  303. this.gizmo[ "scale" ].visible = mode === "scale";
  304. this.picker[ "translate" ].visible = false; // mode === "translate";
  305. this.picker[ "rotate" ].visible = false; // mode === "rotate";
  306. this.picker[ "scale" ].visible = false; // mode === "scale";
  307. var handles = [];
  308. handles = handles.concat( this.picker[ mode ].children );
  309. handles = handles.concat( this.gizmo[ mode ].children );
  310. for ( var i = 0; i < handles.length; i++ ) {
  311. var handle = handles[i];
  312. // hide aligned to camera
  313. handle.visible = true;
  314. handle.scale.set( 1, 1, 1 );
  315. handle.position.set( 0, 0, 0 );
  316. handle.rotation.set( 0, 0, 0 );
  317. if ( mode === 'translate' || mode === 'scale' ) {
  318. // Hide translate and scale axis facing the camera
  319. if ( handle.name === 'X' || handle.name === 'XYZX' ) {
  320. if ( Math.abs( alignVector.set( 1, 0, 0 ).applyEuler( rotation ).dot( eye ) ) > 0.99 ) {
  321. handle.scale.set( 1e-3, 1e-3, 1e-3 );
  322. handle.visible = false;
  323. }
  324. }
  325. if ( handle.name === 'Y' || handle.name === 'XYZY' ) {
  326. if ( Math.abs( alignVector.set( 0, 1, 0 ).applyEuler( rotation ).dot( eye ) ) > 0.99 ) {
  327. handle.scale.set( 1e-3, 1e-3, 1e-3 );
  328. handle.visible = false;
  329. }
  330. }
  331. if ( handle.name === 'Z' || handle.name === 'XYZZ' ) {
  332. if ( Math.abs( alignVector.set( 0, 0, 1 ).applyEuler( rotation ).dot( eye ) ) > 0.99 ) {
  333. handle.scale.set( 1e-3, 1e-3, 1e-3 );
  334. handle.visible = false;
  335. }
  336. }
  337. if ( handle.name === 'XY' ) {
  338. if ( Math.abs( alignVector.set( 0, 0, 1 ).applyEuler( rotation ).dot( eye ) ) < 0.2 ) {
  339. handle.scale.set( 1e-3, 1e-3, 1e-3 );
  340. handle.visible = false;
  341. }
  342. }
  343. if ( handle.name === 'YZ' ) {
  344. if ( Math.abs( alignVector.set( 1, 0, 0 ).applyEuler( rotation ).dot( eye ) ) < 0.2 ) {
  345. handle.scale.set( 1e-3, 1e-3, 1e-3 );
  346. handle.visible = false;
  347. }
  348. }
  349. if ( handle.name === 'XZ' ) {
  350. if ( Math.abs( alignVector.set( 0, 1, 0 ).applyEuler( rotation ).dot( eye ) ) < 0.2 ) {
  351. handle.scale.set( 1e-3, 1e-3, 1e-3 );
  352. handle.visible = false;
  353. }
  354. }
  355. // Flip translate and scale axis ocluded behind another axis
  356. if ( handle.name.search( 'X' ) !== -1 ) {
  357. if ( alignVector.set( 1, 0, 0 ).applyEuler( rotation ).dot( eye ) < -0.4 ) {
  358. if ( handle.tag === 'fwd' ) {
  359. handle.visible = false;
  360. } else {
  361. handle.scale.x = -1;
  362. }
  363. } else if ( handle.tag === 'bwd' ) {
  364. handle.visible = false;
  365. }
  366. }
  367. if ( handle.name.search( 'Y' ) !== -1 ) {
  368. if ( alignVector.set( 0, 1, 0 ).applyEuler( rotation ).dot( eye ) < -0.4 ) {
  369. if ( handle.tag === 'fwd' ) {
  370. handle.visible = false;
  371. } else {
  372. handle.scale.y = -1;
  373. }
  374. } else if ( handle.tag === 'bwd' ) {
  375. handle.visible = false;
  376. }
  377. }
  378. if ( handle.name.search( 'Z' ) !== -1 ) {
  379. if ( alignVector.set( 0, 0, 1 ).applyEuler( rotation ).dot( eye ) < -0.4 ) {
  380. if ( handle.tag === 'fwd' ) {
  381. handle.visible = false;
  382. } else {
  383. handle.scale.z = -1;
  384. }
  385. } else if ( handle.tag === 'bwd' ) {
  386. handle.visible = false;
  387. }
  388. }
  389. // Align handles to current local or world rotation
  390. handle.quaternion.setFromEuler( rotation );
  391. } else if (mode === 'rotate') {
  392. // switch between liner/radial rotation handle affordances
  393. if ( handle.name === 'X' ) {
  394. if ( Math.abs( alignVector.set( 1, 0, 0 ).applyEuler( rotation ).dot( eye ) ) > 0.3 ) {
  395. if ( handle.tag === 'linear' ) {
  396. handle.visible = false;
  397. }
  398. } else if ( handle.tag === 'radial' ) {
  399. handle.visible = false;
  400. }
  401. }
  402. if ( handle.name === 'Y' ) {
  403. if ( Math.abs( alignVector.set( 0, 1, 0 ).applyEuler( rotation ).dot( eye ) ) > 0.3 ) {
  404. if ( handle.tag === 'linear' ) {
  405. handle.visible = false;
  406. }
  407. } else if ( handle.tag === 'radial' ) {
  408. handle.visible = false;
  409. }
  410. }
  411. if ( handle.name === 'Z' ) {
  412. if ( Math.abs( alignVector.set( 0, 0, 1 ).applyEuler( rotation ).dot( eye ) ) > 0.3 ) {
  413. if ( handle.tag === 'linear' ) {
  414. handle.visible = false;
  415. }
  416. } else if ( handle.tag === 'radial' ) {
  417. handle.visible = false;
  418. }
  419. }
  420. // Align handles to current local or world rotation
  421. tempQuaternion2.setFromEuler( rotation );
  422. alignVector.copy( eye ).applyQuaternion( tempQuaternion.setFromEuler( rotation ).inverse());
  423. if ( handle.name.search( "E" ) !== - 1 ) {
  424. alignVector.set( 0, 1, 0 );
  425. handle.quaternion.setFromRotationMatrix( lookAtMatrix.lookAt( eye, vec1, alignVector ) );
  426. }
  427. if ( handle.name === 'X' ) {
  428. tempQuaternion.setFromAxisAngle( unitX, Math.atan2( -alignVector.y, alignVector.z ) );
  429. tempQuaternion.multiplyQuaternions( tempQuaternion2, tempQuaternion );
  430. handle.quaternion.copy( tempQuaternion );
  431. }
  432. if ( handle.name === 'Y' ) {
  433. tempQuaternion.setFromAxisAngle( unitY, Math.atan2( alignVector.x, alignVector.z ) );
  434. tempQuaternion.multiplyQuaternions( tempQuaternion2, tempQuaternion );
  435. handle.quaternion.copy( tempQuaternion );
  436. }
  437. if ( handle.name === 'Z' ) {
  438. tempQuaternion.setFromAxisAngle( unitZ, Math.atan2( alignVector.y, alignVector.x ) );
  439. tempQuaternion.multiplyQuaternions( tempQuaternion2, tempQuaternion );
  440. handle.quaternion.copy( tempQuaternion );
  441. }
  442. }
  443. // highlight selected axis
  444. handle.material._opacity = handle.material._opacity || handle.material.opacity;
  445. handle.material._color = handle.material._color || handle.material.color.clone();
  446. handle.material.color.copy( handle.material._color );
  447. handle.material.opacity = handle.material._opacity;
  448. if ( axis ) {
  449. if ( handle.name === axis ) {
  450. handle.material.opacity *= 2.0;
  451. handle.material.color.lerp( new THREE.Color( 1, 1, 1 ), 0.5 );
  452. } else if ( axis.split('').some( function( a ) { return handle.name === a; } ) ) {
  453. handle.material.opacity *= 2.0;
  454. handle.material.color.lerp( new THREE.Color( 1, 1, 1 ), 0.5 );
  455. } else {
  456. handle.material.opacity *= 0.15;
  457. }
  458. }
  459. }
  460. THREE.Object3D.prototype.updateMatrixWorld.call( this );
  461. };
  462. this.setMode = function() {
  463. console.warn( 'THREE.TransformControlsGizmo: setMode function has been depricated.' );
  464. };
  465. };
  466. THREE.TransformControlsGizmo.prototype = Object.assign( Object.create( THREE.Object3D.prototype ), {
  467. constructor: THREE.TransformControlsGizmo,
  468. isTransformControlsGizmo: true
  469. } );