Override submodel AO setting by parent

Match vanilla behavior

Author: Su5eD

fabric-renderer-api-v1/src/client/java/net/fabricmc/fabric/api/renderer/v1/render/RenderContext.java

@@ -19,6 +19,7 @@
 import java.util.function.Consumer;
 import java.util.function.Supplier;
 
+import net.fabricmc.fabric.api.util.TriState;
 import net.minecraft.client.renderer.RenderType;
 import net.neoforged.neoforge.client.model.data.ModelData;
 import org.jetbrains.annotations.Nullable;
@@ -191,4 +192,10 @@ default RenderType getRenderType() {
 	default void pushModelData(ModelData modelData) {}
 
 	default void popModelData() {}
+
+	default TriState usesAmbientOcclusion() {
+		return TriState.DEFAULT;
+	}
+
+	default void setUsesAmbientOcclusion(TriState state) {}
 }

fabric-renderer-api-v1/src/client/java/net/fabricmc/fabric/impl/renderer/VanillaModelEncoder.java

@@ -45,7 +45,8 @@ public class VanillaModelEncoder {
 
 	public static void emitBlockQuads(BakedModel model, @Nullable BlockState state, Supplier<RandomSource> randomSupplier, RenderContext context) {
 		QuadEmitter emitter = context.getEmitter();
-		final RenderMaterial defaultMaterial = model.useAmbientOcclusion() ? STANDARD_MATERIAL : NO_AO_MATERIAL;
+		final TriState useAO = context.usesAmbientOcclusion();
+		final RenderMaterial defaultMaterial = useAO == TriState.TRUE || useAO == TriState.DEFAULT && model.useAmbientOcclusion() ? STANDARD_MATERIAL : NO_AO_MATERIAL;
 
 		for (int i = 0; i <= ModelHelper.NULL_FACE_ID; i++) {
 			final Direction cullFace = ModelHelper.faceFromIndex(i);

fabric-renderer-api-v1/src/client/java/net/fabricmc/fabric/mixin/renderer/client/MultipartBakedModelMixin.java

@@ -18,6 +18,7 @@
 
 import net.fabricmc.fabric.api.renderer.v1.model.FabricBakedModel;
 import net.fabricmc.fabric.api.renderer.v1.render.RenderContext;
+import net.fabricmc.fabric.api.util.TriState;
 import net.minecraft.client.renderer.RenderType;
 import net.minecraft.client.resources.model.BakedModel;
 import net.minecraft.client.resources.model.MultiPartBakedModel;
@@ -106,9 +107,12 @@ public void emitBlockQuads(BlockAndTintGetter blockView, BlockState state, Block
 				if (renderType != null && !model.getRenderTypes(state, random, context.getModelData()).contains(renderType))
 					continue;
 
+				net.neoforged.neoforge.common.util.TriState usesAO = ((MultiPartBakedModel) (Object) this).useAmbientOcclusion(state, context.getModelData(), renderType);
 				context.pushModelData(data);
+				context.setUsesAmbientOcclusion(usesAO == net.neoforged.neoforge.common.util.TriState.FALSE ? TriState.FALSE : TriState.TRUE);
 				model.emitBlockQuads(blockView, state, pos, subModelRandomSupplier, context);
 				context.popModelData();
+				context.setUsesAmbientOcclusion(TriState.DEFAULT);
 			}
 		}
 	}

fabric-renderer-indigo/src/client/java/net/fabricmc/fabric/impl/client/indigo/renderer/render/AbstractBlockRenderContext.java

@@ -66,6 +66,7 @@ public void emitDirectly() {
 	private final BlockPos.MutableBlockPos lightPos = new BlockPos.MutableBlockPos();
 
 	private final Stack<ModelData> modelDataStack = new Stack<>();
+	private TriState useAO = TriState.DEFAULT;
 
 	protected AbstractBlockRenderContext() {
 		aoCalc = createAoCalc(blockInfo);
@@ -118,7 +119,17 @@ public void pushModelData(ModelData modelData) {
 	public void popModelData() {
 		this.modelDataStack.pop();
 	}
-	
+
+	@Override
+	public TriState usesAmbientOcclusion() {
+		return this.useAO;
+	}
+
+	@Override
+	public void setUsesAmbientOcclusion(TriState state) {
+		this.useAO = state;
+	}
+
 	private void renderQuad(MutableQuadViewImpl quad) {
 		if (!transform(quad)) {
 			return;