Spaces:
Build error
Build error
File size: 4,980 Bytes
d46f4a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
package net.minecraft.resources;
import com.mojang.datafixers.util.Either;
import com.mojang.datafixers.util.Pair;
import com.mojang.serialization.Codec;
import com.mojang.serialization.DataResult;
import com.mojang.serialization.DynamicOps;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import net.minecraft.core.Holder;
import net.minecraft.core.HolderGetter;
import net.minecraft.core.HolderOwner;
import net.minecraft.core.HolderSet;
import net.minecraft.core.Registry;
import net.minecraft.tags.TagKey;
import net.minecraft.util.ExtraCodecs;
public class HolderSetCodec<E> implements Codec<HolderSet<E>> {
private final ResourceKey<? extends Registry<E>> registryKey;
private final Codec<Holder<E>> elementCodec;
private final Codec<List<Holder<E>>> homogenousListCodec;
private final Codec<Either<TagKey<E>, List<Holder<E>>>> registryAwareCodec;
private static <E> Codec<List<Holder<E>>> homogenousList(Codec<Holder<E>> p_206668_, boolean p_206669_) {
Codec<List<Holder<E>>> codec = p_206668_.listOf().validate(ExtraCodecs.ensureHomogenous(Holder::kind));
return p_206669_ ? codec : ExtraCodecs.compactListCodec(p_206668_, codec);
}
public static <E> Codec<HolderSet<E>> create(ResourceKey<? extends Registry<E>> p_206686_, Codec<Holder<E>> p_206687_, boolean p_206688_) {
return new HolderSetCodec<>(p_206686_, p_206687_, p_206688_);
}
private HolderSetCodec(ResourceKey<? extends Registry<E>> p_206660_, Codec<Holder<E>> p_206661_, boolean p_206662_) {
this.registryKey = p_206660_;
this.elementCodec = p_206661_;
this.homogenousListCodec = homogenousList(p_206661_, p_206662_);
this.registryAwareCodec = Codec.either(TagKey.hashedCodec(p_206660_), this.homogenousListCodec);
}
@Override
public <T> DataResult<Pair<HolderSet<E>, T>> decode(DynamicOps<T> p_206696_, T p_206697_) {
if (p_206696_ instanceof RegistryOps<T> registryops) {
Optional<HolderGetter<E>> optional = registryops.getter(this.registryKey);
if (optional.isPresent()) {
HolderGetter<E> holdergetter = optional.get();
return this.registryAwareCodec
.decode(p_206696_, p_206697_)
.flatMap(
p_326147_ -> {
DataResult<HolderSet<E>> dataresult = p_326147_.getFirst()
.map(
p_326145_ -> lookupTag(holdergetter, (TagKey<E>)p_326145_),
p_326140_ -> DataResult.success(HolderSet.direct((List<? extends Holder<E>>)p_326140_))
);
return dataresult.map(p_326149_ -> Pair.of((HolderSet<E>)p_326149_, (T)p_326147_.getSecond()));
}
);
}
}
return this.decodeWithoutRegistry(p_206696_, p_206697_);
}
private static <E> DataResult<HolderSet<E>> lookupTag(HolderGetter<E> p_331398_, TagKey<E> p_328227_) {
return (DataResult)p_331398_.get(p_328227_)
.map(DataResult::success)
.orElseGet(() -> DataResult.error(() -> "Missing tag: '" + p_328227_.location() + "' in '" + p_328227_.registry().location() + "'"));
}
public <T> DataResult<T> encode(HolderSet<E> p_206674_, DynamicOps<T> p_206675_, T p_206676_) {
if (p_206675_ instanceof RegistryOps<T> registryops) {
Optional<HolderOwner<E>> optional = registryops.owner(this.registryKey);
if (optional.isPresent()) {
if (!p_206674_.canSerializeIn(optional.get())) {
return DataResult.error(() -> "HolderSet " + p_206674_ + " is not valid in current registry set");
}
return this.registryAwareCodec.encode(p_206674_.unwrap().mapRight(List::copyOf), p_206675_, p_206676_);
}
}
return this.encodeWithoutRegistry(p_206674_, p_206675_, p_206676_);
}
private <T> DataResult<Pair<HolderSet<E>, T>> decodeWithoutRegistry(DynamicOps<T> p_206671_, T p_206672_) {
return this.elementCodec.listOf().decode(p_206671_, p_206672_).flatMap(p_206666_ -> {
List<Holder.Direct<E>> list = new ArrayList<>();
for (Holder<E> holder : p_206666_.getFirst()) {
if (!(holder instanceof Holder.Direct<E> direct)) {
return DataResult.error(() -> "Can't decode element " + holder + " without registry");
}
list.add(direct);
}
return DataResult.success(new Pair<>(HolderSet.direct(list), p_206666_.getSecond()));
});
}
private <T> DataResult<T> encodeWithoutRegistry(HolderSet<E> p_206690_, DynamicOps<T> p_206691_, T p_206692_) {
return this.homogenousListCodec.encode(p_206690_.stream().toList(), p_206691_, p_206692_);
}
}
|