shared-picking.js 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  1. const state = {
  2. width: 300, // canvas default
  3. height: 150, // canvas default
  4. };
  5. const pickPosition = {x: 0, y: 0};
  6. function init(data) { // eslint-disable-line no-unused-vars
  7. const {canvas} = data;
  8. const renderer = new THREE.WebGLRenderer({canvas});
  9. state.width = canvas.width;
  10. state.height = canvas.height;
  11. const fov = 75;
  12. const aspect = 2; // the canvas default
  13. const near = 0.1;
  14. const far = 100;
  15. const camera = new THREE.PerspectiveCamera(fov, aspect, near, far);
  16. camera.position.z = 4;
  17. const scene = new THREE.Scene();
  18. {
  19. const color = 0xFFFFFF;
  20. const intensity = 1;
  21. const light = new THREE.DirectionalLight(color, intensity);
  22. light.position.set(-1, 2, 4);
  23. scene.add(light);
  24. }
  25. const boxWidth = 1;
  26. const boxHeight = 1;
  27. const boxDepth = 1;
  28. const geometry = new THREE.BoxGeometry(boxWidth, boxHeight, boxDepth);
  29. function makeInstance(geometry, color, x) {
  30. const material = new THREE.MeshPhongMaterial({
  31. color,
  32. });
  33. const cube = new THREE.Mesh(geometry, material);
  34. scene.add(cube);
  35. cube.position.x = x;
  36. return cube;
  37. }
  38. const cubes = [
  39. makeInstance(geometry, 0x44aa88, 0),
  40. makeInstance(geometry, 0x8844aa, -2),
  41. makeInstance(geometry, 0xaa8844, 2),
  42. ];
  43. class PickHelper {
  44. constructor() {
  45. this.raycaster = new THREE.Raycaster();
  46. this.pickedObject = null;
  47. this.pickedObjectSavedColor = 0;
  48. }
  49. pick(normalizedPosition, scene, camera, time) {
  50. // restore the color if there is a picked object
  51. if (this.pickedObject) {
  52. this.pickedObject.material.emissive.setHex(this.pickedObjectSavedColor);
  53. this.pickedObject = undefined;
  54. }
  55. // cast a ray through the frustum
  56. this.raycaster.setFromCamera(normalizedPosition, camera);
  57. // get the list of objects the ray intersected
  58. const intersectedObjects = this.raycaster.intersectObjects(scene.children);
  59. if (intersectedObjects.length) {
  60. // pick the first object. It's the closest one
  61. this.pickedObject = intersectedObjects[0].object;
  62. // save its color
  63. this.pickedObjectSavedColor = this.pickedObject.material.emissive.getHex();
  64. // set its emissive color to flashing red/yellow
  65. this.pickedObject.material.emissive.setHex((time * 8) % 2 > 1 ? 0xFFFF00 : 0xFF0000);
  66. }
  67. }
  68. }
  69. const pickHelper = new PickHelper();
  70. function resizeRendererToDisplaySize(renderer) {
  71. const canvas = renderer.domElement;
  72. const width = state.width;
  73. const height = state.height;
  74. const needResize = canvas.width !== width || canvas.height !== height;
  75. if (needResize) {
  76. renderer.setSize(width, height, false);
  77. }
  78. return needResize;
  79. }
  80. function render(time) {
  81. time *= 0.001;
  82. if (resizeRendererToDisplaySize(renderer)) {
  83. camera.aspect = state.width / state.height;
  84. camera.updateProjectionMatrix();
  85. }
  86. cubes.forEach((cube, ndx) => {
  87. const speed = 1 + ndx * .1;
  88. const rot = time * speed;
  89. cube.rotation.x = rot;
  90. cube.rotation.y = rot;
  91. });
  92. pickHelper.pick(pickPosition, scene, camera, time);
  93. renderer.render(scene, camera);
  94. requestAnimationFrame(render);
  95. }
  96. requestAnimationFrame(render);
  97. }