def __init__(self):
super().__init__()
self.setWindowTitle('Bartender Data File Generator')
self.setWindowIcon(QIcon('Images/espresso-icon.png'))
self.setMinimumSize(350, 350)
self.button_is_checked = True
button = QPushButton('Press me!')
button.setCheckable(True) # set button as toggleable
button.clicked.connect(self.button_was_toggled
) # connect button to signal to perform action
button.setChecked(
self.button_is_checked) # toggles the button to False
self.setCentralWidget(button)
def initUI(self):
self.col = QColor(0, 0, 0)
redb = QPushButton('Red', self)
redb.setCheckable(True)
redb.move(10, 10)
redb.clicked[bool].connect(self.setColor)
greenb = QPushButton('Green', self)
greenb.setCheckable(True)
greenb.move(10, 60)
greenb.clicked[bool].connect(self.setColor)
blueb = QPushButton('Blue', self)
blueb.setCheckable(True)
blueb.move(10, 110)
blueb.clicked[bool].connect(self.setColor)
self.square = QFrame(self)
self.square.setGeometry(150, 20, 100, 100)
self.square.setStyleSheet("QWidget { background-color: %s }" %
self.col.name())
self.setGeometry(300, 300, 300, 250)
self.setWindowTitle('Toggle button')
self.show()
class MainWindow(QMainWindow):
def __init__(self, parent = None):
super(MainWindow, self).__init__()
D = self.screen().availableGeometry()
self.move(0,0)#center.x() + .25*D.width() , center.y() - .5*D.height() )
self.resize( int(.95*D.width()), int(6*D.height()) )
#qr = self.frameGeometry()
#cp = self.screen().availableGeometry().center()
#qr.moveCenter(cp)
#self.move(qr.topLeft())
self.setWindowState(self.windowState() & ~QtCore.Qt.WindowState.WindowMinimized
| QtCore.Qt.WindowState.WindowActive)
self.activateWindow()
self.subWin = Window()
self.iw = imwin()
self.Manual = Manual()
self.setCentralWidget(self.iw)
#Stacked dock widgets
docked1 = QDockWidget("", self)
docked2 = QDockWidget("", self)
self.addDockWidget(QtCore.Qt.DockWidgetArea.LeftDockWidgetArea, docked1)
self.addDockWidget(QtCore.Qt.DockWidgetArea.LeftDockWidgetArea, docked2)
docked1.setWidget(self.subWin)
docked2.setWidget(self.Manual)
docked1.setFeatures(QDockWidget.DockWidgetFeature.DockWidgetFloatable)
self.setCorner(QtCore.Qt.Corner.TopLeftCorner, QtCore.Qt.DockWidgetArea.LeftDockWidgetArea);
self.setCorner(QtCore.Qt.Corner.TopRightCorner, QtCore.Qt.DockWidgetArea.RightDockWidgetArea)
self.setCorner(QtCore.Qt.Corner.BottomLeftCorner, QtCore.Qt.DockWidgetArea.LeftDockWidgetArea);
self.setCorner(QtCore.Qt.Corner.BottomRightCorner, QtCore.Qt.DockWidgetArea.RightDockWidgetArea)
self.resizeDocks( (docked1,docked2), (400,400), QtCore.Qt.Orientation.Horizontal )
self.exportButton = QPushButton("Export Measurements", self)
self.exportButton.clicked.connect(self.export_measurements)
self.exportButton.setEnabled(False)
self.importImage = QPushButton("New Image", self)
self.importImage.clicked.connect(self.file_open)
self.lengthButton = QPushButton("Measure Length", self)
self.lengthButton.clicked.connect(self.measure_length)
self.lengthButton.setEnabled(False)
self.lengthButton.setCheckable(True)
self.lengthNames = []
self.widthsButton = QPushButton("Measure Widths", self)
self.widthsButton.clicked.connect(self.iw.measure_widths)
self.widthsButton.setEnabled(False)
self.widthsButton.setCheckable(True)
self.areaButton = QPushButton("Measure Area", self)
self.areaButton.clicked.connect(self.measure_area)
self.areaButton.setEnabled(False)
self.areaButton.setCheckable(True)
self.areaNames = []
self.angleButton = QPushButton("Measure Angle", self)
self.angleButton.clicked.connect(self.measure_angle)
self.angleButton.setEnabled(False)
self.angleButton.setCheckable(True)
self.angleNames = []
shortcut_polyClose = QShortcut(QtGui.QKeySequence(QtCore.Qt.Key.Key_Tab), self)
shortcut_polyClose.activated.connect(self.iw.polyClose)
self.undoButton = QPushButton("Undo", self)
self.undoButton.clicked.connect(self.undo)
self.undoButton.setEnabled(False)
shortcut_undo = QShortcut(QtGui.QKeySequence('Ctrl+Z'), self)
shortcut_undo.activated.connect(self.undo)
self.bezier = QRadioButton("Bezier fit", self)
self.bezier.setEnabled(True)
self.bezier.setChecked(True)
#self.bezier.toggled.connect(self.onClicked)
self.piecewise = QRadioButton("Piecewise", self)
self.statusbar = self.statusBar()
self.statusbar.showMessage('Select new image to begin')
self.tb = QToolBar('Toolbar')
#self.addToolBar(QtCore.Qt.RightToolBarArea,self.tb)
spacer = QWidget(self)
spacer.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Expanding)
self.tb.addWidget(spacer)
self.addToolBar(self.tb)
self.tb.addWidget(self.importImage)
self.tb.addWidget(self.exportButton)
self.tb.addWidget(self.lengthButton)
self.tb.addWidget(self.widthsButton)
self.tb.addWidget(self.areaButton)
self.tb.addWidget(self.angleButton)
self.tb.addWidget(self.undoButton)
self.tb.addWidget(self.bezier)
self.tb.addWidget(self.piecewise)
#self.tb.setOrientation(QtCore.Qt.Vertical)
def file_open(self):
self.iw.scene.clear()
self.image_name = QFileDialog.getOpenFileName(self, 'Open File')
self.iw.pixmap = QtGui.QPixmap(self.image_name[0])
self.iw.pixmap_fit = self.iw.pixmap.scaled(
self.iw.pixmap.width(),
self.iw.pixmap.height(),
QtCore.Qt.AspectRatioMode.KeepAspectRatio,
transformMode=QtCore.Qt.TransformationMode.SmoothTransformation)
self.iw.scene.addPixmap(self.iw.pixmap_fit) #add image
self.iw.setScene(self.iw.scene)
#Adjust window size automatically?
self.iw.fitInView(self.iw.scene.sceneRect(), QtCore.Qt.AspectRatioMode.KeepAspectRatio)
self.iw.scene.update()
self.statusbar.showMessage('Select a measurement to make from the toolbar')
self.lengthButton.setEnabled(True)
self.areaButton.setEnabled(True)
self.angleButton.setEnabled(True)
self.exportButton.setEnabled(True)
self.undoButton.setEnabled(True)
self.bezier.setEnabled(True)
self.bezier.setChecked(True)
self.widthsButton.setEnabled(False)
self.angleNames = []
self.areaNames = []
self.lengthNames = []
#self.iw.measurements = [[]]
self.iw.widths = []
self.iw.lengths = [[]]
self.iw.L = posData(
np.empty(shape=(0, 0)), np.empty(shape=(0, 0))) #lengths
self.iw.A = posData(
np.empty(shape=(0, 0)), np.empty(shape=(0, 0))) #area
self.iw.W = posData(
np.empty(shape=(0, 0)), np.empty(shape=(0, 0))) #widths
self.iw.T = angleData(np.empty(shape=(0, 0))) #angles
self.iw.angleValues = np.empty((0,0))
self.iw.areaValues = np.empty((0,0))
self.iw._lastpos = None
self.iw._thispos = None
self.iw.measuring_length = False
self.iw.measuring_area = False
self.iw.measuring_widths = False
self.iw.measuring_angle = False
self.iw._zoom = 0
self.iw.factor = 1.0
self.iw.d = {} #dictionary for line items
self.iw.k = 0 #initialize counter so lines turn yellow
self.iw.m = None
self.iw.scene.realline = None
self.iw.scene.testline = None
self.iw.scene.ellipseItem = None
self.iw.scene.area_ellipseItem = None
self.iw.scene.polyItem = None
self.iw.image_name = None
def measure_length(self):
self.lel = QLineEdit(self)
self.lel.move(130, 22)
text, ok = QInputDialog.getText(self, 'Input Dialog', 'Length name')
if ok:
self.lel.setText(str(text))
self.lengthNames.append(self.lel.text())
QApplication.setOverrideCursor(QtCore.Qt.CursorShape.CrossCursor) #change cursor
self.widthsButton.setChecked(False)
self.widthsButton.setEnabled(False)
self.iw.line_count = 0
self.iw.measuring_length = True
self.iw.L = posData(
np.empty(shape=(0, 0)),
np.empty(shape=(0, 0))) #preallocate
self.iw._lastpos = None
self.iw._thispos = None
self.statusbar.showMessage('Click initial point for length measurement')
else:
self.lengthButton.setChecked(False)
def measure_angle(self):
self.lea = QLineEdit(self)
self.lea.move(130, 22)
text, ok = QInputDialog.getText(self, 'Input Dialog', 'Angle name')
if ok:
self.lea.setText(str(text))
self.angleNames.append(self.lea.text())
QApplication.setOverrideCursor(QtCore.Qt.CrossCursor) #change cursor
self.bezier.setEnabled(False)
self.iw.measuring_angle = True
self.iw._lastpos = None
self.iw._thispos = None
self.statusbar.showMessage('Click initial point for angle measurement')
else:
self.angleButton.setChecked(False)
def measure_area(self):
self.lea = QLineEdit(self)
self.lea.move(130, 22)
text, ok = QInputDialog.getText(self, 'Input Dialog', 'Area name')
if ok:
self.lea.setText(str(text))
self.areaNames.append(self.lea.text())
QApplication.setOverrideCursor(QtCore.Qt.CrossCursor) #change cursor
self.bezier.setEnabled(False)
self.iw.line_count = 0
self.iw.measuring_area = True
self.iw._lastpos = None
self.iw._thispos = None
self.iw.A = posData(
np.empty(shape=(0, 0)),
np.empty(shape=(0, 0))) #preallocate
self.statusbar.showMessage('Click initial point for area measurement')
else:
self.areaButton.setChecked(False)
def undo(self):
if self.iw.measuring_length:
self.iw._thispos = self.iw._lastpos
self.iw.L.downdate() #remove data
self.iw.line_count += -1
self.iw.scene.removeItem(self.iw.scene.realline) #remove graphic
self.iw.scene.realline = False
if self.iw.measuring_area:
self.iw._thispos = self.iw._lastpos
self.iw.A.downdate() #remove data
self.iw.line_count += -1
self.iw.scene.removeItem(self.iw.scene.realline) #remove graphic
self.iw.scene.realline = False
if self.iw.measuring_widths:
self.iw.W.downdate() #remove data
self.iw.scene.removeItem(self.iw.scene.ellipseItem) #remove graphic
self.iw.scene.ellipseItem = False
self.iw.d[str(self.iw.k)].setPen(
QtGui.QPen(QtGui.QColor('black'))) #un-highlight next spine
self.iw.k += -1 #reduce count
if self.iw.measuring_angle:
self.iw.T.downdate() #remove data
self.iw._thispos = self.iw_lastpos
self.iw.scene.removeItem(self.iw.scene.realline) #remove graphic
self.iw.scene.realline = False
def export_measurements(self):
fac = max(self.iw.pixmap.width(), self.iw.pixmap.height()) / max(
self.iw.pixmap_fit.width(),
self.iw.pixmap_fit.height()) #scale pixel -> m by scaled image
name = QFileDialog.getSaveFileName(
self, 'Save File', self.image_name[0].split('.', 1)[0])[0]
self.pixeldim = float(self.subWin.pixeldim.text())
self.altitude = float(self.subWin.altitude.text())
self.focal = float(self.subWin.focal.text())
#okay in mm https://www.imaging-resource.com/PRODS/sony-a5100/sony-a5100DAT.HTM
if name:
#Convert pixels to meters
#measurements = [ f * fac * self.pixeldim * self.altitude / self.focal for f in self.iw.measurements]
#lengths = [ f * fac * self.pixeldim * self.altitude / self.focal for f in self.iw.lengths]
#print(self.iw.widths)
areas = self.iw.areaValues * (
fac * self.pixeldim * self.altitude / self.focal)**2
values_optical = np.array([
self.subWin.id.text(), self.image_name[0], self.focal,
self.altitude, self.pixeldim
])
names_optical = [
'Image ID', 'Image Path', 'Focal Length', 'Altitude',
'Pixel Dimension'
]
names_widths = ['Object'] + ['Length (m)'] + ['Widths (%)'] # + self.iw.widthNames[0]
#names_widths.append([self.iw.widthNames[0]])
#Write .csv file
print(f"Writing {name} to file")
with open(name + '.csv', 'w') as csvfile:
writer = csv.writer(csvfile)
for (f, g) in zip(names_optical, values_optical):
writer.writerow([f, g])
writer.writerow(['Notes', self.subWin.notes.toPlainText()])
writer.writerow([''])
writer.writerow(names_widths)
for k,m in enumerate(self.lengthNames):
#format and convert pixel length measurement
l = "{0:.2f}".format( self.iw.lengths[k] * fac * self.pixeldim * self.altitude / self.focal )
if any(self.iw.widths[k]): #check if width measurement exists for length
n = self.iw.widthNames[k]
writer.writerow( [''] + [''] + n )
#format and convert pixel width measurement
vals = [ "{0:.2f}".format(g * fac * self.pixeldim * self.altitude / self.focal) for g in self.iw.widths[k]]
line = [m] + [l] + list(vals)
else:
#vals = l #f.copy()
line = [m] + [l]
writer.writerow(line)
writer.writerow([''])
writer.writerow(['Object'] + ['Angle'])
for k, f in enumerate(self.angleNames): #write angles
line = [[f] + ["{0:.3f}".format(self.iw.angleValues[k])]] #need to convert NaNs to empty
writer.writerows(line)
writer.writerow([''])
writer.writerow(['Object'] + ['Area (m\u00B2)'])
for k, f in enumerate(self.areaNames): #write areas
line = [[f] + ["{0:.3f}".format(areas[k])]] #need to convert NaNs to empty
writer.writerows(line)
#Export image
self.iw.fitInView(self.iw.scene.sceneRect(), QtCore.Qt.AspectRatioMode.KeepAspectRatio)
pix = QtGui.QPixmap(self.iw.viewport().size())
self.iw.viewport().render(pix)
pix.save(name + '-measurements.png')
