
import 'package:vector_math/vector_math_64.dart';
import '../thermion_viewer.dart';

class Gizmo extends AbstractGizmo {
  final ThermionEntity x;
  Vector3 _x = Vector3(0.1, 0, 0);
  final ThermionEntity y;
  Vector3 _y = Vector3(0.0, 0.1, 0);
  final ThermionEntity z;
  Vector3 _z = Vector3(0.0, 0.0, 0.1);

  final ThermionViewer controller;

  ThermionEntity? _activeAxis;
  ThermionEntity? _activeEntity;
  bool get isActive => _activeAxis != null;

  final Set<ThermionEntity> ignore;

  Gizmo(this.x, this.y, this.z, this.controller,
      {this.ignore = const <ThermionEntity>{}}) {
    controller.pickResult.listen(_onPickResult);
  }

  Future _reveal() async {
    await controller.reveal(x, null);
    await controller.reveal(y, null);
    await controller.reveal(z, null);
  }

  void translate(double transX, double transY) async {
    late Vector3 vec;
    if (_activeAxis == x) {
      vec = _x;
    } else if (_activeAxis == y) {
      vec = _y;
    } else if (_activeAxis == z) {
      vec = _z;
    }
    await controller.queuePositionUpdate(
        _activeEntity!, transX * vec.x, -transY * vec.y, -transX * vec.z,
        relative: true);
  }

  void reset() {
    _activeAxis = null;
  }

  void _onPickResult(FilamentPickResult result) async {
    if (ignore.contains(result)) {
      detach();
      return;
    }
    if (result.entity == x || result.entity == y || result.entity == z) {
      _activeAxis = result.entity;
    } else {
      attach(result.entity);
    }
  }

  void attach(ThermionEntity entity) async {
    _activeAxis = null;
    _activeEntity = entity;
    await _reveal();
    await controller.setParent(x, entity);
    await controller.setParent(y, entity);
    await controller.setParent(z, entity);
  }

  void detach() async {
    await controller.setParent(x, 0);
    await controller.setParent(y, 0);
    await controller.setParent(z, 0);
    await controller.hide(x, null);
    await controller.hide(y, null);
    await controller.hide(z, null);
  }
}
