armorpaint/base/Sources/uniforms_ext.ts

let uniforms_ext_vec: vec4_t = vec4_create();
let uniforms_ext_ortho_p: mat4_t = mat4_ortho(-0.5, 0.5, -0.5, 0.5, -0.5, 0.5);

function uniforms_ext_init() {
	uniforms_i32_links = uniforms_ext_i32_link;
	uniforms_f32_links = uniforms_ext_f32_link;
	uniforms_vec2_links = uniforms_ext_vec2_link;
	uniforms_vec3_links = uniforms_ext_vec3_link;
	uniforms_vec4_links = uniforms_ext_vec4_link;
	uniforms_mat4_links = uniforms_ext_mat4_link;
	uniforms_tex_links = uniforms_ext_tex_link;
}

function uniforms_ext_i32_link(object: object_t, mat: material_data_t, link: string): Null<i32> {
	if (link == "_bloomCurrentMip") return RenderPathBase.bloom_current_mip;
	return null;
}

function uniforms_ext_f32_link(object: object_t, mat: material_data_t, link: string): Null<f32> {
	switch (link) {
		case "_brushRadius": {
			///if (is_paint || is_sculpt)
			let decal: bool = context_raw.tool == workspace_tool_t.DECAL || context_raw.tool == workspace_tool_t.TEXT;
			let decal_mask: bool = decal && operator_shortcut(config_keymap.decal_mask + "+" + config_keymap.action_paint, shortcut_type_t.DOWN);
			let brush_decal_mask_radius: f32 = context_raw.brush_decal_mask_radius;
			if (config_raw.brush_3d) {
				brush_decal_mask_radius *= context_raw.paint2d ? 0.55 * ui_view2d_pan_scale : 2.0;
			}
			let radius: f32 = decal_mask ? brush_decal_mask_radius : context_raw.brush_radius;
			let val: f32 = (radius * context_raw.brush_nodes_radius) / 15.0;
			if (config_raw.pressure_radius && pen_down()) {
				val *= pen_pressure * config_raw.pressure_sensitivity;
			}
			let scale2d: f32 = (900 / base_h()) * config_raw.window_scale;

			if (config_raw.brush_3d && !decal) {
				val *= context_raw.paint2d ? 0.55 * scale2d * ui_view2d_pan_scale : 2;
			}
			else {
				val *= scale2d; // Projection ratio
			}
			///end

			///if is_lab
			let radius: f32 = context_raw.brush_radius;
			let val: f32 = radius / 15.0;
			if (config_raw.pressure_radius && pen_down()) {
				val *= pen_pressure * config_raw.pressure_sensitivity;
			}
			val *= 2;
			///end

			return val;
		}
		case "_vignetteStrength": {
			return config_raw.rp_vignette;
		}
		case "_grainStrength": {
			return config_raw.rp_grain;
		}
		case "_coneOffset": {
			return context_raw.vxao_offset;
		}
		case "_coneAperture": {
			return context_raw.vxao_aperture;
		}
		case "_bloomSampleScale": {
			return RenderPathBase.bloom_sample_scale;
		}

		///if (is_paint || is_sculpt)
		case "_brushScaleX": {
			return 1 / context_raw.brush_scale_x;
		}
		case "_brushOpacity": {
			let val: f32 = context_raw.brush_opacity * context_raw.brush_nodes_opacity;
			if (config_raw.pressure_opacity && pen_down()) {
				val *= pen_pressure * config_raw.pressure_sensitivity;
			}
			return val;
		}
		case "_brushHardness": {
			let decal_mask: bool = operator_shortcut(config_keymap.decal_mask + "+" + config_keymap.action_paint, shortcut_type_t.DOWN);
			if (context_raw.tool != workspace_tool_t.BRUSH && context_raw.tool != workspace_tool_t.ERASER && context_raw.tool != workspace_tool_t.CLONE && !decal_mask) return 1.0;
			let val: f32 = context_raw.brush_hardness * context_raw.brush_nodes_hardness;
			if (config_raw.pressure_hardness && pen_down()) {
				val *= pen_pressure * config_raw.pressure_sensitivity;
			}
			if (config_raw.brush_3d) {
				if (context_raw.paint2d) {
					val *= 1.0 / ui_view2d_pan_scale;
				}
				else {
					val *= val;
				}
			}
			return val;
		}
		case "_brushScale": {
			let fill: bool = context_raw.layer.fill_layer != null;
			let val: f32 = (fill ? context_raw.layer.scale : context_raw.brush_scale) * context_raw.brush_nodes_scale;
			return val;
		}
		case "_objectId": {
			return project_paint_objects.indexOf(object.ext);
		}
		///if is_paint
		case "_dilateRadius": {
			return util_uv_dilatemap != null ? config_raw.dilate_radius : 0.0;
		}
		///end
		case "_decalLayerDim": {
			return mat4_get_scale(context_raw.layer.decal_mat).z * 0.5;
		}
		case "_pickerOpacity": {
			return context_raw.picked_color.opacity;
		}
		case "_pickerOcclusion": {
			return context_raw.picked_color.occlusion;
		}
		case "_pickerRoughness": {
			return context_raw.picked_color.roughness;
		}
		case "_pickerMetallic": {
			return context_raw.picked_color.metallic;
		}
		case "_pickerHeight": {
			return context_raw.picked_color.height;
		}
		///end
	}
	if (ParserMaterial.script_links != null) {
		for (let key of ParserMaterial.script_links.keys()) {
			let asciprt_links: any = ParserMaterial.script_links;
			let script: string = asciprt_links[key];
			let result: f32 = 0.0;
			if (script != "") {
				try {
					result = eval(script);
				}
				catch(e: any) {
					console_log(e);
				}
			}
			return result;
		}
	}
	return null;
}

