MarlinFirmware/E3S1PROFORKBYTT_printdata_superslicer_thumbnail.py

#!/usr/bin/env python3
#
# Superslicer Slicer remove headers before jpg.
#
# It also adds the needed printdata to show on the main during print.
#
# This script has been developed for E3S1PROFORKBYTT by Thomas Toka.
#
# Introduced with v008 into E3S1PROFORKBYTT. Extended in v023
# ------------------------------------------------------------------------------

import sys
import os
import math
import base64
from PIL import Image
from io import BytesIO
import re

# Get the g-code source file name
sourceFile = sys.argv[1]

# Read the ENTIRE g-code file into memory
with open(sourceFile, "r", encoding='utf-8') as f:
    lines = f.readlines()

new_lines = []

thumbnail_header_found = False
thumbnail_lines = []  # Initialize a list to store thumbnail lines
remove_lines = False  # Reset the remove_lines flag

# Find the existing thumbnail header and footer lines
thumbnail_start = None
thumbnail_end = None

for i, line in enumerate(lines):
    if line.startswith('; generated by SuperSlicer'):
        remove_lines = True
    elif line.startswith('; thumbnail begin'):
        if not thumbnail_header_found:
            thumbnail_start = i
            thumbnail_header_found = True
        else:
            # If the new thumbnail header has been added, skip this line
            continue
    elif line.startswith('; thumbnail end'):
        thumbnail_end = i
    elif remove_lines and line.strip() == ';':
        remove_lines = False
    elif not remove_lines:
        new_lines.append(line)

# Extract additional information
filament_used_m, filament_used_g, filament_diameter, filament_density, layer_height, layers = "0", "0", "0", "0", "0", "0"
for line in new_lines:
    if line.startswith("; filament used [mm] ="):
        filament_used_mm = float(line.split("=")[1].strip())
        filament_used_m = round(filament_used_mm / 1000, 2)
        if filament_used_m > 0:
            filament_used_m = math.ceil(filament_used_m)
        else:
            filament_used_m = 0
    elif line.startswith("; filament used [g] ="):
        filament_used_g = float(line.split("=")[1].strip())
        filament_used_g = round(filament_used_g, 2)
        if filament_used_g > 0:
            filament_used_g = math.ceil(filament_used_g)
        else:
            filament_used_g = 0
    elif line.startswith("; filament_diameter ="):
        filament_diameter = float(line.split("=")[1].strip())
    elif line.startswith("; filament_density ="):
        filament_density = float(line.split("=")[1].strip())
    elif line.startswith("; layer_height ="):
        layer_height = line.split("=")[1].strip()
        layer_height = "{:.2f}".format(round(float(layer_height), 2))
    elif line.startswith("; total layers count ="):
        layers = line.split("=")[1].strip()
    elif line.startswith("; estimated printing time (normal mode) ="):
        time_parts = line.split("=")[1].strip().split()
        days, hours, minutes, seconds = 0, 0, 0, 0
        for part in time_parts:
            if part.endswith('d'):
                days = int(part[:-1])
            elif part.endswith('h'):
                hours = int(part[:-1])
            elif part.endswith('m'):
                minutes = int(part[:-1])
            elif part.endswith('s'):
                seconds = int(part[:-1])
        total_time_minutes = (days * 24 * 60) + (hours * 60) + minutes + (seconds / 60)

layers = int(layers)
filament_used_m_per_layer = filament_used_m / max(layers, 1)  # Avoid division by zero
remaining_filament_m = filament_used_m

filament_used_g_per_layer = filament_used_g / max(layers, 1)  # Avoid division by zero
remaining_filament_g = filament_used_g

m117_added = 0  # Counter for added M117 commands
first_layer = True

if thumbnail_start is not None and thumbnail_end is not None:
    # Extract the JPEG data without decoding
    original_jpeg_data = "".join(lines[thumbnail_start + 1:thumbnail_end]).replace("; ", "")

    # Define a maximum line length for the JPEG data
    max_line_length = 75 - len("; ")

    # Split the JPEG data into lines with a maximum length
    num_lines = math.ceil(len(original_jpeg_data) / max_line_length)

    # Add new thumbnail header
    new_thumbnail_header = (
        f"; thumbnail begin 250x250 {len(original_jpeg_data)} "
        f"1 {num_lines} {filament_used_m} {filament_used_g} "
        f"{layer_height} {filament_diameter} {filament_density} {layers}\n"
    )
    new_lines.insert(0, new_thumbnail_header)

    # Add JPEG lines after the new thumbnail header
    new_lines.extend([original_jpeg_data[i:i+max_line_length] for i in range(0, len(original_jpeg_data), max_line_length)])

# Process thumbnail section
for i, line in enumerate(new_lines):
    if line.startswith(';AFTER_LAYER_CHANGE'):
        after_layer_change_index = i  # Store the index of ';AFTER_LAYER_CHANGE'
        break  # Exit loop once we find ';AFTER_LAYER_CHANGE'

# Add lines after ';AFTER_LAYER_CHANGE'
for i in range(after_layer_change_index, len(new_lines)):
    if new_lines[i].startswith(';AFTER_LAYER_CHANGE'):
        if first_layer:
            m117_line = "M117 L1 M{} G{} Z{} Q{}".format(math.ceil(remaining_filament_m), math.ceil(remaining_filament_g), layers, layer_height)
            new_lines.insert(i + 1, m117_line + '\n')
            m73_line_r = "M73 R{}".format(int(total_time_minutes * (1 - m117_added / layers)))
            new_lines.insert(i + 2, m73_line_r + '\n')
            m73_line_p = "M73 P{}".format(int((m117_added / layers) * 100))
            new_lines.insert(i + 3, m73_line_p + '\n')
            first_layer = False
        else:
            m117_line = "M117 L{} M{} G{}".format(m117_added + 1, math.ceil(remaining_filament_m), math.ceil(remaining_filament_g))
            new_lines.insert(i + 1, m117_line + '\n')
            if m117_added == layers - 1:
                m73_line_r = "M73 R{}".format(int(total_time_minutes * (1 - m117_added / layers)))
                new_lines.insert(i + 2, m73_line_r + '\n')
                m73_line_p = "M73 P{}".format(100)
                new_lines.insert(i + 3, m73_line_p + '\n')
            else:
                m73_line_r = "M73 R{}".format(int(total_time_minutes * (1 - m117_added / layers)))
                new_lines.insert(i + 2, m73_line_r + '\n')
                m73_line_p = "M73 P{}".format(int((m117_added / layers) * 100))
                new_lines.insert(i + 3, m73_line_p + '\n')
            remaining_filament_m -= filament_used_m_per_layer
            remaining_filament_g -= filament_used_g_per_layer
            m117_added += 1

# Write the modified content back to the original file
with open(sourceFile, "w", encoding='utf-8') as f:
    f.writelines(new_lines)

print(f"Added {m117_added} M117 commands and M73 with time information.")