64 lines
2.1 KiB
Java
64 lines
2.1 KiB
Java
package net.minecraft.advancements.critereon;
|
|
|
|
import com.mojang.serialization.Codec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.Optional;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.HolderSet;
|
|
import net.minecraft.core.RegistryCodecs;
|
|
import net.minecraft.core.registries.Registries;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.world.level.material.Fluid;
|
|
import net.minecraft.world.level.material.FluidState;
|
|
|
|
public record FluidPredicate(Optional<HolderSet<Fluid>> fluids, Optional<StatePropertiesPredicate> properties) {
|
|
public static final Codec<FluidPredicate> CODEC = RecordCodecBuilder.create(
|
|
instance -> instance.group(
|
|
RegistryCodecs.homogeneousList(Registries.FLUID).optionalFieldOf("fluids").forGetter(FluidPredicate::fluids),
|
|
StatePropertiesPredicate.CODEC.optionalFieldOf("state").forGetter(FluidPredicate::properties)
|
|
)
|
|
.apply(instance, FluidPredicate::new)
|
|
);
|
|
|
|
public boolean matches(ServerLevel level, BlockPos pos) {
|
|
if (!level.isLoaded(pos)) {
|
|
return false;
|
|
} else {
|
|
FluidState fluidState = level.getFluidState(pos);
|
|
return this.fluids.isPresent() && !fluidState.is((HolderSet<Fluid>)this.fluids.get())
|
|
? false
|
|
: !this.properties.isPresent() || ((StatePropertiesPredicate)this.properties.get()).matches(fluidState);
|
|
}
|
|
}
|
|
|
|
public static class Builder {
|
|
private Optional<HolderSet<Fluid>> fluids = Optional.empty();
|
|
private Optional<StatePropertiesPredicate> properties = Optional.empty();
|
|
|
|
private Builder() {
|
|
}
|
|
|
|
public static FluidPredicate.Builder fluid() {
|
|
return new FluidPredicate.Builder();
|
|
}
|
|
|
|
public FluidPredicate.Builder of(Fluid fluid) {
|
|
this.fluids = Optional.of(HolderSet.direct(fluid.builtInRegistryHolder()));
|
|
return this;
|
|
}
|
|
|
|
public FluidPredicate.Builder of(HolderSet<Fluid> fluids) {
|
|
this.fluids = Optional.of(fluids);
|
|
return this;
|
|
}
|
|
|
|
public FluidPredicate.Builder setProperties(StatePropertiesPredicate properties) {
|
|
this.properties = Optional.of(properties);
|
|
return this;
|
|
}
|
|
|
|
public FluidPredicate build() {
|
|
return new FluidPredicate(this.fluids, this.properties);
|
|
}
|
|
}
|
|
}
|