Added option to set animation class settings

threedi_plugin.py

@@ -167,6 +167,7 @@ def initGui(self):
         self.model.result_checked.connect(self.map_animator.results_changed)
         self.model.result_unchecked.connect(self.map_animator.results_changed)
         self.model.result_added.connect(self.map_animator.results_changed)
+        self.model.result_removed.connect(self.map_animator.results_changed)
         self.temporal_manager.updated.connect(self.map_animator.update_results)
 
         # flow summary signals
@@ -237,6 +238,10 @@ def write(self, doc: QDomDocument) -> bool:
             if not tool.write(doc, node):
                 return False
 
+        # Also allow animator to persist settings
+        if not self.map_animator.write(doc, node):
+            return False
+
         return True
 
     def write_map_layer(self, layer: QgsMapLayer, elem: QDomElement, _: QDomDocument):
@@ -263,6 +268,11 @@ def read(self, doc: QDomDocument) -> bool:
                 self.model.clear()
                 return False
 
+        # Also allow animator to read saved settings
+        if not self.map_animator.read(tool_node):
+            self.model.clear()
+            return False
+
         return True
 
     def unload(self):

tool_animation/animation_styler.py

@@ -1,25 +1,22 @@
 from pathlib import Path
-from typing import List
-import logging
-
-import numpy as np
-
 from qgis.core import QgsFeatureRequest
 from qgis.core import QgsMarkerSymbol
 from qgis.core import QgsProperty
 from qgis.core import QgsSymbolLayer
 from qgis.core import QgsVectorLayer
 from qgis.utils import iface
-
 from threedi_results_analysis.datasource.result_constants import WET_CROSS_SECTION_AREA
+from threedi_results_analysis.utils.color import color_ramp_from_data
 from threedi_results_analysis.utils.color import COLOR_RAMP_OCEAN_CURL
 from threedi_results_analysis.utils.color import COLOR_RAMP_OCEAN_DEEP
 from threedi_results_analysis.utils.color import COLOR_RAMP_OCEAN_HALINE
-from threedi_results_analysis.utils.color import color_ramp_from_data
+from typing import List
+
+import logging
+import numpy as np
+
 
 STYLES_ROOT = Path(__file__).parent / "layer_styles"
-ANIMATION_LAYERS_NR_LEGEND_CLASSES = 24
-assert ANIMATION_LAYERS_NR_LEGEND_CLASSES % 2 == 0
 DEFAULT_LOWER_THRESHOLD = 1e-6
 
 logger = logging.getLogger(__name__)
@@ -174,7 +171,7 @@ def style_animation_node_current(
 
 
 def style_animation_node_difference(
-    lyr: QgsVectorLayer, percentiles: List[float], variable: str, cells: bool, field_postfix=""
+    lyr: QgsVectorLayer, percentiles: List[float], variable: str, cells: bool, nr_of_classes, field_postfix=""
 ):
     """Applies styling to Animation Toolbar node layer in 'difference' mode"""
 
@@ -201,7 +198,7 @@ def style_animation_node_difference(
                     stop=abs_high,
                     step=(
                         (abs_high - abs_high * -1)
-                        / (ANIMATION_LAYERS_NR_LEGEND_CLASSES - 2)
+                        / (nr_of_classes - 2)
                     ),
                 )
             )