diff options
Diffstat (limited to 'lcode.c')
| -rw-r--r-- | lcode.c | 18 |
1 files changed, 15 insertions, 3 deletions
| @@ -68,6 +68,15 @@ static int tonumeral (const expdesc *e, TValue *v) { | |||
| 68 | 68 | ||
| 69 | 69 | ||
| 70 | /* | 70 | /* |
| 71 | ** Get the constant value from a constant expression | ||
| 72 | */ | ||
| 73 | static TValue *const2val (FuncState *fs, const expdesc *e) { | ||
| 74 | lua_assert(e->k == VCONST); | ||
| 75 | return &fs->ls->dyd->actvar.arr[e->u.info].k; | ||
| 76 | } | ||
| 77 | |||
| 78 | |||
| 79 | /* | ||
| 71 | ** If expression is a constant, fills 'v' with its value | 80 | ** If expression is a constant, fills 'v' with its value |
| 72 | ** and returns 1. Otherwise, returns 0. | 81 | ** and returns 1. Otherwise, returns 0. |
| 73 | */ | 82 | */ |
| @@ -85,6 +94,10 @@ int luaK_exp2const (FuncState *fs, const expdesc *e, TValue *v) { | |||
| 85 | setsvalue(fs->ls->L, v, e->u.strval); | 94 | setsvalue(fs->ls->L, v, e->u.strval); |
| 86 | return 1; | 95 | return 1; |
| 87 | } | 96 | } |
| 97 | case VCONST: { | ||
| 98 | setobj(fs->ls->L, v, const2val(fs, e)); | ||
| 99 | return 1; | ||
| 100 | } | ||
| 88 | default: return tonumeral(e, v); | 101 | default: return tonumeral(e, v); |
| 89 | } | 102 | } |
| 90 | } | 103 | } |
| @@ -730,14 +743,13 @@ void luaK_setoneret (FuncState *fs, expdesc *e) { | |||
| 730 | 743 | ||
| 731 | 744 | ||
| 732 | /* | 745 | /* |
| 733 | ** Ensure that expression 'e' is not a variable. | 746 | ** Ensure that expression 'e' is not a variable (nor a constant). |
| 734 | ** (Expression still may have jump lists.) | 747 | ** (Expression still may have jump lists.) |
| 735 | */ | 748 | */ |
| 736 | void luaK_dischargevars (FuncState *fs, expdesc *e) { | 749 | void luaK_dischargevars (FuncState *fs, expdesc *e) { |
| 737 | switch (e->k) { | 750 | switch (e->k) { |
| 738 | case VCONST: { | 751 | case VCONST: { |
| 739 | TValue *val = &fs->ls->dyd->actvar.arr[e->u.info].k; | 752 | const2exp(const2val(fs, e), e); |
| 740 | const2exp(val, e); | ||
| 741 | break; | 753 | break; |
| 742 | } | 754 | } |
| 743 | case VLOCAL: { /* already in a register */ | 755 | case VLOCAL: { /* already in a register */ |
