armorpaint/Sources/arm/node/LogicNode.hx

package arm.node;

class LogicNode {

	var tree: LogicTree;
	var inputs: Array<LogicNodeInput> = [];
	var outputs: Array<Array<LogicNode>> = [];

	public function new(tree: LogicTree) {
		this.tree = tree;
	}

	public function addInput(node: LogicNode, from: Int) {
		inputs.push(new LogicNodeInput(node, from));
	}

	public function addOutputs(nodes: Array<LogicNode>) {
		outputs.push(nodes);
	}

	/**
	  Called when this node is activated.
	  @param from impulse index
	**/
	function run(from: Int) {}

	/**
	  Call to activate node connected to the output.
	  @param i output index
	**/
	function runOutput(i: Int) {
		if (i >= outputs.length) return;
		for (o in outputs[i]) {
			// Check which input activated the node
			for (j in 0...o.inputs.length) {
				if (o.inputs[j].node == this) {
					o.run(j);
					break;
				}
			}
		}
	}

	@:allow(arm.node.LogicNodeInput)
	function get(from: Int): Dynamic {
		return this;
	}

	@:allow(arm.node.LogicNodeInput)
	function set(value: Dynamic) {}
}

class LogicNodeInput {

	@:allow(arm.node.LogicNode)
	var node: LogicNode;
	var from: Int; // Socket index

	public function new(node: LogicNode, from: Int) {
		this.node = node;
		this.from = from;
	}

	@:allow(arm.node.LogicNode)
	function get(): Dynamic {
		return node.get(from);
	}

	@:allow(arm.node.LogicNode)
	function set(value: Dynamic) {
		node.set(value);
	}
}