78 lines
3.2 KiB
Java
78 lines
3.2 KiB
Java
package net.minecraft.advancements.critereon;
|
|
|
|
import com.google.common.collect.ImmutableMap;
|
|
import com.mojang.serialization.MapCodec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder.Instance;
|
|
import java.util.Map;
|
|
import java.util.function.Predicate;
|
|
import net.minecraft.core.component.DataComponentExactPredicate;
|
|
import net.minecraft.core.component.DataComponentGetter;
|
|
import net.minecraft.core.component.predicates.DataComponentPredicate;
|
|
import net.minecraft.network.RegistryFriendlyByteBuf;
|
|
import net.minecraft.network.codec.StreamCodec;
|
|
|
|
public record DataComponentMatchers(DataComponentExactPredicate exact, Map<DataComponentPredicate.Type<?>, DataComponentPredicate> partial)
|
|
implements Predicate<DataComponentGetter> {
|
|
public static final DataComponentMatchers ANY = new DataComponentMatchers(DataComponentExactPredicate.EMPTY, Map.of());
|
|
public static final MapCodec<DataComponentMatchers> CODEC = RecordCodecBuilder.mapCodec(
|
|
p_392397_ -> p_392397_.group(
|
|
DataComponentExactPredicate.CODEC
|
|
.optionalFieldOf("components", DataComponentExactPredicate.EMPTY)
|
|
.forGetter(DataComponentMatchers::exact),
|
|
DataComponentPredicate.CODEC.optionalFieldOf("predicates", Map.of()).forGetter(DataComponentMatchers::partial)
|
|
)
|
|
.apply(p_392397_, DataComponentMatchers::new)
|
|
);
|
|
public static final StreamCodec<RegistryFriendlyByteBuf, DataComponentMatchers> STREAM_CODEC = StreamCodec.composite(
|
|
DataComponentExactPredicate.STREAM_CODEC,
|
|
DataComponentMatchers::exact,
|
|
DataComponentPredicate.STREAM_CODEC,
|
|
DataComponentMatchers::partial,
|
|
DataComponentMatchers::new
|
|
);
|
|
|
|
public boolean test(DataComponentGetter p_397267_) {
|
|
if (!this.exact.test(p_397267_)) {
|
|
return false;
|
|
} else {
|
|
for (DataComponentPredicate datacomponentpredicate : this.partial.values()) {
|
|
if (!datacomponentpredicate.matches(p_397267_)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
|
|
public boolean isEmpty() {
|
|
return this.exact.isEmpty() && this.partial.isEmpty();
|
|
}
|
|
|
|
public static class Builder {
|
|
private DataComponentExactPredicate exact = DataComponentExactPredicate.EMPTY;
|
|
private final ImmutableMap.Builder<DataComponentPredicate.Type<?>, DataComponentPredicate> partial = ImmutableMap.builder();
|
|
|
|
private Builder() {
|
|
}
|
|
|
|
public static DataComponentMatchers.Builder components() {
|
|
return new DataComponentMatchers.Builder();
|
|
}
|
|
|
|
public <T extends DataComponentPredicate> DataComponentMatchers.Builder partial(DataComponentPredicate.Type<T> p_391465_, T p_394464_) {
|
|
this.partial.put(p_391465_, p_394464_);
|
|
return this;
|
|
}
|
|
|
|
public DataComponentMatchers.Builder exact(DataComponentExactPredicate p_394802_) {
|
|
this.exact = p_394802_;
|
|
return this;
|
|
}
|
|
|
|
public DataComponentMatchers build() {
|
|
return new DataComponentMatchers(this.exact, this.partial.buildOrThrow());
|
|
}
|
|
}
|
|
} |