]> git.sur5r.net Git - cc65/blob - src/cc65/codeseg.c
Working
[cc65] / src / cc65 / codeseg.c
1 /*****************************************************************************/
2 /*                                                                           */
3 /*                                 codeseg.c                                 */
4 /*                                                                           */
5 /*                          Code segment structure                           */
6 /*                                                                           */
7 /*                                                                           */
8 /*                                                                           */
9 /* (C) 2001      Ullrich von Bassewitz                                       */
10 /*               Wacholderweg 14                                             */
11 /*               D-70597 Stuttgart                                           */
12 /* EMail:        uz@cc65.org                                                 */
13 /*                                                                           */
14 /*                                                                           */
15 /* This software is provided 'as-is', without any expressed or implied       */
16 /* warranty.  In no event will the authors be held liable for any damages    */
17 /* arising from the use of this software.                                    */
18 /*                                                                           */
19 /* Permission is granted to anyone to use this software for any purpose,     */
20 /* including commercial applications, and to alter it and redistribute it    */
21 /* freely, subject to the following restrictions:                            */
22 /*                                                                           */
23 /* 1. The origin of this software must not be misrepresented; you must not   */
24 /*    claim that you wrote the original software. If you use this software   */
25 /*    in a product, an acknowledgment in the product documentation would be  */
26 /*    appreciated but is not required.                                       */
27 /* 2. Altered source versions must be plainly marked as such, and must not   */
28 /*    be misrepresented as being the original software.                      */
29 /* 3. This notice may not be removed or altered from any source              */
30 /*    distribution.                                                          */
31 /*                                                                           */
32 /*****************************************************************************/
33
34
35
36 #include <string.h>
37 #include <ctype.h>
38
39 /* common */
40 #include "chartype.h"
41 #include "check.h"
42 #include "hashstr.h"
43 #include "strutil.h"
44 #include "xmalloc.h"
45 #include "xsprintf.h"
46
47 /* cc65 */
48 #include "asmlabel.h"
49 #include "codeent.h"
50 #include "codeinfo.h"
51 #include "datatype.h"
52 #include "error.h"
53 #include "symentry.h"
54 #include "codeseg.h"
55
56
57
58 /*****************************************************************************/
59 /*                             Helper functions                              */
60 /*****************************************************************************/
61
62
63
64 static void MoveLabelsToPool (CodeSeg* S, CodeEntry* E)
65 /* Move the labels of the code entry E to the label pool of the code segment */
66 {
67     unsigned LabelCount = GetCodeLabelCount (E);
68     while (LabelCount--) {
69         CodeLabel* L = GetCodeLabel (E, LabelCount);
70         L->Flags &= ~LF_DEF;
71         L->Owner = 0;
72         CollAppend (&S->Labels, L);
73     }
74     CollDeleteAll (&E->Labels);
75 }
76
77
78
79 static CodeLabel* FindCodeLabel (CodeSeg* S, const char* Name, unsigned Hash)
80 /* Find the label with the given name. Return the label or NULL if not found */
81 {
82     /* Get the first hash chain entry */
83     CodeLabel* L = S->LabelHash[Hash];
84
85     /* Search the list */
86     while (L) {
87         if (strcmp (Name, L->Name) == 0) {
88             /* Found */
89             break;
90         }
91         L = L->Next;
92     }
93     return L;
94 }
95
96
97
98 static CodeLabel* NewCodeSegLabel (CodeSeg* S, const char* Name, unsigned Hash)
99 /* Create a new label and insert it into the label hash table */
100 {
101     /* Not found - create a new one */
102     CodeLabel* L = NewCodeLabel (Name, Hash);
103
104     /* Enter the label into the hash table */
105     L->Next = S->LabelHash[L->Hash];
106     S->LabelHash[L->Hash] = L;
107
108     /* Return the new label */
109     return L;
110 }
111
112
113
114 static void RemoveLabelFromHash (CodeSeg* S, CodeLabel* L)
115 /* Remove the given code label from the hash list */
116 {
117     /* Get the first entry in the hash chain */
118     CodeLabel* List = S->LabelHash[L->Hash];
119     CHECK (List != 0);
120
121     /* First, remove the label from the hash chain */
122     if (List == L) {
123         /* First entry in hash chain */
124         S->LabelHash[L->Hash] = L->Next;
125     } else {
126         /* Must search through the chain */
127         while (List->Next != L) {
128             /* If we've reached the end of the chain, something is *really* wrong */
129             CHECK (List->Next != 0);
130             /* Next entry */
131             List = List->Next;
132         }
133         /* The next entry is the one, we have been searching for */
134         List->Next = L->Next;
135     }
136 }
137
138
139
140 /*****************************************************************************/
141 /*                    Functions for parsing instructions                     */
142 /*****************************************************************************/
143
144
145
146 static const char* SkipSpace (const char* S)
147 /* Skip white space and return an updated pointer */
148 {
149     while (IsSpace (*S)) {
150         ++S;
151     }
152     return S;
153 }
154
155
156
157 static const char* ReadToken (const char* L, const char* Term,
158                               char* Buf, unsigned BufSize)
159 /* Read the next token into Buf, return the updated line pointer. The
160  * token is terminated by one of the characters given in term.
161  */
162 {
163     /* Read/copy the token */
164     unsigned I = 0;
165     unsigned ParenCount = 0;
166     while (*L && (ParenCount > 0 || strchr (Term, *L) == 0)) {
167         if (I < BufSize-1) {
168             Buf[I++] = *L;
169         }
170         if (*L == ')') {
171             --ParenCount;
172         } else if (*L == '(') {
173             ++ParenCount;
174         }
175         ++L;
176     }
177
178     /* Terminate the buffer contents */
179     Buf[I] = '\0';
180
181     /* Return the updated line pointer */
182     return L;
183 }
184
185
186
187 static CodeEntry* ParseInsn (CodeSeg* S, const char* L)
188 /* Parse an instruction nnd generate a code entry from it. If the line contains
189  * errors, output an error message and return NULL.
190  * For simplicity, we don't accept the broad range of input a "real" assembler
191  * does. The instruction and the argument are expected to be separated by
192  * white space, for example.
193  */
194 {
195     char                Mnemo[16];
196     const OPCDesc*      OPC;
197     am_t                AM = 0;         /* Initialize to keep gcc silent */
198     char                Arg[64];
199     char                Reg;
200     CodeEntry*          E;
201     CodeLabel*          Label;
202
203     /* Mnemonic */
204     L = ReadToken (L, " \t", Mnemo, sizeof (Mnemo));
205
206     /* Try to find the opcode description for the mnemonic */
207     OPC = FindOpcode (Mnemo);
208
209     /* If we didn't find the opcode, print an error and bail out */
210     if (OPC == 0) {
211         Error ("ASM code error: %s is not a valid mnemonic", Mnemo);
212         return 0;
213     }
214
215     /* Skip separator white space */
216     L = SkipSpace (L);
217
218     /* Get the addressing mode */
219     Arg[0] = '\0';
220     switch (*L) {
221
222         case '\0':
223             /* Implicit */
224             AM = AM_IMP;
225             break;
226
227         case '#':
228             /* Immidiate */
229             StrCopy (Arg, sizeof (Arg), L+1);
230             AM = AM_IMM;
231             break;
232
233         case '(':
234             /* Indirect */
235             L = ReadToken (L+1, ",)", Arg, sizeof (Arg));
236
237             /* Check for errors */
238             if (*L == '\0') {
239                 Error ("ASM code error: syntax error");
240                 return 0;
241             }
242
243             /* Check the different indirect modes */
244             if (*L == ',') {
245                 /* Expect zp x indirect */
246                 L = SkipSpace (L+1);
247                 if (toupper (*L) != 'X') {
248                     Error ("ASM code error: `X' expected");
249                     return 0;
250                 }
251                 L = SkipSpace (L+1);
252                 if (*L != ')') {
253                     Error ("ASM code error: `)' expected");
254                     return 0;
255                 }
256                 L = SkipSpace (L+1);
257                 if (*L != '\0') {
258                     Error ("ASM code error: syntax error");
259                     return 0;
260                 }
261                 AM = AM_ZPX_IND;
262             } else if (*L == ')') {
263                 /* zp indirect or zp indirect, y */
264                 L = SkipSpace (L+1);
265                 if (*L == ',') {
266                     L = SkipSpace (L+1);
267                     if (toupper (*L) != 'Y') {
268                         Error ("ASM code error: `Y' expected");
269                         return 0;
270                     }
271                     L = SkipSpace (L+1);
272                     if (*L != '\0') {
273                         Error ("ASM code error: syntax error");
274                         return 0;
275                     }
276                     AM = AM_ZP_INDY;
277                 } else if (*L == '\0') {
278                     AM = AM_ZP_IND;
279                 } else {
280                     Error ("ASM code error: syntax error");
281                     return 0;
282                 }
283             }
284             break;
285
286         case 'a':
287         case 'A':
288             /* Accumulator? */
289             if (L[1] == '\0') {
290                 AM = AM_ACC;
291                 break;
292             }
293             /* FALLTHROUGH */
294
295         default:
296             /* Absolute, maybe indexed */
297             L = ReadToken (L, ",", Arg, sizeof (Arg));
298             if (*L == '\0') {
299                 /* Absolute, zeropage or branch */
300                 if ((OPC->Info & OF_BRA) != 0) {
301                     /* Branch */
302                     AM = AM_BRA;
303                 } else if (IsZPName (Arg)) {
304                     AM = AM_ZP;
305                 } else {
306                     AM = AM_ABS;
307                 }
308             } else if (*L == ',') {
309                 /* Indexed */
310                 L = SkipSpace (L+1);
311                 if (*L == '\0') {
312                     Error ("ASM code error: syntax error");
313                     return 0;
314                 } else {
315                     Reg = toupper (*L);
316                     L = SkipSpace (L+1);
317                     if (Reg == 'X') {
318                         if (IsZPName (Arg)) {
319                             AM = AM_ZPX;
320                         } else {
321                             AM = AM_ABSX;
322                         }
323                     } else if (Reg == 'Y') {
324                         AM = AM_ABSY;
325                     } else {
326                         Error ("ASM code error: syntax error");
327                         return 0;
328                     }
329                     if (*L != '\0') {
330                         Error ("ASM code error: syntax error");
331                         return 0;
332                     }
333                 }
334             }
335             break;
336
337     }
338
339     /* If the instruction is a branch, check for the label and generate it
340      * if it does not exist. Ignore anything but local labels here.
341      */
342     Label = 0;
343     if (AM == AM_BRA && Arg[0] == 'L') {
344
345         /* Generate the hash over the label, then search for the label */
346         unsigned Hash = HashStr (Arg) % CS_LABEL_HASH_SIZE;
347         Label = FindCodeLabel (S, Arg, Hash);
348
349         /* If we don't have the label, it's a forward ref - create it */
350         if (Label == 0) {
351             /* Generate a new label */
352             Label = NewCodeSegLabel (S, Arg, Hash);
353         }
354     }
355
356     /* We do now have the addressing mode in AM. Allocate a new CodeEntry
357      * structure and initialize it.
358      */
359     E = NewCodeEntry (OPC->OPC, AM, Arg, Label);
360
361     /* Return the new code entry */
362     return E;
363 }
364
365
366
367 /*****************************************************************************/
368 /*                                   Code                                    */
369 /*****************************************************************************/
370
371
372
373 CodeSeg* NewCodeSeg (const char* SegName, SymEntry* Func)
374 /* Create a new code segment, initialize and return it */
375 {
376     unsigned I;
377
378     /* Allocate memory */
379     CodeSeg* S = xmalloc (sizeof (CodeSeg));
380
381     /* Initialize the fields */
382     S->SegName  = xstrdup (SegName);
383     S->Func     = Func;
384     InitCollection (&S->Entries);
385     InitCollection (&S->Labels);
386     for (I = 0; I < sizeof(S->LabelHash) / sizeof(S->LabelHash[0]); ++I) {
387         S->LabelHash[I] = 0;
388     }
389
390     /* If we have a function given, get the return type of the function.
391      * Assume ANY return type besides void will use the A and X registers.
392      */
393     if (S->Func && !IsTypeVoid (GetFuncReturn (Func->Type))) {
394         S->ExitRegs = REG_AX;
395     } else {
396         S->ExitRegs = REG_NONE;
397     }
398
399     /* Return the new struct */
400     return S;
401 }
402
403
404
405 void AddCodeEntry (CodeSeg* S, const char* Format, va_list ap)
406 /* Add a line to the given code segment */
407 {
408     const char* L;
409     CodeEntry*  E;
410     char        Token[64];
411
412     /* Format the line */
413     char Buf [256];
414     xvsprintf (Buf, sizeof (Buf), Format, ap);
415
416     /* Skip whitespace */
417     L = SkipSpace (Buf);
418
419     /* Check which type of instruction we have */
420     E = 0;      /* Assume no insn created */
421     switch (*L) {
422
423         case '\0':
424             /* Empty line, just ignore it */
425             break;
426
427         case ';':
428             /* Comment or hint, ignore it for now */
429             break;
430
431         case '.':
432             /* Control instruction */
433             ReadToken (L, " \t", Token, sizeof (Token));
434             Error ("ASM code error: Pseudo instruction `%s' not supported", Token);
435             break;
436
437         default:
438             E = ParseInsn (S, L);
439             break;
440     }
441
442     /* If we have a code entry, transfer the labels and insert it */
443     if (E) {
444
445         /* Transfer the labels if we have any */
446         unsigned I;
447         unsigned LabelCount = CollCount (&S->Labels);
448         for (I = 0; I < LabelCount; ++I) {
449
450             /* Get the label */
451             CodeLabel* L = CollAt (&S->Labels, I);
452
453             /* Attach it to the entry */
454             AttachCodeLabel (E, L);
455         }
456
457         /* Delete the transfered labels */
458         CollDeleteAll (&S->Labels);
459
460         /* Add the entry to the list of code entries in this segment */
461         CollAppend (&S->Entries, E);
462
463     }
464 }
465
466
467
468 void InsertCodeEntry (CodeSeg* S, struct CodeEntry* E, unsigned Index)
469 /* Insert the code entry at the index given. Following code entries will be
470  * moved to slots with higher indices.
471  */
472 {
473     /* Insert the entry into the collection */
474     CollInsert (&S->Entries, E, Index);
475 }
476
477
478
479 void DelCodeEntry (CodeSeg* S, unsigned Index)
480 /* Delete an entry from the code segment. This includes moving any associated
481  * labels, removing references to labels and even removing the referenced labels
482  * if the reference count drops to zero.
483  */
484 {
485     /* Get the code entry for the given index */
486     CodeEntry* E = GetCodeEntry (S, Index);
487
488     /* If the entry has a labels, we have to move this label to the next insn.
489      * If there is no next insn, move the label into the code segement label
490      * pool. The operation is further complicated by the fact that the next
491      * insn may already have a label. In that case change all reference to
492      * this label and delete the label instead of moving it.
493      */
494     unsigned Count = GetCodeLabelCount (E);
495     if (Count > 0) {
496
497         /* The instruction has labels attached. Check if there is a next
498          * instruction.
499          */
500         if (Index == GetCodeEntryCount (S)-1) {
501
502             /* No next instruction, move to the codeseg label pool */
503             MoveLabelsToPool (S, E);
504
505         } else {
506
507             /* There is a next insn, get it */
508             CodeEntry* N = GetCodeEntry (S, Index+1);
509
510             /* Move labels to the next entry */
511             MoveCodeLabels (S, E, N);
512
513         }
514     }
515
516     /* If this insn references a label, remove the reference. And, if the
517      * the reference count for this label drops to zero, remove this label.
518      */
519     if (E->JumpTo) {
520         /* Remove the reference */
521         RemoveCodeLabelRef (S, E);
522     }
523
524     /* Delete the pointer to the insn */
525     CollDelete (&S->Entries, Index);
526
527     /* Delete the instruction itself */
528     FreeCodeEntry (E);
529 }
530
531
532
533 void DelCodeEntries (CodeSeg* S, unsigned Start, unsigned Count)
534 /* Delete a range of code entries. This includes removing references to labels,
535  * labels attached to the entries and so on.
536  */
537 {
538     /* Start deleting the entries from the rear, because this involves less
539      * memory moving.
540      */
541     while (Count--) {
542         DelCodeEntry (S, Start + Count);
543     }
544 }
545
546
547
548 void MoveCodeEntry (CodeSeg* S, unsigned OldPos, unsigned NewPos)
549 /* Move an entry from one position to another. OldPos is the current position
550  * of the entry, NewPos is the new position of the entry.
551  */
552 {
553     /* Get the code entry and remove it from the collection */
554     CodeEntry* E = GetCodeEntry (S, OldPos);
555     CollDelete (&S->Entries, OldPos);
556
557     /* Correct NewPos if needed */
558     if (NewPos >= OldPos) {
559         /* Position has changed with removal */
560         --NewPos;
561     }
562
563     /* Now insert it at the new position */
564     CollInsert (&S->Entries, E, NewPos);
565 }
566
567
568
569 struct CodeEntry* GetNextCodeEntry (CodeSeg* S, unsigned Index)
570 /* Get the code entry following the one with the index Index. If there is no
571  * following code entry, return NULL.
572  */
573 {
574     if (Index >= CollCount (&S->Entries)-1) {
575         /* This is the last entry */
576         return 0;
577     } else {
578         /* Code entries left */
579         return CollAtUnchecked (&S->Entries, Index+1);
580     }
581 }
582
583
584
585 int GetCodeEntries (CodeSeg* S, struct CodeEntry** List,
586                     unsigned Start, unsigned Count)
587 /* Get Count code entries into List starting at index start. Return true if
588  * we got the lines, return false if not enough lines were available.
589  */
590 {
591     /* Check if enough entries are available */
592     if (Start + Count > CollCount (&S->Entries)) {
593         return 0;
594     }
595
596     /* Copy the entries */
597     while (Count--) {
598         *List++ = CollAtUnchecked (&S->Entries, Start++);
599     }
600
601     /* We have the entries */
602     return 1;
603 }
604
605
606
607 unsigned GetCodeEntryIndex (CodeSeg* S, struct CodeEntry* E)
608 /* Return the index of a code entry */
609 {
610     int Index = CollIndex (&S->Entries, E);
611     CHECK (Index >= 0);
612     return Index;
613 }
614
615
616
617 void AddCodeLabel (CodeSeg* S, const char* Name)
618 /* Add a code label for the next instruction to follow */
619 {
620     /* Calculate the hash from the name */
621     unsigned Hash = HashStr (Name) % CS_LABEL_HASH_SIZE;
622
623     /* Try to find the code label if it does already exist */
624     CodeLabel* L = FindCodeLabel (S, Name, Hash);
625
626     /* Did we find it? */
627     if (L) {
628         /* We found it - be sure it does not already have an owner */
629         CHECK (L->Owner == 0);
630     } else {
631         /* Not found - create a new one */
632         L = NewCodeSegLabel (S, Name, Hash);
633     }
634
635     /* Safety. This call is quite costly, but safety is better */
636     if (CollIndex (&S->Labels, L) >= 0) {
637         Internal ("AddCodeLabel: Label `%s' already defined", Name);
638     }
639
640     /* We do now have a valid label. Remember it for later */
641     CollAppend (&S->Labels, L);
642 }
643
644
645
646 CodeLabel* GenCodeLabel (CodeSeg* S, struct CodeEntry* E)
647 /* If the code entry E does already have a label, return it. Otherwise
648  * create a new label, attach it to E and return it.
649  */
650 {
651     CodeLabel* L;
652
653     if (CodeEntryHasLabel (E)) {
654
655         /* Get the label from this entry */
656         L = GetCodeLabel (E, 0);
657
658     } else {
659
660         /* Get a new name */
661         const char* Name = LocalLabelName (GetLocalLabel ());
662
663         /* Generate the hash over the name */
664         unsigned Hash = HashStr (Name) % CS_LABEL_HASH_SIZE;
665
666         /* Create a new label */
667         L = NewCodeSegLabel (S, Name, Hash);
668
669         /* Attach this label to the code entry */
670         AttachCodeLabel (E, L);
671
672     }
673
674     /* Return the label */
675     return L;
676 }
677
678
679
680 void DelCodeLabel (CodeSeg* S, CodeLabel* L)
681 /* Remove references from this label and delete it. */
682 {
683     unsigned Count, I;
684
685     /* First, remove the label from the hash chain */
686     RemoveLabelFromHash (S, L);
687
688     /* Remove references from insns jumping to this label */
689     Count = CollCount (&L->JumpFrom);
690     for (I = 0; I < Count; ++I) {
691         /* Get the insn referencing this label */
692         CodeEntry* E = CollAt (&L->JumpFrom, I);
693         /* Remove the reference */
694         E->JumpTo = 0;
695     }
696     CollDeleteAll (&L->JumpFrom);
697
698     /* Remove the reference to the owning instruction if it has one. The
699      * function may be called for a label without an owner when deleting
700      * unfinished parts of the code. This is unfortunate since it allows
701      * errors to slip through.
702      */
703     if (L->Owner) {
704         CollDeleteItem (&L->Owner->Labels, L);
705     }
706
707     /* All references removed, delete the label itself */
708     FreeCodeLabel (L);
709 }
710
711
712
713 void MergeCodeLabels (CodeSeg* S)
714 /* Merge code labels. That means: For each instruction, remove all labels but
715  * one and adjust references accordingly.
716  */
717 {
718     unsigned I;
719
720     /* Walk over all code entries */
721     unsigned EntryCount = GetCodeEntryCount (S);
722     for (I = 0; I < EntryCount; ++I) {
723
724         CodeLabel* RefLab;
725         unsigned   J;
726
727         /* Get a pointer to the next entry */
728         CodeEntry* E = GetCodeEntry (S, I);
729
730         /* If this entry has zero labels, continue with the next one */
731         unsigned LabelCount = GetCodeLabelCount (E);
732         if (LabelCount == 0) {
733             continue;
734         }
735
736         /* We have at least one label. Use the first one as reference label. */
737         RefLab = GetCodeLabel (E, 0);
738
739         /* Walk through the remaining labels and change references to these
740          * labels to a reference to the one and only label. Delete the labels
741          * that are no longer used. To increase performance, walk backwards
742          * through the list.
743          */
744         for (J = LabelCount-1; J >= 1; --J) {
745
746             /* Get the next label */
747             CodeLabel* L = GetCodeLabel (E, J);
748
749             /* Move all references from this label to the reference label */
750             MoveLabelRefs (L, RefLab);
751
752             /* Remove the label completely. */
753             DelCodeLabel (S, L);
754         }
755
756         /* The reference label is the only remaining label. Check if there
757          * are any references to this label, and delete it if this is not
758          * the case.
759          */
760         if (CollCount (&RefLab->JumpFrom) == 0) {
761             /* Delete the label */
762             DelCodeLabel (S, RefLab);
763         }
764     }
765 }
766
767
768
769 void MoveCodeLabels (CodeSeg* S, struct CodeEntry* Old, struct CodeEntry* New)
770 /* Move all labels from Old to New. The routine will move the labels itself
771  * if New does not have any labels, and move references if there is at least
772  * a label for new. If references are moved, the old label is deleted
773  * afterwards.
774  */
775 {
776     /* Get the number of labels to move */
777     unsigned OldLabelCount = GetCodeLabelCount (Old);
778
779     /* Does the new entry have itself a label? */
780     if (CodeEntryHasLabel (New)) {
781
782         /* The new entry does already have a label - move references */
783         CodeLabel* NewLabel = GetCodeLabel (New, 0);
784         while (OldLabelCount--) {
785
786             /* Get the next label */
787             CodeLabel* OldLabel = GetCodeLabel (Old, OldLabelCount);
788
789             /* Move references */
790             MoveLabelRefs (OldLabel, NewLabel);
791
792             /* Delete the label */
793             DelCodeLabel (S, OldLabel);
794
795         }
796
797     } else {
798
799         /* The new entry does not have a label, just move them */
800         while (OldLabelCount--) {
801
802             /* Move the label to the new entry */
803             MoveCodeLabel (GetCodeLabel (Old, OldLabelCount), New);
804
805         }
806
807     }
808 }
809
810
811
812 void RemoveCodeLabelRef (CodeSeg* S, struct CodeEntry* E)
813 /* Remove the reference between E and the label it jumps to. The reference
814  * will be removed on both sides and E->JumpTo will be 0 after that. If
815  * the reference was the only one for the label, the label will get
816  * deleted.
817  */
818 {
819     /* Get a pointer to the label and make sure it exists */
820     CodeLabel* L = E->JumpTo;
821     CHECK (L != 0);
822
823     /* Delete the entry from the label */
824     CollDeleteItem (&L->JumpFrom, E);
825
826     /* The entry jumps no longer to L */
827     E->JumpTo = 0;
828
829     /* If there are no more references, delete the label */
830     if (CollCount (&L->JumpFrom) == 0) {
831         DelCodeLabel (S, L);
832     }
833 }
834
835
836
837 void MoveCodeLabelRef (CodeSeg* S, struct CodeEntry* E, CodeLabel* L)
838 /* Change the reference of E to L instead of the current one. If this
839  * was the only reference to the old label, the old label will get
840  * deleted.
841  */
842 {
843     /* Get the old label */
844     CodeLabel* OldLabel = E->JumpTo;
845
846     /* Be sure that code entry references a label */
847     PRECONDITION (OldLabel != 0);
848
849     /* Remove the reference to our label */
850     RemoveCodeLabelRef (S, E);
851
852     /* Use the new label */
853     AddLabelRef (L, E);
854 }
855
856
857
858 void AddCodeSegHint (CodeSeg* S, unsigned Hint)
859 /* Add a hint for the preceeding instruction */
860 {
861     CodeEntry* E;
862
863     /* Get the number of entries in this segment */
864     unsigned EntryCount = GetCodeEntryCount (S);
865
866     /* Must have at least one entry */
867     CHECK (EntryCount > 0);
868
869     /* Get the last entry */
870     E = GetCodeEntry (S, EntryCount-1);
871
872     /* Add the hint */
873     E->Hints |= Hint;
874 }
875
876
877
878 void DelCodeSegAfter (CodeSeg* S, unsigned Last)
879 /* Delete all entries including the given one */
880 {
881     /* Get the number of entries in this segment */
882     unsigned Count = GetCodeEntryCount (S);
883
884     /* First pass: Delete all references to labels. If the reference count
885      * for a label drops to zero, delete it.
886      */
887     unsigned C = Count;
888     while (Last < C--) {
889
890         /* Get the next entry */
891         CodeEntry* E = GetCodeEntry (S, C);
892
893         /* Check if this entry has a label reference */
894         if (E->JumpTo) {
895             /* If the label is a label in the label pool and this is the last
896              * reference to the label, remove the label from the pool.
897              */
898             CodeLabel* L = E->JumpTo;
899             int Index = CollIndex (&S->Labels, L);
900             if (Index >= 0 && CollCount (&L->JumpFrom) == 1) {
901                 /* Delete it from the pool */
902                 CollDelete (&S->Labels, Index);
903             }
904
905             /* Remove the reference to the label */
906             RemoveCodeLabelRef (S, E);
907         }
908
909     }
910
911     /* Second pass: Delete the instructions. If a label attached to an
912      * instruction still has references, it must be references from outside
913      * the deleted area. Don't delete the label in this case, just make it
914      * ownerless and move it to the label pool.
915      */
916     C = Count;
917     while (Last < C--) {
918
919         /* Get the next entry */
920         CodeEntry* E = GetCodeEntry (S, C);
921
922         /* Check if this entry has a label attached */
923         if (CodeEntryHasLabel (E)) {
924             /* Move the labels to the pool and clear the owner pointer */
925             MoveLabelsToPool (S, E);
926         }
927
928         /* Delete the pointer to the entry */
929         CollDelete (&S->Entries, C);
930
931         /* Delete the entry itself */
932         FreeCodeEntry (E);
933     }
934 }
935
936
937
938 void OutputCodeSeg (const CodeSeg* S, FILE* F)
939 /* Output the code segment data to a file */
940 {
941     unsigned I;
942
943     /* Get the number of entries in this segment */
944     unsigned Count = GetCodeEntryCount (S);
945
946     /* If the code segment is empty, bail out here */
947     if (Count == 0) {
948         return;
949     }
950
951     /* Output the segment directive */
952     fprintf (F, ".segment\t\"%s\"\n\n", S->SegName);
953
954     /* If this is a segment for a function, enter a function */
955     if (S->Func) {
956         fprintf (F, ".proc\t_%s\n\n", S->Func->Name);
957     }
958
959     /* Output all entries */
960     for (I = 0; I < Count; ++I) {
961
962         unsigned char Use;
963
964         OutputCodeEntry (CollConstAt (&S->Entries, I), F);
965                               
966 #if 0
967         /* Print usage info */
968         Use = GetRegInfo ((CodeSeg*) S, I+1);
969         fprintf (F,
970                  "  Use: %c%c%c\n",
971                  (Use & REG_A)? 'A' : '_',
972                  (Use & REG_X)? 'X' : '_',
973                  (Use & REG_Y)? 'Y' : '_');
974 #else
975         fprintf (F, "\n");
976 #endif
977     }
978
979     /* If this is a segment for a function, leave the function */
980     if (S->Func) {
981         fprintf (F, "\n.endproc\n\n");
982     }
983 }
984
985
986
987
988
989
990