#signToolChanged = QtCore.pyqtSignal()

def __init__(self, iface):

self.iface = iface

# Set up log file and collect any relevant messages

"""logFilePath = os.environ.get('QGIS_LOGFILE_PATH') # Set up within TOMs - and connects for all messages ...

if logFilePath:

QgsMessageLog.logMessage("LogFilePath: " + str(logFilePath), tag="TOMs panel")

logfile = 'qgis_' + datetime.date.today().strftime("%Y%m%d") + '.log'

self.filename = os.path.join(logFilePath, logfile)

QgsMessageLog.logMessage("Sorting out log file" + self.filename, tag="TOMs panel")

QgsMessageLog.instance().messageReceived.connect(self.write_log_message)

QgsMessageLog.logMessage("Finished init", tag="TOMs panel")

# self.toolbar = self.iface.addToolBar(u'Test5Class')

# self.toolbar.setObjectName(u'Test5Class')

def write_log_message(self, message, tag, level):

filename = os.path.join('C:\Users\Tim\Documents\MHTC', 'qgis.log')

with open(self.filename, 'a') as logfile:

logfile.write(

'{dateDetails}:: {message}\n'.format(dateDetails=time.strftime("%Y%m%d:%H%M%S"), message=message))"""

def setDefaultRestrictionDetails(self, currRestriction, currRestrictionLayer):

QgsMessageLog.logMessage("In setDefaultRestrictionDetails: ", tag="TOMs panel")

self.userName = (QgsExpressionContextUtils.globalScope().variable("user_account_name"))

QgsMessageLog.logMessage("In setDefaultRestrictionDetails: curr user name: " + self.userName, tag="TOMs panel")

try:

#currRestrictionLayer.changeAttributeValue(currRestriction.id(), currRestriction.fieldNameIndex("Signs_DateTime"), QDateTime.currentDateTime())

currRestriction.setAttribute("Signs_DateTime", QDateTime.currentDateTime())

currRestriction.setAttribute("Surveyor", self.userName)

except:

reply = QMessageBox.information(None, "Information", "Problems with default values ...",

QMessageBox.Ok)

#generateGeometryUtils.setRoadName(currRestriction)

def setupDemandDialog(self, demandDialog, currDemandLayer, currFeature):

# self.restrictionDialog = restrictionDialog

self.demandDialog = demandDialog

#self.demandAttributeForm = self.demandDialog.attributeForm()

#self.demandAttributeForm = self.demandDialog

self.currDemandLayer = currDemandLayer

self.currFeature = currFeature

# self.restrictionTransaction = restrictionTransaction

if self.demandDialog is None:

QgsMessageLog.logMessage(

"In setupDemandDialog. dialog not found",

tag="TOMs panel")

button_box = self.demandDialog.findChild(QDialogButtonBox, "button_box")

if button_box is None:

QgsMessageLog.logMessage(

"In setupDemandDialog. button box not found",

tag="TOMs panel")

reply = QMessageBox.information(None, "Information", "Please reset form. There are missing buttons",

QMessageBox.Ok)

return

#self.demandDialog.attributeForm().disconnectButtonBox()

try:

button_box.accepted.disconnect()

except:

reply = QMessageBox.information(None, "Information", "Problem disconnection Accept",

QMessageBox.Ok)

button_box.accepted.connect(functools.partial(self.onSaveDemandDetails, currFeature,

currDemandLayer, self.demandDialog))

try:

button_box.rejected.disconnect()

except:

reply = QMessageBox.information(None, "Information", "Problem disconnection Reject",

QMessageBox.Ok)

button_box.rejected.connect(functools.partial(self.onRejectDemandDetailsFromForm, currFeature,

currDemandLayer, self.demandDialog))

self.demandDialog.attributeForm().attributeChanged.connect(

functools.partial(self.onAttributeChangedClass2, self.currFeature, self.currDemandLayer))

self.photoDetails()

def onSaveDemandDetails(self, currFeature, currFeatureLayer, dialog):

QgsMessageLog.logMessage("In onSaveDemandDetails: ", tag="TOMs panel")

try:

self.camera1.endCamera()

except:

None

attrs1 = currFeature.attributes()

QgsMessageLog.logMessage("In onSaveDemandDetails: currRestriction: " + str(attrs1),

tag="TOMs panel")

QgsMessageLog.logMessage(

("In onSaveDemandDetails. geometry: " + str(currFeature.geometry().exportToWkt())),

tag="TOMs panel")

currFeatureID = currFeature.id()

