main.py

import pygame
import sys
from PIL import Image
import os

#import image
imageyn = input('import image? [y/n]')
threshold = 360
if imageyn == 'y':
    print('import image')
    inputi = Image.open(input('image name: '))
    threshold = int(input('threshold (percent): '))
    thresholdpix = int(input('threshold of pixel is on/off [1/0]: '))
    if thresholdpix == 1:
        thresholdpixoff = 0
    else:
        thresholdpixoff = 1
    inputi.resize((32, 24)).save('tmp.' + inputi.format)
    image = Image.open('tmp.' + inputi.format)

# Initialize Pygame
pygame.init()

# Set up display
WIDTH, HEIGHT = 1200, 600
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("115esp2 drawer")

# Define colors
BLUE = (32, 30, 82)
LGREEN = (113, 191, 71)
GREEN = (50, 191, 57)
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)

# Define constants
PIXEL_SIZE = 18
SPRITE_WIDTH = 96
SPRITE_HEIGHT = 9
BSPRITE_WIDTH = 32
BSPRITE_HEIGHT = 24
GRID_COLOR = LGREEN

# Create sprite data with every 8th character set to 1
sprite_data = [[0] * SPRITE_WIDTH for _ in range(SPRITE_HEIGHT)]
for row in sprite_data:
    for i in range(7, SPRITE_WIDTH, 8):
        row[i] = 1

#load image or default
if imageyn == 'y':
    pixels = image.load()
    height, width = image.size
else:
    bsprite_data = [[0] * BSPRITE_WIDTH for _ in range(BSPRITE_HEIGHT)]
    for row in bsprite_data:
        for i in range(7, BSPRITE_WIDTH, 8):
            row[i] = 1

#load image
if imageyn == 'y':
    bsprite_data = []
    for x in range(width):
        bsprite_data_sect = []
        for y in range(height):
            cpixel = pixels[y, x]
            cpixsum = cpixel[0] + cpixel[1] + cpixel[2]
            target = 768
            if cpixsum > threshold / 100 * target:
                bsprite_data_sect.append(thresholdpix)
            else:
                bsprite_data_sect.append(thresholdpixoff)
        bsprite_data.append(bsprite_data_sect)

# Function to draw the sprite
def draw_sprite():
    # Draw pixels
    for y in range(BSPRITE_HEIGHT):
        for x in range(BSPRITE_WIDTH):
            color = BLUE if bsprite_data[y][x] == 1 else GREEN
            pygame.draw.rect(screen, color, (x * PIXEL_SIZE, y * PIXEL_SIZE, PIXEL_SIZE, PIXEL_SIZE))
    # Draw grid lines
    for x in range(0, BSPRITE_WIDTH*PIXEL_SIZE, PIXEL_SIZE):
        pygame.draw.line(screen, GRID_COLOR, (x, 0), (x, (BSPRITE_HEIGHT*PIXEL_SIZE) - 1), 2)
    for y in range(0, BSPRITE_HEIGHT*PIXEL_SIZE, PIXEL_SIZE):
        pygame.draw.line(screen, GRID_COLOR, (0, y), ((BSPRITE_WIDTH*PIXEL_SIZE) - 1, y), 2)


# Function to translate sprite data to hexadecimal and then to tokens
def translate(line_wrap):
    i = 0
    hconv = []
    for px in bsprite_data:
        if i <= 3 and not i > 3:
            sprite_data[i+4][32:64] = px
            i+=1
        elif i >= 4 and not i > 12:
            sprite_data[i-4][0:32] = px 
            i+=1
        elif i >= 13 and not i > 20:
            sprite_data[i-13][64:96] = px
            i+=1
        elif i >= 21:
            sprite_data[i-21][32:64] = px
            i+=1
    hex_data = []
    for row in sprite_data:
        binary_string = ''.join(str(bit) for bit in row)
        hex_string = hex(int(binary_string, 2))[2:].zfill(SPRITE_WIDTH // 4)
        hex_data.extend(hex_string[i:i+2] + " " for i in range(0, len(hex_string), 2))
    # Update specific indices with the desired values
    hex_data[40:43] = ["06 ", "75 ", "31 "] #a:bcde de bc xa xx | 1:7506 -> render_waitshift    
    # Add line wrap and conversion

    for hchar in hex_data:
        hconv.append((str(open("Token_List.md").read().split("\n")[int(hchar.replace(" ", ""), 16)+7].split()[3])))
    
    data_wrapped = []
    for i in range(0, len(hconv), line_wrap):
        data_wrapped.append(' '.join(hconv[i:i+line_wrap]))

    return data_wrapped

# Main loop
running = True
drawing = False
output = []
global data

while running:
    screen.fill(BLACK)
    draw_sprite()

    # Display token representation
    font = pygame.font.SysFont(None, 24)
    data = translate(24)
    for i, row in enumerate(data):
        text = font.render(row, True, WHITE)
        screen.blit(text, (20, BSPRITE_HEIGHT * PIXEL_SIZE + 20 + i * 30))
        output.append(row)

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        elif event.type == pygame.MOUSEBUTTONDOWN:
            if event.button == 1:
                x, y = pygame.mouse.get_pos()
                x //= PIXEL_SIZE
                y //= PIXEL_SIZE
                if 0 <= x < BSPRITE_WIDTH and 0 <= y < BSPRITE_HEIGHT:
                    bsprite_data[y][x] = 1 - bsprite_data[y][x]  # Invert the pixel color

    pygame.display.flip()
# Save Hackstring
open("output.txt", "w").write('\n'.join(translate(12)))
# Quit Pygame
if imageyn == 'y':
    os.remove('tmp.' + inputi.format)
pygame.quit()
sys.exit()