From ca116989c7ff6799ac4ce6156c6b497f306d189e Mon Sep 17 00:00:00 2001
From: Julian Weaver <114316166+weavejul@users.noreply.github.com>
Date: Thu, 27 Feb 2025 16:05:01 -0600
Subject: [PATCH] Add functionality to invert colors

Adds a checkbox under the "View" tab that inverts the RGB values of the displayed image (QImage.InvertRGB).

Does not affect trace colors.

Resets to "False" at startup.
---
 .../modules/backend/view/section_layer.py          | 11 +++++++++--
 .../modules/datatypes/default_settings.py          |  1 +
 .../modules/gui/main/field_widget_1_base.py        | 14 +++++++++++++-
 PyReconstruct/modules/gui/main/menubar.py          |  3 ++-
 4 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/PyReconstruct/modules/backend/view/section_layer.py b/PyReconstruct/modules/backend/view/section_layer.py
index af6228a4..3b22623d 100644
--- a/PyReconstruct/modules/backend/view/section_layer.py
+++ b/PyReconstruct/modules/backend/view/section_layer.py
@@ -1,4 +1,4 @@
-from PySide6.QtGui import QPainter, QPixmap
+from PySide6.QtGui import QPainter, QPixmap, QImage
 from PySide6.QtCore import Qt
 
 from .image_layer import ImageLayer
@@ -32,7 +32,8 @@ def generateView(
         generate_traces=True,
         hide_traces=False,
         show_all_traces=False,
-        hide_image=False
+        hide_image=False,
+        invert_image=False
         ):
         """Generate pixmap view for a section.
         
@@ -57,6 +58,12 @@ def generateView(
         elif generate_image:
             self.image_layer = self.generateImageLayer(pixmap_dim, window)
         
+        # Invert base color if requested
+        if invert_image:
+            temp_img = self.image_layer.toImage()
+            temp_img.invertPixels(QImage.InvertRgb)
+            self.image_layer = QPixmap.fromImage(temp_img)
+        
         ## Hide all traces if requested
         if hide_traces:
             return self.image_layer.copy()        
diff --git a/PyReconstruct/modules/datatypes/default_settings.py b/PyReconstruct/modules/datatypes/default_settings.py
index 69279ee9..8fa175c5 100644
--- a/PyReconstruct/modules/datatypes/default_settings.py
+++ b/PyReconstruct/modules/datatypes/default_settings.py
@@ -53,6 +53,7 @@ def get_username() -> str:
     "screenshot_res": 300,
     "show_ztraces": True,  # MFO
     "fill_opacity": 0.2,  # MFO
+    "invert_colors": False,  # MFO
     "find_zoom": 95.0,  # MFO
     "show_flags": "unresolved",  # MFO
     "display_closest": True,  # MFO
diff --git a/PyReconstruct/modules/gui/main/field_widget_1_base.py b/PyReconstruct/modules/gui/main/field_widget_1_base.py
index 3f83d2b2..eef160d6 100644
--- a/PyReconstruct/modules/gui/main/field_widget_1_base.py
+++ b/PyReconstruct/modules/gui/main/field_widget_1_base.py
@@ -127,6 +127,9 @@ def createField(self, series : Series):
         ## Create zarr view if applicable
         self.createZarrLayer()
         
+        # Reset invert color option to false
+        self.series.setOption("invert_colors", False)
+        
         ## Reset b section and layer
         self.b_section = None
         self.b_section_layer = None
@@ -258,6 +261,9 @@ def generateView(self, generate_image=True, generate_traces=True, update=True) -
         assert(abs(x_scaling - y_scaling) < 1e-6)
         
         self.scaling = x_scaling
+        
+        # check if image should be inverted
+        invert_image = self.series.getOption("invert_colors")
 
         ## Generate section view
         view = self.section_layer.generateView(
@@ -267,7 +273,8 @@ def generateView(self, generate_image=True, generate_traces=True, update=True) -
             generate_traces=generate_traces,
             hide_traces=self.hide_trace_layer,
             show_all_traces=self.show_all_traces,
-            hide_image=self.hide_image
+            hide_image=self.hide_image,
+            invert_image=invert_image
         )
 
         # blend b section if requested
@@ -312,6 +319,11 @@ def generateView(self, generate_image=True, generate_traces=True, update=True) -
         if update:
             self.update()
     
+    def toggleInvertColors(self):
+        current = self.series.getOption("invert_colors")
+        self.series.setOption("invert_colors", not current)
+        self.generateView()
+        
     def clearStates(self) -> None:
         """Create/clear the states for each section."""
         self.series_states = SeriesStates(self.series)
diff --git a/PyReconstruct/modules/gui/main/menubar.py b/PyReconstruct/modules/gui/main/menubar.py
index da1df700..3fb3ff05 100644
--- a/PyReconstruct/modules/gui/main/menubar.py
+++ b/PyReconstruct/modules/gui/main/menubar.py
@@ -330,8 +330,9 @@ def return_view_menu(self):
             ("viewmag_act", "View magnification...", "", self.field.setViewMagnification),
             ("findview_act", "Set zoom when finding contours...", "", self.setFindZoom),
             None,
-            ("toggleztraces_act", "Toggle show Z-traces", "", self.toggleZtraces),
+            ("toggleztraces_act", "Toggle show Z-traces", "", self.toggleZtraces),            
             None,
+            ("invertcolors_act", "Invert Colors", "checkbox", self.field.toggleInvertColors),
             {
                 "attr_name": "palettemenu",
                 "text": "Palette",