Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 1 | /* |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 2 | ** $Id: lcode.h $ |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 3 | ** Code generator for Lua |
| 4 | ** See Copyright Notice in lua.h |
| 5 | */ |
| 6 | |
| 7 | #ifndef lcode_h |
| 8 | #define lcode_h |
| 9 | |
| 10 | #include "llex.h" |
| 11 | #include "lobject.h" |
| 12 | #include "lopcodes.h" |
| 13 | #include "lparser.h" |
| 14 | |
| 15 | |
| 16 | /* |
| 17 | ** Marks the end of a patch list. It is an invalid value both as an absolute |
| 18 | ** address, and as a list link (would link an element to itself). |
| 19 | */ |
| 20 | #define NO_JUMP (-1) |
| 21 | |
| 22 | |
| 23 | /* |
| 24 | ** grep "ORDER OPR" if you change these enums (ORDER OP) |
| 25 | */ |
| 26 | typedef enum BinOpr { |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 27 | /* arithmetic operators */ |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 28 | OPR_ADD, OPR_SUB, OPR_MUL, OPR_MOD, OPR_POW, |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 29 | OPR_DIV, OPR_IDIV, |
| 30 | /* bitwise operators */ |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 31 | OPR_BAND, OPR_BOR, OPR_BXOR, |
| 32 | OPR_SHL, OPR_SHR, |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 33 | /* string operator */ |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 34 | OPR_CONCAT, |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 35 | /* comparison operators */ |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 36 | OPR_EQ, OPR_LT, OPR_LE, |
| 37 | OPR_NE, OPR_GT, OPR_GE, |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 38 | /* logical operators */ |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 39 | OPR_AND, OPR_OR, |
| 40 | OPR_NOBINOPR |
| 41 | } BinOpr; |
| 42 | |
| 43 | |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 44 | /* true if operation is foldable (that is, it is arithmetic or bitwise) */ |
| 45 | #define foldbinop(op) ((op) <= OPR_SHR) |
| 46 | |
| 47 | |
| 48 | #define luaK_codeABC(fs,o,a,b,c) luaK_codeABCk(fs,o,a,b,c,0) |
| 49 | |
| 50 | |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 51 | typedef enum UnOpr { OPR_MINUS, OPR_BNOT, OPR_NOT, OPR_LEN, OPR_NOUNOPR } UnOpr; |
| 52 | |
| 53 | |
| 54 | /* get (pointer to) instruction of given 'expdesc' */ |
| 55 | #define getinstruction(fs,e) ((fs)->f->code[(e)->u.info]) |
| 56 | |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 57 | |
| 58 | #define luaK_setmultret(fs,e) luaK_setreturns(fs, e, LUA_MULTRET) |
| 59 | |
| 60 | #define luaK_jumpto(fs,t) luaK_patchlist(fs, luaK_jump(fs), t) |
| 61 | |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 62 | LUAI_FUNC int luaK_code (FuncState *fs, Instruction i); |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 63 | LUAI_FUNC int luaK_codeABx (FuncState *fs, OpCode o, int A, unsigned int Bx); |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 64 | LUAI_FUNC int luaK_codeAsBx (FuncState *fs, OpCode o, int A, int Bx); |
| 65 | LUAI_FUNC int luaK_codeABCk (FuncState *fs, OpCode o, int A, |
| 66 | int B, int C, int k); |
| 67 | LUAI_FUNC int luaK_isKint (expdesc *e); |
| 68 | LUAI_FUNC int luaK_exp2const (FuncState *fs, const expdesc *e, TValue *v); |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 69 | LUAI_FUNC void luaK_fixline (FuncState *fs, int line); |
| 70 | LUAI_FUNC void luaK_nil (FuncState *fs, int from, int n); |
| 71 | LUAI_FUNC void luaK_reserveregs (FuncState *fs, int n); |
| 72 | LUAI_FUNC void luaK_checkstack (FuncState *fs, int n); |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 73 | LUAI_FUNC void luaK_int (FuncState *fs, int reg, lua_Integer n); |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 74 | LUAI_FUNC void luaK_dischargevars (FuncState *fs, expdesc *e); |
| 75 | LUAI_FUNC int luaK_exp2anyreg (FuncState *fs, expdesc *e); |
| 76 | LUAI_FUNC void luaK_exp2anyregup (FuncState *fs, expdesc *e); |
| 77 | LUAI_FUNC void luaK_exp2nextreg (FuncState *fs, expdesc *e); |
| 78 | LUAI_FUNC void luaK_exp2val (FuncState *fs, expdesc *e); |
| 79 | LUAI_FUNC int luaK_exp2RK (FuncState *fs, expdesc *e); |
| 80 | LUAI_FUNC void luaK_self (FuncState *fs, expdesc *e, expdesc *key); |
| 81 | LUAI_FUNC void luaK_indexed (FuncState *fs, expdesc *t, expdesc *k); |
| 82 | LUAI_FUNC void luaK_goiftrue (FuncState *fs, expdesc *e); |
| 83 | LUAI_FUNC void luaK_goiffalse (FuncState *fs, expdesc *e); |
| 84 | LUAI_FUNC void luaK_storevar (FuncState *fs, expdesc *var, expdesc *e); |
| 85 | LUAI_FUNC void luaK_setreturns (FuncState *fs, expdesc *e, int nresults); |
| 86 | LUAI_FUNC void luaK_setoneret (FuncState *fs, expdesc *e); |
| 87 | LUAI_FUNC int luaK_jump (FuncState *fs); |
| 88 | LUAI_FUNC void luaK_ret (FuncState *fs, int first, int nret); |
| 89 | LUAI_FUNC void luaK_patchlist (FuncState *fs, int list, int target); |
| 90 | LUAI_FUNC void luaK_patchtohere (FuncState *fs, int list); |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 91 | LUAI_FUNC void luaK_concat (FuncState *fs, int *l1, int l2); |
| 92 | LUAI_FUNC int luaK_getlabel (FuncState *fs); |
| 93 | LUAI_FUNC void luaK_prefix (FuncState *fs, UnOpr op, expdesc *v, int line); |
| 94 | LUAI_FUNC void luaK_infix (FuncState *fs, BinOpr op, expdesc *v); |
| 95 | LUAI_FUNC void luaK_posfix (FuncState *fs, BinOpr op, expdesc *v1, |
| 96 | expdesc *v2, int line); |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 97 | LUAI_FUNC void luaK_settablesize (FuncState *fs, int pc, |
| 98 | int ra, int asize, int hsize); |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 99 | LUAI_FUNC void luaK_setlist (FuncState *fs, int base, int nelems, int tostore); |
Elliott Hughes | e0e566c | 2020-08-05 12:30:16 -0700 | [diff] [blame] | 100 | LUAI_FUNC void luaK_finish (FuncState *fs); |
| 101 | LUAI_FUNC l_noret luaK_semerror (LexState *ls, const char *msg); |
Lukas Zilka | 6cdd400 | 2018-12-05 23:30:46 +0100 | [diff] [blame] | 102 | |
| 103 | |
| 104 | #endif |