Milling square holes copy 2.py

import tkinter as tk
from tkinter import ttk
from tkinter import filedialog, messagebox

class CNCMillingApp:
    def __init__(self, root):
        self.root = root
        self.root.title("CNC Milling App")
        
        self.create_ui()
        
        self.absolute_mode = False
        self.tool_diameter = 3.0
        self.milling_speed = 500  # Default milling speed in mm/min
        self.hole_depth = 10.0  # Default hole depth in mm
        self.layer_height = 1.0  # Default layer height in mm

    def create_ui(self):
        self.dimensions_label = ttk.Label(self.root, text="Rectangle Dimensions (X, Y):")
        self.dimensions_entry = ttk.Entry(self.root)
        self.tool_diameter_label = ttk.Label(self.root, text="Tool Diameter:")
        self.tool_diameter_entry = ttk.Entry(self.root)
        self.speed_label = ttk.Label(self.root, text="Milling Speed:")
        self.speed_entry = ttk.Entry(self.root)
        self.depth_label = ttk.Label(self.root, text="Hole Depth:")
        self.depth_entry = ttk.Entry(self.root)
        self.layer_height_label = ttk.Label(self.root, text="Layer Height:")
        self.layer_height_entry = ttk.Entry(self.root)
        self.absolute_mode_var = tk.BooleanVar(value=False)
        self.absolute_mode_checkbox = ttk.Checkbutton(self.root, text="Absolute Mode",
                                                      variable=self.absolute_mode_var)
        self.generate_button = ttk.Button(self.root, text="Generate G-code", command=self.generate_gcode)

        self.dimensions_label.pack(pady=5)
        self.dimensions_entry.pack(pady=5)
        self.tool_diameter_label.pack(pady=5)
        self.tool_diameter_entry.pack(pady=5)
        self.speed_label.pack(pady=5)
        self.speed_entry.pack(pady=5)
        self.depth_label.pack(pady=5)
        self.depth_entry.pack(pady=5)
        self.layer_height_label.pack(pady=5)
        self.layer_height_entry.pack(pady=5)
        self.absolute_mode_checkbox.pack(pady=5)
        self.generate_button.pack(pady=10)

    def generate_gcode(self):
        gcode = []  # Initialize an empty list to store G-code lines

        # Convert input values to appropriate types
        self.rect_width = float(self.dimensions_entry.get().split(",")[0].strip())
        try:
            dimensions_parts = self.dimensions_entry.get().split(",")
            if len(dimensions_parts) >= 2:
                self.rect_height = float(dimensions_parts[1].strip())
            else:
                raise ValueError("Invalid dimensions input format")
        except ValueError as e:
            messagebox.showerror("Error", f"Invalid dimensions input: {e}")
            return

        self.tool_diameter = float(self.tool_diameter_entry.get())
        self.milling_speed = float(self.speed_entry.get())
        self.hole_depth = float(self.depth_entry.get())
        self.layer_height = float(self.layer_height_entry.get())
        self.absolute_mode = self.absolute_mode_var.get()

        # Calculate the boundary coordinates of the rectangle
        self.x_min = -self.rect_width / 2
        self.x_max = self.rect_width / 2
        self.y_min = -self.rect_height / 2
        self.y_max = self.rect_height / 2

        # Switch between absolute and relative positioning mode
        if self.absolute_mode:
            gcode.append("G21 ; Set units to millimeters")
            gcode.append("G90 ; Set to absolute positioning mode")
        else:
            gcode.append("G21 ; Set units to millimeters")
            gcode.append("G91 ; Set to relative positioning mode")

        # Move to the initial safe Z position
        gcode.append("G0 Z0 F300")

        # Generate G-code for each layer
        for layer in range(0, int(self.hole_depth / self.layer_height)):
            z_position = layer * self.layer_height

            # Move to the starting position of the layer
            gcode.append(f"G0 Z{z_position:.2f}")

            # Generate the outline of the rectangle
            gcode.append(f"G1 X{self.x_min:.2f} Y{self.y_min:.2f} F{self.milling_speed}")
            gcode.append(f"G1 X{self.x_max:.2f} Y{self.y_min:.2f} F{self.milling_speed}")
            gcode.append(f"G1 X{self.x_max:.2f} Y{self.y_max:.2f} F{self.milling_speed}")
            gcode.append(f"G1 X{self.x_min:.2f} Y{self.y_max:.2f} F{self.milling_speed}")
            gcode.append(f"G1 X{self.x_min:.2f} Y{self.y_min:.2f} F{self.milling_speed}")

            # Generate the milling path within the rectangle
            y_position = self.y_max - self.tool_diameter  # Start from the top of the rectangle
            while y_position >= self.y_min:
                gcode.append(f"G0 X{self.x_max:.2f} Y{y_position:.2f}")
                gcode.append(f"G1 X{self.x_min:.2f} Y{y_position:.2f} F{self.milling_speed}")
                y_position -= self.tool_diameter

        # Save the generated G-code to a file
        file_path = filedialog.asksaveasfilename(defaultextension=".gcode", filetypes=[("G-code Files", "*.gcode")])
        if file_path:
            with open(file_path, 'w') as f:
                for line in gcode:
                    f.write(line + "\n")
                messagebox.showinfo("Success", "G-code generated and saved successfully!")


if __name__ == "__main__":
    root = tk.Tk()
    app = CNCMillingApp(root)
    root.mainloop()