init
This commit is contained in:
commit
9809bb55e5
GPG key ID:
22618F758B5BE2E5
6 changed files with 305 additions and 0 deletions
33
mcdraw.py
Normal file
33
mcdraw.py
Normal file
|
|
@ -0,0 +1,33 @@
|
|||
from PIL import ImageFont, Image, ImageDraw
|
||||
import re
|
||||
try:
|
||||
font = ImageFont.truetype("wqy-zenhei.ttc", 24, 1)
|
||||
emfont = ImageFont.truetype("NotoEmoji-Regular.ttf", 24, 0)
|
||||
except OSError:
|
||||
print("Please download these fonts to the current dir:\n"
|
||||
"wqy-zenhei.ttc\n\tfrom https://downloads.sourceforge.net/project/wqy/wqy-zenhei/0.9.45%20%28Fighting-state%20RC1%29/wqy-zenhei-0.9.45.tar.gz"
|
||||
"\n"
|
||||
"NotoEmoji-Regular.ttf\n\tfrom https://github.com/googlefonts/noto-emoji/raw/2f1ffdd6fbbd05d6f382138a3d3adcd89c5ce800/fonts/NotoEmoji-Regular.ttf"
|
||||
"\n\n"
|
||||
)
|
||||
raise
|
||||
|
||||
def text_to_im(text="紧密团结在鱼塔同志周围"):
|
||||
emoji = re.compile("^[" u"\U0001F600-\U0001F64F" u"\U0001F300-\U0001F5FF" u"\U0001F680-\U0001F6FF" u"\U0001F1E0-\U0001F1FF" "]$", flags=re.UNICODE)
|
||||
brush_location = [0]
|
||||
def draw_single_char(ch, draw=None):
|
||||
m = emoji.match(ch)
|
||||
myfont = emfont if m else font
|
||||
width = myfont.getsize(ch)[0]
|
||||
if draw:
|
||||
draw.text((brush_location[0], -2 if m else 0), ch, fill="black", font=myfont, spacing=0)
|
||||
brush_location[0] += width
|
||||
return width
|
||||
def max_height(text):
|
||||
return max(*[font.getsize(ch)[1] for ch in text], *[emfont.getsize(ch)[1] for ch in text])
|
||||
width, height = sum(draw_single_char(ch) for ch in text), max_height(text)
|
||||
im = Image.new('1', (width, height), color="white")
|
||||
draw = ImageDraw.Draw(im)
|
||||
for ch in text:
|
||||
draw_single_char(ch, draw)
|
||||
return im
|
||||
20
minecraft-concrete.gpl
Normal file
20
minecraft-concrete.gpl
Normal file
|
|
@ -0,0 +1,20 @@
|
|||
GIMP Palette
|
||||
Name: minecraft-concrete
|
||||
Columns: 0
|
||||
#
|
||||
208 214 215 white_concrete
|
||||
226 98 0 orange_concrete
|
||||
170 45 160 magenta_concrete
|
||||
30 137 198 light_blue_concrete
|
||||
242 176 13 yellow_concrete
|
||||
96 171 19 lime_concrete
|
||||
214 100 143 pink_concrete
|
||||
51 55 59 gray_concrete
|
||||
126 126 116 light_gray_concrete
|
||||
15 121 138 cyan_concrete
|
||||
101 26 158 purple_concrete
|
||||
40 42 144 blue_concrete
|
||||
97 59 27 brown_concrete
|
||||
71 90 31 green_concrete
|
||||
144 29 29 red_concrete
|
||||
2 3 7 black_concrete
|
||||
57
minecraft-fullblock.gpl
Normal file
57
minecraft-fullblock.gpl
Normal file
|
|
@ -0,0 +1,57 @@
|
|||
GIMP Palette
|
||||
Name: minecraft-fullblock
|
||||
Columns: 0
|
||||
#
|
||||
247 233 163 smooth_sandstone
|
||||
199 199 199 mushroom_stem[east=true,west=true,north=true,south=true,up=true,down=true]
|
||||
255 0 0 redstone_block
|
||||
160 160 255 packed_ice
|
||||
167 167 167 iron_block
|
||||
0 124 0 oak_leaves[distance=7,persistent=true]
|
||||
255 255 255 white_concrete
|
||||
164 168 184 clay
|
||||
151 109 77 dirt
|
||||
112 112 112 stone
|
||||
143 119 72 oak_planks
|
||||
255 252 245 quartz_block
|
||||
216 127 51 orange_concrete
|
||||
178 76 216 magenta_concrete
|
||||
102 153 216 light_blue_concrete
|
||||
229 229 51 yellow_concrete
|
||||
127 204 25 lime_concrete
|
||||
242 127 165 pink_concrete
|
||||
76 76 76 gray_concrete
|
||||
153 153 153 light_gray_concrete
|
||||
76 127 153 cyan_concrete
|
||||
127 63 178 purple_concrete
|
||||
51 76 178 blue_concrete
|
||||
102 76 51 brown_concrete
|
||||
102 127 51 green_concrete
|
||||
153 51 51 red_concrete
|
||||
25 25 25 black_concrete
|
||||
250 238 77 gold_block
|
||||
92 219 213 diamond_block
|
||||
74 128 255 lapis_block
|
||||
0 217 58 emerald_block
|
||||
112 2 0 netherrack
|
||||
209 177 161 white_terracotta
|
||||
159 82 36 orange_terracotta
|
||||
149 87 108 magenta_terracotta
|
||||
112 108 138 light_blue_terracotta
|
||||
186 133 36 yellow_terracotta
|
||||
103 117 53 lime_terracotta
|
||||
160 77 78 pink_terracotta
|
||||
57 41 35 gray_terracotta
|
||||
135 107 98 light_gray_terracotta
|
||||
87 92 92 cyan_terracotta
|
||||
122 73 88 purple_terracotta
|
||||
76 62 92 blue_terracotta
|
||||
76 50 35 brown_terracotta
|
||||
76 82 42 green_terracotta
|
||||
142 60 46 red_terracotta
|
||||
37 22 16 black_terracotta
|
||||
189 48 49 crimson_nylium
|
||||
148 63 97 crimson_planks
|
||||
22 126 134 warped_nylium
|
||||
58 142 140 warped_planks
|
||||
20 180 133 warped_wart_block
|
||||
62
minecraft.gpl
Normal file
62
minecraft.gpl
Normal file
|
|
@ -0,0 +1,62 @@
|
|||
GIMP Palette
|
||||
Name: minecraft
|
||||
Columns: 0
|
||||
#
|
||||
127 178 56 grass_block[snowy=false]
|
||||
247 233 163 smooth_sandstone
|
||||
199 199 199 mushroom_stem[east=true,west=true,north=true,south=true,up=true,down=true]
|
||||
255 0 0 redstone_block
|
||||
160 160 255 packed_ice
|
||||
167 167 167 iron_block
|
||||
0 124 0 oak_leaves[distance=7,persistent=true]
|
||||
255 255 255 white_concrete
|
||||
164 168 184 clay
|
||||
151 109 77 dirt
|
||||
112 112 112 stone
|
||||
64 64 255 water[level=0]
|
||||
143 119 72 oak_planks
|
||||
255 252 245 quartz_block
|
||||
216 127 51 orange_concrete
|
||||
178 76 216 magenta_concrete
|
||||
102 153 216 light_blue_concrete
|
||||
229 229 51 yellow_concrete
|
||||
127 204 25 lime_concrete
|
||||
242 127 165 pink_concrete
|
||||
76 76 76 gray_concrete
|
||||
153 153 153 light_gray_concrete
|
||||
76 127 153 cyan_concrete
|
||||
127 63 178 purple_concrete
|
||||
51 76 178 blue_concrete
|
||||
102 76 51 brown_concrete
|
||||
102 127 51 green_concrete
|
||||
153 51 51 red_concrete
|
||||
25 25 25 black_concrete
|
||||
250 238 77 gold_block
|
||||
92 219 213 diamond_block
|
||||
74 128 255 lapis_block
|
||||
0 217 58 emerald_block
|
||||
129 86 49 podzol[snowy=false]
|
||||
112 2 0 netherrack
|
||||
209 177 161 white_terracotta
|
||||
159 82 36 orange_terracotta
|
||||
149 87 108 magenta_terracotta
|
||||
112 108 138 light_blue_terracotta
|
||||
186 133 36 yellow_terracotta
|
||||
103 117 53 lime_terracotta
|
||||
160 77 78 pink_terracotta
|
||||
57 41 35 gray_terracotta
|
||||
135 107 98 light_gray_terracotta
|
||||
87 92 92 cyan_terracotta
|
||||
122 73 88 purple_terracotta
|
||||
76 62 92 blue_terracotta
|
||||
76 50 35 brown_terracotta
|
||||
76 82 42 green_terracotta
|
||||
142 60 46 red_terracotta
|
||||
37 22 16 black_terracotta
|
||||
189 48 49 crimson_nylium
|
||||
148 63 97 crimson_planks
|
||||
92 25 29 crimson_hyphae[axis=y]
|
||||
22 126 134 warped_nylium
|
||||
58 142 140 warped_planks
|
||||
86 44 62 warped_hyphae[axis=y]
|
||||
20 180 133 warped_wart_block
|
||||
2
requirements.txt
Normal file
2
requirements.txt
Normal file
|
|
@ -0,0 +1,2 @@
|
|||
pillow==8.3.2
|
||||
PyNBT==3.1.0
|
||||
131
schemgen.py
Normal file
131
schemgen.py
Normal file
|
|
@ -0,0 +1,131 @@
|
|||
from PIL import Image
|
||||
from pathlib import Path
|
||||
from itertools import zip_longest
|
||||
import re
|
||||
|
||||
def read_palette(text):
|
||||
palette = dict()
|
||||
metadata = dict()
|
||||
in_body = False
|
||||
for line in text.split('\n'):
|
||||
line = line.strip()
|
||||
if not line or line == "GIMP Palette":
|
||||
continue
|
||||
if line == '#':
|
||||
in_body = True
|
||||
continue
|
||||
if in_body:
|
||||
*rgb, name = line.split()
|
||||
r, g, b = [int(x) for x in rgb]
|
||||
index = len(palette)
|
||||
palette[f"{r:02x}{g:02x}{b:02x}"] = {"rgb": (r, g, b), "name": name if name.startswith("minecraft:") else f"minecraft:{name}", "index": index}
|
||||
else:
|
||||
k, *v = line.split(':')
|
||||
k, v = k.strip(), ':'.join(v).strip()
|
||||
metadata[k] = v
|
||||
return {"metadata": metadata, "palette": palette}
|
||||
|
||||
def read_imtuple(impath, palette_path):
|
||||
palette = read_palette(Path(palette_path).read_text())
|
||||
im = Image.open(impath)
|
||||
im = im.rotate(180)
|
||||
im = im.convert('RGB')
|
||||
used_color_hex = tuple(set(f"{r:02x}{g:02x}{b:02x}" for r, g, b in im.getdata()))
|
||||
used_mc_palette = tuple(palette["palette"][h]["name"] for h in used_color_hex)
|
||||
imdata = (used_color_hex.index(f"{r:02x}{g:02x}{b:02x}") for r, g, b in im.getdata())
|
||||
#imtuple = tuple(zip_longest(*([iter(imdata)] * im.width), fillvalue=0))
|
||||
#assert len(imtuple) == im.height
|
||||
imtuple = imdata
|
||||
return (used_mc_palette, imtuple, im.width, im.height)
|
||||
|
||||
def read_text(text):
|
||||
from mcdraw import text_to_im
|
||||
im = text_to_im(text)
|
||||
used_mc_palette = ('minecraft:air', 'minecraft:stone')
|
||||
imdata = (0 if i else 1 for i in im.getdata())
|
||||
imtuple = imdata
|
||||
return (used_mc_palette, imtuple, im.width, im.height)
|
||||
|
||||
import gzip
|
||||
from pynbt import NBTFile, TAG_Int, TAG_Compound, TAG_Short, TAG_Byte_Array, TAG_List, TAG_Int_Array
|
||||
from io import BytesIO
|
||||
import struct
|
||||
|
||||
class VarInt:
|
||||
MAX_BYTES = 5
|
||||
def __init__(self, io, set_value=None):
|
||||
self._io = io
|
||||
if set_value:
|
||||
self.setval(set_value)
|
||||
def setval(self, value):
|
||||
ret = bytes()
|
||||
while True:
|
||||
byte = value & 0x7F
|
||||
value >>= 7
|
||||
ret += struct.pack("B", byte | (0x80 if value > 0 else 0))
|
||||
if value == 0:
|
||||
break
|
||||
if len(ret) > self.MAX_BYTES:
|
||||
raise ValueError("Tried to write too long of a VarInt")
|
||||
self._io.write(ret)
|
||||
@property
|
||||
def value(self):
|
||||
number = 0
|
||||
bytes_encountered = 0
|
||||
while True:
|
||||
byte = self._io.read(1)
|
||||
if len(byte) < 1:
|
||||
raise EOFError("Unexpected end of message.")
|
||||
byteord = ord(byte)
|
||||
number |= (byteord & 0x7F) << 7 * bytes_encountered
|
||||
if not byteord & 0x80:
|
||||
break
|
||||
|
||||
bytes_encountered += 1
|
||||
if bytes_encountered > self.MAX_BYTES:
|
||||
raise ValueError("Tried to read too long of a VarInt")
|
||||
return number
|
||||
|
||||
def assemble_nbtfile(used_mc_palette, imtuple, width, height):
|
||||
blackdataio = BytesIO()
|
||||
varint = VarInt(blackdataio)
|
||||
for i in imtuple:
|
||||
varint.setval(i)
|
||||
nbtvalue = {
|
||||
'Version': TAG_Int(2),
|
||||
'PaletteMax': TAG_Int(len(used_mc_palette)),
|
||||
'Palette': TAG_Compound(
|
||||
{name: TAG_Int(i) for i, name in enumerate(used_mc_palette)}
|
||||
),
|
||||
'Metadata': TAG_Compound({
|
||||
'WEOffsetX': TAG_Int(0),
|
||||
'WEOffsetY': TAG_Int(0),
|
||||
'WEOffsetZ': TAG_Int(0)
|
||||
}),
|
||||
'Width': TAG_Short(width),
|
||||
'Height': TAG_Short(1),
|
||||
'Length': TAG_Short(height),
|
||||
'DataVersion': TAG_Int(2586),
|
||||
'BlockData': TAG_Byte_Array(blackdataio.getvalue()),
|
||||
'BlockEntities': TAG_List(TAG_Compound),
|
||||
'Offset': TAG_Int_Array([0, 0, 0])
|
||||
}
|
||||
nbt = NBTFile(value=nbtvalue, name='Schematic')
|
||||
return nbt
|
||||
|
||||
if __name__ == "__main__":
|
||||
import argparse
|
||||
parser = argparse.ArgumentParser(description='schemgen')
|
||||
parser.add_argument('-o', '--output', default='output.schem', help='output worldedit schema file')
|
||||
parser.add_argument('-i', '--input', type=str, default="input.png", help='input image')
|
||||
parser.add_argument('-t', '--text', type=str, default="", help='input text')
|
||||
parser.add_argument('-p', '--palette', type=str, default="minecraft.gpl", help='gimp palette file')
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.text:
|
||||
nbt = assemble_nbtfile(*read_text(args.text))
|
||||
else:
|
||||
nbt = assemble_nbtfile(*read_imtuple(args.input, args.palette))
|
||||
print(nbt.pretty())
|
||||
with gzip.open(args.output, 'w') as f:
|
||||
nbt.save(f)
|
||||
Loading…
Reference in a new issue