mirror of
https://github.com/SPLURT-Station/S.P.L.U.R.T-Station-13.git
synced 2025-12-11 02:12:14 +00:00
101 lines
2.4 KiB
Python
101 lines
2.4 KiB
Python
import os
|
|
import sys
|
|
import inspect
|
|
import shutil
|
|
import PIL.Image as Image
|
|
|
|
current_dir = os.path.split(inspect.getfile(inspect.currentframe()))[0]
|
|
|
|
def PngSave(im, file):
|
|
# From http://blog.client9.com/2007/08/28/python-pil-and-png-metadata-take-2.html
|
|
|
|
# these can be automatically added to Image.info dict
|
|
# they are not user-added metadata
|
|
reserved = ('interlace', 'gamma', 'dpi', 'transparency', 'aspect')
|
|
|
|
# undocumented class
|
|
import PIL.PngImagePlugin as PngImagePlugin
|
|
meta = PngImagePlugin.PngInfo()
|
|
|
|
# copy metadata into new object
|
|
for k,v in im.info.items():
|
|
if k in reserved: continue
|
|
meta.add_text(k, str(v), 0)
|
|
|
|
# and save
|
|
im.save(file, "PNG", pnginfo=meta)
|
|
|
|
def ProcessFile(path):
|
|
name, ext = os.path.splitext(path)
|
|
ext = ext.lower()
|
|
if (ext != ".dmi" and ext != ".png") or os.path.splitext(name)[1] == ".new":
|
|
return
|
|
|
|
try:
|
|
im = Image.open(path)
|
|
print(name + ": " + im.format, im.size, im.mode)
|
|
if im.mode != "RGBA":
|
|
return
|
|
width, height = im.size
|
|
pix = im.load()
|
|
|
|
n_transparent = 0
|
|
|
|
make_opaque = []
|
|
|
|
def add(x, y):
|
|
if pix[x, y][3] == 0:
|
|
make_opaque.append((x, y))
|
|
|
|
for x in range(0, width):
|
|
for y in range(0, height):
|
|
if pix[x, y][3] > 0:
|
|
if x > 0:
|
|
add(x - 1, y)
|
|
if x < width - 1:
|
|
add(x + 1, y)
|
|
if y > 0:
|
|
add(x, y - 1)
|
|
if y < height - 1:
|
|
add(x, y + 1)
|
|
else:
|
|
n_transparent += 1
|
|
|
|
for coords in make_opaque:
|
|
pix[coords] = (0, 0, 0, 1)
|
|
|
|
PngSave(im, path)
|
|
except Exception as e:
|
|
print("Could not process " + name)
|
|
print(e)
|
|
|
|
root_dir = os.path.abspath(os.path.join(current_dir, "../../"))
|
|
icons_dir = os.path.join(root_dir, "icons")
|
|
|
|
def Main():
|
|
if len(sys.argv) != 2:
|
|
if os.name == 'nt':
|
|
print("Usage: drag-and-drop a .dmi onto `Hitbox Expander.bat`\n or")
|
|
with open(os.path.join(current_dir, "README.txt")) as f:
|
|
print(f.read())
|
|
return 0
|
|
|
|
try:
|
|
with open(sys.argv[1]):
|
|
ProcessFile(os.path.abspath(sys.argv[1]))
|
|
return 0
|
|
except IOError:
|
|
pass
|
|
|
|
for root, subdirs, files in os.walk(icons_dir):
|
|
for file in files:
|
|
if file == sys.argv[1]:
|
|
path = os.path.join(root, file)
|
|
ProcessFile(path)
|
|
return 0
|
|
|
|
print("File not found: " + sys.argv[1])
|
|
|
|
if __name__ == "__main__":
|
|
Main()
|