BERATools 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

beratools/gui_tk/bt_widgets.py

@@ -1,442 +0,0 @@
-import os
-import sys
-from PyQt5.QtWidgets import (QApplication, QLineEdit, QFileDialog, QComboBox, QWidget,
-                             QPushButton, QLabel, QSlider, QMessageBox,
-                             QStyleOptionSlider, QStyle, QToolTip, QAbstractSlider,
-                             QHBoxLayout, QVBoxLayout, QSpinBox, QDoubleSpinBox)
-
-from PyQt5.QtCore import pyqtSignal, Qt, QPoint
-from pathlib import Path
-import json
-import re
-
-import inspect
-currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
-parentdir = os.path.dirname(currentdir)
-sys.path.insert(0, parentdir)
-
-from beratools_main import BeraTools
-from common import *
-
-bt = BeraTools()
-
-
-class ToolWidgets(QWidget):
-    signal_save_tool_params = pyqtSignal(object)
-
-    def __init__(self, tool_name, parent=None):
-        super(ToolWidgets, self).__init__(parent)
-
-        self.tool_name = tool_name
-        self.current_tool_api = ''
-        self.widget_list = []
-        self.setWindowTitle("Tool widgets")
-
-        self.create_widgets()
-        layout = QVBoxLayout()
-
-        for item in self.widget_list:
-            layout.addWidget(item)
-
-        self.save_button = QPushButton('Save Parameters')
-        self.save_button.clicked.connect(self.save_tool_parameters)
-        self.save_button.setFixedSize(200, 50)
-        layout.addSpacing(20)
-        layout.addWidget(self.save_button, alignment=Qt.AlignCenter)
-        layout.addStretch()
-        self.setLayout(layout)
-
-    def get_current_tool_parameters(self):
-        tool_params = bt.get_bera_tool_parameters(self.tool_name)
-        self.current_tool_api = tool_params['tool_api']
-        return tool_params
-
-    def create_widgets(self):
-        k = bt.get_bera_tool_info(self.tool_name)
-        print(k)
-        print('\n')
-
-        j = self.get_current_tool_parameters()
-
-        param_num = 0
-        for p in j['parameters']:
-            json_str = json.dumps(p, sort_keys=True, indent=2, separators=(',', ': '))
-            pt = p['parameter_type']
-            widget = None
-
-            if 'ExistingFileOrFloat' in pt:
-                widget = FileOrFloat(json_str, None)
-                param_num = param_num + 1
-            elif 'ExistingFile' in pt or 'NewFile' in pt or 'Directory' in pt:
-                widget = FileSelector(json_str, None)
-                param_num = param_num + 1
-            elif 'FileList' in pt:
-                widget = MultifileSelector(json_str, None)
-                param_num = param_num + 1
-            elif 'Boolean' in pt:
-                widget = BooleanInput(json_str)
-                param_num = param_num + 1
-            elif 'OptionList' in pt:
-                widget = OptionsInput(json_str)
-                param_num = param_num + 1
-            elif ('Float' in pt or 'Integer' in pt or
-                  'Text' in pt or 'String' in pt or 'StringOrNumber' in pt or
-                  'StringList' in pt or 'VectorAttributeField' in pt):
-                widget = DataInput(json_str)
-                param_num = param_num + 1
-            else:
-                msg_box = QMessageBox()
-                msg_box.setIcon(QMessageBox.Warning)
-                msg_box.setText("Unsupported parameter type: {}.".format(pt))
-                msg_box.exec()
-
-            param_value = None
-            if 'saved_value' in p.keys():
-                param_value = p['saved_value']
-            if param_value is None:
-                param_value = p['default_value']
-            if param_value is not None:
-                if type(widget) is OptionsInput:
-                    widget.value = param_value
-                elif widget:
-                    widget.value = param_value
-            else:
-                print('No default value found: {}'.format(p['name']))
-
-            # hide optional widgets
-            if widget:
-                if widget.optional and widget.label:
-                    widget.label.setStyleSheet("QLabel { background-color : transparent; color : blue; }")
-
-                # if widget.optional and not bt.show_advanced:
-                #     widget.hide()
-
-            self.widget_list.append(widget)
-
-    def update_widgets(self, values_dict):
-        for key, value in values_dict.items():
-            for item in self.widget_list:
-                if key == item.flag:
-                    item.set_value(value)
-
-    def save_tool_parameters(self):
-        params = {}
-        for item in self.widget_list:
-            if item.flag:
-                params[item.flag] = item.get_value()
-
-        self.signal_save_tool_params.emit(params)
-
-
-class FileSelector(QWidget):
-    def __init__(self, json_str, runner, master=None, parent=None):
-        super(FileSelector, self).__init__(parent)
-
-        # first make sure that the json data has the correct fields
-        j = json.loads(json_str)
-        self.name = j['name']
-        self.description = j['description']
-        self.flag = j['flag']
-        self.parameter_type = j['parameter_type']
-        self.file_type = ""
-        if "ExistingFile" in self.parameter_type:
-            self.file_type = j['parameter_type']['ExistingFile']
-        elif "NewFile" in self.parameter_type:
-            self.file_type = j['parameter_type']['NewFile']
-        self.optional = j['optional']
-        self.value = j['default_value']
-
-        self.runner = runner
-
-        self.layout = QHBoxLayout()
-        self.label = QLabel(self.name)
-        self.label.setMinimumWidth(BT_LABEL_MIN_WIDTH)
-        self.in_file = QLineEdit()
-        self.btn_select = QPushButton("...")
-        self.btn_select.clicked.connect(self.select_file)
-        self.layout.addWidget(self.label)
-        self.layout.addWidget(self.in_file)
-        self.layout.addWidget(self.btn_select)
-
-        self.setLayout(self.layout)
-
-    def select_file(self):
-        try:
-            dialog = QFileDialog(self)
-            dialog.setViewMode(QFileDialog.Detail)
-            dialog.setDirectory(str(Path(self.value).parent))
-            dialog.selectFile(Path(self.value).name)
-            result = None
-            file_names = None
-
-            if self.parameter_type == "Directory":
-                dialog.setFileMode(QFileDialog.FileMode.Directory)
-            elif "ExistingFile" in self.parameter_type or "NewFile" in self.parameter_type:
-                file_types = "All files '*.*')"
-                if 'RasterAndVector' in self.file_type:
-                    file_types = """Shapefiles (*.shp);; 
-                                    Raster files (*.dep *.tif *.tiff *.bil *.flt *.sdat *.rdc *.asc *grd)"""
-                elif 'Raster' in self.file_type:
-                    file_types = """Tiff raster files (*.tif *.tiff);; 
-                                    Other raster files (*.dep *.bil *.flt *.sdat *.rdc *.asc *.grd)"""
-                elif 'Lidar' in self.file_type:
-                    file_types = "LiDAR files (*.las *.zlidar *.laz *.zip)"
-                elif 'Vector' in self.file_type:
-                    file_types = "Shapefiles (*.shp)"
-                elif 'Text' in self.file_type:
-                    file_types = "Text files (*.txt);; all files (*.*)"
-                elif 'Csv' in self.file_type:
-                    file_types = "CSC files (*.csv);; all files (*.*)"
-                elif 'Dat' in self.file_type:
-                    file_types = "Binary data files (*.dat);; all files (*.*)"
-                elif 'Html' in self.file_type:
-                    file_types = "HTML files (*.html)"
-                elif 'json' in self.file_type or 'JSON' in self.file_type:
-                    file_types = "JSON files (*.json)"
-
-                dialog.setNameFilter(file_types)
-
-                if "ExistingFile" in self.parameter_type:
-                    dialog.setFileMode(QFileDialog.FileMode.ExistingFiles)
-                else:
-                    dialog.setFileMode(QFileDialog.FileMode.AnyFile)
-
-                if dialog.exec_():
-                    file_names = dialog.selectedFiles()
-
-                if not file_names:
-                    return
-
-                if len(file_names) == 0:
-                    print('No file(s) selected.')
-
-                    if file_names[0] == '':
-                        print('File name not valid.')
-                        return
-
-                # append suffix when not
-                # TODO: more consideration for multiple formats
-                result = file_names[0]
-                file_path = Path(result)
-                if result != '':
-                    break_loop = False
-                    selected_filters = self.get_file_filter_list(dialog.selectedNameFilter())
-
-                    if file_path.suffix not in selected_filters:
-                        if selected_filters[0] != '.*':
-                            file_path = file_path.with_suffix(selected_filters[0])
-
-                result = str(file_path)
-                self.set_value(result)
-
-            # update the working
-            # if not self.runner and str(result) != '':
-            #     self.runner.working_dir = os.path.dirname(result)
-        except:
-            t = "file"
-            if self.parameter_type == "Directory":
-                t = "directory"
-
-            msg_box = QMessageBox()
-            msg_box.setIcon(QMessageBox.Warning)
-            msg_box.setText("Could not find {}".format(t))
-            msg_box.exec()
-
-    def get_file_filter_list(self, filter_str):
-        """
-        Extract filters out of full filter string, split int list and replace first '*'
-        Result: ['.shp', '.*']
-        """
-        filter_list = re.search('\((.+?)\)', filter_str).group(1).split(' ')
-        filter_list = [item.replace('*', '', 1) for item in filter_list if item != '']
-        return filter_list
-
-    def get_value(self):
-        return self.value
-
-    def set_value(self, value):
-        self.value = value
-        self.in_file.setText(Path(self.value).name)
-        self.in_file.setToolTip(value)
-
-
-class FileOrFloat(QWidget):
-    def __init__(self, json_str, runner, master=None, parent=None):
-        super(FileOrFloat, self).__init__(parent)
-        pass
-
-
-class MultifileSelector(QWidget):
-    def __init__(self, json_str, runner, master=None, parent=None):
-        super(MultifileSelector, self).__init__(parent)
-        pass
-
-
-class BooleanInput(QWidget):
-    def __init__(self, json_str, master=None, parent=None):
-        super(BooleanInput, self).__init__(parent)
-        pass
-
-
-class OptionsInput(QWidget):
-    def __init__(self, json_str, master=None, parent=None):
-        super(OptionsInput, self).__init__(parent)
-
-        # first make sure that the json data has the correct fields
-        j = json.loads(json_str)
-        self.name = j['name']
-        self.description = j['description']
-        self.flag = j['flag']
-        self.parameter_type = j['parameter_type']
-        self.optional = j['optional']
-        self.data_type = j['data_type']
-        self.default_value = str(j['default_value'])
-        self.value = self.default_value  # initialize in event of no default and no selection
-
-        self.label = QLabel(self.name)
-        self.label.setMinimumWidth(BT_LABEL_MIN_WIDTH)
-        self.combobox = QComboBox()
-        self.combobox.currentIndexChanged.connect(self.selection_change)
-
-        i = 1
-        default_index = -1
-        self.option_list = j['parameter_type']['OptionList']
-        if self.option_list:
-            self.option_list = [str(item) for item in self.option_list]  # convert to strings
-        values = ()
-        for v in self.option_list:
-            values += (v,)
-            if v == str(self.default_value):
-                default_index = i - 1
-            i = i + 1
-
-        self.combobox.addItems(self.option_list)
-        self.combobox.setCurrentIndex(default_index)
-
-        self.layout = QHBoxLayout()
-        self.layout.addWidget(self.label)
-        self.layout.addWidget(self.combobox)
-        self.setLayout(self.layout)
-
-    def selection_change(self, i):
-        self.value = self.option_list[i]
-
-    def set_value(self, value):
-        self.value = self.value
-        for v in self.option_list:
-            if value == v:
-                self.combobox.setCurrentIndex(self.option_list.index(v))
-
-    def get_value(self):
-        return self.value
-
-
-class DataInput(QWidget):
-    def __init__(self, json_str, master=None, parent=None):
-        super(DataInput, self).__init__(parent)
-
-        # first make sure that the json data has the correct fields
-        j = json.loads(json_str)
-        self.name = j['name']
-        self.description = j['description']
-        self.flag = j['flag']
-        self.parameter_type = j['parameter_type']
-        self.optional = j['optional']
-        self.default_value = j['default_value']
-        self.data = j['default_value']
-        self.label = QLabel(self.name)
-        self.label.setMinimumWidth(BT_LABEL_MIN_WIDTH)
-        self.data_input = None
-
-        if "Integer" in self.parameter_type:
-            self.data_input = QSpinBox()
-        elif "Float" in self.parameter_type or "Double" in self.parameter_type:
-            self.data_input = QDoubleSpinBox()
-
-        if self.data_input:
-            self.data_input.setValue(self.data)
-
-        self.data_input.valueChanged.connect(self.update_value)
-
-        self.layout = QHBoxLayout()
-        self.layout.addWidget(self.label)
-        self.layout.addWidget(self.data_input)
-        self.setLayout(self.layout)
-
-    def update_value(self):
-        self.data = self.data_input.value
-
-    def get_value(self):
-        v = self.data
-        if v:
-            if "Integer" in self.parameter_type:
-                return self.flag, int(self.data())
-            elif "Float" in self.parameter_type:
-                return self.flag, float(self.data())
-            elif "Double" in self.parameter_type:
-                return self.flag, float(self.data())
-            else:  # String or StringOrNumber types
-                return self.flag, self.value.get()
-        else:
-            if not self.optional:
-                msg_box = QMessageBox()
-                msg_box.setIcon(QMessageBox.Warning)
-                msg_box.setText("Unspecified non-optional parameter {}.".format(self.flag))
-                msg_box.exec()
-
-        return None
-
-    def set_value(self, value):
-        if self.data_input:
-            self.data_input.setValue(value)
-            self.update_value()
-
-class DoubleSlider(QSlider):
-
-    # create our our signal that we can connect to if necessary
-    doubleValueChanged = pyqtSignal(float)
-
-    def __init__(self, decimals=3, *args, **kargs):
-        super(DoubleSlider, self).__init__(Qt.Horizontal)
-        self._multi = 10 ** decimals
-
-        self.opt = QStyleOptionSlider()
-        self.initStyleOption(self.opt)
-
-        self.valueChanged.connect(self.emitDoubleValueChanged)
-
-    def emitDoubleValueChanged(self):
-        value = float(super(DoubleSlider, self).value())/self._multi
-        self.doubleValueChanged.emit(value)
-
-    def value(self):
-        return float(super(DoubleSlider, self).value()) / self._multi
-
-    def setMinimum(self, value):
-        return super(DoubleSlider, self).setMinimum(value * self._multi)
-
-    def setMaximum(self, value):
-        return super(DoubleSlider, self).setMaximum(value * self._multi)
-
-    def setSingleStep(self, value):
-        return super(DoubleSlider, self).setSingleStep(value * self._multi)
-
-    def singleStep(self):
-        return float(super(DoubleSlider, self).singleStep()) / self._multi
-
-    def setValue(self, value):
-        super(DoubleSlider, self).setValue(int(value * self._multi))
-
-    def sliderChange(self, change):
-        if change == QAbstractSlider.SliderValueChange:
-            sr = self.style().subControlRect(QStyle.CC_Slider, self.opt, QStyle.SC_SliderHandle)
-            bottomRightCorner = sr.bottomLeft()
-            QToolTip.showText(self.mapToGlobal(QPoint(bottomRightCorner.x(), bottomRightCorner.y())),
-                              str(self.value()), self)
-
-
-if __name__ == '__main__':
-    app = QApplication(sys.argv)
-    dlg = ToolWidgets('Raster Line Attributes')
-    dlg.show()
-    sys.exit(app.exec_())

