spacr 0.2.41__py3-none-any.whl → 0.2.46__py3-none-any.whl

spacr/gui_elements.py

@@ -287,6 +287,73 @@ class spacrDropdownMenu(tk.OptionMenu):
         # Create custom button
         self.create_custom_button()
 
+    def create_custom_button(self):
+        self.canvas_width = self.winfo_reqwidth() + 20  # Adjust width based on required size
+        self.canvas_height = 40  # Adjust the height as needed
+        self.canvas = tk.Canvas(self, width=self.canvas_width, height=self.canvas_height, bd=0, highlightthickness=0, relief='ridge', bg='#2B2B2B')
+        self.canvas.pack()
+        self.label = tk.Label(self.canvas, text="Settings Category", bg='#2B2B2B', fg='#ffffff', font=('Arial', 12))
+        self.label.place(relx=0.5, rely=0.5, anchor=tk.CENTER)
+        self.draw_rounded_rectangle('#2B2B2B')
+
+        # Bind the click event to open the dropdown menu
+        self.canvas.bind("<Button-1>", self.on_click)
+        self.label.bind("<Button-1>", self.on_click)
+
+    def draw_rounded_rectangle(self, color):
+        radius = 15
+        x0, y0 = 10, 5
+        x1, y1 = self.canvas_width - 10, self.canvas_height - 5  # Adjust based on canvas size
+
+        self.canvas.delete("all")
+
+        # Create the rounded rectangle
+        self.canvas.create_arc(x0, y0, x0 + 2 * radius, y0 + 2 * radius, start=90, extent=90, fill=color, outline=color)
+        self.canvas.create_arc(x1 - 2 * radius, y0, x1, y0 + 2 * radius, start=0, extent=90, fill=color, outline=color)
+        self.canvas.create_arc(x0, y1 - 2 * radius, x0 + 2 * radius, y1, start=180, extent=90, fill=color, outline=color)
+        self.canvas.create_arc(x1 - 2 * radius, y1 - 2 * radius, x1, y1, start=270, extent=90, fill=color, outline=color)
+
+        self.canvas.create_rectangle(x0 + radius, y0, x1 - radius, y1, fill=color, outline=color)
+        self.canvas.create_rectangle(x0, y0 + radius, x1, y1 - radius, fill=color, outline=color)
+
+        self.label.config(bg=color)  # Update label background to match rectangle color
+
+    def on_click(self, event):
+        self.post_menu()
+
+    def post_menu(self):
+        x, y, width, height = self.winfo_rootx(), self.winfo_rooty(), self.winfo_width(), self.winfo_height()
+        self.menu.post(x, y + height)
+
+    def update_styles(self, active_categories=None):
+        style = ttk.Style()
+        style_out = set_dark_style(style, widgets=[self])
+        self.menu = self['menu']
+        style_out = set_dark_style(style, widgets=[self.menu])
+
+        if active_categories is not None:
+            for idx in range(self.menu.index("end") + 1):
+                option = self.menu.entrycget(idx, "label")
+                if option in active_categories:
+                    self.menu.entryconfig(idx, background=style_out['active_color'], foreground=style_out['fg_color'])
+                else:
+                    self.menu.entryconfig(idx, background=style_out['bg_color'], foreground=style_out['fg_color'])
+
+
+
+class spacrDropdownMenu_v1(tk.OptionMenu):
+    def __init__(self, parent, variable, options, command=None, **kwargs):
+        self.variable = variable
+        self.variable.set("Settings Category")
+        super().__init__(parent, self.variable, *options, command=command, **kwargs)
+        self.update_styles()
+
+        # Hide the original button
+        self.configure(highlightthickness=0, relief='flat', bg='#2B2B2B', fg='#2B2B2B')
+
+        # Create custom button
+        self.create_custom_button()
+
     def create_custom_button(self):
         self.canvas_width = self.winfo_reqwidth()  # Use the required width of the widget
         self.canvas_height = 40  # Adjust the height as needed
@@ -374,19 +441,26 @@ class spacrProgressBar(ttk.Progressbar):
         # Track whether to show the progress label
         self.label = label
 
+        # Create the progress label (defer placement)
         if self.label:
-            # Create the progress label
             self.progress_label = tk.Label(parent, text="Processing: 0/0", anchor='w', justify='left', bg=self.inactive_color, fg=self.fg_color)
