style: running black

Author: lachlangrose

loopstructural/gui/loop_widget.py

@@ -1,8 +1,12 @@
 from PyQt5.QtWidgets import QTabWidget, QVBoxLayout, QWidget
 from .modelling.modelling_widget import ModellingWidget
 from .visualisation.visualisation_widget import VisualisationWidget
+
+
 class LoopWidget(QWidget):
-    def __init__(self, parent=None, *, mapCanvas=None, logger=None, data_manager=None, model_manager=None):
+    def __init__(
+        self, parent=None, *, mapCanvas=None, logger=None, data_manager=None, model_manager=None
+    ):
         super().__init__(parent)
         self.mapCanvas = mapCanvas
         self.logger = logger
@@ -15,12 +19,15 @@ def __init__(self, parent=None, *, mapCanvas=None, logger=None, data_manager=Non
         tabWidget.setTabPosition(QTabWidget.South)
         mainLayout.addWidget(tabWidget)
         self.modelling_widget = ModellingWidget(
-            self, mapCanvas=self.mapCanvas, logger=self.logger, data_manager=self.data_manager, model_manager=self.model_manager
+            self,
+            mapCanvas=self.mapCanvas,
+            logger=self.logger,
+            data_manager=self.data_manager,
+            model_manager=self.model_manager,
         )
 
         self.visualisation_widget = VisualisationWidget(
             self, mapCanvas=self.mapCanvas, logger=self.logger, model_manager=self.model_manager
         )
         tabWidget.addTab(self.modelling_widget, "Modelling")
         tabWidget.addTab(self.visualisation_widget, "Visualisation")
-

loopstructural/gui/modelling/geological_history_tab.py

@@ -1,5 +1,3 @@
-
-
 from loopstructural.gui.modelling.base_tab import BaseTab
 from loopstructural.gui.modelling.stratigraphic_column.stratigraphic_column import StratColumnWidget
 

loopstructural/gui/modelling/model_definition/model_definition_tab.py

@@ -1,4 +1,3 @@
-
 from PyQt5.QtWidgets import QSizePolicy
 
 from loopstructural.gui.modelling.base_tab import BaseTab
@@ -8,6 +7,7 @@
 from .stratigraphic_layers import StratigraphicLayersWidget
 from .dem import DEMWidget
 
+
 class ModelDefinitionTab(BaseTab):
     def __init__(self, parent=None, data_manager=None):
         super().__init__(parent, data_manager, scrollable=True)
@@ -18,10 +18,12 @@ def __init__(self, parent=None, data_manager=None):
         self.stratigraphy_layers = StratigraphicLayersWidget(self, data_manager)
 
         # Set uniform size policy for all widgets
-        for widget in [self.bounding_box, self.fault_layers, self.dem,self.stratigraphy_layers]:
+        for widget in [self.bounding_box, self.fault_layers, self.dem, self.stratigraphy_layers]:
             widget.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
 
         self.add_widget(self.bounding_box, 'Bounding Box')  # , "Bounding Box")
         self.add_widget(self.dem, 'DEM')
         self.add_widget(self.fault_layers, 'Fault Layers')  # , "Fault Layers")
-        self.add_widget(self.stratigraphy_layers, 'Stratigraphic Layers')  # , "Stratigraphic Layers")
+        self.add_widget(
+            self.stratigraphy_layers, 'Stratigraphic Layers'
+        )  # , "Stratigraphic Layers")

loopstructural/gui/modelling/modelling_widget_back.py

@@ -1,4 +1,3 @@
-
 from PyQt5.QtWidgets import (
     QAbstractItemView,
     QListWidget,
@@ -63,6 +62,7 @@ def clearColumn(self):
             self.data_manager._stratigraphic_column.clear()
         else:
             print("Error: Data manager is not initialized.")
+
     def update_display(self):
         """Update the widget display based on the data manager's stratigraphic column."""
         self.unitList.clear()
@@ -72,7 +72,7 @@ def update_display(self):
                     self.add_unit(unit_data=unit.to_dict(), create_new=False)
                 elif unit.element_type == StratigraphicColumnElementType.UNCONFORMITY:
 
-                    self.add_unconformity(unconformity_data=unit.to_dict(),create_new=False)
+                    self.add_unconformity(unconformity_data=unit.to_dict(), create_new=False)
 
     def init_stratigraphic_column_from_basal_contacts(self):
         if self.data_manager:
@@ -99,9 +99,13 @@ def add_unit(self, *, unit_data=None, create_new=True):
                 unit_data.pop(k)
         unit_widget = StratigraphicUnitWidget(**unit_data)
         unit_widget.deleteRequested.connect(self.delete_unit)  # Connect delete signal
-        unit_widget.nameChanged.connect(lambda: self.update_element(unit_widget))  # Connect name change signal
+        unit_widget.nameChanged.connect(
+            lambda: self.update_element(unit_widget)
+        )  # Connect name change signal
 
-        unit_widget.thicknessChanged.connect(lambda: self.update_element(unit_widget))  # Connect thickness change signal
+        unit_widget.thicknessChanged.connect(
+            lambda: self.update_element(unit_widget)
+        )  # Connect thickness change signal
 
         unit_widget.set_thickness(unit_data.get('thickness', 0.0))  # Set initial thickness
         item = QListWidgetItem()

loopstructural/gui/modelling/stratigraphic_column/stratigraphic_column.py

@@ -1,5 +1,3 @@
-
-
 class GeometryObject:
     def __init__(self, name, object, options=None):
         self.name = name

loopstructural/gui/visualisation/geometry_object.py

@@ -1,6 +1,7 @@
 import enum
 from typing import Dict
 
+
 class UnconformityType(enum.Enum):
     """
     An enumeration for different types of unconformities in a stratigraphic column.
@@ -213,6 +214,7 @@ def get_groups(self):
         if group:
             groups.append(group)
         return groups
+
     def get_unitname_groups(self):
         groups = []
         group = []
@@ -226,6 +228,7 @@ def get_unitname_groups(self):
         if group:
             groups.append(group)
         return groups
+
     def __getitem__(self, uuid):
         """
         Retrieves an element by its uuid from the stratigraphic column.
@@ -244,6 +247,7 @@ def update_order(self, new_order):
         self.order = [
             self.__getitem__(uuid) for uuid in new_order if self.__getitem__(uuid) is not None
         ]
+
     def update_element(self, unit_data: Dict):
         """
         Updates an existing element in the stratigraphic column with new data.
@@ -261,6 +265,7 @@ def update_element(self, unit_data: Dict):
             element.unconformity_type = UnconformityType(
                 unit_data.get('unconformity_type', element.unconformity_type.value)
             )
+
     def clear(self):
         """
         Clears the stratigraphic column, removing all elements.

loopstructural/main/stratigraphic_column.py

@@ -2,14 +2,13 @@
 import geopandas as gpd
 from qgis.core import QgsRaster, QgsWkbTypes
 
+
 def qgsLayerToGeoDataFrame(layer) -> gpd.GeoDataFrame:
     if layer is None:
         return None
     features = layer.getFeatures()
     fields = layer.fields()
-    data = {
-        'geometry': []
-    }
+    data = {'geometry': []}
     for f in fields:
         data[f.name()] = []
     for feature in features:
@@ -21,6 +20,7 @@ def qgsLayerToGeoDataFrame(layer) -> gpd.GeoDataFrame:
             data[f.name()].append(feature[f.name()])
     return gpd.GeoDataFrame(data, crs=layer.crs().authid())
 
+
 def qgsLayerToDataFrame(layer, dtm) -> pd.DataFrame:
     """Convert a vector layer to a pandas DataFrame
     samples the geometry using either points or the vertices of the lines

loopstructural/main/vectorLayerWrapper.py

@@ -17,8 +17,7 @@
 
 
 class LoopstructuralProvider(QgsProcessingProvider):
-    """Processing provider class.
-    """
+    """Processing provider class."""
 
     def loadAlgorithms(self):
         """Loads all algorithms belonging to this provider."""

loopstructural/processing/provider.py

@@ -146,6 +146,8 @@ def log(
                     level=log_level,
                     duration=duration,
                 )
+
+
 class PlgLoggerHandler(logging.Handler):
     """
     Standard logging.Handler that forwards logs to PlgLogger.log().