yvan-rcs-web/src/core/controls/EsDragControls.ts

import * as THREE from 'three'
import { DragControls } from './DragControls.js'
import type Viewport from '@/core/engine/Viewport.ts'
import { getItemTypeByName } from '@/model/itemType/ItemTypeDefine'
import type { ItemTypeDefineOption } from '@/model/itemType/ItemTypeDefine'
import { markRaw } from 'vue'
import EventBus from '@/runtime/EventBus'
import { getInteraction, getMeta } from '@/core/manager/ModuleManager.ts'
import type BaseInteraction from '@/core/base/BaseInteraction.ts'

// dragControls 绑定函数
let dragStartFn, dragFn, dragEndFn, clickblankFn

export default class EsDragControls {
  _dragObjects: THREE.Object3D[] = [] // 拖拽对象
  dragControls: any
  private onDownPosition: { x: number; y: number } = { x: -1, y: -1 }

  viewport: Viewport
  currentInteraction: BaseInteraction
  isDragging = false

  constructor(viewport) {
    this.viewport = viewport

    // 物体拖拽控制器
    this.dragControls = new DragControls(this._dragObjects, viewport.camera, viewport.renderer.domElement)
    this.dragControls.deactivate() // 默认禁用
    dragStartFn = this.dragControlsStart.bind(this)
    this.dragControls.addEventListener('dragstart', dragStartFn)
    dragFn = this.drag.bind(this)
    this.dragControls.addEventListener('drag', dragFn)
    dragEndFn = this.dragControlsEnd.bind(this)
    this.dragControls.addEventListener('dragend', dragEndFn)
    // 点击可拖拽物体之外
    clickblankFn = this.clickblank.bind(this)
    this.dragControls.addEventListener('clickblank', clickblankFn)
  }

  set domElement(element: HTMLElement) {
    this.dragControls.setDomElement(element)
  }

  setDragObjects(objects: THREE.Object3D[], type: 'eq' | 'push' | 'remove' = 'eq') {
    // 当前拖拽对象为空时加入对象需激活控制器
    if (this._dragObjects.length === 0) {
      if (objects.length > 0) {
        this.dragControls.activate()
      }

      this._dragObjects = objects
    } else {
      // 当前拖拽对象不为空时
      if (type === 'eq') {
        // 是清空拖拽对象的设置，则禁用控制器
        if (objects.length === 0) {
          this.dragControls.deactivate()
        }

        this._dragObjects = objects
      } else if (type === 'push') {
        this._dragObjects.push(...objects)
      } else if (type === 'remove') {
        this._dragObjects = this._dragObjects.filter((item) => !objects.includes(item))
      }
    }

    this.dragControls.setObjects(this._dragObjects)
  }

  // 拖拽开始
  dragControlsStart(e) {
    // 右键拖拽不响应
    if (e.e.button === 2 || !e.object?.visible) return

    const type = e.object.userData?.t
    const entityId = e.object.userData?.entityId

    if (!type || !entityId) return

    this.currentInteraction = getInteraction(e.object.userData?.t)
    if (!this.currentInteraction) {
      return
    }

    // 有效拖拽对象
    const enable = this.currentInteraction.dragPointStart(this.viewport, {
      object: e.object,
      entityId: entityId
    })
    if (!enable) return

    e.e.preventDefault()

    // 拖拽时禁用其他控制器
    this.viewport.controls.enabled = false

    this.isDragging = true

    // 记录拖拽按下的位置和对象
    this.onDownPosition = { x: e.e.clientX, y: e.e.clientY }

    // const itemType: ItemTypeDefineOption = getItemTypeByName(e.object.userData.type)
    // if (itemType?.clazz) {
    //   itemType.clazz.dragPointStart(this.viewport, e.object)
    // }
    // switch (e.object.userData.type) {
    //   case Constract.MeasureMarker:
    //     this.viewport.measure.dragPointStart(e.object)
    //     break
    // }
  }

  // 拖拽中
  drag(e) {
    this.currentInteraction?.dragPointMove(this.viewport, e)
  }

  // 拖拽结束
  dragControlsEnd(e) {
    // 右键拖拽不响应
    if (e.e.button === 2 || !e.object.visible) return
    if (!e.object.userData?.t) return

    // 单选点击
    if (this.onDownPosition.x === e.e.clientX && this.onDownPosition.y === e.e.clientY) {
      if (e.object.userData.onClick) {
        e.object.userData.onClick(e)
      }
    }

    const ret = this.currentInteraction?.dragPointComplete(this.viewport, e)
    if (!ret) return

    EventBus.dispatch('selectedObjectPropertyChanged', {})

    // 拖拽结束启用其他控制器
    this.viewport.controls.enabled = true
    this.isDragging = false

    // if (e.object.userData?.type) {
    //   const itemType: ItemTypeDefineOption = getItemTypeByName(e.object.userData.type)
    //   if (itemType?.clazz) {
    //     itemType.clazz.dragPointComplete(this.viewport)
    //   }
    // }
    // switch (e.object.userData.type) {
    //   case Constract.MeasureMarker:
    //     this.viewport.measure.dragPointComplete()
    //     break
    // }
  }

  // 点击可拖拽物体之外
  clickblank(e) {
    if (e.e.button === 2) return
  }

  dispose() {
    this._dragObjects = []

    this.dragControls.removeEventListener('dragstart', dragStartFn)
    this.dragControls.removeEventListener('dragend', dragEndFn)
    this.dragControls.dispose()
  }
}