Class AngleTable

ar.py:591–656

Coefficient matrix A for slope(direction).

class AngleTable(GeometricTable):
  """Coefficient matrix A for slope(direction)."""

  def __init__(self, name: str = ''):
    name = name or 'pi'
    super().__init__(name)
    self.pi = self.const

  def modulo(self, e: dict[str, float]) -> dict[str, float]:
    e = strip(e)
    if self.pi not in e:
      return super().modulo(e)

    e[self.pi] = e[self.pi] % 1
    return strip(e)

  def add_para(
      self, d1: gm.Direction, d2: gm.Direction, dep: pr.Dependency
  ) -> None:
    return self.add_const_angle(d1, d2, 0, dep)

  def add_const_angle(
      self, d1: gm.Direction, d2: gm.Direction, ang: float, dep: pr.Dependency
  ) -> None:
    if ang and d2._obj.num > d1._obj.num:  # pylint: disable=protected-access
      d1, d2 = d2, d1
      ang = 180 - ang

    d1, d2 = self.get_name([d1, d2])

    num, den = simplify(ang, 180)
    ang = frac(int(num), int(den))
    return super().add_eq3(d1, d2, ang, dep)

  def add_eqangle(
      self,
      d1: gm.Direction,
      d2: gm.Direction,
      d3: gm.Direction,
      d4: gm.Direction,
      dep: pr.Dependency,
  ) -> None:
    """Add the inequality d1-d2=d3-d4."""
    # Use string as variables.
    l1, l2, l3, l4 = [d._obj.num for d in [d1, d2, d3, d4]]  # pylint: disable=protected-access
    d1, d2, d3, d4 = self.get_name([d1, d2, d3, d4])
    ang1 = {d1: 1, d2: -1}
    ang2 = {d3: 1, d4: -1}

    if l2 > l1:
      ang1 = plus({self.pi: 1}, ang1)
    if l4 > l3:
      ang2 = plus({self.pi: 1}, ang2)

    ang12 = minus(ang1, ang2)
    self.record_eq(d1, d2, d3, d4)
    self.record_eq(d1, d3, d2, d4)