All files / src/views/threejs pickPixelColor.ts

17.65% Statements 3/17
100% Branches 0/0
0% Functions 0/1
17.65% Lines 3/17

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51        1x                       1x         1x                                                          
/*!
 * Copyright 2020 Cognite AS
 */
 
import * as THREE from 'three';
 
export interface PickingInput {
  coords: {
    x: number;
    y: number;
  };
  scene: THREE.Scene;
  camera: THREE.PerspectiveCamera;
  renderer: THREE.WebGLRenderer;
}
 
const storage = {
  renderTarget: new THREE.WebGLRenderTarget(1, 1),
  pixelBuffer: new Uint8Array(4)
};
 
export function pickPixelColor(input: PickingInput, clearColor: THREE.Color, clearAlpha: number) {
  const { renderTarget, pixelBuffer } = storage;
  const { scene, camera, coords, renderer } = input;
 
  const pickCamera = camera.clone();
 
  pickCamera.setViewOffset(
    renderer.domElement.clientWidth,
    renderer.domElement.clientHeight,
    ((coords.x + 1.0) / 2.0) * renderer.domElement.clientWidth,
    ((1.0 - coords.y) / 2.0) * renderer.domElement.clientHeight,
    1,
    1
  );
 
  const currentClearColor = renderer.getClearColor().clone();
  const currentClearAlpha = renderer.getClearAlpha();
 
  renderer.setRenderTarget(renderTarget);
  renderer.setClearColor(clearColor, clearAlpha);
  renderer.clearColor();
  renderer.render(scene, pickCamera);
  renderer.setRenderTarget(null);
 
  renderer.setClearColor(currentClearColor, currentClearAlpha);
 
  renderer.readRenderTargetPixels(renderTarget, 0, 0, 1, 1, pixelBuffer);
  return pixelBuffer;
}