From 34b66cfa0ba454ad9d830e2b68dce4d4ce63882c Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Tue, 25 Apr 2023 17:48:38 +0200
Subject: Adding color attribute.

---
 src/argaze/ArUcoMarkers/ArUcoMarker.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/argaze/ArUcoMarkers/ArUcoMarker.py b/src/argaze/ArUcoMarkers/ArUcoMarker.py
index f41810b..93d44e8 100644
--- a/src/argaze/ArUcoMarkers/ArUcoMarker.py
+++ b/src/argaze/ArUcoMarkers/ArUcoMarker.py
@@ -33,6 +33,9 @@ class ArUcoMarker():
     points: numpy.array = field(init=False, repr=False)
     """Estimated 3D corners positions in camera world referential."""
 
+    color: tuple = field(init=False, repr=False, default_factory=lambda : (0, 255, 0))
+    """Color used to draw marker on frame."""
+
     @property
     def center(self) -> numpy.array:
         """Get 2D center position in camera image referential."""
@@ -57,7 +60,7 @@ class ArUcoMarker():
 
             cv.drawFrameAxes(frame, numpy.array(K), numpy.array(D), self.rotation, self.translation, self.size)
 
-        aruco.drawDetectedMarkers(frame, [self.corners], numpy.array([self.identifier]))
+        aruco.drawDetectedMarkers(frame, [self.corners], numpy.array([self.identifier]), self.color)
 
     def save(self, destination_folder, dpi):
         """Save marker image as .png file into a destination folder."""
-- 
cgit v1.1