PyImageLabeling 1.0.0__py3-none-any.whl

PyImageLabeling/__init__.py

@@ -0,0 +1,22 @@
+from PyImageLabeling.model.Utils import Utils
+from PyImageLabeling.view.View import View
+from PyImageLabeling.controller.Controller import Controller
+from PyImageLabeling.model.Model import Model
+
+from PyQt6.QtWidgets import QApplication
+import sys 
+
+__version__ = "1.0.0"
+
+def __main__():
+    config = Utils.get_config()
+    app = QApplication(sys.argv)
+    
+    controller = Controller(config)
+    view = View(controller, config)
+    model = Model(view, controller, config)
+    controller.set_model(model)
+    
+    sys.exit(app.exec())
+
+__main__()

PyImageLabeling/config.json

@@ -0,0 +1,289 @@
+{
+    "window_size":{
+        "margin": 10,
+        "icon": [20, 20],
+        "icon_save_marker": [12, 12],
+        
+        "graphics_view":{
+            "width":400
+        },
+        "labeling_bar":{
+            "width": 250
+        },
+        "image_bar":{
+            "height": 220,
+            "width": 65
+        },
+        "label_bar":{
+            "height": 75
+        },
+        "file_bar":{
+            "width": 250
+        }
+    },
+    "apply_cancel_bar":[
+        {
+            "name": "apply",
+            "name_view": "Apply",
+            "icon": "apply",
+            "checkable": false,
+            "tooltip": "Apply an operation."
+        },
+        {
+            "name": "cancel",
+            "name_view": "Cancel",
+            "icon": "cancel",
+            "checkable": false,
+            "tooltip": "Cancel an operation."
+        }
+    ],
+    "file_bar":[
+        {
+            "name": "load",
+            "name_view": "Load images and labels",
+            "icon": "open_image",
+            "checkable": false,
+            "tooltip": "Load images and labels."
+        },
+        {
+            "name": "save",
+            "name_view": "Save labels",
+            "icon": "save",
+            "checkable": false,
+            "tooltip": "Save all labels."
+        },
+        {
+            "name": "next_image",
+            "name_view": "Next Image",
+            "icon": "down",
+            "checkable": false,
+            "tooltip": "Next image."
+        },
+        {
+            "name": "previous_image",
+            "name_view": "Previous Image",
+            "icon": "up",
+            "checkable": false,
+            "tooltip": "Previous image."
+        }
+    ],
+
+    "image_bar":
+            [
+                {
+                    "name": "zoom_plus",
+                    "name_view": "Zoom Plus",
+                    "icon": "zoom_plus",
+                    "checkable": true,
+                    "tooltip": "Zoom Plus."
+                },
+                {
+                    "name": "zoom_minus",
+                    "name_view": "Zoom Minus",
+                    "icon": "zoom_minus",
+                    "checkable": true,
+                    "tooltip": "zoom_minus."
+                },
+                {
+                    "name": "move_image",
+                    "icon": "move",
+                    "checkable": true,
+                    "name_view": "Move",
+                    "tooltip": "Click to activate Move mode. This allows you to move the image around in the viewer by dragging it."
+                },
+                {
+                    "name": "reset_move_zoom_image",
+                    "icon": "reset",
+                    "checkable": false,
+                    "name_view": "Move/Zoom Reset",
+                    "tooltip": "Click to reset the image's position and zoom level to the default view."
+                }
+            ],
+    "label_bar":
+            {"permanent":[
+                {
+                    "name": "new_label",
+                    "name_view": "New Label",
+                    "icon": "plus",
+                    "checkable": false,
+                    "tooltip": "Add a new label."
+                },
+                {
+                    "name": "opacity",
+                    "name_view": "Opacity",
+                    "checkable": false,
+                    "icon": "opacity",
+                    "tooltip": "Click to toggle opacity mode, allowing you to adjust the transparency of layers."
+                }
+            ],
+            "layer":[
+                {
+                    "name": "color",
+                    "name_view": "Color",
+                    "icon": "",
+                    "checkable": false,
+                    "tooltip": "Color of the layer/label."
+                },
+                {
+                    "name": "visibility",
+                    "name_view": "Visibility",
+                    "icon": "visibility",
+                    "checkable": true,
+                    "tooltip": "Visibility of the label."
+                },
+                {
+                    "name": "label_setting",
+                    "name_view": "Label Setting",
+                    "icon": "setting",
+                    "checkable": false,
+                    "tooltip": "Parameters of the layer/label."
+                },
+                {
+                    "name": "remove_label",
+                    "name_view": "Remove Label",
+                    "icon": "close",
+                    "checkable": false,
+                    "tooltip": "Remove the label"
+                }
+                
+            ]},
+    "labeling_bar": 
+            {"pixel": {
+                "name_view": "Pixel-by-pixel",
+                "buttons": [
+                    {
+                        "name": "contour_filling",
+                        "icon": "filling",
+                        "checkable": true,
+                        "name_view": "Contour Filling",
+                        "tooltip": "Click to activate Move mode. This allows you to move the image around in the viewer by dragging it."
+                    },
+                    {
+                        "name": "paintbrush",
+                        "name_view": "Paintbrush",
+                        "checkable": true,
+                        "icon": "paint",
+                        "tooltip": "Click to reset the image's position and zoom level to the default view."
+                    },
+                    {
+                        "name": "magic_pen",
+                        "name_view": "Magic Pen",
+                        "checkable": true,
+                        "icon": "magic",
+                        "tooltip": "Click to activate Move mode. This allows you to move the image around in the viewer by dragging it."
+                    }
+                ]
+            }
+            ,
+            "geometric":{
+                "name_view": "Geometric shapes",
+                "buttons": [
+                    {
+                        "name": "ellipse",
+                        "name_view": "Ellipse",
+                        "checkable": true,
+                        "icon": "ellipse",
+                        "tooltip": "."
+                    },
+                    {
+                        "name": "rectangle",
+                        "name_view": "Rectangle",
+                        "checkable": true,
+                        "icon": "rectangle",
+                        "tooltip": "Click to reset the image's position and zoom level to the default view."
+                    },
+                    {
+                        "name": "polygon",
+                        "name_view": "Polygon",
+                        "checkable": true,
+                        "icon": "polygon",
+                        "tooltip": "Click to activate Move mode. This allows you to move the image around in the viewer by dragging it."
+                    }
+                ]}
+            ,
+            "edit":{
+                "name_view": "Edition",
+                "buttons": [
+                {
+                    "name": "undo",
+                    "name_view": "Undo",
+                    "checkable": false,
+                    "icon": "back",
+                    "tooltip": "Click to undo the last drawing action or modification."
+                },
+                {
+                    "name": "eraser",
+                    "name_view": "Eraser",
+                    "checkable": true,
+                    "icon": "eraser",
+                    "tooltip": "Click to activate the eraser tool, allowing you to erase parts of the image or layer."
+                },
+                {
+                    "name": "clear_all",
+                    "name_view": "Clear All",
+                    "checkable": false,
+                    "icon": "cleaner",
+                    "tooltip": "Click to clear all layers and reset the image to its original state."
+                }
+            ]}
+            }
+        ,
+    "labeling_bar_setting": 
+    [
+            {"Edit Setting":[
+                {
+                    "name": "eraser_setting",
+                    "name_view": "Eraser Setting",
+                    "icon": "setting",
+                    "checkable": false,
+                    "tooltip": "Parameters of eraser."
+                }
+            ]},
+            {"Pixel Labeling Setting":[
+                {
+                    "name": "contour_filling_setting",
+                    "name_view": "Contour Filling Setting",
+                    "icon": "setting",
+                    "checkable": false,
+                    "tooltip": "Parameters of contour filling."
+                },
+                {
+                    "name": "paintbrush_setting",
+                    "name_view": "PaintBrush Setting",
+                    "icon": "setting",
+                    "checkable": false,
+                    "tooltip": "Parameters of paintbrush."
+                },
+                {
+                    "name": "magic_pen_setting",
+                    "name_view": "Magic Pen Setting",
+                    "icon": "setting",
+                    "checkable": false,
+                    "tooltip": "Parameters of magic pen."
+                }
+            ]},
+            {"Geometric Labeling Setting":[
+                {
+                    "name": "ellipse_setting",
+                    "name_view": "Ellipse Setting",
+                    "icon": "setting",
+                    "checkable": false,
+                    "tooltip": "Parameters of ellipse."
+                },
+                {
+                    "name": "rectangle_setting",
+                    "name_view": "Rectangle Setting",
+                    "icon": "setting",
+                    "checkable": false,
+                    "tooltip": "Parameters of rectangle."
+                },
+                {
+                    "name": "polygon_setting",
+                    "name_view": "Polygon Setting",
+                    "icon": "setting",
+                    "checkable": false,
+                    "tooltip": "Parameters of polygon."
+                }
+            ]}
+    ]
+}