QgsMessageLog.logMessage("In onSaveDemandDetails: currFeatureID: " + str(currFeatureID),

tag="TOMs panel")

if currFeatureID > 0: # Not sure what this value should if the feature has not been created ...

# TODO: Sort out this for UPDATE

self.setDefaultRestrictionDetails(currFeature, currFeatureLayer)

status = currFeatureLayer.updateFeature(currFeature)

QgsMessageLog.logMessage("In onSaveDemandDetails: updated Feature: ", tag="TOMs panel")

else:

status = currFeatureLayer.addFeature(currFeature)

QgsMessageLog.logMessage("In onSaveDemandDetails: added Feature: " + str(status), tag="TOMs panel")

QgsMessageLog.logMessage("In onSaveDemandDetails: Before commit", tag="TOMs panel")

"""reply = QMessageBox.information(None, "Information",

"Wait a moment ...",

QMessageBox.Ok)"""

attrs1 = currFeature.attributes()

QgsMessageLog.logMessage("In onSaveDemandDetails: currRestriction: " + str(attrs1),

tag="TOMs panel")

QgsMessageLog.logMessage(

("In onSaveDemandDetails. geometry: " + str(currFeature.geometry().exportToWkt())),

tag="TOMs panel")

QgsMessageLog.logMessage("In onSaveDemandDetails: currActiveLayer: " + str(self.iface.activeLayer().name()),

tag="TOMs panel")

#Test

#status = dialog.attributeForm().save()

#status = dialog.accept()

#status = dialog.accept()

"""reply = QMessageBox.information(None, "Information",

"And another ... iseditable: " + str(currFeatureLayer.isEditable()),

QMessageBox.Ok)"""

#currFeatureLayer.blockSignals(True)

if currFeatureID == 0:

self.iface.mapCanvas().unsetMapTool(self.iface.mapCanvas().mapTool())

QgsMessageLog.logMessage("In onSaveDemandDetails: mapTool unset",

tag="TOMs panel")

try:

currFeatureLayer.commitChanges()

except:

reply = QMessageBox.information(None, "Information", "Problem committing changes" + str(currFeatureLayer.commitErrors()), QMessageBox.Ok)

#currFeatureLayer.blockSignals(False)

QgsMessageLog.logMessage("In onSaveDemandDetails: changes committed", tag="TOMs panel")

status = dialog.close()

#status = dialog.accept()

#QgsMessageLog.logMessage("In onSaveDemandDetails: dialog reject: " + str(status), tag="TOMs panel")

def onRejectDemandDetailsFromForm(self, currFeature, currFeatureLayer, dialog):

QgsMessageLog.logMessage("In onRejectDemandDetailsFromForm", tag="TOMs panel")

# self.currDemandLayer.destroyEditCommand()

try:

self.camera1.endCamera()

except:

None

dialog.reject()

if currFeatureLayer.rollBack() == False:

reply = QMessageBox.information(None, "Information", "Problem rolling back changes", QMessageBox.Ok)

else:

QgsMessageLog.logMessage("In onRejectDemandDetailsFromForm: rollBack successful ...", tag="TOMs panel")

def onAttributeChangedClass2(self, currFeature, layer, fieldName, value):

QgsMessageLog.logMessage(

"In FormOpen:onAttributeChangedClass 2 - layer: " + str(layer.name()) + " (" + fieldName + "): " + str(

value), tag="TOMs panel")

# self.currFeature.setAttribute(fieldName, value)

try:

currFeature[layer.fieldNameIndex(fieldName)] = value

# currFeature.setAttribute(layer.fieldNameIndex(fieldName), value)

except:

reply = QMessageBox.information(None, "Error",

"onAttributeChangedClass2. Update failed for: " + str(

layer.name()) + " (" + fieldName + "): " + str(value),

QMessageBox.Ok) # rollback all changes

return

def photoDetails(self):

# Function to deal with photo fields

QgsMessageLog.logMessage("In photoDetails", tag="TOMs panel")

FIELD1 = self.demandDialog.findChild(QLabel, "Photo_Widget_01")

FIELD2 = self.demandDialog.findChild(QLabel, "Photo_Widget_02")

FIELD3 = self.demandDialog.findChild(QLabel, "Photo_Widget_03")

photoPath = QgsExpressionContextUtils.projectScope().variable('PhotoPath')

projectFolder = QgsExpressionContextUtils.projectScope().variable('project_folder')

path_absolute = os.path.join(projectFolder, photoPath)

if path_absolute == None:

