77 lines
2.2 KiB
Java
77 lines
2.2 KiB
Java
package net.minecraft.core;
|
|
|
|
import com.mojang.serialization.Lifecycle;
|
|
import java.util.Optional;
|
|
import net.minecraft.core.Holder.Reference;
|
|
import net.minecraft.resources.ResourceKey;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.util.RandomSource;
|
|
import org.jetbrains.annotations.NotNull;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
public class DefaultedMappedRegistry<T> extends MappedRegistry<T> implements DefaultedRegistry<T> {
|
|
private final ResourceLocation defaultKey;
|
|
private Reference<T> defaultValue;
|
|
|
|
public DefaultedMappedRegistry(String defaultKey, ResourceKey<? extends Registry<T>> key, Lifecycle registryLifecycle, boolean hasIntrusiveHolders) {
|
|
super(key, registryLifecycle, hasIntrusiveHolders);
|
|
this.defaultKey = ResourceLocation.parse(defaultKey);
|
|
}
|
|
|
|
@Override
|
|
public Reference<T> register(ResourceKey<T> key, T value, RegistrationInfo registrationInfo) {
|
|
Reference<T> reference = super.register(key, value, registrationInfo);
|
|
if (this.defaultKey.equals(key.location())) {
|
|
this.defaultValue = reference;
|
|
}
|
|
|
|
return reference;
|
|
}
|
|
|
|
@Override
|
|
public int getId(@Nullable T value) {
|
|
int i = super.getId(value);
|
|
return i == -1 ? super.getId(this.defaultValue.value()) : i;
|
|
}
|
|
|
|
@NotNull
|
|
@Override
|
|
public ResourceLocation getKey(T value) {
|
|
ResourceLocation resourceLocation = super.getKey(value);
|
|
return resourceLocation == null ? this.defaultKey : resourceLocation;
|
|
}
|
|
|
|
@NotNull
|
|
@Override
|
|
public T getValue(@Nullable ResourceLocation key) {
|
|
T object = super.getValue(key);
|
|
return object == null ? this.defaultValue.value() : object;
|
|
}
|
|
|
|
@Override
|
|
public Optional<T> getOptional(@Nullable ResourceLocation name) {
|
|
return Optional.ofNullable(super.getValue(name));
|
|
}
|
|
|
|
@Override
|
|
public Optional<Reference<T>> getAny() {
|
|
return Optional.ofNullable(this.defaultValue);
|
|
}
|
|
|
|
@NotNull
|
|
@Override
|
|
public T byId(int id) {
|
|
T object = super.byId(id);
|
|
return object == null ? this.defaultValue.value() : object;
|
|
}
|
|
|
|
@Override
|
|
public Optional<Reference<T>> getRandom(RandomSource random) {
|
|
return super.getRandom(random).or(() -> Optional.of(this.defaultValue));
|
|
}
|
|
|
|
@Override
|
|
public ResourceLocation getDefaultKey() {
|
|
return this.defaultKey;
|
|
}
|
|
}
|