]> git.sur5r.net Git - cc65/blobdiff - src/cc65/function.c
Optimizations, cleanup in codegen
[cc65] / src / cc65 / function.c
index c0b913ab46e73bf2e11df79c52b1af845ee42ed0..06786e6e8cf16a6fa0f5a7d12f1683db42098f50 100644 (file)
@@ -6,10 +6,10 @@
 /*                                                                           */
 /*                                                                           */
 /*                                                                           */
-/* (C) 2000     Ullrich von Bassewitz                                        */
-/*              Wacholderweg 14                                              */
-/*              D-70597 Stuttgart                                            */
-/* EMail:       uz@musoftware.de                                             */
+/* (C) 2000-2001 Ullrich von Bassewitz                                       */
+/*               Wacholderweg 14                                             */
+/*               D-70597 Stuttgart                                           */
+/* EMail:        uz@cc65.org                                                 */
 /*                                                                           */
 /*                                                                           */
 /* This software is provided 'as-is', without any expressed or implied       */
@@ -47,6 +47,7 @@
 #include "litpool.h"
 #include "locals.h"
 #include "scanner.h"
+#include "segments.h"
 #include "stmt.h"
 #include "symtab.h"
 #include "function.h"
 /* Structure that holds all data needed for function activation */
 struct Function {
     struct SymEntry*           FuncEntry;      /* Symbol table entry */
-    type*              ReturnType;     /* Function return type */
+    type*              ReturnType;     /* Function return type */
     struct FuncDesc*   Desc;           /* Function descriptor */
-    CodeMark           EntryCode;      /* Backpatch addr for entry code */
     int                        Reserved;       /* Reserved local space */
     unsigned           RetLab;         /* Return code label */
+    int                        TopLevelSP;     /* SP at function top level */
 };
 
 /* Pointer to current function */