reply = QMessageBox.information(None, "Information", "Please set value for PhotoPath.", QMessageBox.Ok)

return

# Check path exists ...

if os.path.isdir(path_absolute) == False:

reply = QMessageBox.information(None, "Information", "PhotoPath folder " + str(

path_absolute) + " does not exist. Please check value.", QMessageBox.Ok)

return

layerName = self.currDemandLayer.name()

# Generate the full path to the file

# fileName1 = "Photos"

fileName1 = "Photos_01"

fileName2 = "Photos_02"

fileName3 = "Photos_03"

idx1 = self.currDemandLayer.fieldNameIndex(fileName1)

idx2 = self.currDemandLayer.fieldNameIndex(fileName2)

idx3 = self.currDemandLayer.fieldNameIndex(fileName3)

QgsMessageLog.logMessage("In photoDetails. idx1: " + str(idx1) + "; " + str(idx2) + "; " + str(idx3),

tag="TOMs panel")

# if currFeatureFeature[idx1]:

# QgsMessageLog.logMessage("In photoDetails. photo1: " + str(currFeatureFeature[idx1]), tag="TOMs panel")

# QgsMessageLog.logMessage("In photoDetails. photo2: " + str(currFeatureFeature.attribute(idx2)), tag="TOMs panel")

# QgsMessageLog.logMessage("In photoDetails. photo3: " + str(currFeatureFeature.attribute(idx3)), tag="TOMs panel")

if FIELD1:

QgsMessageLog.logMessage("In photoDetails. FIELD 1 exisits",

tag="TOMs panel")

if self.currFeature[idx1]:

newPhotoFileName1 = os.path.join(path_absolute, self.currFeature[idx1])

else:

newPhotoFileName1 = None

# QgsMessageLog.logMessage("In photoDetails. Photo1: " + str(newPhotoFileName1), tag="TOMs panel")

pixmap1 = QPixmap(newPhotoFileName1)

if pixmap1.isNull():

pass

# FIELD1.setText('Picture could not be opened ({path})'.format(path=newPhotoFileName1))

else:

FIELD1.setPixmap(pixmap1)

FIELD1.setScaledContents(True)

QgsMessageLog.logMessage("In photoDetails. Photo1: " + str(newPhotoFileName1), tag="TOMs panel")

if cv2_available == True:

START_CAMERA_1 = self.demandDialog.findChild(QPushButton, "startCamera1")

TAKE_PHOTO_1 = self.demandDialog.findChild(QPushButton, "getPhoto1")

TAKE_PHOTO_1.setEnabled(False)

self.camera1 = formCamera(path_absolute, newPhotoFileName1)

START_CAMERA_1.clicked.connect(

functools.partial(self.camera1.useCamera, START_CAMERA_1, TAKE_PHOTO_1, FIELD1))

self.camera1.notifyPhotoTaken.connect(functools.partial(self.savePhotoTaken, idx1))

if FIELD2:

QgsMessageLog.logMessage("In photoDetails. FIELD 2 exisits",

tag="TOMs panel")

if self.currFeature[idx2]:

newPhotoFileName2 = os.path.join(path_absolute, self.currFeature[idx2])

else:

newPhotoFileName2 = None

# newPhotoFileName2 = os.path.join(path_absolute, str(self.currFeature[idx2]))

# newPhotoFileName2 = os.path.join(path_absolute, str(self.currFeature.attribute(fileName2)))

# QgsMessageLog.logMessage("In photoDetails. Photo2: " + str(newPhotoFileName2), tag="TOMs panel")

pixmap2 = QPixmap(newPhotoFileName2)

if pixmap2.isNull():

pass

# FIELD1.setText('Picture could not be opened ({path})'.format(path=newPhotoFileName1))

else:

FIELD2.setPixmap(pixmap2)

FIELD2.setScaledContents(True)

QgsMessageLog.logMessage("In photoDetails. Photo2: " + str(newPhotoFileName2), tag="TOMs panel")

if cv2_available == True:

START_CAMERA_2 = self.demandDialog.findChild(QPushButton, "startCamera2")

TAKE_PHOTO_2 = self.demandDialog.findChild(QPushButton, "getPhoto2")

TAKE_PHOTO_2.setEnabled(False)

self.camera2 = formCamera(path_absolute, newPhotoFileName2)

START_CAMERA_2.clicked.connect(

functools.partial(self.camera2.useCamera, START_CAMERA_2, TAKE_PHOTO_2, FIELD2))

self.camera2.notifyPhotoTaken.connect(functools.partial(self.savePhotoTaken, idx2))

