creatumlibre.ui.dialogs.color_adjustment_dialog

  1from PyQt6.QtCore import Qt
  2from PyQt6.QtWidgets import (
  3    QDialog,
  4    QHBoxLayout,
  5    QLabel,
  6    QPushButton,
  7    QSlider,
  8    QVBoxLayout,
  9)
 10
 11from creatumlibre.graphics.filters.enhancers import (
 12    adjust_brightness,
 13    adjust_contrast,
 14    adjust_rgb,
 15    adjust_saturation,
 16)
 17from creatumlibre.ui.dialogs.color_adjustment_dialog_css import (
 18    BTN_APPLY,
 19    BTN_CANCEL,
 20    MAIN_DIALOG,
 21)
 22
 23
 24class ColorAdjustmentDialog(QDialog):
 25    """Dialog for adjusting brightness, saturation, contrast, and color balance."""
 26
 27    def __init__(self, parent, tab_manager):
 28        super().__init__(parent)
 29        self.tab_manager = tab_manager  # Reference to the tab manager
 30        self.apply_button = QPushButton("Apply")
 31        self.cancel_button = QPushButton("Cancel")
 32        self.sliders = {}
 33        self.labels = {}
 34        self.base_image_snapshot = None  # unchanged image
 35
 36        self._init_layout()
 37
 38    def _init_layout(self):
 39
 40        self.setWindowTitle("Color Adjustments")
 41        self.setFixedSize(300, 480)
 42        self.setStyleSheet(MAIN_DIALOG)
 43
 44        button_layout = QHBoxLayout()
 45
 46        self.apply_button.setStyleSheet(BTN_APPLY)
 47        self.apply_button.clicked.connect(self.apply_changes)
 48        button_layout.addWidget(self.apply_button)
 49
 50        self.cancel_button.setStyleSheet(BTN_CANCEL)
 51        self.cancel_button.clicked.connect(self.cancel_changes)
 52        button_layout.addWidget(self.cancel_button)
 53
 54        self.slider_settings = {
 55            "Brightness": {
 56                "default": 0,
 57                "range": (-100, 100),
 58                "tick_interval": 10,
 59                "scale_function": lambda value: 1 + (value / 100 * 1),
 60                "apply_function": adjust_brightness,
 61            },
 62            "Saturation": {
 63                "default": 0,
 64                "range": (-100, 100),
 65                "tick_interval": 10,
 66                "scale_function": lambda value: 1 + (value / 100 * 2),
 67                "apply_function": adjust_saturation,
 68            },
 69            "Contrast": {
 70                "default": 0,
 71                "range": (-100, 100),
 72                "tick_interval": 10,
 73                "scale_function": lambda value: 1 + (value / 100 * 1.0),
 74                "apply_function": adjust_contrast,
 75            },
 76            "Red": {
 77                "default": 0,
 78                "range": (-100, 100),
 79                "tick_interval": 10,
 80                "scale_function": lambda value: value / 100,
 81                "apply_function": lambda img, value: adjust_rgb(img, value, "Red"),
 82            },
 83            "Green": {
 84                "default": 0,
 85                "range": (-100, 100),
 86                "tick_interval": 10,
 87                "scale_function": lambda value: value / 100,
 88                "apply_function": lambda img, value: adjust_rgb(img, value, "Green"),
 89            },
 90            "Blue": {
 91                "default": 0,
 92                "range": (-100, 100),
 93                "tick_interval": 10,
 94                "scale_function": lambda value: value / 100,
 95                "apply_function": lambda img, value: adjust_rgb(img, value, "Blue"),
 96            },
 97        }
 98
 99        dialog_layout = QVBoxLayout(self)
100
101        for adjustment, config in self.slider_settings.items():
102            label = QLabel(
103                adjustment
104            )  #  This will be replaced with translation support later
105            self.labels[adjustment] = label
106            slider = QSlider(Qt.Orientation.Horizontal)
107            slider.setTickPosition(QSlider.TickPosition.TicksBelow)
108            slider.setTickInterval(config["tick_interval"])
109            slider.setRange(*config["range"])  #  Uses dict values
110            slider.setValue(config["default"])
111            self.labels[adjustment].setText(f"{adjustment} [{slider.value():.2f}]")
112
113            slider.valueChanged.connect(self.apply_all_adjustments)
114
115            dialog_layout.addWidget(label)
116            dialog_layout.addWidget(slider)
117            self.sliders[adjustment] = slider
118
119        dialog_layout.addLayout(button_layout)  # add buttons
120        self.setLayout(dialog_layout)
121
122    def showEvent(self, event):
123        """Runs logic when the dialog is shown."""
124        super().showEvent(event)
125
126        promoted = self.get_promoted_object()
127        if promoted:
128            self.base_image_snapshot = promoted.get_image().copy()
129            self.apply_all_adjustments()
130
131            return  # Already has a promoted object
132
133        # Promote base image if no selection is active
134        active_tab = self.tab_manager.get_active_tab()
135        if not active_tab:
136            return
137
138        manager = active_tab.get("manager")
139        base_object = manager.get_base_object()
140
141        if base_object:
142            # Promote full image as fallback
143            height, width = base_object.get_image().shape[:2]
144            fake_selection = (0, 0, width, height)
145
146            # Make RegionManager believe the full image is selected
147            base_object.region_manager.set_bounding_rect(*fake_selection)
148            base_object.region_manager.initialize_mask(base_object.get_image().shape)
149
150            new_obj = base_object.extract_selection_as_new_image()
151            if new_obj:
152                manager.add_object(new_obj)
153                self.base_image_snapshot = new_obj.get_image().copy()
154                self.apply_all_adjustments()
155                self.tab_manager.refresh_tab_display(
156                    self.tab_manager.get_active_tab_index()
157                )
158
159    def apply_all_adjustments(self):
160        promoted = self.get_promoted_object()
161        if promoted is None:
162            return
163
164        image = self.base_image_snapshot.copy()  # Work on a copy for preview
165
166        for name, config in self.slider_settings.items():
167            value = self.sliders[name].value()
168            scale = config["scale_function"](value)
169            image = config["apply_function"](image, scale)
170            mapped_value = config["scale_function"](value)
171            self.labels[name].setText(f"{name} [{mapped_value:.2f}]")
172
173        promoted.set_image(image)
174
175        self.tab_manager.refresh_tab_display(self.tab_manager.get_active_tab_index())
176
177    def get_promoted_object(self):
178        """Finds the promoted object in the current tab's object manager."""
179        active_tab = self.tab_manager.get_active_tab()
180        if not active_tab:
181            return None
182
183        object_manager = active_tab.get("manager")
184        if not object_manager:
185            return None
186
187        for obj in object_manager.object_list:
188            if getattr(obj, "is_promoted", False):
189                return obj
190        return None
191
192    def apply_changes(self):
193        obj_manager = self.tab_manager.get_active_tab().get("manager")
194        obj_manager.merge_selection()
195        self.tab_manager.refresh_tab_display(self.tab_manager.get_active_tab_index())
196        self.accept()
197
198    def cancel_changes(self):
199        """Cancel adjustments, delete the promoted layer"""
200        promoted = self.get_promoted_object()
201        if promoted:
202            active_tab = self.tab_manager.get_active_tab()
203            object_manager = active_tab.get("manager")
204            object_manager.object_list.remove(promoted)
205            self.tab_manager.refresh_tab_display(
206                self.tab_manager.get_active_tab_index()
207            )
208            self.reject()