h6-nandsim/editor/src/editor/states.ts

import { type ComponentKind } from "./Board";
import { ConnectingWire, Selection } from "./Cx";
import { SelectionBox, type Cx, type Tool } from "./Cx";
import { v2, type V2 } from "./V2";

export interface State {
  enterState?(): void;
  leaveState?(): void;
  onMouseDown?(pos: V2): void;
  onMouseUp?(pos: V2): void;
  onMouseMove?(deltaPos: V2, pos: V2): void;
  onKeyDown?(key: string): void;
  onKeyUp?(key: string): void;
  selectedTool?(): Tool | null;
}

export class Normal implements State {
  constructor(private cx: Cx) {}

  onMouseDown(pos: V2): void {
    if (
      this.cx.board.handleMouseClick(
        pos.sub(this.cx.offset),
        (comp, i) => {
          console.log({ comp, i });
          this.cx.connectingWire = new ConnectingWire(
            {
              tag: "InputPin",
              comp,
              i,
            },
            pos,
          );
          this.cx.transitionTo(new Wiring(this.cx));
        },
        (comp, i) => {
          this.cx.connectingWire = new ConnectingWire(
            {
              tag: "OutputPin",
              comp,
              i,
            },
            pos,
          );
          this.cx.transitionTo(new Wiring(this.cx));
        },
        (comp) => {
          this.cx.selection = new Selection();
          this.cx.selection.addComponent(comp);
          this.cx.transitionTo(new Selecting(this.cx));
        },
      ) === "handled"
    ) {
      return;
    } else {
      this.cx.selectionBox = new SelectionBox(pos);
      this.cx.transitionTo(new SelectingBox(this.cx));
    }
  }

  onMouseMove(_deltaPos: V2, pos: V2): void {
    this.cx.board.updateMouseHover(pos.sub(this.cx.offset));
  }

  onKeyDown(key: string): void {
    if (key === "Shift") {
      this.cx.transitionTo(new Panning(this.cx));
      return;
    }
  }

  selectedTool(): Tool | null {
    return "select";
  }
}

export class Panning implements State {
  private dragging = false;

  constructor(private cx: Cx) {}

  onMouseDown(_pos: V2): void {
    this.dragging = true;
  }

  onMouseUp(_pos: V2): void {
    this.dragging = false;
  }

  onMouseMove(deltaPos: V2): void {
    if (this.dragging) {
      this.cx.moveOffset(deltaPos);
    }
  }

  onKeyDown(key: string): void {
    if (key === "Escape") {
      this.cx.transitionTo(new Normal(this.cx));
      return;
    }
  }

  onKeyUp(key: string): void {
    if (key === "Shift") {
      this.cx.transitionTo(new Normal(this.cx));
      return;
    }
  }

  selectedTool(): Tool | null {
    return "pan";
  }
}

export class Placing implements State {
  private compDef: ComponentKind;

  constructor(
    private cx: Cx,
    private tool: Tool,
  ) {
    this.compDef = this.cx.componentRepo.get(this.tool);
  }

  enterState(): void {
    this.cx.addComponentPlacer(v2(0, 0), this.compDef.size);
  }

  leaveState(): void {
    this.cx.removeComponentPlacer();
  }

  onMouseDown(pos: V2): void {
    const boardPos = this.cx.canvasPosToBoard(pos);
    if (this.cx.board.canPlaceComponent(this.compDef, boardPos)) {
      this.cx.board.placeComponent(this.compDef, boardPos);
      this.cx.transitionTo(new Normal(this.cx));
    }
  }

  onMouseMove(_deltaPos: V2, pos: V2): void {
    this.cx.setComponentPlacerPos(pos);
  }

  onKeyDown(key: string): void {
    if (key === "Escape") {
      this.cx.transitionTo(new Normal(this.cx));
      return;
    }
  }

  selectedTool(): Tool | null {
    return this.tool;
  }
}

export class Selecting implements State {
  constructor(private cx: Cx) {}

  onMouseDown(pos: V2): void {
    if (
      this.cx.board.handleMouseClick(
        pos.sub(this.cx.offset),
        (_comp, _i) => {},
        (_comp, _i) => {},
        (comp) => {
          if (this.cx.keysPressed.has("Control")) {
            this.cx.selection?.toggleComponent(comp);
          } else {
            this.cx.selection = new Selection();
            this.cx.selection.addComponent(comp);
          }
        },
      ) === "handled"
    ) {
      return;
    } else {
      if (this.cx.keysPressed.has("Control")) {
        this.cx.selectionBox = new SelectionBox(pos);
        this.cx.transitionTo(new SelectingBox(this.cx));
      } else {
        this.cx.selection = null;
        this.cx.transitionTo(new Normal(this.cx));
      }
    }
  }
}

export class SelectingBox implements State {
  constructor(private cx: Cx) {}

  onMouseUp(_pos: V2): void {
    if (!this.cx.selectionBox) {
      throw new Error("expected selectionBox to active");
    }
    const { pos, size } = this.cx.selectionBox.normalized();
    const selected = this.cx.board.componentsInRect(
      pos.sub(this.cx.offset),
      size,
    );
    if (selected.length > 0) {
      this.cx.selection ??= new Selection();
    }
    for (const comp of selected) {
      this.cx.selection?.addComponent(comp);
    }
    if (this.cx.selection) {
      this.cx.selectionBox = null;
      this.cx.transitionTo(new Selecting(this.cx));
    } else {
      this.cx.selectionBox = null;
      this.cx.transitionTo(new Normal(this.cx));
    }
  }

  onMouseMove(deltaPos: V2): void {
    this.cx.selectionBox?.move(deltaPos);
  }

  selectedTool(): Tool | null {
    return "select";
  }
}

export class Wiring implements State {
  constructor(private cx: Cx) {}

  onMouseMove(_deltaPos: V2, pos: V2): void {
    if (!this.cx.connectingWire) {
      throw new Error("expected connectingWire to be active");
    }
    this.cx.connectingWire.move(pos);
    this.cx.board.updateMouseHover(pos.sub(this.cx.offset));
  }

  onKeyDown(key: string): void {
    if (key === "Escape") {
      this.cx.transitionTo(new Normal(this.cx));
      this.cx.connectingWire = null;
      return;
    }
  }
}