pointvec/desktop/sources/scripts/tool.js

'use strict'

function Tool (dotgrid) {
  this.index = 0
  this.settings = { size: { width: 300, height: 300 }, crest: false }
  this.layers = [[], [], []]
  this.styles = [
    { thickness: 10, strokeLinecap: 'round', strokeLinejoin: 'round', color: '#f00', fill: 'none', mirror_style: 0, transform: 'rotate(45)' },
    { thickness: 10, strokeLinecap: 'round', strokeLinejoin: 'round', color: '#0f0', fill: 'none', mirror_style: 0, transform: 'rotate(45)' },
    { thickness: 10, strokeLinecap: 'round', strokeLinejoin: 'round', color: '#00f', fill: 'none', mirror_style: 0, transform: 'rotate(45)' }
  ]
  this.vertices = []
  this.reqs = { line: 2, arc_c: 2, arc_r: 2, arc_c_full: 2, arc_r_full: 2, bezier: 3, close: 0 }

  this.start = function () {
    this.styles[0].color = dotgrid.theme.active.f_high
    this.styles[1].color = dotgrid.theme.active.f_med
    this.styles[2].color = dotgrid.theme.active.f_low
  }

  this.erase = function () {
    this.layers = [[], [], []]
  }

  this.reset = function () {
    this.settings.crest = false
    this.styles[0].mirror_style = 0
    this.styles[1].mirror_style = 0
    this.styles[2].mirror_style = 0
    this.styles[0].fill = 'none'
    this.styles[1].fill = 'none'
    this.styles[2].fill = 'none'
    this.erase()
    this.vertices = []
    this.index = 0
  }

  this.clear = function () {
    this.vertices = []
    dotgrid.renderer.update()
    dotgrid.interface.update(true)
  }

  this.undo = function () {
    this.layers = dotgrid.history.prev()
    dotgrid.renderer.update()
    dotgrid.interface.update(true)
  }

  this.redo = function () {
    this.layers = dotgrid.history.next()
    dotgrid.renderer.update()
    dotgrid.interface.update(true)
  }

  this.length = function () {
    return this.layers[0].length + this.layers[1].length + this.layers[2].length
  }

  // I/O

  this.export = function (target = { settings: this.settings, layers: this.layers, styles: this.styles }) {
    return JSON.stringify(copy(target), null, 2)
  }

  this.import = function (layer) {
    this.layers[this.index] = this.layers[this.index].concat(layer)
    dotgrid.history.push(this.layers)
    this.clear()
    dotgrid.renderer.update()
    dotgrid.interface.update(true)
  }

  this.replace = function (dot) {
    if (!dot.layers || dot.layers.length != 3) { console.warn('Incompatible version'); return }

    if (dot.settings.width && dot.settings.height) {
      dot.settings.size = { width: dot.settings.width, height: dot.settings.height }
    }
    if (this.settings && (this.settings.size.width != dot.settings.size.width || this.settings.size.height != dot.settings.size.height)) {
      dotgrid.setSize({ width: dot.settings.size.width, height: dot.settings.size.height })
    }

    this.layers = dot.layers
    this.styles = dot.styles
    this.settings = dot.settings

    this.clear()
    dotgrid.renderer.update()
    dotgrid.interface.update(true)
    dotgrid.history.push(this.layers)
  }

  // EDIT

  this.removeSegment = function () {
    if (this.vertices.length > 0) { this.clear(); return }

    this.layer().pop()
    this.clear()
    dotgrid.renderer.update()
    dotgrid.interface.update(true)
  }

  this.removeSegmentsAt = function (pos) {
    for (const segmentId in this.layer()) {
      let segment = this.layer()[segmentId]
      for (const vertexId in segment.vertices) {
        let vertex = segment.vertices[vertexId]
        if (Math.abs(pos.x) == Math.abs(vertex.x) && Math.abs(pos.y) == Math.abs(vertex.y)) {
          segment.vertices.splice(vertexId, 1)
        }
      }
      if (segment.vertices.length < 2) {
        this.layers[this.index].splice(segmentId, 1)
      }
    }
    this.clear()
    dotgrid.renderer.update()
    dotgrid.interface.update(true)
  }

  this.selectSegmentAt = function (pos, source = this.layer()) {
    let target_segment = null
    for (const segmentId in source) {
      let segment = source[segmentId]
      for (const vertexId in segment.vertices) {
        let vertex = segment.vertices[vertexId]
        if (vertex.x == Math.abs(pos.x) && vertex.y == Math.abs(pos.y)) {
          return segment
        }
      }
    }
    return null
  }

  this.addVertex = function (pos) {
    pos = { x: Math.abs(pos.x), y: Math.abs(pos.y) }
    this.vertices.push(pos)
    dotgrid.interface.update(true)
  }

  this.vertexAt = function (pos) {
    for (const segmentId in this.layer()) {
      let segment = this.layer()[segmentId]
      for (const vertexId in segment.vertices) {
        let vertex = segment.vertices[vertexId]
        if (vertex.x == Math.abs(pos.x) && vertex.y == Math.abs(pos.y)) {
          return vertex
        }
      }
    }
    return null
  }

  this.addSegment = function (type, vertices) {
    let append_target = this.canAppend({ type: type, vertices: vertices })
    if (append_target) {
      this.layer()[append_target].vertices = this.layer()[append_target].vertices.concat(vertices)
    } else {
      this.layer().push({ type: type, vertices: vertices })
    }
  }

  this.cast = function (type) {
    if (!this.layer()) { this.layers[this.index] = [] }
    if (!this.canCast(type)) { console.warn('Cannot cast'); return }

    this.addSegment(type, this.vertices.slice())

    dotgrid.history.push(this.layers)

    this.clear()
    dotgrid.renderer.update()
    dotgrid.interface.update(true)

    console.log(`Casted ${type} -> ${this.layer().length} elements`)
  }

  this.i = { linecap: 0, linejoin: 0, thickness: 5 }

  this.toggle = function (type, mod = 1) {
    if (type == 'linecap') {
      let a = ['butt', 'square', 'round']
      this.i.linecap += mod
      this.style().strokeLinecap = a[this.i.linecap % a.length]
    } else if (type == 'linejoin') {
      let a = ['miter', 'round', 'bevel']
      this.i.linejoin += mod
      this.style().strokeLinejoin = a[this.i.linejoin % a.length]
    } else if (type == 'fill') {
      this.style().fill = this.style().fill == 'none' ? this.style().color : 'none'
    } else if (type == 'thickness') {
      this.style().thickness = clamp(this.style().thickness + mod, 1, 100)
    } else if (type == 'mirror') {
      this.style().mirror_style = this.style().mirror_style > 2 ? 0 : this.style().mirror_style + 1
    } else {
      console.warn('Unknown', type)
    }
    dotgrid.interface.update(true)
    dotgrid.renderer.update()
  }

  this.toggleCrest = function () {
    this.settings.crest = this.settings.crest !== true
    dotgrid.interface.update(true)
    dotgrid.renderer.update()
  }

  this.misc = function (type) {
    dotgrid.picker.start()
  }

  this.source = function (type) {
    if (type == 'grid') { dotgrid.renderer.toggle() }
    if (type == 'screen') { app.toggleFullscreen() }

    if (type == 'open') { dotgrid.open() }
    if (type == 'save') { dotgrid.save() }
    if (type == 'render') { dotgrid.render() }
    if (type == 'export') { dotgrid.export() }
  }

  this.canAppend = function (content) {
    for (const id in this.layer()) {
      let stroke = this.layer()[id]
      if (stroke.type != content.type) { continue }
      if (!stroke.vertices) { continue }
      if (!stroke.vertices[stroke.vertices.length - 1]) { continue }
      if (stroke.vertices[stroke.vertices.length - 1].x != content.vertices[0].x) { continue }
      if (stroke.vertices[stroke.vertices.length - 1].y != content.vertices[0].y) { continue }
      return id
    }
    return false
  }

  this.canCast = function (type) {
    if (!type) { return false }
    // Cannot cast close twice
    if (type == 'close') {
      let prev = this.layer()[this.layer().length - 1]
      if (!prev || prev.type == 'close') {
        return false
      }
    }
    if (type == 'bezier') {
      if (this.vertices.length != 3 && this.vertices.length != 5 && this.vertices.length != 7 && this.vertices.length != 9) {
        return false
      }
    }
    return this.vertices.length >= this.reqs[type]
  }

  this.paths = function () {
    let l1 = new Generator(dotgrid.tool.layers[0], dotgrid.tool.styles[0]).toString({ x: 0, y: 0 }, 1)
    let l2 = new Generator(dotgrid.tool.layers[1], dotgrid.tool.styles[1]).toString({ x: 0, y: 0 }, 1)
    let l3 = new Generator(dotgrid.tool.layers[2], dotgrid.tool.styles[2]).toString({ x: 0, y: 0 }, 1)

    return [l1, l2, l3]
  }

  this.path = function () {
    return new Generator(dotgrid.tool.layer(), dotgrid.tool.style()).toString({ x: 0, y: 0 }, 1)
  }

  this.translate = function (a, b) {
    for (const segmentId in this.layer()) {
      let segment = this.layer()[segmentId]
      for (const vertexId in segment.vertices) {
        let vertex = segment.vertices[vertexId]
        if (vertex.x == Math.abs(a.x) && vertex.y == Math.abs(a.y)) {
          segment.vertices[vertexId] = { x: Math.abs(b.x), y: Math.abs(b.y) }
        }
      }
    }
    dotgrid.history.push(this.layers)
    this.clear()
    dotgrid.renderer.update()
  }

  this.translateMulti = function (a, b) {
    const offset = { x: a.x - b.x, y: a.y - b.y }
    const segment = this.selectSegmentAt(a)

    if (!segment) { return }

    for (const vertexId in segment.vertices) {
      let vertex = segment.vertices[vertexId]
      segment.vertices[vertexId] = { x: vertex.x - offset.x, y: vertex.y - offset.y }
    }

    dotgrid.history.push(this.layers)
    this.clear()
    dotgrid.renderer.update()
  }

  this.translateLayer = function (a, b) {
    const offset = { x: a.x - b.x, y: a.y - b.y }
    for (const segmentId in this.layer()) {
      let segment = this.layer()[segmentId]
      for (const vertexId in segment.vertices) {
        let vertex = segment.vertices[vertexId]
        segment.vertices[vertexId] = { x: vertex.x - offset.x, y: vertex.y - offset.y }
      }
    }
    dotgrid.history.push(this.layers)
    this.clear()
    dotgrid.renderer.update()
  }

  this.translateCopy = function (a, b) {
    const offset = { x: a.x - b.x, y: a.y - b.y }
    const segment = this.selectSegmentAt(a, copy(this.layer()))

    if (!segment) { return }

    for (const vertexId in segment.vertices) {
      let vertex = segment.vertices[vertexId]
      segment.vertices[vertexId] = { x: vertex.x - offset.x, y: vertex.y - offset.y }
    }
    this.layer().push(segment)

    dotgrid.history.push(this.layers)
    this.clear()
    dotgrid.renderer.update()
  }

  this.merge = function () {
    const merged = [].concat(this.layers[0]).concat(this.layers[1]).concat(this.layers[2])
    this.erase()
    this.layers[this.index] = merged

    dotgrid.history.push(this.layers)
    this.clear()
    dotgrid.renderer.update()
  }

  // Style

  this.style = function () {
    if (!this.styles[this.index]) {
      this.styles[this.index] = []
    }
    return this.styles[this.index]
  }

  // Layers

  this.layer = function () {
    if (!this.layers[this.index]) {
      this.layers[this.index] = []
    }
    return this.layers[this.index]
  }

  this.selectLayer = function (id) {
    this.index = clamp(id, 0, 2)
    
    if (this.index !== 0) { this.settings.crest = false }

    this.clear()
    dotgrid.renderer.update()
    dotgrid.interface.update(true)
    console.log(`layer:${this.index}`)
  }

  this.selectNextLayer = function () {
    this.index = this.index >= 2 ? 0 : this.index++
    this.selectLayer(this.index)
  }

  this.selectPrevLayer = function () {
    this.index = this.index >= 0 ? 2 : this.index--
    this.selectLayer(this.index)
  }

  function copy (data) { return data ? JSON.parse(JSON.stringify(data)) : [] }
  function clamp (v, min, max) { return v < min ? min : v > max ? max : v }
}