]> git.sur5r.net Git - cc65/blobdiff - src/cc65/codeent.c
Fixed a bug
[cc65] / src / cc65 / codeent.c
index 342c0909c60ac0e6dd54a3f5a4e57b3893c39723..7cca0bd788863c11b41a2b60a53314736de64d02 100644 (file)
@@ -40,6 +40,7 @@
 #include "chartype.h"
 #include "check.h"
 #include "xmalloc.h"
+#include "xsprintf.h"
 
 /* cc65 */
 #include "codeinfo.h"
@@ -136,6 +137,8 @@ static int NumArg (const char* Arg, unsigned long* Num)
 static void SetUseChgInfo (CodeEntry* E, const OPCDesc* D)
 /* Set the Use and Chg in E */
 {
+    const ZPInfo* Info;
+
     /* If this is a subroutine call, or a jump to an external function,
      * lookup the information about this function and use it. The jump itself
      * does not change any registers, so we don't need to use the data from D.
@@ -145,10 +148,53 @@ static void SetUseChgInfo (CodeEntry* E, const OPCDesc* D)
        GetFuncInfo (E->Arg, &E->Use, &E->Chg);
     } else {
        /* Some other instruction. Use the values from the opcode description
-        * plus addressing mode info
+        * plus addressing mode info.
         */
        E->Use = D->Use | GetAMUseInfo (E->AM);
        E->Chg = D->Chg;
+
+       /* Check for special zero page registers used */
+       switch (E->AM) {
+
+           case AM65_ZP:
+           case AM65_ABS:
+           /* Be conservative: */
+           case AM65_ZPX:
+           case AM65_ABSX:
+           case AM65_ABSY:
+               Info = GetZPInfo (E->Arg);
+               if (Info && Info->ByteUse != REG_NONE) {
+                   if (E->OPC == OP65_ASL || E->OPC == OP65_DEC ||
+                       E->OPC == OP65_INC || E->OPC == OP65_LSR ||
+                       E->OPC == OP65_ROL || E->OPC == OP65_ROR ||
+                       E->OPC == OP65_TRB || E->OPC == OP65_TSB) {
+                       /* The zp loc is both, input and output */
+                       E->Chg |= Info->ByteUse;
+                       E->Use |= Info->ByteUse;
+                   } else if ((E->Info & OF_STORE) != 0) {
+                       /* Just output */
+                       E->Chg |= Info->ByteUse;
+                   } else {
+                       /* Input only */
+                       E->Use |= Info->ByteUse;
+                   }
+               }
+               break;
+
+           case AM65_ZPX_IND:
+           case AM65_ZP_INDY:
+           case AM65_ZP_IND:
+               Info = GetZPInfo (E->Arg);
+               if (Info && Info->ByteUse != REG_NONE) {
+                   /* These addressing modes will never change the zp loc */
+                   E->Use |= Info->WordUse;
+               }
+               break;
+
+           default:
+               /* Keep gcc silent */
+               break;
+       }
     }
 }
 
@@ -265,14 +311,35 @@ void CE_MoveLabel (CodeLabel* L, CodeEntry* E)
 
 
 
-void CE_SetArg (CodeEntry* E, const char* Arg)
-/* Set a new argument for the given code entry. An old string is deleted. */
+void CE_SetNumArg (CodeEntry* E, long Num)
+/* Set a new numeric argument for the given code entry that must already
+ * have a numeric argument.
+ */
 {
+    char Buf[16];
+
+    /* Check that the entry has a numerical argument */
+    CHECK (E->Flags & CEF_NUMARG);
+
+    /* Make the new argument string */
+    if (E->Size == 2) {
+       Num &= 0xFF;
+       xsprintf (Buf, sizeof (Buf), "$%02X", (unsigned) Num);
+    } else if (E->Size == 3) {
+       Num &= 0xFFFF;
+       xsprintf (Buf, sizeof (Buf), "$%04X", (unsigned) Num);
+    } else {
+       Internal ("Invalid instruction size in CE_SetNumArg");
+    }
+
     /* Free the old argument */
     FreeArg (E->Arg);
 
     /* Assign the new one */
-    E->Arg = GetArgCopy (Arg);
+    E->Arg = GetArgCopy (Buf);
+
+    /* Use the new numerical value */
+    E->Num = Num;
 }
 
 
