class OutputTabManager(QtCore.QObject):
''' Class to hide managing of relative tab
'''
projectModified = QtCore.pyqtSignal()
def __init__(self, parent=None):
'''constructor'''
super(OutputTabManager, self).__init__(parent)
# parent is the dock widget with all graphical elements
self.gui = parent
self.plugin = parent.parent
# init some globals
self.applicationPath = os.path.dirname(os.path.realpath(__file__))
self.project = None
self.projectPath = None
self.outLayer = None
self.outLayerId = None
self.outLayerRemoved = False
# retrieve the current tab index
self.initTabTabIndex()
# disable tab at the beginning
self.gui.tabWidget.setTabEnabled(self.tabIndex, False)
# add some gui events
self.gui.createNewLayer_RButton.toggled.connect(self.setInputLayer)
self.gui.selectOutfile_TButton.clicked.connect(self.selectNewOutFile)
self.gui.calculate_PButton.clicked.connect(self.calculate)
def initTabTabIndex(self):
''' Retrieve what tab index refer the current tab manager
'''
for tabIndex in range(self.gui.tabWidget.count()):
if self.gui.tabWidget.tabText(tabIndex) == "Output":
self.tabIndex = tabIndex
def setProject(self, project=None):
''' setting the new project on which the tab is based
'''
self.project = project
if self.project:
# write everything on disk
self.project.sync()
# set some globals
confFileName = self.project.fileName()
self.projectPath = os.path.dirname(confFileName)
# emit configurationLoaded with the status of loading
self.setTabGUIBasingOnProject()
# enable current tab because project has been loaded
self.gui.tabWidget.setTabEnabled(self.tabIndex, True)
else:
# disable current tab because no project has been loaded yet
self.gui.tabWidget.setTabEnabled(self.tabIndex, False)
def selectNewOutFile(self):
''' Select the the file
'''
oldOutputLayer = self.gui.outFile_LEdit.text()
# get last conf to start from its path
startPath = self.projectPath
newOutputLayer = self.project.value(
'Processing.OutputFileDefinition/newOutputLayer', '')
if newOutputLayer:
startPath = os.path.dirname(newOutputLayer)
# ask for the new out file
newOutputLayer = QtGui.QFileDialog.getSaveFileName(
self.gui, "Select an autput file", startPath,
self.plugin.tr("All (*)"))
if not newOutputLayer:
return
if oldOutputLayer == newOutputLayer:
return
# add shp extension
components = os.path.splitext(newOutputLayer)
if len(components) == 1:
newOutputLayer = newOutputLayer + '.shp'
else:
if components[1] != '.shp':
newOutputLayer = components[0] + '.shp'
# update gui
self.gui.outFile_LEdit.setText(newOutputLayer)
self.saveTabOnProject()
def setInputLayer(self):
''' Set GUI basing on toggled ComboBox to set if the layer is the input one
or have to select a new one
'''
self.gui.outFile_LEdit.setEnabled(
self.gui.createNewLayer_RButton.isChecked())
self.gui.selectOutfile_TButton.setEnabled(
self.gui.createNewLayer_RButton.isChecked())
def setTabGUIBasingOnProject(self):
'''Set tab basing on project conf
'''
if not self.project:
return
# get conf parameters
addToInputLayer = self.project.value(
'Processing.OutputFileDefinition/addToInputLayer', False, bool)
newOutputLayer = self.project.value(
'Processing.OutputFileDefinition/newOutputLayer', '')
Gasoline_Consumption = self.project.value(
'Processing.Parameters/Gasoline_Consumption', False, bool)
Diesel_Consumption = self.project.value(
'Processing.Parameters/Diesel_Consumption', False, bool)
LPG_Consumption = self.project.value(
'Processing.Parameters/LPG_Consumption', False, bool)
NewFuel_Consumption = self.project.value(
'Processing.Parameters/NewFuel_Consumption', False, bool)
Fuel_Consumption_Total = self.project.value(
'Processing.Parameters/Fuel_Consumption_Total', False, bool)
Energy_Consumption = self.project.value(
'Processing.Parameters/Energy_Consumption', False, bool)
CO = self.project.value('Processing.Parameters/CO', False, bool)
NOx = self.project.value('Processing.Parameters/NOx', False, bool)
NMVOC = self.project.value('Processing.Parameters/NMVOC', False, bool)
CO2 = self.project.value('Processing.Parameters/CO2', False, bool)
CH4 = self.project.value('Processing.Parameters/CH4', False, bool)
N2O = self.project.value('Processing.Parameters/N2O', False, bool)
NH3 = self.project.value('Processing.Parameters/NH3', False, bool)
SO2 = self.project.value('Processing.Parameters/SO2', False, bool)
PM25 = self.project.value('Processing.Parameters/PM', False, bool)
C6H6 = self.project.value('Processing.Parameters/C6H6', False, bool)
# avoid emitting signal in case of reset of indexes
try:
# to avoid add multiple listener, remove previous listener
self.gui.addToOriginaLayer_RButton.toggled.disconnect(
self.saveTabOnProject)
self.gui.createNewLayer_RButton.toggled.disconnect(
self.saveTabOnProject)
self.gui.outFile_LEdit.returnPressed.disconnect(
self.saveTabOnProject)
self.gui.fuelEnergyConsumptionGasoline_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.fuelEnergyConsumptionDiesel_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.fuelEnergyConsumptionLPG_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.fuelEnergyConsumptionNewFuels_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.totalFuelConsumption_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.energyConsumption_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsCO_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsNOx_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsNMVOCs_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsCO2_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsCH4_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsN2O_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsNH3_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsSO2_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsPM25_CBox.clicked.disconnect(
self.saveTabOnProject)
self.gui.pollutantsC6H6_CBox.clicked.disconnect(
self.saveTabOnProject)
except (Exception) as ex:
pass
# now populate interface
self.gui.addToOriginaLayer_RButton.setChecked(addToInputLayer)
self.gui.createNewLayer_RButton.setChecked(not addToInputLayer)
self.gui.outFile_LEdit.setText(newOutputLayer)
self.gui.fuelEnergyConsumptionGasoline_CBox.setChecked(
Gasoline_Consumption)
self.gui.fuelEnergyConsumptionDiesel_CBox.setChecked(
Diesel_Consumption)
self.gui.fuelEnergyConsumptionLPG_CBox.setChecked(LPG_Consumption)
self.gui.fuelEnergyConsumptionNewFuels_CBox.setChecked(
NewFuel_Consumption)
self.gui.totalFuelConsumption_CBox.setChecked(Fuel_Consumption_Total)
self.gui.energyConsumption_CBox.setChecked(Energy_Consumption)
self.gui.pollutantsCO_CBox.setChecked(CO)
self.gui.pollutantsNOx_CBox.setChecked(NOx)
self.gui.pollutantsNMVOCs_CBox.setChecked(NMVOC)
self.gui.pollutantsCO2_CBox.setChecked(CO2)
self.gui.pollutantsCH4_CBox.setChecked(CH4)
self.gui.pollutantsN2O_CBox.setChecked(N2O)
self.gui.pollutantsNH3_CBox.setChecked(NH3)
self.gui.pollutantsSO2_CBox.setChecked(SO2)
self.gui.pollutantsPM25_CBox.setChecked(PM25)
self.gui.pollutantsC6H6_CBox.setChecked(C6H6)
# add all modification events to notify project modification
self.gui.addToOriginaLayer_RButton.toggled.connect(
self.saveTabOnProject)
self.gui.createNewLayer_RButton.toggled.connect(self.saveTabOnProject)
self.gui.outFile_LEdit.returnPressed.connect(self.saveTabOnProject)
self.gui.fuelEnergyConsumptionGasoline_CBox.clicked.connect(
self.saveTabOnProject)
self.gui.fuelEnergyConsumptionDiesel_CBox.clicked.connect(
self.saveTabOnProject)
self.gui.fuelEnergyConsumptionLPG_CBox.clicked.connect(
self.saveTabOnProject)
self.gui.fuelEnergyConsumptionNewFuels_CBox.clicked.connect(
self.saveTabOnProject)
self.gui.totalFuelConsumption_CBox.clicked.connect(
self.saveTabOnProject)
self.gui.energyConsumption_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsCO_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsNOx_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsNMVOCs_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsCO2_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsCH4_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsN2O_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsNH3_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsSO2_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsPM25_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsC6H6_CBox.clicked.connect(self.saveTabOnProject)
# if and output file is specified and it's available => load it
if newOutputLayer and os.path.exists(newOutputLayer):
# check if layer is already loaded in layer list checking it's source
found = False
for layerName, layer in QgsMapLayerRegistry.instance().mapLayers(
).items():
if newOutputLayer == layer.publicSource():
# set the found layer as roadLayer
self.outLayer = layer
self.outLayerId = self.outLayer.id()
found = True
break
if not found:
# get layer name to set as public name in the legend
layerName = os.path.splitext(
os.path.basename(newOutputLayer))[0]
# load layer
self.outLayer = QgsVectorLayer(newOutputLayer, layerName,
'ogr')
if not self.outLayer.isValid():
message = self.tr(
"Error loading layer: %s" %
self.outLayer.error().message(QgsErrorMessage.Text))
iface.messageBar().pushMessage(message,
QgsMessageBar.CRITICAL)
return
self.outLayerId = self.outLayer.id()
# show layer in the canvas
QgsMapLayerRegistry.instance().addMapLayer(self.outLayer)
def saveTabOnProject(self):
''' Save tab configuration in the project basing on GUI values
'''
# get values from the GUI
addToInputLayer = self.gui.addToOriginaLayer_RButton.isChecked()
newOutputLayer = self.gui.outFile_LEdit.text()
Gasoline_Consumption = self.gui.fuelEnergyConsumptionGasoline_CBox.isChecked(
)
Diesel_Consumption = self.gui.fuelEnergyConsumptionDiesel_CBox.isChecked(
)
LPG_Consumption = self.gui.fuelEnergyConsumptionLPG_CBox.isChecked()
NewFuel_Consumption = self.gui.fuelEnergyConsumptionNewFuels_CBox.isChecked(
)
Fuel_Consumption_Total = self.gui.totalFuelConsumption_CBox.isChecked()
Energy_Consumption = self.gui.energyConsumption_CBox.isChecked()
CO = self.gui.pollutantsCO_CBox.isChecked()
NOx = self.gui.pollutantsNOx_CBox.isChecked()
NMVOC = self.gui.pollutantsNMVOCs_CBox.isChecked()
CO2 = self.gui.pollutantsCO2_CBox.isChecked()
CH4 = self.gui.pollutantsCH4_CBox.isChecked()
N2O = self.gui.pollutantsN2O_CBox.isChecked()
NH3 = self.gui.pollutantsNH3_CBox.isChecked()
SO2 = self.gui.pollutantsSO2_CBox.isChecked()
PM25 = self.gui.pollutantsPM25_CBox.isChecked()
C6H6 = self.gui.pollutantsC6H6_CBox.isChecked()
# set conf parameters
self.project.setValue(
'Processing.OutputFileDefinition/addToInputLayer',
int(addToInputLayer))
self.project.setValue('Processing.OutputFileDefinition/newOutputLayer',
newOutputLayer)
self.project.setValue('Processing.Parameters/Gasoline_Consumption',
int(Gasoline_Consumption))
self.project.setValue('Processing.Parameters/Diesel_Consumption',
int(Diesel_Consumption))
self.project.setValue('Processing.Parameters/LPG_Consumption',
int(LPG_Consumption))
self.project.setValue('Processing.Parameters/NewFuel_Consumption',
int(NewFuel_Consumption))
self.project.setValue('Processing.Parameters/Fuel_Consumption_Total',
int(Fuel_Consumption_Total))
self.project.setValue('Processing.Parameters/Energy_Consumption',
int(Energy_Consumption))
self.project.setValue('Processing.Parameters/CO', int(CO))
self.project.setValue('Processing.Parameters/NOx', int(NOx))
self.project.setValue('Processing.Parameters/NMVOC', int(NMVOC))
self.project.setValue('Processing.Parameters/CO2', int(CO2))
self.project.setValue('Processing.Parameters/CH4', int(CH4))
self.project.setValue('Processing.Parameters/N2O', int(N2O))
self.project.setValue('Processing.Parameters/NH3', int(NH3))
self.project.setValue('Processing.Parameters/SO2', int(SO2))
self.project.setValue('Processing.Parameters/PM', int(PM25))
self.project.setValue('Processing.Parameters/C6H6', int(C6H6))
# notify project modification
self.projectModified.emit()
def calculate(self):
''' Prepare environment to run the alg and run it. After run, merge produced
data basing on plugin configuration.
Before calculation a parametere validation will be executed
'''
# perform validation
if not self.gui.validate():
return
else:
# notify successful validation
message = self.tr(
"QTraffic: Parameters validation passed successfully")
iface.messageBar().pushMessage(message, QgsMessageBar.SUCCESS)
# set number of classes in the project config (that is the temporary one... but equal to the official one)
fleetDistributionRoadTypes = self.gui.getRoadTypes()
self.project.setValue('Processing.Parameters/maximum_type',
len(fleetDistributionRoadTypes))
self.project.sync()
# create the algorithm
self.alg = Algorithm()
roadLayer = self.gui.getRoadLayer()
# prepare layer where to add result
addToInputLayer = self.gui.addToOriginaLayer_RButton.isChecked()
newOutputLayer = self.gui.outFile_LEdit.text()
if addToInputLayer:
self.outLayer = roadLayer
self.outLayerId = self.outLayer.id()
else:
# if layer is present... remove it
# out layer would not be the same of input road layer... in thi scase don't remove it
if self.outLayer and self.outLayer.isValid():
# to be sure, remove only if roadLayer and outLayer are different
if self.outLayer.publicSource() != roadLayer.publicSource():
self.outLayerRemoved = False
QgsMapLayerRegistry.instance().layerRemoved.connect(
self.checkOutLayerRemoved)
QgsMapLayerRegistry.instance().removeMapLayer(
self.outLayer.id())
# remove file when it has been removed from qgis
while not self.outLayerRemoved:
sleep(0.1)
QgsMapLayerRegistry.instance().layerRemoved.disconnect(
self.checkOutLayerRemoved)
# reinit outLayer variables
# If not, under windws remain a locking of the related file creating
# an error during QgsVectorFileWriter.deleteShapeFile
self.outLayer = None
self.outLayerId = None
if os.path.exists(newOutputLayer):
if not QgsVectorFileWriter.deleteShapeFile(
newOutputLayer):
message = self.tr(
"Error removing shape: {}".format(
newOutputLayer))
iface.messageBar().pushMessage(
message, QgsMessageBar.CRITICAL)
return
# copy input layer to the new one
writeError = QgsVectorFileWriter.writeAsVectorFormat(
roadLayer, newOutputLayer, 'utf-8', roadLayer.crs())
if writeError != QgsVectorFileWriter.NoError:
message = self.tr(
'Error writing vector file {}'.format(newOutputLayer))
QgsMessageLog.logMessage(message, 'QTraffic',
QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
# load the layer
newLayerName = os.path.splitext(
os.path.basename(newOutputLayer))[0]
self.outLayer = QgsVectorLayer(newOutputLayer, newLayerName, 'ogr')
if not self.outLayer.isValid():
message = self.tr(
'Error loading vector file {}'.format(newOutputLayer))
QgsMessageLog.logMessage(message, 'QTraffic',
QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
self.outLayerId = self.outLayer.id()
# prepare environment
try:
self.alg.setProject(self.project)
self.alg.setLayer(roadLayer)
self.alg.initConfig()
self.alg.prepareRun()
except Exception as ex:
traceback.print_exc()
message = self.tr(
'Error preparing running contex for the algoritm: %s' %
str(ex))
QgsMessageLog.logMessage(message, 'QTraffic',
QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
# run the self.alg
self.thread = QtCore.QThread(self)
self.thread.started.connect(self.alg.run)
self.thread.finished.connect(self.threadCleanup)
self.thread.terminated.connect(self.threadCleanup)
self.alg.moveToThread(self.thread)
self.alg.started.connect(self.manageStarted)
self.alg.progress.connect(self.manageProgress)
self.alg.message.connect(self.manageMessage)
self.alg.error.connect(self.manageError)
self.alg.finished.connect(self.manageFinished)
# set wait cursor and start
QgsApplication.instance().setOverrideCursor(QtCore.Qt.WaitCursor)
self.thread.start()
def checkOutLayerRemoved(self, layerId):
''' check when outLayer has been removed settign semaphore to True
'''
if self.outLayerId == layerId:
QgsMessageLog.logMessage('Completly removed outLayer', 'QTraffic',
QgsMessageLog.INFO)
self.outLayerRemoved = True
def threadCleanup(self):
''' cleanup after thread run
'''
# restore cursor
QgsApplication.instance().restoreOverrideCursor()
if self.alg:
self.alg.deleteLater()
self.alg = None
self.thread.wait()
self.thread.deleteLater()
# remove progress bar
if self.progressMessageBarItem:
iface.messageBar().popWidget(self.progressMessageBarItem)
self.progressMessageBarItem = None
def manageStarted(self):
''' Setup GUI env to notify processing progress
'''
message = self.tr('Processing started')
QgsMessageLog.logMessage(message, 'QTraffic', QgsMessageLog.INFO)
# create message bar to show progress
self.progressMessageBarItem = iface.messageBar().createMessage(
self.tr('Executing alg'))
self.progress = QtGui.QProgressBar()
self.progress.setMaximum(0)
self.progress.setMinimum(0)
self.progress.setAlignment(QtCore.Qt.AlignLeft
| QtCore.Qt.AlignVCenter)
self.progressMessageBarItem.layout().addWidget(self.progress)
iface.messageBar().pushWidget(self.progressMessageBarItem,
QgsMessageBar.INFO)
def manageProgress(self, percentage):
''' Update progress bar basing on alg nofify
'''
self.progress.setValue(percentage)
def manageMessage(self, message, msgType):
''' expose alg thread messages to gui and log
'''
QgsMessageLog.logMessage(message, 'QTraffic', msgType)
duration = 0
if msgType == QgsMessageLog.INFO:
return
if msgType == QgsMessageLog.CRITICAL:
msgType = QgsMessageBar.CRITICAL
if msgType == QgsMessageLog.WARNING:
duration = 3
msgType = QgsMessageBar.WARNING
iface.messageBar().pushMessage(message, msgType, duration)
def manageError(self, ex, exceptionMessage):
''' Do actions in case of alg thread error. Now only notify exception
'''
QgsMessageLog.logMessage(exceptionMessage, 'QTraffic',
QgsMessageLog.CRITICAL)
iface.messageBar().pushMessage(exceptionMessage,
QgsMessageBar.CRITICAL)
def manageFinished(self, success, reason):
''' Do action after notify that alg is finished. These are the postprocessing steps
1) merge result to the output layer
2) add the layer to canvas in case it is new
3) notify edn of processing
4) terminate the thread
'''
# finish the thread
self.thread.quit()
# check result
if not success:
QgsMessageLog.logMessage('Failed execution: {}'.format(reason),
'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical(
'QTraffic',
self.tr("Error executing the algorithm: {}".format(reason)))
return
# prepare result
try:
self.alg.addResultToLayer(self.outLayer)
except Exception as ex:
traceback.print_exc()
QgsMessageLog.logMessage(
'Cannot add result to layer: {}'.format(str(ex)), 'QTraffic',
QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical(
'QTraffic', self.tr("Cannot add result to layer"))
return
# add or refresh rsult vector layer
addToInputLayer = self.gui.addToOriginaLayer_RButton.isChecked()
if not addToInputLayer:
QgsMapLayerRegistry.instance().addMapLayer(self.outLayer)
iface.mapCanvas().refresh()
# notify the user the end of process
iface.messageBar().pushSuccess('QTraffic',
self.tr('Alg terminated successfully'))
def validate(self):
''' pre calcluation validation related only to this tab
Mandatory:
At least an output parameters have to be set
if "create new layer" a new layer name have to be set
'''
addToInputLayer = self.gui.addToOriginaLayer_RButton.isChecked()
newOutputLayer = self.gui.outFile_LEdit.text()
if (not addToInputLayer and not newOutputLayer):
message = self.tr('No output vector specified')
iface.messageBar().pushMessage(message, QgsMessageBar.CRITICAL)
return False
Gasoline_Consumption = self.gui.fuelEnergyConsumptionGasoline_CBox.isChecked(
)
Diesel_Consumption = self.gui.fuelEnergyConsumptionDiesel_CBox.isChecked(
)
LPG_Consumption = self.gui.fuelEnergyConsumptionLPG_CBox.isChecked()
NewFuel_Consumption = self.gui.fuelEnergyConsumptionNewFuels_CBox.isChecked(
)
Fuel_Consumption_Total = self.gui.totalFuelConsumption_CBox.isChecked()
Energy_Consumption = self.gui.energyConsumption_CBox.isChecked()
CO = self.gui.pollutantsCO_CBox.isChecked()
NOx = self.gui.pollutantsNOx_CBox.isChecked()
NMVOC = self.gui.pollutantsNMVOCs_CBox.isChecked()
CO2 = self.gui.pollutantsCO2_CBox.isChecked()
CH4 = self.gui.pollutantsCH4_CBox.isChecked()
N2O = self.gui.pollutantsN2O_CBox.isChecked()
NH3 = self.gui.pollutantsNH3_CBox.isChecked()
SO2 = self.gui.pollutantsSO2_CBox.isChecked()
PM25 = self.gui.pollutantsPM25_CBox.isChecked()
C6H6 = self.gui.pollutantsC6H6_CBox.isChecked()
if (not Gasoline_Consumption and not Diesel_Consumption
and not LPG_Consumption and not NewFuel_Consumption
and not Fuel_Consumption_Total and not Energy_Consumption
and not CO and not NOx and not NMVOC and not CO2 and not CH4
and not N2O and not NH3 and not SO2 and not PM25 and not C6H6):
message = self.tr(
"Validation error: At least an output parameter have to be selected"
)
iface.messageBar().pushMessage(message, QgsMessageBar.CRITICAL)
return False
return True
def tr(self, string, context=''):
if not context:
context = 'QTraffic'
return QtCore.QCoreApplication.translate(context, string)
def calculate(self):
''' Prepare environment to run the alg and run it. After run, merge produced
data basing on plugin configuration
'''
alg = Algorithm()
roadLayer = self.gui.getRoadLayer()
# prepare layer where to add result
addToInputLayer = self.gui.addToOriginaLayer_RButton.isChecked()
newOutputLayer = self.gui.outFile_LEdit.text()
self.outLayer = roadLayer
if not addToInputLayer:
if not newOutputLayer:
message = self.tr('No output vector specified')
QgsMessageLog.logMessage(message, 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
# copy input layer to the new one
writeError = QgsVectorFileWriter.writeAsVectorFormat(roadLayer, newOutputLayer, 'utf-8', roadLayer.crs())
if writeError != QgsVectorFileWriter.NoError:
message = self.tr('Error writing vector file {}'.format(newOutputLayer))
QgsMessageLog.logMessage(message, 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
# load the layer
newLayerName =os.path.splitext(os.path.basename( newOutputLayer ))[0]
self.outLayer = QgsVectorLayer(newOutputLayer, newLayerName, 'ogr')
if not self.outLayer.isValid():
message = self.tr('Error loading vector file {}'.format(newOutputLayer))
QgsMessageLog.logMessage(message, 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
# prepare environment
try:
alg.setProject(self.project)
alg.setLayer( roadLayer )
alg.init()
alg.prepareRun()
except:
traceback.print_exc()
message = self.tr('Error preparing context for the algoritm')
QgsMessageLog.logMessage(message, 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
# run the alg
success = alg.run()
if not success:
return
# prepare result
try:
alg.addResultToLayer(self.outLayer)
except:
return
# add or refresh rsult vector layer
if not addToInputLayer:
QgsMapLayerRegistry.instance().addMapLayer(self.outLayer)
iface.mapCanvas().refresh()
class OutputTabManager(QtCore.QObject):
''' Class to hide managing of relative tab
'''
projectModified = QtCore.pyqtSignal()
def __init__(self, parent=None):
'''constructor'''
super(OutputTabManager, self).__init__(parent)
# parent is the dock widget with all graphical elements
self.gui = parent
self.plugin = parent.parent
# init some globals
self.applicationPath = os.path.dirname(os.path.realpath(__file__))
self.project = None
self.projectPath = None
self.outLayer = None
self.outLayerId = None
self.outLayerRemoved = False
# retrieve the current tab index
self.initTabTabIndex()
# disable tab at the beginning
self.gui.tabWidget.setTabEnabled(self.tabIndex, False)
# add some gui events
self.gui.createNewLayer_RButton.toggled.connect(self.setInputLayer)
self.gui.selectOutfile_TButton.clicked.connect(self.selectNewOutFile)
self.gui.calculate_PButton.clicked.connect(self.calculate)
def initTabTabIndex(self):
''' Retrieve what tab index refer the current tab manager
'''
for tabIndex in range(self.gui.tabWidget.count()):
if self.gui.tabWidget.tabText(tabIndex) == "Output":
self.tabIndex = tabIndex
def setProject(self, project=None):
''' setting the new project on which the tab is based
'''
self.project = project
if self.project:
# write everything on disk
self.project.sync()
# set some globals
confFileName = self.project.fileName()
self.projectPath = os.path.dirname(confFileName)
# emit configurationLoaded with the status of loading
self.setTabGUIBasingOnProject()
# enable current tab because project has been loaded
self.gui.tabWidget.setTabEnabled(self.tabIndex, True)
else:
# disable current tab because no project has been loaded yet
self.gui.tabWidget.setTabEnabled(self.tabIndex, False)
def selectNewOutFile(self):
''' Select the the file
'''
oldOutputLayer = self.gui.outFile_LEdit.text()
# get last conf to start from its path
startPath = self.projectPath
newOutputLayer = self.project.value('Processing.OutputFileDefinition/newOutputLayer', '')
if newOutputLayer:
startPath = os.path.dirname( newOutputLayer )
# ask for the new out file
newOutputLayer = QtGui.QFileDialog.getSaveFileName(self.gui, "Select an autput file", startPath,
self.plugin.tr("All (*)"))
if not newOutputLayer:
return
if oldOutputLayer == newOutputLayer:
return
# add shp extension
components = os.path.splitext(newOutputLayer)
if len(components) == 1:
newOutputLayer = newOutputLayer + '.shp'
else:
if components[1] != '.shp':
newOutputLayer = components[0] + '.shp'
# update gui
self.gui.outFile_LEdit.setText(newOutputLayer)
self.saveTabOnProject()
def setInputLayer(self):
''' Set GUI basing on toggled ComboBox to set if the layer is the input one
or have to select a new one
'''
self.gui.outFile_LEdit.setEnabled( self.gui.createNewLayer_RButton.isChecked() )
self.gui.selectOutfile_TButton.setEnabled( self.gui.createNewLayer_RButton.isChecked() )
def setTabGUIBasingOnProject(self):
'''Set tab basing on project conf
'''
if not self.project:
return
# get conf parameters
addToInputLayer = self.project.value('Processing.OutputFileDefinition/addToInputLayer', False, bool)
newOutputLayer = self.project.value('Processing.OutputFileDefinition/newOutputLayer', '')
Gasoline_Consumption = self.project.value('Processing.Parameters/Gasoline_Consumption', False, bool)
Diesel_Consumption = self.project.value('Processing.Parameters/Diesel_Consumption', False, bool)
LPG_Consumption = self.project.value('Processing.Parameters/LPG_Consumption', False, bool)
NewFuel_Consumption = self.project.value('Processing.Parameters/NewFuel_Consumption', False, bool)
Fuel_Consumption_Total = self.project.value('Processing.Parameters/Fuel_Consumption_Total', False, bool)
Energy_Consumption = self.project.value('Processing.Parameters/Energy_Consumption', False, bool)
CO = self.project.value('Processing.Parameters/CO', False, bool)
NOx = self.project.value('Processing.Parameters/NOx', False, bool)
NMVOC = self.project.value('Processing.Parameters/NMVOC', False, bool)
CO2 = self.project.value('Processing.Parameters/CO2', False, bool)
CH4 = self.project.value('Processing.Parameters/CH4', False, bool)
N2O = self.project.value('Processing.Parameters/N2O', False, bool)
NH3 = self.project.value('Processing.Parameters/NH3', False, bool)
SO2 = self.project.value('Processing.Parameters/SO2', False, bool)
PM25 = self.project.value('Processing.Parameters/PM', False, bool)
C6H6 = self.project.value('Processing.Parameters/C6H6', False, bool)
# avoid emitting signal in case of reset of indexes
try:
# to avoid add multiple listener, remove previous listener
self.gui.addToOriginaLayer_RButton.toggled.disconnect(self.saveTabOnProject)
self.gui.createNewLayer_RButton.toggled.disconnect(self.saveTabOnProject)
self.gui.outFile_LEdit.returnPressed.disconnect(self.saveTabOnProject)
self.gui.fuelEnergyConsumptionGasoline_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.fuelEnergyConsumptionDiesel_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.fuelEnergyConsumptionLPG_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.fuelEnergyConsumptionNewFuels_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.totalFuelConsumption_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.energyConsumption_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsCO_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsNOx_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsNMVOCs_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsCO2_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsCH4_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsN2O_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsNH3_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsSO2_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsPM25_CBox.clicked.disconnect(self.saveTabOnProject)
self.gui.pollutantsC6H6_CBox.clicked.disconnect(self.saveTabOnProject)
except (Exception) as ex:
pass
# now populate interface
self.gui.addToOriginaLayer_RButton.setChecked( addToInputLayer )
self.gui.createNewLayer_RButton.setChecked( not addToInputLayer )
self.gui.outFile_LEdit.setText( newOutputLayer )
self.gui.fuelEnergyConsumptionGasoline_CBox.setChecked( Gasoline_Consumption )
self.gui.fuelEnergyConsumptionDiesel_CBox.setChecked( Diesel_Consumption )
self.gui.fuelEnergyConsumptionLPG_CBox.setChecked( LPG_Consumption )
self.gui.fuelEnergyConsumptionNewFuels_CBox.setChecked( NewFuel_Consumption )
self.gui.totalFuelConsumption_CBox.setChecked( Fuel_Consumption_Total )
self.gui.energyConsumption_CBox.setChecked( Energy_Consumption )
self.gui.pollutantsCO_CBox.setChecked( CO )
self.gui.pollutantsNOx_CBox.setChecked( NOx )
self.gui.pollutantsNMVOCs_CBox.setChecked( NMVOC )
self.gui.pollutantsCO2_CBox.setChecked( CO2 )
self.gui.pollutantsCH4_CBox.setChecked( CH4 )
self.gui.pollutantsN2O_CBox.setChecked( N2O )
self.gui.pollutantsNH3_CBox.setChecked( NH3 )
self.gui.pollutantsSO2_CBox.setChecked( SO2 )
self.gui.pollutantsPM25_CBox.setChecked( PM25 )
self.gui.pollutantsC6H6_CBox.setChecked( C6H6 )
# add all modification events to notify project modification
self.gui.addToOriginaLayer_RButton.toggled.connect(self.saveTabOnProject)
self.gui.createNewLayer_RButton.toggled.connect(self.saveTabOnProject)
self.gui.outFile_LEdit.returnPressed.connect(self.saveTabOnProject)
self.gui.fuelEnergyConsumptionGasoline_CBox.clicked.connect(self.saveTabOnProject)
self.gui.fuelEnergyConsumptionDiesel_CBox.clicked.connect(self.saveTabOnProject)
self.gui.fuelEnergyConsumptionLPG_CBox.clicked.connect(self.saveTabOnProject)
self.gui.fuelEnergyConsumptionNewFuels_CBox.clicked.connect(self.saveTabOnProject)
self.gui.totalFuelConsumption_CBox.clicked.connect(self.saveTabOnProject)
self.gui.energyConsumption_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsCO_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsNOx_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsNMVOCs_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsCO2_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsCH4_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsN2O_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsNH3_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsSO2_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsPM25_CBox.clicked.connect(self.saveTabOnProject)
self.gui.pollutantsC6H6_CBox.clicked.connect(self.saveTabOnProject)
# if and output file is specified and it's available => load it
if newOutputLayer and os.path.exists(newOutputLayer):
# check if layer is already loaded in layer list checking it's source
found = False
for layerName, layer in QgsMapLayerRegistry.instance().mapLayers().items():
if newOutputLayer == layer.publicSource():
# set the found layer as roadLayer
self.outLayer = layer
self.outLayerId = self.outLayer.id()
found = True
break
if not found:
# get layer name to set as public name in the legend
layerName = os.path.splitext( os.path.basename(newOutputLayer) )[0]
# load layer
self.outLayer = QgsVectorLayer(newOutputLayer, layerName, 'ogr')
if not self.outLayer.isValid():
message = self.tr("Error loading layer: %s" % self.outLayer.error().message(QgsErrorMessage.Text))
iface.messageBar().pushMessage(message, QgsMessageBar.CRITICAL)
return
self.outLayerId = self.outLayer.id()
# show layer in the canvas
QgsMapLayerRegistry.instance().addMapLayer(self.outLayer)
def saveTabOnProject(self):
''' Save tab configuration in the project basing on GUI values
'''
# get values from the GUI
addToInputLayer = self.gui.addToOriginaLayer_RButton.isChecked()
newOutputLayer = self.gui.outFile_LEdit.text()
Gasoline_Consumption = self.gui.fuelEnergyConsumptionGasoline_CBox.isChecked()
Diesel_Consumption = self.gui.fuelEnergyConsumptionDiesel_CBox.isChecked()
LPG_Consumption = self.gui.fuelEnergyConsumptionLPG_CBox.isChecked()
NewFuel_Consumption = self.gui.fuelEnergyConsumptionNewFuels_CBox.isChecked()
Fuel_Consumption_Total = self.gui.totalFuelConsumption_CBox.isChecked()
Energy_Consumption = self.gui.energyConsumption_CBox.isChecked()
CO = self.gui.pollutantsCO_CBox.isChecked()
NOx = self.gui.pollutantsNOx_CBox.isChecked()
NMVOC = self.gui.pollutantsNMVOCs_CBox.isChecked()
CO2 = self.gui.pollutantsCO2_CBox.isChecked()
CH4 = self.gui.pollutantsCH4_CBox.isChecked()
N2O = self.gui.pollutantsN2O_CBox.isChecked()
NH3 = self.gui.pollutantsNH3_CBox.isChecked()
SO2 = self.gui.pollutantsSO2_CBox.isChecked()
PM25 = self.gui.pollutantsPM25_CBox.isChecked()
C6H6 = self.gui.pollutantsC6H6_CBox.isChecked()
# set conf parameters
self.project.setValue('Processing.OutputFileDefinition/addToInputLayer', int(addToInputLayer))
self.project.setValue('Processing.OutputFileDefinition/newOutputLayer', newOutputLayer)
self.project.setValue('Processing.Parameters/Gasoline_Consumption', int(Gasoline_Consumption))
self.project.setValue('Processing.Parameters/Diesel_Consumption', int(Diesel_Consumption))
self.project.setValue('Processing.Parameters/LPG_Consumption', int(LPG_Consumption))
self.project.setValue('Processing.Parameters/NewFuel_Consumption', int(NewFuel_Consumption))
self.project.setValue('Processing.Parameters/Fuel_Consumption_Total', int(Fuel_Consumption_Total))
self.project.setValue('Processing.Parameters/Energy_Consumption', int(Energy_Consumption))
self.project.setValue('Processing.Parameters/CO', int(CO))
self.project.setValue('Processing.Parameters/NOx', int(NOx))
self.project.setValue('Processing.Parameters/NMVOC', int(NMVOC))
self.project.setValue('Processing.Parameters/CO2', int(CO2))
self.project.setValue('Processing.Parameters/CH4', int(CH4))
self.project.setValue('Processing.Parameters/N2O', int(N2O))
self.project.setValue('Processing.Parameters/NH3', int(NH3))
self.project.setValue('Processing.Parameters/SO2', int(SO2))
self.project.setValue('Processing.Parameters/PM', int(PM25))
self.project.setValue('Processing.Parameters/C6H6', int(C6H6))
# notify project modification
self.projectModified.emit()
def calculate(self):
''' Prepare environment to run the alg and run it. After run, merge produced
data basing on plugin configuration.
Before calculation a parametere validation will be executed
'''
# perform validation
if not self.gui.validate():
return
else:
# notify successful validation
message = self.tr("QTraffic: Parameters validation passed successfully")
iface.messageBar().pushMessage(message, QgsMessageBar.SUCCESS)
# set number of classes in the project config (that is the temporary one... but equal to the official one)
fleetDistributionRoadTypes = self.gui.getRoadTypes()
self.project.setValue('Processing.Parameters/maximum_type', len(fleetDistributionRoadTypes))
self.project.sync()
# create the algorithm
self.alg = Algorithm()
roadLayer = self.gui.getRoadLayer()
# prepare layer where to add result
addToInputLayer = self.gui.addToOriginaLayer_RButton.isChecked()
newOutputLayer = self.gui.outFile_LEdit.text()
if addToInputLayer:
self.outLayer = roadLayer
self.outLayerId = self.outLayer.id()
else:
# if layer is present... remove it
# out layer would not be the same of input road layer... in thi scase don't remove it
if self.outLayer and self.outLayer.isValid():
# to be sure, remove only if roadLayer and outLayer are different
if self.outLayer.publicSource() != roadLayer.publicSource():
self.outLayerRemoved = False
QgsMapLayerRegistry.instance().layerRemoved.connect(self.checkOutLayerRemoved)
QgsMapLayerRegistry.instance().removeMapLayer(self.outLayer.id())
# remove file when it has been removed from qgis
while not self.outLayerRemoved:
sleep(0.1)
QgsMapLayerRegistry.instance().layerRemoved.disconnect(self.checkOutLayerRemoved)
# reinit outLayer variables
# If not, under windws remain a locking of the related file creating
# an error during QgsVectorFileWriter.deleteShapeFile
self.outLayer = None
self.outLayerId = None
if os.path.exists(newOutputLayer):
if not QgsVectorFileWriter.deleteShapeFile(newOutputLayer):
message = self.tr("Error removing shape: {}".format(newOutputLayer))
iface.messageBar().pushMessage(message, QgsMessageBar.CRITICAL)
return
# copy input layer to the new one
writeError = QgsVectorFileWriter.writeAsVectorFormat(roadLayer, newOutputLayer, 'utf-8', roadLayer.crs())
if writeError != QgsVectorFileWriter.NoError:
message = self.tr('Error writing vector file {}'.format(newOutputLayer))
QgsMessageLog.logMessage(message, 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
# load the layer
newLayerName =os.path.splitext(os.path.basename( newOutputLayer ))[0]
self.outLayer = QgsVectorLayer(newOutputLayer, newLayerName, 'ogr')
if not self.outLayer.isValid():
message = self.tr('Error loading vector file {}'.format(newOutputLayer))
QgsMessageLog.logMessage(message, 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
self.outLayerId = self.outLayer.id()
# prepare environment
try:
self.alg.setProject(self.project)
self.alg.setLayer( roadLayer )
self.alg.initConfig()
self.alg.prepareRun()
except Exception as ex:
traceback.print_exc()
message = self.tr('Error preparing running contex for the algoritm: %s' % str(ex))
QgsMessageLog.logMessage(message, 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', message)
return
# run the self.alg
self.thread = QtCore.QThread(self)
self.thread.started.connect(self.alg.run)
self.thread.finished.connect(self.threadCleanup)
self.thread.terminated.connect(self.threadCleanup)
self.alg.moveToThread(self.thread)
self.alg.started.connect(self.manageStarted)
self.alg.progress.connect(self.manageProgress)
self.alg.message.connect(self.manageMessage)
self.alg.error.connect(self.manageError)
self.alg.finished.connect(self.manageFinished)
# set wait cursor and start
QgsApplication.instance().setOverrideCursor(QtCore.Qt.WaitCursor)
self.thread.start()
def checkOutLayerRemoved(self, layerId):
''' check when outLayer has been removed settign semaphore to True
'''
if self.outLayerId == layerId:
QgsMessageLog.logMessage('Completly removed outLayer', 'QTraffic', QgsMessageLog.INFO)
self.outLayerRemoved = True
def threadCleanup(self):
''' cleanup after thread run
'''
# restore cursor
QgsApplication.instance().restoreOverrideCursor()
if self.alg:
self.alg.deleteLater()
self.alg = None
self.thread.wait()
self.thread.deleteLater()
# remove progress bar
if self.progressMessageBarItem:
iface.messageBar().popWidget(self.progressMessageBarItem)
self.progressMessageBarItem = None
def manageStarted(self):
''' Setup GUI env to notify processing progress
'''
message = self.tr('Processing started')
QgsMessageLog.logMessage(message, 'QTraffic', QgsMessageLog.INFO)
# create message bar to show progress
self.progressMessageBarItem = iface.messageBar().createMessage(self.tr('Executing alg'))
self.progress = QtGui.QProgressBar()
self.progress.setMaximum(0)
self.progress.setMinimum(0)
self.progress.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter)
self.progressMessageBarItem.layout().addWidget(self.progress)
iface.messageBar().pushWidget(self.progressMessageBarItem, QgsMessageBar.INFO)
def manageProgress(self, percentage):
''' Update progress bar basing on alg nofify
'''
self.progress.setValue(percentage)
def manageMessage(self, message, msgType):
''' expose alg thread messages to gui and log
'''
QgsMessageLog.logMessage(message, 'QTraffic', msgType)
duration = 0
if msgType == QgsMessageLog.INFO:
return
if msgType == QgsMessageLog.CRITICAL:
msgType = QgsMessageBar.CRITICAL
if msgType == QgsMessageLog.WARNING:
duration = 3
msgType = QgsMessageBar.WARNING
iface.messageBar().pushMessage(message, msgType, duration)
def manageError(self, ex, exceptionMessage):
''' Do actions in case of alg thread error. Now only notify exception
'''
QgsMessageLog.logMessage(exceptionMessage, 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushMessage(exceptionMessage, QgsMessageBar.CRITICAL)
def manageFinished(self, success, reason):
''' Do action after notify that alg is finished. These are the postprocessing steps
1) merge result to the output layer
2) add the layer to canvas in case it is new
3) notify edn of processing
4) terminate the thread
'''
# finish the thread
self.thread.quit()
# check result
if not success:
QgsMessageLog.logMessage('Failed execution: {}'.format(reason), 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', self.tr("Error executing the algorithm: {}".format(reason)))
return
# prepare result
try:
self.alg.addResultToLayer(self.outLayer)
except Exception as ex:
traceback.print_exc()
QgsMessageLog.logMessage('Cannot add result to layer: {}'.format(str(ex)), 'QTraffic', QgsMessageLog.CRITICAL)
iface.messageBar().pushCritical('QTraffic', self.tr("Cannot add result to layer"))
return
# add or refresh rsult vector layer
addToInputLayer = self.gui.addToOriginaLayer_RButton.isChecked()
if not addToInputLayer:
QgsMapLayerRegistry.instance().addMapLayer(self.outLayer)
iface.mapCanvas().refresh()
# notify the user the end of process
iface.messageBar().pushSuccess('QTraffic', self.tr('Alg terminated successfully'))
def validate(self):
''' pre calcluation validation related only to this tab
Mandatory:
At least an output parameters have to be set
if "create new layer" a new layer name have to be set
'''
addToInputLayer = self.gui.addToOriginaLayer_RButton.isChecked()
newOutputLayer = self.gui.outFile_LEdit.text()
if (not addToInputLayer and
not newOutputLayer):
message = self.tr('No output vector specified')
iface.messageBar().pushMessage(message, QgsMessageBar.CRITICAL)
return False
Gasoline_Consumption = self.gui.fuelEnergyConsumptionGasoline_CBox.isChecked()
Diesel_Consumption = self.gui.fuelEnergyConsumptionDiesel_CBox.isChecked()
LPG_Consumption = self.gui.fuelEnergyConsumptionLPG_CBox.isChecked()
NewFuel_Consumption = self.gui.fuelEnergyConsumptionNewFuels_CBox.isChecked()
Fuel_Consumption_Total = self.gui.totalFuelConsumption_CBox.isChecked()
Energy_Consumption = self.gui.energyConsumption_CBox.isChecked()
CO = self.gui.pollutantsCO_CBox.isChecked()
NOx = self.gui.pollutantsNOx_CBox.isChecked()
NMVOC = self.gui.pollutantsNMVOCs_CBox.isChecked()
CO2 = self.gui.pollutantsCO2_CBox.isChecked()
CH4 = self.gui.pollutantsCH4_CBox.isChecked()
N2O = self.gui.pollutantsN2O_CBox.isChecked()
NH3 = self.gui.pollutantsNH3_CBox.isChecked()
SO2 = self.gui.pollutantsSO2_CBox.isChecked()
PM25 = self.gui.pollutantsPM25_CBox.isChecked()
C6H6 = self.gui.pollutantsC6H6_CBox.isChecked()
if (not Gasoline_Consumption and
not Diesel_Consumption and
not LPG_Consumption and
not NewFuel_Consumption and
not Fuel_Consumption_Total and
not Energy_Consumption and
not CO and
not NOx and
not NMVOC and
not CO2 and
not CH4 and
not N2O and
not NH3 and
not SO2 and
not PM25 and
not C6H6 ):
message = self.tr("Validation error: At least an output parameter have to be selected")
iface.messageBar().pushMessage(message, QgsMessageBar.CRITICAL)
return False
return True
def tr(self, string, context=''):
if not context:
context = 'QTraffic'
return QtCore.QCoreApplication.translate(context, string)
