FinalMix/assets_scaler.py at main · cloudzXIII/FinalMix · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
# Full Credits to Firch for original code of this amazing sprite converter code so that 2x sprites don't look like they have glow :3

import sys, os
from PIL import Image
import pathlib


asset_dir = str(pathlib.Path(__file__).parent.resolve()) + r"\\assets\\"
input_dir = asset_dir + r"\1x"
upscale_dir = asset_dir + r"\2x"

def average_color_of_neighbors(img, x, y):
    width, height = img.size
    r_total, g_total, b_total, count = 0, 0, 0, 0

    neighbors = [(-1, -1), (0, -1), (1, -1),
                 (-1,  0),          (1,  0),
                 (-1,  1), (0,  1), (1,  1)]

    for dx, dy in neighbors:
        nx, ny = x + dx, y + dy
        if 0 <= nx < width and 0 <= ny < height:
            neighbor_pixel = img.getpixel((nx, ny))
            if neighbor_pixel[3] > 0:
                r_total += neighbor_pixel[0]
                g_total += neighbor_pixel[1]
                b_total += neighbor_pixel[2]
                count += 1

    if count > 0:
        return (r_total // count, g_total // count, b_total // count, 255)
    else:
        return (0, 0, 0, 0)

def process_image(path):
    img = Image.open(path).convert("RGBA")

    bottom_layer = img.copy()
    width, height = bottom_layer.size
    new_img = bottom_layer.copy()

    for y in range(height):
        for x in range(width):
            pixel = bottom_layer.getpixel((x, y))
            if pixel[3] == 0:
                new_color = average_color_of_neighbors(bottom_layer, x, y)
                new_img.putpixel((x, y), new_color)

    alpha = new_img.split()[3]
    alpha = alpha.point(lambda p: 0)
    new_img.putalpha(alpha)
    combined = Image.alpha_composite(new_img, img)

    return combined

def upscale_image(image, scale_factor=2):
    width, height = image.size
    new_size = (width * scale_factor, height * scale_factor)
    upscaled_image = image.resize(new_size, Image.NEAREST)
    return upscaled_image

def clear_directory(directory):
    print("Clearing directory: " + directory)

    for root, dirs, files in os.walk(directory, topdown=False):
        for name in files:
            os.remove(os.path.join(root, name))
        for name in dirs:
            os.rmdir(os.path.join(root, name))

def process_and_upscale(input_path, input_directory, upscale_directory):
    relative_path = os.path.relpath(input_path, input_directory)
    output_path = os.path.join(upscale_directory, relative_path)

    os.makedirs(os.path.dirname(output_path), exist_ok=True)

    print("Processing image: " + input_path)
    processed_image = process_image(input_path)
    processed_image.save(output_path, "PNG")

    print("Upscaling image: " + output_path)
    upscaled_image = upscale_image(processed_image)
    upscaled_image.save(output_path, "PNG")

def process_directory(input_directory, upscale_directory):
    print("Processing directory: " + input_directory)

    clear_directory(upscale_directory)

    for root, _, files in os.walk(input_directory):
        for file in files:
            input_path = os.path.join(root, file)
            process_and_upscale(input_path, input_directory, upscale_directory)

if __name__ == "__main__":
    print("Script start")
    if len(sys.argv) > 1:
        file_path = sys.argv[1]
        process_and_upscale(file_path, input_dir, upscale_dir)
    else:
        process_directory(input_dir, upscale_dir)
    print("Script end")