function uniforms_ext_vec2_link(object: object_t, mat: material_data_t, link: string): vec4_t {
	switch (link) {
		case "_gbufferSize": {
			vec4_set(uniforms_ext_vec, 0, 0, 0);
			let gbuffer2: render_target_t = render_path_render_targets.get("gbuffer2");
			vec4_set(uniforms_ext_vec, gbuffer2._image.width, gbuffer2._image.height, 0);
			return uniforms_ext_vec;
		}
		case "_cloneDelta": {
			vec4_set(uniforms_ext_vec, context_raw.clone_delta_x, context_raw.clone_delta_y, 0);
			return uniforms_ext_vec;
		}
		case "_texpaintSize": {
			vec4_set(uniforms_ext_vec, config_get_texture_res_x(), config_get_texture_res_y(), 0);
			return uniforms_ext_vec;
		}
		///if (is_paint || is_sculpt)
		case "_brushAngle": {
			let brush_angle: f32 = context_raw.brush_angle + context_raw.brush_nodes_angle;
			let angle: f32 = context_raw.layer.fill_layer != null ? context_raw.layer.angle : brush_angle;
			angle *= (math_pi() / 180);
			if (config_raw.pressure_angle && pen_down()) {
				angle *= pen_pressure * config_raw.pressure_sensitivity;
			}
			vec4_set(uniforms_ext_vec, math_cos(-angle), math_sin(-angle), 0);
			return uniforms_ext_vec;
		}
		///end
	}
	return null;
}

function uniforms_ext_vec3_link(object: object_t, mat: material_data_t, link: string): vec4_t {
	let v: vec4_t = null;
	switch (link) {
		///if (is_paint || is_sculpt)
		case "_brushDirection": {
			v = _uniforms_vec;
			// Discard first paint for directional brush
			let allow_paint: bool = context_raw.prev_paint_vec_x != context_raw.last_paint_vec_x &&
									context_raw.prev_paint_vec_y != context_raw.last_paint_vec_y &&
									context_raw.prev_paint_vec_x > 0 &&
									context_raw.prev_paint_vec_y > 0;
			let x: f32 = context_raw.paint_vec.x;
			let y: f32 = context_raw.paint_vec.y;
			let lastx: f32 = context_raw.prev_paint_vec_x;
			let lasty: f32 = context_raw.prev_paint_vec_y;
			if (context_raw.paint2d) {
				x = vec2d(x);
				lastx = vec2d(lastx);
			}
			let angle: f32 = math_atan2(-y + lasty, x - lastx) - math_pi() / 2;
			vec4_set(v, math_cos(angle), math_sin(angle), allow_paint ? 1 : 0);
			context_raw.prev_paint_vec_x = context_raw.last_paint_vec_x;
			context_raw.prev_paint_vec_y = context_raw.last_paint_vec_y;
			return v;
		}
		case "_decalLayerLoc": {
			v = _uniforms_vec;
			vec4_set(v, context_raw.layer.decal_mat.m[12], context_raw.layer.decal_mat.m[13], context_raw.layer.decal_mat.m[14]);
			return v;
		}
		case "_decalLayerNor": {
			v = _uniforms_vec;
			vec4_normalize(vec4_set(v, context_raw.layer.decal_mat.m[8], context_raw.layer.decal_mat.m[9], context_raw.layer.decal_mat.m[10]));
			return v;
		}
		case "_pickerBase": {
			v = _uniforms_vec;
			vec4_set(v,
				color_get_rb(context_raw.picked_color.base) / 255,
				color_get_gb(context_raw.picked_color.base) / 255,
				color_get_bb(context_raw.picked_color.base) / 255
			);
			return v;
		}
		case "_pickerNormal": {
			v = _uniforms_vec;
			vec4_set(v,
				color_get_rb(context_raw.picked_color.normal) / 255,
				color_get_gb(context_raw.picked_color.normal) / 255,
				color_get_bb(context_raw.picked_color.normal) / 255
			);
			return v;
		}
		///if arm_physics
		case "_particleHit": {
			v = _uniforms_vec;
			vec4_set(v, context_raw.particle_hit_x, context_raw.particle_hit_y, context_raw.particle_hit_z);
			return v;
		}
		case "_particleHitLast": {
			v = _uniforms_vec;
			vec4_set(v, context_raw.last_particle_hit_x, context_raw.last_particle_hit_y, context_raw.last_particle_hit_z);
			return v;
		}
		///end
		///end
	}

	return v;
}

