Files
armorpaint/base/sources/export_texture.ts
T
2024-09-06 21:11:46 +02:00

571 lines
19 KiB
TypeScript

///if (is_paint || is_lab)
let export_texture_gamma: f32 = 1.0 / 2.2;
function export_texture_run(path: string, bake_material: bool = false) {
///if is_paint
if (bake_material) {
export_texture_run_bake_material(path);
}
else if (context_raw.layers_export == export_mode_t.PER_UDIM_TILE) {
let udim_tiles: string[] = [];
for (let i: i32 = 0; i < project_layers.length; ++i) {
let l: slot_layer_t = project_layers[i];
if (slot_layer_get_object_mask(l) > 0) {
let name: string = project_paint_objects[slot_layer_get_object_mask(l) - 1].base.name;
if (substring(name, name.length - 5, 2) == ".1") { // tile.1001
array_push(udim_tiles, substring(name, name.length - 5, name.length));
}
}
}
if (udim_tiles.length > 0) {
for (let i: i32 = 0; i < udim_tiles.length; ++i) {
let udim_tile: string = udim_tiles[i];
export_texture_run_layers(path, project_layers, udim_tile);
}
}
else {
export_texture_run_layers(path, project_layers);
}
}
else if (context_raw.layers_export == export_mode_t.PER_OBJECT) {
let object_names: string[] = [];
for (let i: i32 = 0; i < project_layers.length; ++i) {
let l: slot_layer_t = project_layers[i];
if (slot_layer_get_object_mask(l) > 0) {
let name: string = project_paint_objects[slot_layer_get_object_mask(l) - 1].base.name;
if (array_index_of(object_names, name) == -1) {
array_push(object_names, name);
}
}
}
if (object_names.length > 0) {
for (let i: i32 = 0; i < object_names.length; ++i) {
let name: string = object_names[i];
export_texture_run_layers(path, project_layers, name);
}
}
else {
export_texture_run_layers(path, project_layers);
}
}
else { // Visible or selected
let atlas_export: bool = false;
if (project_atlas_objects != null) {
for (let i: i32 = 1; i < project_atlas_objects.length; ++i) {
if (project_atlas_objects[i - 1] != project_atlas_objects[i]) {
atlas_export = true;
break;
}
}
}
if (atlas_export) {
for (let atlas_index: i32 = 0; atlas_index < project_atlas_objects.length; ++atlas_index) {
let layers: slot_layer_t[] = [];
for (let object_index: i32 = 0; object_index < project_atlas_objects.length; ++object_index) {
if (project_atlas_objects[object_index] == atlas_index) {
for (let i: i32 = 0; i < project_layers.length; ++i) {
let l: slot_layer_t = project_layers[i];
if (slot_layer_get_object_mask(l) == 0 || // shared object
slot_layer_get_object_mask(l) - 1 == object_index) {
array_push(layers, l);
}
}
}
}
if (layers.length > 0) {
export_texture_run_layers(path, layers, project_atlas_names[atlas_index]);
}
}
}
else {
let layers: slot_layer_t[];
if (context_raw.layers_export == export_mode_t.SELECTED) {
if (slot_layer_is_group(context_raw.layer)) {
layers = slot_layer_get_children(context_raw.layer);
}
else {
layers = [context_raw.layer];
}
}
else {
layers = project_layers;
}
export_texture_run_layers(path, layers);
}
}
///end
///if is_lab
let layers: slot_layer_t[] = [brush_output_node_inst];
export_texture_run_layers(path, layers);
///end
///if arm_ios
console_info(tr("Textures exported") + " (\"Files/On My iPad/" + manifest_title + "\")");
///elseif arm_android
console_info(tr("Textures exported") + " (\"Files/Internal storage/Pictures/" + manifest_title + "\")");
///else
console_info(tr("Textures exported"));
///end
ui_files_last_path = "";
}
///if is_paint
function export_texture_run_bake_material(path: string) {
if (render_path_paint_live_layer == null) {
render_path_paint_live_layer = slot_layer_create("_live");
}
let _tool: workspace_tool_t = context_raw.tool;
context_raw.tool = workspace_tool_t.FILL;
make_material_parse_paint_material();
let _paint_object: mesh_object_t = context_raw.paint_object;
let planeo: mesh_object_t = scene_get_child(".Plane").ext;
planeo.base.visible = true;
context_raw.paint_object = planeo;
context_raw.pdirty = 1;
let _visibles: bool[] = [];
for (let i: i32 = 0; i < project_paint_objects.length; ++i) {
let p: mesh_object_t = project_paint_objects[i];
array_push(_visibles, p.base.visible);
p.base.visible = false;
}
render_path_paint_use_live_layer(true);
render_path_paint_commands_paint(false);
render_path_paint_use_live_layer(false);
context_raw.tool = _tool;
make_material_parse_paint_material();
context_raw.pdirty = 0;
planeo.base.visible = false;
context_raw.paint_object = _paint_object;
for (let i: i32 = 0; i < project_paint_objects.length; ++i) {
project_paint_objects[i].base.visible = _visibles[i];
}
let layers: slot_layer_t[] = [render_path_paint_live_layer];
export_texture_run_layers(path, layers, "", true);
}
///end
///if is_paint
function export_texture_run_layers(path: string, layers: slot_layer_t[], object_name: string = "", bake_material: bool = false) {
///end
///if is_lab
function export_texture_run_layers(path: string, layers: any[], object_name: string = "") {
///end
let texture_size_x: i32 = config_get_texture_res_x();
let texture_size_y: i32 = config_get_texture_res_y();
///if (arm_android || arm_ios)
let f: string = sys_title();
///else
let f: string = ui_files_filename;
///end
if (f == "") {
f = tr("untitled");
}
let format_type: texture_ldr_format_t = context_raw.format_type;
let bits: i32 = base_bits_handle.position == texture_bits_t.BITS8 ? 8 : 16;
let ext: string = bits == 16 ? ".exr" : format_type == texture_ldr_format_t.PNG ? ".png" : ".jpg";
if (ends_with(f, ext)) {
f = substring(f, 0, f.length - 4);
}
///if is_paint
let is_udim: bool = context_raw.layers_export == export_mode_t.PER_UDIM_TILE;
if (is_udim) {
ext = object_name + ext;
}
base_make_temp_img();
base_make_export_img();
if (base_pipe_merge == null) {
base_make_pipe();
}
if (const_data_screen_aligned_vb == null) {
const_data_create_screen_aligned_data();
}
let rt: render_target_t = map_get(render_path_render_targets, "empty_white");
let empty: image_t = rt._image;
// Append object mask name
let export_selected: bool = context_raw.layers_export == export_mode_t.SELECTED;
if (export_selected && slot_layer_get_object_mask(layers[0]) > 0) {
f += "_" + project_paint_objects[slot_layer_get_object_mask(layers[0]) - 1].base.name;
}
if (!is_udim && !export_selected && object_name != "") {
f += "_" + object_name;
}
// Clear export layer
g4_begin(base_expa);
g4_clear(color_from_floats(0.0, 0.0, 0.0, 0.0));
g4_end();
g4_begin(base_expb);
g4_clear(color_from_floats(0.5, 0.5, 1.0, 0.0));
g4_end();
g4_begin(base_expc);
g4_clear(color_from_floats(1.0, 0.0, 0.0, 0.0));
g4_end();
// Flatten layers
for (let i: i32 = 0; i < layers.length; ++i) {
let l1: slot_layer_t = layers[i];
if (!export_selected && !slot_layer_is_visible(l1)) {
continue;
}
if (!slot_layer_is_layer(l1)) {
continue;
}
if (object_name != "" && slot_layer_get_object_mask(l1) > 0) {
if (is_udim && !ends_with(project_paint_objects[slot_layer_get_object_mask(l1) - 1].base.name, object_name)) {
continue;
}
let per_object: bool = context_raw.layers_export == export_mode_t.PER_OBJECT;
if (per_object && project_paint_objects[slot_layer_get_object_mask(l1) - 1].base.name != object_name) {
continue;
}
}
let mask: image_t = empty;
let l1masks: slot_layer_t[] = slot_layer_get_masks(l1);
if (l1masks != null && !bake_material) {
if (l1masks.length > 1) {
base_make_temp_mask_img();
g2_begin(base_temp_mask_image);
g2_clear(0x00000000);
g2_end();
let l1: slot_layer_t = {
texpaint: base_temp_mask_image
};
for (let i: i32 = 0; i < l1masks.length; ++i) {
base_merge_layer(l1, l1masks[i]);
}
mask = base_temp_mask_image;
}
else {
mask = l1masks[0].texpaint;
}
}
if (l1.paint_base) {
g2_begin(base_temp_image); // Copy to temp
g2_set_pipeline(base_pipe_copy);
g2_draw_image(base_expa, 0, 0);
g2_set_pipeline(null);
g2_end();
g4_begin(base_expa);
g4_set_pipeline(base_pipe_merge);
g4_set_tex(base_tex0, l1.texpaint);
g4_set_tex(base_tex1, empty);
g4_set_tex(base_texmask, mask);
g4_set_tex(base_texa, base_temp_image);
g4_set_float(base_opac, slot_layer_get_opacity(l1));
g4_set_int(base_blending, layers.length > 1 ? l1.blending : 0);
g4_set_vertex_buffer(const_data_screen_aligned_vb);
g4_set_index_buffer(const_data_screen_aligned_ib);
g4_draw();
g4_end();
}
if (l1.paint_nor) {
g2_begin(base_temp_image);
g2_set_pipeline(base_pipe_copy);
g2_draw_image(base_expb, 0, 0);
g2_set_pipeline(null);
g2_end();
g4_begin(base_expb);
g4_set_pipeline(base_pipe_merge);
g4_set_tex(base_tex0, l1.texpaint);
g4_set_tex(base_tex1, l1.texpaint_nor);
g4_set_tex(base_texmask, mask);
g4_set_tex(base_texa, base_temp_image);
g4_set_float(base_opac, slot_layer_get_opacity(l1));
g4_set_int(base_blending, l1.paint_nor_blend ? -2 : -1);
g4_set_vertex_buffer(const_data_screen_aligned_vb);
g4_set_index_buffer(const_data_screen_aligned_ib);
g4_draw();
g4_end();
}
if (l1.paint_occ || l1.paint_rough || l1.paint_met || l1.paint_height) {
g2_begin(base_temp_image);
g2_set_pipeline(base_pipe_copy);
g2_draw_image(base_expc, 0, 0);
g2_set_pipeline(null);
g2_end();
if (l1.paint_occ && l1.paint_rough && l1.paint_met && l1.paint_height) {
base_commands_merge_pack(base_pipe_merge, base_expc, l1.texpaint, l1.texpaint_pack, slot_layer_get_opacity(l1), mask, l1.paint_height_blend ? -3 : -1);
}
else {
if (l1.paint_occ) base_commands_merge_pack(base_pipe_merge_r, base_expc, l1.texpaint, l1.texpaint_pack, slot_layer_get_opacity(l1), mask);
if (l1.paint_rough) base_commands_merge_pack(base_pipe_merge_g, base_expc, l1.texpaint, l1.texpaint_pack, slot_layer_get_opacity(l1), mask);
if (l1.paint_met) base_commands_merge_pack(base_pipe_merge_b, base_expc, l1.texpaint, l1.texpaint_pack, slot_layer_get_opacity(l1), mask);
}
}
}
///if arm_metal
// Flush command list
g2_begin(base_expa);
g2_end();
g2_begin(base_expb);
g2_end();
g2_begin(base_expc);
g2_end();
///end
///end
///if is_paint
let texpaint: image_t = base_expa;
let texpaint_nor: image_t = base_expb;
let texpaint_pack: image_t = base_expc;
///end
///if is_lab
let texpaint: image_t = brush_output_node_inst.texpaint;
let texpaint_nor: image_t = brush_output_node_inst.texpaint_nor;
let texpaint_pack: image_t = brush_output_node_inst.texpaint_pack;
///end
let pixpaint: buffer_t = null;
let pixpaint_nor: buffer_t = null;
let pixpaint_pack: buffer_t = null;
let preset: export_preset_t = box_export_preset;
let pix: buffer_t = null;
for (let i: i32 = 0; i < preset.textures.length; ++i) {
let t: export_preset_texture_t = preset.textures[i];
for (let i: i32 = 0; i < t.channels.length; ++i) {
let c: string = t.channels[i];
if ((c == "base_r" || c == "base_g" || c == "base_b" || c == "opac") && pixpaint == null) {
pixpaint = image_get_pixels(texpaint);
}
else if ((c == "nor_r" || c == "nor_g" || c == "nor_g_directx" || c == "nor_b" || c == "emis" || c == "subs") && pixpaint_nor == null) {
pixpaint_nor = image_get_pixels(texpaint_nor);
}
else if ((c == "occ" || c == "rough" || c == "metal" || c == "height" || c == "smooth") && pixpaint_pack == null) {
pixpaint_pack = image_get_pixels(texpaint_pack);
}
}
}
for (let i: i32 = 0; i < preset.textures.length; ++i) {
let t: export_preset_texture_t = preset.textures[i];
let c: string[] = t.channels;
let tex_name: string = t.name != "" ? "_" + t.name : "";
let single_channel: bool = c[0] == c[1] && c[1] == c[2] && c[3] == "1.0";
if (c[0] == "base_r" && c[1] == "base_g" && c[2] == "base_b" && c[3] == "1.0" && t.color_space == "linear") {
export_texture_write_texture(path + path_sep + f + tex_name + ext, pixpaint, 1);
}
else if (c[0] == "nor_r" && c[1] == "nor_g" && c[2] == "nor_b" && c[3] == "1.0" && t.color_space == "linear") {
export_texture_write_texture(path + path_sep + f + tex_name + ext, pixpaint_nor, 1);
}
else if (c[0] == "occ" && c[1] == "rough" && c[2] == "metal" && c[3] == "1.0" && t.color_space == "linear") {
export_texture_write_texture(path + path_sep + f + tex_name + ext, pixpaint_pack, 1);
}
else if (single_channel && c[0] == "occ" && t.color_space == "linear") {
export_texture_write_texture(path + path_sep + f + tex_name + ext, pixpaint_pack, 2, 0);
}
else if (single_channel && c[0] == "rough" && t.color_space == "linear") {
export_texture_write_texture(path + path_sep + f + tex_name + ext, pixpaint_pack, 2, 1);
}
else if (single_channel && c[0] == "metal" && t.color_space == "linear") {
export_texture_write_texture(path + path_sep + f + tex_name + ext, pixpaint_pack, 2, 2);
}
else if (single_channel && c[0] == "height" && t.color_space == "linear") {
export_texture_write_texture(path + path_sep + f + tex_name + ext, pixpaint_pack, 2, 3);
}
else if (single_channel && c[0] == "opac" && t.color_space == "linear") {
export_texture_write_texture(path + path_sep + f + tex_name + ext, pixpaint, 2, 3);
}
else {
if (pix == null) {
pix = buffer_create(texture_size_x * texture_size_y * 4 * math_floor(bits / 8));
}
for (let i: i32 = 0; i < 4; ++i) {
let c: string = t.channels[i];
if (c == "base_r") {
export_texture_copy_channel(pixpaint, 0, pix, i, t.color_space == "linear");
}
else if (c == "base_g") {
export_texture_copy_channel(pixpaint, 1, pix, i, t.color_space == "linear");
}
else if (c == "base_b") {
export_texture_copy_channel(pixpaint, 2, pix, i, t.color_space == "linear");
}
else if (c == "height") {
export_texture_copy_channel(pixpaint_pack, 3, pix, i, t.color_space == "linear");
}
else if (c == "metal") {
export_texture_copy_channel(pixpaint_pack, 2, pix, i, t.color_space == "linear");
}
else if (c == "nor_r") {
export_texture_copy_channel(pixpaint_nor, 0, pix, i, t.color_space == "linear");
}
else if (c == "nor_g") {
export_texture_copy_channel(pixpaint_nor, 1, pix, i, t.color_space == "linear");
}
else if (c == "nor_g_directx") {
export_texture_copy_channel_inv(pixpaint_nor, 1, pix, i, t.color_space == "linear");
}
else if (c == "nor_b") {
export_texture_copy_channel(pixpaint_nor, 2, pix, i, t.color_space == "linear");
}
else if (c == "occ") {
export_texture_copy_channel(pixpaint_pack, 0, pix, i, t.color_space == "linear");
}
else if (c == "opac") {
export_texture_copy_channel(pixpaint, 3, pix, i, t.color_space == "linear");
}
else if (c == "rough") {
export_texture_copy_channel(pixpaint_pack, 1, pix, i, t.color_space == "linear");
}
else if (c == "smooth") {
export_texture_copy_channel_inv(pixpaint_pack, 1, pix, i, t.color_space == "linear");
}
else if (c == "emis") {
export_texture_extract_channel(pixpaint_nor, 3, pix, i, 3, 1, t.color_space == "linear");
}
else if (c == "subs") {
export_texture_extract_channel(pixpaint_nor, 3, pix, i, 3, 2, t.color_space == "linear");
}
else if (c == "0.0") {
export_texture_set_channel(0, pix, i);
}
else if (c == "1.0") {
export_texture_set_channel(255, pix, i);
}
}
export_texture_write_texture(path + path_sep + f + tex_name + ext, pix, 3);
}
}
// Release staging memory allocated in image_get_pixels()
texpaint.pixels = null;
texpaint_nor.pixels = null;
texpaint_pack.pixels = null;
}
function export_texture_write_texture(file: string, pixels: buffer_t, type: i32 = 1, off: i32 = 0) {
let res_x: i32 = config_get_texture_res_x();
let res_y: i32 = config_get_texture_res_y();
let bits_handle: i32 = base_bits_handle.position;
let bits: i32 = bits_handle == texture_bits_t.BITS8 ? 8 : bits_handle == texture_bits_t.BITS16 ? 16 : 32;
let format: i32 = 0; // RGBA
if (type == 1) {
format = 2; // RGB1
}
if (type == 2 && off == 0) {
format = 3; // RRR1
}
if (type == 2 && off == 1) {
format = 4; // GGG1
}
if (type == 2 && off == 2) {
format = 5; // BBB1
}
if (type == 2 && off == 3) {
format = 6; // AAA1
}
if (context_raw.layers_destination == export_destination_t.PACKED) {
let image: image_t = image_from_bytes(pixels, res_x, res_y);
map_set(data_cached_images, file, image);
let ar: string[] = string_split(file, path_sep);
let name: string = ar[ar.length - 1];
let asset: asset_t = {
name: name,
file: file,
id: project_asset_id++
};
array_push(project_assets, asset);
if (project_raw.assets == null) {
project_raw.assets = [];
}
array_push(project_raw.assets, asset.file);
array_push(project_asset_names, asset.name);
map_set(project_asset_map, asset.id, image);
let assets: asset_t[] = [asset];
export_arm_pack_assets(project_raw, assets);
return;
}
if (bits == 8 && context_raw.format_type == texture_ldr_format_t.PNG) {
iron_write_png(file, pixels, res_x, res_y, format);
}
else if (bits == 8 && context_raw.format_type == texture_ldr_format_t.JPG) {
iron_write_jpg(file, pixels, res_x, res_y, format, math_floor(context_raw.format_quality));
}
else { // Exr
let b: buffer_t = parser_exr_run(res_x, res_y, pixels, bits, type, off);
iron_file_save_bytes(file, b, b.length);
}
}
function export_texture_copy_channel(from: buffer_t, from_channel: i32, to: buffer_t, to_channel: i32, linear: bool = true) {
for (let i: i32 = 0; i < math_floor((to.length) / 4); ++i) {
buffer_set_u8(to, i * 4 + to_channel, buffer_get_u8(from, i * 4 + from_channel));
}
if (!linear) {
export_texture_to_srgb(to, to_channel);
}
}
function export_texture_copy_channel_inv(from: buffer_t, from_channel: i32, to: buffer_t, to_channel: i32, linear: bool = true) {
for (let i: i32 = 0; i < math_floor((to.length) / 4); ++i) {
buffer_set_u8(to, i * 4 + to_channel, 255 - buffer_get_u8(from, i * 4 + from_channel));
}
if (!linear) {
export_texture_to_srgb(to, to_channel);
}
}
function export_texture_extract_channel(from: buffer_t, from_channel: i32, to: buffer_t, to_channel: i32, step: i32, mask: i32, linear: bool = true) {
for (let i: i32 = 0; i < math_floor((to.length) / 4); ++i) {
buffer_set_u8(to, i * 4 + to_channel, buffer_get_u8(from, i * 4 + from_channel) % step == mask ? 255 : 0);
}
if (!linear) {
export_texture_to_srgb(to, to_channel);
}
}
function export_texture_set_channel(value: i32, to: buffer_t, to_channel: i32, linear: bool = true) {
for (let i: i32 = 0; i < math_floor((to.length) / 4); ++i) {
buffer_set_u8(to, i * 4 + to_channel, value);
}
if (!linear) {
export_texture_to_srgb(to, to_channel);
}
}
function export_texture_to_srgb(to: buffer_t, to_channel: i32) {
for (let i: i32 = 0; i < math_floor((to.length) / 4); ++i) {
buffer_set_u8(to, i * 4 + to_channel, math_floor(math_pow(buffer_get_u8(to, i * 4 + to_channel) / 255, export_texture_gamma) * 255));
}
}
type export_preset_t = {
textures?: export_preset_texture_t[];
};
type export_preset_texture_t = {
name?: string;
channels?: string[];
color_space?: string;
};
///end