mapillary_settings.py

# -*- coding: utf-8 -*-
"""
/***************************************************************************
 go2mapillaryDockWidget
                                 A QGIS plugin
 mapillary filter
                             -------------------
        begin                : 2016-01-21
        git sha              : $Format:%H$
        copyright            : (C) 2016 by enrico ferreguti
        email                : enricofer@gmail.com
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
"""

import os
import sys
import datetime
import json

from PyQt5 import QtWidgets, uic
from PyQt5.QtCore import pyqtSignal
from PyQt5.QtWidgets import QTableWidgetItem
from PyQt5.QtGui import QColor

from qgis.core import QgsPointXY, QgsCoordinateReferenceSystem, QgsCoordinateTransform, QgsProject, Qgis, QgsExpressionContextUtils
from qgis.gui import QgsFileWidget

FORM_CLASS, _ = uic.loadUiType(os.path.join(
    os.path.dirname(__file__), 'mapillary_settings_dialog_base.ui'))


class mapillarySettings(QtWidgets.QDialog, FORM_CLASS):

    closingPlugin = pyqtSignal()

    def __init__(self,parentInstance, parent=None):
        """Constructor."""
        super(mapillarySettings, self).__init__(parent)
        # Set up the user interface from Designer.
        # After setupUI you can access any designer object by doing
        # self.<objectname>, and you can use autoconnect slots - see
        # http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
        # #widgets-and-dialogs-with-auto-connect
        self.setupUi(self)
        self.parentInstance = parentInstance
        self.iface = parentInstance.iface
        self.setWindowTitle("go2mapillary settings")
        self.fileWidget.setStorageMode(QgsFileWidget.SaveFile)
        self.fileWidget.setFilter("SHP files (*.shp)")
        self.buttonBox.accepted.connect(self.applySettings)
        self.addCategoryButton.clicked.connect(self.addCategoryAction)
        self.removeCategoryButton.clicked.connect(self.removeCategoryAction)
        self.tableWidget.setColumnWidth(1 , 150)
        self.newCategoryEdit.setMaxLength(20)
        self.settings = {
            'sample_source': 'memory',
            'auto_open_form': True,
            'categories': {
                'cat_a':'#ff0000',
                'cat_b':'#0000ff',
                'cat_c':'#00ff00',
            }
        }
        if not  QgsExpressionContextUtils.globalScope().hasVariable('mapillarySettings'):
            QgsExpressionContextUtils.setGlobalVariable("mapillarySettings", json.dumps(self.settings))
        self.loadSettings()

    def loadSettings(self):
        self.settings = json.loads(QgsExpressionContextUtils.globalScope().variable("mapillarySettings"))
        if self.settings['sample_source'] == 'memory':
            self.radioButtonMemorySource.setChecked(True)
        else:
            self.radioButtonShpSource.setChecked(True)
            self.fileWidget.setFilePath(self.settings['sample_source'])
        self.autoOpenCheckBox.setChecked(self.settings['auto_open_form'])
        self.tableWidget.clear()
        self.tableWidget.setRowCount(0)
        for cat,color in self.settings['categories'].items():
            self.tableWidget.insertRow(0)
            self.tableWidget.setItem(0, 0, QTableWidgetItem())
            self.tableWidget.item(0, 0).setBackground(QColor(color))
            self.tableWidget.setItem(0, 1, QTableWidgetItem(cat))


    def applySettings(self):
        self.settings = {}
        if self.radioButtonMemorySource.isChecked():
            self.settings['sample_source'] = 'memory'
        else:
            self.settings['sample_source'] = self.fileWidget.filePath()
        self.settings['auto_open_form'] = self.autoOpenCheckBox.isChecked()
        categories = {}
        for row in range(0,self.tableWidget.rowCount()):
            categories[self.tableWidget.item(row,1).text()[:20]] = self.tableWidget.item(row,0).background().color().name()
        self.settings['categories'] = categories
        self.parentInstance.sample_cursor.update_ds(self.settings['sample_source'])
        QgsExpressionContextUtils.setGlobalVariable("mapillarySettings", json.dumps(self.settings))


    def addCategoryAction(self):
        if self.newCategoryEdit.text():
            self.tableWidget.setRowCount(self.tableWidget.rowCount()+1)
            self.tableWidget.setItem(self.tableWidget.rowCount()-1, 0, QTableWidgetItem())
            self.tableWidget.item(self.tableWidget.rowCount()-1, 0).setBackground(self.mColorButton.color())
            self.tableWidget.setItem(self.tableWidget.rowCount()-1, 1, QTableWidgetItem(self.newCategoryEdit.text()[:20]))
            self.newCategoryEdit.setText("")

    def removeCategoryAction(self):
        if self.tableWidget.selectedItems():
            del_row = self.tableWidget.selectedItems()[0].row()
            self.tableWidget.clearSelection()
            self.tableWidget.removeRow(del_row)
            self.tableWidget.clearSelection()
        elif self.tableWidget.rowCount() > 0:
            self.tableWidget.removeRow(self.tableWidget.rowCount()-1)