beratools/gui_tk/cli.py

@@ -1,18 +0,0 @@
-# -*- coding: utf-8 -*-
-
-"""Console script for BERA tools."""
-import sys
-import click
-
-
-@click.command()
-def main(args=None):
-    """Console script for BERA tools."""
-    click.echo("Replace this message by putting your code into "
-               "BERA_tools.cli.main")
-    click.echo("See click documentation at http://click.pocoo.org/")
-    return 0
-
-
-if __name__ == "__main__":
-    sys.exit(main())  # pragma: no cover

beratools/gui_tk/main.py

@@ -1,14 +0,0 @@
-import sys
-from pathlib import Path
-from inspect import getsourcefile
-
-if __name__ == "__main__":
-    current_file = Path(getsourcefile(lambda: 0)).resolve()
-    btool_dir = current_file.parents[2]
-    sys.path.insert(0, btool_dir.as_posix())
-
-from beratools.gui_tk.runner import main_runner
-
-
-if __name__ == "__main__":
-    main_runner()

beratools/gui_tk/map_window.py

@@ -1,144 +0,0 @@
-import os
-import sys
-os.environ['QT_API'] = 'pyqt5'
-from qtpy.QtWidgets import (QApplication, QVBoxLayout, QHBoxLayout, QWidget, QTreeWidget, QTreeWidgetItem,
-                            QPushButton, QGroupBox, QDialog, QDialogButtonBox)
-from qtpy.QtCore import (Qt, Signal)
-from beratools.third_party.pyqtlet2 import L, MapWidget
-
-
-class MapWindow(QDialog):
-    def __init__(self, parent=None):
-        # Setting up the widgets and layout
-        super(MapWindow, self).__init__(parent)
-        self.setWindowTitle('Tiler map')
-        self.setGeometry(0, 0, 1200, 800)
-
-        # delete dialog when close
-        self.setAttribute(Qt.WA_DeleteOnClose)
-
-        # Add OK/cancel buttons
-        self.ok_btn_box = QDialogButtonBox(Qt.Vertical)
-        self.ok_btn_box.addButton("Run Tiler", QDialogButtonBox.AcceptRole)
-        self.ok_btn_box.addButton("Cancel", QDialogButtonBox.RejectRole)
-        self.ok_btn_box.addButton("Help", QDialogButtonBox.HelpRole)
-
-        self.ok_btn_box.buttons()[0].setFixedSize(120, 40)
-        self.ok_btn_box.buttons()[1].setFixedSize(120, 40)
-        self.ok_btn_box.buttons()[2].setFixedSize(120, 40)
-
-        self.ok_btn_box.accepted.connect(self.run)
-        self.ok_btn_box.rejected.connect(self.cancel)
-        self.ok_btn_box.helpRequested.connect(self.help)
-
-        self.info_layout = QVBoxLayout()  # layout reserved for tiles info widgets
-        self.vbox_group = QVBoxLayout()
-        self.vbox_group.addLayout(self.info_layout)
-        self.vbox_group.addStretch()
-        self.vbox_group.addWidget(self.ok_btn_box, alignment=Qt.AlignCenter)
-
-        groupbox_info = QGroupBox('Tiles')
-        groupbox_info.setLayout(self.vbox_group)
-
-        central_widget = QWidget()
-        map_layout = QHBoxLayout(central_widget)
-        map_layout.addWidget(groupbox_info)
-
-        groupbox_map = QGroupBox('Map')
-        self.map_widget = MapWidget()
-        self.map_widget.setContentsMargins(10, 10, 10, 10)
-        self.vbox_map = QVBoxLayout()
-        self.vbox_map.addWidget(self.map_widget)
-        groupbox_map.setLayout(self.vbox_map)
-        map_layout.addWidget(groupbox_map, 10)
-        self.setLayout(map_layout)
-
-        # Working with the maps with pyqtlet
-        self.map = L.map(self.map_widget)
-        self.map.setView([0, 0], 10)  # this is necessary
-
-        L.tileLayer('http://{s}.tile.osm.org/{z}/{x}/{y}.png').addTo(self.map)
-
-        # add marker layer
-        # self.add_marker_layer()
-        self.show()
-
-    def add_polygons_to_map(self, layer_name, polygons, color):
-        style = {'fillOpacity': 0.1, 'color': color}
-        vars()[layer_name] = L.polygon(polygons, style)
-        self.map.addLayer(vars()[layer_name])
-
-        # this works too. addLayer has to be called first
-        # self.map.runJavaScript("var stylePoly = {fillColor:'red',color: 'blue',weight:2,fillOpacity:0.8};", 0)
-        # self.map.runJavaScript(f'{self.multipolygon.jsName}.setStyle(stylePoly);', 0)
-    def add_polylines_to_map(self, polylines, color):
-        style = {'color': color}
-        lines = L.polyline(polylines, style)
-        self.map.addLayer(lines)
-
-    def set_view(self, point, zoom):
-        self.map = self.map.setView(point, zoom)
-
-    # bounds is a pair of corner points, LL and UR
-    def fit_bounds(self, bounds):
-        # self.map.fitBounds(bounds)
-        self.map.runJavaScript(f'{self.map.jsName}.fitBounds(bounds);', 0)
-
-    def add_marker_layer(self):
-        self.marker = L.marker([12.934056, -77.610029])
-        self.marker.bindPopup('Maps are a treasure.')
-        self.map.addLayer(self.marker)
-
-        # Create a icon called markerIcon in the js runtime.
-        self.map.runJavaScript('var markerIcon '
-                               '= L.icon({iconUrl: "https://leafletjs.com/examples/custom-icons/leaf-red.png"});', 0)
-
-        # Edit the existing python object by accessing it's jsName property
-        self.map.runJavaScript(f'{self.marker.jsName}.setIcon(markerIcon);', 0)
-
-    def accept(self):
-        print("Run the tiling.")
-        QDialog.accept(self)
-
-    def run(self):
-        self.accept()
-
-    def cancel(self):
-        print("Tiling canceled.")
-        self.reject()
-
-    def help(self):
-        print("Help requested.")
-
-    def set_tiles_info(self, tiles_info):
-        tree = QTreeWidget()
-        tree.setColumnCount(2)
-        tree.setHeaderLabels(["Item", "Value"])
-        item = QTreeWidgetItem(['Tiles'])
-        for key, value in tiles_info.items():
-            child = QTreeWidgetItem([key, str(value)])
-            item.addChild(child)
-
-        tree.insertTopLevelItem(0, item)
-        tree.expandAll()
-
-        # add to group widget
-        self.vbox_group.insertWidget(0, tree)
-
-
-if __name__ == '__main__':
-    # supress web engine logging
-    os.environ["QTWEBENGINE_CHROMIUM_FLAGS"] = "--enable-logging --log-level=3"
-
-    app = QApplication(sys.argv)
-    widget = MapWindow()
-
-    # add polygons to map
-    polygon_coords_base = [[[17.285044, 78.286671], [16.606174, 80.748015], [17.886816, 83.518482]]]
-    widget.add_polygons_to_map(polygon_coords_base, 'blue')
-
-    polygon_coords = [[[17.385044, 78.486671], [16.506174, 80.648015], [17.686816, 83.218482]],
-                      [[13.082680, 80.270718], [12.971599, 77.594563], [15.828126, 78.037279]]]
-    widget.add_polygons_to_map(polygon_coords, 'red')
-
-    sys.exit(app.exec_())