adf

2025-03-10 04:07:37 -06:00
parent 0c16b74b49
commit e56b5a860f
1 changed files with 0 additions and 0 deletions
--- a/Purpose/identification_overlay.py
+++ b/Purpose/identification_overlay.py
@ -1,103 +0,0 @@
-#!/usr/bin/env python3
-"""
-Identification Overlay Node:
- Users can configure threads/slices for parallel processing.
-"""
-
-import re
-from OdenGraphQt import BaseNode
-from PyQt5.QtCore import QTimer
-from PyQt5.QtGui import QColor
-from Modules import data_collector
-
-
-class IdentificationOverlayNode(BaseNode):
-    __identifier__ = "bunny-lab.io.identification_overlay_node"
-    NODE_NAME = "Identification Overlay"
-
-    def __init__(self):
-        super(IdentificationOverlayNode, self).__init__()
-
-        # User-configurable options
-        self.add_text_input("search_term", "Search Term", text="Aibatt")
-        self.add_text_input("offset_value", "Offset Value (X,Y)", text="0,0")  # X,Y Offset
-        self.add_text_input("margin", "Margin", text="5")  # Box Margin
-        self.add_text_input("polling_freq", "Polling Frequency (ms)", text="500")  # Polling Rate
-        self.add_combo_menu("ocr_engine", "Type", items=["CPU", "GPU"])
-        self.set_property("ocr_engine", "CPU")  # Default to CPU mode
-
-        # Custom overlay options
-        self.add_text_input("overlay_color", "Overlay Color (RGB)", text="0,0,255")  # Default blue
-        self.add_text_input("thickness", "Line Thickness", text="2")  # Default 2px
-        self.add_text_input("threads_slices", "Threads / Slices", text="8")  # Default 8 threads/slices
-
-        self.region_id = "identification_overlay"
-        data_collector.create_ocr_region(self.region_id, x=250, y=50, w=300, h=200, color=(0, 0, 255), thickness=2)
-
-        data_collector.start_collector()
-        self.set_name("Identification Overlay")
-
-        # Timer for updating overlays
-        self.timer = QTimer()
-        self.timer.timeout.connect(self.update_overlay)
-
-        # Set initial polling frequency
-        self.update_polling_frequency()
-
-    def update_polling_frequency(self):
-        polling_text = self.get_property("polling_freq")
-        try:
-            polling_interval = max(50, int(polling_text))  
-        except ValueError:
-            polling_interval = 500  
-
-        self.timer.start(polling_interval)
-
-    def update_overlay(self):
-        search_term = self.get_property("search_term")
-        offset_text = self.get_property("offset_value")
-        margin_text = self.get_property("margin")
-        ocr_engine = self.get_property("ocr_engine")
-        threads_slices_text = self.get_property("threads_slices")
-
-        self.update_polling_frequency()
-
-        try:
-            offset_x, offset_y = map(int, offset_text.split(","))
-        except ValueError:
-            offset_x, offset_y = 0, 0  
-
-        try:
-            margin = int(margin_text)
-        except ValueError:
-            margin = 5  
-
-        color_text = self.get_property("overlay_color")
-        try:
-            color = tuple(map(int, color_text.split(",")))  
-        except ValueError:
-            color = (0, 0, 255)  
-
-        thickness_text = self.get_property("thickness")
-        try:
-            thickness = max(1, int(thickness_text))  
-        except ValueError:
-            thickness = 2  
-
-        try:
-            num_slices = max(1, int(threads_slices_text))  # Ensure at least 1 slice
-        except ValueError:
-            num_slices = 1  
-
-        if not search_term:
-            return
-
-        detected_positions = data_collector.find_word_positions(
-            self.region_id, search_term, offset_x, offset_y, margin, ocr_engine, num_slices
-        )
-
-        # Ensure slice count is updated visually in the region widget
-        data_collector.update_region_slices(self.region_id, num_slices)
-
-        data_collector.draw_identification_boxes(self.region_id, detected_positions, color=color, thickness=thickness)
-