Browse code

(uxnasm) Lambdas support

Devine Lu Linvega authored on 29/07/2023 23:14:19
Showing 1 changed files
... ...
@@ -35,8 +35,9 @@ typedef struct {
35 35
 
36 36
 typedef struct {
37 37
 	Uint8 data[LENGTH];
38
+	Uint8 lambda_stack[0x100], lambda_ptr, lambda_count;
38 39
 	unsigned int ptr, length;
39
-	Uint16 llen, mlen, rlen;
40
+	Uint16 label_len, macro_len, refs_len;
40 41
 	Label labels[0x400];
41 42
 	Macro macros[0x100];
42 43
 	Reference refs[0x800];
... ...
@@ -87,7 +88,7 @@ static Macro *
87 88
 findmacro(char *name)
88 89
 {
89 90
 	int i;
90
-	for(i = 0; i < p.mlen; i++)
91
+	for(i = 0; i < p.macro_len; i++)
91 92
 		if(scmp(p.macros[i].name, name, 0x40))
92 93
 			return &p.macros[i];
93 94
 	return NULL;
... ...
@@ -97,7 +98,7 @@ static Label *
97 98
 findlabel(char *name)
98 99
 {
99 100
 	int i;
100
-	for(i = 0; i < p.llen; i++)
101
+	for(i = 0; i < p.label_len; i++)
101 102
 		if(scmp(p.labels[i].name, name, 0x40))
102 103
 			return &p.labels[i];
103 104
 	return NULL;
... ...
@@ -139,9 +140,9 @@ makemacro(char *name, FILE *f)
139 140
 		return error("Macro name is hex number", name);
140 141
 	if(findopcode(name) || scmp(name, "BRK", 4) || !slen(name))
141 142
 		return error("Macro name is invalid", name);
142
-	if(p.mlen == 0x100)
143
+	if(p.macro_len == 0x100)
143 144
 		return error("Macros limit exceeded", name);
144
-	m = &p.macros[p.mlen++];
145
+	m = &p.macros[p.macro_len++];
145 146
 	scpy(name, m->name, 0x40);
146 147
 	while(fscanf(f, "%63s", word) == 1) {
147 148
 		if(word[0] == '{') continue;
... ...
@@ -165,9 +166,9 @@ makelabel(char *name)
165 166
 		return error("Label name is hex number", name);
166 167
 	if(findopcode(name) || scmp(name, "BRK", 4) || !slen(name))
167 168
 		return error("Label name is invalid", name);
168
-	if(p.llen == 0x400)
169
+	if(p.label_len == 0x400)
169 170
 		return error("Labels limit exceeded", name);
170
-	l = &p.labels[p.llen++];
171
+	l = &p.labels[p.label_len++];
171 172
 	l->addr = p.ptr;
172 173
 	l->refs = 0;
173 174
 	scpy(name, l->name, 0x40);
... ...
@@ -179,9 +180,9 @@ makereference(char *scope, char *label, char rune, Uint16 addr)
179 180
 {
180 181
 	char subw[0x40], parent[0x40];
181 182
 	Reference *r;
182
-	if(p.rlen >= 0x800)
183
+	if(p.refs_len >= 0x800)
183 184
 		return error("References limit exceeded", label);
184
-	r = &p.refs[p.rlen++];
185
+	r = &p.refs[p.refs_len++];
185 186
 	if(label[0] == '&') {
186 187
 		if(!sublabel(subw, scope, label + 1))
187 188
 			return error("Invalid sublabel", label);
... ...
@@ -351,6 +352,18 @@ parse(char *w, FILE *f)
351 352
 		while((c = w[++i]))
352 353
 			if(!writebyte(c)) return 0;
353 354
 		break;
355
+	case '{': /* lambda start */
356
+		char laba[8] = "lambda0\0";
357
+		p.lambda_stack[p.lambda_ptr++] = p.lambda_count;
358
+		laba[6] = '0' + p.lambda_count++;
359
+		makereference(p.scope, laba, ' ', p.ptr + 1);
360
+		return writebyte(0x60) && writeshort(0xffff, 0);
361
+	case '}': /* lambda end */
362
+		char labb[8] = "lambda0\0";
363
+		labb[6] = '0' + p.lambda_stack[--p.lambda_ptr];
364
+		if(!makelabel(labb))
365
+			return error("Invalid label", labb);
366
+		return writebyte(0x6f);
354 367
 	case '[':
355 368
 	case ']':
356 369
 		if(slen(w) == 1) break; /* else fallthrough */
... ...
@@ -384,7 +397,7 @@ resolve(void)
384 397
 	Label *l;
385 398
 	int i;
386 399
 	Uint16 a;
387
-	for(i = 0; i < p.rlen; i++) {
400
+	for(i = 0; i < p.refs_len; i++) {
388 401
 		Reference *r = &p.refs[i];
389 402
 		switch(r->rune) {
390 403
 		case '_':
... ...
@@ -443,7 +456,7 @@ static void
443 456
 review(char *filename)
444 457
 {
445 458
 	int i;
446
-	for(i = 0; i < p.llen; i++)
459
+	for(i = 0; i < p.label_len; i++)
447 460
 		if(p.labels[i].name[0] >= 'A' && p.labels[i].name[0] <= 'Z')
448 461
 			continue; /* Ignore capitalized labels(devices) */
449 462
 		else if(!p.labels[i].refs)
... ...
@@ -453,8 +466,8 @@ review(char *filename)
453 466
 		filename,
454 467
 		p.length - TRIM,
455 468
 		(p.length - TRIM) / 652.80,
456
-		p.llen,
457
-		p.mlen);
469
+		p.label_len,
470
+		p.macro_len);
458 471
 }
459 472
 
460 473
 static void
... ...
@@ -467,7 +480,7 @@ writesym(char *filename)
467 480
 		return;
468 481
 	fp = fopen(scat(scpy(filename, symdst, slen(filename) + 1), ".sym"), "w");
469 482
 	if(fp != NULL) {
470
-		for(i = 0; i < p.llen; i++) {
483
+		for(i = 0; i < p.label_len; i++) {
471 484
 			Uint8 hb = p.labels[i].addr >> 8, lb = p.labels[i].addr & 0xff;
472 485
 			fwrite(&hb, 1, 1, fp);
473 486
 			fwrite(&lb, 1, 1, fp);