///if (is_paint || is_sculpt)
function vec2d(x: f32) {
	// Transform from 3d viewport coord to 2d view coord
	context_raw.paint2d_view = false;
	let res: f32 = (x * base_w() - base_w()) / ui_view2d_ww;
	context_raw.paint2d_view = true;
	return res;
}
///end

function uniforms_ext_vec4_link(object: object_t, mat: material_data_t, link: string): vec4_t {
	switch (link) {
		case "_inputBrush": {
			let down: bool = mouse_down() || pen_down();
			vec4_set(uniforms_ext_vec, context_raw.paint_vec.x, context_raw.paint_vec.y, down ? 1.0 : 0.0, 0.0);

			///if (is_paint || is_sculpt)
			if (context_raw.paint2d) {
				uniforms_ext_vec.x = vec2d(uniforms_ext_vec.x);
			}
			///end

			return uniforms_ext_vec;
		}
		case "_inputBrushLast": {
			let down: bool = mouse_down() || pen_down();
			vec4_set(uniforms_ext_vec, context_raw.last_paint_vec_x, context_raw.last_paint_vec_y, down ? 1.0 : 0.0, 0.0);

			///if (is_paint || is_sculpt)
			if (context_raw.paint2d) {
				uniforms_ext_vec.x = vec2d(uniforms_ext_vec.x);
			}
			///end

			return uniforms_ext_vec;
		}
		case "_envmapData": {
			vec4_set(uniforms_ext_vec, context_raw.envmap_angle, math_sin(-context_raw.envmap_angle), math_cos(-context_raw.envmap_angle), scene_world.strength);
			return uniforms_ext_vec;
		}
		case "_envmapDataWorld": {
			vec4_set(uniforms_ext_vec, context_raw.envmap_angle, math_sin(-context_raw.envmap_angle), math_cos(-context_raw.envmap_angle), context_raw.show_envmap ? scene_world.strength : 1.0);
			return uniforms_ext_vec;
		}
		///if (is_paint || is_sculpt)
		case "_stencilTransform": {
			vec4_set(uniforms_ext_vec, context_raw.brush_stencil_x, context_raw.brush_stencil_y, context_raw.brush_stencil_scale, context_raw.brush_stencil_angle);
			if (context_raw.paint2d) uniforms_ext_vec.x = vec2d(uniforms_ext_vec.x);
			return uniforms_ext_vec;
		}
		case "_decalMask": {
			let decal_mask: bool = operator_shortcut(config_keymap.decal_mask + "+" + config_keymap.action_paint, shortcut_type_t.DOWN);
			let val: f32 = (context_raw.brush_radius * context_raw.brush_nodes_radius) / 15.0;
			let scale2d: f32 = (900 / base_h()) * config_raw.window_scale;
			val *= scale2d; // Projection ratio
			vec4_set(uniforms_ext_vec, context_raw.decal_x, context_raw.decal_y, decal_mask ? 1 : 0, val);
			if (context_raw.paint2d) uniforms_ext_vec.x = vec2d(uniforms_ext_vec.x);
			return uniforms_ext_vec;
		}
		///end
	}
	return null;
}

function uniforms_ext_mat4_link(object: object_t, mat: material_data_t, link: string): mat4_t {
	switch (link) {
		///if (is_paint || is_sculpt)
		case "_decalLayerMatrix": { // Decal layer
			let m: mat4_t = _uniforms_mat;
			mat4_set_from(m, context_raw.layer.decal_mat);
			mat4_get_inv(m, m);
			mat4_mult_mat(m, uniforms_ext_ortho_p);
			return m;
		}
		///end
	}
	return null;
}

