From 25cbc57ecb87d75760f6c345721a5f81e361f0ce Mon Sep 17 00:00:00 2001
From: Brandon Rodriguez <brodriguez8774@gmail.com>
Date: Thu, 4 Nov 2021 03:50:21 -0400
Subject: [PATCH] Move some functions for organization
---
src/misc.py | 97 +++++++++++++++++++++++++++--------------------------
1 file changed, 49 insertions(+), 48 deletions(-)
diff --git a/src/misc.py b/src/misc.py
index ff57c55..e7831ac 100644
--- a/src/misc.py
+++ b/src/misc.py
@@ -169,6 +169,55 @@ def handle_mouse_click(data_manager, button_state, pos_x, pos_y):
data_manager.ideal_trash_paths = calc_trash_distances(data_manager)
calc_traveling_salesman(data_manager)
+
+def toggle_roomba_ai(data_manager):
+ """
+ Toggles roomba AI on or off. Program start default is off.
+ :param data_manager: Data manager data structure. Consolidates useful program data to one location.
+ """
+ logger.info('Toggling roomba ai.')
+ if data_manager.ai_active:
+ data_manager.ai_active = False
+ else:
+ data_manager.ai_active = True
+
+
+def set_roomba_vision_range_0(data_manager):
+ """
+ Adjusts roomba AI sight to see 0 tiles out from current location.
+ :param data_manager: Data manager data structure. Consolidates useful program data to one location.
+ """
+ logger.info('Setting roomba vision to "0 tiles" (bump sensor).')
+ data_manager.roomba_vision = 0
+
+
+def set_roomba_vision_range_1(data_manager):
+ """
+ Adjusts roomba AI sight to see 1 tiles out from current location.
+ :param data_manager: Data manager data structure. Consolidates useful program data to one location.
+ """
+ logger.info('Setting roomba vision to "1 tiles".')
+ data_manager.roomba_vision = 1
+
+
+def set_roomba_vision_range_2(data_manager):
+ """
+ Adjusts roomba AI sight to see 2 tiles out from current location.
+ :param data_manager: Data manager data structure. Consolidates useful program data to one location.
+ """
+ logger.info('Setting roomba vision to "2 tiles".')
+ data_manager.roomba_vision = 2
+
+
+def set_roomba_vision_range_full(data_manager):
+ """
+ Adjusts roomba AI sight to see all tiles on map.
+ :param data_manager: Data manager data structure. Consolidates useful program data to one location.
+ """
+ logger.info('Setting roomba vision to "full sight".')
+ data_manager.roomba_vision = -1
+
+
# endregion GUI Logic Functions
@@ -810,54 +859,6 @@ def calc_traveling_salesman(data_manager, debug=False):
data_manager.debug_entities.append(debug_entity)
-def toggle_roomba_ai(data_manager):
- """
- Toggles roomba AI on or off. Program start default is off.
- :param data_manager: Data manager data structure. Consolidates useful program data to one location.
- """
- logger.info('Toggling roomba ai.')
- if data_manager.ai_active:
- data_manager.ai_active = False
- else:
- data_manager.ai_active = True
-
-
-def set_roomba_vision_range_0(data_manager):
- """
- Adjusts roomba AI sight to see 0 tiles out from current location.
- :param data_manager: Data manager data structure. Consolidates useful program data to one location.
- """
- logger.info('Setting roomba vision to "0 tiles" (bump sensor).')
- data_manager.roomba_vision = 0
-
-
-def set_roomba_vision_range_1(data_manager):
- """
- Adjusts roomba AI sight to see 1 tiles out from current location.
- :param data_manager: Data manager data structure. Consolidates useful program data to one location.
- """
- logger.info('Setting roomba vision to "1 tiles".')
- data_manager.roomba_vision = 1
-
-
-def set_roomba_vision_range_2(data_manager):
- """
- Adjusts roomba AI sight to see 2 tiles out from current location.
- :param data_manager: Data manager data structure. Consolidates useful program data to one location.
- """
- logger.info('Setting roomba vision to "2 tiles".')
- data_manager.roomba_vision = 2
-
-
-def set_roomba_vision_range_full(data_manager):
- """
- Adjusts roomba AI sight to see all tiles on map.
- :param data_manager: Data manager data structure. Consolidates useful program data to one location.
- """
- logger.info('Setting roomba vision to "full sight".')
- data_manager.roomba_vision = -1
-
-
def clear_debug_entities(data_manager):
"""
Removes all debug entities, so that the screen does not become cluttered with redundant/overlapping debug info.
--
GitLab