vr.js 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302
  1. import {
  2. DebugTilesRenderer as TilesRenderer,
  3. NONE,
  4. SCREEN_ERROR,
  5. GEOMETRIC_ERROR,
  6. DISTANCE,
  7. DEPTH,
  8. RELATIVE_DEPTH,
  9. IS_LEAF,
  10. RANDOM_COLOR,
  11. } from '../src/index.js';
  12. import {
  13. Scene,
  14. DirectionalLight,
  15. AmbientLight,
  16. WebGLRenderer,
  17. PerspectiveCamera,
  18. Box3,
  19. Raycaster,
  20. Mesh,
  21. MeshBasicMaterial,
  22. Group,
  23. TorusBufferGeometry,
  24. sRGBEncoding,
  25. GridHelper,
  26. BufferGeometry,
  27. Float32BufferAttribute,
  28. LineBasicMaterial,
  29. AdditiveBlending,
  30. Line,
  31. Vector3,
  32. } from 'three';
  33. import * as dat from 'three/examples/jsm/libs/dat.gui.module.js';
  34. import { VRButton } from 'three/examples/jsm/webxr/VRButton.js';
  35. import { XRControllerModelFactory } from 'three/examples/jsm/webxr/XRControllerModelFactory.js';
  36. let camera, scene, renderer, tiles;
  37. let workspace;
  38. let box, grid;
  39. let raycaster, fwdVector, intersectRing;
  40. let offsetParent;
  41. let controller, controllerGrip;
  42. let params = {
  43. 'displayBoxBounds': false,
  44. 'colorMode': 0,
  45. 'displayGrid': true,
  46. };
  47. init();
  48. animate();
  49. function init() {
  50. scene = new Scene();
  51. // primary camera view
  52. renderer = new WebGLRenderer( { antialias: true } );
  53. renderer.setPixelRatio( window.devicePixelRatio );
  54. renderer.setSize( window.innerWidth, window.innerHeight );
  55. renderer.setClearColor( 0x151c1f );
  56. renderer.outputEncoding = sRGBEncoding;
  57. renderer.xr.enabled = true;
  58. document.body.appendChild( renderer.domElement );
  59. renderer.domElement.tabIndex = 1;
  60. // create workspace
  61. workspace = new Group();
  62. scene.add( workspace );
  63. grid = new GridHelper( 10, 10, 0xffffff, 0xffffff );
  64. grid.material.transparent = true;
  65. grid.material.opacity = 0.5;
  66. grid.material.depthWrite = false;
  67. workspace.add( grid );
  68. camera = new PerspectiveCamera( 60, window.innerWidth / window.innerHeight, 0.1, 4000 );
  69. camera.position.set( 0, 1, 0 );
  70. workspace.add( camera );
  71. // lights
  72. const dirLight = new DirectionalLight( 0xffffff );
  73. dirLight.position.set( 1, 2, 3 );
  74. scene.add( dirLight );
  75. const ambLight = new AmbientLight( 0xffffff, 0.2 );
  76. scene.add( ambLight );
  77. // tile set
  78. box = new Box3();
  79. // parent for centering the tileset
  80. offsetParent = new Group();
  81. scene.add( offsetParent );
  82. tiles = new TilesRenderer( '../data/tileset.json' );
  83. offsetParent.add( tiles.group );
  84. // Raycasting init
  85. raycaster = new Raycaster();
  86. fwdVector = new Vector3( 0, 0, 1 );
  87. const rayIntersectMat = new MeshBasicMaterial( { color: 0xb2dfdb } );
  88. intersectRing = new Mesh( new TorusBufferGeometry( 1.5, 0.2, 16, 100 ), rayIntersectMat );
  89. intersectRing.visible = false;
  90. scene.add( intersectRing );
  91. // vr setup
  92. document.body.appendChild( VRButton.createButton( renderer ) );
  93. controller = renderer.xr.getController( 0 );
  94. controller.addEventListener( 'selectstart', () => {
  95. workspace.position.copy( intersectRing.position );
  96. } );
  97. controller.addEventListener( 'connected', function ( event ) {
  98. this.controllerActive = true;
  99. this.add( buildController( event.data ) );
  100. } );
  101. controller.addEventListener( 'disconnected', function () {
  102. this.controllerActive = false;
  103. this.remove( this.children[ 0 ] );
  104. } );
  105. workspace.add( controller );
  106. // controller models
  107. const controllerModelFactory = new XRControllerModelFactory();
  108. controllerGrip = renderer.xr.getControllerGrip( 0 );
  109. controllerGrip.add( controllerModelFactory.createControllerModel( controllerGrip ) );
  110. workspace.add( controllerGrip );
  111. onWindowResize();
  112. window.addEventListener( 'resize', onWindowResize, false );
  113. // GUI
  114. const gui = new dat.GUI();
  115. gui.width = 300;
  116. gui.add( params, 'displayGrid' );
  117. gui.add( params, 'displayBoxBounds' );
  118. gui.add( params, 'colorMode', {
  119. NONE,
  120. SCREEN_ERROR,
  121. GEOMETRIC_ERROR,
  122. DISTANCE,
  123. DEPTH,
  124. RELATIVE_DEPTH,
  125. IS_LEAF,
  126. RANDOM_COLOR,
  127. } );
  128. gui.open();
  129. }
  130. function buildController( data ) {
  131. let geometry, material;
  132. switch ( data.targetRayMode ) {
  133. case 'tracked-pointer':
  134. geometry = new BufferGeometry();
  135. geometry.setAttribute( 'position', new Float32BufferAttribute( [ 0, 0, 0, 0, 0, - 1 ], 3 ) );
  136. geometry.setAttribute( 'color', new Float32BufferAttribute( [ 0.5, 0.5, 0.5, 0, 0, 0 ], 3 ) );
  137. material = new LineBasicMaterial( {
  138. vertexColors: true,
  139. blending: AdditiveBlending,
  140. depthWrite: false,
  141. transparent: true,
  142. } );
  143. return new Line( geometry, material );
  144. case 'gaze':
  145. geometry = new RingBufferGeometry( 0.02, 0.04, 32 ).translate( 0, 0, - 1 );
  146. material = new MeshBasicMaterial( { opacity: 0.5, transparent: true } );
  147. return new Mesh( geometry, material );
  148. }
  149. }
  150. function onWindowResize() {
  151. camera.updateProjectionMatrix();
  152. renderer.setPixelRatio( window.devicePixelRatio );
  153. renderer.setSize( window.innerWidth, window.innerHeight );
  154. camera.aspect = window.innerWidth / window.innerHeight;
  155. camera.updateProjectionMatrix();
  156. }
  157. function animate() {
  158. renderer.setAnimationLoop( render );
  159. }
  160. function render() {
  161. requestAnimationFrame( animate );
  162. grid.visible = params.displayGrid;
  163. // update options
  164. tiles.displayBoxBounds = params.displayBoxBounds;
  165. tiles.colorMode = parseFloat( params.colorMode );
  166. // update tiles center
  167. if ( tiles.getBounds( box ) ) {
  168. box.getCenter( tiles.group.position );
  169. tiles.group.position.multiplyScalar( - 1 );
  170. }
  171. // remove all cameras so we can use the VR camera instead
  172. tiles.cameras.forEach( c => tiles.deleteCamera( camera ) );
  173. // get the XR camera with a combined frustum for culling
  174. if ( renderer.xr.isPresenting ) {
  175. const currCamera = renderer.xr.getCamera( camera );
  176. tiles.setCamera( currCamera );
  177. const leftCam = currCamera.cameras[0];
  178. if ( leftCam ) {
  179. tiles.setResolution( currCamera, leftCam.viewport.z, leftCam.viewport.w );
  180. }
  181. } else {
  182. tiles.setCamera( camera );
  183. tiles.setResolutionFromRenderer( camera, renderer );
  184. }
  185. tiles.update();
  186. if ( controller.controllerActive ) {
  187. const { ray } = raycaster;
  188. raycaster.firstHitOnly = true;
  189. // get the controller ray
  190. ray.origin
  191. .copy( controller.position )
  192. .applyMatrix4( workspace.matrixWorld );
  193. controller
  194. .getWorldDirection( ray.direction )
  195. .multiplyScalar( - 1 );
  196. const results = raycaster.intersectObject( tiles.group, true );
  197. if ( results.length ) {
  198. const hit = results[ 0 ];
  199. hit.face.normal.transformDirection( tiles.group.matrixWorld );
  200. intersectRing.position.copy( hit.point );
  201. intersectRing.quaternion.setFromUnitVectors(
  202. fwdVector,
  203. hit.face.normal,
  204. );
  205. intersectRing.visible = true;
  206. // scale ring based on distance
  207. const scale = workspace.position.distanceTo( intersectRing.position ) * camera.fov / 4000;
  208. intersectRing.scale.setScalar( scale );
  209. } else {
  210. intersectRing.visible = false;
  211. }
  212. } else {
  213. intersectRing.visible = false;
  214. }
  215. // render primary view
  216. renderer.render( scene, camera );
  217. }