HobbyCNC fórum - UCCNC vezérlő program

import pandas as pd import numpy as np import re import tkinter as tk from tkinter import filedialog, messagebox # Hibák betöltésére szolgáló függvény def load_errors(error_file): # Hibák betöltése a CSV fájlból pontosvessző elválasztóval error_df = pd.read_csv(error_file, delimiter=';', encoding='utf-8') return error_df # Lineáris interpoláció a hibákhoz def interpolate_error(coord, coord_name, errors): if coord < errors[coord_name].min() or coord > errors[coord_name].max(): print(f"{coord_name} koordináta ({coord}) kívül esik a táblázat értékein.") # Debug return 0 # Ha a koordináta kívül esik a táblázat értékein, nem alkalmazunk hibát lower_bound = errors[errors[coord_name] <= coord].iloc[-1] upper_bound = errors[errors[coord_name] >= coord].iloc[0] if lower_bound[coord_name] == upper_bound[coord_name]: print(f"Pontos találat {coord_name} tengelyen: {lower_bound[coord_name]}") # Debug return lower_bound[f'error_{coord_name.lower()}'] # Ha pontosan egy adatponttal egyezik c1, e1 = lower_bound[coord_name], lower_bound[f'error_{coord_name.lower()}'] c2, e2 = upper_bound[coord_name], upper_bound[f'error_{coord_name.lower()}'] # Lineáris interpoláció interpolated_error = e1 + (coord - c1) * (e2 - e1) / (c2 - c1) return interpolated_error # G-kód javító függvény, amely X, Y és Z tengelyek hibáit korrigálja def correct_gcode(input_file, output_file, errors): try: with open(input_file, 'r', encoding='utf-8') as file: lines = file.readlines() with open(output_file, 'w', encoding='utf-8') as file: for line in lines: # Regex a koordináták beolvasásához (formátum: Y29.702Z-9.727 vagy X-4.375Y31.695Z1.) matches = re.findall(r'[XYZ]-?\d+\.\d*', line) coords = {match[0]: float(match[1:]) for match in matches} x_corrected = y_corrected = z_corrected = None # Kezdetben None-ra állítjuk if 'X' in coords or 'Y' in coords or 'Z' in coords: if 'X' in coords: x = coords['X'] error_x = interpolate_error(x, 'X', errors) x_corrected = x + error_x if 'Y' in coords: y = coords['Y'] error_y = interpolate_error(y, 'Y', errors) y_corrected = y + error_y if 'Z' in coords: z = coords['Z'] error_z = interpolate_error(z, 'Z', errors) z_corrected = z + error_z # Új sor összeállítása a javított koordinátákkal corrected_line_parts = [] for part in re.finditer(r'([XYZ]-?\d+\.\d*)', line): prefix = part.group()[0] value = part.group()[1:] corrected_value = float(value) if prefix == 'X' and x_corrected is not None: corrected_value = x_corrected elif prefix == 'Y' and y_corrected is not None: corrected_value = y_corrected elif prefix == 'Z' and z_corrected is not None: corrected_value = z_corrected corrected_line_parts.append(f"{prefix}{corrected_value:.4f}") corrected_line = ' '.join(corrected_line_parts) + '\n' file.write(corrected_line) else: file.write(line) except FileNotFoundError: print(f"A fájl nem található: {input_file}") except Exception as e: print(f"Hiba történt: {e}") # GUI funkció def process_files(): input_file = filedialog.askopenfilename(title="Válassza ki a bemeneti G-kód fájlt", filetypes=[("G-kód fájlok", "*.tap")]) error_file = filedialog.askopenfilename(title="Válassza ki a hibafájlt", filetypes=[("CSV fájlok", "*.csv")]) output_file = filedialog.asksaveasfilename(title="Mentse a kimeneti G-kód fájlt", defaultextension=".nc", filetypes=[("G-kód fájlok", "*.nc")]) # Ellenőrizd, hogy a fájlok kiválasztásra kerültek if input_file and error_file and output_file: errors = load_errors(error_file) correct_gcode(input_file, output_file, errors) messagebox.showinfo("Kész", "A G-kód javítása befejeződött!") # Ablak létrehozása root = tk.Tk() root.title("G-Kód Hibakorrekciós Program") process_button = tk.Button(root, text="Fájlok feldolgozása", command=process_files) process_button.pack(pady=20) root.mainloop()