if FIELD3:

QgsMessageLog.logMessage("In photoDetails. FIELD 3 exisits",

tag="TOMs panel")

if self.currFeature[idx3]:

newPhotoFileName3 = os.path.join(path_absolute, self.currFeature[idx3])

else:

newPhotoFileName3 = None

# newPhotoFileName3 = os.path.join(path_absolute, str(self.currFeature[idx3]))

# newPhotoFileName3 = os.path.join(path_absolute,

# str(self.currFeature.attribute(fileName3)))

# newPhotoFileName3 = os.path.join(path_absolute, str(layerName + "_Photos_03"))

# QgsMessageLog.logMessage("In photoDetails. Photo3: " + str(newPhotoFileName3), tag="TOMs panel")

pixmap3 = QPixmap(newPhotoFileName3)

if pixmap3.isNull():

pass

# FIELD1.setText('Picture could not be opened ({path})'.format(path=newPhotoFileName1))

else:

FIELD3.setPixmap(pixmap3)

FIELD3.setScaledContents(True)

QgsMessageLog.logMessage("In photoDetails. Photo3: " + str(newPhotoFileName3), tag="TOMs panel")

if cv2_available == True:

START_CAMERA_3 = self.demandDialog.findChild(QPushButton, "startCamera3")

TAKE_PHOTO_3 = self.demandDialog.findChild(QPushButton, "getPhoto3")

TAKE_PHOTO_3.setEnabled(False)

self.camera3 = formCamera(path_absolute, newPhotoFileName3)

START_CAMERA_3.clicked.connect(

functools.partial(self.camera3.useCamera, START_CAMERA_3, TAKE_PHOTO_3, FIELD3))

self.camera3.notifyPhotoTaken.connect(functools.partial(self.savePhotoTaken, idx3))

pass

@pyqtSlot(str)

def savePhotoTaken(self, idx, fileName):

QgsMessageLog.logMessage("In demandFormUtils::savePhotoTaken ... " + fileName + " idx: " + str(idx),

tag="TOMs panel")

if len(fileName) > 0:

simpleFile = ntpath.basename(fileName)

QgsMessageLog.logMessage("In demandFormUtils::savePhotoTaken. Simple file: " + simpleFile,

tag="TOMs panel")

try:

self.currFeature[idx] = simpleFile

#self.currFeature.setAttribute(idx, simpleFile)

QgsMessageLog.logMessage("In demandFormUtils::savePhotoTaken. attrib value changed",

tag="TOMs panel")

except:

QgsMessageLog.logMessage("In demandFormUtils::savePhotoTaken. problem changing attrib value",

tag="TOMs panel")

