infiniteserver/fabric
1
0
Fork 0
forked from FabricMC/fabric
Code Pull requests Activity

fix FabricEntityTypeBuilder

Browse source
This commit is contained in:
asie 2018-12-12 13:41:56 +01:00
parent d3e6e9ca73
commit 7699ef9dc5
1 changed files with 17 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/main/java/net/fabricmc/fabric/entity/FabricEntityTypeBuilder.java
View file

@ -25,30 +25,34 @@ import java.util.function.Function;
// TODO: javadocs // TODO: javadocs
public class FabricEntityTypeBuilder<T extends Entity> { public class FabricEntityTypeBuilder<T extends Entity> {
protected final EntityType.Builder<T> delegate; private final Class<? extends T> entityClass;
private final Function<? super World, ? extends T> function;
private boolean saveable = true;
private boolean summonable = true;
private int trackingDistance = -1; private int trackingDistance = -1;
private int updateIntervalTicks = -1; private int updateIntervalTicks = -1;
private boolean alwaysUpdateVelocity = true; private boolean alwaysUpdateVelocity = true;
protected FabricEntityTypeBuilder(EntityType.Builder<T> delegate) { protected FabricEntityTypeBuilder(Class<? extends T> entityClass, Function<? super World, ? extends T> function) {
this.delegate = delegate; this.entityClass = entityClass;
this.function = function;
} }
public static <T extends Entity> FabricEntityTypeBuilder<T> create(Class<? extends T> entityClass) { public static <T extends Entity> FabricEntityTypeBuilder<T> create(Class<? extends T> entityClass) {
return new FabricEntityTypeBuilder<>(EntityType.Builder.create(entityClass)); return new FabricEntityTypeBuilder<>(entityClass, (w) -> null);
} }
public static <T extends Entity> FabricEntityTypeBuilder<T> create(Class<? extends T> entityClass, Function<? super World, ? extends T> function) { public static <T extends Entity> FabricEntityTypeBuilder<T> create(Class<? extends T> entityClass, Function<? super World, ? extends T> function) {
return new FabricEntityTypeBuilder<>(EntityType.Builder.create(entityClass, function)); return new FabricEntityTypeBuilder<>(entityClass, function);
} }
public FabricEntityTypeBuilder<T> disableSummon() { public FabricEntityTypeBuilder<T> disableSummon() {
delegate.disableSummon(); this.summonable = false;
return this; return this;
} }
public FabricEntityTypeBuilder<T> disableSaving() { public FabricEntityTypeBuilder<T> disableSaving() {
delegate.disableSaving(); this.saveable = false;
return this; return this;
} }
@ -64,7 +68,12 @@ public class FabricEntityTypeBuilder<T extends Entity> {
} }
public EntityType<T> build(String id) { public EntityType<T> build(String id) {
EntityType type = delegate.build(id); if (this.saveable) {
// SNIP! Modded datafixers are not supported anyway.
// TODO: Flesh out once modded datafixers exist.
}
EntityType<T> type = new EntityType<>(this.entityClass, this.function, this.saveable, this.summonable, null);
if (trackingDistance != -1) { if (trackingDistance != -1) {
EntityTrackingRegistry.INSTANCE.register(type, trackingDistance, updateIntervalTicks, alwaysUpdateVelocity); EntityTrackingRegistry.INSTANCE.register(type, trackingDistance, updateIntervalTicks, alwaysUpdateVelocity);
} }