armorpaint/Sources/Main.hx

package;

import kha.Window;
import kha.System;
import iron.object.Object;
import iron.Scene;
import iron.RenderPath;
import arm.render.Inc;
import arm.render.RenderPathForward;
import arm.render.RenderPathDeferred;
import arm.render.Uniforms;
import arm.sys.BuildMacros;
import arm.Config;
import arm.Context;
import arm.Res;
#if arm_vr
import arm.render.RenderPathForwardVR;
#end

class Main {

	public static inline var title = "ArmorPaint";
	public static var version = "0.8";
	public static var sha = BuildMacros.sha().substr(1, 7);
	public static var date = BuildMacros.date().split(" ")[0];
	static var tasks: Int;

	public static function main() {
		#if arm_snapshot

		var global = js.Syntax.code("globalThis");
		global.kickstart = kickstart;

		Res.embedRaw("Scene", "Scene.arm", untyped global['data/Scene.arm']);
		untyped global['data/Scene.arm'] = null;

		Res.embedRaw("shader_datas", "shader_datas.arm", untyped global['data/shader_datas.arm']);
		untyped global['data/shader_datas.arm'] = null;

		Res.embedFont("font.ttf", untyped global['data/font.ttf']);
		untyped global['data/font.ttf'] = null;

		Res.embedFont("font_mono.ttf", untyped global['data/font_mono.ttf']);
		untyped global['data/font_mono.ttf'] = null;

		var files = [
			"font13.bin",
			"ltc_mag.arm",
			"ltc_mat.arm",
			"default_brush.arm",
			"default_material.arm",
			"World_irradiance.arm",
			"World_radiance.k",
			"World_radiance_0.k",
			"World_radiance_1.k",
			"World_radiance_2.k",
			"World_radiance_3.k",
			"World_radiance_4.k",
			"World_radiance_5.k",
			"World_radiance_6.k",
			"World_radiance_7.k",
			"World_radiance_8.k",
			"brdf.k",
			"color_wheel.k",
			"cursor.k",
			"icons.k",
			"icons2x.k",
			"noise256.k",
			"smaa_search.k",
			"smaa_area.k"
		];

		for (file in files) {
			Res.embedBlob(file, untyped global['data/' + file]);
			untyped global['data/' + file] = null;
		}

		#if (kha_direct3d12 || kha_vulkan)

		var ext = #if kha_direct3d12 ".cso" #else ".spirv" #end ;

		var files_renderlib = [
			"bnoise_rank.k",
			"bnoise_scramble.k",
			"bnoise_sobol.k",
			"raytrace_bake_ao" + ext,
			"raytrace_bake_bent" + ext,
			"raytrace_bake_light" + ext,
			"raytrace_bake_thick" + ext,
			"raytrace_brute_core" + ext,
			"raytrace_brute_full" + ext
		];

		for (file in files_renderlib) {
			Res.embedBlob(file, untyped global['data/' + file]);
			untyped global['data/' + file] = null;
		}

		#end

		#end // arm_snapshot

		#if (!arm_snapshot)
		iron.data.ShaderData.shaderPath = ""; // Use arm_data_dir
		kickstart();
		#end
	}

	@:keep
	public static function kickstart() {
		// Used to locate external application data folder
		Krom.setApplicationName(Main.title);

		tasks = 1;
		tasks++; Config.load(function() { tasks--; start(); });
		#if arm_physics
		tasks++; arm.plugin.PhysicsWorld.load();
		#end
		tasks--; start();
	}

	static function start() {
		if (tasks > 0) return;

		Config.init();
		System.start(Config.getOptions(), function(window: Window) {
			if (Config.raw.layout == null) arm.App.initLayout();
			Krom.setApplicationName(Main.title);
			iron.App.init(function() {
				Scene.setActive("Scene", function(o: Object) {
					Uniforms.init();
					var path = new RenderPath();
					Inc.init(path);

					#if arm_vr
					RenderPathDeferred.init(path); // Allocate gbuffer
					RenderPathForward.init(path);
					RenderPathForwardVR.init(path);
					path.commands = RenderPathForwardVR.commands;
					#else
					if (Context.renderMode == RenderForward) {
						RenderPathDeferred.init(path); // Allocate gbuffer
						RenderPathForward.init(path);
						path.commands = RenderPathForward.commands;
					}
					else {
						RenderPathDeferred.init(path);
						path.commands = RenderPathDeferred.commands;
					}
					#end

					RenderPath.setActive(path);
					new arm.App();
					#if arm_physics
					o.addTrait(new arm.plugin.PhysicsWorld());
					#end
				});
			});
		});
	}
}