Spaces:
Build error
Build error
File size: 5,830 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 107 108 109 110 111 112 113 114 115 116 117 118 |
package net.minecraft.data.loot;
import com.google.common.collect.Multimap;
import com.google.common.collect.Sets;
import com.mojang.logging.LogUtils;
import com.mojang.serialization.Lifecycle;
import it.unimi.dsi.fastutil.objects.Object2ObjectOpenHashMap;
import java.nio.file.Path;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Map.Entry;
import java.util.concurrent.CompletableFuture;
import java.util.function.Function;
import net.minecraft.Util;
import net.minecraft.core.Holder;
import net.minecraft.core.HolderGetter;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.MappedRegistry;
import net.minecraft.core.RegistrationInfo;
import net.minecraft.core.RegistryAccess;
import net.minecraft.core.WritableRegistry;
import net.minecraft.core.registries.Registries;
import net.minecraft.data.CachedOutput;
import net.minecraft.data.DataProvider;
import net.minecraft.data.PackOutput;
import net.minecraft.resources.ResourceKey;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.util.ProblemReporter;
import net.minecraft.util.context.ContextKeySet;
import net.minecraft.world.RandomSequence;
import net.minecraft.world.level.levelgen.RandomSupport;
import net.minecraft.world.level.storage.loot.LootTable;
import net.minecraft.world.level.storage.loot.ValidationContext;
import net.minecraft.world.level.storage.loot.parameters.LootContextParamSets;
import org.slf4j.Logger;
public class LootTableProvider implements DataProvider {
private static final Logger LOGGER = LogUtils.getLogger();
private final PackOutput.PathProvider pathProvider;
private final Set<ResourceKey<LootTable>> requiredTables;
private final List<LootTableProvider.SubProviderEntry> subProviders;
private final CompletableFuture<HolderLookup.Provider> registries;
public LootTableProvider(
PackOutput p_254123_,
Set<ResourceKey<LootTable>> p_254481_,
List<LootTableProvider.SubProviderEntry> p_253798_,
CompletableFuture<HolderLookup.Provider> p_330862_
) {
this.pathProvider = p_254123_.createRegistryElementsPathProvider(Registries.LOOT_TABLE);
this.subProviders = p_253798_;
this.requiredTables = p_254481_;
this.registries = p_330862_;
}
@Override
public CompletableFuture<?> run(CachedOutput p_254060_) {
return this.registries.thenCompose(p_325860_ -> this.run(p_254060_, p_325860_));
}
private CompletableFuture<?> run(CachedOutput p_327970_, HolderLookup.Provider p_331092_) {
WritableRegistry<LootTable> writableregistry = new MappedRegistry<>(Registries.LOOT_TABLE, Lifecycle.experimental());
Map<RandomSupport.Seed128bit, ResourceLocation> map = new Object2ObjectOpenHashMap<>();
this.subProviders.forEach(p_341016_ -> p_341016_.provider().apply(p_331092_).generate((p_358218_, p_358219_) -> {
ResourceLocation resourcelocation = sequenceIdForLootTable(p_358218_);
ResourceLocation resourcelocation1 = map.put(RandomSequence.seedForKey(resourcelocation), resourcelocation);
if (resourcelocation1 != null) {
Util.logAndPauseIfInIde("Loot table random sequence seed collision on " + resourcelocation1 + " and " + p_358218_.location());
}
p_358219_.setRandomSequence(resourcelocation);
LootTable loottable = p_358219_.setParamSet(p_341016_.paramSet).build();
writableregistry.register(p_358218_, loottable, RegistrationInfo.BUILT_IN);
}));
writableregistry.freeze();
ProblemReporter.Collector problemreporter$collector = new ProblemReporter.Collector();
HolderGetter.Provider holdergetter$provider = new RegistryAccess.ImmutableRegistryAccess(List.of(writableregistry)).freeze();
ValidationContext validationcontext = new ValidationContext(problemreporter$collector, LootContextParamSets.ALL_PARAMS, holdergetter$provider);
for (ResourceKey<LootTable> resourcekey : Sets.difference(this.requiredTables, writableregistry.registryKeySet())) {
problemreporter$collector.report("Missing built-in table: " + resourcekey.location());
}
writableregistry.listElements()
.forEach(
p_358221_ -> p_358221_.value()
.validate(
validationcontext.setContextKeySet(p_358221_.value().getParamSet())
.enterElement("{" + p_358221_.key().location() + "}", p_358221_.key())
)
);
Multimap<String, String> multimap = problemreporter$collector.get();
if (!multimap.isEmpty()) {
multimap.forEach((p_124446_, p_124447_) -> LOGGER.warn("Found validation problem in {}: {}", p_124446_, p_124447_));
throw new IllegalStateException("Failed to validate loot tables, see logs");
} else {
return CompletableFuture.allOf(writableregistry.entrySet().stream().map(p_325852_ -> {
ResourceKey<LootTable> resourcekey1 = p_325852_.getKey();
LootTable loottable = p_325852_.getValue();
Path path = this.pathProvider.json(resourcekey1.location());
return DataProvider.saveStable(p_327970_, p_331092_, LootTable.DIRECT_CODEC, loottable, path);
}).toArray(CompletableFuture[]::new));
}
}
private static ResourceLocation sequenceIdForLootTable(ResourceKey<LootTable> p_331928_) {
return p_331928_.location();
}
@Override
public final String getName() {
return "Loot Tables";
}
public static record SubProviderEntry(Function<HolderLookup.Provider, LootTableSubProvider> provider, ContextKeySet paramSet) {
}
} |