#define Maxarg 4                /* maximum number of args an insn can have */
#define Wordsz 4                /* the size of a "natural int" */
#define Ptrsz 8                 /* the size of a machine word (ie, pointer size) */

typedef size_t regid;

typedef struct Insn Insn;
typedef struct Loc Loc;
typedef struct Func Func;
typedef struct Blob Blob;
typedef struct Isel Isel;
typedef struct Asmbb Asmbb;

typedef enum {
#define Insn(val, gasfmt, p9fmt, use, def) val,
#include "insns.def"
#undef Insn
} AsmOp;

typedef enum {
#define Reg(r, gasname, p9name, mode) r,
#include "regs.def"
#undef Reg
	Nreg
} Reg;

typedef enum {
	Locnone,
	Loclbl,  /* label */
	Locreg,  /* register */
	Locmem,  /* reg offset mem */
	Locmeml, /* label offset mem */
	Loclit,  /* literal value */
	Loclitl  /* label address */
} Loctype;

typedef enum {
	ModeNone,
	ModeB, /* byte */
	ModeW, /* short */
	ModeL, /* long */
	ModeQ, /* quad */
	ModeF, /* float32 */
	ModeD, /* float64 */
	Nmode,
} Mode;

typedef enum {
	PassInNoPref,
	PassInSSE,
	PassInInt,
	PassInMemory,
} PassIn;

typedef enum {
	ArgVoid,
	/* Either int or flt, depending on Loc* type */
	ArgReg,
	/* Small aggregates packed into registers */
	ArgAggrI,
	ArgAggrF,
	ArgAggrII,
	ArgAggrFI,
	ArgAggrIF,
	ArgAggrFF,
	ArgBig,
} ArgType;

typedef enum {
	Classbad,
	Classint,
	Classflt,
	Nclass,
} Rclass;

typedef enum {
	Gnugaself,
	Gnugasmacho,
	Plan9,
} Asmsyntax;

typedef enum {
	Bti8,
	Bti16,
	Bti32,
	Bti64,
	Btimin,     /* byte-packed uint */
	Btref,
	Btbytes,
	Btseq,
	Btpad,
} Blobtype;

struct Blob {
	Blobtype type;
	size_t align;
	char *lbl;  /* may be null */
	char isglobl;
	char iscomdat;
	union {
		uvlong npad;
		uvlong ival;
		struct {
			char *str;
			char isextern;
			size_t off;
		} ref;
		struct {
			size_t len;
			char *buf;
		} bytes;
		struct {
			size_t nsub;
			Blob **sub;
		} seq;
	};
};

/* a register, label, or memory location */
struct Loc {
	Loctype type; /* the type of loc */
	Mode mode;    /* the mode of this location */
	void *list;
	union {
		char *lbl;  /* for Loclbl, Loclitl */
		struct {    /* for Locreg */
			regid id;
			Reg   colour;
		} reg;
		long  lit;  /* for Loclit */
		/*
		 * for Locmem, Locmeml.
		 * address format is
		 *    disp(base + index)
		 */
		struct {
			/* only one of lbldisp and constdisp may be used */
			char *lbldisp;
			long constdisp;
			int scale; /* 0,1,2,4, or 8 */
			Loc *base; /* needed */
			Loc *idx;  /* optional */
		} mem;
	};
};

struct Insn {
	size_t uid;
	AsmOp op;
	Loc *args[Maxarg];
	size_t nargs;
};

struct Func {
	char *name;	/* function name */
	Srcloc loc;	/* location of definition */
	Type *type;	/* type of function */

	Node **args;	/* argument list */
	size_t nargs;	/* number of args, including hidden ones */
	Htab *stkoff;	/* Loc* -> int stkoff map */
	Htab *envoff;	/* Loc* -> int envoff map */
	size_t stksz;	/* stack size */
	Node *ret;	/* return value */
	ArgType rettype;	/* how to actually get ret out */

	Cfg  *cfg;	/* flow graph */
	char isexport;	/* is this exported from the asm? */
	char hasenv;	/* do we have an environment? */
};

struct Asmbb {
	int id;		/* unique identifier */
	char **lbls;	/* list of BB labels */
	size_t nlbls;	/* number of labels */
	Insn **il;	/* instructions */
	size_t ni;	/* number of instructions */

	Bitset *pred;	/* set of predecessor BB ids */
	Bitset *succ;	/* set of successor BB ids */
	Bitset *use;	/* registers used by this BB */
	Bitset *def;	/* registers defined by this BB */
	Bitset *livein;	/* variables live on entrance to BB */
	Bitset *liveout;	/* variables live on exit from BB */
};

/* instruction selection state */
struct Isel {
	char *name;
	char *cwd;

	Cfg  *cfg;          /* cfg built with nodes */

	Asmbb **bb;         /* 1:1 mappings with the Node bbs in the CFG */
	size_t nbb;
	Asmbb *curbb;

