#include <stdio.h>
#include <string.h>
#include "asm.h"
#include "tree.h"

#define MAX_CODES 4096		/* コードバッファのサイズ */
#define STRING_TABLE_SIZE 1024	/* 文字列テーブルのサイズ */

/* stringf.c で定義 */
char *Stringf(char *fmt, ...);	/* 文字列を生成して返す (malloc 使用) */

/********************************/
/* コードバッファ関係		*/
/********************************/

/* コードバッファ */
Code CodeBuf[MAX_CODES];
int CBp=0;

/* 新しいラベルの生成 */
int labelNo=0;
char *newLabel(char prefix) {
  return Stringf("_%c%04d", prefix, labelNo++);
}

/* ラベルを追加する */
void L(char *label) {
  CodeBuf[CBp].label = label;
  CodeBuf[CBp].op = CodeBuf[CBp].arg1
    = CodeBuf[CBp].arg2 = CodeBuf[CBp].comment = NUL;
  CBp++;
}

/* 命令を追加する */
void I(char *op, char *arg1, char *arg2) {
  CodeBuf[CBp].op = op;
  CodeBuf[CBp].arg1 = arg1;
  CodeBuf[CBp].arg2 = arg2;
  CodeBuf[CBp].label = CodeBuf[CBp].comment = NUL;
  CBp++;
}

/* ラベルつき命令を追加する */
void LI(char *label, char *op, char *arg1, char *arg2) {
  CodeBuf[CBp].label = label;
  CodeBuf[CBp].op = op;
  CodeBuf[CBp].arg1 = arg1;
  CodeBuf[CBp].arg2 = arg2;
  CodeBuf[CBp].comment = NUL;
  CBp++;
}

/* 直前の命令にコメントをつける */
void C(char *comment) { CodeBuf[CBp-1].comment = comment; }

/* コードの出力 */
void putCode() {
  int i;
  for (i=0; i<CBp; i++) {
    if (CodeBuf[i].label!=NUL) {
      if(CodeBuf[i].label[0]=='.') printf("%s", CodeBuf[i].label);
      else printf("%s:", CodeBuf[i].label);
    }
    if (CodeBuf[i].op!=NUL) printf("\t%s", CodeBuf[i].op);
    if (CodeBuf[i].arg1!=NUL) printf("\t%s", CodeBuf[i].arg1);
    if (CodeBuf[i].arg2!=NUL) printf(", %s", CodeBuf[i].arg2);
    if (CodeBuf[i].comment!=NUL) printf("\t# %s", CodeBuf[i].comment);
    printf("\n");
  }
}

/********************************/
/* 文字列テーブル関係		*/
/********************************/

/* 文字列テーブル */
struct {
  char *string, *label;
} StringTable[STRING_TABLE_SIZE];
int STp=0;

char *getStrLabel(char *s) {
  int i;
  for (i=0; i<STp; i++) {
    if (strcmp(s, StringTable[i].string)==0)
      return StringTable[i].label;
  }
  StringTable[STp].string = s;
  return (StringTable[STp++].label = newLabel('S'));
}

/* 文字列定数の生成 */
void genStrings(void) {
  int i;
  for (i=0; i<STp; i++) {
    LI(StringTable[i].label,
       ".ascii", StringTable[i].string, "\"\\0\"");
  }
}