utils.py

import arduboy.fxcart
import arduboy.arduhex
import arduboy.shortcuts
import arduboy.image

from constants import *
from arduboy.constants import *

import sys
import os
import time
import textwrap
import slugify
import logging
import demjson3

from typing import List
from PIL import Image, ImageDraw, ImageFont

EXPORT_SLOTS_DIGITS = 3


def set_app_id():
    # Some initial setup
    try:
        # This apparently only matters for windows and for GUI apps
        from ctypes import windll  # Only exists on Windows.
        myappid = 'Haloopdy.ArduboyToolset'
        windll.shell32.SetCurrentProcessExplicitAppUserModelID(myappid)
    except ImportError:
        pass

def set_basic_logging():
    if sys.platform == "darwin":
        log_dir = os.path.expanduser("~/Library/Logs/arduboy_toolset")
    else:
        log_dir = SCRIPTDIR

    level=logging.DEBUG
    log_format="%(asctime)s - %(levelname)s - %(message)s"

    try:
        if not os.path.exists(log_dir):
            os.makedirs(log_dir)
        logging.basicConfig(level=level, format=log_format,
            handlers=[
                logging.FileHandler(os.path.join(log_dir, "arduboy_toolset_gui_log.txt")),
                logging.StreamHandler()
            ]
        )
    except Exception as ex:
        logging.basicConfig(level=level, format=log_format,
            handlers=[ logging.StreamHandler() ]
        )
        logging.warning(f"Couldn't set up file logging: {ex}")



def get_filesafe_datetime():
    return time.strftime("%Y%m%d-%H%M%S", time.localtime())

def get_sketch_backup_filename():
    return f"sketch-backup-{get_filesafe_datetime()}.hex"

def get_eeprom_backup_filename():
    return f"eeprom-backup-{get_filesafe_datetime()}.bin"

def get_fx_backup_filename():
    return f"flashcart-backup-{get_filesafe_datetime()}.bin"

def get_arduhex_backup_filename(arduparsed: arduboy.arduhex.ArduboyParsed):
    return slugify.slugify(arduparsed.title or arduparsed.original_filename or "") + f"_{get_filesafe_datetime()}.arduboy"

def get_meta_backup_filename(meta: arduboy.fxcart.FxSlotMeta, extension):
    return slugify.slugify(meta.title if meta.title else "") + f"_{get_filesafe_datetime()}.{extension}"

def resource_file(name):
    basedir = os.path.dirname(__file__)
    return os.path.join(basedir, 'appresource', name)

# Create a default titlescreen using the given text. We use our little font we got from the internet (free to use, attribution given)
def make_titlescreen(text):
    img = Image.new('1', (SCREEN_WIDTH, SCREEN_HEIGHT), 0)  # 1-bit black and white image
    font = ImageFont.truetype(resource_file(TINYFONT), 16) # I think the thing said 16, 32, etc
    draw = ImageDraw.Draw(img)
    # We know each character takes up a fixed amount of pixels, so this works.
    wrapped_text = textwrap.fill(text, width=((SCREEN_WIDTH - 8)//TINYFONT_WIDTH))  
    _, _, text_width, text_height = draw.textbbox((0,0), wrapped_text, font=font)

    # Calculate text position to center it in the image
    x = (SCREEN_WIDTH - text_width) // 2
    y = (SCREEN_HEIGHT - text_height) // 2

    # Draw the wrapped text in white color
    draw.text((x, y), wrapped_text, font=font, fill=1)

    return img

def make_titlescreen_from_slot(slot: arduboy.fxcart.FxParsedSlot):
    logging.debug(f"Creating title image for {slot.meta.title}")
    base = "Category: " if slot.is_category() else "Game: "
    if slot.meta.title:
        return make_titlescreen(f"{base}{slot.meta.title}")
    else:
        return make_titlescreen(f"{base}{slot.category}")

def export_slots_name(slot, number):
    return str(number).zfill(EXPORT_SLOTS_DIGITS) + "_" + slugify.slugify(slot.meta.title)

def export_slots_as_arduboy(slots: List[arduboy.fxcart.FxParsedSlot], device: str, folderpath, report_progress):
    logging.debug(f"Exporting {len(slots)} slots as a bunch of arduboy files to {folderpath}")
    if not os.path.isdir(folderpath):
        raise Exception(f"Folder {folderpath} does not exist!")
    category = -1
    program = 0
    current_path = folderpath
    for index,slot in enumerate(slots):
        if slot.is_category():
            category += 1
            program = 0
            current_path = os.path.join(folderpath, export_slots_name(slot, category))
            os.mkdir(current_path)
            arduboy.image.bin_to_pilimage(slot.image_raw).save(os.path.join(current_path, "category.png"))
            data = { "title" : slot.meta.title, "info" : slot.meta.info, "image" : "category.png" }
            demjson3.encode_to_file(os.path.join(current_path, "category.json"), data, compactly = False)
        else:
            program += 1
            ardparsed = arduboy.shortcuts.arduboy_from_slot(slot, device)
            arduboy.arduhex.write_arduboy(ardparsed, os.path.join(current_path, export_slots_name(slot, program) + ".arduboy"))
        if report_progress:
            report_progress(index + 1, len(slots))