X-Git-Url: http://j8takagi.net/cgi-bin/gitweb.cgi?p=YACASL2.git;a=blobdiff_plain;f=src%2Flabel.c;h=a8baa1df2b585e91c12e7be3d09f799d6cafc338;hp=5a0b4cc9aa540cbaa3cd20bbf9938a81076f10fb;hb=6eb54846c27b216d7602a5107d9383f8f4324d71;hpb=0423c8cffe1e1a480222b0a80ecd31957edec06d

diff --git a/src/label.c b/src/label.c
index 5a0b4cc..a8baa1d 100644
--- a/src/label.c
+++ b/src/label.c
@@ -1,72 +1,148 @@
-#include "casl2.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <assert.h>
+
+#include "cerr.h"
+#include "cmem.h"
+#include "hash.h"
 #include "assemble.h"
 
-LABELTAB *labels[LABELTABSIZE];
+static int labelcnt = 0;                /* ラベル数 */
+static LABELTAB *labels[LABELTABSIZE];  /* ラベル表 */
 
-/* ラベル表からアドレスを検索する */
-WORD getlabel(const char *label, const char *prog)
+/**
+ * プログラム名とラベルに対応するハッシュ値を返す
+ */
+unsigned labelhash(const char *prog, const char *label)
 {
-    LABELTAB *np;
+    HKEY *keys[2];
+    int i = 0, j;
+    unsigned h;
+
+    if(prog != NULL) {
+        keys[i] = malloc_chk(sizeof(HKEY), "labelhash.key");
+        keys[i]->type = CHARS;
+        keys[i]->val.s = strdup_chk(prog, "labelhash.key.val");
+        i++;
+    }
+    keys[i] = malloc_chk(sizeof(HKEY), "labelhash.key");
+    keys[i]->type = CHARS;
+    keys[i]->val.s = strdup_chk(label, "labelhash.key.val");
+    h = hash(i+1, keys, LABELTABSIZE);
+    for(j = 0; j < i + 1; j++) {
+        FREE(keys[j]->val.s);
+        FREE(keys[j]);
+    }
+    return h;
+}
+
+/**
+ * プログラム名とラベルに対応するアドレスをラベル表から検索する
+ */
+WORD getlabel(const char *prog, const char *label)
+{
+    assert(label != NULL);
+    LABELTAB *p;
 
-    for(np = labels[hash(label, LABELTABSIZE)]; np != NULL; np = np->next) {
-        if(strcmp(label, np->label) == 0) {
-            return np->adr;
+    for(p = labels[labelhash(prog, label)]; p != NULL; p = p->next) {
+        if((prog == NULL || (p->prog != NULL && strcmp(prog, p->prog) == 0)) &&
+           strcmp(label, p->label) == 0)
+        {
+            return p->adr;
         }
     }
     return 0xFFFF;
 }
 
-/* ラベルを表に追加する */
+/**
+ * プログラム名、ラベル、アドレスをラベル表に追加する
+ */
 bool addlabel(const char *prog, const char *label, WORD adr)
 {
+    assert(label != NULL);
     LABELTAB *np;
     unsigned hashval;
 
-    if(getlabel(label, prog) != 0xFFFF) {
+    /* 登録されたラベルを検索。すでに登録されている場合はエラー発生 */
+    if(getlabel(prog, label) != 0xFFFF) {
         setcerr(101, label);    /* label already defined */
         return false;
     }
-    np = (LABELTAB *) malloc(sizeof(*np));
-    if(np == NULL || (np->label = strdup(label)) == NULL ||
-       (prog != NULL && (np->prog = strdup(prog)) == NULL))
-    {
-        setcerr(102, NULL);    /* label table is full */
-        return false;
+    /* メモリを確保 */
+    np = malloc_chk(sizeof(LABELTAB), "labels.next");
+    /* プログラム名を設定 */
+    if(prog == NULL) {
+        np->prog = NULL;
+    } else {
+        np->prog = strdup_chk(prog, "labels.prog");
     }
-    hashval = hash(label, LABELTABSIZE);
+    /* ラベルを設定 */
+    np->label = strdup_chk(label, "labels.label");
+    /* アドレスを設定 */
+    np->adr = adr;
+    /* ラベル数を設定 */
+    labelcnt++;
+    /* ハッシュ表へ追加 */
+    hashval = labelhash(prog, label);
     np->next = labels[hashval];
     labels[hashval] = np;
-    np->adr = adr;
     return true;
 }
 
-/* ラベル表を表示する */
+/**
+ * ラベルを比較した結果を返す
+ */
+int compare_adr(const void *a, const void *b)
+{
+    return (**(LABELARRAY **)a).adr - (**(LABELARRAY **)b).adr;
+}
+
+/**
+ * ラベル表を表示する
+ */
 void printlabel()
 {
-    int i;
-    LABELTAB *np;
+    int i, asize = 0;
+    LABELTAB *p;
+    LABELARRAY *ar[labelcnt];
+
     for(i = 0; i < LABELTABSIZE; i++) {
-        for(np = labels[i]; np != NULL; np = np->next) {
-            if(np->prog == NULL) {
-                fprintf(stdout, "%s ---> #%04X\n", np->label, np->adr);
+        for(p = labels[i]; p != NULL; p = p->next) {
+            assert(p->label != NULL);
+            ar[asize] = malloc_chk(sizeof(LABELARRAY), "ar[]");
+            if(p->prog == NULL) {
+                ar[asize]->prog = NULL;
             } else {
-                fprintf(stdout, "%s.%s ---> #%04X\n", np->prog, np->label, np->adr);
+                ar[asize]->prog = strdup_chk(p->prog, "ar[].prog");
             }
+            ar[asize]->label = strdup_chk(p->label, "ar[].label");
+            ar[asize++]->adr = p->adr;
         }
     }
+    qsort(ar, asize, sizeof(*ar), compare_adr);
+    for(i = 0; i < asize; i++) {
+        if(ar[i]->prog != NULL) {
+            fprintf(stdout, "%s.", ar[i]->prog);
+        }
+        fprintf(stdout, "%s ---> #%04X\n", ar[i]->label, ar[i]->adr);
+    }
 }
 
-/* ラベル表を解放する */
+/**
+ * ラベル表を解放する
+ */
 void freelabel()
 {
     int i;
-    LABELTAB *np, *nq;
+    LABELTAB *p, *q;
+
     for(i = 0; i < LABELTABSIZE; i++) {
-        for(np = labels[i]; np != NULL; np = nq){
-            nq = np->next;
-            free(np->prog);
-            free(np->label);
-            free(np);
+        for(p = labels[i]; p != NULL; p = q) {
+            q = p->next;
+            FREE(p->prog);
+            FREE(p->label);
+            FREE(p);
         }
     }
 }