function uniforms_ext_tex_link(object: object_t, mat: material_data_t, link: string): image_t {
	switch (link) {
		case "_texpaint_undo": {
			///if (is_paint || is_sculpt)
			let i: i32 = history_undo_i - 1 < 0 ? config_raw.undo_steps - 1 : history_undo_i - 1;
			return render_path_render_targets.get("texpaint_undo" + i)._image;
			///end

			///if is_lab
			return null;
			///end
		}
		case "_texpaint_nor_undo": {
			///if (is_paint || is_sculpt)
			let i: i32 = history_undo_i - 1 < 0 ? config_raw.undo_steps - 1 : history_undo_i - 1;
			return render_path_render_targets.get("texpaint_nor_undo" + i)._image;
			///end

			///if is_lab
			return null;
			///end
		}
		case "_texpaint_pack_undo": {
			///if (is_paint || is_sculpt)
			let i: i32 = history_undo_i - 1 < 0 ? config_raw.undo_steps - 1 : history_undo_i - 1;
			return render_path_render_targets.get("texpaint_pack_undo" + i)._image;
			///end

			///if is_lab
			return null;
			///end
		}

		case "_ltcMat": {
			if (const_data_ltc_mat_tex == null) const_data_init_ltc();
			return const_data_ltc_mat_tex;
		}
		case "_ltcMag": {
			if (const_data_ltc_mag_tex == null) const_data_init_ltc();
			return const_data_ltc_mag_tex;
		}

		///if (is_paint || is_sculpt)
		case "_texcolorid": {
			if (project_assets.length == 0) return render_path_render_targets.get("empty_white")._image;
			else return project_get_image(project_assets[context_raw.colorid_handle.position]);
		}
		case "_textexttool": { // Opacity map for text
			return context_raw.text_tool_image;
		}
		case "_texbrushmask": {
			return context_raw.brush_mask_image;
		}
		case "_texbrushstencil": {
			return context_raw.brush_stencil_image;
		}
		case "_texparticle": {
			return render_path_render_targets.get("texparticle")._image;
		}
		///end

		///if is_paint
		case "_texuvmap": {
			if (!util_uv_uvmap_cached) {
				let _init = () => {
					util_uv_cache_uv_map();
				}
				app_notify_on_init(_init);
			}
			return util_uv_uvmap;
		}
		case "_textrianglemap": {
			if (!util_uv_trianglemap_cached) {
				let _init = () => {
					util_uv_cache_triangle_map();
				}
				app_notify_on_init(_init);
			}
			return util_uv_trianglemap;
		}
		case "_texuvislandmap": {
			let _init = () => {
				util_uv_cache_uv_island_map();
			}
			app_notify_on_init(_init);
			return util_uv_uvislandmap_cached ? util_uv_uvislandmap :render_path_render_targets.get("empty_black")._image;
		}
		case "_texdilatemap": {
			return util_uv_dilatemap;
		}
		///end
	}

	if (link.startsWith("_texpaint_pack_vert")) {
		let tid: string = link.substr(link.length - 1);
		return render_path_render_targets.get("texpaint_pack" + tid)._image;
	}

	if (link.startsWith("_texpaint_vert")) {
		///if (is_paint || is_sculpt)
		let tid: i32 = Number(link.substr(link.length - 1));
		return tid < project_layers.length ? project_layers[tid].texpaint : null;
		///end

		///if is_lab
		return BrushOutputNode.inst.texpaint;
		///end
	}
	if (link.startsWith("_texpaint_nor")) {
		///if is_paint
		let tid: i32 = Number(link.substr(link.length - 1));
		return tid < project_layers.length ? project_layers[tid].texpaint_nor : null;
		///end

		///if is_lab
		return BrushOutputNode.inst.texpaint_nor;
		///end
	}
	if (link.startsWith("_texpaint_pack")) {
		///if is_paint
		let tid: i32 = Number(link.substr(link.length - 1));
		return tid < project_layers.length ? project_layers[tid].texpaint_pack : null;
		///end

		///if is_lab
		return BrushOutputNode.inst.texpaint_pack;
		///end
	}
	if (link.startsWith("_texpaint")) {
		///if (is_paint || is_sculpt)
		let tid: i32 = Number(link.substr(link.length - 1));
		return tid < project_layers.length ? project_layers[tid].texpaint : null;
		///end

		///if is_lab
		return BrushOutputNode.inst.texpaint;
		///end
	}

	///if (is_paint || is_sculpt)
	if (link.startsWith("_texblur_")) {
		let id: string = link.substr(9);
		return context_raw.node_previews != null ? context_raw.node_previews.get(id) :render_path_render_targets.get("empty_black")._image;
	}
	if (link.startsWith("_texwarp_")) {
		let id: string = link.substr(9);
		return context_raw.node_previews != null ? context_raw.node_previews.get(id) :render_path_render_targets.get("empty_black")._image;
	}
	if (link.startsWith("_texbake_")) {
		let id: string = link.substr(9);
		return context_raw.node_previews != null ? context_raw.node_previews.get(id) :render_path_render_targets.get("empty_black")._image;
	}
	///end

	return null;
}