/*****************************************************************************/
/* */
-/* condasm.c */
+/* condasm.c */
/* */
-/* Conditional assembly support for ca65 */
+/* Conditional assembly support for ca65 */
/* */
/* */
/* */
/*****************************************************************************/
-/* Data */
+/* Data */
/*****************************************************************************/
/* Maximum count of nested .ifs */
-#define MAX_IFS 256
+#define MAX_IFS 256
/* Set of bitmapped flags for the if descriptor */
enum {
- ifNone = 0x0000, /* No flag */
- ifCond = 0x0001, /* IF condition was true */
- ifElse = 0x0002, /* We had a .ELSE branch */
- ifNeedTerm = 0x0004 /* Need .ENDIF termination */
+ ifNone = 0x0000, /* No flag */
+ ifCond = 0x0001, /* IF condition was true */
+ ifParentCond= 0x0002, /* IF condition of parent */
+ ifElse = 0x0004, /* We had a .ELSE branch */
+ ifNeedTerm = 0x0008, /* Need .ENDIF termination */
};
+/* The overall .IF condition */
+int IfCond = 1;
+
/*****************************************************************************/
-/* struct IfDesc */
+/* struct IfDesc */
/*****************************************************************************/
/* One .IF descriptor */
typedef struct IfDesc IfDesc;
struct IfDesc {
- unsigned Flags; /* Bitmapped flags, see above */
+ unsigned Flags; /* Bitmapped flags, see above */
Collection LineInfos; /* File position of the .IF */
- const char* Name; /* Name of the directive */
+ const char* Name; /* Name of the directive */
};
/* The .IF stack */
-static IfDesc* AllocIf (const char* Directive, int NeedTerm)
-/* Alloc a new element from the .IF stack */
-{
- IfDesc* ID;
-
- /* Check for stack overflow */
- if (IfCount >= MAX_IFS) {
- Fatal ("Too many nested .IFs");
- }
-
- /* Alloc one element */
- ID = &IfStack [IfCount++];
-
- /* Initialize elements */
- ID->Flags = NeedTerm? ifNeedTerm : ifNone;
- ID->LineInfos = EmptyCollection;
- GetFullLineInfo (&ID->LineInfos, 0);
- ID->Name = Directive;
-
- /* Return the result */
- return ID;
-}
-
-
-
static IfDesc* GetCurrentIf (void)
/* Return the current .IF descriptor */
{
if (IfCount == 0) {
- return 0;
+ return 0;
} else {
- return &IfStack [IfCount-1];
+ return &IfStack[IfCount-1];
}
}
-static void FreeIf (void)
-/* Free all .IF descriptors until we reach one with the NeedTerm bit set */
+static int GetOverallIfCond (void)
+/* Get the overall condition based on all conditions on the stack. */
{
- int Done;
- do {
- IfDesc* D = GetCurrentIf();
- if (D == 0) {
- Error (" Unexpected .ENDIF");
- Done = 1;
- } else {
- Done = (D->Flags & ifNeedTerm) != 0;
- --IfCount;
- }
- } while (!Done);
+ /* Since the last entry contains the overall condition of the parent, we
+ * must check it in combination of the current condition. If there is no
+ * last entry, the overall condition is true.
+ */
+ return (IfCount == 0) ||
+ ((IfStack[IfCount-1].Flags & (ifCond | ifParentCond)) == (ifCond | ifParentCond));
}
-static int GetCurrentIfCond (void)
-/* Return the current condition based on all conditions on the stack */
+static void CalcOverallIfCond (void)
+/* Caclulate the overall condition based on all conditions on the stack. */
{
- unsigned Count;
- for (Count = 0; Count < IfCount; ++Count) {
- if ((IfStack[Count].Flags & ifCond) == 0) {
- return 0;
- }
- }
- return 1;
+ IfCond = GetOverallIfCond ();
}
/* Set the .IF condition */
{
if (C) {
- ID->Flags |= ifCond;
+ ID->Flags |= ifCond;
} else {
- ID->Flags &= ~ifCond;
+ ID->Flags &= ~ifCond;
}
}
-static void InvertIfCond (IfDesc* ID)
-/* Invert the current condition */
+static void ElseClause (IfDesc* ID, const char* Directive)
+/* Enter an .ELSE clause */
{
+ /* Check if we have an open .IF - otherwise .ELSE is not allowed */
+ if (ID == 0) {
+ Error ("Unexpected %s", Directive);
+ return;
+ }
+
+ /* Check for a duplicate else, then remember that we had one */
+ if (ID->Flags & ifElse) {
+ /* We already had a .ELSE ! */
+ Error ("Duplicate .ELSE");
+ }
+ ID->Flags |= ifElse;
+
+ /* Condition is inverted now */
ID->Flags ^= ifCond;
}
-static int GetElse (const IfDesc* ID)
-/* Return true if we had a .ELSE */
+static IfDesc* AllocIf (const char* Directive, int NeedTerm)
+/* Alloc a new element from the .IF stack */
{
- return (ID->Flags & ifElse) != 0;
+ IfDesc* ID;
+
+ /* Check for stack overflow */
+ if (IfCount >= MAX_IFS) {
+ Fatal ("Too many nested .IFs");
+ }
+
+ /* Get the next element */
+ ID = &IfStack[IfCount];
+
+ /* Initialize elements */
+ ID->Flags = NeedTerm? ifNeedTerm : ifNone;
+ if (GetOverallIfCond ()) {
+ /* The parents .IF condition is true */
+ ID->Flags |= ifParentCond;
+ }
+ ID->LineInfos = EmptyCollection;
+ GetFullLineInfo (&ID->LineInfos);
+ ID->Name = Directive;
+
+ /* One more slot allocated */
+ ++IfCount;
+
+ /* Return the result */
+ return ID;
}
-static void SetElse (IfDesc* ID, int E)
-/* Set the .ELSE flag */
+static void FreeIf (void)
+/* Free all .IF descriptors until we reach one with the NeedTerm bit set */
{
- if (E) {
- ID->Flags |= ifElse;
- } else {
- ID->Flags &= ~ifElse;
- }
+ int Done;
+ do {
+ IfDesc* ID = GetCurrentIf();
+ if (ID == 0) {
+ Error (" Unexpected .ENDIF");
+ Done = 1;
+ } else {
+ Done = (ID->Flags & ifNeedTerm) != 0;
+ ReleaseFullLineInfo (&ID->LineInfos);
+ DoneCollection (&ID->LineInfos);
+ --IfCount;
+ }
+ } while (!Done);
}
/*****************************************************************************/
-/* Code */
+/* Code */
/*****************************************************************************/
{
IfDesc* D;
- int IfCond = GetCurrentIfCond ();
do {
- switch (CurTok.Tok) {
-
- case TOK_ELSE:
- D = GetCurrentIf ();
- if (D == 0) {
- Error ("Unexpected .ELSE");
- } else if (GetElse(D)) {
- /* We already had a .ELSE ! */
- Error ("Duplicate .ELSE");
- } else {
- /* Allow an .ELSE */
- InvertIfCond (D);
- SetElse (D, 1);
- GetFullLineInfo (&D->LineInfos, 0);
- D->Name = ".ELSE";
- IfCond = GetCurrentIfCond ();
- }
- NextTok ();
- ExpectSep ();
- break;
-
- case TOK_ELSEIF:
- D = GetCurrentIf ();
- if (D == 0) {
- Error ("Unexpected .ELSEIF");
- } else if (GetElse(D)) {
- /* We already had a .ELSE */
- Error ("Duplicate .ELSE");
- } else {
- /* Handle as if there was an .ELSE first */
- InvertIfCond (D);
- SetElse (D, 1);
-
- /* Allocate and prepare a new descriptor */
- D = AllocIf (".ELSEIF", 0);
- NextTok ();
-
- /* Ignore the new condition if we are inside a false .ELSE
- * branch. This way we won't get any errors about undefined
- * symbols or similar...
- */
- if (IfCond == 0) {
- SetIfCond (D, ConstExpression ());
- ExpectSep ();
- }
-
- /* Get the new overall condition */
- IfCond = GetCurrentIfCond ();
- }
- break;
-
- case TOK_ENDIF:
- /* We're done with this .IF.. - remove the descriptor(s) */
- FreeIf ();
-
- /* Be sure not to read the next token until the .IF stack
- * has been cleanup up, since we may be at end of file.
- */
- NextTok ();
- ExpectSep ();
-
- /* Get the new overall condition */
- IfCond = GetCurrentIfCond ();
- break;
-
- case TOK_IF:
- D = AllocIf (".IF", 1);
- NextTok ();
- if (IfCond) {
- SetIfCond (D, ConstExpression ());
- ExpectSep ();
- }
- IfCond = GetCurrentIfCond ();
- break;
-
- case TOK_IFBLANK:
- D = AllocIf (".IFBLANK", 1);
- NextTok ();
- if (IfCond) {
+ switch (CurTok.Tok) {
+
+ case TOK_ELSE:
+ D = GetCurrentIf ();
+
+ /* Allow an .ELSE */
+ ElseClause (D, ".ELSE");
+
+ /* Remember the data for the .ELSE */
+ if (D) {
+ ReleaseFullLineInfo (&D->LineInfos);
+ GetFullLineInfo (&D->LineInfos);
+ D->Name = ".ELSE";
+ }
+
+ /* Calculate the new overall condition */
+ CalcOverallIfCond ();
+
+ /* Skip .ELSE */
+ NextTok ();
+ ExpectSep ();
+ break;
+
+ case TOK_ELSEIF:
+ D = GetCurrentIf ();
+ /* Handle as if there was an .ELSE first */
+ ElseClause (D, ".ELSEIF");
+
+ /* Calculate the new overall if condition */
+ CalcOverallIfCond ();
+
+ /* Allocate and prepare a new descriptor */
+ D = AllocIf (".ELSEIF", 0);
+ NextTok ();
+
+ /* Ignore the new condition if we are inside a false .ELSE
+ * branch. This way we won't get any errors about undefined
+ * symbols or similar...
+ */
+ if (IfCond) {
+ SetIfCond (D, ConstExpression ());
+ ExpectSep ();
+ }
+
+ /* Get the new overall condition */
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_ENDIF:
+ /* We're done with this .IF.. - remove the descriptor(s) */
+ FreeIf ();
+
+ /* Be sure not to read the next token until the .IF stack
+ * has been cleanup up, since we may be at end of file.
+ */
+ NextTok ();
+ ExpectSep ();
+
+ /* Get the new overall condition */
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IF:
+ D = AllocIf (".IF", 1);
+ NextTok ();
+ if (IfCond) {
+ SetIfCond (D, ConstExpression ());
+ ExpectSep ();
+ }
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFBLANK:
+ D = AllocIf (".IFBLANK", 1);
+ NextTok ();
+ if (IfCond) {
if (TokIsSep (CurTok.Tok)) {
SetIfCond (D, 1);
} else {
- SetIfCond (D, 0);
+ SetIfCond (D, 0);
SkipUntilSep ();
}
- }
- IfCond = GetCurrentIfCond ();
- break;
-
- case TOK_IFCONST:
- D = AllocIf (".IFCONST", 1);
- NextTok ();
- if (IfCond) {
- ExprNode* Expr = Expression();
- SetIfCond (D, IsConstExpr (Expr, 0));
- FreeExpr (Expr);
- ExpectSep ();
- }
- IfCond = GetCurrentIfCond ();
- break;
-
- case TOK_IFDEF:
- D = AllocIf (".IFDEF", 1);
- NextTok ();
- if (IfCond) {
- SymEntry* Sym = ParseAnySymName (SYM_FIND_EXISTING);
- SetIfCond (D, Sym != 0 && SymIsDef (Sym));
- }
- IfCond = GetCurrentIfCond ();
- break;
-
- case TOK_IFNBLANK:
- D = AllocIf (".IFNBLANK", 1);
- NextTok ();
- if (IfCond) {
+ }
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFCONST:
+ D = AllocIf (".IFCONST", 1);
+ NextTok ();
+ if (IfCond) {
+ ExprNode* Expr = Expression();
+ SetIfCond (D, IsConstExpr (Expr, 0));
+ FreeExpr (Expr);
+ ExpectSep ();
+ }
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFDEF:
+ D = AllocIf (".IFDEF", 1);
+ NextTok ();
+ if (IfCond) {
+ SymEntry* Sym = ParseAnySymName (SYM_FIND_EXISTING);
+ SetIfCond (D, Sym != 0 && SymIsDef (Sym));
+ }
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFNBLANK:
+ D = AllocIf (".IFNBLANK", 1);
+ NextTok ();
+ if (IfCond) {
if (TokIsSep (CurTok.Tok)) {
SetIfCond (D, 0);
} else {
- SetIfCond (D, 1);
+ SetIfCond (D, 1);
SkipUntilSep ();
}
- }
- IfCond = GetCurrentIfCond ();
- break;
-
- case TOK_IFNCONST:
- D = AllocIf (".IFNCONST", 1);
- NextTok ();
- if (IfCond) {
- ExprNode* Expr = Expression();
- SetIfCond (D, !IsConstExpr (Expr, 0));
- FreeExpr (Expr);
- ExpectSep ();
- }
- IfCond = GetCurrentIfCond ();
- break;
-
- case TOK_IFNDEF:
- D = AllocIf (".IFNDEF", 1);
- NextTok ();
- if (IfCond) {
- SymEntry* Sym = ParseAnySymName (SYM_FIND_EXISTING);
- SetIfCond (D, Sym == 0 || !SymIsDef (Sym));
- ExpectSep ();
- }
- IfCond = GetCurrentIfCond ();
- break;
-
- case TOK_IFNREF:
- D = AllocIf (".IFNREF", 1);
- NextTok ();
- if (IfCond) {
- SymEntry* Sym = ParseAnySymName (SYM_FIND_EXISTING);
- SetIfCond (D, Sym == 0 || !SymIsRef (Sym));
- ExpectSep ();
- }
- IfCond = GetCurrentIfCond ();
- break;
-
- case TOK_IFP02:
- D = AllocIf (".IFP02", 1);
- NextTok ();
- if (IfCond) {
- SetIfCond (D, GetCPU() == CPU_6502);
- }
- IfCond = GetCurrentIfCond ();
- ExpectSep ();
- break;
-
- case TOK_IFP816:
- D = AllocIf (".IFP816", 1);
- NextTok ();
- if (IfCond) {
- SetIfCond (D, GetCPU() == CPU_65816);
- }
- IfCond = GetCurrentIfCond ();
- ExpectSep ();
- break;
-
- case TOK_IFPC02:
- D = AllocIf (".IFPC02", 1);
- NextTok ();
- if (IfCond) {
- SetIfCond (D, GetCPU() == CPU_65C02);
- }
- IfCond = GetCurrentIfCond ();
- ExpectSep ();
- break;
-
- case TOK_IFPSC02:
- D = AllocIf (".IFPSC02", 1);
- NextTok ();
- if (IfCond) {
- SetIfCond (D, GetCPU() == CPU_65SC02);
- }
- IfCond = GetCurrentIfCond ();
- ExpectSep ();
- break;
-
- case TOK_IFREF:
- D = AllocIf (".IFREF", 1);
- NextTok ();
- if (IfCond) {
- SymEntry* Sym = ParseAnySymName (SYM_FIND_EXISTING);
- SetIfCond (D, Sym != 0 && SymIsRef (Sym));
- ExpectSep ();
- }
- IfCond = GetCurrentIfCond ();
- break;
-
- default:
- /* Skip tokens */
- NextTok ();
-
- }
+ }
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFNCONST:
+ D = AllocIf (".IFNCONST", 1);
+ NextTok ();
+ if (IfCond) {
+ ExprNode* Expr = Expression();
+ SetIfCond (D, !IsConstExpr (Expr, 0));
+ FreeExpr (Expr);
+ ExpectSep ();
+ }
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFNDEF:
+ D = AllocIf (".IFNDEF", 1);
+ NextTok ();
+ if (IfCond) {
+ SymEntry* Sym = ParseAnySymName (SYM_FIND_EXISTING);
+ SetIfCond (D, Sym == 0 || !SymIsDef (Sym));
+ ExpectSep ();
+ }
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFNREF:
+ D = AllocIf (".IFNREF", 1);
+ NextTok ();
+ if (IfCond) {
+ SymEntry* Sym = ParseAnySymName (SYM_FIND_EXISTING);
+ SetIfCond (D, Sym == 0 || !SymIsRef (Sym));
+ ExpectSep ();
+ }
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFP02:
+ D = AllocIf (".IFP02", 1);
+ NextTok ();
+ if (IfCond) {
+ SetIfCond (D, GetCPU() == CPU_6502);
+ }
+ ExpectSep ();
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFP816:
+ D = AllocIf (".IFP816", 1);
+ NextTok ();
+ if (IfCond) {
+ SetIfCond (D, GetCPU() == CPU_65816);
+ }
+ ExpectSep ();
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFPC02:
+ D = AllocIf (".IFPC02", 1);
+ NextTok ();
+ if (IfCond) {
+ SetIfCond (D, GetCPU() == CPU_65C02);
+ }
+ ExpectSep ();
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFPSC02:
+ D = AllocIf (".IFPSC02", 1);
+ NextTok ();
+ if (IfCond) {
+ SetIfCond (D, GetCPU() == CPU_65SC02);
+ }
+ ExpectSep ();
+ CalcOverallIfCond ();
+ break;
+
+ case TOK_IFREF:
+ D = AllocIf (".IFREF", 1);
+ NextTok ();
+ if (IfCond) {
+ SymEntry* Sym = ParseAnySymName (SYM_FIND_EXISTING);
+ SetIfCond (D, Sym != 0 && SymIsRef (Sym));
+ ExpectSep ();
+ }
+ CalcOverallIfCond ();
+ break;
+
+ default:
+ /* Skip tokens */
+ NextTok ();
+
+ }
} while (IfCond == 0 && CurTok.Tok != TOK_EOF);
}
const LineInfo* LI;
while (1) {
- /* Get the current file number and check if the topmost entry on the
- * .IF stack was inserted with this file number
- */
- IfDesc* D = GetCurrentIf ();
- if (D == 0) {
- /* There are no open .IFs */
- break;
- }
+ /* Get the current file number and check if the topmost entry on the
+ * .IF stack was inserted with this file number
+ */
+ IfDesc* D = GetCurrentIf ();
+ if (D == 0) {
+ /* There are no open .IFs */
+ break;
+ }
LI = CollConstAt (&D->LineInfos, 0);
- if (LI->Pos.Name != CurTok.Pos.Name) {
- /* The .if is from another file, bail out */
- break;
- }
-
- /* Start of .if is in the file we're about to leave */
- LIError (&D->LineInfos, "Conditional assembly branch was never closed");
- FreeIf ();
+ if (GetSourcePos (LI)->Name != CurTok.Pos.Name) {
+ /* The .if is from another file, bail out */
+ break;
+ }
+
+ /* Start of .if is in the file we're about to leave */
+ LIError (&D->LineInfos, "Conditional assembly branch was never closed");
+ FreeIf ();
}
+
+ /* Calculate the new overall .IF condition */
+ CalcOverallIfCond ();
}
/* Cleanup the .IF stack, remove anything above the given stack pointer */
{
while (IfCount > SP) {
- FreeIf ();
+ FreeIf ();
}
-}
-
-
-
+ /* Calculate the new overall .IF condition */
+ CalcOverallIfCond ();
+}