Exemple #1
0
    def loadTrayImage(self, image_fname, csv_fname):
        """Load the tray scan image and activate the camera.

        Keyword Arguments:
        image_fname -- the string filepath of the tray scan image"""

        # Export current csv if there is one
        ret = self.exportToCSV()
        if not ret:
            return

        # Clear data
        self.reset()
        self.cvImpl.reset()
        self.barcodeEntry.setCurrentBugId("")

        # Load the image
        self.trayPath = image_fname
        self.logger.log("LOAD image scan file %s" % self.trayPath, 1)
        self.trayImage = cv2.imread(self.trayPath, cv2.IMREAD_COLOR)
        self.csvPath = csv_fname

        # Load csv file
        if os.path.isfile(csv_fname):
            with open(csv_fname) as csvfile:
                reader = csv.reader(csvfile)
                self.bugBoxList = Util.BugBoxList()

                if (reader.next()[1] == " Rectangle x1"):
                    for b in reader:
                        box = Util.BugBox(
                            b[0], None,
                            (int(b[1]), int(b[2]), int(b[3]), int(b[4])),
                            (int(b[5]), int(b[6])))
                        self.bugBoxList.newBox(box)

                    self.window.statusBar().showMessage(
                        "Also loaded CSV file: %s" %
                        str(os.path.split(csv_fname)[1]))
                    self.logger.log(
                        "LOAD found corresponding csv file %s" % self.csvPath,
                        1)
                else:
                    QtGui.QMessageBox.information(
                        self.window, "Error Reading CSV",
                        "It seems the file %s id badly formatted. Cannot load."
                        % str(os.path.split(csv_fname)[1]))

        # Start the camera loop
        self.startCameraFeed()

        self.sigShowHint.emit(C.HINT_TRAYAREA_1)
        self.bugBoxList.clearUndoRedoStacks()
Exemple #2
0
    def editMousePress(self, ev):
        """Called when the application is in edit mode and the mouse is clicked
        on the big label.

        Keyword Arguments:
        ev -- PySide.QtGui.QMouseEvent object"""

        (mx, my) = self.mousePos
        pad = AppData.BOX_RESIZE_EDGE_PADDING
        sp = int(AppData.DRAW_DELTA / self.lblBig.imageScaleRatio)

        if ev.button() == QtCore.Qt.MouseButton.LeftButton:
            if self.selectedEditBox is not None:
                # If a box is selected, check for clicking on an editable point
                # (e. corners for resizing, center for panning)
                p = self.mousePos
                (x1, y1, x2, y2) =\
                    self.bugBoxList[self.selectedEditBox].static

                if Util.pointInBox(
                        p, (x2 - 3 * sp, y1 + sp, x2 - sp, y1 + 3 * sp)):
                    self.bugBoxList.delete(self.selectedEditBox)
                    self.selectedEditBox = None
                    self.sigDeletedBox.emit(self.selectedEditBox)
                    self.barcodeEntry.setCurrentBugId("")
                elif Util.pointInBox(p,
                                     (x1 - pad, y1 - pad, x1 + pad, y1 + pad)):
                    self.editAction = AppData.DG_NW
                elif Util.pointInBox(p,
                                     (x2 - pad, y2 - pad, x2 + pad, y2 + pad)):
                    self.editAction = AppData.DG_SE
                elif Util.pointInBox(p,
                                     (x1 - pad, y2 - pad, x1 + pad, y2 + pad)):
                    self.editAction = AppData.DG_SW
                elif Util.pointInBox(p,
                                     (x2 - pad, y1 - pad, x2 + pad, y1 + pad)):
                    self.editAction = AppData.DG_NE
                elif Util.pointInBox(p,
                                     (x1 + pad, y1 - pad, x2 - pad, y1 + pad)):
                    self.editAction = AppData.DG_N
                elif Util.pointInBox(p,
                                     (x1 + pad, y2 - pad, x2 - pad, y2 + pad)):
                    self.editAction = AppData.DG_S
                elif Util.pointInBox(p,
                                     (x1 - pad, y1 + pad, x1 + pad, y2 - pad)):
                    self.editAction = AppData.DG_W
                elif Util.pointInBox(p,
                                     (x2 - pad, y1 + pad, x2 + pad, y2 - pad)):
                    self.editAction = AppData.DG_E
                elif Util.pointInBox(p,
                                     (x1 + pad, y1 + pad, x2 - pad, y2 - pad)):
                    self.editAction = AppData.PAN

                if self.editAction != AppData.NO_ACTION:
                    return

            # If a box is not selected, check for clicking on a box to
            # to select it
            clickedOn = False
            for i in range(len(self.bugBoxList)):
                (x1, y1, x2, y2) = self.bugBoxList[i].static
                if Util.pointInBox((mx, my),
                                   (x1 - pad, y1 - pad, x2 + pad, y2 + pad)):
                    self.sigSelectedBox.emit(i)
                    clickedOn = True
                    self.selectedEditBox = i
                    self.barcodeEntry.setCurrentBugId(self.bugBoxList[i].name)
                    self.sigShowHint.emit(C.HINT_EDITBOX)

            # Check for deselect of box
            if not clickedOn:
                self.selectedEditBox = None
                self.sigShowHint.emit(C.HINT_REMOVEBUG)
                self.barcodeEntry.setCurrentBugId("")
        else:
            box = Util.BugBox(
                "Box %s" % str(len(self.bugBoxList)), None,
                (mx - 3 * sp, my - 3 * sp, mx + 3 * sp, my + 3 * sp), (mx, my))
            self.bugBoxList.newBox(box)
            self.selectedEditBox = len(self.bugBoxList) - 1
            self.barcodeEntry.setCurrentBugId(
                self.bugBoxList[self.selectedEditBox].name)