Initial commit

2026-04-28 14:39:54 +08:00
commit fd29e955e5
22 changed files with 3313 additions and 0 deletions
@@ -0,0 +1,177 @@
+from __future__ import annotations
+
+import hashlib
+import os
+import shutil
+import tempfile
+import zipfile
+from pathlib import Path
+
+from .archive import GCODE_MEMBER_RE, MD5_MEMBER_RE, list_gcode_members
+from .gcode import (
+    apply_line_ending,
+    apply_plate_number_offset,
+    build_swap_block,
+    insert_swap_block,
+    normalize_newlines,
+    read_swap_gcode_file,
+    resolve_swap_gcode_path,
+)
+from .metadata import read_filament_metadata, read_slice_plate_metadata, safe_float, update_first_slice_info
+from .models import BuildOptions, BuildResult, GcodePatchConfig, PlateJob, PlateSource
+from .patches import apply_gcode_patches, parse_patch_config
+
+try:
+    from PIL import Image, ImageDraw, ImageFont
+except Exception:  # pragma: no cover
+    Image = None
+    ImageDraw = None
+    ImageFont = None
+
+
+def load_plate_sources(job: PlateJob) -> list[PlateSource]:
+    if job.copies < 1:
+        raise ValueError(f"Invalid copy count for {job.source_3mf}: {job.copies}")
+    with zipfile.ZipFile(job.source_3mf, "r") as archive:
+        members = list_gcode_members(archive)
+        if not members:
+            raise ValueError(f"No Metadata/plate_N.gcode member was found in {job.source_3mf}")
+        plate_metadata = read_slice_plate_metadata(archive)
+        filament_metadata = read_filament_metadata(archive)
+        sources: list[PlateSource] = []
+        for member in members:
+            match = GCODE_MEMBER_RE.match(member)
+            plate_index = int(match.group(1)) if match else 1
+            raw = archive.read(member)
+            text = raw.decode("utf-8-sig", errors="replace")
+            metadata = plate_metadata.get(plate_index, {})
+            filament = filament_metadata.get(plate_index, {})
+            weight = safe_float(metadata.get("weight"))
+            used_g = safe_float(filament.get("used_g"))
+            sources.append(
+                PlateSource(
+                    source_3mf=job.source_3mf,
+                    member_name=member,
+                    gcode_text=text,
+                    prediction_seconds=safe_float(metadata.get("prediction")),
+                    weight_grams=weight if weight is not None else used_g,
+                    filament_used_m=safe_float(filament.get("used_m")),
+                    filament_used_g=used_g if used_g is not None else weight,
+                )
+            )
+        return sources
+
+
+def expand_jobs(jobs: list[PlateJob]) -> list[PlateSource]:
+    expanded: list[PlateSource] = []
+    for job in jobs:
+        sources = load_plate_sources(job)
+        for _ in range(job.copies):
+            expanded.extend(sources)
+    if not expanded:
+        raise ValueError("No input plates were provided.")
+    return expanded
+
+
+def process_plate_gcode(
+    source: PlateSource,
+    plate_number: int,
+    total_plates: int,
+    options: BuildOptions,
+    swap_gcode_text: str,
+    patch_config: GcodePatchConfig,
+) -> str:
+    text = normalize_newlines(source.gcode_text)
+    if options.apply_gcode_patches:
+        text = apply_gcode_patches(text, patch_config)
+    if options.show_plate_number:
+        text = apply_plate_number_offset(text, plate_number)
+    should_swap = options.swap_after_final or plate_number < total_plates
+    if should_swap:
+        swap_block = build_swap_block(swap_gcode_text, options.cool_bed_temp, options.wait_after_eject_seconds)
+        text = insert_swap_block(text, swap_block, patch_config.insert_before_marker)
+    return text.rstrip("\n") + "\n"
+
+
+def update_preview_label(png_bytes: bytes, label: str, small: bool = False) -> bytes:
+    if Image is None or ImageDraw is None:
+        return png_bytes
+    import io
+
+    try:
+        image = Image.open(io.BytesIO(png_bytes)).convert("RGBA")
+    except Exception:
+        return png_bytes
+    draw = ImageDraw.Draw(image)
+    size = max(16, image.width // (12 if small else 8))
+    font = None
+    for font_path in (
+        Path(os.environ.get("WINDIR", "")) / "Fonts" / "arialbd.ttf",
+        Path("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"),
+        Path("/System/Library/Fonts/Supplemental/Arial Bold.ttf"),
+    ):
+        if font_path.exists() and ImageFont is not None:
+            try:
+                font = ImageFont.truetype(str(font_path), size)
+                break
+            except Exception:
+                font = None
+    if font is None and ImageFont is not None:
+        font = ImageFont.load_default()
+    margin = max(8, image.width // 24)
+    text_bbox = draw.textbbox((0, 0), label, font=font)
+    x = margin
+    y = image.height - margin - (text_bbox[3] - text_bbox[1])
+    draw.text((x, y), label, font=font, fill=(0, 255, 0, 255))
+    output = io.BytesIO()
+    image.save(output, format="PNG")
+    return output.getvalue()
+
+
+def write_output_3mf(base_3mf: Path, output_3mf: Path, gcode_bytes: bytes, sources: list[PlateSource], options: BuildOptions) -> str:
+    md5 = hashlib.md5(gcode_bytes).hexdigest()
+    with zipfile.ZipFile(base_3mf, "r") as src, zipfile.ZipFile(output_3mf, "w", compression=zipfile.ZIP_DEFLATED, compresslevel=1) as dst:
+        for item in src.infolist():
+            name = item.filename
+            if GCODE_MEMBER_RE.match(name) or MD5_MEMBER_RE.match(name):
+                continue
+            data = src.read(name)
+            if name == "Metadata/slice_info.config":
+                data = update_first_slice_info(data, sources, options)
+            elif options.add_preview_label and name in {"Metadata/plate_1.png", "Metadata/plate_1_small.png"}:
+                data = update_preview_label(data, f"{len(sources)} plates", small=name.endswith("_small.png"))
+            dst.writestr(item, data)
+        dst.writestr("Metadata/plate_1.gcode", gcode_bytes)
+        dst.writestr("Metadata/plate_1.gcode.md5", md5)
+    return md5
+
+
+def build_packed_3mf(jobs: list[PlateJob], options: BuildOptions) -> BuildResult:
+    sources = expand_jobs(jobs)
+    swap_gcode_path = resolve_swap_gcode_path(options.swap_gcode, options.swap_gcode_dir)
+    swap_gcode_text = read_swap_gcode_file(swap_gcode_path)
+    patch_config = parse_patch_config() if options.apply_gcode_patches else GcodePatchConfig(rules=())
+    processed: list[str] = []
+    for plate_number, source in enumerate(sources, start=1):
+        processed.append(process_plate_gcode(source, plate_number, len(sources), options, swap_gcode_text, patch_config))
+    final_gcode = "\n".join(item.rstrip("\n") for item in processed) + "\n"
+    gcode_bytes = apply_line_ending(final_gcode, options.line_ending)
+    output_3mf = options.output_3mf
+    output_3mf.parent.mkdir(parents=True, exist_ok=True)
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".3mf", dir=str(output_3mf.parent)) as temp_file:
+        temp_path = Path(temp_file.name)
+    try:
+        md5 = write_output_3mf(sources[0].source_3mf, temp_path, gcode_bytes, sources, options)
+        shutil.move(str(temp_path), str(output_3mf))
+    finally:
+        if temp_path.exists():
+            temp_path.unlink(missing_ok=True)
+    total_prediction = sum(value for value in (source.prediction_seconds for source in sources) if value is not None)
+    total_weight = sum(value for value in (source.weight_grams for source in sources) if value is not None)
+    return BuildResult(
+        output_3mf=output_3mf,
+        plate_count=len(sources),
+        total_prediction_seconds=total_prediction if total_prediction > 0 else None,
+        total_weight_grams=total_weight if total_weight > 0 else None,
+        gcode_md5=md5,
+    )