Mechanicus_LASER_SHAPE_TRACING copy 15.py

import cv2
import numpy as np
import serial
from PIL import Image, ImageTk
import tkinter as tk
import time

# Initialize default values
video_width, video_height = 480, 480
cnc_width, cnc_height = 400, 400  # in millimeters
serial_port = 'COM4'

# Initialize variables
video_center_x = video_width // 2
video_center_y = video_height // 2

# Initialize the current CNC position at the center
current_x = cnc_width // 2
current_y = cnc_height // 2

# Function to set the current position to the center
def set_current_position_to_center():
    global current_x, current_y
    current_x = cnc_width // 2
    current_y = cnc_height // 2
    send_gcode(f'G0 X{current_x} Y{current_y}')

# Function to activate the laser
def activate_laser():
    send_gcode("M3 S15")

# Function to turn off the laser
def turn_off_laser():
    send_gcode("M5 S0")

# Function to home the CNC machine
def home():
    send_gcode("G28")

# Function to move the CNC to a specific position
def move_to_position(x, y):
    global current_x, current_y
    x = max(0, min(x, cnc_width))
    y = max(0, min(y, cnc_height))
    gcode_line = f"G0 X{x} Y{y}"
    send_gcode(gcode_line)
    current_x = x
    current_y = y

# Function to send G-code commands to control the CNC
def send_gcode(gcode_line):
    try:
        ser = serial.Serial(serial_port, 115200)
        ser.write((gcode_line + "\n").encode())
        ser.close()
        #print("Sent G-code command:", gcode_line)  # Print the sent command
    except serial.SerialException as e:
        print(f"Serial Error: {e}")

# Function to process the video frame and detect the green object
def process_frame(cap):
    ret, frame = cap.read()
    if not ret:
        return None, None, None

    lower_green = np.array([35, 70, 70])
    upper_green = np.array([90, 255, 255])

    hsv_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    mask = cv2.inRange(hsv_frame, lower_green, upper_green)

    contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    if contours:
        largest_contour = max(contours, key=cv2.contourArea)
        moments = cv2.moments(largest_contour)
        if moments["m00"] != 0:
            green_x = int(moments["m10"] / moments["m00"])
            green_y = int(moments["m01"] / moments["m00"])
            return frame, green_x, green_y
        else:
            turn_off_laser()
    return frame, None, None

# Updated tracker function
def tracker():
    # Open the webcam
    cap = cv2.VideoCapture(1)

    while True:
        frame, green_x, green_y = process_frame(cap)
        if frame is None:
            break

        if green_x is not None and green_y is not None:
            # Calculate the absolute coordinates of the green point from the origin
            absolute_x = current_x + (green_x - video_width / 2) * (cnc_width / video_width)
            absolute_y = current_y + (green_y - video_height / 2) * (cnc_height / video_height)

            # Print the absolute coordinates for debugging purposes
            print("Absolute Coordinates:", absolute_x, absolute_y)

            # Move the CNC to the absolute coordinates of the green point
            move_to_position(absolute_x, absolute_y)

        frame = frame[:video_width, :video_height]

        img = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
        img_tk = ImageTk.PhotoImage(image=img)
        label.config(image=img_tk)
        label.image = img_tk

        root.update()
        time.sleep(0.1)

    cap.release()
    root.destroy()


# Entry point
if __name__ == "__main__":
    # Create the tkinter app
    root = tk.Tk()
    root.title("CNC Control App")

    # Create input fields for dimensions and serial port
    tk.Label(root, text="Video Width:").grid(row=0, column=0)
    video_width_entry = tk.Entry(root)
    video_width_entry.insert(0, str(video_width))
    video_width_entry.grid(row=0, column=1)

    tk.Label(root, text="Video Height:").grid(row=1, column=0)
    video_height_entry = tk.Entry(root)
    video_height_entry.insert(0, str(video_height))
    video_height_entry.grid(row=1, column=1)

    tk.Label(root, text="CNC Width (mm):").grid(row=2, column=0)
    cnc_width_entry = tk.Entry(root)
    cnc_width_entry.insert(0, str(cnc_width))
    cnc_width_entry.grid(row=2, column=1)

    tk.Label(root, text="CNC Height (mm):").grid(row=3, column=0)
    cnc_height_entry = tk.Entry(root)
    cnc_height_entry.insert(0, str(cnc_height))
    cnc_height_entry.grid(row=3, column=1)

    tk.Label(root, text="Serial Port:").grid(row=4, column=0)
    serial_port_entry = tk.Entry(root)
    serial_port_entry.insert(0, serial_port)
    serial_port_entry.grid(row=4, column=1)

    # Create buttons to trigger CNC functions
    tk.Button(root, text="Set Current Position to Center", command=set_current_position_to_center).grid(row=5, column=0, columnspan=2)

    label = tk.Label(root)
    label.grid(row=6, column=0, columnspan=2)

    tracker()  # Start the tracker function

    # Main loop
    root.mainloop()