From 6d7dee82bc19940397b3bd9ecd1c1bbbbfcbea1f Mon Sep 17 00:00:00 2001
From: gdamms <damguillotin@gmail.com>
Date: Fri, 25 Mar 2022 18:00:37 +0100
Subject: [PATCH] fix: start and stop point

---
 src/garage.py |  6 +++---
 src/main.py   | 14 ++++++++++++--
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/src/garage.py b/src/garage.py
index f87b77b..54f0764 100644
--- a/src/garage.py
+++ b/src/garage.py
@@ -31,10 +31,10 @@ NB_WHEELS = 23
 NB_HATS = 20
 NB_TEAMS = 2
 
-DELAY = 0.1
+DELAY = 0.01
 
-NB_PRIMARY_COLORS = 5
-NB_SECONDARY_COLORS = 5
+NB_PRIMARY_COLORS = 3
+NB_SECONDARY_COLORS = 1
 
 
 old_model = None
diff --git a/src/main.py b/src/main.py
index fa2ca84..d42bcd5 100644
--- a/src/main.py
+++ b/src/main.py
@@ -31,10 +31,20 @@ if __name__ == "__main__":
     database = connect_to_database("dataset.db")
 
     loadouts = generate_loadouts()
-    NB_ROTATIONS = 20
+    NB_ROTATIONS = 1
+    N_START = 9984
+    N_STOP = 1e10
 
     while detect_focus(device):
+
         for (i, loadout) in enumerate(loadouts):
+            # Sauter les N premiers
+            if i < N_START:
+                continue
+
+            if i >= N_STOP:
+                exit(0)
+
             ((model, sticker), wheel, hat, team, primary_color, secondary_color) = loadout
 
             logging.debug(f"loadout {i}: {((model, sticker), wheel, hat, team, primary_color, secondary_color)}")
@@ -42,9 +52,9 @@ if __name__ == "__main__":
 
             sleep(0.5)
             for x_rotation in range(NB_ROTATIONS):
-                rotate(130, 0)
                 uuid = take_screenshot(device)
                 insert_into_database(database, uuid, loadout, x_rotation, 0)
+                # rotate(130, 0)
 
     set_notifications(device, True)