TransformControls.js 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149
  1. /**
  2. * @author arodic / https://github.com/arodic
  3. */
  4. ( function () {
  5. 'use strict';
  6. var GizmoMaterial = function ( parameters ) {
  7. THREE.MeshBasicMaterial.call( this );
  8. this.depthTest = false;
  9. this.depthWrite = false;
  10. this.side = THREE.FrontSide;
  11. this.transparent = true;
  12. this.setValues( parameters );
  13. this.oldColor = this.color.clone();
  14. this.oldOpacity = this.opacity;
  15. this.highlight = function( highlighted ) {
  16. if ( highlighted ) {
  17. this.color.setRGB( 1, 1, 0 );
  18. this.opacity = 1;
  19. } else {
  20. this.color.copy( this.oldColor );
  21. this.opacity = this.oldOpacity;
  22. }
  23. };
  24. };
  25. GizmoMaterial.prototype = Object.create( THREE.MeshBasicMaterial.prototype );
  26. GizmoMaterial.prototype.constructor = GizmoMaterial;
  27. var GizmoLineMaterial = function ( parameters ) {
  28. THREE.LineBasicMaterial.call( this );
  29. this.depthTest = false;
  30. this.depthWrite = false;
  31. this.transparent = true;
  32. this.linewidth = 1;
  33. this.setValues( parameters );
  34. this.oldColor = this.color.clone();
  35. this.oldOpacity = this.opacity;
  36. this.highlight = function( highlighted ) {
  37. if ( highlighted ) {
  38. this.color.setRGB( 1, 1, 0 );
  39. this.opacity = 1;
  40. } else {
  41. this.color.copy( this.oldColor );
  42. this.opacity = this.oldOpacity;
  43. }
  44. };
  45. };
  46. GizmoLineMaterial.prototype = Object.create( THREE.LineBasicMaterial.prototype );
  47. GizmoLineMaterial.prototype.constructor = GizmoLineMaterial;
  48. var pickerMaterial = new GizmoMaterial( { visible: false, transparent: false } );
  49. THREE.TransformGizmo = function () {
  50. this.init = function () {
  51. THREE.Object3D.call( this );
  52. this.handles = new THREE.Object3D();
  53. this.pickers = new THREE.Object3D();
  54. this.planes = new THREE.Object3D();
  55. this.add( this.handles );
  56. this.add( this.pickers );
  57. this.add( this.planes );
  58. //// PLANES
  59. var planeGeometry = new THREE.PlaneBufferGeometry( 50, 50, 2, 2 );
  60. var planeMaterial = new THREE.MeshBasicMaterial( { visible: false, side: THREE.DoubleSide } );
  61. var planes = {
  62. "XY": new THREE.Mesh( planeGeometry, planeMaterial ),
  63. "YZ": new THREE.Mesh( planeGeometry, planeMaterial ),
  64. "XZ": new THREE.Mesh( planeGeometry, planeMaterial ),
  65. "XYZE": new THREE.Mesh( planeGeometry, planeMaterial )
  66. };
  67. this.activePlane = planes[ "XYZE" ];
  68. planes[ "YZ" ].rotation.set( 0, Math.PI / 2, 0 );
  69. planes[ "XZ" ].rotation.set( - Math.PI / 2, 0, 0 );
  70. for ( var i in planes ) {
  71. planes[ i ].name = i;
  72. this.planes.add( planes[ i ] );
  73. this.planes[ i ] = planes[ i ];
  74. }
  75. //// HANDLES AND PICKERS
  76. var setupGizmos = function( gizmoMap, parent ) {
  77. for ( var name in gizmoMap ) {
  78. for ( i = gizmoMap[ name ].length; i --; ) {
  79. var object = gizmoMap[ name ][ i ][ 0 ];
  80. var position = gizmoMap[ name ][ i ][ 1 ];
  81. var rotation = gizmoMap[ name ][ i ][ 2 ];
  82. object.name = name;
  83. if ( position ) object.position.set( position[ 0 ], position[ 1 ], position[ 2 ] );
  84. if ( rotation ) object.rotation.set( rotation[ 0 ], rotation[ 1 ], rotation[ 2 ] );
  85. parent.add( object );
  86. }
  87. }
  88. };
  89. setupGizmos( this.handleGizmos, this.handles );
  90. setupGizmos( this.pickerGizmos, this.pickers );
  91. // reset Transformations
  92. this.traverse( function ( child ) {
  93. if ( child instanceof THREE.Mesh ) {
  94. child.updateMatrix();
  95. var tempGeometry = child.geometry.clone();
  96. tempGeometry.applyMatrix( child.matrix );
  97. child.geometry = tempGeometry;
  98. child.position.set( 0, 0, 0 );
  99. child.rotation.set( 0, 0, 0 );
  100. child.scale.set( 1, 1, 1 );
  101. }
  102. } );
  103. };
  104. this.highlight = function ( axis ) {
  105. this.traverse( function( child ) {
  106. if ( child.material && child.material.highlight ) {
  107. if ( child.name === axis ) {
  108. child.material.highlight( true );
  109. } else {
  110. child.material.highlight( false );
  111. }
  112. }
  113. } );
  114. };
  115. };
  116. THREE.TransformGizmo.prototype = Object.create( THREE.Object3D.prototype );
  117. THREE.TransformGizmo.prototype.constructor = THREE.TransformGizmo;
  118. THREE.TransformGizmo.prototype.update = function ( rotation, eye ) {
  119. var vec1 = new THREE.Vector3( 0, 0, 0 );
  120. var vec2 = new THREE.Vector3( 0, 1, 0 );
  121. var lookAtMatrix = new THREE.Matrix4();
  122. this.traverse( function( child ) {
  123. if ( child.name.search( "E" ) !== - 1 ) {
  124. child.quaternion.setFromRotationMatrix( lookAtMatrix.lookAt( eye, vec1, vec2 ) );
  125. } else if ( child.name.search( "X" ) !== - 1 || child.name.search( "Y" ) !== - 1 || child.name.search( "Z" ) !== - 1 ) {
  126. child.quaternion.setFromEuler( rotation );
  127. }
  128. } );
  129. };
  130. THREE.TransformGizmoTranslate = function () {
  131. THREE.TransformGizmo.call( this );
  132. var arrowGeometry = new THREE.Geometry();
  133. var mesh = new THREE.Mesh( new THREE.CylinderGeometry( 0, 0.05, 0.2, 12, 1, false ) );
  134. mesh.position.y = 0.5;
  135. mesh.updateMatrix();
  136. arrowGeometry.merge( mesh.geometry, mesh.matrix );
  137. var lineXGeometry = new THREE.BufferGeometry();
  138. lineXGeometry.addAttribute( 'position', new THREE.Float32BufferAttribute( [ 0, 0, 0, 1, 0, 0 ], 3 ) );
  139. var lineYGeometry = new THREE.BufferGeometry();
  140. lineYGeometry.addAttribute( 'position', new THREE.Float32BufferAttribute( [ 0, 0, 0, 0, 1, 0 ], 3 ) );
  141. var lineZGeometry = new THREE.BufferGeometry();
  142. lineZGeometry.addAttribute( 'position', new THREE.Float32BufferAttribute( [ 0, 0, 0, 0, 0, 1 ], 3 ) );
  143. this.handleGizmos = {
  144. X: [
  145. [ new THREE.Mesh( arrowGeometry, new GizmoMaterial( { color: 0xff0000 } ) ), [ 0.5, 0, 0 ], [ 0, 0, - Math.PI / 2 ] ],
  146. [ new THREE.Line( lineXGeometry, new GizmoLineMaterial( { color: 0xff0000 } ) ) ]
  147. ],
  148. Y: [
  149. [ new THREE.Mesh( arrowGeometry, new GizmoMaterial( { color: 0x00ff00 } ) ), [ 0, 0.5, 0 ] ],
  150. [ new THREE.Line( lineYGeometry, new GizmoLineMaterial( { color: 0x00ff00 } ) ) ]
  151. ],
  152. Z: [
  153. [ new THREE.Mesh( arrowGeometry, new GizmoMaterial( { color: 0x0000ff } ) ), [ 0, 0, 0.5 ], [ Math.PI / 2, 0, 0 ] ],
  154. [ new THREE.Line( lineZGeometry, new GizmoLineMaterial( { color: 0x0000ff } ) ) ]
  155. ],
  156. XYZ: [
  157. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.1, 0 ), new GizmoMaterial( { color: 0xffffff, opacity: 0.25 } ) ), [ 0, 0, 0 ], [ 0, 0, 0 ] ]
  158. ],
  159. XY: [
  160. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.29, 0.29 ), new GizmoMaterial( { color: 0xffff00, opacity: 0.25 } ) ), [ 0.15, 0.15, 0 ] ]
  161. ],
  162. YZ: [
  163. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.29, 0.29 ), new GizmoMaterial( { color: 0x00ffff, opacity: 0.25 } ) ), [ 0, 0.15, 0.15 ], [ 0, Math.PI / 2, 0 ] ]
  164. ],
  165. XZ: [
  166. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.29, 0.29 ), new GizmoMaterial( { color: 0xff00ff, opacity: 0.25 } ) ), [ 0.15, 0, 0.15 ], [ - Math.PI / 2, 0, 0 ] ]
  167. ]
  168. };
  169. this.pickerGizmos = {
  170. X: [
  171. [ new THREE.Mesh( new THREE.CylinderBufferGeometry( 0.2, 0, 1, 4, 1, false ), pickerMaterial ), [ 0.6, 0, 0 ], [ 0, 0, - Math.PI / 2 ] ]
  172. ],
  173. Y: [
  174. [ new THREE.Mesh( new THREE.CylinderBufferGeometry( 0.2, 0, 1, 4, 1, false ), pickerMaterial ), [ 0, 0.6, 0 ] ]
  175. ],
  176. Z: [
  177. [ new THREE.Mesh( new THREE.CylinderBufferGeometry( 0.2, 0, 1, 4, 1, false ), pickerMaterial ), [ 0, 0, 0.6 ], [ Math.PI / 2, 0, 0 ] ]
  178. ],
  179. XYZ: [
  180. [ new THREE.Mesh( new THREE.OctahedronGeometry( 0.2, 0 ), pickerMaterial ) ]
  181. ],
  182. XY: [
  183. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.4, 0.4 ), pickerMaterial ), [ 0.2, 0.2, 0 ] ]
  184. ],
  185. YZ: [
  186. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.4, 0.4 ), pickerMaterial ), [ 0, 0.2, 0.2 ], [ 0, Math.PI / 2, 0 ] ]
  187. ],
  188. XZ: [
  189. [ new THREE.Mesh( new THREE.PlaneBufferGeometry( 0.4, 0.4 ), pickerMaterial ), [ 0.2, 0, 0.2 ], [ - Math.PI / 2, 0, 0 ] ]
  190. ]
  191. };
  192. this.setActivePlane = function ( axis, eye ) {
  193. var tempMatrix = new THREE.Matrix4();
  194. eye.applyMatrix4( tempMatrix.getInverse( tempMatrix.extractRotation( this.planes[ "XY" ].matrixWorld ) ) );
  195. if ( axis === "X" ) {
  196. this.activePlane = this.planes[ "XY" ];
  197. if ( Math.abs( eye.y ) > Math.abs( eye.z ) ) this.activePlane = this.planes[ "XZ" ];
  198. }
  199. if ( axis === "Y" ) {
  200. this.activePlane = this.planes[ "XY" ];
  201. if ( Math.abs( eye.x ) > Math.abs( eye.z ) ) this.activePlane = this.planes[ "YZ" ];
  202. }
  203. if ( axis === "Z" ) {
  204. this.activePlane = this.planes[ "XZ" ];
  205. if ( Math.abs( eye.x ) > Math.abs( eye.y ) ) this.activePlane = this.planes[ "YZ" ];
  206. }
  207. if ( axis === "XYZ" ) this.activePlane = this.planes[ "XYZE" ];
  208. if ( axis === "XY" ) this.activePlane = this.planes[ "XY" ];
  209. if ( axis === "YZ" ) this.activePlane = this.planes[ "YZ" ];
  210. if ( axis === "XZ" ) this.activePlane = this.planes[ "XZ" ];
  211. };
  212. this.init();
  213. };
  214. THREE.TransformGizmoTranslate.prototype = Object.create( THREE.TransformGizmo.prototype );
  215. THREE.TransformGizmoTranslate.prototype.constructor = THREE.TransformGizmoTranslate;
  216. THREE.TransformGizmoRotate = function () {
  217. THREE.TransformGizmo.call( this );
  218. var CircleGeometry = function ( radius, facing, arc ) {
  219. var geometry = new THREE.BufferGeometry();
  220. var vertices = [];
  221. arc = arc ? arc : 1;
  222. for ( var i = 0; i <= 64 * arc; ++ i ) {
  223. if ( facing === 'x' ) vertices.push( 0, Math.cos( i / 32 * Math.PI ) * radius, Math.sin( i / 32 * Math.PI ) * radius );
  224. if ( facing === 'y' ) vertices.push( Math.cos( i / 32 * Math.PI ) * radius, 0, Math.sin( i / 32 * Math.PI ) * radius );
  225. if ( facing === 'z' ) vertices.push( Math.sin( i / 32 * Math.PI ) * radius, Math.cos( i / 32 * Math.PI ) * radius, 0 );
  226. }
  227. geometry.addAttribute( 'position', new THREE.Float32BufferAttribute( vertices, 3 ) );
  228. return geometry;
  229. };
  230. this.handleGizmos = {
  231. X: [
  232. [ new THREE.Line( new CircleGeometry( 1, 'x', 0.5 ), new GizmoLineMaterial( { color: 0xff0000 } ) ) ]
  233. ],
  234. Y: [
  235. [ new THREE.Line( new CircleGeometry( 1, 'y', 0.5 ), new GizmoLineMaterial( { color: 0x00ff00 } ) ) ]
  236. ],
  237. Z: [
  238. [ new THREE.Line( new CircleGeometry( 1, 'z', 0.5 ), new GizmoLineMaterial( { color: 0x0000ff } ) ) ]
  239. ],
  240. E: [
  241. [ new THREE.Line( new CircleGeometry( 1.25, 'z', 1 ), new GizmoLineMaterial( { color: 0xcccc00 } ) ) ]
  242. ],
  243. XYZE: [
  244. [ new THREE.Line( new CircleGeometry( 1, 'z', 1 ), new GizmoLineMaterial( { color: 0x787878 } ) ) ]
  245. ]
  246. };
  247. this.pickerGizmos = {
  248. X: [
  249. [ new THREE.Mesh( new THREE.TorusBufferGeometry( 1, 0.12, 4, 12, Math.PI ), pickerMaterial ), [ 0, 0, 0 ], [ 0, - Math.PI / 2, - Math.PI / 2 ] ]
  250. ],
  251. Y: [
  252. [ new THREE.Mesh( new THREE.TorusBufferGeometry( 1, 0.12, 4, 12, Math.PI ), pickerMaterial ), [ 0, 0, 0 ], [ Math.PI / 2, 0, 0 ] ]
  253. ],
  254. Z: [
  255. [ new THREE.Mesh( new THREE.TorusBufferGeometry( 1, 0.12, 4, 12, Math.PI ), pickerMaterial ), [ 0, 0, 0 ], [ 0, 0, - Math.PI / 2 ] ]
  256. ],
  257. E: [
  258. [ new THREE.Mesh( new THREE.TorusBufferGeometry( 1.25, 0.12, 2, 24 ), pickerMaterial ) ]
  259. ],
  260. XYZE: [
  261. [ new THREE.Mesh() ]// TODO
  262. ]
  263. };
  264. this.setActivePlane = function ( axis ) {
  265. if ( axis === "E" ) this.activePlane = this.planes[ "XYZE" ];
  266. if ( axis === "X" ) this.activePlane = this.planes[ "YZ" ];
  267. if ( axis === "Y" ) this.activePlane = this.planes[ "XZ" ];
  268. if ( axis === "Z" ) this.activePlane = this.planes[ "XY" ];
  269. };
  270. this.update = function ( rotation, eye2 ) {
  271. THREE.TransformGizmo.prototype.update.apply( this, arguments );
  272. var tempMatrix = new THREE.Matrix4();
  273. var worldRotation = new THREE.Euler( 0, 0, 1 );
  274. var tempQuaternion = new THREE.Quaternion();
  275. var unitX = new THREE.Vector3( 1, 0, 0 );
  276. var unitY = new THREE.Vector3( 0, 1, 0 );
  277. var unitZ = new THREE.Vector3( 0, 0, 1 );
  278. var quaternionX = new THREE.Quaternion();
  279. var quaternionY = new THREE.Quaternion();
  280. var quaternionZ = new THREE.Quaternion();
  281. var eye = eye2.clone();
  282. worldRotation.copy( this.planes[ "XY" ].rotation );
  283. tempQuaternion.setFromEuler( worldRotation );
  284. tempMatrix.makeRotationFromQuaternion( tempQuaternion ).getInverse( tempMatrix );
  285. eye.applyMatrix4( tempMatrix );
  286. this.traverse( function( child ) {
  287. tempQuaternion.setFromEuler( worldRotation );
  288. if ( child.name === "X" ) {
  289. quaternionX.setFromAxisAngle( unitX, Math.atan2( - eye.y, eye.z ) );
  290. tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionX );
  291. child.quaternion.copy( tempQuaternion );
  292. }
  293. if ( child.name === "Y" ) {
  294. quaternionY.setFromAxisAngle( unitY, Math.atan2( eye.x, eye.z ) );
  295. tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionY );
  296. child.quaternion.copy( tempQuaternion );
  297. }
  298. if ( child.name === "Z" ) {
  299. quaternionZ.setFromAxisAngle( unitZ, Math.atan2( eye.y, eye.x ) );
  300. tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionZ );
  301. child.quaternion.copy( tempQuaternion );
  302. }
  303. } );
  304. };
  305. this.init();
  306. };
  307. THREE.TransformGizmoRotate.prototype = Object.create( THREE.TransformGizmo.prototype );
  308. THREE.TransformGizmoRotate.prototype.constructor = THREE.TransformGizmoRotate;
  309. THREE.TransformGizmoScale = function () {
  310. THREE.TransformGizmo.call( this );
  311. var arrowGeometry = new THREE.Geometry();
  312. var mesh = new THREE.Mesh( new THREE.BoxGeometry( 0.125, 0.125, 0.125 ) );
  313. mesh.position.y = 0.5;
  314. mesh.updateMatrix();
  315. arrowGeometry.merge( mesh.geometry, mesh.matrix );
  316. var lineXGeometry = new THREE.BufferGeometry();
  317. lineXGeometry.addAttribute( 'position', new THREE.Float32BufferAttribute( [ 0, 0, 0, 1, 0, 0 ], 3 ) );
  318. var lineYGeometry = new THREE.BufferGeometry();
  319. lineYGeometry.addAttribute( 'position', new THREE.Float32BufferAttribute( [ 0, 0, 0, 0, 1, 0 ], 3 ) );
  320. var lineZGeometry = new THREE.BufferGeometry();
  321. lineZGeometry.addAttribute( 'position', new THREE.Float32BufferAttribute( [ 0, 0, 0, 0, 0, 1 ], 3 ) );
  322. this.handleGizmos = {
  323. X: [
  324. [ new THREE.Mesh( arrowGeometry, new GizmoMaterial( { color: 0xff0000 } ) ), [ 0.5, 0, 0 ], [ 0, 0, - Math.PI / 2 ] ],
  325. [ new THREE.Line( lineXGeometry, new GizmoLineMaterial( { color: 0xff0000 } ) ) ]
  326. ],
  327. Y: [
  328. [ new THREE.Mesh( arrowGeometry, new GizmoMaterial( { color: 0x00ff00 } ) ), [ 0, 0.5, 0 ] ],
  329. [ new THREE.Line( lineYGeometry, new GizmoLineMaterial( { color: 0x00ff00 } ) ) ]
  330. ],
  331. Z: [
  332. [ new THREE.Mesh( arrowGeometry, new GizmoMaterial( { color: 0x0000ff } ) ), [ 0, 0, 0.5 ], [ Math.PI / 2, 0, 0 ] ],
  333. [ new THREE.Line( lineZGeometry, new GizmoLineMaterial( { color: 0x0000ff } ) ) ]
  334. ],
  335. XYZ: [
  336. [ new THREE.Mesh( new THREE.BoxBufferGeometry( 0.125, 0.125, 0.125 ), new GizmoMaterial( { color: 0xffffff, opacity: 0.25 } ) ) ]
  337. ]
  338. };
  339. this.pickerGizmos = {
  340. X: [
  341. [ new THREE.Mesh( new THREE.CylinderBufferGeometry( 0.2, 0, 1, 4, 1, false ), pickerMaterial ), [ 0.6, 0, 0 ], [ 0, 0, - Math.PI / 2 ] ]
  342. ],
  343. Y: [
  344. [ new THREE.Mesh( new THREE.CylinderBufferGeometry( 0.2, 0, 1, 4, 1, false ), pickerMaterial ), [ 0, 0.6, 0 ] ]
  345. ],
  346. Z: [
  347. [ new THREE.Mesh( new THREE.CylinderBufferGeometry( 0.2, 0, 1, 4, 1, false ), pickerMaterial ), [ 0, 0, 0.6 ], [ Math.PI / 2, 0, 0 ] ]
  348. ],
  349. XYZ: [
  350. [ new THREE.Mesh( new THREE.BoxBufferGeometry( 0.4, 0.4, 0.4 ), pickerMaterial ) ]
  351. ]
  352. };
  353. this.setActivePlane = function ( axis, eye ) {
  354. var tempMatrix = new THREE.Matrix4();
  355. eye.applyMatrix4( tempMatrix.getInverse( tempMatrix.extractRotation( this.planes[ "XY" ].matrixWorld ) ) );
  356. if ( axis === "X" ) {
  357. this.activePlane = this.planes[ "XY" ];
  358. if ( Math.abs( eye.y ) > Math.abs( eye.z ) ) this.activePlane = this.planes[ "XZ" ];
  359. }
  360. if ( axis === "Y" ) {
  361. this.activePlane = this.planes[ "XY" ];
  362. if ( Math.abs( eye.x ) > Math.abs( eye.z ) ) this.activePlane = this.planes[ "YZ" ];
  363. }
  364. if ( axis === "Z" ) {
  365. this.activePlane = this.planes[ "XZ" ];
  366. if ( Math.abs( eye.x ) > Math.abs( eye.y ) ) this.activePlane = this.planes[ "YZ" ];
  367. }
  368. if ( axis === "XYZ" ) this.activePlane = this.planes[ "XYZE" ];
  369. };
  370. this.init();
  371. };
  372. THREE.TransformGizmoScale.prototype = Object.create( THREE.TransformGizmo.prototype );
  373. THREE.TransformGizmoScale.prototype.constructor = THREE.TransformGizmoScale;
  374. THREE.TransformControls = function ( camera, domElement ) {
  375. // TODO: Make non-uniform scale and rotate play nice in hierarchies
  376. // TODO: ADD RXYZ contol
  377. THREE.Object3D.call( this );
  378. domElement = ( domElement !== undefined ) ? domElement : document;
  379. this.object = undefined;
  380. this.visible = false;
  381. this.translationSnap = null;
  382. this.rotationSnap = null;
  383. this.space = "world";
  384. this.size = 1;
  385. this.axis = null;
  386. var scope = this;
  387. var _mode = "translate";
  388. var _dragging = false;
  389. var _gizmo = {
  390. "translate": new THREE.TransformGizmoTranslate(),
  391. "rotate": new THREE.TransformGizmoRotate(),
  392. "scale": new THREE.TransformGizmoScale()
  393. };
  394. for ( var type in _gizmo ) {
  395. var gizmoObj = _gizmo[ type ];
  396. gizmoObj.visible = ( type === _mode );
  397. this.add( gizmoObj );
  398. }
  399. var changeEvent = { type: "change" };
  400. var mouseDownEvent = { type: "mouseDown" };
  401. var mouseUpEvent = { type: "mouseUp", mode: _mode };
  402. var objectChangeEvent = { type: "objectChange" };
  403. var ray = new THREE.Raycaster();
  404. var pointerVector = new THREE.Vector2();
  405. var point = new THREE.Vector3();
  406. var offset = new THREE.Vector3();
  407. var rotation = new THREE.Vector3();
  408. var offsetRotation = new THREE.Vector3();
  409. var scale = 1;
  410. var lookAtMatrix = new THREE.Matrix4();
  411. var eye = new THREE.Vector3();
  412. var tempMatrix = new THREE.Matrix4();
  413. var tempVector = new THREE.Vector3();
  414. var tempQuaternion = new THREE.Quaternion();
  415. var unitX = new THREE.Vector3( 1, 0, 0 );
  416. var unitY = new THREE.Vector3( 0, 1, 0 );
  417. var unitZ = new THREE.Vector3( 0, 0, 1 );
  418. var quaternionXYZ = new THREE.Quaternion();
  419. var quaternionX = new THREE.Quaternion();
  420. var quaternionY = new THREE.Quaternion();
  421. var quaternionZ = new THREE.Quaternion();
  422. var quaternionE = new THREE.Quaternion();
  423. var oldPosition = new THREE.Vector3();
  424. var oldScale = new THREE.Vector3();
  425. var oldRotationMatrix = new THREE.Matrix4();
  426. var parentRotationMatrix = new THREE.Matrix4();
  427. var parentScale = new THREE.Vector3();
  428. var worldPosition = new THREE.Vector3();
  429. var worldRotation = new THREE.Euler();
  430. var worldRotationMatrix = new THREE.Matrix4();
  431. var camPosition = new THREE.Vector3();
  432. var camRotation = new THREE.Euler();
  433. domElement.addEventListener( "mousedown", onPointerDown, false );
  434. domElement.addEventListener( "touchstart", onPointerDown, false );
  435. domElement.addEventListener( "mousemove", onPointerHover, false );
  436. domElement.addEventListener( "touchmove", onPointerHover, false );
  437. domElement.addEventListener( "mousemove", onPointerMove, false );
  438. domElement.addEventListener( "touchmove", onPointerMove, false );
  439. domElement.addEventListener( "mouseup", onPointerUp, false );
  440. domElement.addEventListener( "mouseout", onPointerUp, false );
  441. domElement.addEventListener( "touchend", onPointerUp, false );
  442. domElement.addEventListener( "touchcancel", onPointerUp, false );
  443. domElement.addEventListener( "touchleave", onPointerUp, false );
  444. this.dispose = function () {
  445. domElement.removeEventListener( "mousedown", onPointerDown );
  446. domElement.removeEventListener( "touchstart", onPointerDown );
  447. domElement.removeEventListener( "mousemove", onPointerHover );
  448. domElement.removeEventListener( "touchmove", onPointerHover );
  449. domElement.removeEventListener( "mousemove", onPointerMove );
  450. domElement.removeEventListener( "touchmove", onPointerMove );
  451. domElement.removeEventListener( "mouseup", onPointerUp );
  452. domElement.removeEventListener( "mouseout", onPointerUp );
  453. domElement.removeEventListener( "touchend", onPointerUp );
  454. domElement.removeEventListener( "touchcancel", onPointerUp );
  455. domElement.removeEventListener( "touchleave", onPointerUp );
  456. };
  457. this.attach = function ( object ) {
  458. this.object = object;
  459. this.visible = true;
  460. this.update();
  461. };
  462. this.detach = function () {
  463. this.object = undefined;
  464. this.visible = false;
  465. this.axis = null;
  466. };
  467. this.getMode = function () {
  468. return _mode;
  469. };
  470. this.setMode = function ( mode ) {
  471. _mode = mode ? mode : _mode;
  472. if ( _mode === "scale" ) scope.space = "local";
  473. for ( var type in _gizmo ) _gizmo[ type ].visible = ( type === _mode );
  474. this.update();
  475. scope.dispatchEvent( changeEvent );
  476. };
  477. this.setTranslationSnap = function ( translationSnap ) {
  478. scope.translationSnap = translationSnap;
  479. };
  480. this.setRotationSnap = function ( rotationSnap ) {
  481. scope.rotationSnap = rotationSnap;
  482. };
  483. this.setSize = function ( size ) {
  484. scope.size = size;
  485. this.update();
  486. scope.dispatchEvent( changeEvent );
  487. };
  488. this.setSpace = function ( space ) {
  489. scope.space = space;
  490. this.update();
  491. scope.dispatchEvent( changeEvent );
  492. };
  493. this.update = function () {
  494. if ( scope.object === undefined ) return;
  495. scope.object.updateMatrixWorld();
  496. worldPosition.setFromMatrixPosition( scope.object.matrixWorld );
  497. worldRotation.setFromRotationMatrix( tempMatrix.extractRotation( scope.object.matrixWorld ) );
  498. camera.updateMatrixWorld();
  499. camPosition.setFromMatrixPosition( camera.matrixWorld );
  500. camRotation.setFromRotationMatrix( tempMatrix.extractRotation( camera.matrixWorld ) );
  501. scale = worldPosition.distanceTo( camPosition ) / 6 * scope.size;
  502. this.position.copy( worldPosition );
  503. this.scale.set( scale, scale, scale );
  504. if ( camera instanceof THREE.PerspectiveCamera ) {
  505. eye.copy( camPosition ).sub( worldPosition ).normalize();
  506. } else if ( camera instanceof THREE.OrthographicCamera ) {
  507. eye.copy( camPosition ).normalize();
  508. }
  509. if ( scope.space === "local" ) {
  510. _gizmo[ _mode ].update( worldRotation, eye );
  511. } else if ( scope.space === "world" ) {
  512. _gizmo[ _mode ].update( new THREE.Euler(), eye );
  513. }
  514. _gizmo[ _mode ].highlight( scope.axis );
  515. };
  516. function onPointerHover( event ) {
  517. if ( scope.object === undefined || _dragging === true || ( event.button !== undefined && event.button !== 0 ) ) return;
  518. var pointer = event.changedTouches ? event.changedTouches[ 0 ] : event;
  519. var intersect = intersectObjects( pointer, _gizmo[ _mode ].pickers.children );
  520. var axis = null;
  521. if ( intersect ) {
  522. axis = intersect.object.name;
  523. event.preventDefault();
  524. }
  525. if ( scope.axis !== axis ) {
  526. scope.axis = axis;
  527. scope.update();
  528. scope.dispatchEvent( changeEvent );
  529. }
  530. }
  531. function onPointerDown( event ) {
  532. if ( scope.object === undefined || _dragging === true || ( event.button !== undefined && event.button !== 0 ) ) return;
  533. var pointer = event.changedTouches ? event.changedTouches[ 0 ] : event;
  534. if ( pointer.button === 0 || pointer.button === undefined ) {
  535. var intersect = intersectObjects( pointer, _gizmo[ _mode ].pickers.children );
  536. if ( intersect ) {
  537. event.preventDefault();
  538. event.stopPropagation();
  539. scope.dispatchEvent( mouseDownEvent );
  540. scope.axis = intersect.object.name;
  541. scope.update();
  542. eye.copy( camPosition ).sub( worldPosition ).normalize();
  543. _gizmo[ _mode ].setActivePlane( scope.axis, eye );
  544. var planeIntersect = intersectObjects( pointer, [ _gizmo[ _mode ].activePlane ] );
  545. if ( planeIntersect ) {
  546. oldPosition.copy( scope.object.position );
  547. oldScale.copy( scope.object.scale );
  548. oldRotationMatrix.extractRotation( scope.object.matrix );
  549. worldRotationMatrix.extractRotation( scope.object.matrixWorld );
  550. parentRotationMatrix.extractRotation( scope.object.parent.matrixWorld );
  551. parentScale.setFromMatrixScale( tempMatrix.getInverse( scope.object.parent.matrixWorld ) );
  552. offset.copy( planeIntersect.point );
  553. }
  554. }
  555. }
  556. _dragging = true;
  557. }
  558. function onPointerMove( event ) {
  559. if ( scope.object === undefined || scope.axis === null || _dragging === false || ( event.button !== undefined && event.button !== 0 ) ) return;
  560. var pointer = event.changedTouches ? event.changedTouches[ 0 ] : event;
  561. var planeIntersect = intersectObjects( pointer, [ _gizmo[ _mode ].activePlane ] );
  562. if ( planeIntersect === false ) return;
  563. event.preventDefault();
  564. event.stopPropagation();
  565. point.copy( planeIntersect.point );
  566. if ( _mode === "translate" ) {
  567. point.sub( offset );
  568. point.multiply( parentScale );
  569. if ( scope.space === "local" ) {
  570. point.applyMatrix4( tempMatrix.getInverse( worldRotationMatrix ) );
  571. if ( scope.axis.search( "X" ) === - 1 ) point.x = 0;
  572. if ( scope.axis.search( "Y" ) === - 1 ) point.y = 0;
  573. if ( scope.axis.search( "Z" ) === - 1 ) point.z = 0;
  574. point.applyMatrix4( oldRotationMatrix );
  575. scope.object.position.copy( oldPosition );
  576. scope.object.position.add( point );
  577. }
  578. if ( scope.space === "world" || scope.axis.search( "XYZ" ) !== - 1 ) {
  579. if ( scope.axis.search( "X" ) === - 1 ) point.x = 0;
  580. if ( scope.axis.search( "Y" ) === - 1 ) point.y = 0;
  581. if ( scope.axis.search( "Z" ) === - 1 ) point.z = 0;
  582. point.applyMatrix4( tempMatrix.getInverse( parentRotationMatrix ) );
  583. scope.object.position.copy( oldPosition );
  584. scope.object.position.add( point );
  585. }
  586. if ( scope.translationSnap !== null ) {
  587. if ( scope.space === "local" ) {
  588. scope.object.position.applyMatrix4( tempMatrix.getInverse( worldRotationMatrix ) );
  589. }
  590. if ( scope.axis.search( "X" ) !== - 1 ) scope.object.position.x = Math.round( scope.object.position.x / scope.translationSnap ) * scope.translationSnap;
  591. if ( scope.axis.search( "Y" ) !== - 1 ) scope.object.position.y = Math.round( scope.object.position.y / scope.translationSnap ) * scope.translationSnap;
  592. if ( scope.axis.search( "Z" ) !== - 1 ) scope.object.position.z = Math.round( scope.object.position.z / scope.translationSnap ) * scope.translationSnap;
  593. if ( scope.space === "local" ) {
  594. scope.object.position.applyMatrix4( worldRotationMatrix );
  595. }
  596. }
  597. } else if ( _mode === "scale" ) {
  598. point.sub( offset );
  599. point.multiply( parentScale );
  600. if ( scope.space === "local" ) {
  601. if ( scope.axis === "XYZ" ) {
  602. scale = 1 + ( ( point.y ) / Math.max( oldScale.x, oldScale.y, oldScale.z ) );
  603. scope.object.scale.x = oldScale.x * scale;
  604. scope.object.scale.y = oldScale.y * scale;
  605. scope.object.scale.z = oldScale.z * scale;
  606. } else {
  607. point.applyMatrix4( tempMatrix.getInverse( worldRotationMatrix ) );
  608. if ( scope.axis === "X" ) scope.object.scale.x = oldScale.x * ( 1 + point.x / oldScale.x );
  609. if ( scope.axis === "Y" ) scope.object.scale.y = oldScale.y * ( 1 + point.y / oldScale.y );
  610. if ( scope.axis === "Z" ) scope.object.scale.z = oldScale.z * ( 1 + point.z / oldScale.z );
  611. }
  612. }
  613. } else if ( _mode === "rotate" ) {
  614. point.sub( worldPosition );
  615. point.multiply( parentScale );
  616. tempVector.copy( offset ).sub( worldPosition );
  617. tempVector.multiply( parentScale );
  618. if ( scope.axis === "E" ) {
  619. point.applyMatrix4( tempMatrix.getInverse( lookAtMatrix ) );
  620. tempVector.applyMatrix4( tempMatrix.getInverse( lookAtMatrix ) );
  621. rotation.set( Math.atan2( point.z, point.y ), Math.atan2( point.x, point.z ), Math.atan2( point.y, point.x ) );
  622. offsetRotation.set( Math.atan2( tempVector.z, tempVector.y ), Math.atan2( tempVector.x, tempVector.z ), Math.atan2( tempVector.y, tempVector.x ) );
  623. tempQuaternion.setFromRotationMatrix( tempMatrix.getInverse( parentRotationMatrix ) );
  624. quaternionE.setFromAxisAngle( eye, rotation.z - offsetRotation.z );
  625. quaternionXYZ.setFromRotationMatrix( worldRotationMatrix );
  626. tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionE );
  627. tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionXYZ );
  628. scope.object.quaternion.copy( tempQuaternion );
  629. } else if ( scope.axis === "XYZE" ) {
  630. quaternionE.setFromEuler( point.clone().cross( tempVector ).normalize() ); // rotation axis
  631. tempQuaternion.setFromRotationMatrix( tempMatrix.getInverse( parentRotationMatrix ) );
  632. quaternionX.setFromAxisAngle( quaternionE, - point.clone().angleTo( tempVector ) );
  633. quaternionXYZ.setFromRotationMatrix( worldRotationMatrix );
  634. tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionX );
  635. tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionXYZ );
  636. scope.object.quaternion.copy( tempQuaternion );
  637. } else if ( scope.space === "local" ) {
  638. point.applyMatrix4( tempMatrix.getInverse( worldRotationMatrix ) );
  639. tempVector.applyMatrix4( tempMatrix.getInverse( worldRotationMatrix ) );
  640. rotation.set( Math.atan2( point.z, point.y ), Math.atan2( point.x, point.z ), Math.atan2( point.y, point.x ) );
  641. offsetRotation.set( Math.atan2( tempVector.z, tempVector.y ), Math.atan2( tempVector.x, tempVector.z ), Math.atan2( tempVector.y, tempVector.x ) );
  642. quaternionXYZ.setFromRotationMatrix( oldRotationMatrix );
  643. if ( scope.rotationSnap !== null ) {
  644. quaternionX.setFromAxisAngle( unitX, Math.round( ( rotation.x - offsetRotation.x ) / scope.rotationSnap ) * scope.rotationSnap );
  645. quaternionY.setFromAxisAngle( unitY, Math.round( ( rotation.y - offsetRotation.y ) / scope.rotationSnap ) * scope.rotationSnap );
  646. quaternionZ.setFromAxisAngle( unitZ, Math.round( ( rotation.z - offsetRotation.z ) / scope.rotationSnap ) * scope.rotationSnap );
  647. } else {
  648. quaternionX.setFromAxisAngle( unitX, rotation.x - offsetRotation.x );
  649. quaternionY.setFromAxisAngle( unitY, rotation.y - offsetRotation.y );
  650. quaternionZ.setFromAxisAngle( unitZ, rotation.z - offsetRotation.z );
  651. }
  652. if ( scope.axis === "X" ) quaternionXYZ.multiplyQuaternions( quaternionXYZ, quaternionX );
  653. if ( scope.axis === "Y" ) quaternionXYZ.multiplyQuaternions( quaternionXYZ, quaternionY );
  654. if ( scope.axis === "Z" ) quaternionXYZ.multiplyQuaternions( quaternionXYZ, quaternionZ );
  655. scope.object.quaternion.copy( quaternionXYZ );
  656. } else if ( scope.space === "world" ) {
  657. rotation.set( Math.atan2( point.z, point.y ), Math.atan2( point.x, point.z ), Math.atan2( point.y, point.x ) );
  658. offsetRotation.set( Math.atan2( tempVector.z, tempVector.y ), Math.atan2( tempVector.x, tempVector.z ), Math.atan2( tempVector.y, tempVector.x ) );
  659. tempQuaternion.setFromRotationMatrix( tempMatrix.getInverse( parentRotationMatrix ) );
  660. if ( scope.rotationSnap !== null ) {
  661. quaternionX.setFromAxisAngle( unitX, Math.round( ( rotation.x - offsetRotation.x ) / scope.rotationSnap ) * scope.rotationSnap );
  662. quaternionY.setFromAxisAngle( unitY, Math.round( ( rotation.y - offsetRotation.y ) / scope.rotationSnap ) * scope.rotationSnap );
  663. quaternionZ.setFromAxisAngle( unitZ, Math.round( ( rotation.z - offsetRotation.z ) / scope.rotationSnap ) * scope.rotationSnap );
  664. } else {
  665. quaternionX.setFromAxisAngle( unitX, rotation.x - offsetRotation.x );
  666. quaternionY.setFromAxisAngle( unitY, rotation.y - offsetRotation.y );
  667. quaternionZ.setFromAxisAngle( unitZ, rotation.z - offsetRotation.z );
  668. }
  669. quaternionXYZ.setFromRotationMatrix( worldRotationMatrix );
  670. if ( scope.axis === "X" ) tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionX );
  671. if ( scope.axis === "Y" ) tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionY );
  672. if ( scope.axis === "Z" ) tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionZ );
  673. tempQuaternion.multiplyQuaternions( tempQuaternion, quaternionXYZ );
  674. scope.object.quaternion.copy( tempQuaternion );
  675. }
  676. }
  677. scope.update();
  678. scope.dispatchEvent( changeEvent );
  679. scope.dispatchEvent( objectChangeEvent );
  680. }
  681. function onPointerUp( event ) {
  682. event.preventDefault(); // Prevent MouseEvent on mobile
  683. if ( event.button !== undefined && event.button !== 0 ) return;
  684. if ( _dragging && ( scope.axis !== null ) ) {
  685. mouseUpEvent.mode = _mode;
  686. scope.dispatchEvent( mouseUpEvent );
  687. }
  688. _dragging = false;
  689. if ( 'TouchEvent' in window && event instanceof TouchEvent ) {
  690. // Force "rollover"
  691. scope.axis = null;
  692. scope.update();
  693. scope.dispatchEvent( changeEvent );
  694. } else {
  695. onPointerHover( event );
  696. }
  697. }
  698. function intersectObjects( pointer, objects ) {
  699. var rect = domElement.getBoundingClientRect();
  700. var x = ( pointer.clientX - rect.left ) / rect.width;
  701. var y = ( pointer.clientY - rect.top ) / rect.height;
  702. pointerVector.set( ( x * 2 ) - 1, - ( y * 2 ) + 1 );
  703. ray.setFromCamera( pointerVector, camera );
  704. var intersections = ray.intersectObjects( objects, true );
  705. return intersections[ 0 ] ? intersections[ 0 ] : false;
  706. }
  707. };
  708. THREE.TransformControls.prototype = Object.create( THREE.Object3D.prototype );
  709. THREE.TransformControls.prototype.constructor = THREE.TransformControls;
  710. }() );