Merge branch 'labelconfig'

buttonarea.py

@@ -0,0 +1,119 @@
+import sys, os
+from PyQt4.QtGui import *
+from PyQt4.QtCore import *
+
+class ButtonArea(QWidget):
+    def __init__(self, parent=None):
+        QWidget.__init__(self, parent)
+
+        self.label_names = []
+        self.label_properties = {}
+
+        self.properties = {}
+        
+        self.last_checked_label = None
+        
+    def create_button(self, button_name):
+        button = QPushButton(button_name)
+        button.setFlat(True)
+        button.setCheckable(True)
+        button.clicked.connect(self.clickedButton)
+        return button
+
+    def update_label_buttons(self):
+        label_vlayout = self.create_choice_list("Labels")
+        self.label_buttongroup = QButtonGroup()
+        for label_name in self.label_names:
+            button = QPushButton(label_name)
+            button.setFlat(True)
+            button.setCheckable(True)
+            button.clicked.connect(self.clickedLabelButton)
+            label_vlayout.addWidget(button)
+            self.label_buttongroup.addButton(button)
+            if label_name == self.last_checked_label:
+                button.setChecked(True)
+
+        self.hlayout = QHBoxLayout()
+        self.hlayout.addLayout(label_vlayout)
+        self.setLayout(self.hlayout)
+
+    def update_buttons(self):
+        layout = self.hlayout.takeAt(1)
+
+        dynamic_hlayout = QHBoxLayout()
+        
+        if self.last_checked_label != None:
+            for key in self.label_properties[self.last_checked_label].keys():
+                if key in ["type", "class"]:
+                    continue
+                vlayout = self.create_choice_list(key)
+                buttongroup = QButtonGroup()
+                for value in self.properties[key]:
+                    button = self.create_button(value)
+                    vlayout.addWidget(button)
+                    buttongroup.addButton(button)
+                dynamic_hlayout.addLayout(vlayout)
+
+        self.hlayout.addLayout(dynamic_hlayout)
+        self.update()
+        self.updateGeometry()
+
+    def add_label(self, label_name, properties = {}):
+        self.label_names.append(label_name)
+        self.label_properties[label_name] = properties
+        for key, value in properties.iteritems():
+            if self.properties.has_key(key):
+                self.properties[key] |= set(value)
+            else:
+                self.properties[key] = set(value)
+
+    def get_checked_label_button(self):
+        return self.label_buttongroup.checkedButton()
+
+    def create_choice_list(self, list_name, elements = []):
+        vlayout = QVBoxLayout()
+        vlayout.setSpacing(0)
+        #vlayout.setMargin(0)
+        #vlayout.setContentsMargins(0, 0, 0, 0)
+        vlayout.addWidget(QLabel("<center><b>" + list_name + "</b></center>"))
+        if len(elements) > 0:
+            buttongroup = QButtonGroup()
+            for element in elements:
+                self.create_button(element)
+        return vlayout
+
+    def clickedButton(self):
+        button = self.get_checked_label_button()
+        label_name = str(button.text())
+        print label_name
+
+    def clickedLabelButton(self):
+        button = self.get_checked_label_button()
+        label_name = str(button.text())
+        if label_name != self.last_checked_label:
+            print label_name, self.last_checked_label
+            self.last_checked_label = label_name
+            self.update_buttons()
+
+    def load(self, config_filepath):
+        execfile(config_filepath)
+        #if self.get_checked_label_button() == None:
+        #    if len(self.label_buttongroup.buttons()) != 0:
+        #        self.label_buttongroup.buttons()[0].setChecked(True)
+        #        
+        #self.last_checked_label = str(self.get_checked_label_button().text())
+        self.update_label_buttons()
+        self.update_buttons()
+        
+def main():
+    app = QApplication(sys.argv)
+
+    ba = ButtonArea()
+    ba.load("example_config.py")
+    ba.show()
+
+    return app.exec_()
+
+if __name__ == '__main__':
+    sys.exit(main())
+

example_config.py

@@ -0,0 +1,22 @@
+obj_type_choice = ["Apple", "Chair", "Carpet"]
+obj_type_choice2 = ["Peach", "Table", "Window"]
+
+ID_choice = ["Martin", "Mika", "Alexander", "Lukas", "Tobias"]
+ID_choice2 = ["Boris", "Manel", "Rainer", "Hua", "Hazim"]
+
+self.add_label("Head", {"type": "rect",
+                        "class": "head",
+                        "id": ID_choice})
+self.add_label("Left Eye", {"type": "point",
+                            "class": "left_eye",
+                            "id": ID_choice,
+                            "obj_type": obj_type_choice})
+self.add_label("Right Eye", {"type": "point",
+                             "class": "right_eye",
+                             "id": ID_choice2,
+                             "obj_type": obj_type_choice2})
+self.add_label("Left Hand", {"type": "rect",
+                             "class": "left_hand",
+                             "obj_type": obj_type_choice})
+self.add_label("Right Hand", {"type": "rect",
+                              "class": "right_hand"})