HeapsIO.hide/hrt/shgraph/nodes/Cond.hx

package hrt.shgraph.nodes;

using hxsl.Ast;

@name("Condition")
@description("Create a custom condition between two inputs")
@group("Condition")
class Cond extends ShaderNode {

	@input("Left") var leftVar = SType.Number;
	@input("Right") var rightVar = SType.Number;

	@output("Boolean") var output = SType.Bool;

	@prop() var condition : Binop;

	override public function checkValidityInput(key : String, type : ShaderType.SType) : Bool {

		if (key == "leftVar" && rightVar != null && !rightVar.isEmpty())
			return ShaderType.checkCompatibilities(type, ShaderType.getSType(rightVar.getType()));

		if (key == "rightVar" && leftVar != null && !leftVar.isEmpty())
			return ShaderType.checkCompatibilities(type, ShaderType.getSType(leftVar.getType()));

		return true;
	}

	override public function computeOutputs() {
		if (leftVar != null && !leftVar.isEmpty() && rightVar != null && !rightVar.isEmpty()) {
			var type = leftVar.getVar(rightVar.getType()).t;
			switch(type) {
				case TVec(s, t):
					removeOutput("output");
					throw ShaderException.t("Vector of bools is not supported", this.id); //addOutput("output", TVec(s, VBool));
				case TFloat:
					addOutput("output", TBool);
				default:
					removeOutput("output");
			}
		} else
			removeOutput("output");
	}

	override public function build(key : String) : TExpr {
		return {
				p : null,
				t : output.type,
				e : TBinop(OpAssign, {
						e: TVar(output),
						p: null,
						t: output.type
					}, {e: TBinop(this.condition,
							leftVar.getVar(rightVar.getType()),
							rightVar.getVar(leftVar.getType())),
						p: null, t: output.type })
			};
	}

	var availableConditions = [OpEq, OpNotEq, OpGt, OpGte, OpLt, OpLte, OpAnd, OpOr];
	var conditionStrings 	= ["==", "!=",    ">",  ">=",  "<",  "<=",  "AND", "OR"];

	override public function loadProperties(props : Dynamic) {
		this.condition = std.Type.createEnum(Binop, Reflect.field(props, "Condition"));
	}

	override public function saveProperties() : Dynamic {
		if (this.condition == null)
			this.condition = availableConditions[0];
		var properties = {
			condition: this.condition.getName()
		};

		return properties;
	}

	#if editor
	override public function getPropertiesHTML(width : Float) : Array<hide.Element> {
		var elements = super.getPropertiesHTML(width);
		var element = new hide.Element('<div style="width: ${width * 0.8}px; height: 40px"></div>');
		element.append('<span>Condition</span>');
		element.append(new hide.Element('<select id="condition"></select>'));

		if (this.condition == null) {
			this.condition = availableConditions[0];
		}
		var input = element.children("select");
		var indexOption = 0;
		for (c in conditionStrings) {
			input.append(new hide.Element('<option value="${indexOption}">${c}</option>'));
			if (this.condition == availableConditions[indexOption]) {
				input.val(indexOption);
			}
			indexOption++;
		}
		input.on("change", function(e) {
			var value = input.val();
			this.condition = availableConditions[value];
		});

		elements.push(element);

		return elements;
	}
	#end

}