Bake up axis

Sources/arm/node/MakeBake.hx

@@ -68,10 +68,16 @@ class MakeBake {
 		else if (UITrait.inst.bakeType == 3) { // Normal (World)
 			frag.n = true;
 			frag.write('fragColor[0] = vec4(n * vec3(0.5, 0.5, 0.5) + vec3(0.5, 0.5, 0.5), 1.0);');
+			if (UITrait.inst.bakeUpAxis == 1) { // Y up
+				frag.write('fragColor[0].rgb = vec3(fragColor[0].r, fragColor[0].b, 1.0 - fragColor[0].g);');
+			}
 		}
 		else if (UITrait.inst.bakeType == 4) { // Position
 			frag.wposition = true;
 			frag.write('fragColor[0] = vec4(wposition * vec3(0.5, 0.5, 0.5) + vec3(0.5, 0.5, 0.5), 1.0);');
+			if (UITrait.inst.bakeUpAxis == 1) { // Y up
+				frag.write('fragColor[0].rgb = vec3(fragColor[0].r, fragColor[0].b, 1.0 - fragColor[0].g);');
+			}
 		}
 		else if (UITrait.inst.bakeType == 5) { // TexCoord
 			frag.write('fragColor[0] = vec4(texCoord.xy, 0.0, 1.0);');

Sources/arm/render/RenderPathRaytrace.hx

@@ -191,6 +191,7 @@ class RenderPathRaytrace {
 			f32[2] = UITrait.inst.bakeAoRadius;
 			f32[3] = UITrait.inst.bakeAoOffset;
 			f32[4] = Scene.active.world.probe.raw.strength;
+			f32[5] = UITrait.inst.bakeUpAxis;
 
 			var baketex2 = path.renderTargets.get("baketex2").image;
 			Krom.raytraceDispatchRays(baketex2.renderTarget_, f32.buffer);

Sources/arm/ui/UITrait.hx

@@ -992,7 +992,7 @@ class UITrait {
 					bakes.push("BentNormal (DXR)");
 					#end
 					bakeType = ui.combo(bakeHandle, bakes, "Bake");
-					if (bakeType == 3 || bakeType == 4 || bakeType == 8) {
+					if (bakeType == 3 || bakeType == 4 || bakeType == 9) {
 						var bakeUpAxisHandle = Id.handle({position: bakeUpAxis});
 						bakeUpAxis = ui.combo(bakeUpAxisHandle, ["Z", "Y"], "Up Axis");
 					}
@@ -1031,8 +1031,6 @@ class UITrait {
 					}
 					if (ui.changed) {
 						MaterialParser.parsePaintMaterial();
-						// Context.pdirty = 4;
-						// pushUndo
 					}
 				}
 				else {