diff --git a/src/misc.py b/src/misc.py index ff57c5553c64519b6ba4f2966404819584435b0b..e7831aca94a7b5534b32bc18db84a8f3f0a1d4e2 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.