expo-three/src/AR/Light.ts

Utilities for using THREE.js on Expo

import { AR } from 'expo';
import ct from 'color-temperature';

import THREE from '../Three';

// @ts-ignore
const ARFrameAttribute = AR.FrameAttribute || AR.FrameAttributes;

class Light extends THREE.PointLight {
  constructor() {
    super(0xffee88, 1, 100, 2);
  }

  _data: AR.LightEstimation | null = null;
  set data(value: AR.LightEstimation | null) {
    if (value === this._data) {
      return;
    }
    this._data = value;
    if (!value) {
      return;
    }

    const { ambientIntensity, ambientColorTemperature } = value;

    this.power = ambientIntensity; // 1000;

    const { red, green, blue } = ct.colorTemperature2rgb(ambientColorTemperature);
    this.color = new THREE.Color(red / 255.0, green / 255.0, blue / 255.0);
  }

  update(): void {
    const { lightEstimation = null } = AR.getCurrentFrame({
      [ARFrameAttribute.LightEstimation]: true,
    });
    this.data = lightEstimation;
  }
}

export default Light;