reply = QMessageBox.information(None, "Error",

"savePhotoTaken. problem changing attrib value",

notifyPhotoTaken = QtCore.pyqtSignal(str)

def __init__(self, path_absolute, currFileName):

QtCore.QObject.__init__(self)

self.path_absolute = path_absolute

self.currFileName = currFileName

self.camera = cvCamera()

@pyqtSlot(QPixmap)

def displayFrame(self, pixmap):

# QgsMessageLog.logMessage("In formCamera::displayFrame ... ", tag="TOMs panel")

self.FIELD.setPixmap(pixmap)

self.FIELD.setScaledContents(True)

QApplication.processEvents() # processes the event queue - https://stackoverflow.com/questions/43094589/opencv-imshow-prevents-qt-python-crashing

def useCamera(self, START_CAMERA_BUTTON, TAKE_PHOTO_BUTTON, FIELD):

QgsMessageLog.logMessage("In formCamera::useCamera ... ", tag="TOMs panel")

self.START_CAMERA_BUTTON = START_CAMERA_BUTTON

self.TAKE_PHOTO_BUTTON = TAKE_PHOTO_BUTTON

self.FIELD = FIELD

# self.blockSignals(True)

self.START_CAMERA_BUTTON.clicked.disconnect()

self.START_CAMERA_BUTTON.clicked.connect(self.endCamera)

""" Camera code """

self.camera.changePixmap.connect(self.displayFrame)

self.camera.closeCamera.connect(self.endCamera)

self.TAKE_PHOTO_BUTTON.setEnabled(True)

self.TAKE_PHOTO_BUTTON.clicked.connect(functools.partial(self.camera.takePhoto, self.path_absolute))

self.camera.photoTaken.connect(self.checkPhotoTaken)

self.photoTaken = False

QgsMessageLog.logMessage("In formCamera::useCamera: starting camera ... ", tag="TOMs panel")

self.camera.startCamera()

def endCamera(self):

QgsMessageLog.logMessage("In formCamera::endCamera: stopping camera ... ", tag="TOMs panel")

self.camera.stopCamera()

self.camera.changePixmap.disconnect(self.displayFrame)

self.camera.closeCamera.disconnect(self.endCamera)

# del self.camera

self.TAKE_PHOTO_BUTTON.setEnabled(False)

self.START_CAMERA_BUTTON.setChecked(False)

self.TAKE_PHOTO_BUTTON.clicked.disconnect()

self.START_CAMERA_BUTTON.clicked.disconnect()

self.START_CAMERA_BUTTON.clicked.connect(

functools.partial(self.useCamera, self.START_CAMERA_BUTTON, self.TAKE_PHOTO_BUTTON, self.FIELD))

if self.photoTaken == False:

self.resetPhoto()

@pyqtSlot(str)

def checkPhotoTaken(self, fileName):

QgsMessageLog.logMessage("In formCamera::photoTaken: file: " + fileName, tag="TOMs panel")

if len(fileName) > 0:

self.photoTaken = True

self.notifyPhotoTaken.emit(fileName)

else:

self.resetPhoto()

self.photoTaken = False

def resetPhoto(self):

QgsMessageLog.logMessage("In formCamera::resetPhoto ... ", tag="TOMs panel")

pixmap = QPixmap(self.currFileName)

if pixmap.isNull():

pass

else:

self.FIELD.setPixmap(pixmap)

changePixmap = QtCore.pyqtSignal(QPixmap)

closeCamera = QtCore.pyqtSignal()

photoTaken = QtCore.pyqtSignal(str)

def __init__(self):

QtCore.QThread.__init__(self)

def stopCamera(self):

QgsMessageLog.logMessage("In cvCamera::stopCamera ... ", tag="TOMs panel")

self.cap.release()

def startCamera(self):

QgsMessageLog.logMessage("In cvCamera::startCamera: ... ", tag="TOMs panel")

self.cap = cv2.VideoCapture(0) # video capture source camera (Here webcam of laptop)

self.cap.set(3, 640) # width=640

self.cap.set(4, 480) # height=480

while self.cap.isOpened():

self.getFrame()

# cv2.imshow('img1',self.frame) #display the captured image

# cv2.waitKey(1)

time.sleep(0.1) # QTimer::singleShot()

else:

QgsMessageLog.logMessage("In cvCamera::startCamera: camera closed ... ", tag="TOMs panel")

self.closeCamera.emit()

def getFrame(self):

""" Camera code """

# QgsMessageLog.logMessage("In cvCamera::getFrame ... ", tag="TOMs panel")

ret, self.frame = self.cap.read() # return a single frame in variable `frame`

if ret == True:

# Need to change from BRG (cv::mat) to RGB image

cvRGBImg = cv2.cvtColor(self.frame, cv2.cv.CV_BGR2RGB)

qimg = QImage(cvRGBImg.data, cvRGBImg.shape[1], cvRGBImg.shape[0], QImage.Format_RGB888)

# Now display ...

pixmap = QPixmap.fromImage(qimg)

self.changePixmap.emit(pixmap)

else:

QgsMessageLog.logMessage("In cvCamera::useCamera: frame not returned ... ", tag="TOMs panel")

self.closeCamera.emit()

def takePhoto(self, path_absolute):

QgsMessageLog.logMessage("In cvCamera::takePhoto ... ", tag="TOMs panel")

# Save frame to file

fileName = 'Photo_{}.png'.format(datetime.datetime.now().strftime('%Y%m%d_%H%M%S%z'))

newPhotoFileName = os.path.join(path_absolute, fileName)

QgsMessageLog.logMessage("Saving photo: file: " + newPhotoFileName, tag="TOMs panel")

writeStatus = cv2.imwrite(newPhotoFileName, self.frame)

if writeStatus is True:

reply = QMessageBox.information(None, "Information", "Photo captured.", QMessageBox.Ok)

self.photoTaken.emit(newPhotoFileName)

else:

reply = QMessageBox.information(None, "Information", "Problem taking photo.", QMessageBox.Ok)

self.photoTaken.emit()

# Now stop camera (and display image)

self.cap.release()

"""def fps(self):

fps = int(cv.GetCaptureProperty(self._cameraDevice, cv.CV_CAP_PROP_FPS))

if not fps > 0:

fps = self._DEFAULT_FPS

return fps"""