graxpert 3.2.0a0.dev4__py3-none-any.whl

graxpert/stretch.py

@@ -0,0 +1,151 @@
+import multiprocessing
+multiprocessing.freeze_support()
+
+import logging
+from concurrent.futures import wait
+from multiprocessing import shared_memory
+
+import numpy as np
+
+from graxpert.mp_logging import get_logging_queue, worker_configurer
+from graxpert.parallel_processing import executor
+
+from dataclasses import dataclass
+
+@dataclass
+class MTFStretchParameters:
+    midtone: float
+    shadow_clipping: float
+    highlight_clipping: float = 1.0
+
+class StretchParameters:
+    stretch_option: str
+    bg: float
+    sigma: float
+    do_stretch: bool = True
+    channels_linked: bool = False
+    images_linked: bool = False
+    
+    def __init__(self, stretch_option: str, channels_linked: bool = False, images_linked: bool = False): 
+        self.stretch_option = stretch_option
+        self.channels_linked = channels_linked
+        self. images_linked = images_linked
+        
+        if stretch_option == "No Stretch":
+            self.do_stretch = False
+        
+        elif stretch_option == "10% Bg, 3 sigma":
+            self.bg = 0.1
+            self.sigma = 3.0
+
+        elif stretch_option == "15% Bg, 3 sigma":
+            self.bg = 0.15
+            self.sigma = 3.0
+
+        elif stretch_option == "20% Bg, 3 sigma":
+            self.bg = 0.2
+            self.sigma = 3.0
+
+        elif stretch_option == "30% Bg, 2 sigma":
+            self.bg = 0.3
+            self.sigma = 2.0
+
+    
+
+def stretch(data, stretch_params: StretchParameters):
+    if not stretch_params.do_stretch:
+        return data
+    
+    mtf_stretch_param = calculate_mtf_stretch_parameters_for_image(stretch_params, data)
+    return stretch_all([data], [mtf_stretch_param])[0]
+
+
+def stretch_all(datas, mtf_stretch_params: list[MTFStretchParameters]):
+    
+    futures = []
+    shms = []
+    copies = []
+    result = []
+    logging_queue = get_logging_queue()             
+    
+    for data, mtf_stretch_param in zip(datas, mtf_stretch_params):
+        shm = shared_memory.SharedMemory(create=True, size=data.nbytes)
+        copy = np.ndarray(data.shape, dtype=data.dtype, buffer=shm.buf)
+        np.copyto(copy, data)
+        shms.append(shm)
+        copies.append(copy)
+        
+        for c in range(copy.shape[-1]):
+            futures.insert(c, executor.submit(stretch_channel, shm.name, c, mtf_stretch_param[c], copy.shape, copy.dtype, logging_queue, worker_configurer))
+    wait(futures)
+
+    for copy in copies:
+        copy = np.copy(copy)
+        result.append(copy)
+    
+    for shm in shms:
+        shm.close()
+        shm.unlink()
+    
+    return result
+
+
+def calculate_mtf_stretch_parameters_for_image(stretch_params, image):
+    if stretch_params.channels_linked:
+        mtf_stretch_param = calculate_mtf_stretch_parameters_for_channel(stretch_params, image)
+        return [mtf_stretch_param] * image.shape[-1]
+    
+    else:
+        return [calculate_mtf_stretch_parameters_for_channel(stretch_params, image[:,:,i]) for i in range(image.shape[-1])]
+
+def calculate_mtf_stretch_parameters_for_channel(stretch_params, channel):
+    channel = channel.flatten()[::4]
+    
+    indx_clip = np.logical_and(channel < 1.0, channel > 0.0)
+    median = np.median(channel[indx_clip])
+    mad = np.median(np.abs(channel[indx_clip]-median))
+    
+    shadow_clipping = np.clip(median - stretch_params.sigma*mad, 0, 1.0)
+    highlight_clipping = 1.0
+    midtone = MTF((median-shadow_clipping)/(highlight_clipping - shadow_clipping), stretch_params.bg)
+    
+    return MTFStretchParameters(midtone, shadow_clipping)
+
+
+def stretch_channel(shm_name, c, mtf_stretch_params, shape, dtype, logging_queue, logging_configurer):
+
+    logging_configurer(logging_queue)
+    logging.info("stretch.stretch_channel started")
+
+    existing_shm = shared_memory.SharedMemory(name=shm_name)
+    channels = np.ndarray(shape, dtype, buffer=existing_shm.buf) #[:,:,channel_idx]
+    channel = channels[:,:,c]
+    
+    try:
+        channel[channel <= mtf_stretch_params.shadow_clipping] = 0.0
+        channel[channel >= mtf_stretch_params.highlight_clipping] = 1.0
+
+        indx_inside = np.logical_and(channel > mtf_stretch_params.shadow_clipping, channel < mtf_stretch_params.highlight_clipping)
+
+        channel[indx_inside] = (channel[indx_inside]-mtf_stretch_params.shadow_clipping)/(mtf_stretch_params.highlight_clipping - mtf_stretch_params.shadow_clipping)
+
+        channel = MTF(channel, mtf_stretch_params.midtone)
+
+    except:
+        logging.exception("An error occured while stretching a color channel")
+    finally:
+        existing_shm.close()
+    
+    logging.info("stretch.stretch_channel finished")
+    
+
+def MTF(data, midtone):
+
+    if type(data) is np.ndarray:
+        data[:] = (midtone-1)*data[:]/((2*midtone-1)*data[:]-midtone)
+    else:
+        data = (midtone-1) * data / ((2*midtone-1) * data - midtone)
+
+    return data
+
+    

