From 67a7224f427986f3f23a1ee0059bdd185cc9f591 Mon Sep 17 00:00:00 2001
From: didier amyot <didier.amyot@gmail.com>
Date: Sun, 21 Jan 2024 12:06:17 -0500
Subject: [PATCH] Added some code.

---
 .gitignore                |  1 +
 dodgeballtracker/logic.py | 25 +++++++++++++++++++++++++
 frontend/style.css        |  5 +++--
 3 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index e7e1e81..5416127 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,3 +3,4 @@ env
 .idea
 .DS_Store
 *.pt
+*.png
diff --git a/dodgeballtracker/logic.py b/dodgeballtracker/logic.py
index 0457ac8..aab86dd 100644
--- a/dodgeballtracker/logic.py
+++ b/dodgeballtracker/logic.py
@@ -63,6 +63,31 @@ def crazy_experiment(frame):
     cv2.polylines(frame, pts, True, (0, 255, 255), 8)
 
 
+def get_court_perspective_transform():
+    """
+    """
+    empirical_top_left = np.float32([416, 175])
+    empirical_top_right = np.float32([905, 173])
+    empirical_bottom_left = np.float32([51, 702])
+    empirical_bottom_right = np.float32([1188, 695])
+
+    court_top_left = np.float32([0, 0])
+    court_top_right = np.float32([9, 0])
+    court_bottom_left = np.float32([0, 18])
+    court_bottom_right = np.float32([9, 18])
+
+    empirical_source = np.float32([empirical_top_left,
+                                  empirical_top_right,
+                                  empirical_bottom_left,
+                                  empirical_bottom_right])
+    pixel_per_meter = 50
+    empirical_target = np.float32([court_top_left,
+                                  court_top_right,
+                                  court_bottom_left,
+                                  court_bottom_right]) * pixel_per_meter
+    return cv2.getPerspectiveTransform(empirical_source, empirical_target)
+
+
 def main(input_filename: Path, output_filename: Path, detected_box_objects):
     # define some constants
     detected_box_objects = sorted(detected_box_objects, reverse=True)
diff --git a/frontend/style.css b/frontend/style.css
index f7912f7..d7a1533 100644
--- a/frontend/style.css
+++ b/frontend/style.css
@@ -1,11 +1,12 @@
 .court-border {
     stroke: red;
     stroke-width: 4px;
-    fill-opacity: 0;
+    /*fill-opacity: 0;*/
+    fill: gray;
 }
 
 .return-line {
-    stroke: yellow;
+    stroke: blue;
     stroke-width: 2px;
     fill-opacity: 0;
 }
\ No newline at end of file