forked from RavinMaddHatter/Structura
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathstructura_core.py
230 lines (216 loc) · 10.7 KB
/
structura_core.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
import armor_stand_geo_class as asgc
import armor_stand_class ,structure_reader ,animation_class ,manifest ,os ,glob ,json ,shutil
import render_controller_class as rcc
import big_render_controller as brc
from shutil import copyfile
from zipfile import ZIP_DEFLATED, ZipFile
import time
import os
debug=False
with open("lookups/nbt_defs.json") as f:
nbt_def = json.load(f)
class structura:
def __init__(self,pack_name):
os.makedirs(pack_name)
self.timers={"start":time.time(),"previous":time.time()}
self.pack_name=pack_name
self.structure_files={}
self.rc=rcc.render_controller()
self.armorstand_entity = armor_stand_class.armorstand()
visual_name=pack_name
self.animation = animation_class.animations()
self.exclude_list=["minecraft:structure_block","minecraft:air"]
self.opacity=0.8
self.longestY=0
self.unsupported_blocks=[]
self.all_blocks={}
self.icon="lookups/pack_icon.png"
self.dead_blocks={}
def set_icon(self,icon):
self.icon=icon
def set_opacity(self,opacity):
self.opacity=opacity
def add_model(self,name,file_name):
self.structure_files[name]={}
self.structure_files[name]["file"]=file_name
self.structure_files[name]["offsets"]=None
def set_model_offset(self,name,offset):
self.structure_files[name]["offsets"]=offset
def generate_nametag_file(self):
## temp folder would be a good idea
name_tags=self.structure_files.keys()
fileName="{} Nametags.txt".format(self.pack_name)
with open(fileName,"w+") as text_file:
text_file.write("These are the nametags used in this file\n")
for name in name_tags:
text_file.write("{}\n".format(name))
def make_big_model(self,offset):
self.rc=brc.render_controller()
file_names=[]
for name in list(self.structure_files.keys()):
file_names.append(self.structure_files[name]["file"])
struct2make=structure_reader.combined_structures(file_names,exclude_list=self.exclude_list)
self.structure_files[""]={}
self.structure_files[""]["offsets"]=[0,0,0]
self.structure_files[""]["offsets"][1]= 0
for i in range(12):
self.armorstand_entity.add_model(str(i))
self.rc.add_geometry(str(i))
self.big_offset=offset
self.all_blocks=self._add_blocks_to_geo(struct2make,"",export_big=True)
self.armorstand_entity.export(self.pack_name)
def generate_with_nametags(self):
update_animation=True
for model_name in self.structure_files.keys():
if self.structure_files[model_name]["offsets"] is None:
offset=[0,0,0]
else:
offset=self.structure_files[model_name]["offsets"]
self.rc.add_model(model_name)
self.armorstand_entity.add_model(model_name)
## temp folder would be a good idea
copyfile(self.structure_files[model_name]["file"], "{}/{}.mcstructure".format(self.pack_name,model_name))
if debug:
print(self.structure_files[model_name]['offsets'])
struct2make = structure_reader.process_structure(self.structure_files[model_name]["file"])
blocks=self._add_blocks_to_geo(struct2make,model_name)
self.structure_files[model_name]["block_list"]=blocks
##consider temp folder
self.armorstand_entity.export(self.pack_name)## this may be in the wrong spot, but transfered from 1.5
def make_nametag_block_lists(self):
## consider temp file
file_names=[]
for model_name in self.structure_files.keys():
file_name="{}-{} block list.txt".format(self.pack_name,model_name)
file_names.append(file_name)
all_blocks = self.structure_files[model_name]["block_list"]
with open(file_name,"w+") as text_file:
text_file.write("This is a list of blocks, there is a known issue with variants, all blocks are reported as minecraft stores them\n")
for name in all_blocks.keys():
commonName = name.replace("minecraft:","")
text_file.write("{}: {}\n".format(commonName,all_blocks[name]))
return file_names
def make_big_blocklist(self):
## consider temp file
file_name="{} block list.txt".format(self.pack_name)
with open(file_name,"w+") as text_file:
text_file.write("This is a list of blocks, there is a known issue with variants, all blocks are reported as minecraft stores them\n")
for name in self.all_blocks.keys():
commonName = name.replace("minecraft:","")
text_file.write("{}: {}\n".format(commonName,self.all_blocks[name]))
def _add_blocks_to_geo(self,struct2make,model_name,export_big=False):
[xlen, ylen, zlen] = struct2make.get_size()
if export_big:
self.structure_files[model_name]['offsets'][0]-=xlen.item()+7
self.structure_files[model_name]['offsets'][2]-=zlen.item()+7
armorstand = asgc.armorstandgeo(model_name,alpha = self.opacity, size=[xlen, ylen, zlen], offsets=self.structure_files[model_name]['offsets'])
if ylen > self.longestY:
update_animation=True
longestY = ylen
else:
update_animation=False
for y in range(ylen):
#creates the layer for controlling. Note there is implied formating here
#for layer names
if y<12:
armorstand.make_layer(y)
#adds links the layer name to an animation
if update_animation and not export_big:
self.animation.insert_layer(y)
non_air=struct2make.get_layer_blocks(y)
for loc in non_air:
x=int(loc[0])
z=int(loc[1])
block = struct2make.get_block(x, y, z)
blk_name=block["name"].replace("minecraft:", "")
blockProp=self._process_block(block)
rot = blockProp[0]
top = blockProp[1]
variant = blockProp[2]
open_bit = blockProp[3]
data = blockProp[4]
if debug:
armorstand.make_block(x, y, z, blk_name, rot = rot, top = top,variant = variant, trap_open=open_bit, data=data, big = export_big)
else:
try:
armorstand.make_block(x, y, z, blk_name, rot = rot, top = top,variant = variant, trap_open=open_bit, data=data, big = export_big)
except:
self.unsupported_blocks.append("x:{} Y:{} Z:{}, Block:{}, Variant: {}".format(x,y,z,block["name"],variant))
print("There is an unsuported block in this world and it was skipped")
print("x:{} Y:{} Z:{}, Block:{}, Variant: {}".format(x,y,z,block["name"],variant))
if block["name"] not in self.dead_blocks.keys():
self.dead_blocks[block["name"]]={}
if type(variant) is list:
variant="_".join(variant)
if variant not in self.dead_blocks[block["name"]].keys():
self.dead_blocks[block["name"]][variant]=0
self.dead_blocks[block["name"]][variant]+=1
## consider temp file
if export_big:
armorstand.export_big(self.pack_name)
self.animation.export_big(self.pack_name,self.big_offset)
else:
armorstand.export(self.pack_name)
self.animation.export(self.pack_name)
return struct2make.get_block_list()
def compile_pack(self):
## consider temp file
nametags=list(self.structure_files.keys())
if len(nametags)>1:
manifest.export(self.pack_name,nameTags=nametags)
else:
manifest.export(self.pack_name)
copyfile(self.icon, f"{self.pack_name}/pack_icon.png")
larger_render = "lookups/armor_stand.larger_render.geo.json"
larger_render_path = f"{self.pack_name}/models/entity/armor_stand.larger_render.geo.json"
copyfile(larger_render, larger_render_path)
self.rc.export(self.pack_name)
file_paths = []
shutil.make_archive("{}".format(self.pack_name), 'zip', self.pack_name)
os.rename(f'{self.pack_name}.zip',f'{self.pack_name}.mcpack')
shutil.rmtree(self.pack_name)
print("Pack Making Completed")
self.timers["finished"]=time.time()-self.timers["previous"]
self.timers["total"]=time.time()-self.timers["start"]
return f'{self.pack_name}.mcpack'
def _process_block(self,block):
rot = None
top = False
open_bit = False
data=0
variant="default"
for key in nbt_def.keys():
if nbt_def[key]== "variant" and key in block["states"].keys():
variant = [key,block["states"][key]]
if nbt_def[key] == "rot" and key in block["states"].keys():
try:
rot = int(block["states"][key])
except:
rot = str(block["states"][key])
if nbt_def[key]== "top" and key in block["states"].keys():
top = bool(block["states"][key])
if nbt_def[key]== "open_bit" and "open_bit" in block["states"].keys():
open_bit = bool(block["states"][key])
if nbt_def[key]== "data" and key in block["states"].keys():
data = int(block["states"][key])
if key == "rail_direction" and key in block["states"].keys():
data = str(block["states"][key].as_unsigned)
if "rail_data_bit" in block["states"].keys():
data += "-"+str(block["states"]["rail_data_bit"].as_unsigned)
if "wood_type" in block["states"].keys():
variant = ["wood_type",block["states"]["wood_type"]]
if block["name"] == "minecraft:wood":
keys = block["states"]["wood_type"]
if bool(block["states"]["stripped_bit"]):
keys+="_stripped"
variant = ["wood",keys]
return [rot, top, variant, open_bit, data]
def get_skipped(self):
## temp folder would be a good idea
if len(self.unsupported_blocks)>1:
fileName="{} skipped.txt".format(self.pack_name)
with open(fileName,"w+") as text_file:
text_file.write("These are the skipped blocks\n")
for skipped in self.unsupported_blocks:
text_file.write(f"{skipped}\n")
return self.dead_blocks