コード例 #1
0
    def __init__(self):
        super(MainWindow, self).__init__()
        self.ui = Ui_Dialog()
        self.ui.setupUi(self)

        # flags = self.windowFlags() | Qt.WindowStaysOnTopHint ^ Qt.WindowContextHelpButtonHint
        flags = self.windowFlags() ^ Qt.WindowContextHelpButtonHint
        self.setWindowFlags(flags)
        self.ui.progressBar.setVisible(False)
        self.ui.stopButton.hide()

        self._connect_gui_signals()
        self._load_settings()
        self._setup_thread()

        # http://lists.osgeo.org/pipermail/qgis-developer/2012-July/021252.html
        self.timer = QTimer(self)
        self.timer.timeout.connect(self.on_idle)
コード例 #2
0
class MainWindow(QDialog):
    """ Dialog to fetch data over OPeNDAP from NLDAS-2 """

    stop_request = pyqtSignal()

    def __init__(self):
        super(MainWindow, self).__init__()
        self.ui = Ui_Dialog()
        self.ui.setupUi(self)

        # flags = self.windowFlags() | Qt.WindowStaysOnTopHint ^ Qt.WindowContextHelpButtonHint
        flags = self.windowFlags() ^ Qt.WindowContextHelpButtonHint
        self.setWindowFlags(flags)
        self.ui.progressBar.setVisible(False)
        self.ui.stopButton.hide()

        self._connect_gui_signals()
        self._load_settings()
        self._setup_thread()

        # http://lists.osgeo.org/pipermail/qgis-developer/2012-July/021252.html
        self.timer = QTimer(self)
        self.timer.timeout.connect(self.on_idle)

        # sc = QShortcut(QKeySequence(Qt.ALT + Qt.Key_A), self)
        # sc.activatedAmbiguously.connect(self.ui.coverageBox.setFocus)
        # sc.activated.connect(self.ui.coverageBox.setFocus)

        # self.sb = QStatusBar()
        # self.ui.verticalLayout.addWidget(self.sb)

    def _connect_gui_signals(self):
        self.ui.nldasButton.clicked.connect(self.browse_cache)
        self.ui.coverageButton.clicked.connect(self.browse_coverage)
        self.ui.outputButton.clicked.connect(self.browse_output)
        self.ui.runButton.clicked.connect(self.accept)
        self.ui.stopButton.clicked.connect(self.terminate)
        self.ui.hideButton.clicked.connect(self.close)
        QObject.connect(self.ui.coverageBox, SIGNAL("currentIndexChanged(int)"),
                        self.coverage_changed)
        QObject.connect(self.ui.coverageBox, SIGNAL("editTextChanged(QString)"),
                        self.coverage_text_changed)

    def _load_settings(self):
        s = QSettings()
        self.ui.nldasEdit.setText(s.value("pihm_helper/nldas/cache", "./cache").toString())
        o = s.value("pihm_helper/nldas/output", ".").toString()
        self.ui.coverageBox.setEditText(s.value("pihm_helper/nldas/input", ".").toString())
        self.ui.beginEdit.setDateTime(QDateTime.fromString(s.value("pihm_helper/nldas/begin", "1999-12-31T19:00:00").toString(), Qt.ISODate))
        self.ui.endEdit.setDateTime(QDateTime.fromString(s.value("pihm_helper/nldas/end", "2000-01-01T02:00:00").toString(), Qt.ISODate))
        if _iface:
            p = QgsProject.instance()
            o = p.readPath(p.readEntry("pihm_helper", "nldas/output", o)[0])
        self.ui.outputEdit.setText(o)

    def _save_settings(self):
        s = QSettings()
        s.setValue("pihm_helper/nldas/cache", self.ui.nldasEdit.text())
        s.setValue("pihm_helper/nldas/output", self.ui.outputEdit.text())
        if -1 == self.ui.coverageBox.currentIndex():
            s.setValue("pihm_helper/nldas/input", self.ui.coverageBox.currentText())
        s.setValue("pihm_helper/nldas/begin", self.ui.beginEdit.dateTime().toString(Qt.ISODate))
        s.setValue("pihm_helper/nldas/end", self.ui.endEdit.dateTime().toString(Qt.ISODate))
        if _iface:
            p = QgsProject.instance()
            p.writeEntry("pihm_helper", "nldas/output", p.writePath(self.ui.outputEdit.text()))

    def _setup_thread(self):
        self.thread = QThread(self) # self
        self.extractor = Worker()
        self.thread.started.connect(self.extractor.start)
        self.thread.finished.connect(self.on_finished)
        # self.extractor.finished.connect(self.thread.quit)
        QObject.connect(self.extractor, SIGNAL("finished()"),
                        self.thread.quit)
        QObject.connect(self, SIGNAL("stop_request()"),
                        self.extractor.stop)
        QObject.connect(self.extractor, SIGNAL("progress(QString,int,int)"),
                        self.on_progress)
        QObject.connect(self.extractor, SIGNAL("error(QString)"),
                        self.on_error)
        self.extractor.moveToThread(self.thread)

    def accept(self):
        # if self.thread.isRunning():
        #     QMessageBox.warning(self, "Extraction is in progress",
        #                         "I'm already working so I ignore this new request.")
        #     return
        self._save_settings()
        self.extractor.begin = self.ui.beginEdit.dateTime().toUTC().toPyDateTime()
        self.extractor.end = self.ui.endEdit.dateTime().toUTC().toPyDateTime()
        self.extractor.output = str(self.ui.outputEdit.text())
        CACHE = str(self.ui.nldasEdit.text())
        idx = self.ui.coverageBox.currentIndex()
        if idx > -1:
            id = self.ui.coverageBox.itemData(idx).toPyObject()
            layer = QgsMapLayerRegistry.instance().mapLayer(id)
            cnt = layer.selectedFeatureCount()
            if cnt and self.ui.selectedBox.isEnabled() and self.ui.selectedBox.isChecked():
                nldasid = layer.fieldNameIndex('NLDAS_ID')
                nldasx = layer.fieldNameIndex('NLDAS_X')
                nldasy = layer.fieldNameIndex('NLDAS_Y')
                if (any((x==-1 for x in (nldasid, nldasx, nldasy)))):
                    QMessageBox.critical(self, self.tr("Necessary attributes are not found"),
                                         self.tr("Layer must have NLDAS_ID, NLDAS_X, NLDAS_Y fields"))
                    return
                out = []
                r = layer.boundingBoxOfSelected()
                ext = [r.xMinimum(), r.xMaximum(), r.yMinimum(), r.yMaximum()]
                for feature in layer.selectedFeatures():
                    m = feature.attributeMap()
                    id = str(m[nldasid].toPyObject())
                    x = m[nldasx].toPyObject()
                    y = m[nldasy].toPyObject()
                    out.append((id, x, y))
                self.extractor.coverage = (ext, out)
            else:               # no selection
                self.extractor.setCoverage(str(layer.source()))
        else:
            self.extractor.setCoverage(str(self.ui.coverageBox.currentText()))
        self.extractor.csv = self.ui.csvButton.isChecked()
        self.ui.progressBar.setMaximum(0)
        self.ui.progressBar.setVisible(True)
        self.thread.start()
        self.start = datetime.now()
        self.ui.runButton.hide()
        self.ui.stopButton.show()
        if _iface:
            self.timer.start(2000)

    def browse_cache(self):
        folder = QFileDialog.getExistingDirectory(self,
                                                  self.tr("Select OPeNDAP cache location"),
                                                  self.ui.nldasEdit.text(),
                                                  QFileDialog.ShowDirsOnly)
        if folder:
            self.ui.nldasEdit.setText(folder)

    def browse_coverage(self):
        file = QFileDialog.getOpenFileName(self,
                                           self.tr("Select a shapefile"),
                                           self.ui.coverageBox.currentText(),
                                           self.tr("ESRI shapefile (*.shp);;All files (*.*)"))
        if file:
            self.ui.coverageBox.setEditText(file)

    def browse_output(self):
        folder = QFileDialog.getExistingDirectory(self,
                                                  self.tr("Select output location"),
                                                  self.ui.outputEdit.text(),
                                                  QFileDialog.ShowDirsOnly)
        if folder:
            self.ui.outputEdit.setText(folder)

    def coverage_changed(self, idx):
        id = self.ui.coverageBox.itemData(idx).toPyObject()
        if _iface and id:
            try:
                layer = QgsMapLayerRegistry.instance().mapLayer(id)
                sel = layer.selectedFeatureCount() > 0
                self.ui.selectedBox.setEnabled(sel)
                self.ui.selectedBox.setChecked(sel)
            except TypeError:
                self.ui.selectedBox.setEnabled(False)
                self.ui.selectedBox.setChecked(False)

    def coverage_text_changed(self, text):
        idx = self.ui.coverageBox.findText(text)
        if idx > -1:
            self.ui.coverageBox.setCurrentIndex(idx)

    # def event(self, e):
    #     if e.type() == QEvent.StatusTip:
    #         self.sb.showMessage(e.tip())
    #         return True
    #     return super(MainWindow, self).event(e)

    def on_error(self, what):
        self.ui.progressBar.setVisible(False)
        raise Exception(what)

    def on_finished(self):
        self.timer.stop()
        self.ui.stopButton.hide()
        self.ui.stopButton.setEnabled()
        self.ui.runButton.show()
        if _iface:
            td = datetime.now() - self.start
            s = "Done! It took me {:s}".format(td)
            sb = _iface.mainWindow().statusBar()
            sb.showMessage(s)

    def on_idle(self):
        time.sleep(.01)

    def on_progress(self, var, at, end):
        self.ui.progressBar.setMaximum(end)
        self.ui.progressBar.setValue(at)
        if _iface:
            sb = _iface.mainWindow().statusBar()
            s = "NLDAS-2 extractor: {:.1f}% (working on {:s})".format(100.*at/end, var)
            sb.showMessage(s)

    def show(self):
        self.ui.coverageBox.clear()
        if _iface:
            legend = _iface.legendInterface()
            for x in legend.layers():
                if x.type() == x.VectorLayer:
                # if x.type() == x.VectorLayer and \
                #         x.crs().toProj4() == '+proj=longlat +datum=WGS84 +no_defs':
                    self.ui.coverageBox.addItem(x.name(), x.id())
            active = _iface.activeLayer()
            if active:
                idx = self.ui.coverageBox.findData(active.id())
                self.ui.coverageBox.setCurrentIndex(idx)
            # p = QgsProject.instance()
            # self.ui.outputEdit.setText(p.readPath('.'))
            # self.ui.nldasEdit.setText(p.readPath('./cache'))
        return super(QDialog, self).show()

    def terminate(self):
        self.ui.stopButton.setDisabled()
        self.stop_request.emit()