	Node *ret;          /* we store the return into here */
	ArgType rettype;    /* how ret actually gets out of the function */
	Htab *spillslots;   /* reg id  => int stkoff */
	Htab *reglocs;      /* decl id => Loc *reg */
	Htab *stkoff;       /* decl id => int stkoff */
	Htab *envoff;       /* decl id => int envoff */
	Htab *globls;       /* decl id => char *globlname */

	Loc *envp;

	/* increased when we spill */
	Loc *stksz;
	Loc *calleesave[Nreg];
	size_t nsaved;

	/* register allocator state */

	size_t *gbits;      /* igraph matrix repr */
	regid **gadj;      /* igraph adj set repr */
	size_t *ngadj;
	size_t *gadjsz;
	size_t nreg;      /* maxregid at time of alloc */
	int *degree;        /* degree of nodes */
	int *nuses;         /* number of uses of nodes */
	Loc **aliasmap;     /* mapping of aliases */

	Bitset *shouldspill;        /* the first registers we should try to spill */
	Bitset *neverspill;        /* registers we should never spill */

	Insn ***rmoves;
	size_t *nrmoves;

	/* move sets */
	Insn **mcoalesced;
	size_t nmcoalesced;

	Insn **mconstrained;
	size_t nmconstrained;

	Insn **mfrozen;
	size_t nmfrozen;

	Bitset *mactiveset;
	Insn **mactive;
	size_t nmactive;


	/* worklists */
	Bitset *wlmoveset;
	Insn **wlmove;
	size_t nwlmove;

	Loc **wlspill;
	size_t nwlspill;

	Loc **wlfreeze;
	size_t nwlfreeze;

	Loc **wlsimp;
	size_t nwlsimp;

	Loc  **selstk;
	size_t nselstk;

	Bitset *coalesced;
	Bitset *spilled;
	Bitset *prepainted; /* locations that need to be a specific colour */
	Bitset *initial;    /* initial set of locations used by this fn */
};

/* globals */
extern Type *tyintptr;
extern Type *tyword;
extern Type *tyvoid;
extern Node *abortoob;

/* options */
extern int extracheck;
extern Asmsyntax asmsyntax;

void initconsts(Htab *globls);
void fillglobls(Stab *st, Htab *globls);
void simpglobl(Node *dcl, Htab *globls, Func ***fn, size_t *nfn, Node ***blob, size_t *nblob);
void selfunc(Isel *is, Func *fn, Htab *globls, Htab *strtab);
void gen(char *out);
void gengas(FILE *fd);
void genp9(FILE *fd);

/* blob stuff */
Blob *mkblobpad(size_t sz);
Blob *mkblobi(Blobtype type, uint64_t ival);
Blob *mkblobbytes(char *buf, size_t len);
Blob *mkblobseq(Blob **sub, size_t nsub);
Blob *mkblobref(char *lbl, size_t off, int isextern);
void blobfree(Blob *b);

Blob *tydescblob(Type *t);
Blob *litblob(Htab *globls, Htab *strtab, Node *n);
size_t blobsz(Blob *b);

/* location generation */
extern size_t maxregid;
extern Loc **locmap; /* mapping from reg id => Loc * */

void resetregs(void);
char *genlocallblstr(char *buf, size_t sz);
Type *codetype(Type *ft);
Type *closuretype(Type *ft);
Loc *loclbl(Node *lbl);
Loc *locstrlbl(char *lbl);
Loc *locreg(Mode m);
Loc *locphysreg(Reg r);
Loc *locmem(long disp, Loc *base, Loc *idx, Mode mode);
Loc *locmeml(char *disp, Loc *base, Loc *idx, Mode mode);
Loc *locmems(long disp, Loc *base, Loc *idx, int scale, Mode mode);
Loc *locmemls(char *disp, Loc *base, Loc *idx, int scale, Mode mode);
Loc *loclit(long val, Mode m);
Loc *loclitl(char *lbl);
char *asmname(Node *dcl);
char *tydescid(char *buf, size_t bufsz, Type *ty);
Loc *coreg(Reg r, Mode m);
int isfloatmode(Mode m);
int isintmode(Mode m);
int issubreg(Loc *, Loc *);
int dumbmov(Loc *, Loc *);

/* emitting instructions */
Insn *mkinsn(int op, ...);
void dbgiprintf(FILE *fd, Insn *insn);
void dbglocprint(FILE *fd, Loc *l, char spec);

/* register allocation */
void regalloc(Isel *s);
Rclass rclass(Loc *l);

/* machine dependent optimization */
void peep(Isel *s);

/* useful functions */
size_t tysize(Type *t);
size_t tyalign(Type *t);
size_t size(Node *n);
ssize_t tyoffset(Type *ty, Node *memb);
ssize_t offset(Node *aggr, Node *memb);
size_t countargs(Type *t);
ArgType classify(Type *t);
int isaggregate(Type *t);
int stacknode(Node *n);
int floatnode(Node *n);
void breakhere();
void dumpasm(Isel *s, FILE *fd);

size_t alignto(size_t sz, Type *t);

