GeoSubSampler_dockwidget.py

# -*- coding: utf-8 -*-
"""
GeoSubSamplerDockWidget — pure-Python UI (no .ui file required).
"""

from qgis.PyQt import QtGui, QtWidgets
from qgis.PyQt.QtCore import pyqtSignal, Qt
from qgis.gui import QgsMapLayerComboBox, QgsFieldComboBox, QgsDoubleSpinBox
from qgis.core import QgsFieldProxyModel

try:
    _SB_ALWAYS = Qt.ScrollBarPolicy.ScrollBarAlwaysOn
except AttributeError:
    _SB_ALWAYS = Qt.ScrollBarAlwaysOn


class GeoSubSamplerDockWidget(QtWidgets.QDockWidget):

    closingPlugin = pyqtSignal()

    def __init__(self, parent=None):
        super().__init__(parent)
        self.setWindowTitle("GeoSubSampler")
        self._build_ui()

    # ------------------------------------------------------------------
    # Helpers
    # ------------------------------------------------------------------

    @staticmethod
    def _lbl(text, align=Qt.AlignRight | Qt.AlignVCenter):
        w = QtWidgets.QLabel(text)
        w.setAlignment(align)
        return w

    @staticmethod
    def _hline():
        line = QtWidgets.QFrame()
        line.setFrameShape(QtWidgets.QFrame.HLine)
        line.setFrameShadow(QtWidgets.QFrame.Sunken)
        return line

    # ------------------------------------------------------------------
    # Top-level build
    # ------------------------------------------------------------------

    def _build_ui(self):
        scroll = QtWidgets.QScrollArea()
        scroll.setWidgetResizable(True)
        scroll.setHorizontalScrollBarPolicy(_SB_ALWAYS)
        scroll.setVerticalScrollBarPolicy(_SB_ALWAYS)

        content = QtWidgets.QWidget()
        content.setMinimumWidth(520)
        ml = QtWidgets.QVBoxLayout(content)
        ml.setSpacing(6)
        ml.setContentsMargins(6, 6, 6, 6)

        self._build_layer_selectors(ml)
        self._build_orientation_group(ml)
        self._build_fault_group(ml)
        self._build_geology_group(ml)
        self._build_tp_group(ml)
        ml.addStretch()

        scroll.setWidget(content)
        self.setWidget(scroll)

    # ------------------------------------------------------------------
    # Layer selectors (top, outside groups)
    # ------------------------------------------------------------------

    def _build_layer_selectors(self, layout):
        self.mMapLayerComboBox_points          = QgsMapLayerComboBox()
        self.mMapLayerComboBox_fault_polylines  = QgsMapLayerComboBox()
        self.mMapLayerComboBox_maps_polygons    = QgsMapLayerComboBox()

        pairs = [
            ("Select Structure Point Layer",    self.mMapLayerComboBox_points),
            ("Select Fault Polyline Layer",      self.mMapLayerComboBox_fault_polylines),
            ("Select Litho/strat Polygon Layer", self.mMapLayerComboBox_maps_polygons),
        ]
        for lbl_text, combo in pairs:
            row = QtWidgets.QHBoxLayout()
            row.addWidget(self._lbl(lbl_text))
            row.addWidget(combo)
            layout.addLayout(row)

    # ------------------------------------------------------------------
    # Orientation Points group
    # ------------------------------------------------------------------

    def _build_orientation_group(self, layout):
        grp = QtWidgets.QGroupBox("Orientation Points")
        vbox = QtWidgets.QVBoxLayout(grp)

        # Dip / Dip Dir / convention row
        self.mFieldComboBox_dip     = QgsFieldComboBox()
        self.mFieldComboBox_dip_dir = QgsFieldComboBox()
        self.mFieldComboBox_dip.setFilters(QgsFieldProxyModel.Numeric)
        self.mFieldComboBox_dip_dir.setFilters(QgsFieldProxyModel.Numeric)
        self.checkBox_dip_dir = QtWidgets.QCheckBox("Dip Direction")
        self.checkBox_dip_dir.setChecked(True)

        dip_row = QtWidgets.QHBoxLayout()
        dip_row.addWidget(self._lbl("Dip"))
        dip_row.addWidget(self.mFieldComboBox_dip)
        dip_row.addWidget(self._lbl("Dip Direction/Strike"))
        dip_row.addWidget(self.mFieldComboBox_dip_dir)
        dip_row.addWidget(self.checkBox_dip_dir)
        vbox.addLayout(dip_row)

        vbox.addWidget(self._hline())

        # Radio group — subsampling algorithms
        self._subsample_group = QtWidgets.QButtonGroup(self)

        # (o) Stochastic
        self.radioButton_stochastic = QtWidgets.QRadioButton("Stochastic")
        self.radioButton_stochastic.setChecked(True)
        self.mQgsDoubleSpinBox_percent = QgsDoubleSpinBox()
        self.mQgsDoubleSpinBox_percent.setDecimals(1)
        self.mQgsDoubleSpinBox_percent.setSingleStep(5.0)
        self.mQgsDoubleSpinBox_percent.setValue(50.0)
        self.mQgsDoubleSpinBox_percent.setFixedWidth(75)
        row_stoch = QtWidgets.QHBoxLayout()
        row_stoch.addWidget(self.radioButton_stochastic)
        row_stoch.addWidget(self.mQgsDoubleSpinBox_percent)
        row_stoch.addWidget(self._lbl("%", Qt.AlignLeft | Qt.AlignVCenter))
        row_stoch.addStretch()
        vbox.addLayout(row_stoch)
        self._subsample_group.addButton(self.radioButton_stochastic)

        # (o) Grid Cell Avg
        self.radioButton_gsa = QtWidgets.QRadioButton("Grid Cell Avg")
        self.lineEdit_grid_size = QtWidgets.QLineEdit("5000")
        self.lineEdit_grid_size.setFixedWidth(75)
        row_gsa = QtWidgets.QHBoxLayout()
        row_gsa.addWidget(self.radioButton_gsa)
        row_gsa.addWidget(self._lbl("Grid Size (m)"))
        row_gsa.addWidget(self.lineEdit_grid_size)
        row_gsa.addStretch()
        vbox.addLayout(row_gsa)
        self._subsample_group.addButton(self.radioButton_gsa)

        # (o) Kent
        self.radioButton_kent = QtWidgets.QRadioButton("Kent")
        self.lineEdit_grid_size_kent = QtWidgets.QLineEdit("5000")
        self.lineEdit_grid_size_kent.setFixedWidth(75)
        row_kent = QtWidgets.QHBoxLayout()
        row_kent.addWidget(self.radioButton_kent)
        row_kent.addWidget(self._lbl("Grid Size (m)"))
        row_kent.addWidget(self.lineEdit_grid_size_kent)
        row_kent.addStretch()
        vbox.addLayout(row_kent)
        self._subsample_group.addButton(self.radioButton_kent)

        # (o) Kent Outlier
        self.radioButton_kent_outlier = QtWidgets.QRadioButton("Kent Outlier")
        self.lineEdit_grid_size_kent_2 = QtWidgets.QLineEdit("5000")
        self.lineEdit_grid_size_kent_2.setFixedWidth(75)
        self.lineEdit_kent_threshold = QtWidgets.QLineEdit("2.0")
        self.lineEdit_kent_threshold.setFixedWidth(50)
        self.lineEdit_kent_threshold.setEnabled(False)
        self.pushButton_subsample_points = QtWidgets.QPushButton("Subsample Points")
        row_ko = QtWidgets.QHBoxLayout()
        row_ko.addWidget(self.radioButton_kent_outlier)
        row_ko.addWidget(self._lbl("Grid Size (m)"))
        row_ko.addWidget(self.lineEdit_grid_size_kent_2)
        row_ko.addStretch()
        row_ko.addWidget(self.pushButton_subsample_points)
        vbox.addLayout(row_ko)
        self._subsample_group.addButton(self.radioButton_kent_outlier)

        # Target N — shared by all three grid cell methods
        self.checkBox_target_n = QtWidgets.QCheckBox("Auto-fit to N pts")
        self.spinBox_target_n = QtWidgets.QSpinBox()
        self.spinBox_target_n.setRange(1, 100000)
        self.spinBox_target_n.setValue(100)
        self.spinBox_target_n.setFixedWidth(75)
        self.spinBox_target_n.setEnabled(False)
        self.checkBox_target_n.toggled.connect(self.spinBox_target_n.setEnabled)
        row_target = QtWidgets.QHBoxLayout()
        row_target.addSpacing(20)
        row_target.addWidget(self.checkBox_target_n)
        row_target.addWidget(self.spinBox_target_n)
        row_target.addWidget(self._lbl("points (±15%)", Qt.AlignLeft | Qt.AlignVCenter))
        row_target.addStretch()
        vbox.addLayout(row_target)

        vbox.addWidget(self._hline())

        # 1o Sampling — separate standalone button
        self.pushButton_1o_sampling = QtWidgets.QPushButton("1o Sampling")
        f8 = QtGui.QFont()
        f8.setPointSize(8)
        self.pushButton_1o_sampling.setFont(f8)
        self.lineEdit_1o_distance = QtWidgets.QLineEdit("1000")
        self.lineEdit_1o_distance.setFixedWidth(70)
        self.lineEdit_1o_angle = QtWidgets.QLineEdit("15")
        self.lineEdit_1o_angle.setFixedWidth(50)
        row_1o = QtWidgets.QHBoxLayout()
        row_1o.addWidget(self.pushButton_1o_sampling)
        row_1o.addWidget(self._lbl("Distance"))
        row_1o.addWidget(self.lineEdit_1o_distance)
        row_1o.addWidget(self._lbl("Angle"))
        row_1o.addWidget(self.lineEdit_1o_angle)
        row_1o.addStretch()
        vbox.addLayout(row_1o)

        layout.addWidget(grp)

    # ------------------------------------------------------------------
    # Fault Polylines group
    # ------------------------------------------------------------------

    def _build_fault_group(self, layout):
        grp = QtWidgets.QGroupBox("Fault Polylines")
        vbox = QtWidgets.QVBoxLayout(grp)

        # Merge Segments — kept as a direct action button
        self.pushButton_merge_segments = QtWidgets.QPushButton("Merge Segments")
        self.lineEdit_merge_tolerance    = QtWidgets.QLineEdit("10")
        self.lineEdit_merge_tolerance.setFixedWidth(50)
        self.lineEdit_merge_search_angle = QtWidgets.QLineEdit("30")
        self.lineEdit_merge_search_angle.setFixedWidth(50)
        self.lineEdit_merge_join_angle   = QtWidgets.QLineEdit("150")
        self.lineEdit_merge_join_angle.setFixedWidth(50)
        row_merge = QtWidgets.QHBoxLayout()
        row_merge.addWidget(self.pushButton_merge_segments)
        row_merge.addWidget(self._lbl("Dist.(m)"))
        row_merge.addWidget(self.lineEdit_merge_tolerance)
        row_merge.addWidget(self._lbl("Search Angle"))
        row_merge.addWidget(self.lineEdit_merge_search_angle)
        row_merge.addWidget(self._lbl("Join Angle"))
        row_merge.addWidget(self.lineEdit_merge_join_angle)
        vbox.addLayout(row_merge)

        vbox.addWidget(self._hline())

        # Radio group — fault attribute methods
        self._fault_group = QtWidgets.QButtonGroup(self)

        self.radioButton_fault_length = QtWidgets.QRadioButton("Length Only")
        self.radioButton_fault_length.setChecked(True)
        self._fault_group.addButton(self.radioButton_fault_length)
        vbox.addWidget(self.radioButton_fault_length)

        self.radioButton_fault_graph = QtWidgets.QRadioButton("Graph then Length")
        self._fault_group.addButton(self.radioButton_fault_graph)
        vbox.addWidget(self.radioButton_fault_graph)

        self.radioButton_fault_strat_offset = QtWidgets.QRadioButton("Strat Offset  then Length")
        self._fault_group.addButton(self.radioButton_fault_strat_offset)
        vbox.addWidget(self.radioButton_fault_strat_offset)

        self.radioButton_fault_clusters = QtWidgets.QRadioButton("Orientation Clusters then Length")
        self._fault_group.addButton(self.radioButton_fault_clusters)
        self.pushButton_process_faults = QtWidgets.QPushButton("Process Faults")
        row_fc = QtWidgets.QHBoxLayout()
        row_fc.addWidget(self.radioButton_fault_clusters)
        row_fc.addStretch()
        row_fc.addWidget(self.pushButton_process_faults)
        vbox.addLayout(row_fc)

        layout.addWidget(grp)

    # ------------------------------------------------------------------
    # Geology Polygons group
    # ------------------------------------------------------------------

    def _build_geology_group(self, layout):
        grp = QtWidgets.QGroupBox("Geology Polygons")
        vbox = QtWidgets.QVBoxLayout(grp)

        # Priority fields — row 1
        self.mFieldComboBox_priority_1 = QgsFieldComboBox()
        self.mFieldComboBox_priority_2 = QgsFieldComboBox()
        self.mFieldComboBox_priority_3 = QgsFieldComboBox()
        row_p1 = QtWidgets.QHBoxLayout()
        row_p1.addWidget(self._lbl("Priority 1"))
        row_p1.addWidget(self.mFieldComboBox_priority_1)
        row_p1.addWidget(self._lbl("2"))
        row_p1.addWidget(self.mFieldComboBox_priority_2)
        row_p1.addWidget(self._lbl("3"))
        row_p1.addWidget(self.mFieldComboBox_priority_3)
        vbox.addLayout(row_p1)

        # Priority fields — row 2 + Dyke Field
        self.mFieldComboBox_priority_4 = QgsFieldComboBox()
        self.mFieldComboBox_priority_5 = QgsFieldComboBox()
        self.mFieldComboBox_dyke       = QgsFieldComboBox()
        row_p2 = QtWidgets.QHBoxLayout()
        row_p2.addWidget(self._lbl("4"))
        row_p2.addWidget(self.mFieldComboBox_priority_4)
        row_p2.addWidget(self._lbl("5"))
        row_p2.addWidget(self.mFieldComboBox_priority_5)
        row_p2.addWidget(self._lbl("Dyke Field"))
        row_p2.addWidget(self.mFieldComboBox_dyke)
        vbox.addLayout(row_p2)

        # Dyke codes text box
        self.plainTextEdit_dyke_Codes = QtWidgets.QPlainTextEdit()
        self.plainTextEdit_dyke_Codes.setMaximumHeight(55)
        row_dyke = QtWidgets.QHBoxLayout()
        row_dyke.addWidget(self._lbl("Ignore Codes\n(comma sep)"))
        row_dyke.addWidget(self.plainTextEdit_dyke_Codes)
        vbox.addLayout(row_dyke)

        # Tolerance / diameter / series / increment row
        self.lineEdit_node_tolerance = QtWidgets.QLineEdit("1")
        self.lineEdit_node_tolerance.setFixedWidth(50)
        self.lineEdit_polygon_area = QtWidgets.QLineEdit("5")
        self.lineEdit_polygon_area.setFixedWidth(50)
        self.checkBox_series = QtWidgets.QCheckBox("Series")
        self.mQgsDoubleSpinBox_upinc = QgsDoubleSpinBox()
        self.mQgsDoubleSpinBox_upinc.setDecimals(1)
        self.mQgsDoubleSpinBox_upinc.setMinimum(0.1)
        self.mQgsDoubleSpinBox_upinc.setSingleStep(1.0)
        self.mQgsDoubleSpinBox_upinc.setValue(1.0)
        self.mQgsDoubleSpinBox_upinc.setFixedWidth(75)
        row_tol = QtWidgets.QHBoxLayout()
        row_tol.addWidget(self._lbl("Tolerance (m)"))
        row_tol.addWidget(self.lineEdit_node_tolerance)
        row_tol.addWidget(self._lbl("Diameter Threshold (km)"))
        row_tol.addWidget(self.lineEdit_polygon_area)
        row_tol.addWidget(self.checkBox_series)
        row_tol.addWidget(self._lbl("Increment (km)"))
        row_tol.addWidget(self.mQgsDoubleSpinBox_upinc)
        vbox.addLayout(row_tol)

        # Rescale Map button
        self.pushButton_minPolyArea = QtWidgets.QPushButton("Rescale Map")
        vbox.addWidget(self.pushButton_minPolyArea)

        # Area tolerance + Simplify Map
        self.lineEdit_area_tolerance = QtWidgets.QLineEdit("10000")
        self.lineEdit_area_tolerance.setFixedWidth(80)
        self.pushButton_simplifyMap = QtWidgets.QPushButton("Simplify Map")
        row_area = QtWidgets.QHBoxLayout()
        row_area.addWidget(self._lbl("Area Tolerance (m2)"))
        row_area.addWidget(self.lineEdit_area_tolerance)
        row_area.addWidget(self.pushButton_simplifyMap)
        row_area.addStretch()
        vbox.addLayout(row_area)


        layout.addWidget(grp)

    # ---- Töpfer & Pillewizer scaling group ---------------------------

    def _build_tp_group(self, layout):
        grp = QtWidgets.QGroupBox("Töpfer & Pillewizer Scaling")
        vbox = QtWidgets.QVBoxLayout(grp)

        # Equation reminder label
        eq_lbl = QtWidgets.QLabel("TN = ON × (OS/TS)^(x/2)   x: 1=points  2=lines  3=polygons")
        eq_lbl.setAlignment(Qt.AlignCenter)
        f = eq_lbl.font()
        f.setItalic(True)
        eq_lbl.setFont(f)
        eq_lbl.setToolTip(
            "Töpfer & Pillewizer (1966) selection equation\n\n"
            "TN  — target number of objects after generalisation\n"
            "ON  — original number of objects before generalisation\n"
            "OS  — original map scale denominator (e.g. 50 000)\n"
            "TS  — target map scale denominator (e.g. 500 000)\n"
            "x   — object type: 1 = points, 2 = lines, 3 = polygons\n\n"
            "Example: going from 1:50 000 to 1:500 000\n"
            "  OS/TS = 0.1,  points kept = ON × 0.1^0.5 ≈ 32 %\n"
            "                lines  kept = ON × 0.1^1.0  = 10 %\n"
            "                polys  kept = ON × 0.1^1.5  ≈  3 %"
        )
        vbox.addWidget(eq_lbl)

        vbox.addWidget(self._hline())

        # OS/TS ratio + increment row
        self.lineEdit_tp_ratio = QtWidgets.QLineEdit("0.1")
        self.lineEdit_tp_ratio.setFixedWidth(70)
        self.lineEdit_tp_increment = QtWidgets.QLineEdit("0.1")
        self.lineEdit_tp_increment.setFixedWidth(70)
        self.label_tp_steps = QtWidgets.QLabel("")
        self.pushButton_tp_run = QtWidgets.QPushButton("Run T&P Scaling")
        row_params = QtWidgets.QHBoxLayout()
        row_params.addWidget(self._lbl("OS/TS ratio"))
        row_params.addWidget(self.lineEdit_tp_ratio)
        row_params.addWidget(self._lbl("Increment"))
        row_params.addWidget(self.lineEdit_tp_increment)
        row_params.addWidget(self.label_tp_steps)
        row_params.addStretch()
        row_params.addWidget(self.pushButton_tp_run)
        vbox.addLayout(row_params)

        self.lineEdit_tp_ratio.textChanged.connect(self._update_tp_steps_label)
        self.lineEdit_tp_increment.textChanged.connect(self._update_tp_steps_label)
        self._update_tp_steps_label()

        layout.addWidget(grp)

    # ------------------------------------------------------------------

    def _update_tp_steps_label(self, _=None):
        try:
            ratio     = float(self.lineEdit_tp_ratio.text())
            increment = float(self.lineEdit_tp_increment.text())
        except ValueError:
            self.label_tp_steps.setText("")
            return

        if increment <= 0:
            n = 1
        else:
            n, current = 0, round(1.0 - increment, 12)
            while current > ratio:
                n += 1
                current = round(current - increment, 12)
            n += 1  # the final step that reaches target_ratio

        self.label_tp_steps.setText(f"({n} step{'s' if n != 1 else ''})")

    def closeEvent(self, event):
        self.closingPlugin.emit()
        event.accept()