/
ModelBakeEventListener.java
45 lines (35 loc) · 1.75 KB
/
ModelBakeEventListener.java
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
package slimeknights.tmechworks.common.event;
import net.minecraft.client.renderer.model.IBakedModel;
import net.minecraft.util.ResourceLocation;
import net.minecraftforge.client.event.ModelBakeEvent;
import net.minecraftforge.eventbus.api.SubscribeEvent;
import net.minecraftforge.fml.common.Mod;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import slimeknights.tmechworks.TMechworks;
import slimeknights.tmechworks.client.model.DisguiseBakedModel;
import java.util.*;
import java.util.stream.Collectors;
@Mod.EventBusSubscriber(modid = TMechworks.modId, bus = Mod.EventBusSubscriber.Bus.MOD)
public class ModelBakeEventListener {
private static Logger log = LogManager.getLogger(TMechworks.modId + ".model");
private static final Set<String> disguiseables = new HashSet<>();
public static void registerDisguiseBlock(ResourceLocation location) {
disguiseables.add(location.toString());
}
@SubscribeEvent
public static void onModelBake(ModelBakeEvent event){
float time = System.nanoTime();
log.info("Starting model bake at " + time);
List<Map.Entry<ResourceLocation, IBakedModel>> registry = new ArrayList<>(event.getModelRegistry().entrySet());
for(int i = 0; i < registry.size(); i++){
Map.Entry<ResourceLocation, IBakedModel> model = registry.get(i);
String modelId = model.getKey().toString();
modelId = modelId.substring(0, modelId.indexOf("#"));
if(disguiseables.contains(modelId)) {
event.getModelRegistry().put(model.getKey(), new DisguiseBakedModel(model.getValue()));
}
}
log.info("Model bake finished in " + (System.nanoTime() - time) / 1000000000F + " seconds");
}
}