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

Merge pull request #71 from alexcpatel/op-oreq-xoreq

Browse Source 
Operators |= and ^= implemented
This commit is contained in:
Thakee Nathees 2021-06-15 10:25:27 +05:30 committed by GitHub
commit da5d5d6993
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 30 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/pk_compiler.c
View File

@ -94,7 +94,8 @@ typedef enum {
TK_DIVEQ, // /=
TK_ANDEQ, // &=
// TODO: TK_OREQ, TK_XOREQ
TK_OREQ, // |=
TK_XOREQ, // ^=
TK_SRIGHT, // >>
TK_SLEFT, // <<
@ -666,10 +667,12 @@ static void lexToken(Compiler* compiler) {
case '&':
setNextTwoCharToken(compiler, '=', TK_AMP, TK_ANDEQ);
return;
// TODO: TK_OREQ, TK_XOREQ
case '|': setNextToken(compiler, TK_PIPE); return;
case '^': setNextToken(compiler, TK_CARET); return;
case '|':
setNextTwoCharToken(compiler, '=', TK_PIPE, TK_OREQ);
return;
case '^':
setNextTwoCharToken(compiler, '=', TK_CARET, TK_XOREQ);
return;
case '\n': setNextToken(compiler, TK_LINE); return;
@ -875,7 +878,8 @@ static bool matchAssignment(Compiler* compiler) {
if (match(compiler, TK_STAREQ)) return true;
if (match(compiler, TK_DIVEQ)) return true;
if (match(compiler, TK_ANDEQ)) return true;
// TODO: match TK_OREQ, TK_XOREQ
if (match(compiler, TK_OREQ)) return true;
if (match(compiler, TK_XOREQ)) return true;
return false;
}
@ -1050,7 +1054,8 @@ GrammarRule rules[] = { // Prefix Infix Infix Precedence
/* TK_STAREQ */ NO_RULE,
/* TK_DIVEQ */ NO_RULE,
/* TK_ANDEQ */ NO_RULE,
// TODO: add TK_OREQ, TK_XOREQ
/* TK_OREQ */ NO_RULE,
/* TK_XOREQ */ NO_RULE,
/* TK_SRIGHT */ { NULL, exprBinaryOp, PREC_BITWISE_SHIFT },
/* TK_SLEFT */ { NULL, exprBinaryOp, PREC_BITWISE_SHIFT },
/* TK_MODULE */ NO_RULE,
@ -1730,7 +1735,8 @@ static void emitAssignment(Compiler* compiler, TokenType assignment) {
case TK_STAREQ: emitOpcode(compiler, OP_MULTIPLY); break;
case TK_DIVEQ: emitOpcode(compiler, OP_DIVIDE); break;
case TK_ANDEQ: emitOpcode(compiler, OP_BIT_AND); break;
// TODO: TK_OREQ, TK_XOREQ
case TK_OREQ: emitOpcode(compiler, OP_BIT_OR); break;
case TK_XOREQ: emitOpcode(compiler, OP_BIT_XOR); break;
default:
UNREACHABLE();
break;

16
tests/lang/basics.pk
View File

@ -65,6 +65,22 @@ for i in 0..1000
x = i; assert((x&=-1) == i)
end
x = 43690 ; assert((x|=21845) == 65535)
x = 204 ; assert((x|=119) == 255)
for i in 0..1000
x = i; assert((x|=0) == i)
x = i; assert((x|=-1) == -1)
x = i; assert((x|=i) == i)
end
x = 4321 ; assert((x^=1234) == 5171)
x = 77 ; assert((x^=88) == 21)
for i in 0..1000
x = i; assert((x^=i) == 0)
x = i; assert((x^=0) == i)
x = i; assert((x^=-1) == -i-1)
end
# If we got here with no errors, then all test passed
print('All tests PASSED (100%)')