@@ -84,15 +85,15 @@ static Function* NewFunction (struct SymEntry* Sym)
 /* Create a new function activation structure and return it */
 {
     /* Allocate a new structure */
-    Function* F = xmalloc (sizeof (Function));
+    Function* F = (Function*) xmalloc (sizeof (Function));
 
     /* Initialize the fields */
     F->FuncEntry  = Sym;
-    F->ReturnType = Sym->Type + 1 + DECODE_SIZE;
-    F->Desc      = DecodePtr (Sym->Type + 1);
-    F->EntryCode  = 0;
+    F->ReturnType = GetFuncReturn (Sym->Type);
+    F->Desc      = (FuncDesc*) DecodePtr (Sym->Type + 1);
     F->Reserved          = 0;
-    F->RetLab    = GetLabel ();
+    F->RetLab    = GetLocalLabel ();
+    F->TopLevelSP = 0;
 
     /* Return the new structure */
     return F;
@@ -148,10 +149,10 @@ int HasVoidReturn (const Function* F)
 
 
 
-void RememberEntry (Function* F)
-/* Remember the current output position for local space creation later */
+int IsVariadic (const Function* F)
+/* Return true if this is a variadic function */
 {
-    F->EntryCode = GetCodePos ();
+    return (F->Desc->Flags & FD_VARIADIC) != 0;
 }
 
 
@@ -164,6 +165,14 @@ unsigned GetRetLab (const Function* F)
 
 
 
+int GetTopLevelSP (const Function* F)
+/* Get the value of the stack pointer on function top level */
+{
+    return F->TopLevelSP;
+}
+
+
+
 int ReserveLocalSpace (Function* F, unsigned Size)
 /* Reserve (but don't allocate) the given local space and return the stack
  * offset.
@@ -182,9 +191,6 @@ void AllocLocalSpace (Function* F)
 {
     if (F->Reserved > 0) {
 
-       /* Switch to the code segment */
-       g_usecode ();
-
        /* Create space on the stack */
        g_space (F->Reserved);
 
@@ -207,11 +213,11 @@ void AllocLocalSpace (Function* F)
 void NewFunc (SymEntry* Func)
 /* Parse argument declarations and function body. */
 {
-    int isbrk;
-    unsigned Flags;
+    int HadReturn;
+    int IsVoidFunc;
 
     /* Get the function descriptor from the function entry */
-    FuncDesc* D = DecodePtr (Func->Type+1);
+    FuncDesc* D = Func->V.F.Func;
 
     /* Allocate the function activation record for the function */
     CurrentFunc = NewFunction (Func);
@@ -223,30 +229,24 @@ void NewFunc (SymEntry* Func)
      * The latter is different depending on the type of the function (variadic
      * or not).
      */
-    AddLocalSym ("__fixargs__", type_uint, SC_DEF | SC_CONST, D->ParamSize);
-    if (D->Flags & FD_ELLIPSIS) {
+    AddConstSym ("__fixargs__", type_uint, SC_DEF | SC_CONST, D->ParamSize);
+    if (D->Flags & FD_VARIADIC) {
        /* Variadic function. The variable must be const. */
        static const type T [] = { T_UCHAR | T_QUAL_CONST, T_END };
        AddLocalSym ("__argsize__", T, SC_DEF | SC_REF | SC_AUTO, 0);
     } else {
        /* Non variadic */
-       AddLocalSym ("__argsize__", type_uchar, SC_DEF | SC_CONST, D->ParamSize);
+               AddConstSym ("__argsize__", type_uchar, SC_DEF | SC_CONST, D->ParamSize);
     }
 
     /* Function body now defined */
     Func->Flags |= SC_DEF;
 
-    /* Need a starting curly brace */
-    if (curtok != TOK_LCURLY) {
-       Error ("`{' expected");
-    }
-
     /* Setup register variables */
     InitRegVars ();
 
-    /* Switch to the code segment and define the function name label */
-    g_usecode ();
-    g_defgloblabel (Func->Name);
+    /* Allocate code and data segments for this function */
+    Func->V.F.Seg = PushSegments (Func);
 
     /* If this is a fastcall function, push the last parameter onto the stack */
     if (IsFastCallFunc (Func->Type) && D->ParamCount > 0) {
@@ -255,7 +255,7 @@ void NewFunc (SymEntry* Func)
        unsigned Flags;
 
        /* Fastcall functions may never have an ellipsis or the compiler is buggy */
-       CHECK ((D->Flags & FD_ELLIPSIS) == 0);
+       CHECK ((D->Flags & FD_VARIADIC) == 0);
 
        /* Get a pointer to the last parameter entry */
        LastParam = D->SymTab->SymTail;
@@ -278,33 +278,55 @@ void NewFunc (SymEntry* Func)
     /* Generate function entry code if needed */
     g_enter (TypeOf (Func->Type), GetParamSize (CurrentFunc));
 
-    /* Remember the current code position. This may be used later to create
-     * local variable space once we have created the function body itself.
-     * Currently this is not possible because the stack offsets of all locals
-     * have to be known in advance.
-     */
-    RememberEntry (CurrentFunc);
-
-    /* Parse the function body */
+    /* Setup the stack */
     oursp = 0;
-    isbrk = compound ();
 
-    /* If the function did not end with an return statement, create exit code */
-    if (!isbrk) {
-#if 0
-       /* If the function has a return type, flag an error */
-       if (!voidfunc) {
-           Error ("Function `%s' must return a value", Func->Name);
+    /* Need a starting curly brace */
+    ConsumeLCurly ();
+
+    /* Parse local variable declarations if any */
+    DeclareLocals ();
+
+    /* Remember the current stack pointer. All variables allocated elsewhere
+     * must be dropped when doing a return from an inner block.
+     */
+    CurrentFunc->TopLevelSP = oursp;
+
+    /* Now process statements in this block */
+    HadReturn = 0;
+    while (CurTok.Tok != TOK_RCURLY) {
+       if (CurTok.Tok != TOK_CEOF) {
+           HadReturn = Statement (0);
+       } else {
+           break;
        }
-#endif
-       RestoreRegVars (0);
+    }
 
-       Flags = HasVoidReturn (CurrentFunc)? CF_NONE : CF_REG;
-        g_leave (Flags, 0);
+    /* If the function has a return type but no return statement, flag
+     * a warning
+     */
+    IsVoidFunc = HasVoidReturn (CurrentFunc);
+#if 0
+    /* Does not work reliably */
+    if (!IsVoidFunc && !HadReturn) {
+       Warning ("Function `%s' should return a value", Func->Name);
     }
+#endif
 
-    /* Dump literal data created by the function */
-    DumpLiteralPool ();
+    /* Output the function exit code label */
+    g_defcodelabel (GetRetLab (CurrentFunc));
+
+    /* Restore the register variables */
+    RestoreRegVars (!IsVoidFunc);
+
+    /* Generate the exit code */
+    g_leave ();
+
+    /* Eat the closing brace */
+    ConsumeRCurly ();
+
+    /* Emit references to imports/exports */
+    EmitExternals ();
 
     /* Cleanup register variables */
     DoneRegVars ();
@@ -312,6 +334,9 @@ void NewFunc (SymEntry* Func)
     /* Leave the lexical level */
     LeaveFunctionLevel ();
 
+    /* Switch back to the old segments */
+    PopSegments ();
+
     /* Reset the current function pointer */
     FreeFunction (CurrentFunc);
     CurrentFunc = 0;
@@ -319,4 +344,3 @@ void NewFunc (SymEntry* Func)
 
 
 
-