armorpaint/paint/sources/args.c

#include "global.h"

void args_parse() {
	if (iron_get_arg_count() > 1) {
		args_use = true;

		i32 i = 0;
		while (i < iron_get_arg_count()) {
			// Process each arg
			char *current_arg = iron_get_arg(i);

			if (path_is_project(current_arg)) {
				gc_unroot(project_filepath);
				project_filepath = string_copy(current_arg);
				gc_root(project_filepath);
			}
			else if (string_equals(current_arg, "--b") || string_equals(current_arg, "--background")) {
				args_background = true;
			}
			else if (path_is_texture(current_arg)) {
				gc_unroot(args_asset_path);
				args_asset_path = string_copy(current_arg);
				gc_root(args_asset_path);
			}
			else if (string_equals(current_arg, "--export-textures") && (i + 3) <= iron_get_arg_count()) {
				args_export_textures = true;
				++i;
				gc_unroot(args_export_textures_type);
				args_export_textures_type = string_copy(iron_get_arg(i));
				gc_root(args_export_textures_type);
				++i;
				gc_unroot(args_export_textures_preset);
				args_export_textures_preset = string_copy(iron_get_arg(i));
				gc_root(args_export_textures_preset);
				++i;
				gc_unroot(args_export_textures_path);
				args_export_textures_path = string_copy(iron_get_arg(i));
				gc_root(args_export_textures_path);
			}
			else if (string_equals(current_arg, "--reload-mesh")) {
				args_reimport_mesh = true;
			}
			else if (string_equals(current_arg, "--export-mesh") && (i + 1) <= iron_get_arg_count()) {
				args_export_mesh = true;
				++i;
				gc_unroot(args_export_mesh_path);
				args_export_mesh_path = string_copy(iron_get_arg(i));
				gc_root(args_export_mesh_path);
			}
			else if (path_is_mesh(current_arg) || (i > 1 && !starts_with(current_arg, "-") && path_is_folder(current_arg))) {
				gc_unroot(args_asset_path);
				args_asset_path = string_copy(current_arg);
				gc_root(args_asset_path);
			}
			else if (string_equals(current_arg, "--export-material") && (i + 1) <= iron_get_arg_count()) {
				args_export_material = true;
				++i;
				gc_unroot(args_export_material_path);
				args_export_material_path = string_copy(iron_get_arg(i));
				gc_root(args_export_material_path);
			}
			++i;
		}
	}
}

void args_run_export_queue(void *_) {
	export_texture_run(args_export_textures_path, false);
}

void args_run_on_next_frame(void *_) {
	if (!string_equals(project_filepath, "")) {
		import_arm_run_project(project_filepath);
	}
	else if (!string_equals(args_asset_path, "")) {
		import_asset_run(args_asset_path, -1, -1, false, true, NULL);
		if (path_is_texture(args_asset_path)) {
			ui_base_show_2d_view(VIEW_2D_TYPE_ASSET);
		}
	}
	else if (args_reimport_mesh) {
		project_reimport_mesh();
	}

	if (args_export_textures) {
		if (!path_is_folder(args_export_textures_path)) {
			iron_log(tr("Invalid export directory"));
		}

		if (string_equals(args_export_textures_type, "png")) {
			base_bits_handle->i      = TEXTURE_BITS_BITS8;
			context_raw->format_type = TEXTURE_LDR_FORMAT_PNG;
		}
		else if (string_equals(args_export_textures_type, "jpg")) {
			base_bits_handle->i      = TEXTURE_BITS_BITS8;
			context_raw->format_type = TEXTURE_LDR_FORMAT_JPG;
		}
		else if (string_equals(args_export_textures_type, "exr16")) {
			base_bits_handle->i = TEXTURE_BITS_BITS16;
		}
		else if (string_equals(args_export_textures_type, "exr32")) {
			base_bits_handle->i = TEXTURE_BITS_BITS32;
		}
		else {
			iron_log(tr("Invalid texture type"));
		}

		context_raw->layers_export = EXPORT_MODE_VISIBLE;

		// Get export preset and apply the correct one from args
		gc_unroot(box_export_files);
		box_export_files = file_read_directory(string("%s%sexport_presets", path_data(), PATH_SEP));
		gc_root(box_export_files);
		for (i32 i = 0; i < box_export_files->length; ++i) {
			char *s                     = box_export_files->buffer[i];
			box_export_files->buffer[i] = substring(s, 0, string_length(s) - 5); // Strip .json
		}

		char *file = string("export_presets/%s.json", box_export_files->buffer[0]);
		for (i32 i = 0; i < box_export_files->length; ++i) {
			char *f = box_export_files->buffer[i];
			if (string_equals(f, args_export_textures_preset)) {
				file = string("export_presets/%s.json", box_export_files->buffer[array_index_of(box_export_files, f)]);
			}
		}

		buffer_t *blob = data_get_blob(file);
		gc_unroot(box_export_preset);
		box_export_preset = json_parse(sys_buffer_to_string(blob));
		gc_root(box_export_preset);
		data_delete_blob(string("export_presets/%s", file));

		// Export queue
		sys_notify_on_next_frame(&args_run_export_queue, NULL);
	}
	else if (args_export_mesh) {
		if (!path_is_folder(args_export_mesh_path)) {
			iron_log(tr("Invalid export directory"));
		}

		char *f = ui_files_filename;
		if (string_equals(f, "")) {
			f = string_copy(tr("untitled"));
		}
		export_mesh_run(string("%s%s%s", args_export_mesh_path, PATH_SEP, f), NULL, false, true);
	}
	else if (args_export_material) {
		context_raw->write_icon_on_export = true;
		export_arm_run_material(args_export_material_path);
	}

	if (args_background) {
		iron_stop();
	}
}

void args_run() {
	if (args_use) {
		sys_notify_on_next_frame(&args_run_on_next_frame, NULL);
	}
}