-            self.progress_label.grid(row=1, column=0, columnspan=2, pady=5, padx=5, sticky='ew')
+            self.progress_label.grid_forget()  # Temporarily hide it
+
+        # Initialize attributes for time and operation
+        self.operation_type = None
+        self.time_image = None
+        self.time_batch = None
+        self.time_left = None
 
-            # Initialize attributes for time and operation
-            self.operation_type = None
-            self.time_image = None
-            self.time_batch = None
-            self.time_left = None
+    def set_label_position(self):
+        if self.label and self.progress_label:
+            row_info = self.grid_info().get('row', 0)
+            col_info = self.grid_info().get('column', 0)
+            col_span = self.grid_info().get('columnspan', 1)
+            self.progress_label.grid(row=row_info + 1, column=col_info, columnspan=col_span, pady=5, padx=5, sticky='ew')
 
     def update_label(self):
-        if self.label:
+        if self.label and self.progress_label:
             # Update the progress label with current progress and additional info
             label_text = f"Processing: {self['value']}/{self['maximum']}"
             if self.operation_type:
@@ -399,44 +473,120 @@ class spacrProgressBar(ttk.Progressbar):
                 label_text += f", Time_left: {self.time_left:.3f} min"
             self.progress_label.config(text=label_text)
 
+def spacrScrollbarStyle(style, inactive_color, active_color):
+    # Check if custom elements already exist to avoid duplication
+    if not style.element_names().count('custom.Vertical.Scrollbar.trough'):
+        style.element_create('custom.Vertical.Scrollbar.trough', 'from', 'clam')
+    if not style.element_names().count('custom.Vertical.Scrollbar.thumb'):
+        style.element_create('custom.Vertical.Scrollbar.thumb', 'from', 'clam')
+
+    style.layout('Custom.Vertical.TScrollbar',
+                 [('Vertical.Scrollbar.trough', {'children': [('Vertical.Scrollbar.thumb', {'expand': '1', 'sticky': 'nswe'})], 'sticky': 'ns'})])
+
+    style.configure('Custom.Vertical.TScrollbar',
+                    background=inactive_color,
+                    troughcolor=inactive_color,
+                    bordercolor=inactive_color,
+                    lightcolor=inactive_color,
+                    darkcolor=inactive_color)
+
+    style.map('Custom.Vertical.TScrollbar',
+              background=[('!active', inactive_color), ('active', active_color)],
+              troughcolor=[('!active', inactive_color), ('active', inactive_color)],
+              bordercolor=[('!active', inactive_color), ('active', inactive_color)],
+              lightcolor=[('!active', inactive_color), ('active', active_color)],
+              darkcolor=[('!active', inactive_color), ('active', active_color)])
+
 class spacrFrame(ttk.Frame):
-    def __init__(self, container, width=None, *args, bg='black', **kwargs):
+    def __init__(self, container, width=None, *args, bg='black', radius=20, scrollbar=True, textbox=False, **kwargs):
         super().__init__(container, *args, **kwargs)
         self.configure(style='TFrame')
         if width is None:
             screen_width = self.winfo_screenwidth()
             width = screen_width // 4
-        canvas = tk.Canvas(self, bg=bg, width=width)
-        scrollbar = ttk.Scrollbar(self, orient="vertical", command=canvas.yview)
+
+        # Create the canvas
+        canvas = tk.Canvas(self, bg=bg, width=width, highlightthickness=0)
+        self.rounded_rectangle(canvas, 0, 0, width, self.winfo_screenheight(), radius, fill=bg)
+
+        # Define scrollbar styles
+        style_out = set_dark_style(ttk.Style())
+        self.inactive_color = style_out['inactive_color']
+        self.active_color = style_out['active_color']
+        self.fg_color = style_out['fg_color']  # Foreground color for text
+
+        # Set custom scrollbar style
+        style = ttk.Style()
+        spacrScrollbarStyle(style, self.inactive_color, self.active_color)
+
+        # Create scrollbar with custom style if scrollbar option is True
+        if scrollbar:
+            scrollbar_widget = ttk.Scrollbar(self, orient="vertical", command=canvas.yview, style='Custom.Vertical.TScrollbar')
+        
+        if textbox:
+            self.scrollable_frame = tk.Text(canvas, bg=bg, fg=self.fg_color, wrap=tk.WORD)
+        else:
+            self.scrollable_frame = ttk.Frame(canvas, style='TFrame')
         