PyImageLabeling/controller/Controller.py

@@ -0,0 +1,25 @@
+
+from PyImageLabeling.controller.FileEvents import FileEvents
+from PyImageLabeling.controller.LabelingEvents import LabelingEvents
+from PyImageLabeling.controller.ImageEvents import ImageEvents
+from PyImageLabeling.controller.LabelEvents import LabelEvents
+
+from PyQt6.QtWidgets import QMessageBox
+
+class Controller(FileEvents, LabelingEvents, ImageEvents, LabelEvents):
+    def __init__(self, config):
+        super().__init__()
+        
+        self.config = config
+        self.view = None
+        self.model = None
+
+    def set_view(self, view):
+        super().set_view(view)
+        self.view = view
+
+    def set_model(self, model):
+        super().set_model(model)
+        self.model = model
+
+    

PyImageLabeling/controller/Events.py

@@ -0,0 +1,147 @@
+from PyQt6.QtWidgets import QMessageBox, QGraphicsView, QApplication, QGraphicsItem
+from PyQt6.QtCore import QObject, QEvent, Qt, QRectF, QRect
+from PyQt6.QtGui import QPixmap, QMouseEvent, QKeyEvent, QBrush, QColor
+from PyQt6.QtWidgets import QLabel
+import os
+
+from PyImageLabeling.model.Utils import Utils
+
+
+
+
+class eventEater(QObject):
+    def __init__(self, controler, view, model):
+        super().__init__()
+        self.controler = controler
+        self.view = view
+        self.model = model
+
+    def set_model(self, model):
+        self.model = model
+
+    def eventFilter(self, event):
+        #self.view.zoomable_graphics_view.setDragMode(QGraphicsView.DragMode.NoDrag)
+        #print("event.type() QObject", event.type())
+        #print("event.type() QObject", obj)
+        
+        if event.type() == QEvent.Type.GraphicsSceneMousePress and event.button() == Qt.MouseButton.LeftButton:
+            if self.model.checked_button == "zoom_plus":
+                self.model.start_zoom_plus()
+            elif self.model.checked_button == "zoom_minus":
+                self.model.start_zoom_minus()
+            elif self.model.checked_button == "move_image":
+                self.model.start_move_tool(event)
+            elif self.model.checked_button == "paint_brush":
+                self.model.start_paint_brush(event.scenePos())
+            elif self.model.checked_button == "magic_pen":
+                self.model.start_magic_pen(event.scenePos())
+            elif self.model.checked_button == "contour_filling":
+                # Fill the contour clicked
+                self.model.fill_contour(event.scenePos())
+            elif self.model.checked_button == "eraser":
+                self.model.start_eraser(event.scenePos())
+            elif self.model.checked_button == "rectangle":
+                self.model.start_rectangle_tool(event.scenePos())
+            elif self.model.checked_button == "polygon":
+                self.model.start_polygon_tool(event.scenePos())
+            elif self.model.checked_button == "ellipse":
+                self.model.start_ellipse_tool(event.scenePos())
+
+        elif event.type() == QEvent.Type.GraphicsSceneMouseMove and event.buttons() == Qt.MouseButton.LeftButton: 
+            if self.model.checked_button == "paint_brush":
+                self.model.move_paint_brush(event.scenePos())
+            elif self.model.checked_button == "move_image":
+                self.model.move_move_tool(event)
+            elif self.model.checked_button == "eraser":
+                self.model.move_eraser(event.scenePos())
+            elif self.model.checked_button == "rectangle":
+                self.model.move_rectangle_tool(event.scenePos())
+            elif self.model.checked_button == "polygon":
+                self.model.move_polygon_tool(event.scenePos())
+            elif self.model.checked_button == "ellipse":
+                self.model.move_ellipse_tool(event.scenePos())
+                
+
+        elif event.type() == QEvent.Type.GraphicsSceneMouseRelease and event.button() == Qt.MouseButton.LeftButton: 
+            if self.model.checked_button == "paint_brush":
+                self.model.end_paint_brush()
+            elif self.model.checked_button == "move_image":
+                self.model.end_move_tool()
+            elif self.model.checked_button == "contour_filling":
+                self.model.end_contour_filling()
+            elif self.model.checked_button == "magic_pen":
+                
+                self.view.zoomable_graphics_view.change_cursor("magic")
+            elif self.model.checked_button == "eraser":
+                self.model.end_eraser()
+            elif self.model.checked_button == "rectangle":
+                self.model.end_rectangle_tool()
+            elif self.model.checked_button == "ellipse":
+                self.model.end_ellipse_tool()
+
+        elif event.type() == QEvent.Type.GraphicsSceneMousePress and event.button() == Qt.MouseButton.RightButton:
+        #elif event.type() == QEvent.Type.KeyPress:
+            #if event.key() == Qt.Key.Key_Delete:
+            if self.model.checked_button == "rectangle":
+                self.model.clear_rectangle()
+            if self.model.checked_button == "polygon":
+                self.model.clear_polygon()
+            if self.model.checked_button == "ellipse":
+                self.model.clear_ellipse()
+                
+        #elif event.type() == QEvent.Type.GraphicsSceneMousePress and event.button() == Qt.MouseButton.RightButton:
+        #    if self.model.checked_button == "contour_filling":
+        #        if self.view.layer_activation == True :
+        #            self.model.fill_contour(event.scenePos())
+        
+        #MouseButton.MiddleButton: move tool
+        elif event.type() == QEvent.Type.GraphicsSceneMousePress and event.button() == Qt.MouseButton.MiddleButton:
+            self.model.start_move_tool(event)
+        elif event.type() == QEvent.Type.GraphicsSceneMouseMove:
+            self.model.move_move_tool(event)
+        elif event.type() == QEvent.Type.GraphicsSceneMouseRelease and event.button() == Qt.MouseButton.MiddleButton:
+            self.model.end_move_tool()
+
+        #MouseButton.MiddleButton: zoom tool
+        elif event.type() == QEvent.Type.Wheel and self.model.move_tool_activation == False:
+            #if QApplication.mouseButtons() & Qt.MiddleButton:
+            #    return False  
+            if hasattr(self.view, 'zoomable_graphics_view'):
+                self.view.zoomable_graphics_view.wheelEvent(event)
+        return True
+
+        
+class Events:
+    def __init__(self):
+        self.view = None
+        self.model = None
+        self.event_eater = None
+
+    def set_view(self, view):
+        self.view = view
+        self.event_eater = eventEater(self, self.view, self.model)
+        #self.view.zoomable_graphics_view.scene.installEventFilter(self.event_eater)
+    
+    def set_model(self, model):
+        self.model = model
+        self.event_eater.set_model(model)
+
+    def all_events(self, event_name):
+        print("all_events")
+    
+    def desactivate_buttons_labeling_image_bar(self, event_name):
+        self.view.desactivate_buttons(event_name, [self.view.buttons_labeling_bar, self.view.buttons_image_bar])
+        
+    def desactivate_buttons_label_bar(self, event_name):
+        buttons_bar = {key: self.view.buttons_label_bar_temporary[key] for key in self.view.buttons_label_bar_temporary.keys() if key.startswith("activation_")}
+        self.view.desactivate_buttons(event_name, [buttons_bar])
+    
+    def error_message(self, title, text):
+        msg_box = QMessageBox(self.view)
+        for button in msg_box.buttons():
+            button.setObjectName("dialog")
+        msg_box.setObjectName("dialog")
+        msg_box.setWindowTitle("Error: "+str(title))
+        msg_box.setText(text)
+        msg_box.exec()
+        

