parser.py

import re

# Matches a letter followed by a number (int or float, optionally negative)
# e.g. "G81", "X-50.", "F1000."
WORD_RE = re.compile(r'([A-Za-z])(-?\d*\.?\d+)')


def parse_line(raw_line):
    """
    Turn one line of G-code into a structured dict.
    Returns None for blank lines, '%' markers, or pure comment lines.
    """
    line = raw_line.strip()

    if not line or line == '%':
        return None

    # Strip inline comments like (2D POCKET2) but keep them for reference
    comment_match = re.search(r'\((.*?)\)', line)
    comment = comment_match.group(1) if comment_match else None
    code_part = re.sub(r'\(.*?\)', '', line).strip()

    if not code_part:
        # it was a comment-only line
        return {'type': 'comment', 'comment': comment, 'raw': raw_line.rstrip()}

    words = WORD_RE.findall(code_part)

    result = {'type': 'code', 'comment': comment, 'raw': raw_line.rstrip(), 'g': [], 'm': []}
    for letter, value in words:
        letter = letter.upper()
        num = float(value)
        if letter == 'G':
            result['g'].append(int(num))
        elif letter == 'M':
            result['m'].append(int(num))
        elif letter == 'N':
            result['n'] = int(num)
        else:
            result[letter] = num

    return result


def parse_file(path):
    """Read an .nc file and return a list of parsed line-dicts (skipping None)."""
    parsed = []
    with open(path, 'r', errors='ignore') as f:
        for raw_line in f:
            result = parse_line(raw_line)
            if result is not None:
                parsed.append(result)
    return parsed