armory3d.armortools/paint/sources/box_export.ts

let box_export_htab: ui_handle_t        = ui_handle_create();
let box_export_files: string[]          = null;
let box_export_mesh_handle: ui_handle_t = ui_handle_create();
let box_export_hpreset: ui_handle_t     = ui_handle_create();
let box_export_preset: export_preset_t  = null;
let box_export_channels: string[]       = [
    "base_r", "base_g", "base_b", "height", "metal", "nor_r", "nor_g", "nor_g_directx", "nor_b", "occ", "opac", "rough", "smooth", "emis", "subs", "0.0", "1.0"
];
let box_export_color_spaces: string[] = [ "linear", "srgb" ];
let _box_export_bake_material: bool;
let _box_export_t: export_preset_texture_t;
let _box_export_apply_displacement: bool;
let _box_export_merge_vertices: bool;

function box_export_show_textures() {
	ui_box_show_custom(function() {
		if (box_export_files == null) {
			box_export_fetch_presets();
			box_export_hpreset.i = array_index_of(box_export_files, "generic");
		}
		if (box_export_preset == null) {
			box_export_parse_preset();
			box_export_hpreset.children = null;
		}

		box_export_tab_export_textures(tr("Export Textures"));
		box_export_tab_presets();

		box_export_tab_atlases();
		/// if (arm_android || arm_ios)
		box_export_tab_export_mesh(box_export_htab);
		/// end
	}, 600, 400, null, true, tr("Export"));
}

function box_export_show_bake_material() {
	ui_box_show_custom(function() {
		if (box_export_files == null) {
			box_export_fetch_presets();
			box_export_hpreset.i = array_index_of(box_export_files, "generic");
		}
		if (box_export_preset == null) {
			box_export_parse_preset();
			box_export_hpreset.children = null;
		}

		box_export_tab_export_textures(tr("Bake to Textures"), true);
		box_export_tab_presets();
	}, 600, 400, null, true, tr("Export"));
}

