50 lines
1.8 KiB
Java
50 lines
1.8 KiB
Java
package moe.nea.firmament.mixins;
|
|
|
|
import moe.nea.firmament.events.BakeExtraModelsEvent;
|
|
import net.minecraft.client.render.model.ModelLoader;
|
|
import net.minecraft.client.render.model.UnbakedModel;
|
|
import net.minecraft.client.util.ModelIdentifier;
|
|
import net.minecraft.util.Identifier;
|
|
import org.spongepowered.asm.mixin.Final;
|
|
import org.spongepowered.asm.mixin.Mixin;
|
|
import org.spongepowered.asm.mixin.Shadow;
|
|
import org.spongepowered.asm.mixin.Unique;
|
|
import org.spongepowered.asm.mixin.injection.At;
|
|
import org.spongepowered.asm.mixin.injection.Inject;
|
|
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
|
|
|
|
import java.util.Map;
|
|
|
|
@Mixin(ModelLoader.class)
|
|
public abstract class CustomModelBakerPatch {
|
|
|
|
@Shadow
|
|
@Final
|
|
private Map<ModelIdentifier, UnbakedModel> modelsToBake;
|
|
|
|
@Shadow
|
|
protected abstract void loadItemModel(ModelIdentifier id);
|
|
|
|
@Shadow
|
|
abstract UnbakedModel getOrLoadModel(Identifier id);
|
|
|
|
@Shadow
|
|
protected abstract void add(ModelIdentifier id, UnbakedModel model);
|
|
|
|
@Unique
|
|
private void loadNonItemModel(ModelIdentifier identifier) {
|
|
UnbakedModel unbakedModel = this.getOrLoadModel(identifier.id());
|
|
this.add(identifier, unbakedModel);
|
|
}
|
|
|
|
|
|
@Inject(method = "bake", at = @At("HEAD"))
|
|
public void onBake(ModelLoader.SpriteGetter spliteGetter, CallbackInfo ci) {
|
|
BakeExtraModelsEvent.Companion.publish(new BakeExtraModelsEvent(this::loadItemModel, this::loadNonItemModel));
|
|
modelsToBake.values().forEach(model -> model.setParents(this::getOrLoadModel));
|
|
// modelsToBake.keySet().stream()
|
|
// .filter(it -> !it.id().getNamespace().equals("minecraft"))
|
|
// .forEach(it -> System.out.println("Non minecraft texture is being loaded: " + it));
|
|
}
|
|
}
|