92 lines
3.5 KiB
Java
92 lines
3.5 KiB
Java
package ru.betterend.mixin.client;
|
|
|
|
import java.io.InputStreamReader;
|
|
import java.io.Reader;
|
|
import java.nio.charset.StandardCharsets;
|
|
import java.util.Optional;
|
|
|
|
import org.spongepowered.asm.mixin.Final;
|
|
import org.spongepowered.asm.mixin.Mixin;
|
|
import org.spongepowered.asm.mixin.Shadow;
|
|
import org.spongepowered.asm.mixin.injection.At;
|
|
import org.spongepowered.asm.mixin.injection.Inject;
|
|
import org.spongepowered.asm.mixin.injection.ModifyVariable;
|
|
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
|
|
|
|
import net.minecraft.block.Block;
|
|
import net.minecraft.client.render.model.ModelLoader;
|
|
import net.minecraft.client.render.model.json.JsonUnbakedModel;
|
|
import net.minecraft.item.Item;
|
|
import net.minecraft.resource.Resource;
|
|
import net.minecraft.resource.ResourceManager;
|
|
import net.minecraft.util.Identifier;
|
|
import net.minecraft.util.registry.Registry;
|
|
import ru.betterend.BetterEnd;
|
|
import ru.betterend.patterns.Patterned;
|
|
import ru.betterend.world.generator.GeneratorOptions;
|
|
|
|
@Mixin(ModelLoader.class)
|
|
public class ModelLoaderMixin {
|
|
@Final
|
|
@Shadow
|
|
private ResourceManager resourceManager;
|
|
|
|
@Inject(method = "loadModelFromJson", at = @At("HEAD"), cancellable = true)
|
|
private void be_loadModelPattern(Identifier id, CallbackInfoReturnable<JsonUnbakedModel> info) {
|
|
if (id.getNamespace().equals(BetterEnd.MOD_ID)) {
|
|
Identifier modelId = new Identifier(id.getNamespace(), "models/" + id.getPath() + ".json");
|
|
JsonUnbakedModel model;
|
|
try (Resource resource = this.resourceManager.getResource(modelId)) {
|
|
Reader reader = new InputStreamReader(resource.getInputStream(), StandardCharsets.UTF_8);
|
|
model = JsonUnbakedModel.deserialize(reader);
|
|
model.id = id.toString();
|
|
info.setReturnValue(model);
|
|
} catch (Exception ex) {
|
|
String data[] = id.getPath().split("/");
|
|
if (data.length > 1) {
|
|
Identifier itemId = new Identifier(id.getNamespace(), data[1]);
|
|
Optional<Block> block = Registry.BLOCK.getOrEmpty(itemId);
|
|
if (block.isPresent()) {
|
|
if (block.get() instanceof Patterned) {
|
|
Patterned patterned = (Patterned) block.get();
|
|
model = this.be_getModel(data, id, patterned);
|
|
info.setReturnValue(model);
|
|
}
|
|
} else {
|
|
Optional<Item> item = Registry.ITEM.getOrEmpty(itemId);
|
|
if (item.isPresent() && item.get() instanceof Patterned) {
|
|
Patterned patterned = (Patterned) item.get();
|
|
model = this.be_getModel(data, id, patterned);
|
|
info.setReturnValue(model);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private JsonUnbakedModel be_getModel(String data[], Identifier id, Patterned patterned) {
|
|
String pattern;
|
|
if (id.getPath().contains("item")) {
|
|
pattern = patterned.getModelPattern(id.getPath());
|
|
} else {
|
|
if (data.length > 2) {
|
|
pattern = patterned.getModelPattern(data[2]);
|
|
} else {
|
|
pattern = patterned.getModelPattern(data[1]);
|
|
}
|
|
}
|
|
JsonUnbakedModel model = JsonUnbakedModel.deserialize(pattern);
|
|
model.id = id.toString();
|
|
|
|
return model;
|
|
}
|
|
|
|
@ModifyVariable(method = "loadModel", ordinal = 2, at = @At(value = "INVOKE"))
|
|
public Identifier be_SwitchModel(Identifier id) {
|
|
if (GeneratorOptions.changeChorusPlant() && id.getNamespace().equals("minecraft") && id.getPath().startsWith("blockstates/") && id.getPath().contains("chorus") && !id.getPath().contains("custom_")) {
|
|
id = new Identifier(id.getPath().replace("chorus", "custom_chorus"));
|
|
}
|
|
return id;
|
|
}
|
|
}
|