88 lines
3.6 KiB
Java
88 lines
3.6 KiB
Java
/*
|
|
* Minecraft Forge
|
|
* Copyright (c) 2016-2020.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation version 2.1
|
|
* of the License.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
package net.minecraftforge.client.model.animation;
|
|
|
|
import java.util.List;
|
|
import java.util.function.Function;
|
|
|
|
import javax.annotation.Nullable;
|
|
|
|
import net.minecraft.client.Minecraft;
|
|
import net.minecraft.client.renderer.model.*;
|
|
import net.minecraft.client.renderer.texture.TextureAtlasSprite;
|
|
import net.minecraft.client.world.ClientWorld;
|
|
import net.minecraft.entity.LivingEntity;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.util.ResourceLocation;
|
|
import net.minecraft.world.World;
|
|
import net.minecraftforge.client.model.ModelLoader;
|
|
import net.minecraftforge.client.model.ModelTransformComposition;
|
|
import net.minecraftforge.common.model.animation.CapabilityAnimation;
|
|
|
|
public final class AnimationItemOverrideList extends ItemOverrideList
|
|
{
|
|
private final ModelBakery bakery;
|
|
private final IUnbakedModel model;
|
|
private final ResourceLocation modelLoc;
|
|
private final IModelTransform state;
|
|
|
|
|
|
private final Function<RenderMaterial, TextureAtlasSprite> bakedTextureGetter;
|
|
|
|
public AnimationItemOverrideList(ModelBakery bakery, IUnbakedModel model, ResourceLocation modelLoc, IModelTransform state, Function<RenderMaterial, TextureAtlasSprite> bakedTextureGetter, ItemOverrideList overrides)
|
|
{
|
|
this(bakery, model, modelLoc, state, bakedTextureGetter, overrides.getOverrides().reverse());
|
|
}
|
|
|
|
public AnimationItemOverrideList(ModelBakery bakery, IUnbakedModel model, ResourceLocation modelLoc, IModelTransform state, Function<RenderMaterial, TextureAtlasSprite> bakedTextureGetter, List<ItemOverride> overrides)
|
|
{
|
|
super(bakery, model, ModelLoader.defaultModelGetter(), bakedTextureGetter, overrides);
|
|
this.bakery = bakery;
|
|
this.model = model;
|
|
this.modelLoc = modelLoc;
|
|
this.state = state;
|
|
this.bakedTextureGetter = bakedTextureGetter;
|
|
}
|
|
|
|
@SuppressWarnings("resource")
|
|
@Override
|
|
public IBakedModel getOverrideModel(IBakedModel originalModel, ItemStack stack, @Nullable ClientWorld world, @Nullable LivingEntity entity)
|
|
{
|
|
return stack.getCapability(CapabilityAnimation.ANIMATION_CAPABILITY, null)
|
|
.map(asm ->
|
|
{
|
|
World w = world;
|
|
// TODO caching?
|
|
if(w == null && entity != null)
|
|
{
|
|
w = entity.world;
|
|
}
|
|
if(world == null)
|
|
{
|
|
w = Minecraft.getInstance().world;
|
|
}
|
|
return asm.apply(Animation.getWorldTime(world, Animation.getPartialTickTime())).getLeft();
|
|
})
|
|
// TODO where should uvlock data come from?
|
|
.map(state -> model.bakeModel(bakery, bakedTextureGetter, new ModelTransformComposition(state, this.state), modelLoc))
|
|
.orElseGet(() -> super.getOverrideModel(originalModel, stack, world, entity));
|
|
}
|
|
}
|