Adding Context menu buttons

Author: PeterWaIIace

app/components.py

@@ -97,6 +97,9 @@ def __init__(self, text, id = None, signal=None, parent=None,
         self.signal = signal
         self.clicked.connect(self.click)
 
+    def update_text(self, new_text):
+        self.setText(new_text)
+
     def click(self):
 
         if self.signal:

app/contextMenu.py

@@ -4,6 +4,7 @@
 from PySide2.QtCore import Qt, QTimer
 from  components import EyePilotButton
 
+import time
 import random
 import resources_rc  # Import the compiled resource file
 class ContextMenuBtn(QWidget):
@@ -37,14 +38,16 @@ def __init__(self,text,x,y):
 
         self.setGeometry(x, y, self.width, self.height)
 
-
         self.to_y = y
         self.to_x = x
 
     def setPosition(self,x,y):
         self.to_x = x - self.diameter/2
         self.to_y = y - self.diameter/2
 
+    def setText(self,text):
+        self.btn.setText(text)
+
     def setTransparency(self,transparency):
         self.transparency = transparency
         if self.transparency > 0:
@@ -56,14 +59,42 @@ def setTransparency(self,transparency):
 
         self.setColor(self.brush_color.red(),self.brush_color.green(),self.brush_color.blue())
 
+class ContextMenu(QWidget):
+
+    def __init__(self) -> None:
+        self.btn1 = ContextMenuBtn("Lorem ipsum morem solem",100,100)
+        self.btn2 = ContextMenuBtn("Lorem ipsum morem solem",150,300)
+        self.btn3 = ContextMenuBtn("Lorem ipsum morem solem",200,500)
+        self.btn4 = ContextMenuBtn("Lorem ipsum morem solem",300,700)
+
+    def show(self):
+        self.btn1.show()
+        self.btn2.show()
+        self.btn3.show()
+        self.btn4.show()
+
+    def hide(self):
+        self.btn1.hide()
+        self.btn2.hide()
+        self.btn3.hide()
+        self.btn4.hide()
+
+    def setText(self,t1,t2,t3,t4):
+        self.btn1.setText(t1)
+        self.btn2.setText(t2)
+        self.btn3.setText(t3)
+        self.btn4.setText(t4)
+
+
 if __name__ == "__main__":
     app = QApplication(sys.argv)
-    btn1 = ContextMenuBtn("Lorem ipsum morem solem",100,100)
-    btn1.show()
-    btn2 = ContextMenuBtn("Lorem ipsum morem solem",150,300)
-    btn2.show()
-    btn3 = ContextMenuBtn("Lorem ipsum morem solem",200,500)
-    btn3.show()
-    btn4 = ContextMenuBtn("Lorem ipsum morem solem",300,700)
-    btn4.show()
+    menu = ContextMenu()
+
+    menu.show()
+    menu.setText("Text1","Text2","Text3","Text4")
+    # for n in range(10):
+    # menu.hide()
+    menu.show()
+    menu.setText("Text1","Text2","Text3","LOL4")
+
     sys.exit(app.exec_())

app/contextTracker.py

@@ -7,6 +7,7 @@
 import threading
 import numpy as np
 import time
+import re
 
 class VisContext:
 
@@ -21,7 +22,9 @@ def __init__(self,before = lambda : None ,after = lambda : None):
 
     def getDescription(self):
         img, description = self.cursorTracker.getRichContext()
-        return description
+        pattern = r'(?<=\d\. )(.*?)(?=\n)'
+        tokens = re.findall(pattern, description)
+        return tokens
 
     def before_scan(self):
         self.before()
@@ -142,7 +145,7 @@ def rescan(self,x,y,w,h):
         self.DSB.loadData(rectangles)
 
     def getRichContext(self):
-        description = ""
+        description = "output: 0. Open .git\n1. Open Documents\n2. Open Downloads\n3. Open dist\n4. Open build\n5. Open Downloads" # test input
         # description = self.advisorModel.respond(self.rich_context.getPath(self))
         return (self.rich_context, description)
 

app/main.py

@@ -15,6 +15,7 @@
 from contextTracker import VisContext
 
 from dot import CircleWidget
+from contextMenu import ContextMenu
 
 from tracker import Tracker
 
@@ -128,6 +129,7 @@ def __init__(self):
         self.calibrationON = False
 
         self.vizContext = VisContext()
+        self.contextMenu = ContextMenu()
 
         self.model = ModelSaver()
 
@@ -171,11 +173,15 @@ def main_loop(self):
                     self.fix_start = time.time()
                     self.vizContext.start()
                     self.vizContext.setPosition(point[0], point[1])
-                    self.vizContext.getDescription()
+                    description = self.vizContext.getDescription()
+                    print(description)
+                    self.contextMenu.show()
+                    self.contextMenu.setText(description[0],description[1],description[2],"Exit")
                 elif not fix:
                     self.vizContext.close()
 
                 if blink and fix:
+                    self.contextMenu.hide()
                     x,y = self.vizContext.setPosition(point[0], point[1])
                     self.press(x,y)