46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
from PyQt6.QtWidgets import QVBoxLayout, QLabel, QSlider
|
|
from PyQt6.QtCore import Qt
|
|
from .ImageParameterDialog import ImageParameterDialog
|
|
import ImageProcessingWorker
|
|
import numpy as np
|
|
|
|
class TemperatureAdjustmentDialog(ImageParameterDialog):
|
|
def __init__(self, image):
|
|
super().__init__(image, ImageProcessingWorker.RGBImageProcessingWorker)
|
|
self.setWindowTitle("Temperature Adjustment")
|
|
self.layout = QVBoxLayout()
|
|
|
|
self.temperature_slider = QSlider(Qt.Orientation.Horizontal)
|
|
self.temperature_slider.setRange(-500, 500)
|
|
self.temperature_slider.setValue(0)
|
|
self.temperature_slider.setTickInterval(10)
|
|
self.temperature_slider.setTickPosition(QSlider.TickPosition.TicksBelow)
|
|
|
|
self.label = QLabel("Temperature: 0")
|
|
|
|
self.temperature_slider.valueChanged.connect(self.update_temperature)
|
|
|
|
self.layout.addWidget(self.label)
|
|
self.layout.addWidget(self.temperature_slider)
|
|
self.layout.addWidget(self.button_box)
|
|
self.setLayout(self.layout)
|
|
|
|
def update_temperature(self, value):
|
|
self.label.setText(f"Temperature: {value}")
|
|
self.send_to_process({'temperature': value})
|
|
|
|
def process_image(self, image, values):
|
|
temperature = values.get('temperature', 0)
|
|
|
|
increment = temperature * 0.1
|
|
|
|
image[..., 2] += increment * -1
|
|
image[..., 0] += increment
|
|
|
|
image = np.clip(image, 0, 255)
|
|
|
|
return image.astype(np.uint8)
|
|
|
|
@classmethod
|
|
def dialog_name(cls):
|
|
return "Temperature Adjustment" |