mirror of
https://github.com/Bubberstation/Bubberstation.git
synced 2025-12-16 12:43:09 +00:00
3df5d3b42b
Each of these is an extra operation TGS has to do when copying the repo. Who is committing these?
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()
|