function box_export_tab_export_textures(title: string, bake_material: bool = false) {
	let tab_vertical: bool = config_raw.touch_ui;
	if (ui_tab(box_export_htab, title, tab_vertical)) {

		ui_row2();

		/// if (arm_android || arm_ios)
		let base_res_combo: string[] = [ "128", "256", "512", "1024", "2048", "4096" ];
		/// else
		let base_res_combo: string[] = [ "128", "256", "512", "1024", "2048", "4096", "8192", "16384" ];
		/// end

		ui_combo(base_res_handle, base_res_combo, tr("Resolution"), true);
		if (base_res_handle.changed) {
			layers_on_resized();
		}

		/// if (arm_android || arm_ios)
		let base_bits_combo: string[] = [ "8bit" ];
		/// else
		let base_bits_combo: string[] = [ "8bit", "16bit", "32bit" ];
		/// end

		ui_combo(base_bits_handle, base_bits_combo, tr("Color"), true);
		if (base_bits_handle.changed) {
			sys_notify_on_next_frame(layers_set_bits);
		}

		ui_row2();
		if (base_bits_handle.i == texture_bits_t.BITS8) {
			let h: ui_handle_t = ui_handle(__ID__);
			if (h.init) {
				h.i = context_raw.format_type;
			}
			let format_combo: string[] = [ "png", "jpg" ];
			context_raw.format_type    = ui_combo(h, format_combo, tr("Format"), true);
		}
		else {
			let h: ui_handle_t = ui_handle(__ID__);
			if (h.init) {
				h.i = context_raw.format_type;
			}
			let format_combo: string[] = [ "exr" ];
			context_raw.format_type    = ui_combo(h, format_combo, tr("Format"), true);
		}

		ui.enabled = context_raw.format_type == texture_ldr_format_t.JPG && base_bits_handle.i == texture_bits_t.BITS8;

		let h_quality: ui_handle_t = ui_handle(__ID__);
		if (h_quality.init) {
			h_quality.f = context_raw.format_quality;
		}
		context_raw.format_quality = ui_slider(h_quality, tr("Quality"), 0.0, 100.0, true, 1);

		ui.enabled = true;

		ui_row2();
		ui.enabled                            = !bake_material;
		let layers_export_handle: ui_handle_t = ui_handle(__ID__);
		layers_export_handle.i                = context_raw.layers_export;
		let layers_export_combo: string[]     = [ tr("Visible"), tr("Selected"), tr("Per Object"), tr("Per Udim Tile") ];
		context_raw.layers_export             = ui_combo(layers_export_handle, layers_export_combo, tr("Layers"), true);
		ui.enabled                            = true;

		ui_combo(box_export_hpreset, box_export_files, tr("Preset"), true);
		if (box_export_hpreset.changed) {
			box_export_preset = null;
		}

		let layers_destination_handle: ui_handle_t = ui_handle(__ID__);
		layers_destination_handle.i                = context_raw.layers_destination;

		let layers_destination_combo: string[] = [ tr("Disk"), tr("Pack into Project") ];
		context_raw.layers_destination         = ui_combo(layers_destination_handle, layers_destination_combo, tr("Destination"), true);

		ui_end_element();

		ui_row2();
		if (ui_icon_button(tr("Cancel"), icon_t.CLOSE)) {
			ui_box_hide();
		}
		if (ui_icon_button(tr("Export"), icon_t.CHECK)) {
			ui_box_hide();
			if (context_raw.layers_destination == export_destination_t.PACK_INTO_PROJECT) {
				_box_export_bake_material       = bake_material;
				context_raw.texture_export_path = "/";
				sys_notify_on_next_frame(function() {
					export_texture_run(context_raw.texture_export_path, _box_export_bake_material);
				});
			}
			else {
				let filters: string = base_bits_handle.i != texture_bits_t.BITS8 ? "exr" : context_raw.format_type == texture_ldr_format_t.PNG ? "png" : "jpg";
				_box_export_bake_material = bake_material;
				ui_files_show(filters, true, false, function(path: string) {
					context_raw.texture_export_path = path;
					/// if (arm_android || arm_ios)
					console_toast(tr("Exporting textures"));
					/// end
					sys_notify_on_next_frame(function() {
						export_texture_run(context_raw.texture_export_path, _box_export_bake_material);
					});
				});
			}
		}
		if (ui.is_hovered) {
			let key: string = map_get(config_keymap, "file_export_textures");
			let tip: string = tr("Export texture files") + " (" + key + ")";
			ui_tooltip(tip);
		}
	}
}

