Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 11:30:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update compiler.c

Browse Source
This commit is contained in:
blueloveTH 2024-06-29 14:38:52 +08:00
parent a03a3bdbf8
commit 14a01c0e6d
1 changed files with 52 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
src/compiler/compiler.c
View File

@ -6,7 +6,6 @@
#include <ctype.h> #include <ctype.h>
/* expr.h */ /* expr.h */
typedef struct Expr Expr; typedef struct Expr Expr;
typedef struct Ctx Ctx; typedef struct Ctx Ctx;
@ -39,21 +38,22 @@ typedef struct ExprVt {
((self)->vt->emit_inplace ? vtcall(emit_inplace, self, ctx) : vtemit_(self, ctx)) ((self)->vt->emit_inplace ? vtcall(emit_inplace, self, ctx) : vtemit_(self, ctx))
#define vtemit_istore(self, ctx) \ #define vtemit_istore(self, ctx) \
((self)->vt->emit_istore ? vtcall(emit_istore, self, ctx) : vtemit_store(self, ctx)) ((self)->vt->emit_istore ? vtcall(emit_istore, self, ctx) : vtemit_store(self, ctx))
#define vtdelete(self) \
do { \
if(self) { \
if((self)->vt->dtor) (self)->vt->dtor(self); \
PoolExpr_dealloc(self); \
} \
} while(0)
#define COMMON_HEADER \ #define EXPR_COMMON_HEADER \
const ExprVt* vt; \ const ExprVt* vt; \
int line; int line;
typedef struct Expr { typedef struct Expr {
COMMON_HEADER EXPR_COMMON_HEADER
} Expr; } Expr;
static void Expr__delete(Expr* self) {
if(!self) return;
if(self->vt->dtor) self->vt->dtor(self);
PoolExpr_dealloc(self);
}
/* context.h */ /* context.h */
typedef struct Ctx { typedef struct Ctx {
CodeObject* co; // 1 CodeEmitContext <=> 1 CodeObject* CodeObject* co; // 1 CodeEmitContext <=> 1 CodeObject*
@ -104,7 +104,7 @@ void Ctx__s_emit_decorators(Ctx*, int count);
/* expr.c */ /* expr.c */
typedef struct NameExpr { typedef struct NameExpr {
COMMON_HEADER EXPR_COMMON_HEADER
StrName name; StrName name;
NameScope scope; NameScope scope;
} NameExpr; } NameExpr;
@ -167,7 +167,7 @@ NameExpr* NameExpr__new(StrName name, NameScope scope) {
} }
typedef struct StarredExpr { typedef struct StarredExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* child; Expr* child;
int level; int level;
} StarredExpr; } StarredExpr;
@ -201,14 +201,14 @@ StarredExpr* StarredExpr__new(Expr* child, int level) {
// InvertExpr, NotExpr, NegatedExpr // InvertExpr, NotExpr, NegatedExpr
// NOTE: NegatedExpr always contains a non-const child. Should not generate -1 or -0.1 // NOTE: NegatedExpr always contains a non-const child. Should not generate -1 or -0.1
typedef struct UnaryExpr { typedef struct UnaryExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* child; Expr* child;
Opcode opcode; Opcode opcode;
} UnaryExpr; } UnaryExpr;
void UnaryExpr__dtor(Expr* self_) { void UnaryExpr__dtor(Expr* self_) {
UnaryExpr* self = (UnaryExpr*)self_; UnaryExpr* self = (UnaryExpr*)self_;
Expr__delete(self->child); vtdelete(self->child);
} }
static void UnaryExpr__emit_(Expr* self_, Ctx* ctx) { static void UnaryExpr__emit_(Expr* self_, Ctx* ctx) {
@ -230,7 +230,7 @@ UnaryExpr* UnaryExpr__new(Expr* child, Opcode opcode) {
// LongExpr, BytesExpr // LongExpr, BytesExpr
typedef struct RawStringExpr { typedef struct RawStringExpr {
COMMON_HEADER EXPR_COMMON_HEADER
c11_string value; c11_string value;
Opcode opcode; Opcode opcode;
} RawStringExpr; } RawStringExpr;
@ -254,7 +254,7 @@ RawStringExpr* RawStringExpr__new(c11_string value, Opcode opcode) {
} }
typedef struct ImagExpr { typedef struct ImagExpr {
COMMON_HEADER EXPR_COMMON_HEADER
double value; double value;
} ImagExpr; } ImagExpr;
@ -278,7 +278,7 @@ ImagExpr* ImagExpr__new(double value) {
} }
typedef struct LiteralExpr { typedef struct LiteralExpr {
COMMON_HEADER EXPR_COMMON_HEADER
const TokenValue* value; const TokenValue* value;
} LiteralExpr; } LiteralExpr;
@ -320,7 +320,7 @@ LiteralExpr* LiteralExpr__new(const TokenValue* value) {
} }
typedef struct SliceExpr { typedef struct SliceExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* start; Expr* start;
Expr* stop; Expr* stop;
Expr* step; Expr* step;
@ -328,9 +328,9 @@ typedef struct SliceExpr {
void SliceExpr__dtor(Expr* self_) { void SliceExpr__dtor(Expr* self_) {
SliceExpr* self = (SliceExpr*)self_; SliceExpr* self = (SliceExpr*)self_;
Expr__delete(self->start); vtdelete(self->start);
Expr__delete(self->stop); vtdelete(self->stop);
Expr__delete(self->step); vtdelete(self->step);
} }
void SliceExpr__emit_(Expr* self_, Ctx* ctx) { void SliceExpr__emit_(Expr* self_, Ctx* ctx) {
@ -364,7 +364,7 @@ SliceExpr* SliceExpr__new() {
// ListExpr, DictExpr, SetExpr, TupleExpr // ListExpr, DictExpr, SetExpr, TupleExpr
typedef struct SequenceExpr { typedef struct SequenceExpr {
COMMON_HEADER EXPR_COMMON_HEADER
c11_array /*T=Expr* */ items; c11_array /*T=Expr* */ items;
Opcode opcode; Opcode opcode;
} SequenceExpr; } SequenceExpr;
@ -380,7 +380,7 @@ static void SequenceExpr__emit_(Expr* self_, Ctx* ctx) {
void SequenceExpr__dtor(Expr* self_) { void SequenceExpr__dtor(Expr* self_) {
SequenceExpr* self = (SequenceExpr*)self_; SequenceExpr* self = (SequenceExpr*)self_;
c11__foreach(Expr*, &self->items, e) Expr__delete(*e); c11__foreach(Expr*, &self->items, e) vtdelete(*e);
c11_array__dtor(&self->items); c11_array__dtor(&self->items);
} }
@ -483,7 +483,7 @@ SequenceExpr* TupleExpr__new(int count) {
} }
typedef struct CompExpr { typedef struct CompExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* expr; // loop expr Expr* expr; // loop expr
Expr* vars; // loop vars Expr* vars; // loop vars
Expr* iter; // loop iter Expr* iter; // loop iter
@ -495,10 +495,10 @@ typedef struct CompExpr {
void CompExpr__dtor(Expr* self_) { void CompExpr__dtor(Expr* self_) {
CompExpr* self = (CompExpr*)self_; CompExpr* self = (CompExpr*)self_;
Expr__delete(self->expr); vtdelete(self->expr);
Expr__delete(self->vars); vtdelete(self->vars);
Expr__delete(self->iter); vtdelete(self->iter);
Expr__delete(self->cond); vtdelete(self->cond);
} }
void CompExpr__emit_(Expr* self_, Ctx* ctx) { void CompExpr__emit_(Expr* self_, Ctx* ctx) {
@ -543,7 +543,7 @@ CompExpr* CompExpr__new(Opcode op0, Opcode op1) {
} }
typedef struct LambdaExpr { typedef struct LambdaExpr {
COMMON_HEADER EXPR_COMMON_HEADER
int index; int index;
} LambdaExpr; } LambdaExpr;
@ -563,7 +563,7 @@ LambdaExpr* LambdaExpr__new(int index) {
} }
typedef struct FStringExpr { typedef struct FStringExpr {
COMMON_HEADER EXPR_COMMON_HEADER
c11_string src; c11_string src;
} FStringExpr; } FStringExpr;
@ -744,7 +744,7 @@ FStringExpr* FStringExpr__new(c11_string src) {
// AndExpr, OrExpr // AndExpr, OrExpr
typedef struct LogicBinaryExpr { typedef struct LogicBinaryExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* lhs; Expr* lhs;
Expr* rhs; Expr* rhs;
Opcode opcode; Opcode opcode;
@ -752,8 +752,8 @@ typedef struct LogicBinaryExpr {
void LogicBinaryExpr__dtor(Expr* self_) { void LogicBinaryExpr__dtor(Expr* self_) {
LogicBinaryExpr* self = (LogicBinaryExpr*)self_; LogicBinaryExpr* self = (LogicBinaryExpr*)self_;
Expr__delete(self->lhs); vtdelete(self->lhs);
Expr__delete(self->rhs); vtdelete(self->rhs);
} }
void LogicBinaryExpr__emit_(Expr* self_, Ctx* ctx) { void LogicBinaryExpr__emit_(Expr* self_, Ctx* ctx) {
@ -777,13 +777,13 @@ LogicBinaryExpr* LogicBinaryExpr__new(Expr* lhs, Expr* rhs, Opcode opcode) {
} }
typedef struct GroupedExpr { typedef struct GroupedExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* child; Expr* child;
} GroupedExpr; } GroupedExpr;
void GroupedExpr__dtor(Expr* self_) { void GroupedExpr__dtor(Expr* self_) {
GroupedExpr* self = (GroupedExpr*)self_; GroupedExpr* self = (GroupedExpr*)self_;
Expr__delete(self->child); vtdelete(self->child);
} }
void GroupedExpr__emit_(Expr* self_, Ctx* ctx) { void GroupedExpr__emit_(Expr* self_, Ctx* ctx) {
@ -815,7 +815,7 @@ GroupedExpr* GroupedExpr__new(Expr* child) {
} }
typedef struct BinaryExpr { typedef struct BinaryExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* lhs; Expr* lhs;
Expr* rhs; Expr* rhs;
TokenIndex op; TokenIndex op;
@ -824,8 +824,8 @@ typedef struct BinaryExpr {
static void BinaryExpr__dtor(Expr* self_) { static void BinaryExpr__dtor(Expr* self_) {
BinaryExpr* self = (BinaryExpr*)self_; BinaryExpr* self = (BinaryExpr*)self_;
Expr__delete(self->lhs); vtdelete(self->lhs);
Expr__delete(self->rhs); vtdelete(self->rhs);
} }
static Opcode cmp_token2op(TokenIndex token) { static Opcode cmp_token2op(TokenIndex token) {
@ -915,7 +915,7 @@ static void BinaryExpr__emit_(Expr* self_, Ctx* ctx) {
} }
typedef struct TernaryExpr { typedef struct TernaryExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* cond; Expr* cond;
Expr* true_expr; Expr* true_expr;
Expr* false_expr; Expr* false_expr;
@ -923,9 +923,9 @@ typedef struct TernaryExpr {
void TernaryExpr__dtor(Expr* self_) { void TernaryExpr__dtor(Expr* self_) {
TernaryExpr* self = (TernaryExpr*)self_; TernaryExpr* self = (TernaryExpr*)self_;
Expr__delete(self->cond); vtdelete(self->cond);
Expr__delete(self->true_expr); vtdelete(self->true_expr);
Expr__delete(self->false_expr); vtdelete(self->false_expr);
} }
void TernaryExpr__emit_(Expr* self_, Ctx* ctx) { void TernaryExpr__emit_(Expr* self_, Ctx* ctx) {
@ -952,15 +952,15 @@ TernaryExpr* TernaryExpr__new() {
} }
typedef struct SubscrExpr { typedef struct SubscrExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* lhs; Expr* lhs;
Expr* rhs; Expr* rhs;
} SubscrExpr; } SubscrExpr;
void SubscrExpr__dtor(Expr* self_) { void SubscrExpr__dtor(Expr* self_) {
SubscrExpr* self = (SubscrExpr*)self_; SubscrExpr* self = (SubscrExpr*)self_;
Expr__delete(self->lhs); vtdelete(self->lhs);
Expr__delete(self->rhs); vtdelete(self->rhs);
} }
void SubscrExpr__emit_(Expr* self_, Ctx* ctx) { void SubscrExpr__emit_(Expr* self_, Ctx* ctx) {
@ -1034,7 +1034,7 @@ SubscrExpr* SubscrExpr__new(Expr* lhs, Expr* rhs) {
} }
typedef struct AttribExpr { typedef struct AttribExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* child; Expr* child;
StrName name; StrName name;
} AttribExpr; } AttribExpr;
@ -1096,7 +1096,7 @@ typedef struct CallExprKwArg {
} CallExprKwArg; } CallExprKwArg;
typedef struct CallExpr { typedef struct CallExpr {
COMMON_HEADER EXPR_COMMON_HEADER
Expr* callable; Expr* callable;
c11_vector /*T=Expr* */ args; c11_vector /*T=Expr* */ args;
// **a will be interpreted as a special keyword argument: {"**": a} // **a will be interpreted as a special keyword argument: {"**": a}
@ -1105,9 +1105,9 @@ typedef struct CallExpr {
void CallExpr__dtor(Expr* self_) { void CallExpr__dtor(Expr* self_) {
CallExpr* self = (CallExpr*)self_; CallExpr* self = (CallExpr*)self_;
Expr__delete(self->callable); vtdelete(self->callable);
c11__foreach(Expr*, &self->args, e) Expr__delete(*e); c11__foreach(Expr*, &self->args, e) vtdelete(*e);
c11__foreach(CallExprKwArg, &self->kwargs, e) Expr__delete(e->val); c11__foreach(CallExprKwArg, &self->kwargs, e) vtdelete(e->val);
c11_vector__dtor(&self->args); c11_vector__dtor(&self->args);
c11_vector__dtor(&self->kwargs); c11_vector__dtor(&self->kwargs);
} }
@ -1207,7 +1207,7 @@ void Ctx__s_emit_top(Ctx* self) {
Expr* top = c11_vector__back(Expr*, &self->s_expr); Expr* top = c11_vector__back(Expr*, &self->s_expr);
top->vt->emit_(top, self); top->vt->emit_(top, self);
c11_vector__pop(&self->s_expr); c11_vector__pop(&self->s_expr);
Expr__delete(top); vtdelete(top);
} }
// push // push
@ -1221,7 +1221,7 @@ int Ctx__s_size(Ctx* self) { return self->s_expr.count; }
// pop -> delete // pop -> delete
void Ctx__s_pop(Ctx* self) { void Ctx__s_pop(Ctx* self) {
Expr__delete(c11_vector__back(Expr*, &self->s_expr)); vtdelete(c11_vector__back(Expr*, &self->s_expr));
c11_vector__pop(&self->s_expr); c11_vector__pop(&self->s_expr);
} }
@ -1236,7 +1236,7 @@ Expr* Ctx__s_popx(Ctx* self) {
void Ctx__s_clean(Ctx* self) { void Ctx__s_clean(Ctx* self) {
c11_smallmap_s2n__dtor(&self->co_consts_string_dedup_map); // ?? c11_smallmap_s2n__dtor(&self->co_consts_string_dedup_map); // ??
for(int i = 0; i < self->s_expr.count; i++) { for(int i = 0; i < self->s_expr.count; i++) {
Expr__delete(c11__getitem(Expr*, &self->s_expr, i)); vtdelete(c11__getitem(Expr*, &self->s_expr, i));
} }
c11_vector__clear(&self->s_expr); c11_vector__clear(&self->s_expr);
} }