@@ -347,6 +414,12 @@ void CE_GenRegInfo (CodeEntry* E, RegContents* InputRegs)
        case OP65_ASL:
            if (E->AM == AM65_ACC && In->RegA >= 0) {
                Out->RegA = (In->RegA << 1) & 0xFF;
+           } else if (E->AM == AM65_ZP) {
+               if ((E->Chg & REG_SREG_LO) != 0 && In->SRegLo >= 0) {
+                   Out->SRegLo = (In->SRegLo << 1) & 0xFF;
+               } else if ((E->Chg & REG_SREG_HI) != 0 && In->SRegHi >= 0) {
+                   Out->SRegHi = (In->SRegHi << 1) & 0xFF;
+               }
            }
            break;
 
@@ -406,25 +479,31 @@ void CE_GenRegInfo (CodeEntry* E, RegContents* InputRegs)
 
        case OP65_DEA:
            if (In->RegA >= 0) {
-               Out->RegA = In->RegA - 1;
+               Out->RegA = (In->RegA - 1) & 0xFF;
            }
            break;
 
        case OP65_DEC:
            if (E->AM == AM65_ACC && In->RegA >= 0) {
-               Out->RegA = In->RegA - 1;
+               Out->RegA = (In->RegA - 1) & 0xFF;
+           } else if (E->AM == AM65_ZP) {
+               if ((E->Chg & REG_SREG_LO) != 0 && In->SRegLo >= 0) {
+                   Out->SRegLo = (In->SRegLo - 1) & 0xFF;
+               } else if ((E->Chg & REG_SREG_HI) != 0 && In->SRegHi >= 0) {
+                   Out->SRegHi = (In->SRegHi - 1) & 0xFF;
+               }
            }
            break;
 
        case OP65_DEX:
                    if (In->RegX >= 0) {
-               Out->RegX = In->RegX - 1;
+               Out->RegX = (In->RegX - 1) & 0xFF;
            }
            break;
 
        case OP65_DEY:
                    if (In->RegY >= 0) {
-               Out->RegY = In->RegY - 1;
+               Out->RegY = (In->RegY - 1) & 0xFF;
            }
            break;
 
@@ -440,25 +519,31 @@ void CE_GenRegInfo (CodeEntry* E, RegContents* InputRegs)
 
        case OP65_INA:
            if (In->RegA >= 0) {
-               Out->RegA = In->RegA + 1;
+               Out->RegA = (In->RegA + 1) & 0xFF;
            }
            break;
 
        case OP65_INC:
            if (E->AM == AM65_ACC && In->RegA >= 0) {
-               Out->RegA = In->RegA + 1;
+               Out->RegA = (In->RegA + 1) & 0xFF;
+           } else if (E->AM == AM65_ZP) {
+               if ((E->Chg & REG_SREG_LO) != 0 && In->SRegLo >= 0) {
+                   Out->SRegLo = (In->SRegLo + 1) & 0xFF;
+               } else if ((E->Chg & REG_SREG_HI) != 0 && In->SRegHi >= 0) {
+                   Out->SRegHi = (In->SRegHi + 1) & 0xFF;
+               }
            }
            break;
 
        case OP65_INX:
            if (In->RegX >= 0) {
-               Out->RegX = In->RegX + 1;
+               Out->RegX = (In->RegX + 1) & 0xFF;
            }
            break;
 
        case OP65_INY:
            if (In->RegY >= 0) {
-               Out->RegY = In->RegY + 1;
+               Out->RegY = (In->RegY + 1) & 0xFF;
            }
            break;
 
@@ -495,6 +580,12 @@ void CE_GenRegInfo (CodeEntry* E, RegContents* InputRegs)
            if (Chg & REG_Y) {
                Out->RegY = -1;
            }
+            if (Chg & REG_SREG_LO) {
+               Out->SRegLo = -1;
+           }
+           if (Chg & REG_SREG_HI) {
+               Out->SRegHi = -1;
+           }
            break;
 
        case OP65_JVC:
