epam/miew

import gfxutils from '../gfxutils'
import { Matrix4, Object3D, Ray } from 'three'

class TransformGroup extends Object3D {
  static _inverseMatrix = new Matrix4()

  static _ray = new Ray()

  constructor(geometry, geoParams, material, transforms) {
    super()
    this._geometry = geometry
    this._geoParams = geoParams
    const mat = material.createInstance()
    geoParams.initMaterial(mat)
    this._material = mat
    this._transforms = transforms.length > 0 ? transforms : [new Matrix4()]
    const meshes = this._createMeshes(geometry)
    for (let i = 0, n = meshes.length; i < n; ++i) {
      this.add(meshes[i])
    }
  }

  raycast(raycaster, intersects) {
    const ray = TransformGroup._ray
    const inverseMatrix = TransformGroup._inverseMatrix
    const { children } = this
    ray.copy(raycaster.ray)
    for (let i = 0, n = children.length; i < n; ++i) {
      const child = children[i]

      if (!gfxutils.belongToSelectLayers(child)) {
        continue
      }

      child.updateMatrixWorld()
      const mtx = child.matrixWorld
      inverseMatrix.copy(mtx).invert()
      raycaster.ray.copy(ray).applyMatrix4(inverseMatrix)
      const childIntersects = []
      this._geometry.raycast(raycaster, childIntersects)

      for (let j = 0, ciCount = childIntersects.length; j < ciCount; ++j) {
        const inters = childIntersects[j]
        if (inters.point) {
          inters.point.applyMatrix4(mtx)
          inters.distance = ray.origin.distanceTo(inters.point)
        }
        inters.object = child
        intersects[intersects.length] = inters
      }
    }
    raycaster.ray.copy(ray)
  }

  getSubset(chunkIndices) {
    const geos = this._geometry.getSubset(chunkIndices)
    const subset = []
    let subIdx = 0

    for (let i = 0, n = geos.length; i < n; ++i) {
      const meshes = this._createMeshes(geos[i])
      for (let j = 0, meshCnt = meshes.length; j < meshCnt; ++j) {
        subset[subIdx++] = meshes[j]
      }
    }

    return subset
  }

  _createMeshes(geometry) {
    const transforms = this._transforms
    const Mesh = this._geoParams.Object
    const material = this._material
    const meshes = []
    for (let i = 0, n = transforms.length; i < n; ++i) {
      const mesh = new Mesh(geometry, material)
      mesh.applyMatrix4(transforms[i])
      meshes[i] = mesh
    }

    return meshes
  }
}

export default TransformGroup