Files
armorpaint/armorlab/sources/render_path_paint.ts
T
2024-09-08 21:53:39 +02:00

311 lines
10 KiB
TypeScript

let render_path_paint_live_layer_drawn: i32 = 0; ////
function render_path_paint_init() {
{
let t: render_target_t = render_target_create();
t.name = "texpaint_blend0";
t.width = config_get_texture_res_x();
t.height = config_get_texture_res_y();
t.format = "R8";
render_path_create_render_target(t);
}
{
let t: render_target_t = render_target_create();
t.name = "texpaint_blend1";
t.width = config_get_texture_res_x();
t.height = config_get_texture_res_y();
t.format = "R8";
render_path_create_render_target(t);
}
{
let t: render_target_t = render_target_create();
t.name = "texpaint_picker";
t.width = 1;
t.height = 1;
t.format = "RGBA32";
render_path_create_render_target(t);
}
{
let t: render_target_t = render_target_create();
t.name = "texpaint_nor_picker";
t.width = 1;
t.height = 1;
t.format = "RGBA32";
render_path_create_render_target(t);
}
{
let t: render_target_t = render_target_create();
t.name = "texpaint_pack_picker";
t.width = 1;
t.height = 1;
t.format = "RGBA32";
render_path_create_render_target(t);
}
{
let t: render_target_t = render_target_create();
t.name = "texpaint_uv_picker";
t.width = 1;
t.height = 1;
t.format = "RGBA32";
render_path_create_render_target(t);
}
render_path_load_shader("shader_datas/copy_mrt3_pass/copy_mrt3_pass");
}
function render_path_paint_commands_paint(dilation: bool = true) {
let tid: string = "";
if (context_raw.pdirty > 0) {
if (context_raw.tool == workspace_tool_t.PICKER) {
///if arm_metal
// render_path_set_target("texpaint_picker");
// render_path_clear_target(0xff000000);
// render_path_set_target("texpaint_nor_picker");
// render_path_clear_target(0xff000000);
// render_path_set_target("texpaint_pack_picker");
// render_path_clear_target(0xff000000);
render_path_set_target("texpaint_picker", ["texpaint_nor_picker", "texpaint_pack_picker", "texpaint_uv_picker"]);
///else
render_path_set_target("texpaint_picker", ["texpaint_nor_picker", "texpaint_pack_picker", "texpaint_uv_picker"]);
// render_path_clear_target(0xff000000);
///end
render_path_bind_target("gbuffer2", "gbuffer2");
// tid = context_raw.layer.id;
render_path_bind_target("texpaint" + tid, "texpaint");
render_path_bind_target("texpaint_nor" + tid, "texpaint_nor");
render_path_bind_target("texpaint_pack" + tid, "texpaint_pack");
render_path_draw_meshes("paint");
ui_header_handle.redraws = 2;
let texpaint_picker: image_t = map_get(render_path_render_targets, "texpaint_picker")._image;
let texpaint_nor_picker: image_t = map_get(render_path_render_targets, "texpaint_nor_picker")._image;
let texpaint_pack_picker: image_t = map_get(render_path_render_targets, "texpaint_pack_picker")._image;
let texpaint_uv_picker: image_t = map_get(render_path_render_targets, "texpaint_uv_picker")._image;
let a: buffer_t = image_get_pixels(texpaint_picker);
let b: buffer_t = image_get_pixels(texpaint_nor_picker);
let c: buffer_t = image_get_pixels(texpaint_pack_picker);
let d: buffer_t = image_get_pixels(texpaint_uv_picker);
if (context_raw.color_picker_callback != null) {
context_raw.color_picker_callback(context_raw.picked_color);
}
// Picked surface values
// ///if (arm_metal || arm_vulkan)
// context_raw.pickedColor.base.Rb = a.get(2);
// context_raw.pickedColor.base.Gb = a.get(1);
// context_raw.pickedColor.base.Bb = a.get(0);
// context_raw.pickedColor.normal.Rb = b.get(2);
// context_raw.pickedColor.normal.Gb = b.get(1);
// context_raw.pickedColor.normal.Bb = b.get(0);
// context_raw.pickedColor.occlusion = c.get(2) / 255;
// context_raw.pickedColor.roughness = c.get(1) / 255;
// context_raw.pickedColor.metallic = c.get(0) / 255;
// context_raw.pickedColor.height = c.get(3) / 255;
// context_raw.pickedColor.opacity = a.get(3) / 255;
// context_raw.uvxPicked = d.get(2) / 255;
// context_raw.uvyPicked = d.get(1) / 255;
// ///else
// context_raw.pickedColor.base.Rb = a.get(0);
// context_raw.pickedColor.base.Gb = a.get(1);
// context_raw.pickedColor.base.Bb = a.get(2);
// context_raw.pickedColor.normal.Rb = b.get(0);
// context_raw.pickedColor.normal.Gb = b.get(1);
// context_raw.pickedColor.normal.Bb = b.get(2);
// context_raw.pickedColor.occlusion = c.get(0) / 255;
// context_raw.pickedColor.roughness = c.get(1) / 255;
// context_raw.pickedColor.metallic = c.get(2) / 255;
// context_raw.pickedColor.height = c.get(3) / 255;
// context_raw.pickedColor.opacity = a.get(3) / 255;
// context_raw.uvxPicked = d.get(0) / 255;
// context_raw.uvyPicked = d.get(1) / 255;
// ///end
}
else {
let texpaint: string = "texpaint_node_target";
render_path_set_target("texpaint_blend1");
render_path_bind_target("texpaint_blend0", "tex");
render_path_draw_shader("shader_datas/copy_pass/copyR8_pass");
render_path_set_target(texpaint, ["texpaint_nor" + tid, "texpaint_pack" + tid, "texpaint_blend0"]);
render_path_bind_target("_main", "gbufferD");
render_path_bind_target("texpaint_blend1", "paintmask");
// Read texcoords from gbuffer
let read_tc: bool = context_raw.tool == workspace_tool_t.CLONE ||
context_raw.tool == workspace_tool_t.BLUR ||
context_raw.tool == workspace_tool_t.SMUDGE;
if (read_tc) {
render_path_bind_target("gbuffer2", "gbuffer2");
}
render_path_draw_meshes("paint");
}
}
}
function render_path_paint_commands_cursor() {
let tool: workspace_tool_t = context_raw.tool;
if (tool != workspace_tool_t.ERASER &&
tool != workspace_tool_t.CLONE &&
tool != workspace_tool_t.BLUR &&
tool != workspace_tool_t.SMUDGE) {
return;
}
let nodes: ui_nodes_t = ui_nodes_get_nodes();
let canvas: ui_node_canvas_t = ui_nodes_get_canvas(true);
let inpaint: bool = nodes.nodes_selected_id.length > 0 && ui_get_node(canvas.nodes, nodes.nodes_selected_id[0]).type == "InpaintNode";
if (!base_ui_enabled || base_is_dragging || !inpaint) {
return;
}
let mx: f32 = context_raw.paint_vec.x;
let my: f32 = 1.0 - context_raw.paint_vec.y;
if (context_raw.brush_locked) {
mx = (context_raw.lock_started_x - app_x()) / app_w();
my = 1.0 - (context_raw.lock_started_y - app_y()) / app_h();
}
let radius: f32 = context_raw.brush_radius;
render_path_paint_draw_cursor(mx, my, radius / 3.4);
}
function render_path_paint_draw_cursor(mx: f32, my: f32, radius: f32, tint_r: f32 = 1.0, tint_g: f32 = 1.0, tint_b: f32 = 1.0) {
let plane: mesh_object_t = scene_get_child(".Plane").ext;
let geom: mesh_data_t = plane.data;
if (base_pipe_cursor == null) {
base_make_cursor_pipe();
}
render_path_set_target("");
g4_set_pipeline(base_pipe_cursor);
let img: image_t = resource_get("cursor.k");
g4_set_tex(base_cursor_tex, img);
let gbuffer0: image_t = map_get(render_path_render_targets, "gbuffer0")._image;
g4_set_tex_depth(base_cursor_gbufferd, gbuffer0);
g4_set_float2(base_cursor_mouse, mx, my);
g4_set_float2(base_cursor_tex_step, 1 / gbuffer0.width, 1 / gbuffer0.height);
g4_set_float(base_cursor_radius, radius);
let right: vec4_t = vec4_norm(camera_object_right_world(scene_camera));
g4_set_float3(base_cursor_camera_right, right.x, right.y, right.z);
g4_set_float3(base_cursor_tint, tint_r, tint_g, tint_b);
g4_set_mat(base_cursor_vp, scene_camera.vp);
let help_mat: mat4_t = mat4_identity();
help_mat = mat4_get_inv(scene_camera.vp);
g4_set_mat(base_cursor_inv_vp, help_mat);
///if (arm_metal || arm_vulkan)
let vs: vertex_structure_t[] = [{name: "tex", data: "short2norm"}];
g4_set_vertex_buffer(mesh_data_get(geom, vs));
///else
g4_set_vertex_buffer(geom._vertexBuffer);
///end
g4_set_index_buffer(geom._indexBuffers[0]);
g4_draw();
g4_disable_scissor();
render_path_end();
}
function render_path_paint_paint_enabled(): bool {
return !context_raw.foreground_event;
}
function render_path_paint_begin() {
if (!render_path_paint_paint_enabled()) {
return;
}
}
function render_path_paint_end() {
render_path_paint_commands_cursor();
context_raw.ddirty--;
context_raw.rdirty--;
if (!render_path_paint_paint_enabled()) {
return;
}
context_raw.pdirty--;
}
function render_path_paint_draw() {
if (!render_path_paint_paint_enabled()) {
return;
}
render_path_paint_commands_paint();
if (context_raw.brush_blend_dirty) {
context_raw.brush_blend_dirty = false;
///if arm_metal
render_path_set_target("texpaint_blend0");
render_path_clear_target(0x00000000);
render_path_set_target("texpaint_blend1");
render_path_clear_target(0x00000000);
///else
render_path_set_target("texpaint_blend0", ["texpaint_blend1"]);
render_path_clear_target(0x00000000);
///end
}
}
function render_path_paint_bind_layers() {
let image: image_t = null;
let nodes: ui_nodes_t = ui_nodes_get_nodes();
let canvas: ui_node_canvas_t = ui_nodes_get_canvas(true);
if (nodes.nodes_selected_id.length > 0) {
let node: ui_node_t = ui_get_node(canvas.nodes, nodes.nodes_selected_id[0]);
let brush_node: logic_node_t = parser_logic_get_logic_node(node);
if (brush_node != null) {
image = logic_node_get_cached_image(brush_node);
}
}
if (image != null) {
if (map_get(render_path_render_targets, "texpaint_node") == null) {
let t: render_target_t = render_target_create();
t.name = "texpaint_node";
t.width = config_get_texture_res_x();
t.height = config_get_texture_res_y();
t.format = "RGBA32";
map_set(render_path_render_targets, t.name, t);
}
if (map_get(render_path_render_targets, "texpaint_node_target") == null) {
let t: render_target_t = render_target_create();
t.name = "texpaint_node_target";
t.width = config_get_texture_res_x();
t.height = config_get_texture_res_y();
t.format = "RGBA32";
map_set(render_path_render_targets, t.name, t);
}
map_get(render_path_render_targets, "texpaint_node")._image = image;
render_path_bind_target("texpaint_node", "texpaint");
render_path_bind_target("texpaint_nor_empty", "texpaint_nor");
render_path_bind_target("texpaint_pack_empty", "texpaint_pack");
let nodes: ui_nodes_t = ui_nodes_get_nodes();
let canvas: ui_node_canvas_t = ui_nodes_get_canvas(true);
let node: ui_node_t = ui_get_node(canvas.nodes, nodes.nodes_selected_id[0]);
let inpaint: bool = node.type == "InpaintNode";
if (inpaint) {
map_get(render_path_render_targets, "texpaint_node_target")._image = inpaint_node_get_target();
}
}
else {
render_path_bind_target("texpaint", "texpaint");
render_path_bind_target("texpaint_nor", "texpaint_nor");
render_path_bind_target("texpaint_pack", "texpaint_pack");
}
}
function render_path_paint_unbind_layers() {
}