PyImageLabeling/controller/FileEvents.py

@@ -0,0 +1,69 @@
+
+from PyImageLabeling.controller.Events import Events
+
+from PyQt6.QtWidgets import QFileDialog, QLabel
+from PyQt6.QtGui import QPixmap, QImage
+
+import os
+
+class FileEvents(Events):
+    def __init__(self):
+        super().__init__()
+        
+    def set_view(self, view):
+        super().set_view(view)
+
+    def set_model(self, model):
+        super().set_model(model)
+        
+    def load(self):
+        self.all_events(self.load.__name__)
+        self.model.load()
+        for button_names in self.view.buttons_image_bar:
+            self.view.buttons_image_bar[button_names].setEnabled(True)
+        for button_names in self.view.buttons_label_bar_permanent:
+            self.view.buttons_label_bar_permanent[button_names].setEnabled(True)
+        
+        self.move_image() # we active the move button :) 
+        print("load")
+
+
+    def save(self):
+        self.model.save()
+
+
+    def select_image(self, item):
+        print("item:", item)
+        
+        #filename = item.filename
+        self.model.select_image(item.file_path)
+        
+        if self.model.get_current_label_item() is not None:
+            self.model.update_labeling_overlays(self.model.get_current_label_item().get_label_id())
+
+        self.view.file_bar_list.setCurrentItem(item)
+        
+        # item_widget = self.view.file_bar_list.itemWidget(item)
+        # if item_widget:
+        #     file_label = item_widget.findChild(QLabel)
+        #     if file_label:
+        #         filename = file_label.text()
+        #         matching_path = None
+        #         for path in self.model.loaded_image_paths:
+        #             if os.path.basename(path) == filename:
+        #                 matching_path = path
+        #                 break
+        #         if matching_path:
+        #             self.model.load_image(matching_path)
+        #             self.view.file_bar_list.setCurrentItem(item)
+        #             print(f"Loaded image: {filename}")
+        #         else:
+        #             print(f"Error: Could not find path for {filename}")
+    
+    def next_image(self):
+        self.all_events(self.next_image.__name__)
+        self.model.next_image()
+
+    def previous_image(self):
+        self.all_events(self.previous_image.__name__)
+        self.model.previous_image()

PyImageLabeling/controller/ImageEvents.py

@@ -0,0 +1,32 @@
+from  controller.Events import Events
+
+
+class ImageEvents(Events):
+    def __init__(self):
+        super().__init__()
+
+    def zoom_plus(self):
+        self.desactivate_buttons_labeling_image_bar(self.zoom_plus.__name__)
+        self.all_events(self.zoom_plus.__name__)
+        self.view.zoomable_graphics_view.change_cursor("zoom_plus")
+        self.model.zoom_plus()
+        
+    def zoom_minus(self):
+        self.desactivate_buttons_labeling_image_bar(self.zoom_minus.__name__)
+        self.all_events(self.zoom_minus.__name__)
+        self.view.zoomable_graphics_view.change_cursor("zoom_minus")
+        self.model.zoom_minus()
+
+    def move_image(self):
+        self.desactivate_buttons_labeling_image_bar(self.move_image.__name__)
+        self.all_events(self.move_image.__name__)
+        self.view.zoomable_graphics_view.change_cursor("move")
+        self.model.move_image()
+        
+    def reset_move_zoom_image(self):
+        self.all_events(self.reset_move_zoom_image.__name__)
+        self.model.reset_move_zoom_image()
+
+
+
+