/
export_dialog.py
87 lines (68 loc) · 3.15 KB
/
export_dialog.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
# -*- coding: utf-8 -*-
"""
/***************************************************************************
ExportDialog
A QGIS plugin
A crater counting tool for planetary science
-------------------
begin : 2014-12-31
git sha : $Format:%H$
copyright : (C) 2014 by Sarah E Braden
email : braden.sarah@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
from PyQt5 import QtGui, QtCore, uic, QtWidgets
from .errors import CircleCraterError
ExportDialogBase, _ = uic.loadUiType(os.path.join(
os.path.dirname(__file__), 'export_dialog_base.ui'))
class ExportDialog(QtWidgets.QDialog, ExportDialogBase):
selected = QtCore.pyqtSignal(object, object, str)
def __init__(self, parent=None):
"""Constructor."""
super(ExportDialog, 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.accepted.connect(self.on_accept)
self.filename_choose_button.clicked.connect(self.choose_file)
def choose_file(self):
self.filename_input.setText(QtWidgets.QFileDialog.getSaveFileName()[0])
def show(self, choices):
if not choices:
raise CircleCraterError(
'No choice of layers available. '
'Please create polygon type vector layers.'
)
self.crater_layer_select.clear()
self.area_layer_select.clear()
for layer in choices:
# Add these layers to the combobox (dropdown menu)
self.crater_layer_select.addItem(layer.name(), layer)
self.area_layer_select.addItem(layer.name(), layer)
super(ExportDialog, self).show()
def _get_layer(self, selector):
return selector.itemData(selector.currentIndex())
def get_crater_layer(self):
return self._get_layer(self.crater_layer_select)
def get_area_layer(self):
return self._get_layer(self.area_layer_select)
def get_filename(self):
return self.filename_input.text()
def on_accept(self):
self.selected.emit(
self.get_crater_layer(),
self.get_area_layer(),
self.get_filename(),
)