#import "/src/cetz.typ": draw
#import "gate.typ"

#let draw-shape(id, tl, tr, br, bl, fill, stroke) = {
  let (x, y) = bl
  let (width, height) = (tr.at(0) - x, tr.at(1) - y)

  let t = (x + width / 2, y + height)
  let b = (x + width / 2, y)

  let ctrl-bl = (x + width / 2, y)
  let ctrl-br = (x + width * 0.8, y + height * 0.1)
  let ctrl-tl = (x + width / 2, y + height)
  let ctrl-tr = (x + width * 0.8, y + height * 0.9)

  let l = (x + width * 0.2, y + height / 2)
  let r = (x + width, y + height / 2)

  let f = draw.group(name: id, {
    draw.merge-path(
      inset: 0.5em,
      fill: fill,
      stroke: stroke,
      name: id + "-path", {
        draw.bezier-through(bl, l, tl)
        draw.bezier((), r, ctrl-tl, ctrl-tr)
        draw.bezier((), bl, ctrl-br, ctrl-bl)
      }
    )
    
    draw.intersections("i",
      id + "-path",
      draw.hide(draw.line(t, b))
    )
    draw.anchor("north", "i.0")
    draw.anchor("south", "i.1")
  })
  return (f, tl, tr, br, bl)
}

/// Draws an OR gate. This function is also available as `element.gate-or()`
/// 
/// For parameters, see #doc-ref("gates.gate")
/// #examples.gate-or
#let gate-or(
  x: none,
  y: none,
  w: none,
  h: none,
  inputs: 2,
  fill: none,
  stroke: black + 1pt,
  id: "",
  inverted: (),
  debug: (
    ports: false
  )
) = {
  gate.gate(
    draw-shape: draw-shape,
    x: x,
    y: y,
    w: w,
    h: h,
    inputs: inputs,
    fill: fill,
    stroke: stroke,
    id: id,
    inverted: inverted,
    debug: debug
  )
}

/// Draws a NOR gate. This function is also available as `element.gate-nor()`
/// 
/// For parameters, see #doc-ref("gates.gate")
/// #examples.gate-nor
#let gate-nor(
  x: none,
  y: none,
  w: none,
  h: none,
  inputs: 2,
  fill: none,
  stroke: black + 1pt,
  id: "",
  inverted: (),
  debug: (
    ports: false
  )
) = {
  gate-or(
    x: x,
    y: y,
    w: w,
    h: h,
    inputs: inputs,
    fill: fill,
    stroke: stroke,
    id: id,
    inverted: if inverted != "all" {inverted + ("out",)} else {inverted},
    debug: debug
  )
}