HeapsIO.hide/hrt/shgraph/ShaderFunction.hx

package hrt.shgraph;

import hxsl.*;

using hxsl.Ast;

class ShaderFunction extends ShaderNode {

	@output() var output = SType.Variant;

	var func : TGlobal;

	public function new(func : TGlobal) {
		this.func = func;
	}

	override public function build(key : String) : TExpr {
		var args = [];
		var varArgs = [];

		for (k in getInputsKey()) {
			args.push({ name: k, type: getInput(k).getType() });
			varArgs.push(getInput(k).getVar());
		}

		return {
					p : null,
					t : TVec(3,VFloat),
					e : TBinop(OpAssign, {
						e: TVar(output),
						p: null,
						t: output.type
					}, {
						e: TCall({
							e: TGlobal(func),
							p: null,
							t: TFun([
								{
									ret: output.type,
									args: args
								}
							])
						}, varArgs),
						p: null,
						t: output.type
					})
				};
	}

}