]> git.sur5r.net Git - cc65/blobdiff - src/cc65/stmt.c
cleanup
[cc65] / src / cc65 / stmt.c
index cad1f866d373b6e96c3dd710d88eab4491c96ca8..1cb03576a5c8abc6e8c33c26b928369487e0641a 100644 (file)
 
 
 
+/*****************************************************************************/
+/*                                Forwards                                  */
+/*****************************************************************************/
+
+
+
 /*****************************************************************************/
 /*                                  Code                                    */
 /*****************************************************************************/
 
 
 
-static int doif (void)
-/* Handle 'if' statement here */
+static int IfStatement (void)
+/* Handle an 'if' statement */
 {
-    int flab1;
-    int flab2;
-    int gotbreak;
+    unsigned Label1;
+    int GotBreak;
 
     /* Skip the if */
     NextToken ();
 
     /* Generate a jump label and parse the condition */
-    flab1 = GetLocalLabel ();
-    test (flab1, 0);
+    Label1 = GetLocalLabel ();
+    test (Label1, 0);
 
     /* Parse the if body */
-    gotbreak = Statement ();
+    GotBreak = Statement ();
 
     /* Else clause present? */
-    if (curtok != TOK_ELSE) {
+    if (CurTok.Tok != TOK_ELSE) {
 
-       g_defcodelabel (flab1);
+       g_defcodelabel (Label1);
        /* Since there's no else clause, we're not sure, if the a break
         * statement is really executed.
         */
@@ -79,91 +84,112 @@ static int doif (void)
 
     } else {
 
-       /* Skip the else */
+       /* Generate a jump around the else branch */
+       unsigned Label2 = GetLocalLabel ();
+       g_jump (Label2);
+
+       /* Skip the else */
        NextToken ();
 
-       /* If we had some sort of break statement at the end of the if clause,
-        * there's no need to generate an additional jump around the else
-        * clause, since the jump is never reached.
-        */
-       if (!gotbreak) {
-           flab2 = GetLocalLabel ();
-           g_jump (flab2);
-       } else {
-           /* Mark the label as unused */
-           flab2 = 0;
-       }
-       g_defcodelabel (flab1);
-       gotbreak &= Statement ();
+       /* Define the target for the first test */
+       g_defcodelabel (Label1);
+
+       /* Total break only if both branches had a break. */
+       GotBreak &= Statement ();
 
        /* Generate the label for the else clause */
-       if (flab2) {
-           g_defcodelabel (flab2);
-       }
+       g_defcodelabel (Label2);
 
        /* Done */
-       return gotbreak;
+       return GotBreak;
     }
 }
 
 
 
-static void dowhile (char wtype)
-/* Handle 'while' statement here */
+static void DoStatement (void)
+/* Handle the 'do' statement */
 {
-    int loop;
-    int lab;
+    /* Get the loop control labels */
+    unsigned loop = GetLocalLabel ();
+    unsigned lab = GetLocalLabel ();
 
+    /* Skip the while token */
     NextToken ();
-    loop = GetLocalLabel ();
-    lab = GetLocalLabel ();
+
+    /* Add the loop to the loop stack */
     AddLoop (oursp, loop, lab, 0, 0);
+
+    /* Define the head label */
     g_defcodelabel (loop);
-    if (wtype == 'w') {
-
-       /* While loop */
-               test (lab, 0);
-
-       /* If the statement following the while loop is empty, that is, we have
-        * something like "while (1) ;", the test function ommitted the jump as
-        * an optimization. Since we know, the condition codes are set, we can
-        * do another small optimization here, and use a conditional jump
-        * instead an absolute one.
-        */
-       if (curtok == TOK_SEMI) {
-           /* Shortcut */
-           NextToken ();
-           /* Use a conditional jump */
-           g_truejump (CF_NONE, loop);
-       } else {
-           /* There is code inside the while loop */
-           Statement ();
-           g_jump (loop);
-           g_defcodelabel (lab);
-       }
 
-    } else {
+    /* Parse the loop body */
+    Statement ();
+
+    /* Parse the end condition */
+    Consume (TOK_WHILE, "`while' expected");
+    test (loop, 1);
+    ConsumeSemi ();
+
+    /* Define the break label */
+    g_defcodelabel (lab);
+
+    /* Remove the loop from the loop stack */
+    DelLoop ();
+}
 
-       /* Do loop */
-               Statement ();
-       Consume (TOK_WHILE, "`while' expected");
-       test (loop, 1);
-       ConsumeSemi ();
-       g_defcodelabel (lab);
 
+
+static void WhileStatement (void)
+/* Handle the 'while' statement */
+{
+    /* Get the loop control labels */
+    unsigned loop = GetLocalLabel ();
+    unsigned lab = GetLocalLabel ();
+
+    /* Skip the while token */
+    NextToken ();
+
+    /* Add the loop to the loop stack */
+    AddLoop (oursp, loop, lab, 0, 0);
+
+    /* Define the head label */
+    g_defcodelabel (loop);
+
+    /* Test the loop condition */
+    test (lab, 0);
+
+    /* If the statement following the while loop is empty, that is, we have
+     * something like "while (1) ;", the test function ommitted the jump as
+     * an optimization. Since we know, the condition codes are set, we can
+     * do another small optimization here, and use a conditional jump
+     * instead an absolute one.
+     */
+    if (CurTok.Tok == TOK_SEMI) {
+       /* Use a conditional jump */
+       g_truejump (CF_NONE, loop);
+       /* Shortcut */
+       NextToken ();
+    } else {
+       /* There is code inside the while loop, parse the body */
+       Statement ();
+       g_jump (loop);
+       g_defcodelabel (lab);
     }
+
+    /* Remove the loop from the loop stack */
     DelLoop ();
 }
 
 
 
-static void DoReturn (void)
-/* Handle 'return' statement here */
+static void ReturnStatement (void)
+/* Handle the 'return' statement */
 {
     struct expent lval;
 
     NextToken ();
-    if (curtok != TOK_SEMI) {
+    if (CurTok.Tok != TOK_SEMI) {
                if (HasVoidReturn (CurrentFunc)) {
                    Error ("Returning a value in function with return type void");
                }
@@ -188,8 +214,8 @@ static void DoReturn (void)
 
 
 
-static void dobreak (void)
-/* Handle 'break' statement here */
+static void BreakStatement (void)
+/* Handle the 'break' statement */
 {
     LoopDesc* L;
 
@@ -215,8 +241,8 @@ static void dobreak (void)
 
 
 
-static void docontinue (void)
-/* Handle 'continue' statement here */
+static void ContinueStatement (void)
+/* Handle the 'continue' statement */
 {
     LoopDesc* L;
 
@@ -254,11 +280,11 @@ static void docontinue (void)
 
 
 
-static void cascadeswitch (struct expent* eval)
+static void CascadeSwitch (struct expent* eval)
 /* Handle a switch statement for chars with a cmp cascade for the selector */
 {
-    unsigned ExitLab;                  /* Exit label */
-    unsigned NextLab;                  /* Next case label */
+    unsigned ExitLab;                  /* Exit label */
+    unsigned NextLab;                  /* Next case label */
     unsigned CodeLab;          /* Label that starts the actual selector code */
     int HaveBreak;             /* Remember if we exited with break */
     int HaveDefault;           /* Remember if we had a default label */
@@ -280,9 +306,9 @@ static void cascadeswitch (struct expent* eval)
 
     /* Parse the labels */
     lcount = 0;
-    while (curtok != TOK_RCURLY) {
+    while (CurTok.Tok != TOK_RCURLY) {
 
-       if (curtok == TOK_CASE || curtok == TOK_DEFAULT) {
+       if (CurTok.Tok == TOK_CASE || CurTok.Tok == TOK_DEFAULT) {
 
            /* If the code for the previous selector did not end with a
             * break statement, we must jump over the next selector test.
@@ -301,10 +327,10 @@ static void cascadeswitch (struct expent* eval)
                NextLab = 0;
            }
 
-           while (curtok == TOK_CASE || curtok == TOK_DEFAULT) {
+           while (CurTok.Tok == TOK_CASE || CurTok.Tok == TOK_DEFAULT) {
 
                /* Parse the selector */
-               if (curtok == TOK_CASE) {
+               if (CurTok.Tok == TOK_CASE) {
 
                    /* Count labels */
                    ++lcount;
@@ -325,71 +351,71 @@ static void cascadeswitch (struct expent* eval)
                        case T_SCHAR:
                            /* Signed char */
                            if (Val < -128 || Val > 127) {
-                               Error ("Range error");
-                           }
-                           break;
-
-                       case T_UCHAR:
-                           if (Val < 0 || Val > 255) {
-                               Error ("Range error");
-                           }
-                           break;
-
-                       case T_INT:
-                           if (Val < -32768 || Val > 32767) {
-                               Error ("Range error");
-                           }
-                           break;
-
-                       case T_UINT:
-                           if (Val < 0 || Val > 65535) {
-                               Error ("Range error");
-                           }
-                           break;
-
-                       default:
+                               Error ("Range error");
+                           }
+                           break;
+
+                       case T_UCHAR:
+                           if (Val < 0 || Val > 255) {
+                               Error ("Range error");
+                           }
+                           break;
+
+                       case T_INT:
+                           if (Val < -32768 || Val > 32767) {
+                               Error ("Range error");
+                           }
+                           break;
+
+                       case T_UINT:
+                           if (Val < 0 || Val > 65535) {
+                               Error ("Range error");
+                           }
+                           break;
+
+                       default:
                            Internal ("Invalid type: %02X", *eval->e_tptr & 0xFF);
                    }
 
-                   /* Skip the colon */
-                   ConsumeColon ();
-
                    /* Emit a compare */
                    g_cmp (Flags, Val);
 
                    /* If another case follows, we will jump to the code if
                     * the condition is true.
                     */
-                   if (curtok == TOK_CASE) {
-                       /* Create a code label if needed */
-                       if (CodeLab == 0) {
-                           CodeLab = GetLocalLabel ();
-                       }
-                       g_falsejump (CF_NONE, CodeLab);
-                   } else if (curtok != TOK_DEFAULT) {
+                   if (CurTok.Tok == TOK_CASE) {
+                       /* Create a code label if needed */
+                       if (CodeLab == 0) {
+                           CodeLab = GetLocalLabel ();
+                       }
+                       g_falsejump (CF_NONE, CodeLab);
+                   } else if (CurTok.Tok != TOK_DEFAULT) {
                        /* No case follows, jump to next selector */
                        if (NextLab == 0) {
                            NextLab = GetLocalLabel ();
                        }
-                       g_truejump (CF_NONE, NextLab);
+                       g_truejump (CF_NONE, NextLab);
                    }
 
+                   /* Skip the colon */
+                   ConsumeColon ();
+
                } else {
 
                    /* Default case */
                    NextToken ();
 
-                   /* Skip the colon */
-                   ConsumeColon ();
-
                    /* Handle the pathologic case: DEFAULT followed by CASE */
-                   if (curtok == TOK_CASE) {
+                   if (CurTok.Tok == TOK_CASE) {
                        if (CodeLab == 0) {
-                           CodeLab = GetLocalLabel ();
+                           CodeLab = GetLocalLabel ();
                        }
-                       g_jump (CodeLab);
+                       g_jump (CodeLab);
                    }
 
+                   /* Skip the colon */
+                   ConsumeColon ();
+
                    /* Remember that we had a default label */
                    HaveDefault = 1;
                }
@@ -405,7 +431,7 @@ static void cascadeswitch (struct expent* eval)
        }
 
        /* Parse statements */
-       if (curtok != TOK_RCURLY) {
+       if (CurTok.Tok != TOK_RCURLY) {
                    HaveBreak = Statement ();
        }
     }
@@ -415,9 +441,6 @@ static void cascadeswitch (struct expent* eval)
        Warning ("No case labels");
     }
 
-    /* Eat the closing curly brace */
-    NextToken ();
-
     /* Define the exit label and, if there's a next label left, create this
      * one, too.
      */
@@ -426,13 +449,16 @@ static void cascadeswitch (struct expent* eval)
     }
     g_defcodelabel (ExitLab);
 
+    /* Eat the closing curly brace */
+    NextToken ();
+
     /* End the loop */
     DelLoop ();
 }
 
 
 
-static void tableswitch (struct expent* eval)
+static void TableSwitch (struct expent* eval)
 /* Handle a switch statement via table based selector */
 {
     /* Entry for one case in a switch statement */
@@ -448,7 +474,7 @@ static void tableswitch (struct expent* eval)
     int lcount;                        /* Label count */
     int HaveBreak;             /* Last statement has a break */
     int HaveDefault;           /* Remember if we had a default label */
-    unsigned Flags;            /* Code generator flags */
+    unsigned Flags;            /* Code generator flags */
     struct expent lval;                /* Case label expression */
     struct swent *p;
     struct swent *swtab;
@@ -467,14 +493,14 @@ static void tableswitch (struct expent* eval)
     /* Jump behind the code for the CASE labels */
     g_jump (lcase = GetLocalLabel ());
     lcount = 0;
-    while (curtok != TOK_RCURLY) {
-       if (curtok == TOK_CASE || curtok == TOK_DEFAULT) {
+    while (CurTok.Tok != TOK_RCURLY) {
+       if (CurTok.Tok == TOK_CASE || CurTok.Tok == TOK_DEFAULT) {
            if (lcount >= CASE_MAX) {
                        Fatal ("Too many case labels");
            }
            label = GetLocalLabel ();
            do {
-               if (curtok == TOK_CASE) {
+               if (CurTok.Tok == TOK_CASE) {
                            NextToken ();
                    constexpr (&lval);
                    if (!IsClassInt (lval.e_tptr)) {
@@ -490,11 +516,11 @@ static void tableswitch (struct expent* eval)
                    HaveDefault = 1;
                }
                ConsumeColon ();
-           } while (curtok == TOK_CASE || curtok == TOK_DEFAULT);
+           } while (CurTok.Tok == TOK_CASE || CurTok.Tok == TOK_DEFAULT);
            g_defcodelabel (label);
            HaveBreak = 0;
        }
-       if (curtok != TOK_RCURLY) {
+       if (CurTok.Tok != TOK_RCURLY) {
            HaveBreak = Statement ();
        }
     }
@@ -523,7 +549,6 @@ static void tableswitch (struct expent* eval)
     g_defdata (CF_INT | CF_CONST, -((int)lcount)-1, 0);
 
     /* Create the case selector table */
-    AddCodeHint ("casetable");
     p = swtab;
     while (lcount) {
                g_case (Flags, p->sw_lab, p->sw_const); /* Create one label */
@@ -543,10 +568,10 @@ static void tableswitch (struct expent* eval)
 
 
 
-static void doswitch (void)
-/* Handle 'switch' statement here */
+static void SwitchStatement (void)
+/* Handle a 'switch' statement */
 {
-    struct expent eval;                /* Switch statement expression */
+    struct expent eval;                /* Switch statement expression */
 
     /* Eat the "switch" */
     NextToken ();
@@ -561,106 +586,130 @@ static void doswitch (void)
 
     /* Now decide which sort of switch we will create: */
     if (IsTypeChar (eval.e_tptr) || (CodeSizeFactor >= 200 && IsClassInt (eval.e_tptr))) {
-               cascadeswitch (&eval);
+               CascadeSwitch (&eval);
     } else {
-       tableswitch (&eval);
+       TableSwitch (&eval);
     }
 }
 
 
 
-static void dofor (void)
-/* Handle 'for' statement here */
+static void ForStatement (void)
+/* Handle a 'for' statement */
 {
-    int loop;
-    int lab;
-    int linc;
-    int lstat;
     struct expent lval1;
     struct expent lval2;
     struct expent lval3;
 
+    /* Get several local labels needed later */
+    unsigned TestLabel = GetLocalLabel ();
+    unsigned lab       = GetLocalLabel ();
+    unsigned IncLabel  = GetLocalLabel ();
+    unsigned lstat     = GetLocalLabel ();
+
+    /* Skip the FOR token */
     NextToken ();
-    loop = GetLocalLabel ();
-    lab = GetLocalLabel ();
-    linc = GetLocalLabel ();
-    lstat = GetLocalLabel ();
-    AddLoop (oursp, loop, lab, linc, lstat);
+
+    /* Add the loop to the loop stack */
+    AddLoop (oursp, TestLabel, lab, IncLabel, lstat);
+
+    /* Skip the opening paren */
     ConsumeLParen ();
-    if (curtok != TOK_SEMI) {  /* exp1 */
+
+    /* Parse the initializer expression */
+    if (CurTok.Tok != TOK_SEMI) {
        expression (&lval1);
     }
     ConsumeSemi ();
-    g_defcodelabel (loop);
-    if (curtok != TOK_SEMI) {  /* exp2 */
-       boolexpr (&lval2);
-       g_truejump (CF_NONE, lstat);
-       g_jump (lab);
+
+    /* Label for the test expressions */
+    g_defcodelabel (TestLabel);
+
+    /* Parse the test expression */
+    if (CurTok.Tok != TOK_SEMI) {
+       boolexpr (&lval2);
+       g_truejump (CF_NONE, lstat);
+       g_jump (lab);
     } else {
-       g_jump (lstat);
+       g_jump (lstat);
     }
     ConsumeSemi ();
-    g_defcodelabel (linc);
-    if (curtok != TOK_RPAREN) {        /* exp3 */
+
+    /* Label for the increment expression */
+    g_defcodelabel (IncLabel);
+
+    /* Parse the increment expression */
+    if (CurTok.Tok != TOK_RPAREN) {
        expression (&lval3);
     }
+
+    /* Jump to the test */
+    g_jump (TestLabel);
+
+    /* Skip the closing paren */
     ConsumeRParen ();
-    g_jump (loop);
+
+    /* Loop body */
     g_defcodelabel (lstat);
     Statement ();
-    g_jump (linc);
+
+    /* Jump back to the increment expression */
+    g_jump (IncLabel);
+
+    /* Declare the break label */
     g_defcodelabel (lab);
+
+    /* Remove the loop from the loop stack */
     DelLoop ();
 }
 
 
 
 static int CompoundStatement (void)
-/* Compound statement. Allow any number of statements inside braces. */
+/* Compound statement. Allow any number of statements inside braces. The
+ * function returns true if the last statement was a break or return.
+ */
 {
-    int isbrk;
-    int oldsp;
-
-    /* eat LCURLY */
-    NextToken ();
+    int GotBreak;
 
     /* Remember the stack at block entry */
-    oldsp = oursp;
+    int OldStack = oursp;
 
     /* Enter a new lexical level */
     EnterBlockLevel ();
 
+    /* Skip the rcurly */
+    NextToken ();
+
     /* Parse local variable declarations if any */
     DeclareLocals ();
 
     /* Now process statements in this block */
-    isbrk = 0;
-    while (curtok != TOK_RCURLY) {
-       if (curtok != TOK_CEOF) {
-           isbrk = Statement ();
+    GotBreak = 0;
+    while (CurTok.Tok != TOK_RCURLY) {
+       if (CurTok.Tok != TOK_CEOF) {
+           GotBreak = Statement ();
        } else {
            break;
        }
     }
 
-    /* Emit references to imports/exports for this block */
-    EmitExternals ();
-
     /* Clean up the stack. */
-    if (isbrk) {
-       oursp = oldsp;
-    } else {
-       g_space (oursp - oldsp);
-       oursp = oldsp;
+    if (!GotBreak) {
+       g_space (oursp - OldStack);
     }
+    oursp = OldStack;
+
+    /* Skip the closing brace */
+    ConsumeRCurly ();
+
+    /* Emit references to imports/exports for this block */
+    EmitExternals ();
 
     /* Leave the lexical level */
     LeaveBlockLevel ();
 
-    /* Eat closing brace */
-    ConsumeRCurly ();
-
-    return isbrk;
+    return GotBreak;
 }
 
 
@@ -672,60 +721,60 @@ int Statement (void)
 {
     struct expent lval;
 
-    /* */
-    if (curtok == TOK_IDENT && nxttok == TOK_COLON) {
+    /* Check for a label */
+    if (CurTok.Tok == TOK_IDENT && NextTok.Tok == TOK_COLON) {
 
        /* Special handling for a label */
        DoLabel ();
 
     } else {
 
-       switch (curtok) {
+       switch (CurTok.Tok) {
 
            case TOK_LCURLY:
-               return CompoundStatement ();
+               return CompoundStatement ();
 
            case TOK_IF:
-               return doif ();
+               return IfStatement ();
 
            case TOK_WHILE:
-               dowhile ('w');
-               break;
+               WhileStatement ();
+               break;
 
            case TOK_DO:
-               dowhile ('d');
-               break;
+               DoStatement ();
+               break;
 
            case TOK_SWITCH:
-               doswitch ();
-               break;
+               SwitchStatement ();
+               break;
 
            case TOK_RETURN:
-               DoReturn ();
-               ConsumeSemi ();
-               return 1;
+               ReturnStatement ();
+               ConsumeSemi ();
+               return 1;
 
            case TOK_BREAK:
-               dobreak ();
+               BreakStatement ();
                ConsumeSemi ();
                return 1;
 
            case TOK_CONTINUE:
-               docontinue ();
+               ContinueStatement ();
                ConsumeSemi ();
                return 1;
 
            case TOK_FOR:
-               dofor ();
+               ForStatement ();
                break;
 
            case TOK_GOTO:
-               DoGoto ();
+               GotoStatement ();
                ConsumeSemi ();
                return 1;
 
            case TOK_SEMI:
-               /* ignore it. */
+               /* Ignore it */
                NextToken ();
                break;
 
@@ -734,9 +783,8 @@ int Statement (void)
                break;
 
            default:
-               AddCodeHint ("stmt:start");
+               /* Actual statement */
                expression (&lval);
-               AddCodeHint ("stmt:end");
                ConsumeSemi ();
        }
     }