class SurveyingCalculation:
"""SurveyingCalculation QGIS Plugin Implementation."""
def __init__(self, iface):
"""Constructor.
:param iface: an interface instance that will be passed to this class which provides the hook by which you can manipulate the QGIS application at run time (QgsInterface)
"""
# Save reference to the QGIS interface
self.iface = iface
# initialize plugin directory
self.plugin_dir = QDir().cleanPath(QFileInfo(__file__).absolutePath())
# initialize locale
locale = QSettings().value('locale/userLocale')[0:2]
print locale
locale_path = QDir.cleanPath(self.plugin_dir + QDir.separator() +
'i18n' + QDir.separator() +
'{}.qm'.format(locale))
print locale_path
if QFileInfo(locale_path).exists():
print "exists"
self.translator = QTranslator()
self.translator.load(locale_path)
if qVersion() > '4.3.3':
QCoreApplication.installTranslator(self.translator)
# init result log
log_path_2 = QSettings().value("SurveyingCalculation/log_path",
config.log_path)
if len(log_path_2) > 0:
log_path = log_path_2
else:
log_path = QDir.cleanPath(self.plugin_dir + QDir.separator() +
'log' + QDir.separator() + 'log.txt')
self.log = ResultLog(log_path)
self.newp_dlg = NewPointDialog()
self.single_dlg = SingleDialog(self.log)
self.traverse_dlg = TraverseDialog(self.log)
self.network_dlg = NetworkDialog(self.log)
self.transformation_dlg = TransformationDialog(self.log)
self.plotbytemplate_dlg = BatchPlottingDialog(self.iface, False)
self.batchplotting_dlg = BatchPlottingDialog(self.iface, True)
# Declare instance attributes
# noinspection PyMethodMayBeStatic
#def tr(self, message):
# """Get the translation for a string using Qt translation API. We implement this ourselves since we do not inherit QObject.
# :param message: string for translation (str, QString)
# :returns: translated version of message (QString)
# """
# noinspection PyTypeChecker,PyArgumentList,PyCallByClass
#return QCoreApplication.translate('SurveyingCalculation', message)
# return tr(message)
def add_action(self,
icon_path,
text,
callback,
enabled_flag=True,
add_to_menu=True,
add_to_toolbar=True,
status_tip=None,
whats_this=None,
parent=None):
"""Add a toolbar icon to the InaSAFE toolbar.
:param icon_path: path to the icon for this action. Can be a resource path (e.g. ':/plugins/foo/bar.png') or a normal file system path (str)
:param text: text that should be shown in menu items for this action (str)
:param callback: function to be called when the action is triggered (function)
:param enabled_flag: a flag indicating if the action should be enabled by default (bool). Defaults to True.
:param add_to_menu: flag indicating whether the action should also be added to the menu (bool). Defaults to True.
:param add_to_toolbar: flag indicating whether the action should also be added to the toolbar (bool). Defaults to True.
:param status_tip: optional text to show in a popup when mouse pointer hovers over the action (str)
:param parent: parent widget for the new action (QWidget). Defaults None.
:param whats_this: optional text to show in the status bar when the mouse pointer hovers over the action (str)
:returns: the action that was created (Qaction). Note that the action is also added to self.actions list.
"""
icon = QIcon(icon_path)
action = QAction(icon, text, parent)
action.triggered.connect(callback)
action.setEnabled(enabled_flag)
if status_tip is not None:
action.setStatusTip(status_tip)
if whats_this is not None:
action.setWhatsThis(whats_this)
if add_to_toolbar:
self.toolbar.addAction(action)
if add_to_menu:
self.iface.addPluginToMenu(self.menu, action)
self.actions.append(action)
return action
def initGui(self):
"""Create the menu entries and toolbar icons inside the QGIS GUI."""
icon_path = ':/plugins/SurveyingCalculation/icon.png'
icon_dir = QDir.cleanPath(self.plugin_dir + QDir.separator() + 'icons')
# build menu
self.actions = []
self.menu = QMenu()
self.menu.setTitle(tr(u'&SurveyingCalculation'))
self.sc_coord = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('new_coord.png')),
tr("New coordinate list ..."), self.iface.mainWindow())
self.sc_fb = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('new_fb.png')),
tr("New fieldbook ..."), self.iface.mainWindow())
self.sc_load = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('import_fieldbook.png')),
tr("Import fieldbook ..."), self.iface.mainWindow())
self.sc_addp = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('addp.png')),
tr("Add new point ..."), self.iface.mainWindow())
self.sc_calc = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('single_calc.png')),
tr("Single point calculations ..."), self.iface.mainWindow())
self.sc_trav = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('traverse_calc.png')),
tr("Traverse calculations ..."), self.iface.mainWindow())
self.sc_netw = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('network_calc.png')),
tr("Network adjustment ..."), self.iface.mainWindow())
self.sc_tran = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('coord_calc.png')),
tr("Coordinate transformation ..."), self.iface.mainWindow())
self.sc_pdiv = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('poly_div.png')),
tr("Polygon division ..."), self.iface.mainWindow())
self.sc_plot = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('plot.png')),
tr("Plot by template ..."), self.iface.mainWindow())
self.sc_batchplot = QAction(
QIcon(QDir(icon_dir).absoluteFilePath('batch_plot.png')),
tr("Batch plotting ..."), self.iface.mainWindow())
self.sc_settings = QAction(tr("Settings ..."), self.iface.mainWindow())
self.sc_help = QAction(tr("Help"), self.iface.mainWindow())
self.sc_about = QAction(tr("About"), self.iface.mainWindow())
self.menu.addActions([
self.sc_coord, self.sc_fb, self.sc_load, self.sc_addp,
self.sc_calc, self.sc_trav, self.sc_netw, self.sc_tran,
self.sc_plot, self.sc_batchplot, self.sc_settings, self.sc_help,
self.sc_about
])
self.menu.insertSeparator(self.sc_calc)
self.menu.insertSeparator(self.sc_plot)
self.menu.insertSeparator(self.sc_settings)
self.menu.insertSeparator(self.sc_help)
menu_bar = self.iface.mainWindow().menuBar()
actions = menu_bar.actions()
lastAction = actions[len(actions) - 1]
menu_bar.insertMenu(lastAction, self.menu)
self.sc_coord.triggered.connect(self.create_coordlist)
self.sc_fb.triggered.connect(self.create_fb)
self.sc_load.triggered.connect(self.load_fieldbook)
self.sc_addp.triggered.connect(self.addp)
self.sc_calc.triggered.connect(self.calculations)
self.sc_trav.triggered.connect(self.traverses)
self.sc_netw.triggered.connect(self.networks)
self.sc_tran.triggered.connect(self.transformation)
self.sc_pdiv.setCheckable(True)
self.tool_pdiv = LineMapTool(self.iface)
self.tool_pdiv.setAction(self.sc_pdiv)
self.sc_pdiv.triggered.connect(self.polygon_division)
self.sc_plot.triggered.connect(self.plot_by_temp)
self.sc_batchplot.triggered.connect(self.batch_plotting)
self.sc_settings.triggered.connect(self.settings)
self.sc_about.triggered.connect(self.about)
self.sc_help.triggered.connect(self.help)
# add icons to toolbar
self.toolbar = self.iface.addToolBar(u'SurveyingCalculation')
self.toolbar.setObjectName(u'SurveyingCalculation')
self.toolbar.addActions([
self.sc_load, self.sc_addp, self.sc_calc, self.sc_trav,
self.sc_netw, self.sc_tran, self.sc_pdiv, self.sc_plot,
self.sc_batchplot
])
self.toolbar.insertSeparator(self.sc_calc)
self.toolbar.insertSeparator(self.sc_plot)
def unload(self):
""" Removes the plugin menu item and icon from QGIS GUI.
"""
for action in self.actions:
self.iface.removePluginMenu(tr(u'&SurveyingCalculation'), action)
self.iface.removeToolBarIcon(action)
del self.menu
del self.toolbar
def create_coordlist(self):
""" Create a new coordinate list from template and add to layer list. Layer/file name changed to start with 'coord\_' if neccessary.
"""
ofname = QFileDialog.getSaveFileName(self.iface.mainWindow(),
tr('QGIS co-ordinate list'),
filter=tr('Shape file (*.shp)'))
if not ofname:
return
if QRegExp('coord_').indexIn(QFileInfo(ofname).baseName()):
ofname = QDir.cleanPath(
QFileInfo(ofname).absolutePath() + QDir().separator() +
'coord_' + QFileInfo(ofname).fileName())
ofbase = QDir.cleanPath(
QFileInfo(ofname).absolutePath() + QDir().separator() +
QFileInfo(ofname).baseName())
tempbase = QDir.cleanPath(self.plugin_dir + QDir().separator() +
'template' + QDir().separator() +
'coord_template')
for ext in ['.shp', '.shx', '.dbf']:
QFile(tempbase + ext).copy(ofbase + ext)
coord = QgsVectorLayer(ofbase + '.shp',
QFileInfo(ofbase).baseName(), "ogr")
if coord.isValid():
QgsMapLayerRegistry.instance().addMapLayer(coord)
def create_fb(self):
""" Create a new empty fieldbook from template and add to layer list. Layer/file name changed to start with 'fb\_' if neccessary.
"""
ofname = QFileDialog.getSaveFileName(
self.iface.mainWindow(),
tr('New fieldbook'),
filter=tr('Fieldbook file (*.dbf)'))
if not ofname:
return
if QRegExp('fb_').indexIn(QFileInfo(ofname).baseName()):
ofname = QDir.cleanPath(
QFileInfo(ofname).absolutePath() + QDir().separator() + 'fb_' +
QFileInfo(ofname).fileName())
ofbase = QDir.cleanPath(
QFileInfo(ofname).absolutePath() + QDir().separator() +
QFileInfo(ofname).baseName())
tempbase = QDir.cleanPath(self.plugin_dir + QDir().separator() +
'template' + QDir().separator() +
'fb_template')
for ext in ['.dbf']:
QFile(tempbase + ext).copy(ofbase + ext)
fb = QgsVectorLayer(ofbase + '.dbf',
QFileInfo(ofbase).baseName(), "ogr")
if fb.isValid():
QgsMapLayerRegistry.instance().addMapLayer(fb)
def load_fieldbook(self):
""" Import an electric fieldbook from file (GSI, JOB/ARE, ...)
"""
if get_coordlist() is None:
QMessageBox.warning(
self.iface.mainWindow(), tr("Warning"),
tr("No coordinate list is opened, coordinates will be lost from the fieldbook"
))
homedir = QSettings().value("SurveyingCalculation/homedir",
config.homedir)
fname = QFileDialog.getOpenFileName(self.iface.mainWindow(), \
tr('Electric fieldbook'), homedir, \
filter = tr('Leica GSI (*.gsi);;Geodimeter JOB/ARE (*.job *.are);;Sokkia CRD (*.crd);;SurvCE RW5 (*.rw5);;STONEX DAT (*.dat)'))
if fname:
# file selected
# make a copy of dbf template if not are is loaded
if QRegExp('\.are$', Qt.CaseInsensitive).indexIn(fname) == -1:
# ask for table name
ofname = QFileDialog.getSaveFileName(
self.iface.mainWindow(),
tr('QGIS fieldbook'),
QFileInfo(fname).absolutePath(),
filter=tr('DBF file (*.dbf)'))
if not ofname:
return
# remember last input dir
QSettings().setValue("SurveyingCalculation/homedir",
QFileInfo(fname).absolutePath())
QSettings().sync()
if QRegExp('fb_').indexIn(QFileInfo(ofname).baseName()):
ofname = QDir.cleanPath(
QFileInfo(ofname).absolutePath() + QDir().separator() +
'fb_' + QFileInfo(ofname).fileName())
tempname = QDir.cleanPath(self.plugin_dir +
QDir().separator() + 'template' +
QDir().separator() +
'fb_template.dbf')
QFile(tempname).copy(ofname)
fb_dbf = QgsVectorLayer(ofname,
QFileInfo(ofname).baseName(), "ogr")
QgsMapLayerRegistry.instance().addMapLayer(fb_dbf)
if QRegExp('\.gsi$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = LeicaGsi(fname)
elif QRegExp('\.job$', Qt.CaseInsensitive).indexIn(fname) > -1 or \
QRegExp('\.are$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = JobAre(fname)
elif QRegExp('\.crd$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = Sdr(fname)
elif QRegExp('\.rw5$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = SurvCE(fname)
elif QRegExp('\.dat$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = Stonex(fname)
else:
QMessageBox.warning(self.iface.mainWindow(),
tr('File warning'),
tr('Unknown fieldbook type'), tr('OK'))
return
i = 10 # ordinal number for fieldbook records
#fb_dbf.startEditing()
fb.open()
n_fb = 0 # fieldbook records stored
n_co = 0 # points stored in coordinate list
while True:
# get next observation/station data from fieldbook
r = fb.parse_next()
if r is None:
break # end of file
if 'station' in r:
# add row to fieldbook table
record = QgsFeature()
# add & initialize attributes
record.setFields(fb_dbf.pendingFields(), True)
j = fb_dbf.dataProvider().fieldNameIndex('id')
if j != -1:
record.setAttribute(j, i)
for key in r:
j = fb_dbf.dataProvider().fieldNameIndex(key)
if j != -1:
record.setAttribute(j, r[key])
fb_dbf.dataProvider().addFeatures([record])
n_fb += 1
if 'station_e' in r or 'station_z' in r:
# store station coordinates too
dimension = 0
if 'station_z' in r:
dimension += 1
else:
r['station_z'] = None
if 'station_e' in r and 'station_n' in r:
dimension += 2
else:
r['station_e'] = None
r['station_n'] = None
if not 'pc' in r:
r['pc'] = None
p = Point(r['point_id'], r['station_e'], r['station_n'],
r['station_z'], r['pc'])
qp = ScPoint(p)
qp.store_coord(dimension)
n_co += 1
if 'e' in r or 'z' in r:
# store coordinates too
dimension = 0
if 'z' in r:
dimension += 1
else:
r['z'] = None
if 'e' in r and 'n' in r:
dimension += 2
else:
r['e'] = None
r['n'] = None
if not 'pc' in r:
r['pc'] = None
p = Point(r['point_id'], r['e'], r['n'], r['z'], r['pc'])
qp = ScPoint(p)
qp.store_coord(dimension)
n_co += 1
i += 10
#fb_dbf.commitChanges()
if QRegExp('\.are$', Qt.CaseInsensitive).indexIn(fname) == -1:
if n_fb == 0: # no observations
QgsMapLayerRegistry.instance().removeMapLayer(fb_dbf.id())
# remove empty file
QFile(ofname).remove()
if n_co == 0: # no coordinates
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"),\
tr("Neither coordinates nor observations found"))
else:
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"),\
tr("No observations found"))
self.log.write()
self.log.write_log(tr("Fieldbook loaded: ") + fname)
self.log.write(" %d observations, %d coordinates" %
(n_fb, n_co))
return
def addp(self):
""" Add point(s) to coordinate list entering coordinates
"""
if get_coordlist() is None:
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"),
tr("A coordinate list must be opened!"))
return
# show the dialog
self.newp_dlg.show()
self.newp_dlg.activateWindow()
# Run the dialog event loop
result = self.newp_dlg.exec_()
def calculations(self):
""" Single point calculations (orientation, intersection,
resection, freestation)
"""
# show the dialog
self.single_dlg.show()
self.single_dlg.activateWindow()
# Run the dialog event loop
result = self.single_dlg.exec_()
def traverses(self):
""" Various traverse claculations
"""
# show the dialog
self.traverse_dlg.show()
self.traverse_dlg.activateWindow()
# Run the dialog event loop
result = self.traverse_dlg.exec_()
def networks(self):
""" Various network adjustments (1D/2D/3D)
"""
# show the dialog
self.network_dlg.show()
self.network_dlg.activateWindow()
# Run the dialog event loop
result = self.network_dlg.exec_()
def transformation(self):
""" Various coordinate transformations (orthogonal, affine, polynomial)
"""
# show the dialog
self.transformation_dlg.show()
self.transformation_dlg.activateWindow()
# Run the dialog event loop
result = self.transformation_dlg.exec_()
def polygon_division(self):
""" accept a line from the user to divide the selected polygon on
active layer
"""
al = self.iface.activeLayer()
if al is None or al.type() != QgsMapLayer.VectorLayer or \
al.geometryType() != QGis.Polygon:
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"),
tr("Actual layer contains no polygons"))
return
if len(al.selectedFeatures()) != 1:
QMessageBox.warning(
self.iface.mainWindow(), tr("Warning"),
tr("Not a single polygon is selected in active layer"))
return
self.iface.mapCanvas().setMapTool(self.tool_pdiv)
def plot_by_temp(self):
# show the dialog
self.plotbytemplate_dlg.show()
self.plotbytemplate_dlg.activateWindow()
# Run the dialog event loop
result = self.plotbytemplate_dlg.exec_()
def batch_plotting(self):
""" Batch plots selected geometry items using the selected template and scale.
"""
#check if there are polygon layers in the project
polygon_layers = get_vector_layers_by_type(QGis.Polygon)
if polygon_layers is None:
QMessageBox.warning(
self.iface.mainWindow(), tr("Warning"),
tr("This utility needs at least one polygon type layer!"))
return
# show the dialog
self.batchplotting_dlg.show()
self.batchplotting_dlg.activateWindow()
# Run the dialog event loop
result = self.batchplotting_dlg.exec_()
def settings(self):
""" Setting of the plugin.
"""
settings_dlg = PluginSettingsDialog()
result = settings_dlg.exec_()
if result == QDialog.Accepted:
log_path = QSettings().value("SurveyingCalculation/log_path",
config.log_path)
self.log.set_log_path(log_path)
def about(self):
""" About box of the plugin
"""
QMessageBox.information(
self.iface.mainWindow(), tr('About'),
tr('Surveying Calculation Plugin\n\n (c) DigiKom Ltd 2014- http://digikom.hu, mail (at) digikom.hu\nVersion 0.2'
))
def help(self):
""" Open user's guide of the plugin in the default web browser.
"""
webbrowser.open(
"http://www.digikom.hu/SurveyingCalculation/usersguide.html")
class MenuManager(QWidget, Ui_nodeActions):
def __init__(self, selection, listmodel):
super(QWidget, self).__init__()
self.__iconView = False
self.setupUi(self)
self.processusManager = ProcessusManager()
self.loader = loader.loader()
self.lmodules = self.loader.modules
self.taskmanager = TaskManager()
self.mainwindow = QApplication.instance().mainWindow
self.createActions()
self.checkedSelection = selection
self.selection = None
self.model = listmodel
self.bookManager = BookmarkManager(self.model)
#self.document = QtWebKit.QWebView()
#self.document.loadFinished.connect(self.__print)
self.__corrupt = base64.b64encode(
str(QtGui.QImage(":file_broken.png").bits()))
self.__printer = QtGui.QPrinter(QtGui.QPrinter.ScreenResolution)
self.__printer.setOutputFormat(QtGui.QPrinter.PdfFormat)
self.__printer.setPaperSize(QtGui.QPrinter.A4)
self.__printer.setFullPage(True)
def setupUi(self, nodeActions):
self.actionScan = QAction(self)
icon = QIcon()
icon.addPixmap(QPixmap(QString.fromUtf8(":/scan")), QIcon.Normal,
QIcon.On)
self.actionScan.setIcon(icon)
self.actionScan.setObjectName(QString.fromUtf8("actionScan"))
self.actionReport_node = QAction(self)
icon = QIcon()
icon.addPixmap(QPixmap(QString.fromUtf8(":/report")), QIcon.Normal,
QIcon.On)
self.actionReport_node.setIcon(icon)
self.actionReport_node.setObjectName(
QString.fromUtf8("actionReport_Node"))
Ui_nodeActions.setupUi(self, nodeActions)
def retranslateUi(self, nodeActions):
Ui_nodeActions.retranslateUi(self, nodeActions)
self.actionScan.setText(
QApplication.translate("nodeActions", "Scan", None,
QApplication.UnicodeUTF8))
self.actionScan.setToolTip(
QApplication.translate("nodeActions", "Launch recursive scan",
None, QApplication.UnicodeUTF8))
self.actionReport_node.setText(
QApplication.translate("nodeActions", "Report", None,
QApplication.UnicodeUTF8))
self.actionReport_node.setToolTip(
QApplication.translate("nodeActions", "Tag nodes", None,
QApplication.UnicodeUTF8))
def setIconView(self, enable):
self.__iconView = enable
def createActions(self):
self.extractor = Extractor(self.mainwindow)
self.connect(self.extractor, SIGNAL("filled"), self.launchExtract)
self.actionOpen.setParent(self.mainwindow)
self.actionOpen_in_new_tab.setParent(self.mainwindow)
self.copyToHtmlTable = QAction(self.tr("Export selection to pdf"),
self)
self.copyToHtmlTable.triggered.connect(self.__exportToPdf)
self.connect(self.actionOpen, SIGNAL("triggered()"), self.openDefault)
self.connect(self.actionOpen_in_new_tab, SIGNAL("triggered()"),
self.openAsNewTab)
self.connect(self.actionOpen_parent_folder, SIGNAL("triggered()"),
self.openParentFolder)
self.connect(self.actionHex_viewer, SIGNAL("triggered()"),
self.launchHexedit)
self.connect(self.actionExtract, SIGNAL("triggered()"),
self.extractNodes)
self.connect(self.actionBookmark, SIGNAL("triggered()"), self.bookmark)
self.connect(self.actionScan, SIGNAL('triggered()'), self.scan)
self.connect(self.actionReport_node, SIGNAL('triggered()'),
self.reportNode)
def createMenu(self):
nodeclicked = self.model.currentNode()
self.mainmenu = QMenu(self.mainwindow)
self.selection = self.model.currentNode()
self.setOpenRelevant()
self.setOpenWith()
self.mainmenu.addAction(self.actionOpen)
self.mainmenu.addAction(self.actionOpen_with)
self.mainmenu.addAction(self.actionOpen_in_new_tab)
self.mainmenu.addAction(self.actionOpen_parent_folder)
if nodeclicked.isDir() or nodeclicked.hasChildren():
self.actionOpen_parent_folder.setVisible(False)
self.actionOpen_parent_folder.setEnabled(False)
self.actionOpen_in_new_tab.setVisible(True)
self.actionOpen_in_new_tab.setEnabled(True)
else:
self.actionOpen_in_new_tab.setVisible(False)
self.actionOpen_in_new_tab.setEnabled(False)
self.actionOpen_parent_folder.setVisible(True)
self.actionOpen_parent_folder.setEnabled(True)
self.mainmenu.addSeparator()
selection = selectionMenu(self, self.model)
self.mainmenu.addMenu(selection)
tags = tagMenu(self, self.mainwindow, self.model)
self.actionTags.setMenu(tags)
self.mainmenu.addAction(self.actionTags)
self.mainmenu.addAction(self.actionBookmark)
if nodeclicked.path().find('/Bookmarks/') != -1:
self.mainmenu.addAction(QIcon(":trash"),
self.tr("Delete bookmark"),
self.deleteBookmark)
self.bookseparator = self.mainmenu.addSeparator()
self.mainmenu.addAction(self.actionHex_viewer)
self.mainmenu.addAction(self.actionExtract)
if self.__iconView:
self.mainmenu.addAction(self.copyToHtmlTable)
self.mainmenu.popup(QCursor.pos())
self.mainmenu.insertSeparator(self.actionOpen_parent_folder)
self.mainmenu.insertAction(self.actionOpen_parent_folder,
self.actionScan)
self.mainmenu.addSeparator()
self.mainmenu.insertAction(self.bookseparator, self.actionReport_node)
self.mainmenu.show()
def reportNode(self):
ReportNodesAction(self.model)
def scan(self):
autoWiz = AutoWizard(self, root=self.model.currentNode())
autoWiz.exec_()
def deleteBookmark(self):
vfs = VFS.Get()
try:
vfs.unregister(self.model.currentNode())
except Exception as e:
print 'TreeMenu.deleteNode exceptions : ', str(e)
def setOpenRelevant(self):
if self.selection != None:
node = self.selection
modules = node.compatibleModules()
if len(modules):
relevant = QMenu()
for modname in modules:
module = self.loader.modules[modname]
relevant.addAction(
newAction(self, self.mainwindow, modname, module.tags,
module.icon))
self.actionOpen.setMenu(relevant)
def __exportToPdf(self):
pdfFile = QtGui.QFileDialog.getSaveFileName(
self, self.tr("Export to pdf file"), os.path.expanduser("~"),
self.tr("Pdf files (*.pdf)"))
if not pdfFile.endsWith(".pdf"):
pdfFile.append(".pdf")
self.__printer.setOutputFileName(pdfFile)
self.__createPdf()
def __createPdf(self):
html = """
<html>
<head>
<style type="text/css">
img {
max-width: 340px;
max-height: 340px;
width: expression(this.width > 340 ? "340px" : true);
height: expression(this.height > 340 ? "340px" : true);
}
.break { page-break-before: always; }
</style>
</head>
<body>
"""
start = """<table style="height: 100%; margin: 1px auto; border-spacing: 10px; border-collapse: separate;">"""
end = """</table><p class="break"></p>"""
count = 1
row = ""
for uid in self.checkedSelection._selection:
cell = self.__nodeToHtmlCell(uid)
if len(cell):
row += cell
if count == 1:
html += start
if count % 3 == 0:
html += "<tr>\n{}\n</tr>".format(row)
row = ""
if count == 9:
html += end
count = 0
row = ""
count += 1
if len(row):
html += "<tr>{}</tr></table>".format(row)
html += "</body></html>"
self.document.setHtml(html)
def __nodeToHtmlCell(self, uid):
imagecell = """<td style="text-align: center;"><img src="data:image/jpg;base64,{}"/><br />{} {} {}</td>"""
node = VFS.Get().getNodeById(uid)
timestamp = ""
device = ""
model = ""
make = ""
b64image = self.__corrupt
if node is not None:
dtype = node.dataType()
try:
if dtype.find("image") != -1:
vfile = node.open()
image = vfile.read()
vfile.close()
data = node.attributesByName("exif.Model",
ABSOLUTE_ATTR_NAME)
if len(data):
model = data[0].toString()
data = node.attributesByName("exif.Make",
ABSOLUTE_ATTR_NAME)
if len(data):
make = data[0].toString()
if len(model) or len(make):
device = "<br />{} {}".format(model, make)
data = node.attributesByName("exif.DateTimeDigitized",
ABSOLUTE_ATTR_NAME)
if len(data):
timestamp = "<br /> {}".format(data[0].toString())
b64image = base64.b64encode(image)
except:
pass
return imagecell.format(b64image, node.name(), device, timestamp)
return ""
def __print(self, ok):
self.document.print_(self.__printer)
def setOpenWith(self):
owmenu = QMenu()
setags = Utils.getSetTags()
selist = list(setags)
selist.sort()
owmenu.addAction(self.mainwindow.actionBrowse_modules)
owmenu.addSeparator()
for tags in selist:
if not tags == "builtins":
action = QAction(QString(tags), self.mainwindow)
menu = self.getMenuFromModuleTag(tags)
action.setMenu(menu)
owmenu.addAction(action)
self.actionOpen_with.setMenu(owmenu)
def getMenuFromModuleTag(self, tagname):
menu = QMenu()
modules = self.loader.modules
for mod in modules:
m = modules[mod]
try:
if m.tags == tagname:
menu.addAction(
newAction(self, self.mainwindow, mod, tagname, m.icon))
# actions.append(newAction(self, self.__mainWindow, mod, self.tags, m.icon))
except AttributeError, e:
pass
return menu
class Historize(QObject):
"""This class handles the initialization and calls of the menus"""
def __init__(self, iface):
QObject.__init__(self)
self.iface = iface
self.dbconn = DBConn(iface)
plugin_dir = os.path.dirname(__file__)
# initialize locale
locale = QSettings().value('locale/userLocale')[0:2]
locale_path = os.path.join(plugin_dir, 'i18n',
'Historize_{}.qm'.format(locale))
if os.path.exists(locale_path):
translator = QTranslator()
translator.load(locale_path)
QCoreApplication.installTranslator(translator)
if qVersion() > '4.3.3':
QCoreApplication.installTranslator(translator)
def initGui(self):
self.menu = QMenu()
self.menu.setTitle("Historize")
self.layerMenu = QMenu()
self.layerMenu.setTitle("Layer")
# Create menu actions
self.actionInitDB = QAction(self.tr(u"Initialize Database"),
self.iface.mainWindow())
self.actionInitLayer = QAction(self.tr(u"Initialize Layer"),
self.iface.mainWindow())
self.actionLayerUpdate = QAction(self.tr(u"Update Layer"),
self.iface.mainWindow())
self.actionLayerLoad = QAction(self.tr(u"Load Layer"),
self.iface.mainWindow())
self.actionAbout = QAction(self.tr(u"About"), self.iface.mainWindow())
# Connect menu actions
self.actionInitDB.triggered.connect(self.initialize_database)
self.actionInitLayer.triggered.connect(self.initialize_layer)
self.actionLayerLoad.triggered.connect(self.show_load_layer_dialog)
self.actionLayerUpdate.triggered.connect(self.show_update_layer_dialog)
self.actionAbout.triggered.connect(self.show_about_dialog)
self.iface.legendInterface().currentLayerChanged.connect(
self.enable_disable_gui)
# Add actions to menu
self.layerMenu.addActions([
self.actionInitLayer, self.actionLayerLoad, self.actionLayerUpdate
])
self.menu.addAction(self.actionInitDB)
self.menu.addMenu(self.layerMenu)
self.menu.addAction(self.actionAbout)
self.menu.insertSeparator(self.actionAbout)
menuBar = self.iface.mainWindow().menuBar()
menuBar.addMenu(self.menu)
# Disable unusable actions
self.actionInitDB.setEnabled(False)
self.actionInitLayer.setEnabled(False)
self.actionLayerUpdate.setEnabled(False)
self.actionLayerLoad.setEnabled(False)
def unload(self):
self.menu.deleteLater()
def initialize_database(self):
"""Use Database info from layer and run historisation.sql on it."""
selectedLayer = self.iface.activeLayer()
provider = selectedLayer.dataProvider()
if provider.name() != 'postgres':
QMessageBox.warning(
self.iface.mainWindow(), self.tr(u"Invalid Layer"),
self.tr(u"Layer must be provided by postgres!"))
return
uri = QgsDataSourceURI(provider.dataSourceUri())
conn = self.dbconn.connect_to_DB(uri)
cur = conn.cursor()
if conn is False:
return
result = QMessageBox.warning(
self.iface.mainWindow(), self.tr(u"Initialize Historisation"),
self.tr(u"Initialize historisation on this layers database?"),
QMessageBox.No | QMessageBox.Yes)
if result == QMessageBox.Yes:
sqlPath = os.path.dirname(
os.path.realpath(__file__)) + '/sql/historisierung.sql'
try:
# Ignore first three characters
# which invalidate the SQL command
cur.execute(open(sqlPath, "r").read())
conn.commit()
QMessageBox.warning(
self.iface.mainWindow(), self.tr(u"Success"),
self.tr(u"Database initialized successfully!"))
except psycopg2.Error as e:
conn.rollback()
QMessageBox.warning(
self.iface.mainWindow(), self.tr(u"Error"),
self.tr(u"Couldn't initialize Database.\n" + e.message))
conn.close()
self.enable_disable_gui(selectedLayer)
else:
return
def initialize_layer(self):
"""Use Layer info and run init() .sql query"""
selectedLayer = self.iface.activeLayer()
provider = selectedLayer.dataProvider()
uri = QgsDataSourceURI(provider.dataSourceUri())
conn = self.dbconn.connect_to_DB(uri)
if conn is False:
return
result = QMessageBox.warning(
self.iface.mainWindow(), self.tr(u"Initialize Layer"),
self.tr(u"Are you sure you wish to proceed?"),
QMessageBox.No | QMessageBox.Yes)
if result == QMessageBox.Yes:
# Get SQL vars
hasGeometry = selectedLayer.hasGeometryType()
schema = uri.schema()
table = uri.table()
execute = SQLExecute(self.iface, self.iface.mainWindow(), uri)
success, msg = execute.Init_hist_tabs(hasGeometry, schema, table)
if success:
QMessageBox.warning(
self.iface.mainWindow(), self.tr(u"Success"),
self.tr(u"Layer successfully initialized!"))
else:
QMessageBox.warning(self.iface.mainWindow(), self.tr(u"Error"),
self.tr(u"Initialization failed!\n" + msg))
self.enable_disable_gui(selectedLayer)
else:
return
def show_update_layer_dialog(self):
"""Open ImportUpdate dialog"""
self.updateDialog = ImportUpdateDialog(self.iface)
self.updateDialog.show()
def show_load_layer_dialog(self):
"""Open selectDate dialog"""
self.dateDialog = SelectDateDialog(self.iface)
self.dateDialog.show()
def show_about_dialog(self):
"""Show About dialog"""
self.aboutDialog = AboutDialog()
self.aboutDialog.show()
def enable_disable_gui(self, layer):
"""Enable/Disable menu options based on selected layer"""
self.actionInitDB.setEnabled(False)
self.layerMenu.setEnabled(False)
self.actionInitLayer.setEnabled(False)
self.actionLayerUpdate.setEnabled(False)
self.actionLayerLoad.setEnabled(False)
selectedLayer = self.iface.activeLayer()
if selectedLayer:
provider = layer.dataProvider()
if provider.name() == "postgres":
self.actionInitDB.setEnabled(True)
uri = QgsDataSourceURI(provider.dataSourceUri())
execute = SQLExecute(self.iface, self.iface.mainWindow(), uri)
historised = execute.check_if_historised(
uri.schema(),
self.iface.activeLayer().name())
db_initialized = execute.db_initialize_check(uri.schema())
if db_initialized:
self.actionInitDB.setEnabled(False)
self.layerMenu.setEnabled(True)
else:
self.layerMenu.setEnabled(False)
if historised:
self.actionLayerUpdate.setEnabled(True)
self.actionLayerLoad.setEnabled(True)
else:
self.actionInitLayer.setEnabled(True)
class SurveyingCalculation:
"""SurveyingCalculation QGIS Plugin Implementation."""
def __init__(self, iface):
"""Constructor.
:param iface: an interface instance that will be passed to this class which provides the hook by which you can manipulate the QGIS application at run time (QgsInterface)
"""
# Save reference to the QGIS interface
self.iface = iface
# initialize plugin directory
self.plugin_dir = QDir().cleanPath( QFileInfo(__file__).absolutePath() )
# initialize locale
locale = QSettings().value('locale/userLocale')[0:2]
locale_path = QDir.cleanPath(self.plugin_dir + QDir.separator() + 'i18n' +
QDir.separator() + '{}.qm'.format(locale))
print locale_path
if QFileInfo(locale_path).exists():
print "exists"
self.translator = QTranslator()
self.translator.load(locale_path)
if qVersion() > '4.3.3':
QCoreApplication.installTranslator(self.translator)
# init result log
log_path_2 = QSettings().value("SurveyingCalculation/log_path",config.log_path)
if len(log_path_2) > 0:
log_path = log_path_2
else:
log_path = QDir.cleanPath(self.plugin_dir + QDir.separator() + 'log' +
QDir.separator() + 'log.txt')
self.log = ResultLog(log_path)
self.newp_dlg = NewPointDialog()
self.single_dlg = SingleDialog(self.log)
self.traverse_dlg = TraverseDialog(self.log)
self.network_dlg = NetworkDialog(self.log)
self.transformation_dlg = TransformationDialog(self.log)
self.plotbytemplate_dlg = BatchPlottingDialog(self.iface, False)
self.batchplotting_dlg = BatchPlottingDialog(self.iface, True)
# Declare instance attributes
# noinspection PyMethodMayBeStatic
#def tr(self, message):
# """Get the translation for a string using Qt translation API. We implement this ourselves since we do not inherit QObject.
# :param message: string for translation (str, QString)
# :returns: translated version of message (QString)
# """
# noinspection PyTypeChecker,PyArgumentList,PyCallByClass
#return QCoreApplication.translate('SurveyingCalculation', message)
# return tr(message)
def add_action(self, icon_path, text, callback, enabled_flag=True,
add_to_menu=True, add_to_toolbar=True, status_tip=None,
whats_this=None, parent=None):
"""Add a toolbar icon to the InaSAFE toolbar.
:param icon_path: path to the icon for this action. Can be a resource path (e.g. ':/plugins/foo/bar.png') or a normal file system path (str)
:param text: text that should be shown in menu items for this action (str)
:param callback: function to be called when the action is triggered (function)
:param enabled_flag: a flag indicating if the action should be enabled by default (bool). Defaults to True.
:param add_to_menu: flag indicating whether the action should also be added to the menu (bool). Defaults to True.
:param add_to_toolbar: flag indicating whether the action should also be added to the toolbar (bool). Defaults to True.
:param status_tip: optional text to show in a popup when mouse pointer hovers over the action (str)
:param parent: parent widget for the new action (QWidget). Defaults None.
:param whats_this: optional text to show in the status bar when the mouse pointer hovers over the action (str)
:returns: the action that was created (Qaction). Note that the action is also added to self.actions list.
"""
icon = QIcon(icon_path)
action = QAction(icon, text, parent)
action.triggered.connect(callback)
action.setEnabled(enabled_flag)
if status_tip is not None:
action.setStatusTip(status_tip)
if whats_this is not None:
action.setWhatsThis(whats_this)
if add_to_toolbar:
self.toolbar.addAction(action)
if add_to_menu:
self.iface.addPluginToMenu(
self.menu,
action)
self.actions.append(action)
return action
def initGui(self):
"""Create the menu entries and toolbar icons inside the QGIS GUI."""
icon_path = ':/plugins/SurveyingCalculation/icon.png'
icon_dir = QDir.cleanPath( self.plugin_dir + QDir.separator() + 'icons')
# build menu
self.actions = []
self.menu = QMenu()
self.menu.setTitle(tr(u'&SurveyingCalculation'))
self.sc_coord = QAction(QIcon(QDir(icon_dir).absoluteFilePath('new_coord.png')), tr("New coordinate list ..."), self.iface.mainWindow())
self.sc_fb = QAction(QIcon(QDir(icon_dir).absoluteFilePath('new_fb.png')),tr("New fieldbook ..."), self.iface.mainWindow())
self.sc_load = QAction(QIcon(QDir(icon_dir).absoluteFilePath('import_fieldbook.png')), tr("Import fieldbook ..."), self.iface.mainWindow())
self.sc_addp = QAction(QIcon(QDir(icon_dir).absoluteFilePath('addp.png')), tr("Add new point ..."), self.iface.mainWindow())
self.sc_calc = QAction(QIcon(QDir(icon_dir).absoluteFilePath('single_calc.png')), tr("Single point calculations ..."), self.iface.mainWindow())
self.sc_trav = QAction(QIcon(QDir(icon_dir).absoluteFilePath('traverse_calc.png')), tr("Traverse calculations ..."), self.iface.mainWindow())
self.sc_netw = QAction(QIcon(QDir(icon_dir).absoluteFilePath('network_calc.png')), tr("Network adjustment ..."), self.iface.mainWindow())
self.sc_tran = QAction(QIcon(QDir(icon_dir).absoluteFilePath('coord_calc.png')), tr("Coordinate transformation ..."), self.iface.mainWindow())
self.sc_pdiv = QAction(QIcon(QDir(icon_dir).absoluteFilePath('poly_div.png')), tr("Polygon division ..."), self.iface.mainWindow())
self.sc_plot = QAction(QIcon(QDir(icon_dir).absoluteFilePath('plot.png')), tr("Plot by template ..."), self.iface.mainWindow())
self.sc_batchplot = QAction(QIcon(QDir(icon_dir).absoluteFilePath('batch_plot.png')), tr("Batch plotting ..."), self.iface.mainWindow())
self.sc_settings = QAction(tr("Settings ..."), self.iface.mainWindow())
self.sc_help = QAction(tr("Help"), self.iface.mainWindow())
self.sc_about = QAction(tr("About"), self.iface.mainWindow())
self.menu.addActions([self.sc_coord, self.sc_fb, self.sc_load,
self.sc_addp, self.sc_calc, self.sc_trav, self.sc_netw,
self.sc_tran, self.sc_plot, self.sc_batchplot, self.sc_settings,
self.sc_help, self.sc_about])
self.menu.insertSeparator(self.sc_calc)
self.menu.insertSeparator(self.sc_plot)
self.menu.insertSeparator(self.sc_settings)
self.menu.insertSeparator(self.sc_help)
menu_bar = self.iface.mainWindow().menuBar()
actions = menu_bar.actions()
lastAction = actions[len(actions) - 1]
menu_bar.insertMenu(lastAction, self.menu)
self.sc_coord.triggered.connect(self.create_coordlist)
self.sc_fb.triggered.connect(self.create_fb)
self.sc_load.triggered.connect(self.load_fieldbook)
self.sc_addp.triggered.connect(self.addp)
self.sc_calc.triggered.connect(self.calculations)
self.sc_trav.triggered.connect(self.traverses)
self.sc_netw.triggered.connect(self.networks)
self.sc_tran.triggered.connect(self.transformation)
self.sc_pdiv.setCheckable(True)
self.tool_pdiv = LineMapTool(self.iface)
self.tool_pdiv.setAction(self.sc_pdiv)
self.sc_pdiv.triggered.connect(self.polygon_division)
self.sc_plot.triggered.connect(self.plot_by_temp)
self.sc_batchplot.triggered.connect(self.batch_plotting)
self.sc_settings.triggered.connect(self.settings)
self.sc_about.triggered.connect(self.about)
self.sc_help.triggered.connect(self.help)
# add icons to toolbar
self.toolbar = self.iface.addToolBar(u'SurveyingCalculation')
self.toolbar.setObjectName(u'SurveyingCalculation')
self.toolbar.addActions([self.sc_load, self.sc_addp, self.sc_calc, self.sc_trav,
self.sc_netw, self.sc_tran, self.sc_pdiv, self.sc_plot, self.sc_batchplot])
self.toolbar.insertSeparator(self.sc_calc)
self.toolbar.insertSeparator(self.sc_plot)
def unload(self):
""" Removes the plugin menu item and icon from QGIS GUI.
"""
for action in self.actions:
self.iface.removePluginMenu(
tr(u'&SurveyingCalculation'),
action)
self.iface.removeToolBarIcon(action)
del self.menu
del self.toolbar
def create_coordlist(self):
""" Create a new coordinate list from template and add to layer list. Layer/file name changed to start with 'coord\_' if neccessary.
"""
ofname = QFileDialog.getSaveFileName(self.iface.mainWindow(),
tr('QGIS co-ordinate list'),
filter = tr('Shape file (*.shp)'))
if not ofname:
return
if QRegExp('coord_').indexIn(QFileInfo(ofname).baseName()):
ofname = QDir.cleanPath(QFileInfo(ofname).absolutePath() +
QDir().separator() + 'coord_' + QFileInfo(ofname).fileName())
ofbase = QDir.cleanPath(QFileInfo(ofname).absolutePath() +
QDir().separator() + QFileInfo(ofname).baseName())
tempbase = QDir.cleanPath(self.plugin_dir + QDir().separator() +
'template' + QDir().separator() + 'coord_template')
for ext in ['.shp', '.shx', '.dbf']:
QFile(tempbase+ext).copy(ofbase+ext)
coord = QgsVectorLayer(ofbase+'.shp', QFileInfo(ofbase).baseName(), "ogr")
if coord.isValid():
QgsMapLayerRegistry.instance().addMapLayer(coord)
def create_fb(self):
""" Create a new empty fieldbook from template and add to layer list. Layer/file name changed to start with 'fb\_' if neccessary.
"""
ofname = QFileDialog.getSaveFileName(self.iface.mainWindow(),
tr('New fieldbook'),
filter = tr('Fieldbook file (*.dbf)'))
if not ofname:
return
if QRegExp('fb_').indexIn(QFileInfo(ofname).baseName()):
ofname = QDir.cleanPath(QFileInfo(ofname).absolutePath() +
QDir().separator() + 'fb_' + QFileInfo(ofname).fileName())
ofbase = QDir.cleanPath(QFileInfo(ofname).absolutePath() +
QDir().separator() + QFileInfo(ofname).baseName())
tempbase = QDir.cleanPath(self.plugin_dir + QDir().separator() +
'template' + QDir().separator() + 'fb_template')
for ext in ['.dbf']:
QFile(tempbase+ext).copy(ofbase+ext)
fb = QgsVectorLayer(ofbase+'.dbf', QFileInfo(ofbase).baseName(), "ogr")
if fb.isValid():
QgsMapLayerRegistry.instance().addMapLayer(fb)
def load_fieldbook(self):
""" Import an electric fieldbook from file (GSI, JOB/ARE, ...)
"""
if get_coordlist() is None:
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"), tr("No coordinate list is opened, coordinates will be lost from the fieldbook"))
homedir = QSettings().value("SurveyingCalculation/homedir",config.homedir)
fname = QFileDialog.getOpenFileName(self.iface.mainWindow(), \
tr('Electric fieldbook'), homedir, \
filter = tr('Leica GSI (*.gsi);;Leica IDX (*.idx);;Geodimeter JOB/ARE (*.job *.are);;Sokkia CRD (*.crd);;SurvCE RW5 (*.rw5);;STONEX DAT (*.dat);;Text dump (*.dmp)'))
if fname:
# file selected
# make a copy of dbf template if not are is loaded
if QRegExp('\.are$', Qt.CaseInsensitive).indexIn(fname) == -1:
# ask for table name
ofname = QFileDialog.getSaveFileName(self.iface.mainWindow(),
tr('QGIS fieldbook'),
QFileInfo(fname).absolutePath(),
filter = tr('DBF file (*.dbf)'))
if not ofname:
return
# remember last input dir
QSettings().setValue("SurveyingCalculation/homedir",QFileInfo(fname).absolutePath())
QSettings().sync()
# start with 'fb_'?
if QRegExp('fb_').indexIn(QFileInfo(ofname).baseName()):
ofname = QDir.cleanPath(QFileInfo(ofname).absolutePath() +
QDir().separator() + 'fb_' + QFileInfo(ofname).fileName())
# extension is .dbf?
if QRegExp('\.dbf$', Qt.CaseInsensitive).indexIn(ofname) == -1:
ofname += '.dbf'
tempname = QDir.cleanPath(self.plugin_dir + QDir().separator() +
'template' + QDir().separator() + 'fb_template.dbf')
if not QFile(tempname).copy(ofname):
QMessageBox.warning(self.iface.mainWindow(),
tr('File warning'),
tr('Error copying fieldbook template, target file exists?'),
tr('OK'))
return
fb_dbf = QgsVectorLayer(ofname, QFileInfo(ofname).baseName(), "ogr")
if not fb_dbf or not fb_dbf.isValid():
QMessageBox.warning(self.iface.mainWindow(),
tr('File warning'),
tr('Fieldbook loading error'),
tr('OK'))
return
QgsMapLayerRegistry.instance().addMapLayer(fb_dbf)
if QRegExp('\.gsi$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = LeicaGsi(fname)
elif QRegExp('\.job$', Qt.CaseInsensitive).indexIn(fname) > -1 or \
QRegExp('\.are$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = JobAre(fname)
elif QRegExp('\.crd$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = Sdr(fname)
elif QRegExp('\.rw5$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = SurvCE(fname)
elif QRegExp('\.dat$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = Stonex(fname)
elif QRegExp('\.dmp$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = Dump(fname)
elif QRegExp('\.idx$', Qt.CaseInsensitive).indexIn(fname) > -1:
fb = Idex(fname)
else:
QMessageBox.warning(self.iface.mainWindow(),
tr('File warning'),
tr('Unknown fieldbook type'),
tr('OK'))
return
i = 10 # ordinal number for fieldbook records
#fb_dbf.startEditing()
fb.open()
n_fb = 0 # fieldbook records stored
n_co = 0 # points stored in coordinate list
while True:
# get next observation/station data from fieldbook
r = fb.parse_next()
if r is None:
break # end of file
if 'station' in r:
# add row to fieldbook table
record = QgsFeature()
# add & initialize attributes
record.setFields(fb_dbf.pendingFields(), True)
j = fb_dbf.dataProvider().fieldNameIndex('id')
if j != -1:
record.setAttribute(j, i)
for key in r:
j = fb_dbf.dataProvider().fieldNameIndex(key)
if j != -1:
record.setAttribute(j, r[key])
(xxx, yyy) = fb_dbf.dataProvider().addFeatures([record])
if not xxx:
QMessageBox.warning(self.iface.mainWindow(),
tr('File warning'),
tr('Fieldbook record creation error'),
tr('OK'))
return
n_fb += 1
if 'station_e' in r or 'station_z' in r:
# store station coordinates too
dimension = 0
if 'station_z' in r:
dimension += 1
else:
r['station_z'] = None
if 'station_e' in r and 'station_n' in r:
dimension += 2
else:
r['station_e'] = None
r['station_n'] = None
if not 'pc' in r:
r['pc'] = None
p = Point(r['point_id'], r['station_e'], r['station_n'], r['station_z'], r['pc'])
qp = ScPoint(p)
qp.store_coord(dimension)
n_co += 1
if 'e' in r or 'z' in r:
# store coordinates too
dimension = 0
if 'z' in r:
dimension += 1
else:
r['z'] = None
if 'e' in r and 'n' in r:
dimension += 2
else:
r['e'] = None
r['n'] = None
if not 'pc' in r:
r['pc'] = None
p = Point(r['point_id'], r['e'], r['n'], r['z'], r['pc'])
qp = ScPoint(p)
qp.store_coord(dimension)
n_co += 1
i += 10
#fb_dbf.commitChanges()
if QRegExp('\.are$', Qt.CaseInsensitive).indexIn(fname) == -1:
if n_fb == 0: # no observations
QgsMapLayerRegistry.instance().removeMapLayer(fb_dbf.id())
# remove empty file
QFile(ofname).remove()
if n_co == 0: # no coordinates
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"),\
tr("Neither coordinates nor observations found"))
else:
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"),\
tr("No observations found"))
self.log.write()
self.log.write_log(tr("Fieldbook loaded: ") + fname)
self.log.write(" %d observations, %d coordinates" % (n_fb, n_co))
return
def addp(self):
""" Add point(s) to coordinate list entering coordinates
"""
if get_coordlist() is None:
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"), tr("A coordinate list must be opened!"))
return
# show the dialog
self.newp_dlg.show()
self.newp_dlg.activateWindow()
# Run the dialog event loop
result = self.newp_dlg.exec_()
def calculations(self):
""" Single point calculations (orientation, intersection,
resection, freestation)
"""
# show the dialog
self.single_dlg.show()
self.single_dlg.activateWindow()
# Run the dialog event loop
result = self.single_dlg.exec_()
def traverses(self):
""" Various traverse claculations
"""
# show the dialog
self.traverse_dlg.show()
self.traverse_dlg.activateWindow()
# Run the dialog event loop
result = self.traverse_dlg.exec_()
def networks(self):
""" Various network adjustments (1D/2D/3D)
"""
# show the dialog
self.network_dlg.show()
self.network_dlg.activateWindow()
# Run the dialog event loop
result = self.network_dlg.exec_()
def transformation(self):
""" Various coordinate transformations (orthogonal, affine, polynomial)
"""
# show the dialog
self.transformation_dlg.show()
self.transformation_dlg.activateWindow()
# Run the dialog event loop
result = self.transformation_dlg.exec_()
def polygon_division(self):
""" accept a line from the user to divide the selected polygon on
active layer
"""
al = self.iface.activeLayer()
if al is None or al.type() != QgsMapLayer.VectorLayer or \
al.geometryType() != QGis.Polygon:
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"), tr("Actual layer contains no polygons"))
return
if len(al.selectedFeatures()) != 1:
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"), tr("Not a single polygon is selected in active layer"))
return
self.iface.mapCanvas().setMapTool(self.tool_pdiv)
def plot_by_temp(self):
# show the dialog
self.plotbytemplate_dlg.show()
self.plotbytemplate_dlg.activateWindow()
# Run the dialog event loop
result = self.plotbytemplate_dlg.exec_()
def batch_plotting(self):
""" Batch plots selected geometry items using the selected template and scale.
"""
#check if there are polygon layers in the project
polygon_layers = get_vector_layers_by_type(QGis.Polygon)
if polygon_layers is None:
QMessageBox.warning(self.iface.mainWindow(), tr("Warning"),
tr("This utility needs at least one polygon type layer!"))
return
# show the dialog
self.batchplotting_dlg.show()
self.batchplotting_dlg.activateWindow()
# Run the dialog event loop
result = self.batchplotting_dlg.exec_()
def settings(self):
""" Setting of the plugin.
"""
settings_dlg = PluginSettingsDialog()
result = settings_dlg.exec_()
if result == QDialog.Accepted:
log_path = QSettings().value("SurveyingCalculation/log_path",config.log_path)
self.log.set_log_path(log_path)
def about(self):
""" About box of the plugin
"""
QMessageBox.information(self.iface.mainWindow(),
tr('About'),
tr('Surveying Calculation Plugin\n\n (c) DigiKom Ltd 2014- http://digikom.hu, mail (at) digikom.hu\nVersion 0.2'))
def help(self):
""" Open user's guide of the plugin in the default web browser.
"""
webbrowser.open("http://www.digikom.hu/SurveyingCalculation/usersguide.html")