-        self.scrollable_frame = ttk.Frame(canvas, style='TFrame')
         self.scrollable_frame.bind(
             "<Configure>",
             lambda e: canvas.configure(scrollregion=canvas.bbox("all"))
         )
         canvas.create_window((0, 0), window=self.scrollable_frame, anchor="nw")
-        canvas.configure(yscrollcommand=scrollbar.set)
+        if scrollbar:
+            canvas.configure(yscrollcommand=scrollbar_widget.set)
         
         canvas.grid(row=0, column=0, sticky="nsew")
-        scrollbar.grid(row=0, column=1, sticky="ns")
+        if scrollbar:
+            scrollbar_widget.grid(row=0, column=1, sticky="ns")
 
         self.grid_rowconfigure(0, weight=1)
         self.grid_columnconfigure(0, weight=1)
-        self.grid_columnconfigure(1, weight=0)
+        if scrollbar:
+            self.grid_columnconfigure(1, weight=0)
         
-        style = ttk.Style()
-        _ = set_dark_style(style, containers=[self], widgets=[canvas, scrollbar, self.scrollable_frame])
+        _ = set_dark_style(style, containers=[self], widgets=[canvas, self.scrollable_frame])
+        if scrollbar:
+            _ = set_dark_style(style, widgets=[scrollbar_widget])
+
+    def rounded_rectangle(self, canvas, x1, y1, x2, y2, radius=20, **kwargs):
+        points = [
+            x1 + radius, y1,
+            x2 - radius, y1,
+            x2 - radius, y1,
+            x2, y1,
+            x2, y1 + radius,
+            x2, y2 - radius,
+            x2, y2 - radius,
+            x2, y2,
+            x2 - radius, y2,
+            x1 + radius, y2,
+            x1 + radius, y2,
+            x1, y2,
+            x1, y2 - radius,
+            x1, y2 - radius,
+            x1, y1 + radius,
+            x1, y1 + radius,
+            x1, y1
+        ]
+        return canvas.create_polygon(points, **kwargs, smooth=True)
 
 class spacrLabel(tk.Frame):
-    def __init__(self, parent, text="", font=None, style=None, align="right", **kwargs):
+    def __init__(self, parent, text="", font=None, style=None, align="right", height=None, **kwargs):
         valid_kwargs = {k: v for k, v in kwargs.items() if k not in ['foreground', 'background', 'font', 'anchor', 'justify', 'wraplength']}
         super().__init__(parent, **valid_kwargs)
         
         self.text = text
         self.align = align
-        screen_height = self.winfo_screenheight()
-        label_height = screen_height // 50
-        label_width = label_height * 10
+
+        if height is None:
+            screen_height = self.winfo_screenheight()
+            label_height = screen_height // 50
+            label_width = label_height * 10
+        else:
+            label_height = height
+            label_width = label_height * 10
+
         style_out = set_dark_style(ttk.Style())
 
         self.canvas = tk.Canvas(self, width=label_width, height=label_height, highlightthickness=0, bg=style_out['bg_color'])
@@ -471,7 +621,7 @@ class spacrLabel(tk.Frame):
             self.canvas.itemconfig(self.label_text, text=text)
 
 class spacrButton(tk.Frame):
-    def __init__(self, parent, text="", command=None, font=None, icon_name=None, size=50, show_text=True, outline=False, *args, **kwargs):
+    def __init__(self, parent, text="", command=None, font=None, icon_name=None, size=50, show_text=True, outline=False, animation=True, *args, **kwargs):
         super().__init__(parent, *args, **kwargs)
         
         self.text = text.capitalize()  # Capitalize only the first letter of the text
@@ -480,6 +630,7 @@ class spacrButton(tk.Frame):
         self.size = size
         self.show_text = show_text
         self.outline = outline
+        self.animation = animation  # Add animation attribute
 
         style_out = set_dark_style(ttk.Style())
 
@@ -546,13 +697,13 @@ class spacrButton(tk.Frame):
     def on_enter(self, event=None):
         self.canvas.itemconfig(self.button_bg, fill=self.active_color)
         self.update_description(event)
-        if not self.is_zoomed_in:
+        if self.animation and not self.is_zoomed_in:
             self.animate_zoom(0.85)  # Zoom in the icon to 85% of button size
 
     def on_leave(self, event=None):
         self.canvas.itemconfig(self.button_bg, fill=self.inactive_color)
         self.clear_description(event)
-        if self.is_zoomed_in:
+        if self.animation and self.is_zoomed_in:
             self.animate_zoom(0.65)  # Reset the icon size to 65% of button size
 
     def on_click(self, event=None):