From c236fdb81b1249e1bcab0a027af81c987f96a60b Mon Sep 17 00:00:00 2001
From: Jan Petykiewicz <anewusername@gmail.com>
Date: Mon, 11 May 2020 20:31:07 -0700
Subject: [PATCH] Add __repr__ methods to most objects`

These are closer to what __str__ should be and will likely change in the
future.
---
 masque/label.py          |  4 ++++
 masque/pattern.py        |  4 ++++
 masque/repetition.py     | 11 +++++++++++
 masque/shapes/arc.py     |  7 +++++++
 masque/shapes/circle.py  |  4 ++++
 masque/shapes/ellipse.py |  6 ++++++
 masque/shapes/path.py    |  6 ++++++
 masque/shapes/polygon.py |  6 ++++++
 masque/shapes/text.py    |  7 +++++++
 masque/subpattern.py     |  9 +++++++++
 10 files changed, 64 insertions(+)

diff --git a/masque/label.py b/masque/label.py
index 972370b..e09c60e 100644
--- a/masque/label.py
+++ b/masque/label.py
@@ -170,3 +170,7 @@ class Label:
         object.__setattr__(self, 'locked', False)
         self.offset.flags.writeable = True
         return self
+
+    def __repr__(self) -> str:
+        locked = ' L' if self.locked else ''
+        return f'<Label "{self.string}" l{self.layer} o{self.offset}{locked}>'
diff --git a/masque/pattern.py b/masque/pattern.py
index a0313a6..eb25c97 100644
--- a/masque/pattern.py
+++ b/masque/pattern.py
@@ -876,3 +876,7 @@ class Pattern:
 
         toplevel = list(patterns - not_toplevel)
         return toplevel
+
+    def __repr__(self) -> str:
+        locked = ' L' if self.locked else ''
+        return (f'<Pattern "{self.name}": sh{len(self.shapes)} sp{len(self.subpatterns)} la{len(self.labels)}{locked}>')
diff --git a/masque/repetition.py b/masque/repetition.py
index 6c7894b..774c8da 100644
--- a/masque/repetition.py
+++ b/masque/repetition.py
@@ -531,3 +531,14 @@ class GridRepetition:
         self.unlock()
         self.pattern.deepunlock()
         return self
+
+    def __repr__(self) -> str:
+        name = self.pattern.name if self.pattern is not None else None
+        rotation = f' r{self.rotation*180/pi:g}' if self.rotation != 0 else ''
+        scale = f' d{self.scale:g}' if self.scale != 1 else ''
+        mirrored = ' m{:d}{:d}'.format(*self.mirrored) if self.mirrored.any() else ''
+        dose = f' d{self.dose:g}' if self.dose != 1 else ''
+        locked = ' L' if self.locked else ''
+        bv = f', {self.b_vector}' if b_vector is not None else ''
+        return (f'<GridRepetition "{name}" at {self.offset} {rotation}{scale}{mirrored}{dose}'
+                f' {self.a_count}x{self.b_count} ({self.a_vector}{bv}){locked}>')
diff --git a/masque/shapes/arc.py b/masque/shapes/arc.py
index 6b1a5bf..7b1cedc 100644
--- a/masque/shapes/arc.py
+++ b/masque/shapes/arc.py
@@ -398,3 +398,10 @@ class Arc(Shape):
         self.radii.flags.writeable = True
         self.angles.flags.writeable = True
         return self
+
+    def __repr__(self) -> str:
+        angles = f' a°{self.angles*180/pi}'
+        rotation = f' r°{self.rotation*180/pi:g}' if self.rotation != 0 else ''
+        dose = f' d{self.dose:g}' if self.dose != 1 else ''
+        locked = ' L' if self.locked else ''
+        return f'<Arc l{self.layer} o{self.offset} r{self.radii}{angles} w{self.width:g}{rotation}{dose}{locked}>'
diff --git a/masque/shapes/circle.py b/masque/shapes/circle.py
index 8f008dd..f7221d5 100644
--- a/masque/shapes/circle.py
+++ b/masque/shapes/circle.py
@@ -110,3 +110,7 @@ class Circle(Shape):
                (self.offset, magnitude, rotation, False, self.dose), \
                lambda: Circle(radius=norm_value, layer=self.layer)
 
+    def __repr__(self) -> str:
+        dose = f' d{self.dose:g}' if self.dose != 1 else ''
+        locked = ' L' if self.locked else ''
+        return f'<Circle l{self.layer} o{self.offset} r{self.radius:g}{dose}{locked}>'
diff --git a/masque/shapes/ellipse.py b/masque/shapes/ellipse.py
index 12a2baf..9779e69 100644
--- a/masque/shapes/ellipse.py
+++ b/masque/shapes/ellipse.py
@@ -189,3 +189,9 @@ class Ellipse(Shape):
         Shape.unlock(self)
         self.radii.flags.writeable = True
         return self
+
+    def __repr__(self) -> str:
+        rotation = f' r{self.rotation*180/pi:g}' if self.rotation != 0 else ''
+        dose = f' d{self.dose:g}' if self.dose != 1 else ''
+        locked = ' L' if self.locked else ''
+        return f'<Ellipse l{self.layer} o{self.offset} r{self.radii}{rotation}{dose}{locked}>'
diff --git a/masque/shapes/path.py b/masque/shapes/path.py
index 57ae873..db7831a 100644
--- a/masque/shapes/path.py
+++ b/masque/shapes/path.py
@@ -407,3 +407,9 @@ class Path(Shape):
         if self.cap_extensions is not None:
             self.cap_extensions.flags.writeable = True
         return self
+
+    def __repr__(self) -> str:
+        centroid = self.offset + self.vertices.mean(axis=0)
+        dose = f' d{self.dose:g}' if self.dose != 1 else ''
+        locked = ' L' if self.locked else ''
+        return f'<Path l{self.layer} centroid {centroid} v{len(self.vertices)} w{self.width} c{self.cap}{dose}{locked}>'
diff --git a/masque/shapes/polygon.py b/masque/shapes/polygon.py
index 6d19714..71c9491 100644
--- a/masque/shapes/polygon.py
+++ b/masque/shapes/polygon.py
@@ -339,3 +339,9 @@ class Polygon(Shape):
         Shape.unlock(self)
         self.vertices.flags.writeable = True
         return self
+
+    def __repr__(self) -> str:
+        centroid = self.offset + self.vertices.mean(axis=0)
+        dose = f' d{self.dose:g}' if self.dose != 1 else ''
+        locked = ' L' if self.locked else ''
+        return f'<Polygon l{self.layer} centroid {centroid} v{len(self.vertices)}{dose}{locked}>'
diff --git a/masque/shapes/text.py b/masque/shapes/text.py
index 355600e..bb8ed0d 100644
--- a/masque/shapes/text.py
+++ b/masque/shapes/text.py
@@ -253,3 +253,10 @@ def get_char_as_polygons(font_path: str,
         Shape.unlock(self)
         self.mirrored.flags.writeable = True
         return self
+
+    def __repr__(self) -> str:
+        rotation = f' r°{self.rotation*180/pi:g}' if self.rotation != 0 else ''
+        dose = f' d{self.dose:g}' if self.dose != 1 else ''
+        locked = ' L' if self.locked else ''
+        mirrored = ' m{:d}{:d}'.format(*self.mirrored) if self.mirrored.any() else ''
+        return f'<TextShape "{self.string}" l{self.layer} o{self.offset} h{self.height:g}{rotation}{mirrored}{dose}{locked}>'
diff --git a/masque/subpattern.py b/masque/subpattern.py
index a6cc123..3f78d62 100644
--- a/masque/subpattern.py
+++ b/masque/subpattern.py
@@ -339,3 +339,12 @@ class SubPattern:
         self.unlock()
         self.pattern.deepunlock()
         return self
+
+    def __repr__(self) -> str:
+        name = self.pattern.name if self.pattern is not None else None
+        rotation = f' r{self.rotation*180/pi:g}' if self.rotation != 0 else ''
+        scale = f' d{self.scale:g}' if self.scale != 1 else ''
+        mirrored = ' m{:d}{:d}'.format(*self.mirrored) if self.mirrored.any() else ''
+        dose = f' d{self.dose:g}' if self.dose != 1 else ''
+        locked = ' L' if self.locked else ''
+        return f'<SubPattern "{name}" at {self.offset}{rotation}{scale}{mirrored}{dose}{locked}>'