function box_export_tab_presets() {
	let tab_vertical: bool = config_raw.touch_ui;
	if (ui_tab(box_export_htab, tr("Presets"), tab_vertical)) {

		let row: f32[] = [ 3 / 5, 1 / 5, 1 / 5 ];
		ui_row(row);

		ui_combo(box_export_hpreset, box_export_files, tr("Preset"));
		if (box_export_hpreset.changed) {
			box_export_preset = null;
		}

		if (ui_icon_button(tr("New"), icon_t.PLUS)) {
			ui_box_show_custom(function() {
				let tab_vertical: bool = config_raw.touch_ui;
				if (ui_tab(ui_handle(__ID__), tr("New Preset"), tab_vertical)) {
					ui_row2();
					let h_preset: ui_handle_t = ui_handle(__ID__);
					if (h_preset.init) {
						h_preset.text = "new_preset";
					}
					let preset_name: string = ui_text_input(h_preset, tr("Name"));
					if (ui_icon_button(tr("OK"), icon_t.CHECK) || ui.is_return_down) {
						box_export_new_preset(preset_name);
						box_export_fetch_presets();
						box_export_preset    = null;
						box_export_hpreset.i = array_index_of(box_export_files, preset_name);
						ui_box_hide();
						box_export_htab.i = 1; // Presets
						box_export_show_textures();
					}
				}
			});
		}

		if (ui_icon_button(tr("Import"), icon_t.IMPORT)) {
			ui_files_show("json", false, false, function(path: string) {
				path = to_lower_case(path);
				if (ends_with(path, ".json")) {
					let filename: string = substring(path, string_last_index_of(path, path_sep) + 1, path.length);
					let dst_path: string = path_data() + path_sep + "export_presets" + path_sep + filename;
					file_copy(path, dst_path); // Copy to presets folder
					box_export_fetch_presets();
					box_export_preset    = null;
					box_export_hpreset.i = array_index_of(box_export_files, substring(filename, 0, filename.length - 5)); // Strip .json
					console_info(tr("Preset imported:") + " " + filename);
				}
				else {
					console_error(strings_unknown_asset_format());
				}
			});
		}

		if (box_export_preset == null) {
			box_export_parse_preset();
			box_export_hpreset.children = null;
		}

		// Texture list
		ui_separator(10, false);
		ui_row6();
		ui_text(tr("Texture"));
		ui_text(tr("R"));
		ui_text(tr("G"));
		ui_text(tr("B"));
		ui_text(tr("A"));
		ui_text(tr("Color Space"));
		ui.changed = false;
		for (let i: i32 = 0; i < box_export_preset.textures.length; ++i) {
			let t: export_preset_texture_t = box_export_preset.textures[i];
			ui_row6();
			let htex: ui_handle_t = ui_nest(box_export_hpreset, i);
			htex.text             = t.name;
			t.name                = ui_text_input(htex);

			if (ui.is_hovered && ui.input_released_r) {
				_box_export_t = t;
				ui_menu_draw(function() {
					if (ui_menu_button(tr("Delete"), "", icon_t.DELETE)) {
						array_remove(box_export_preset.textures, _box_export_t);
						box_export_save_preset();
					}
				});
			}

			let hr: ui_handle_t = ui_nest(htex, 0);
			hr.i                = array_index_of(box_export_channels, t.channels[0]);
			let hg: ui_handle_t = ui_nest(htex, 1);
			hg.i                = array_index_of(box_export_channels, t.channels[1]);
			let hb: ui_handle_t = ui_nest(htex, 2);
			hb.i                = array_index_of(box_export_channels, t.channels[2]);
			let ha: ui_handle_t = ui_nest(htex, 3);
			ha.i                = array_index_of(box_export_channels, t.channels[3]);

			ui_combo(hr, box_export_channels, tr("R"));
			if (hr.changed) {
				t.channels[0] = box_export_channels[hr.i];
			}
			ui_combo(hg, box_export_channels, tr("G"));
			if (hg.changed) {
				t.channels[1] = box_export_channels[hg.i];
			}
			ui_combo(hb, box_export_channels, tr("B"));
			if (hb.changed) {
				t.channels[2] = box_export_channels[hb.i];
			}
			ui_combo(ha, box_export_channels, tr("A"));
			if (ha.changed) {
				t.channels[3] = box_export_channels[ha.i];
			}

			let hspace: ui_handle_t = ui_nest(htex, 4);
			hspace.i                = array_index_of(box_export_color_spaces, t.color_space);
			ui_combo(hspace, box_export_color_spaces, tr("Color Space"));
			if (hspace.changed) {
				t.color_space = box_export_color_spaces[hspace.i];
			}
		}

		if (ui.changed) {
			box_export_save_preset();
		}

		row = [ 1 / 6 ];
		ui_row(row);
		if (ui_icon_button(tr("Add"), icon_t.PLUS)) {
			let tex: export_preset_texture_t = {name : "base", channels : [ "base_r", "base_g", "base_b", "1.0" ], color_space : "linear"};
			array_push(box_export_preset.textures, tex);
			box_export_hpreset.children = null;
			box_export_save_preset();
		}
	}
}

function box_export_tab_atlases() {
	let tab_vertical: bool = config_raw.touch_ui;
	if (ui_tab(box_export_htab, tr("Atlases"), tab_vertical)) {
		if (project_atlas_objects == null || project_atlas_objects.length != project_paint_objects.length) {
			project_atlas_objects = [];
			project_atlas_names   = [];
			for (let i: i32 = 0; i < project_paint_objects.length; ++i) {
				array_push(project_atlas_objects, 0);
				let i1: i32 = i + 1;
				array_push(project_atlas_names, tr("Atlas") + " " + i1);
			}
		}
		for (let i: i32 = 0; i < project_paint_objects.length; ++i) {
			ui_row2();
			ui_text(project_paint_objects[i].base.name);
			let hatlas: ui_handle_t  = ui_nest(ui_handle(__ID__), i);
			hatlas.i                 = project_atlas_objects[i];
			project_atlas_objects[i] = ui_combo(hatlas, project_atlas_names, tr("Atlas"));
		}
	}
}

