def openCSVFile(self, activated=False, filePath=None): if filePath is None: filePath, _ = QFileDialog.getOpenFileName(None, 'Open CSV File', userDir, 'CSV files (*.csv)') if len(filePath) is not 0: self.df = pd.read_csv(filePath, index_col=0) if self.handInputSystem is not None: self.inputScene.removeItem(self.handInputSystem) self.handInputSystem = HandInputSystem() self.handInputSystem.setRect(self.inputScene.sceneRect()) self.inputScene.addItem(self.handInputSystem) self.handInputSystem.setDataFrame(self.df) self.handInputSystem.setPoints() self.dataFrameWidget.clear() self.dataFrameWidget.dataFrameNo = self.handInputSystem.dataFrameNo self.dataFrameWidget.editingNo = 0 for item in range(self.handInputSystem.dataFrameNo + 1): color = self.handInputSystem.itemList[item].getColor() print(item, color) self.dataFrameWidget.colorList.append(color) self.dataFrameWidget.setUniqueIDLabel() self.evaluate()
def __init__(self): super(Ui_MainWindow, self).__init__() self.setupUi(self) self.videoPlaybackInit() self.imgInit() self.menuInit() self.df = None self.trackingPathGroup = None self.drawingFlag = False self.handInputSystem = None self.handInputSystem = HandInputSystem() self.handInputSystem.setRect(self.inputScene.sceneRect()) self.inputScene.addItem(self.handInputSystem) self.handInputSystem.addNewDataFrame() self.currentFrameNo = 0 self.colors = [] self.circleCheckBox.stateChanged.connect( self.polyLineCheckBoxStateChanged) self.lineCheckBox.stateChanged.connect( self.polyLineCheckBoxStateChanged) self.overlayCheckBox.stateChanged.connect( self.overlayCheckBoxStateChanged) self.radiusSpinBox.valueChanged.connect(self.radiusSpinBoxValueChanged) self.frameNoSpinBox.valueChanged.connect( self.frameNoSpinBoxValueChanged) self.groupBox_2.hide() self.inputGraphicsView.viewport().setCursor(QtCore.Qt.ArrowCursor) # self.optionViewButton.pressed.connect(self.optionViewButtonPressed) self.zoomedGraphicsView.hide() self.dataFrameWidget.dataFrameChanged.connect(self.dataFrameChanged) self.dataFrameWidget.hide() self.handInputSystem.setColor(self.dataFrameWidget.getColor())