バージョンアップ
[YACASL2.git] / src / label.c
index 035665a..385922d 100644 (file)
-#include "casl2.h"
 #include "assemble.h"
 
-LABELTAB *labels[LABELTABSIZE];
+/**
+ * ラベルのハッシュ値をセットしたキーを返す
+ *
+ * @return ハッシュ値をセットしたキー
+ *
+ * @param value 値
+ */
+HKEY *label_hashkey(const char *value);
+
+/**
+ * プログラム名とラベルに対応するハッシュ値を返す
+ *
+ * @return ハッシュ値
+ *
+ * @param prog プログラム名
+ * @param label ラベル
+ */
+unsigned labelhash(const char *prog, const char *label);
+
+/**
+ * ラベルを比較した結果を返す。qsort内で使われる関数
+ *
+ * @return ラベルが同一の場合は0、異なる場合は0以外
+ *
+ * @param *a ラベルa
+ * @param *b ラベルb
+ */
+int compare_adr(const void *a, const void *b);
+
+/**
+ * @brief ラベル数
+ */
+static int labelcnt = 0;
+
+/**
+ * @brief ラベル表
+ */
+static LABELTAB *labels[LABELTABSIZE];
+
+/**
+ * @brief ラベルのエラー
+ */
+static CERR cerr_label[] = {
+    { 101, "label already defined" },
+    { 102, "label table is full" },
+    { 103, "label not found" },
+};
+
+HKEY *label_hashkey(const char *value) {
+    HKEY *key = NULL;
+
+    key = malloc_chk(sizeof(HKEY), "label_hashkey");
+    key->type = CHARS;
+    key->val.s = strdup_chk(value, "label_hashkey->value");
+    return key;
+}
+
+unsigned labelhash(const char *prog, const char *label)
+{
+    HKEY *keys[2] = {NULL};
+    int i = 0;
+    unsigned h = 0;
+
+    if(prog[0]) {
+        keys[i++] = label_hashkey(prog);
+    }
+    keys[i] = label_hashkey(label);
+    h = hash(i+1, keys, LABELTABSIZE);
+    for(int j = 0; j < i + 1; j++) {
+        FREE(keys[j]->val.s);
+        FREE(keys[j]);
+    }
+    return h;
+}
+
+int compare_adr(const void *a, const void *b)
+{
+    return (**(LABELARRAY **)a).adr - (**(LABELARRAY **)b).adr;
+}
+
+/* assemble.hで定義された関数群 */
+void addcerrlist_label()
+{
+    addcerrlist(ARRAYSIZE(cerr_label), cerr_label);
+}
 
-/* ラベル表からアドレスを検索する */
 WORD getlabel(const char *prog, const char *label)
 {
-    LABELTAB *np;
+    assert(prog != NULL && label != NULL);
+    LABELTAB *p = NULL;
+    LABELARRAY *l = NULL;
 
-    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) {
+        l = p->label;
+        if((!prog[0] || (strcmp(prog, l->prog) == 0)) &&
+           strcmp(label, l->label) == 0)
+        {
+            return l->adr;
         }
     }
     return 0xFFFF;
 }
 
-/* ラベルを表に追加する */
 bool addlabel(const char *prog, const char *label, WORD adr)
 {
-    LABELTAB *np;
-    unsigned hashval;
+    assert(label != NULL);
+    LABELTAB *p = NULL;
+    LABELARRAY *l = NULL;
+    unsigned h = 0;
 
+    /* 登録されたラベルを検索。すでに登録されている場合はエラー発生 */
     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;
-    }
-    hashval = hash(label, LABELTABSIZE);
-    np->next = labels[hashval];
-    labels[hashval] = np;
-    np->adr = adr;
+    /* メモリを確保 */
+    p = malloc_chk(sizeof(LABELTAB), "labels.next");
+    l = p->label = malloc_chk(sizeof(LABELARRAY), "labels.label");
+    /* プログラム名を設定 */
+    l->prog = strdup_chk(prog, "label.prog");
+    /* ラベルを設定 */
+    l->label = strdup_chk(label, "label.label");
+    /* アドレスを設定 */
+    l->adr = adr;
+    /* ラベル数を設定 */
+    labelcnt++;
+    /* ハッシュ表へ追加 */
+    p->next = labels[h = labelhash(prog, label)];
+    labels[h] = p;
     return true;
 }
 
-/* ラベル表を表示する */
 void printlabel()
 {
-    int i;
-    LABELTAB *np;
-    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);
-            } else {
-                fprintf(stdout, "%s.%s ---> #%04X\n", np->prog, np->label, np->adr);
-            }
+    int s = 0;
+    LABELTAB *p = NULL;
+    LABELARRAY **l = {NULL};
+
+    l = calloc_chk(labelcnt, sizeof(LABELARRAY **), "labels");
+    for(int i = 0; i < LABELTABSIZE; i++) {
+        for(p = labels[i]; p != NULL; p = p->next) {
+            assert(p->label != NULL);
+            l[s++] = p->label;
         }
     }
+    qsort(l, s, sizeof(*l), compare_adr);
+    for(int i = 0; i < s; i++) {
+        if(l[i]->prog[0]) {
+            fprintf(stdout, "%s.", l[i]->prog);
+        }
+        fprintf(stdout, "%s ---> #%04X\n", l[i]->label, l[i]->adr);
+    }
+    FREE(l);
 }
 
-/* ラベル表を解放する */
 void freelabel()
 {
     int i;
-    LABELTAB *np, *nq;
+    LABELTAB *p = NULL;
+    LABELTAB *q = NULL;
+
     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->label->prog);
+            FREE(p->label->label);
+            FREE(p->label);
+            FREE(p);
         }
+        labels[i] = NULL;
     }
 }