function box_export_show_mesh() {
	box_export_mesh_handle.i = context_raw.export_mesh_index;
	ui_box_show_custom(function() {
		let htab: ui_handle_t = ui_handle(__ID__);
		box_export_tab_export_mesh(htab);
	}, 420, 260, null, true, tr("Export"));
}

function box_export_tab_export_mesh(htab: ui_handle_t) {
	let tab_vertical: bool = config_raw.touch_ui;
	if (ui_tab(htab, tr("Export Mesh"), tab_vertical)) {

		ui_row2();

		let h_export_mesh_format: ui_handle_t = ui_handle(__ID__);
		if (h_export_mesh_format.init) {
			h_export_mesh_format.i = context_raw.export_mesh_format;
		}
		let export_mesh_format_combo: string[] = [ "obj", "arm" ];
		context_raw.export_mesh_format         = ui_combo(h_export_mesh_format, export_mesh_format_combo, tr("Format"), true);

		let ar: string[] = [ tr("All") ];
		for (let i: i32 = 0; i < project_paint_objects.length; ++i) {
			let p: mesh_object_t = project_paint_objects[i];
			array_push(ar, p.base.name);
		}
		ui_combo(box_export_mesh_handle, ar, tr("Meshes"), true);

		let apply_displacement: bool = ui_check(ui_handle(__ID__), tr("Apply Displacement"));

		let hmerge: ui_handle_t = ui_handle(__ID__);
		if (hmerge.init) {
			hmerge.b = true;
		}
		let merge_vertices: bool = ui_check(hmerge, tr("Merge Shared Vertices"));

		let tris: i32 = 0;
		let pos: i32  = box_export_mesh_handle.i;
		let paint_objects: mesh_object_t[];
		if (pos == 0) {
			paint_objects = project_paint_objects;
		}
		else {
			let po: mesh_object_t = project_paint_objects[pos - 1];
			paint_objects         = [ po ];
		}
		for (let i: i32 = 0; i < paint_objects.length; ++i) {
			let po: mesh_object_t = paint_objects[i];
			tris += math_floor(po.data.index_array.length / 3);
		}
		ui_text(tris + " " + tr("triangles"));

		ui_row2();
		if (ui_icon_button(tr("Cancel"), icon_t.CLOSE)) {
			ui_box_hide();
		}
		if (ui_icon_button(tr("Export"), icon_t.CHECK)) {
			ui_box_hide();
			_box_export_apply_displacement = apply_displacement;
			_box_export_merge_vertices     = merge_vertices;
			ui_files_show(context_raw.export_mesh_format == mesh_format_t.OBJ ? "obj" : "arm", true, false, function(path: string) {
				/// if (arm_android || arm_ios)
				let f: string = sys_title();
				/// else
				let f: string = ui_files_filename;
				/// end
				if (f == "") {
					f = tr("untitled");
				}
				/// if (arm_android || arm_ios)
				console_toast(tr("Exporting mesh"));
				/// end

				let paint_objects: mesh_object_t[];
				if (box_export_mesh_handle.i == 0) {
					paint_objects = null;
				}
				else {
					let po: mesh_object_t = project_paint_objects[box_export_mesh_handle.i - 1];
					paint_objects         = [ po ];
				}

				export_mesh_run(path + path_sep + f, paint_objects, _box_export_apply_displacement, _box_export_merge_vertices);
			});
		}
	}
}

