script_svg.py

# -*- coding: utf-8 -*-
"""Dessiner un chemin de profondeur constante."""

from datetime import date
from pathlib import Path

from xml.etree import ElementTree as ET

import svg.path

import matplotlib
from matplotlib import pyplot

# Régler l'origine
# On assume que l'origine est au coin supérieur gauche, contre la plaque
# sacrificielle.
# î
# .->---------------.
# |                 |
# |                 |
# .-----------------.

namespace = {'svg': 'http://www.w3.org/2000/svg'}  # Pour le SVG


def extraire_chemins(fichier_svg: Path,  # Entrée
                     résolution: float = 0.1  # mm
                     ):
    # Aller chercher les chemins dans le fichier SVG
    document_svg = ET.parse(fichier_svg)
    chemins_svg = [svg.path.parse_path(c.attrib['d'])
                   for c in document_svg.findall('.//svg:path', namespace)]

    chemins = []
    for chemin in chemins_svg:
        longueur = int(chemin.length() / résolution)

        xs, ys = [], []
        for i in range(longueur + 1):
            index = i / longueur
            point = chemin.point(index)
            x, y = point.real, point.imag
            xs.append(x)
            ys.append(y)
        chemins.append([xs, ys])

    return chemins


def extraire_gcode(chemins: list[list[list[float]]],  # Entrée
                   # Paramètres de programme
                   z_0: float = 10,  # mm, niveau de déplacement rapide
                   vitesse_de_rotation: float = 10000,  # tr/min
                   avance: float = 800,  # mm/min
                   ):
    programme = []  # Programme à écrire

    # Début du programme
    # Réglages de base
    programme.append('G71')  # mm
    programme.append('T1')  # Outil 1
    programme.append(f'S{vitesse_de_rotation}')  # tr/min
    programme.append(f'F{avance}')  # mm/min
    programme.append('G90')  # Coordonnées absolues

    for xs, ys in chemins:
        # Déplacement hors-matière
        programme.append('')
        programme.append(f'G0 X{xs[0]} Y{ys[0]} Z{z_0}')

        # On itère sur une série de points.
        for x, y in zip(xs, ys):
            # Retrait de matière
            programme.append(f'G1 X{x} Y{y} Z0.')

        programme.append(f'G0 Z{z_0}')  # Plan de déplacement sécure

    # Fin & arrêt du programme
    programme.append('')
    programme.append('M5')  # Arrêt broche
    programme.append('M2')  # Fin de programme
    programme.append('')

    return '\n'.join(programme)


def extraire_graphique(chemins: list[list[list[float]]]):
    for i, (xs, ys) in enumerate(chemins):
        pyplot.plot(xs, ys, '-')

        if i < (len(chemins) - 1):
            dernier = xs[-1], ys[-1]
            suivant = chemins[i + 1]
            premier = suivant[0][-1], suivant[1][-1]
            pyplot.plot([dernier[0], premier[0]],
                        [dernier[1], premier[1]],
                        '--', color='gray')


def main():
    chemins = extraire_chemins('dessin.svg')
    gcode = extraire_gcode(chemins)

    matplotlib.style.use('seaborn')
    pyplot.gca().set_aspect('equal')

    extraire_graphique(chemins)

    with open(f'parcours {date.today()}.iso', 'w') as f:
        f.write(gcode)

    pyplot.savefig(f'parcours {date.today()}.svg')
    pyplot.show()


if __name__ == '__main__':
    main()