luboslenco
432 lines
18 KiB
TypeScript
432 lines
18 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 l of project_layers) {
|
|
if (slot_layer_get_object_mask(l) > 0) {
|
|
let name: string = project_paint_objects[slot_layer_get_object_mask(l) - 1].base.name;
|
|
if (name.substr(name.length - 5, 2) == ".1") { // tile.1001
|
|
udim_tiles.push(name.substr(name.length - 5));
|
|
}
|
|
}
|
|
}
|
|
if (udim_tiles.length > 0) {
|
|
for (let udim_tile of udim_tiles) 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 l of project_layers) {
|
|
if (slot_layer_get_object_mask(l) > 0) {
|
|
let name: string = project_paint_objects[slot_layer_get_object_mask(l) - 1].base.name;
|
|
if (object_names.indexOf(name) == -1) {
|
|
object_names.push(name);
|
|
}
|
|
}
|
|
}
|
|
if (object_names.length > 0) {
|
|
for (let name of object_names) 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: SlotLayerRaw[] = [];
|
|
for (let object_index: i32 = 0; object_index < project_atlas_objects.length; ++object_index) {
|
|
if (project_atlas_objects[object_index] == atlas_index) {
|
|
for (let l of project_layers) {
|
|
if (slot_layer_get_object_mask(l) == 0 /* shared object */ || slot_layer_get_object_mask(l) - 1 == object_index) layers.push(l);
|
|
}
|
|
}
|
|
}
|
|
if (layers.length > 0) {
|
|
export_texture_run_layers(path, layers, project_atlas_names[atlas_index]);
|
|
}
|
|
}
|
|
}
|
|
else export_texture_run_layers(path, context_raw.layers_export == export_mode_t.SELECTED ? (slot_layer_is_group(context_raw.layer) ? slot_layer_get_children(context_raw.layer) : [context_raw.layer]) : project_layers);
|
|
}
|
|
///end
|
|
|
|
///if is_lab
|
|
export_texture_run_layers(path, [brush_output_node_inst]);
|
|
///end
|
|
|
|
///if krom_ios
|
|
console_info(tr("Textures exported") + " ('Files/On My iPad/" + manifest_title + "')");
|
|
///elseif krom_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;
|
|
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;
|
|
|
|
export_texture_run_layers(path, [render_path_paint_live_layer], "", true);
|
|
}
|
|
///end
|
|
|
|
///if is_paint
|
|
function export_texture_run_layers(path: string, layers: SlotLayerRaw[], 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 (krom_android || krom_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 (f.endsWith(ext)) f = f.substr(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 empty: image_t = render_path_render_targets.get("empty_white")._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 l1 of layers) {
|
|
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 && !project_paint_objects[slot_layer_get_object_mask(l1) - 1].base.name.endsWith(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: SlotLayerRaw[] = 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: any = { 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 krom_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: ArrayBuffer = null;
|
|
let pixpaint_nor: ArrayBuffer = null;
|
|
let pixpaint_pack: ArrayBuffer = null;
|
|
let preset: export_preset_t = box_export_preset;
|
|
let pix: ArrayBuffer = null;
|
|
|
|
for (let t of preset.textures) {
|
|
for (let c of t.channels) {
|
|
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 t of preset.textures) {
|
|
let c: string[] = t.channels;
|
|
let tex_name = 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 = new ArrayBuffer(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(new DataView(pixpaint), 0, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "base_g") export_texture_copy_channel(new DataView(pixpaint), 1, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "base_b") export_texture_copy_channel(new DataView(pixpaint), 2, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "height") export_texture_copy_channel(new DataView(pixpaint_pack), 3, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "metal") export_texture_copy_channel(new DataView(pixpaint_pack), 2, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "nor_r") export_texture_copy_channel(new DataView(pixpaint_nor), 0, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "nor_g") export_texture_copy_channel(new DataView(pixpaint_nor), 1, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "nor_g_directx") export_texture_copy_channel_inv(new DataView(pixpaint_nor), 1, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "nor_b") export_texture_copy_channel(new DataView(pixpaint_nor), 2, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "occ") export_texture_copy_channel(new DataView(pixpaint_pack), 0, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "opac") export_texture_copy_channel(new DataView(pixpaint), 3, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "rough") export_texture_copy_channel(new DataView(pixpaint_pack), 1, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "smooth") export_texture_copy_channel_inv(new DataView(pixpaint_pack), 1, new DataView(pix), i, t.color_space == "linear");
|
|
else if (c == "emis") export_texture_extract_channel(new DataView(pixpaint_nor), 3, new DataView(pix), i, 3, 1, t.color_space == "linear");
|
|
else if (c == "subs") export_texture_extract_channel(new DataView(pixpaint_nor), 3, new DataView(pix), i, 3, 2, t.color_space == "linear");
|
|
else if (c == "0.0") export_texture_set_channel(0, new DataView(pix), i);
|
|
else if (c == "1.0") export_texture_set_channel(255, new DataView(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: ArrayBuffer, 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);
|
|
data_cached_images.set(file, image);
|
|
let ar: string[] = file.split(path_sep);
|
|
let name: string = ar[ar.length - 1];
|
|
let asset: asset_t = {name: name, file: file, id: project_asset_id++};
|
|
project_assets.push(asset);
|
|
if (project_raw.assets == null) project_raw.assets = [];
|
|
project_raw.assets.push(asset.file);
|
|
project_asset_names.push(asset.name);
|
|
project_asset_map.set(asset.id, image);
|
|
export_arm_pack_assets(project_raw, [asset]);
|
|
return;
|
|
}
|
|
|
|
if (bits == 8 && context_raw.format_type == texture_ldr_format_t.PNG) {
|
|
krom_write_png(file, pixels, res_x, res_y, format);
|
|
}
|
|
else if (bits == 8 && context_raw.format_type == texture_ldr_format_t.JPG) {
|
|
krom_write_jpg(file, pixels, res_x, res_y, format, math_floor(context_raw.format_quality));
|
|
}
|
|
else { // Exr
|
|
let b: ArrayBuffer = parser_exr_run(res_x, res_y, pixels, bits, type, off);
|
|
krom_file_save_bytes(file, b, b.byteLength);
|
|
}
|
|
}
|
|
|
|
function export_texture_copy_channel(from: DataView, from_channel: i32, to: DataView, to_channel: i32, linear: bool = true) {
|
|
for (let i: i32 = 0; i < math_floor(to.byteLength / 4); ++i) {
|
|
to.setUint8(i * 4 + to_channel, from.getUint8(i * 4 + from_channel));
|
|
}
|
|
if (!linear) export_texture_to_srgb(to, to_channel);
|
|
}
|
|
|
|
function export_texture_copy_channel_inv(from: DataView, from_channel: i32, to: DataView, to_channel: i32, linear: bool = true) {
|
|
for (let i: i32 = 0; i < math_floor(to.byteLength / 4); ++i) {
|
|
to.setUint8(i * 4 + to_channel, 255 - from.getUint8(i * 4 + from_channel));
|
|
}
|
|
if (!linear) export_texture_to_srgb(to, to_channel);
|
|
}
|
|
|
|
function export_texture_extract_channel(from: DataView, from_channel: i32, to: DataView, to_channel: i32, step: i32, mask: i32, linear: bool = true) {
|
|
for (let i: i32 = 0; i < math_floor(to.byteLength / 4); ++i) {
|
|
to.setUint8(i * 4 + to_channel, from.getUint8(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: DataView, to_channel: i32, linear: bool = true) {
|
|
for (let i: i32 = 0; i < math_floor(to.byteLength / 4); ++i) {
|
|
to.setUint8(i * 4 + to_channel, value);
|
|
}
|
|
if (!linear) export_texture_to_srgb(to, to_channel);
|
|
}
|
|
|
|
function export_texture_to_srgb(to: DataView, to_channel: i32) {
|
|
for (let i: i32 = 0; i < math_floor(to.byteLength / 4); ++i) {
|
|
to.setUint8(i * 4 + to_channel, math_floor(math_pow(to.getUint8(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
|