package armory.logicnode;

import iron.Scene;
import armory.trait.internal.CanvasScript;

class CanvasGetPBNode extends LogicNode {

	var canvas: CanvasScript;
	var element: String;
	var at: Int;
	var max: Int;

	public function new(tree: LogicTree) {
		super(tree);
	}

#if arm_ui
	function update() {
		if (!canvas.ready) return;
		tree.removeUpdate(update);

		var e = canvas.getElement(element);
		if (e == null) return;

		at = canvas.getElement(element).progress_at;
        max = canvas.getElement(element).progress_total;
		runOutput(0);
	}

	override function run(from: Int) {
		element = inputs[1].get();
		canvas = Scene.active.getTrait(CanvasScript);
		if (canvas == null) canvas = Scene.active.camera.getTrait(CanvasScript);

		// Ensure canvas is ready
		tree.notifyOnUpdate(update);
		update();
	}
    override function get(from: Int): Dynamic {
		if (from == 1) return at;
		else if (from == 2) return max;
		else return 0;
	}
#end
}