function box_export_show_material() {
	ui_box_show_custom(function() {
		let htab: ui_handle_t  = ui_handle(__ID__);
		let tab_vertical: bool = config_raw.touch_ui;
		if (ui_tab(htab, tr("Export Material"), tab_vertical)) {
			let h1: ui_handle_t               = ui_handle(__ID__);
			let h2: ui_handle_t               = ui_handle(__ID__);
			h1.b                              = context_raw.pack_assets_on_export;
			h2.b                              = context_raw.write_icon_on_export;
			context_raw.pack_assets_on_export = ui_check(h1, tr("Pack Assets"));
			context_raw.write_icon_on_export  = ui_check(h2, tr("Export Icon"));
			ui_row2();
			if (ui_icon_button(tr("Cancel"), icon_t.CLOSE)) {
				ui_box_hide();
			}
			if (ui_icon_button(tr("Export"), icon_t.CHECK)) {
				ui_box_hide();
				ui_files_show("arm", true, false, function(path: string) {
					let f: string = ui_files_filename;
					if (f == "") {
						f = tr("untitled");
					}
					sys_notify_on_next_frame(function(path: string) {
						export_arm_run_material(path);
					}, path + path_sep + f);
				});
			}
		}
	});
}

function box_export_show_brush() {
	ui_box_show_custom(function() {
		let htab: ui_handle_t  = ui_handle(__ID__);
		let tab_vertical: bool = config_raw.touch_ui;
		if (ui_tab(htab, tr("Export Brush"), tab_vertical)) {
			let h1: ui_handle_t               = ui_handle(__ID__);
			let h2: ui_handle_t               = ui_handle(__ID__);
			h1.b                              = context_raw.pack_assets_on_export;
			h2.b                              = context_raw.write_icon_on_export;
			context_raw.pack_assets_on_export = ui_check(h1, tr("Pack Assets"));
			context_raw.write_icon_on_export  = ui_check(h2, tr("Export Icon"));
			ui_row2();
			if (ui_icon_button(tr("Cancel"), icon_t.CLOSE)) {
				ui_box_hide();
			}
			if (ui_icon_button(tr("Export"), icon_t.CHECK)) {
				ui_box_hide();
				ui_files_show("arm", true, false, function(path: string) {
					let f: string = ui_files_filename;
					if (f == "")
						f = tr("untitled");
					sys_notify_on_next_frame(function(path: string) {
						export_arm_run_brush(path);
					}, path + path_sep + f);
				});
			}
		}
	});
}

function box_export_fetch_presets() {
	box_export_files = file_read_directory(path_data() + path_sep + "export_presets");
	for (let i: i32 = 0; i < box_export_files.length; ++i) {
		let s: string       = box_export_files[i];
		box_export_files[i] = substring(s, 0, s.length - 5); // Strip .json
	}
}

function box_export_parse_preset() {
	let file: string   = "export_presets/" + box_export_files[box_export_hpreset.i] + ".json";
	let blob: buffer_t = data_get_blob(file);
	box_export_preset  = json_parse(sys_buffer_to_string(blob));
	data_delete_blob(file);
}

function box_export_new_preset(name: string) {
	let template: string = "{\
\"textures\": [\
	{ \"name\": \"base\", \"channels\": [\"base_r\", \"base_g\", \"base_b\", \"1.0\"], \"color_space\": \"linear\" }\
]\
}\
";
	if (!ends_with(name, ".json")) {
		name += ".json";
	}
	let path: string = path_data() + path_sep + "export_presets" + path_sep + name;
	iron_file_save_bytes(path, sys_string_to_buffer(template), 0);
}

function box_export_save_preset() {
	let name: string = box_export_files[box_export_hpreset.i];
	if (name == "generic") {
		return; // generic is const
	}
	let path: string = path_data() + path_sep + "export_presets" + path_sep + name + ".json";
	iron_file_save_bytes(path, sys_string_to_buffer(box_export_preset_to_json(box_export_preset)), 0);
}

function box_export_preset_to_json(p: export_preset_t): string {
	json_encode_begin();
	json_encode_begin_array("textures");
	for (let i: i32 = 0; i < p.textures.length; ++i) {
		json_encode_begin_object();
		json_encode_string("name", p.textures[i].name);
		json_encode_string_array("channels", p.textures[i].channels);
		json_encode_string("color_space", p.textures[i].color_space);
		json_encode_end_object();
	}
	json_encode_end_array();
	return json_encode_end();
}