graxpert/theme/graxpert-dark-blue.json

@@ -0,0 +1,367 @@
+{
+    "CTk": {
+        "fg_color": [
+            "gray95",
+            "gray10"
+        ]
+    },
+    "CTkToplevel": {
+        "fg_color": [
+            "gray95",
+            "gray10"
+        ]
+    },
+    "CTkFrame": {
+        "corner_radius": 6,
+        "border_width": 0,
+        "fg_color": [
+            "gray90",
+            "gray13"
+        ],
+        "top_fg_color": [
+            "gray85",
+            "gray16"
+        ],
+        "border_color": [
+            "gray65",
+            "gray28"
+        ]
+    },
+    "CTkButton": {
+        "corner_radius": 6,
+        "border_width": 0,
+        "fg_color": [
+            "#3a7ebf",
+            "#1f538d"
+        ],
+        "hover_color": [
+            "#325882",
+            "#14375e"
+        ],
+        "border_color": [
+            "#3E454A",
+            "#949A9F"
+        ],
+        "text_color": [
+            "#DCE4EE",
+            "#DCE4EE"
+        ],
+        "text_color_disabled": [
+            "gray74",
+            "gray60"
+        ]
+    },
+    "Accent.CTkButton": {
+        "fg_color": "#247f4c",
+        "hover_color": "#154b2d"
+    },
+    "Help.CTkButton": {
+        "fg_color": "#c46f1a",
+        "hover_color": "#73410f"
+    },
+    "CTkLabel": {
+        "corner_radius": 0,
+        "fg_color": "transparent",
+        "text_color": [
+            "gray14",
+            "gray84"
+        ]
+    },
+    "CTkEntry": {
+        "corner_radius": 6,
+        "border_width": 2,
+        "fg_color": [
+            "#F9F9FA",
+            "#343638"
+        ],
+        "border_color": [
+            "#979DA2",
+            "#565B5E"
+        ],
+        "text_color": [
+            "gray14",
+            "gray84"
+        ],
+        "placeholder_text_color": [
+            "gray52",
+            "gray62"
+        ]
+    },
+    "CTkCheckBox": {
+        "corner_radius": 6,
+        "border_width": 3,
+        "fg_color": [
+            "#3a7ebf",
+            "#1f538d"
+        ],
+        "border_color": [
+            "#3E454A",
+            "#949A9F"
+        ],
+        "hover_color": [
+            "#325882",
+            "#14375e"
+        ],
+        "checkmark_color": [
+            "#DCE4EE",
+            "gray90"
+        ],
+        "text_color": [
+            "gray14",
+            "gray84"
+        ],
+        "text_color_disabled": [
+            "gray60",
+            "gray45"
+        ]
+    },
+    "CTkSwitch": {
+        "corner_radius": 1000,
+        "border_width": 3,
+        "button_length": 0,
+        "fg_color": [
+            "#939BA2",
+            "#4A4D50"
+        ],
+        "progress_color": [
+            "#3a7ebf",
+            "#1f538d"
+        ],
+        "button_color": [
+            "gray36",
+            "#D5D9DE"
+        ],
+        "button_hover_color": [
+            "gray20",
+            "gray100"
+        ],
+        "text_color": [
+            "gray14",
+            "gray84"
+        ],
+        "text_color_disabled": [
+            "gray60",
+            "gray45"
+        ]
+    },
+    "CTkRadioButton": {
+        "corner_radius": 1000,
+        "border_width_checked": 6,
+        "border_width_unchecked": 3,
+        "fg_color": [
+            "#3a7ebf",
+            "#1f538d"
+        ],
+        "border_color": [
+            "#3E454A",
+            "#949A9F"
+        ],
+        "hover_color": [
+            "#325882",
+            "#14375e"
+        ],
+        "text_color": [
+            "gray14",
+            "gray84"
+        ],
+        "text_color_disabled": [
+            "gray60",
+            "gray45"
+        ]
+    },
+    "CTkProgressBar": {
+        "corner_radius": 1000,
+        "border_width": 0,
+        "fg_color": [
+            "#939BA2",
+            "#4A4D50"
+        ],
+        "progress_color": [
+            "#3a7ebf",
+            "#1f538d"
+        ],
+        "border_color": [
+            "gray",
+            "gray"
+        ]
+    },
+    "CTkSlider": {
+        "corner_radius": 1000,
+        "button_corner_radius": 1000,
+        "border_width": 6,
+        "button_length": 0,
+        "fg_color": [
+            "#939BA2",
+            "#4A4D50"
+        ],
+        "progress_color": [
+            "gray40",
+            "#AAB0B5"
+        ],
+        "button_color": [
+            "#3a7ebf",
+            "#1f538d"
+        ],
+        "button_hover_color": [
+            "#325882",
+            "#14375e"
+        ]
+    },
+    "CTkOptionMenu": {
+        "corner_radius": 6,
+        "fg_color": [
+            "#3a7ebf",
+            "#1f538d"
+        ],
+        "button_color": [
+            "#325882",
+            "#14375e"
+        ],
+        "button_hover_color": [
+            "#234567",
+            "#1e2c40"
+        ],
+        "text_color": [
+            "#DCE4EE",
+            "#DCE4EE"
+        ],
+        "text_color_disabled": [
+            "gray74",
+            "gray60"
+        ]
+    },
+    "CTkComboBox": {
+        "corner_radius": 6,
+        "border_width": 2,
+        "fg_color": [
+            "#F9F9FA",
+            "#343638"
+        ],
+        "border_color": [
+            "#979DA2",
+            "#565B5E"
+        ],
+        "button_color": [
+            "#979DA2",
+            "#565B5E"
+        ],
+        "button_hover_color": [
+            "#6E7174",
+            "#7A848D"
+        ],
+        "text_color": [
+            "gray14",
+            "gray84"
+        ],
+        "text_color_disabled": [
+            "gray50",
+            "gray45"
+        ]
+    },
+    "CTkScrollbar": {
+        "corner_radius": 1000,
+        "border_spacing": 4,
+        "fg_color": "transparent",
+        "button_color": [
+            "gray55",
+            "gray41"
+        ],
+        "button_hover_color": [
+            "gray40",
+            "gray53"
+        ]
+    },
+    "CTkSegmentedButton": {
+        "corner_radius": 6,
+        "border_width": 2,
+        "fg_color": [
+            "#979DA2",
+            "gray29"
+        ],
+        "selected_color": [
+            "#3a7ebf",
+            "#1f538d"
+        ],
+        "selected_hover_color": [
+            "#325882",
+            "#14375e"
+        ],
+        "unselected_color": [
+            "#979DA2",
+            "gray29"
+        ],
+        "unselected_hover_color": [
+            "gray70",
+            "gray41"
+        ],
+        "text_color": [
+            "#DCE4EE",
+            "#DCE4EE"
+        ],
+        "text_color_disabled": [
+            "gray74",
+            "gray60"
+        ]
+    },
+    "CTkTextbox": {
+        "corner_radius": 6,
+        "border_width": 0,
+        "fg_color": [
+            "gray100",
+            "gray20"
+        ],
+        "border_color": [
+            "#979DA2",
+            "#565B5E"
+        ],
+        "text_color": [
+            "gray14",
+            "gray84"
+        ],
+        "scrollbar_button_color": [
+            "gray55",
+            "gray41"
+        ],
+        "scrollbar_button_hover_color": [
+            "gray40",
+            "gray53"
+        ]
+    },
+    "CTkScrollableFrame": {
+        "label_fg_color": [
+            "gray80",
+            "gray21"
+        ]
+    },
+    "DropdownMenu": {
+        "fg_color": [
+            "gray90",
+            "gray20"
+        ],
+        "hover_color": [
+            "gray75",
+            "gray28"
+        ],
+        "text_color": [
+            "gray14",
+            "gray84"
+        ]
+    },
+    "CTkFont": {
+        "macOS": {
+            "family": "SF Display",
+            "size": 13,
+            "weight": "normal"
+        },
+        "Windows": {
+            "family": "Roboto",
+            "size": 13,
+            "weight": "normal"
+        },
+        "Linux": {
+            "family": "Roboto",
+            "size": 13,
+            "weight": "normal"
+        }
+    }
+}

