advent22/ui/src/components/door_map/RectangleCanvas.vue

<template>
  <ThouCanvas
    @mousedown.left="draw_start"
    @mouseup.left="draw_finish"
    @mousedown.right="drag_start"
    @mouseup.right="drag_finish"
    @mousemove="on_mousemove"
    @click.middle="remove_rect"
    @dblclick.left="remove_rect"
  >
    <SVGRect
      v-for="(rect, index) in rectangles"
      :key="`rect-${index}`"
      :rectangle="rect"
    />
    <SVGRect v-if="preview_visible" :focused="true" :rectangle="preview_rect" />
  </ThouCanvas>
</template>

<script lang="ts">
import { Vue, Options } from "vue-class-component";
import { Vector2D, Rectangle } from "../rects/rectangles";
import ThouCanvas from "../rects/ThouCanvas.vue";
import SVGRect from "../rects/SVGRect.vue";

enum CanvasState {
  Idle,
  Drawing,
  Dragging,
}

@Options({
  components: {
    ThouCanvas,
    SVGRect,
  },
  props: {
    rectangles: Array,
  },
  emits: ["add_rect", "edit_rect", "remove_rect"],
})
export default class extends Vue {
  private readonly min_rect_area = 300;
  private state = CanvasState.Idle;
  private preview_rect = new Rectangle();
  private drag_rect?: Rectangle;
  private drag_origin = new Vector2D();
  private rectangles: Rectangle[] = [];

  private get preview_visible(): boolean {
    return this.state !== CanvasState.Idle;
  }

  private pop_rectangle(point: Vector2D): Rectangle | undefined {
    const idx = this.rectangles.findIndex((rect) => rect.contains(point));

    if (idx === -1) {
      return;
    }

    return this.rectangles.splice(idx, 1)[0];
  }

  private draw_start(event: MouseEvent, point: Vector2D) {
    if (this.preview_visible) {
      return;
    }

    this.state = CanvasState.Drawing;
    this.preview_rect = new Rectangle(point, point);
  }

  private draw_finish() {
    if (this.state !== CanvasState.Drawing || this.preview_rect === undefined) {
      return;
    }

    this.state = CanvasState.Idle;

    if (this.preview_rect.area < this.min_rect_area) {
      return;
    }

    this.rectangles.push(this.preview_rect);
    this.$emit("add_rect", this.preview_rect);
  }

  private drag_start(event: MouseEvent, point: Vector2D) {
    if (this.preview_visible) {
      return;
    }

    this.drag_rect = this.pop_rectangle(point);

    if (this.drag_rect === undefined) {
      return;
    }

    this.state = CanvasState.Dragging;
    this.drag_origin = point;

    this.preview_rect = this.drag_rect;
  }

  private drag_finish() {
    if (
      this.state !== CanvasState.Dragging ||
      this.preview_rect === undefined
    ) {
      return;
    }

    this.state = CanvasState.Idle;
    this.rectangles.push(this.preview_rect);
    this.$emit("edit_rect", this.drag_rect, this.preview_rect);
  }

  private on_mousemove(event: MouseEvent, point: Vector2D) {
    if (this.preview_rect === undefined) {
      return;
    }

    if (this.state === CanvasState.Drawing) {
      this.preview_rect = this.preview_rect.update(undefined, point);
    } else if (this.state === CanvasState.Dragging && this.drag_rect) {
      const movement = point.minus(this.drag_origin);
      this.preview_rect = this.drag_rect.move(movement);
    }
  }

  private remove_rect(event: MouseEvent, point: Vector2D) {
    if (this.preview_visible) {
      return;
    }

    this.pop_rectangle(point);
  }
}
</script>

<style scoped>
* {
  cursor: crosshair;
}
</style>