@@ -505,25 +596,49 @@ void CE_GenRegInfo (CodeEntry* E, RegContents* InputRegs)
 
        case OP65_LDA:
            if (CE_KnownImm (E)) {
-               Out->RegA = (unsigned char) E->Num;
+               Out->RegA = (unsigned char) E->Num;
+           } else if (E->AM == AM65_ZP) {
+               if (E->Use & REG_SREG_LO) {
+                   Out->RegA = In->SRegLo;
+               } else if (E->Use & REG_SREG_HI) {
+                   Out->RegA = In->SRegHi;
+               } else {
+                   Out->RegA = -1;
+               }
            } else {
-               /* A is now unknown */
-               Out->RegA = -1;
+               /* A is now unknown */
+               Out->RegA = -1;
            }
            break;
 
        case OP65_LDX:
            if (CE_KnownImm (E)) {
-               Out->RegX = (unsigned char) E->Num;
+               Out->RegX = (unsigned char) E->Num;
+           } else if (E->AM == AM65_ZP) {
+               if (E->Use & REG_SREG_LO) {
+                   Out->RegX = In->SRegLo;
+               } else if (E->Use & REG_SREG_HI) {
+                   Out->RegX = In->SRegHi;
+               } else {
+                   Out->RegX = -1;
+               }
            } else {
-               /* X is now unknown */
-               Out->RegX = -1;
+               /* X is now unknown */
+               Out->RegX = -1;
            }
            break;
 
        case OP65_LDY:
            if (CE_KnownImm (E)) {
-               Out->RegY = (unsigned char) E->Num;
+               Out->RegY = (unsigned char) E->Num;
+           } else if (E->AM == AM65_ZP) {
+               if (E->Use & REG_SREG_LO) {
+                   Out->RegY = In->SRegLo;
+               } else if (E->Use & REG_SREG_HI) {
+                   Out->RegY = In->SRegHi;
+               } else {
+                   Out->RegY = -1;
+               }
            } else {
                /* Y is now unknown */
                Out->RegY = -1;
@@ -533,6 +648,12 @@ void CE_GenRegInfo (CodeEntry* E, RegContents* InputRegs)
        case OP65_LSR:
            if (E->AM == AM65_ACC && In->RegA >= 0) {
                Out->RegA = (In->RegA >> 1) & 0xFF;
+           } else if (E->AM == AM65_ZP) {
+               if ((E->Chg & REG_SREG_LO) != 0 && In->SRegLo >= 0) {
+                   Out->SRegLo = (In->SRegLo >> 1) & 0xFF;
+               } else if (E->Chg & REG_SREG_HI) {
+                   Out->SRegHi = (In->SRegHi >> 1) & 0xFF;
+               }
            }
            break;
 
@@ -569,49 +690,86 @@ void CE_GenRegInfo (CodeEntry* E, RegContents* InputRegs)
        case OP65_PLP:
            break;
 
-       case OP65_PLX:
-           Out->RegX = -1;
-           break;
+       case OP65_PLX:
+           Out->RegX = -1;
+           break;
 
-       case OP65_PLY:
-           Out->RegY = -1;
-           break;
+       case OP65_PLY:
+           Out->RegY = -1;
+           break;
 
-       case OP65_ROL:
-           Out->RegA = -1;
-           break;
+       case OP65_ROL:
+           if (E->AM == AM65_ACC) {
+               Out->RegA = -1;
+           } else if (E->AM == AM65_ZP) {
+               if (E->Chg & REG_SREG_LO) {
+                   Out->SRegLo = -1;
+               } else if (E->Chg & REG_SREG_HI) {
+                   Out->SRegHi = -1;
+               }
+           }
+           break;
 
-       case OP65_ROR:
-           Out->RegA = -1;
-           break;
+       case OP65_ROR:
+           if (E->AM == AM65_ACC) {
+               Out->RegA = -1;
+           } else if (E->AM == AM65_ZP) {
+               if (E->Chg & REG_SREG_LO) {
+                   Out->SRegLo = -1;
+               } else if (E->Chg & REG_SREG_HI) {
+                   Out->SRegHi = -1;
+               }
+           }
+           break;
 
-       case OP65_RTI:
-           break;
+       case OP65_RTI:
+           break;
 
-       case OP65_RTS:
-           break;
+       case OP65_RTS:
+           break;
 
-       case OP65_SBC:
-           /* We don't know the value of the carry bit */
-           Out->RegA = -1;
-           break;
+       case OP65_SBC:
+           /* We don't know the value of the carry bit */
+           Out->RegA = -1;
+           break;
 
-       case OP65_SEC:
-           break;
+       case OP65_SEC:
+           break;
 
-       case OP65_SED:
-           break;
+       case OP65_SED:
+           break;
 
-       case OP65_SEI:
-           break;
+       case OP65_SEI:
+           break;
 
-       case OP65_STA:
+       case OP65_STA:
+           if (E->AM == AM65_ZP) {
+               if (E->Chg & REG_SREG_LO) {
+                   Out->SRegLo = In->RegA;
+               } else if (E->Chg & REG_SREG_HI) {
+                   Out->SRegHi = In->RegA;
+               }
+           }
            break;
 
        case OP65_STX:
+           if (E->AM == AM65_ZP) {
+               if (E->Chg & REG_SREG_LO) {
+                   Out->SRegLo = In->RegX;
+               } else if (E->Chg & REG_SREG_HI) {
+                   Out->SRegHi = In->RegX;
+               }
+           }
            break;
 
        case OP65_STY:
+           if (E->AM == AM65_ZP) {
+               if (E->Chg & REG_SREG_LO) {
+                   Out->SRegLo = In->RegY;
+               } else if (E->Chg & REG_SREG_HI) {
+                   Out->SRegHi = In->RegY;
+               }
+           }
            break;
 
        case OP65_TAX:
@@ -655,6 +813,26 @@ void CE_GenRegInfo (CodeEntry* E, RegContents* InputRegs)
 
 
 
+static char* RegInfoDesc (unsigned U, char* Buf)
+/* Return a string containing register info */
+{
+    Buf[0] = '\0';
+
+    strcat (Buf, U & REG_SREG_HI? "H" : "_");
+    strcat (Buf, U & REG_SREG_LO? "L" : "_");
+    strcat (Buf, U & REG_A?       "A" : "_");
+    strcat (Buf, U & REG_X?       "X" : "_");
+    strcat (Buf, U & REG_Y?       "Y" : "_");
+    strcat (Buf, U & REG_TMP1?    "T1" : "__");
+    strcat (Buf, U & REG_PTR1?    "1" : "_");
+    strcat (Buf, U & REG_PTR2?    "2" : "_");
+    strcat (Buf, U & REG_SAVE?    "V"  : "_");
+
+    return Buf;
+}
+
+
+
 void CE_Output (const CodeEntry* E, FILE* F)
 /* Output the code entry to a file */
 {
@@ -699,7 +877,7 @@ void CE_Output (const CodeEntry* E, FILE* F)
        case AM65_ABS:
            /* zeropage and absolute */
            Chars += fprintf (F, "%*s%s", 9-Chars, "", E->Arg);
-           break;
+                   break;
 
        case AM65_ZPX:
        case AM65_ABSX:
@@ -740,16 +918,14 @@ void CE_Output (const CodeEntry* E, FILE* F)
 
     /* Print usage info if requested by the debugging flag */
     if (Debug) {
-       fprintf (F,
-                "%*s; USE: %c%c%c CHG: %c%c%c SIZE: %u\n",
-                30-Chars, "",
-                (E->Use & REG_A)? 'A' : '_',
-                (E->Use & REG_X)? 'X' : '_',
-                (E->Use & REG_Y)? 'Y' : '_',
-                (E->Chg & REG_A)? 'A' : '_',
-                (E->Chg & REG_X)? 'X' : '_',
-                (E->Chg & REG_Y)? 'Y' : '_',
-                E->Size);
+       char Use [128];
+       char Chg [128];
+               fprintf (F,
+                        "%*s; USE: %-20s CHG: %-20s SIZE: %u\n",
+                        30-Chars, "",
+                RegInfoDesc (E->Use, Use),
+                RegInfoDesc (E->Chg, Chg),
+                E->Size);
     } else {
        /* Terminate the line */
        fprintf (F, "\n");