243 lines
7.2 KiB
Python
243 lines
7.2 KiB
Python
import zlib, gzip
|
|
import struct
|
|
import os
|
|
import sys
|
|
from io import StringIO
|
|
|
|
def nbt_read_tag(bytes, index, tag):
|
|
if tag <= 6:
|
|
plen = (None, 1, 2, 4, 8, 4, 8)[tag]
|
|
value = struct.unpack(">"+(None, "b", "h", "i", "q", "f", "d")[tag],
|
|
bytes[index:index+plen])[0]
|
|
index += plen
|
|
return value, index
|
|
elif tag == 7:
|
|
plen, index = nbt_read_tag(bytes, index, 3)
|
|
value = list(struct.unpack(">"+str(plen)+"B", bytes[index:index+plen]))
|
|
index += plen
|
|
return value, index
|
|
elif tag == 11:
|
|
plen, index = nbt_read_tag(bytes, index, 3)
|
|
value = list(struct.unpack(">"+str(plen)+"i", bytes[index:index+4*plen]))
|
|
index += 4*plen
|
|
return value, index
|
|
elif tag == 8:
|
|
plen, index = nbt_read_tag(bytes, index, 2)
|
|
value = bytes[index:index+plen].decode('utf-8')
|
|
index += plen
|
|
return value, index
|
|
elif tag == 9:
|
|
tagid = bytes[index]
|
|
index += 1
|
|
plen, index = nbt_read_tag(bytes, index, 3)
|
|
value = []
|
|
for i in range(plen):
|
|
v, index = nbt_read_tag(bytes, index, tagid)
|
|
value.append(v)
|
|
return value, index
|
|
elif tag == 10:
|
|
return nbt_read_named(bytes, index)
|
|
|
|
def nbt_read_named(bytes, index):
|
|
d = {}
|
|
while True:
|
|
if index >= len(bytes):
|
|
return d, index
|
|
tag = bytes[index]
|
|
index += 1
|
|
if tag == 0:
|
|
return d, index
|
|
name, index = nbt_read_tag(bytes, index, 8)
|
|
value, index = nbt_read_tag(bytes, index, tag)
|
|
d[name] = value
|
|
|
|
def nbt_read(bytes):
|
|
return nbt_read_named(bytes, 0)[0]
|
|
|
|
def to_int(bytes):
|
|
s = 0
|
|
for x in bytes:
|
|
s = (s<<8) + x
|
|
return s
|
|
|
|
def remove_half_bytes(d, key):
|
|
l = []
|
|
for x in d[key]:
|
|
l.append(x&0xf)
|
|
l.append((x>>4)&0xf)
|
|
d[key] = l
|
|
|
|
def to_anvil_data(block, key):
|
|
data = block[key]
|
|
del block[key]
|
|
l = [block["Sections"][i][key] for i in range(8)]
|
|
for yslice in range(8):
|
|
l_yslice = l[yslice]
|
|
k = 0
|
|
k2 = yslice << 4
|
|
for y in range(16):
|
|
for z in range(16):
|
|
for x in range(16):
|
|
l_yslice[k] = data[k2]
|
|
k += 1
|
|
k2 += 2048
|
|
k2 = (k2&0x7ff)+128
|
|
k2 = (k2&0x7f)+1
|
|
|
|
def format_block(block):
|
|
if block.get("isFormatted", False):
|
|
return block
|
|
block["isFormatted"] = True
|
|
if block.get("Sections", None) != None:
|
|
for section in block["Sections"]:
|
|
remove_half_bytes(section, "Data")
|
|
remove_half_bytes(section, "BlockLight")
|
|
remove_half_bytes(section, "SkyLight")
|
|
return block
|
|
remove_half_bytes(block, "Data")
|
|
remove_half_bytes(block, "BlockLight")
|
|
remove_half_bytes(block, "SkyLight")
|
|
block["Sections"] = [{"Y":i, "Blocks":[0]*4096, "Data":[0]*4096,
|
|
"BlockLight":[0]*4096, "SkyLight":[0]*4096}
|
|
for i in range(8)]
|
|
to_anvil_data(block, "Blocks")
|
|
to_anvil_data(block, "Data")
|
|
to_anvil_data(block, "BlockLight")
|
|
to_anvil_data(block, "SkyLight")
|
|
return block
|
|
|
|
|
|
def read_block(f, blockx, blockz):
|
|
ofs = (blockx%32 + (blockz%32)*32)*4
|
|
f.seek(ofs)
|
|
offset = to_int(f.read(3))<<12
|
|
if offset == 0:
|
|
return
|
|
f.seek(offset)
|
|
length = struct.unpack(">i", f.read(4))[0]
|
|
ctype = struct.unpack(">b", f.read(1))[0]
|
|
data = f.read(length - 1)
|
|
udata = zlib.decompress(data)
|
|
block = nbt_read(udata)['']["Level"]
|
|
return block
|
|
|
|
def read_dir(dirname):
|
|
filenames = [i for i in os.listdir(dirname) if i.endswith(".mcr") or i.endswith(".mca")]
|
|
blocks = {}
|
|
for filename in filenames:
|
|
#for filename in ("r.0.0.mca", "r.0.-1.mca", "r.-1.0.mca", "r.-1.-1.mca"):
|
|
s = filename.split(".")
|
|
chkx, chkz = int(s[1])*32, int(s[2])*32
|
|
with open(dirname+filename, "rb") as f:
|
|
for blockx in range(chkx, chkx+32):
|
|
for blockz in range(chkz, chkz+32):
|
|
#print("Reading block", blockx, blockz)
|
|
block = read_block(f, blockx, blockz)
|
|
#print("Block read")
|
|
if block != None:
|
|
blocks[(blockx, blockz)] = block
|
|
return blocks
|
|
|
|
def read_content():
|
|
f = open("map_content.txt", "r")
|
|
lines = f.readlines()
|
|
f.close()
|
|
|
|
lines.reverse()
|
|
|
|
bd = {}
|
|
for line in lines:
|
|
if line[-1] == "\n":
|
|
line = line[:-1]
|
|
line = line.split("#")[0] # Remove comment
|
|
s = line.split("\t")
|
|
if len(s) >= 2:
|
|
r = s[1].split(" ")
|
|
if len(r) == 0:
|
|
print(line)
|
|
raise ValueError("Malformed data")
|
|
name = r[0]
|
|
param2 = 0
|
|
for i in range(1, len(r)):
|
|
if r[i] != "":
|
|
param2 = int(r[i])
|
|
break
|
|
t = s[0].split(" ")
|
|
if len(t) == 2:
|
|
bd[(int(t[0]), int(t[1]))] = (name, param2)
|
|
elif len(t) == 1:
|
|
for data in range(16):
|
|
bd[(int(t[0]), data)] = (name, param2)
|
|
|
|
blocks_len = max([i[0] for i in bd.keys()])+1
|
|
blocks = [[(None, 0),]*16 for i in range(blocks_len)]
|
|
for (id, data), value in bd.items():
|
|
blocks[id][data] = value
|
|
return blocks
|
|
|
|
blocks_id = read_content()
|
|
|
|
def get_name(id, data):
|
|
if id >= len(blocks_id):
|
|
return None, 0
|
|
return blocks_id[id][data]
|
|
|
|
def convert_section(section, tileentities, yslice):
|
|
n = len(section["Blocks"])
|
|
blocksdata = section["Blocks"]
|
|
data = section["Data"]
|
|
blocks = [None]*n
|
|
param1 = [0]*n
|
|
param2 = [0]*n
|
|
metadata = [None]*n
|
|
for i in range(n):
|
|
blocks[i], param2[i] = get_name(blocksdata[i], data[i])
|
|
return (blocks, param1, param2, metadata)
|
|
|
|
def convert_block(block, yslice):
|
|
for section in block["Sections"]:
|
|
if section["Y"] == yslice:
|
|
return convert_section(section, block.get("TileEntities", []), yslice)
|
|
return None
|
|
|
|
def export_we(b, f):
|
|
blocks, param1, param2, metadata = b
|
|
L = []
|
|
f.write("return {")
|
|
write_comma = False
|
|
for i in range(len(blocks)):
|
|
x = i%16
|
|
z = (i//16)%16
|
|
y = i//256
|
|
if blocks[i] != None:
|
|
if write_comma:
|
|
f.write(",")
|
|
else:
|
|
write_comma = True
|
|
f.write("{x="+str(x)+",y="+str(y)+",z="+str(z)+',name="'+blocks[i]+'",param1='+str(param1[i])+",param2="+str(param2[i])+",meta={fields={},inventory={}}}")
|
|
f.write("}")
|
|
|
|
inputdir = sys.argv[1] + "region/"
|
|
blocks = read_dir(inputdir)
|
|
print("Reading done")
|
|
|
|
outputdir = sys.argv[2] + "blocks/"
|
|
if not os.path.exists(outputdir):
|
|
os.mkdir(outputdir)
|
|
|
|
done = 0
|
|
max_number = len(blocks)
|
|
for key, value in blocks.items():
|
|
done += 1
|
|
if done%20 == 0:
|
|
print("Exported {} blocks on {}".format(done, max_number))
|
|
if value == None:
|
|
continue
|
|
for yslice in range(16):
|
|
value = format_block(value)
|
|
b = convert_block(value, yslice)
|
|
if b == None:
|
|
continue
|
|
with open(outputdir+str(key[0])+"."+str(yslice-4)+"."+str(key[1])+".we", "w") as f:
|
|
export_we(b, f)
|