graxpert/ui/__init__.py

@@ -0,0 +1 @@
+

graxpert/ui/application_frame.py

@@ -0,0 +1,115 @@
+import tkinter as tk
+
+from customtkinter import CTkFrame
+
+from graxpert.application.app import graxpert
+from graxpert.application.app_events import AppEvents
+from graxpert.application.eventbus import eventbus
+from graxpert.commands import InitHandler
+from graxpert.ui.canvas import Canvas
+from graxpert.ui.left_menu import LeftMenu
+from graxpert.ui.right_menu import AdvancedFrame, HelpFrame
+from graxpert.ui.statusbar import StatusBar
+from graxpert.ui.ui_events import UiEvents
+from graxpert.ui.widgets import default_label_width, padx
+
+
+class ApplicationFrame(CTkFrame):
+    def __init__(self, master, **kwargs):
+        super().__init__(master, **kwargs)
+
+        self.initial_title = master.title()
+        self.show_help = False
+        self.show_advanced = False
+
+        self.create_children()
+        self.setup_layout()
+        self.place_children()
+        self.create_bindings()
+        self.register_events()
+
+    def create_children(self):
+        self.left_menu = LeftMenu(self, fg_color="transparent", width=default_label_width + padx + 16)
+        self.canvas = Canvas(self)
+        self.help_frame = HelpFrame(self, fg_color="transparent", width=300)
+        self.advanced_frame = AdvancedFrame(self, fg_color="transparent", width=300)
+        self.statusbar_frame = StatusBar(self, fg_color="transparent")
+
+    def setup_layout(self):
+        self.columnconfigure(0, weight=0)
+        self.columnconfigure(1, weight=100)
+        self.columnconfigure(2, weight=0)
+        self.rowconfigure(0, weight=1)
+        self.rowconfigure(1, weight=0)
+
+    def place_children(self):
+        self.left_menu.grid(column=0, row=0, rowspan=2, ipadx=padx, sticky=tk.NS)
+        self.canvas.grid(column=1, row=0, sticky=tk.NSEW)
+        self.statusbar_frame.grid(column=1, row=1, sticky=tk.NSEW)
+
+    def create_bindings(self):
+        self.master.bind("<BackSpace>", lambda e: eventbus.emit(UiEvents.RESET_ZOOM_REQUEST))  # backspace -> reset zoom
+        self.master.bind("<Control-Key-0>", lambda e: eventbus.emit(UiEvents.RESET_ZOOM_REQUEST))  # ctrl + 0 -> reset zoom (Windows)
+        self.master.bind("<Command-Key-0>", lambda e: eventbus.emit(UiEvents.RESET_ZOOM_REQUEST))  # cmd  + 0 -> reset zoom (Mac)
+        self.master.bind("<Control-Key-KP_0>", lambda e: eventbus.emit(UiEvents.RESET_ZOOM_REQUEST))  # ctrl + numpad 0 -> reset zoom (Windows)
+        self.master.bind("<Command-Key-KP_0>", lambda e: eventbus.emit(UiEvents.RESET_ZOOM_REQUEST))  # cmd  + numpad 0 -> reset zoom (Mac)
+        self.master.bind("<Control-l>", lambda e: eventbus.emit(AppEvents.OPEN_FILE_DIALOG_REQUEST))
+        self.master.bind("<Command-l>", lambda e: eventbus.emit(AppEvents.OPEN_FILE_DIALOG_REQUEST))
+        self.master.bind("<Control-c>", lambda e: eventbus.emit(AppEvents.CALCULATE_REQUEST))
+        self.master.bind("<Command-c>", lambda e: eventbus.emit(AppEvents.CALCULATE_REQUEST))
+        self.master.bind("<Control-d>", lambda e: eventbus.emit(AppEvents.DENOISE_REQUEST))
+        self.master.bind("<Command-d>", lambda e: eventbus.emit(AppEvents.DENOISE_REQUEST))
+        self.master.bind("<Control-s>", lambda e: eventbus.emit(AppEvents.SAVE_REQUEST))
+        self.master.bind("<Command-s>", lambda e: eventbus.emit(AppEvents.SAVE_REQUEST))
+        self.master.bind("<Control-z>", self.undo)  # undo
+        self.master.bind("<Control-y>", self.redo)  # redo
+        self.master.bind("<Command-z>", self.undo)  # undo on macs
+        self.master.bind("<Command-y>", self.redo)  # redo on macs
+
+    def register_events(self):
+        eventbus.add_listener(UiEvents.HELP_FRAME_TOGGLED, self.toggle_help)
+        eventbus.add_listener(UiEvents.ADVANCED_FRAME_TOGGLED, self.toggle_advanced)
+        eventbus.add_listener(AppEvents.LOAD_IMAGE_END, self.on_load_image_end)
+
+    def place_right_frame(self):
+        self.help_frame.grid_forget()
+        self.advanced_frame.grid_forget()
+
+        if self.show_help:
+            self.help_frame.grid(column=2, row=0, rowspan=2, sticky=tk.NSEW)
+
+        if self.show_advanced:
+            self.advanced_frame.grid(column=2, row=0, rowspan=2, sticky=tk.NSEW)
+
+    # event handling
+    def on_load_image_end(self, event):
+        self.master.title(f'{self.initial_title} - {event["filename"]}')
+
+    def redo(self, event):
+        if graxpert.cmd.next is not None:
+            redo = graxpert.cmd.redo()
+            graxpert.cmd = redo
+            eventbus.emit(AppEvents.REDRAW_POINTS_REQUEST)
+
+    # widget logic
+    def toggle_help(self, event):
+        if self.show_help:
+            self.show_help = False
+        else:
+            self.show_advanced = False
+            self.show_help = True
+        self.place_right_frame()
+
+    def toggle_advanced(self, event):
+        if self.show_advanced:
+            self.show_advanced = False
+        else:
+            self.show_help = False
+            self.show_advanced = True
+        self.place_right_frame()
+
+    def undo(self, event):
+        if not type(graxpert.cmd.handler) is InitHandler:
+            undo = graxpert.cmd.undo()
+            graxpert.cmd = undo
+            eventbus.emit(AppEvents.REDRAW_POINTS_REQUEST)