46 lines
1.9 KiB
Java
46 lines
1.9 KiB
Java
package net.minecraft.util.datafix.fixes;
|
|
|
|
import java.util.function.Function;
|
|
import com.mojang.datafixers.types.DynamicOps;
|
|
import java.util.Optional;
|
|
import com.mojang.datafixers.Dynamic;
|
|
import com.mojang.datafixers.util.Pair;
|
|
import com.mojang.datafixers.types.Type;
|
|
import java.util.Objects;
|
|
import com.mojang.datafixers.DSL;
|
|
import com.mojang.datafixers.TypeRewriteRule;
|
|
import net.minecraft.world.scores.criteria.ObjectiveCriteria;
|
|
import com.mojang.datafixers.schemas.Schema;
|
|
import com.mojang.datafixers.DataFix;
|
|
|
|
public class ObjectiveRenderTypeFix extends DataFix {
|
|
public ObjectiveRenderTypeFix(final Schema schema, final boolean boolean2) {
|
|
super(schema, boolean2);
|
|
}
|
|
|
|
private static ObjectiveCriteria.RenderType getRenderType(final String string) {
|
|
return string.equals("health") ? ObjectiveCriteria.RenderType.HEARTS : ObjectiveCriteria.RenderType.INTEGER;
|
|
}
|
|
|
|
protected TypeRewriteRule makeRule() {
|
|
final Type<Pair<String, Dynamic<?>>> type2 = (Type<Pair<String, Dynamic<?>>>)DSL.named(References.OBJECTIVE.typeName(), DSL.remainderType());
|
|
if (!Objects.equals(type2, this.getInputSchema().getType(References.OBJECTIVE))) {
|
|
throw new IllegalStateException("Objective type is not what was expected.");
|
|
}
|
|
final Optional<String> optional2;
|
|
String string3;
|
|
ObjectiveCriteria.RenderType a4;
|
|
return this.fixTypeEverywhere("ObjectiveRenderTypeFix", (Type)type2, dynamicOps -> pair -> pair.mapSecond(dynamic -> {
|
|
optional2 = (Optional<String>)dynamic.get("RenderType").asString();
|
|
if (!optional2.isPresent()) {
|
|
string3 = dynamic.get("CriteriaName").asString("");
|
|
a4 = getRenderType(string3);
|
|
return dynamic.set("RenderType", dynamic.createString(a4.getId()));
|
|
}
|
|
else {
|
|
return dynamic;
|
|
}
|
|
}));
|
|
}
|
|
}
|