MinedMap/resource/extract.py
Matthias Schiffer 48a6e242ea
resource: add WallSign flag to block types
Allow to distinguish wall signs from freestanding or -hanging signs.
2024-01-10 22:56:29 +01:00

74 lines
1.5 KiB
Python
Executable file

#!/usr/bin/env python3
import json
import os
import sys
from PIL import Image
if len(sys.argv) != 4:
sys.exit('Usage: extract.py <blocks.json> <asset directory> <colors.json>')
def mean_color(texture):
path = os.path.join(sys.argv[2], texture + '.png')
im = Image.open(path)
data = im.convert('RGBA').getdata()
a = sum([a for (r, g, b, a) in data])
if a == 0:
return None
r = sum([r * a for (r, g, b, a) in data])
g = sum([g * a for (r, g, b, a) in data])
b = sum([b * a for (r, g, b, a) in data])
return {
'r': r / a,
'g': g / a,
'b': b / a,
}
with open(sys.argv[1]) as f:
blocks = json.load(f)
output = {}
for name, info in blocks.items():
id = name
output[id] = {
'color': {'r': 0, 'g': 0, 'b': 0},
'opaque': False,
'grass': False,
'foliage': False,
'birch': False,
'spruce': False,
'water': False,
'wall_sign': False,
'sign_material': None,
}
if info is None:
continue
texture = info.get('texture', name)
color = None
if texture:
color = mean_color(texture)
if color:
output[id]['color'] = color
output[id]['opaque'] = True
output[id]['grass'] = info.get('grass', False)
output[id]['foliage'] = info.get('foliage', False)
output[id]['birch'] = info.get('birch', False)
output[id]['spruce'] = info.get('spruce', False)
output[id]['water'] = info.get('water', False)
output[id]['wall_sign'] = info.get('wall_sign', False)
output[id]['sign_material'] = info.get('sign_material')
with open(sys.argv[3], 'w') as f:
json.dump(output, f)