mirrors/pocketlang
1
0
Fork 0
mirror of https://github.com/zekexiao/pocketlang.git synced 2025-02-05 20:26:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add modeq implementation (#108)

Browse Source 
* add modeq implementation

* code review feedback
This commit is contained in:
Takashi Idobe 2021-06-22 08:19:52 -04:00 committed by GitHub
parent e177f80410
commit c0f00f6cae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/pk_compiler.c
View File

@ -96,7 +96,7 @@ typedef enum {
TK_MINUSEQ, // -=
TK_STAREQ, // *=
TK_DIVEQ, // /=
//TK_MODEQ, // %=
TK_MODEQ, // %=
TK_ANDEQ, // &=
TK_OREQ, // |=
@ -742,7 +742,9 @@ static void lexToken(Compiler* compiler) {
case ']': setNextToken(compiler, TK_RBRACKET); return;
case '{': setNextToken(compiler, TK_LBRACE); return;
case '}': setNextToken(compiler, TK_RBRACE); return;
case '%': setNextToken(compiler, TK_PERCENT); return;
case '%':
setNextTwoCharToken(compiler, '=', TK_PERCENT, TK_MODEQ);
return;
case '~': setNextToken(compiler, TK_TILD); return;
@ -966,6 +968,7 @@ static bool matchAssignment(Compiler* compiler) {
if (match(compiler, TK_ANDEQ)) return true;
if (match(compiler, TK_OREQ)) return true;
if (match(compiler, TK_XOREQ)) return true;
if (match(compiler, TK_MODEQ)) return true;
return false;
}
@ -1149,6 +1152,7 @@ GrammarRule rules[] = { // Prefix Infix Infix Precedence
/* TK_MINUSEQ */ NO_RULE,
/* TK_STAREQ */ NO_RULE,
/* TK_DIVEQ */ NO_RULE,
/* TK_MODEQ */ NO_RULE,
/* TK_ANDEQ */ NO_RULE,
/* TK_OREQ */ NO_RULE,
/* TK_XOREQ */ NO_RULE,
@ -1853,6 +1857,7 @@ static void emitAssignment(Compiler* compiler, TokenType assignment) {
case TK_ANDEQ: emitOpcode(compiler, OP_BIT_AND); break;
case TK_OREQ: emitOpcode(compiler, OP_BIT_OR); break;
case TK_XOREQ: emitOpcode(compiler, OP_BIT_XOR); break;
case TK_MODEQ: emitOpcode(compiler, OP_MOD); break;
default:
UNREACHABLE();
break;

3
tests/lang/basics.pk
View File

@ -84,6 +84,9 @@ for i in 0..1000
x = i; assert((x&=-1) == i)
end
x = 42 ; assert((x%=3) == 0)
x = 27 ; assert((x%=8) == 3)
x = 43690 ; assert((x|=21845) == 65535)
x = 204 ; assert((x|=119) == 255)
for i in 0..1000