nqasm: support binary arithmetic operators in immediate operands

1 files changed, 5 insertions, 0 deletions

diff --git a/nqasm.c b/nqasm.c
index 49b4ec6..e915bd7 100644
--- a/nqasm.c
+++ b/nqasm.c
@@ -78,6 +78,11 @@ static long eval_argument(const struct argument *arg, uint16_t pc)
 	case ARG_UNARY_NOT: return !eval_argument(arg->children[0], pc);
 	case ARG_UNARY_INV: return ~eval_argument(arg->children[0], pc);
 
+	case ARG_ADD: return eval_argument(arg->children[0], pc) + eval_argument(arg->children[1], pc);
+	case ARG_SUB: return eval_argument(arg->children[0], pc) - eval_argument(arg->children[1], pc);
+	case ARG_MUL: return eval_argument(arg->children[0], pc) * eval_argument(arg->children[1], pc);
+	case ARG_DIV: return eval_argument(arg->children[0], pc) / eval_argument(arg->children[1], pc);
+
 	case ARG_LABEL:
 		l = vector_search(&labels->v, &(struct label) { .name = arg->label }, cmp_label);
 		if (l) return l->addr;