From 3fc005ada833407047fb1f0860bb6dd07708f360 Mon Sep 17 00:00:00 2001
From: Lukas Krickl <lukas@krickl.dev>
Date: Sun, 24 Dec 2023 18:33:24 +0100
Subject: [PATCH] Added counter symbol naming to .rep

---
 src/ulas.c   |  22 ++++++++++++++++++++++
 tests/t0.bin | Bin 189 -> 204 bytes
 tests/t0.s   |   3 ++-
 3 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/src/ulas.c b/src/ulas.c
index 0a11ad1..dffea89 100644
--- a/src/ulas.c
+++ b/src/ulas.c
@@ -2585,6 +2585,26 @@ int ulas_asmdirchr(FILE *dst, const char **line, unsigned long n, int *rc) {
 }
 
 int ulas_asmdirrep(FILE *dst, FILE *src, const char **line, unsigned long n) {
+  char name[ULAS_SYMNAMEMAX];
+  ulas_tok(&ulas.tok, line, n);
+  if (!ulas_isname(ulas.tok.buf, ulas.tok.maxlen)) {
+    ULASERR("Unexpected token '%s'\n", ulas.tok.buf);
+    return -1;
+  }
+  strncpy(name, ulas.tok.buf, ULAS_SYMNAMEMAX);
+
+  // consume ,
+  ulas_tok(&ulas.tok, line, n);
+  if (strncmp(ulas.tok.buf, ",", ulas.tok.maxlen) != 0) {
+    ULASERR("Unexpected token '%s'. Expected ','\n", ulas.tok.buf);
+    return -1;
+  }
+
+  union ulas_val val = {0};
+  val.intv = 0;
+  struct ulas_tok tok = {ULAS_INT, val};
+  ulas_symbolset(name, ulas.scope, tok, 0);
+
   int repval = 0;
   int rc = 0;
 
@@ -2609,6 +2629,8 @@ int ulas_asmdirrep(FILE *dst, FILE *src, const char **line, unsigned long n) {
   unsigned long argline_len = strlen(*line);
 
   for (int i = 0; i < repval; i += step) {
+    tok.val.intv = i;
+    ulas_symbolset(name, ulas.scope, tok, 0);
     rc = ulas_asmline(dst, src, *line, argline_len);
     if (rc == -1) {
       break;
diff --git a/tests/t0.bin b/tests/t0.bin
index d480d73b283696097da413d86de46765d34ebc83..ec638b95dde8bd2d4f4eed37942e391c1444394a 100644
GIT binary patch
delta 25
ccmdnXc!qJpE+IPxJ0?38Ajt@%nSm@;09O+QD*ylh

delta 10
RcmX@ZxR-IlF6OebG5{CM1Umo#

diff --git a/tests/t0.s b/tests/t0.s
index 296c6fd..17e0ab0 100644
--- a/tests/t0.s
+++ b/tests/t0.s
@@ -131,4 +131,5 @@ l3: .db 1
 .db 1, 2, 3
 .chr 01233213
 
-.rep 6, 2, halt
+.rep repc, 6, 2, ld a, repc
+.rep repc, 6, 1, ld a, repc
-- 
2.30.2