Commit a753c53f authored by dmMaze's avatar dmMaze
Browse files

impl pen tool

parent 5e8bb4f7

ballontranslator/ui/canvas.py

+8 −2
Original line number Diff line number Diff line
@@ -356,7 +356,7 @@ class Canvas(QGraphicsScene): 


    def setDrawingLayer(self, img: Union[QPixmap, np.ndarray] = None):

        

        ditems = self.drawingLayer.childItems()

        ditems = self.get_drawings(visible=False)

        for item in ditems:

            self.removeItem(item)
@@ -433,3 +433,9 @@ class Canvas(QGraphicsScene): 
        if item == self.stroke_path_item:

            self.stroke_path_item = None

        return super().removeItem(item)



    def get_drawings(self, visible=False) -> List[QGraphicsItem]:

        ditems = self.drawingLayer.childItems()

        if visible:

            ditems = [item for item in ditems if item.isVisible()]

        return ditems
 
 No newline at end of file

ballontranslator/ui/mainwindow.py

+1 −1
Original line number Diff line number Diff line
@@ -410,7 +410,7 @@ class MainWindow(QMainWindow): 
            os.makedirs(self.imgtrans_proj.result_dir())



        # save drawings to inpainted

        ditems = self.canvas.drawingLayer.childItems()

        ditems = self.canvas.get_drawings(visible=True)

        if len(ditems) > 0:

            inpainted = self.canvas.inpaintLayer.pixmap().toImage().convertToFormat(QImage.Format.Format_ARGB32)

            painter = QPainter(inpainted)