armory3d.armortools/armorpaint/Sources/make_discard.ts

function make_discard_color_id(vert: node_shader_t, frag: node_shader_t) {
	node_shader_add_uniform(frag, "sampler2D texpaint_colorid"); // 1x1 picker
	node_shader_add_uniform(frag, "sampler2D texcolorid", "_texcolorid"); // color map
	node_shader_write(frag, "vec3 colorid_c1 = texelFetch(texpaint_colorid, ivec2(0, 0), 0).rgb;");
	node_shader_write(frag, "vec3 colorid_c2 = textureLod(texcolorid, texCoordPick, 0).rgb;");
	///if (krom_direct3d11 || krom_direct3d12 || krom_metal)
	node_shader_write(frag, "if (any(colorid_c1 != colorid_c2)) discard;");
	///else
	node_shader_write(frag, "if (colorid_c1 != colorid_c2) discard;");
	///end
}

function make_discard_face(vert: node_shader_t, frag: node_shader_t) {
	node_shader_add_uniform(frag, "sampler2D gbuffer2");
	node_shader_add_uniform(frag, "sampler2D textrianglemap", "_textrianglemap");
	node_shader_add_uniform(frag, "vec2 textrianglemapSize", "_texpaintSize");
	node_shader_add_uniform(frag, "vec2 gbufferSize", "_gbufferSize");
	///if (krom_direct3d11 || krom_direct3d12 || krom_metal || krom_vulkan)
	node_shader_write(frag, "vec2 texCoordInp = texelFetch(gbuffer2, ivec2(inp.x * gbufferSize.x, inp.y * gbufferSize.y), 0).ba;");
	///else
	node_shader_write(frag, "vec2 texCoordInp = texelFetch(gbuffer2, ivec2(inp.x * gbufferSize.x, (1.0 - inp.y) * gbufferSize.y), 0).ba;");
	///end
	node_shader_write(frag, "vec4 face_c1 = texelFetch(textrianglemap, ivec2(texCoordInp * textrianglemapSize), 0);");
	node_shader_write(frag, "vec4 face_c2 = textureLod(textrianglemap, texCoordPick, 0);");
	///if (krom_direct3d11 || krom_direct3d12 || krom_metal)
	node_shader_write(frag, "if (any(face_c1 != face_c2)) discard;");
	///else
	node_shader_write(frag, "if (face_c1 != face_c2) discard;");
	///end
}

function make_discard_uv_island(vert: node_shader_t, frag: node_shader_t) {
	node_shader_add_uniform(frag, "sampler2D texuvislandmap", "_texuvislandmap");
	node_shader_write(frag, "if (textureLod(texuvislandmap, texCoordPick, 0).r == 0.0) discard;");
}

function make_discard_material_id(vert: node_shader_t, frag: node_shader_t) {
	frag.wvpposition = true;
	node_shader_write(frag, "vec2 picker_sample_tc = vec2(wvpposition.x / wvpposition.w, wvpposition.y / wvpposition.w) * 0.5 + 0.5;");
	///if (krom_direct3d11 || krom_direct3d12 || krom_metal || krom_vulkan)
	node_shader_write(frag, "picker_sample_tc.y = 1.0 - picker_sample_tc.y;");
	///end
	node_shader_add_uniform(frag, "sampler2D texpaint_nor_undo", "_texpaint_nor_undo");
	let matid: i32 = context_raw.materialid_picked / 255;
	node_shader_write(frag, "if (" + matid + " != textureLod(texpaint_nor_undo, picker_sample_tc, 0.0).a) discard;");
}