1
0
mirror of https://github.com/sys4-fr/server-nalc.git synced 2024-12-26 10:40:37 +01:00
server-nalc/mods/automappercolors/process.py
2015-12-17 18:45:15 +01:00

122 lines
5.1 KiB
Python
Executable File

#!/usr/bin/env python2
# Automappercolors by gravgun
# WTFPL
# Note: running this in Pypy has been tested to be ~3x faster
from __future__ import print_function
import sys, os, re
from PIL import Image
predefined = {
#"^default:river_water_([a-z]+)": {'r': 39, 'g': 66, 'b': 106, 'a': 128, 't': 224},
#"^default:lava_([a-z]+)": {'r': 255, 'g': 100, 'b': 0},
"^default:([a-z_]*)glass": {'a': 64, 't': 16},
"^default:torch": {'r': 255, 'g': 255, 'b': 0, 'a': 255},
"^default:ice": {'r': 74, 'g': 105, 'b': 159, 'a': 159},
"^default:water_([a-z]+)": {'r': 43, 'g': 97, 'b': 183, 'a': 128},
"^default:dirt_with_grass": {'r': 107, 'g': 134, 'b': 51, 'a': 255},
"^flowers:cotton_plant": {'r': 199, 'g': 218, 'b': 158},
"^flowers:seaweed": {'r': 48, 'g': 114, 'b': 107},
"^flowers:waterlily": {'r': 119, 'g': 166, 'b': 100},
"^flowers:waterlily_225": {'r': 119, 'g': 166, 'b': 100},
"^flowers:waterlily_45": {'r': 119, 'g': 166, 'b': 100},
"^flowers:waterlily_675": {'r': 119, 'g': 166, 'b': 100},
"^flowers:dandelion_white": {'r': 161, 'g': 174, 'b': 149},
"^flowers:dandelion_yellow": {'r': 144, 'g': 138, 'b': 0},
"^flowers:geranium": {'r': 75, 'g': 101, 'b': 84},
"^flowers:rose": {'r': 153, 'g': 9, 'b': 0},
"^flowers:tulip": {'r': 175, 'g': 114, 'b': 0},
"^flowers:viola": {'r': 84, 'g': 90, 'b': 64},
"^glass_arena:wall": {'r': 255, 'g': 0, 'b': 0}
}
predef_compiled = {}
for k in predefined:
predef_compiled[re.compile(k)] = predefined[k]
if len(sys.argv) <= 2:
print("Usage: %s <world path> <minetest data root path>" % sys.argv[0])
else:
pngpaths = {}
for root, dirs, files in os.walk(sys.argv[2]):
for dir in dirs:
if dir[0] == ".": # No dotdirs
dirs.remove(dir)
for file in files:
pngpaths[file] = os.path.join(root, file)
out = open(os.path.join(sys.argv[1], "colors.txt"), 'w')
f = open(os.path.join(sys.argv[1], "amc_nodes.txt"), 'r')
for line in f:
ldata = line.split(' ')
if len(ldata) == 2:
nodename = ldata[0]
tex = ldata[1][:-1] # Strip newline char
try:
r_override = None
g_override = None
b_override = None
a_override = None
t_override = None
compute = True
for k in predef_compiled:
if k.match(nodename) is not None:
v = predef_compiled[k]
if 'r' in v:
r_override = ccumul[0] = v['r']
if 'g' in v:
g_override = ccumul[1] = v['g']
if 'b' in v:
b_override = ccumul[2] = v['b']
if 'a' in v:
a_override = a = v['a']
if 't' in v:
t_override = t = v['t']
compute = not('r' in v and 'g' in v and 'b' in v and 'a' in v)
if compute:
inp = Image.open(pngpaths[tex])
# Flaky PILlow bug causing unclosed fp loss during convert
# resulting in a Too many files open IOError
inpfp = inp.fp
inp2 = inp.convert('RGBA')
ind = inp2.load()
inpfp.close()
pixcount = 0
ccumul = [0, 0, 0, 0]
for x in range(inp.size[0]):
for y in range(inp.size[1]):
pxl = ind[x, y]
ccumul[0] += (pxl[0]*pxl[3])/255
ccumul[1] += (pxl[1]*pxl[3])/255
ccumul[2] += (pxl[2]*pxl[3])/255
ccumul[3] += pxl[3]
a = ccumul[3]/255
if a > 0:
for i in range(3):
ccumul[i] /= a
if r_override is not None:
ccumul[0] = r_override
if g_override is not None:
ccumul[1] = g_override
if b_override is not None:
ccumul[2] = b_override
if a_override is None:
a = ccumul[3]/(inp.size[0]*inp.size[1])
else:
a = a_override
if t_override is None:
t = 255-a
else:
t = t_override
if t != 0:
out.write("%s %d %d %d %d %d\n" % (nodename, ccumul[0], ccumul[1], ccumul[2], a, t))
elif a != 255:
out.write("%s %d %d %d %d\n" % (nodename, ccumul[0], ccumul[1], ccumul[2], a))
else:
out.write("%s %d %d %d\n" % (nodename, ccumul[0], ccumul[1], ccumul[2]))
except KeyError:
print("Skip texture %s for %s" % (tex, nodename))
out.close()
f.close()