rep putn test

asm.y

@@ -75,6 +75,7 @@ Construct
 | I_REP I_DUP NumericalConstant { $$ = node(DUP, $3); }
 | I_REP I_DROP NumericalConstant { $$ = node(DROP, $3); }
 | I_REP I_ADD NumericalConstant { $$ = node2(ADD, $3, imm_val(1)); }
+| I_REP I_PUTN NumericalConstant { $$ = node2(PUTN, $3, imm_val(1)); }
 | I_REP I_SUB NumericalConstant { $$ = node2(SUB, $3, imm_val(1)); }
 | I_REP I_MUL NumericalConstant { $$ = node2(MUL, $3, imm_val(1)); }
 | I_REP I_DIV NumericalConstant { $$ = node2(DIV, $3, imm_val(1)); }

asm_gen.c

@@ -171,6 +171,15 @@ void asm_gen(FILE * output, vector(struct node_t) data, int optlevel) {
                     T;N;T;T; // GETN
                     break;
                 case PUTN:
+                    if(it->data2.type != IMM_NONE) {
+                        // REP PUTN X times
+                        rep(it->data1.value) {
+                            T;N;S;T;
+                        }
+
+                        break;
+                    }
+
                     if(it->data1.type != IMM_NONE) {
                         S;S;numeral(output, it->data1.value); // PUTN N => PUSH N
                     }

mktest.sh

@@ -35,10 +35,10 @@ case $ans in
         cd "ws-build-run"
 
         _prompt "program" "asm"
-        _prompt "input" "ws.in"
-        _prompt "output" "ws.out"
+        _prompt "input" "asm.ws.in"
+        _prompt "output" "asm.ws.out"
 
-        [ $(wc -c < "$name.in") -eq 0 ] && rm "$name.in"
+        [ $(wc -c < "$name.asm.ws.in") -eq 0 ] && rm "$name.asm.ws.in"
         ;;
     *)
         exit;;

tests/ws-build-run/rep_putn.asm

@@ -0,0 +1,2 @@
+1/2/3/4
+REP PUTN 4

tests/ws-build-run/rep_putn.asm.ws.out

@@ -0,0 +1 @@
+4321