new center text positions

ikkez · ikkez · commit bfa36fd4b556 · 2020-03-03T10:48:38.000+01:00
diff --git a/anki_vector/annotate.py b/anki_vector/annotate.py
@@ -72,6 +72,7 @@ class AnnotationPosition(Enum):
     RIGHT = 2
     TOP = 4
     BOTTOM = 8
+    CENTER = 16
 
     #: Top left position
     TOP_LEFT = TOP | LEFT
@@ -85,6 +86,15 @@ class AnnotationPosition(Enum):
     #: Bottom right position
     BOTTOM_RIGHT = BOTTOM | RIGHT
 
+    #: Top center position
+    TOP_CENTER = TOP | CENTER
+
+    #: Middle of the screen position
+    CENTER_CENTER = CENTER | CENTER
+
+    #: Bottom center position
+    BOTTOM_CENTER = BOTTOM | CENTER
+
 
 class ImageText:  # pylint: disable=too-few-public-methods
     """ImageText represents some text that can be applied to an image.
@@ -156,15 +166,25 @@ def render(self, draw: ImageDraw.ImageDraw, bounds: tuple) -> ImageDraw.ImageDra
         (bx1, by1, bx2, by2) = bounds
         text_width, text_height = draw.textsize(self.text, font=self.font)
 
-        if self.position.value & AnnotationPosition.TOP.value:
-            y = by1
+        if self.position.value == AnnotationPosition.CENTER_CENTER.value:
+            y = by1 + (by2 - by1 - (text_height))/2
+            x = bx1 + (bx2 - bx1 - (text_width))/2
         else:
-            y = by2 - text_height
+            if (self.position.value & AnnotationPosition.LEFT.value or self.position.value & AnnotationPosition.RIGHT.value)\
+                    and self.position.value & AnnotationPosition.CENTER.value:
+                y = by1 + (by2 - by1 - (text_height)) / 2
+            if self.position.value & AnnotationPosition.TOP.value:
+                y = by1
+            else:
+                y = by2 - text_height
 
-        if self.position.value & AnnotationPosition.LEFT.value:
-            x = bx1
-        else:
-            x = bx2 - text_width
+            if (self.position.value & AnnotationPosition.TOP.value or self.position.value & AnnotationPosition.BOTTOM.value)\
+                    and self.position.value & AnnotationPosition.CENTER.value:
+                x = bx1 + (bx2 - bx1 - (text_width)) / 2
+            elif self.position.value & AnnotationPosition.LEFT.value:
+                x = bx1
+            else:
+                x = bx2 - text_width
 
         # helper method for each draw call below
         def _draw_text(pos, color):
