#pragma once
#include "app.h"

#define OPHANDLER(ACB) std::bind(ACB, this, std::placeholders::_1, std::placeholders::_2)

#define MODE_6502 0
#define MODE_65C02 1
#define MODE_65816 2

#define SYNTAX_MERLIN 0
#define SYNTAX_APW	  1
#define SYNTAX_ORCA	  2

#define FLAG_FORCELONG 0x01
#define FLAG_FORCEABS  0x02
#define FLAG_FORCEDP   0x04
#define FLAG_DP		   0x08
#define FLAG_BIGNUM    0x10
#define FLAG_INDUM     0x20

#define FLAG_FORCEADDRPRINT 0x0100
#define FLAG_NOLINEPRINT 0x2000

#define OP_A       0x0001
#define OP_XY      0x0002
#define OP_PSUEDO  0x0004
#define OP_SPECIAL 0x0008

// these bits are actually the CC (last 2 bits) of opcode addressing
#define OP_CLASS0  0x0000
#define OP_CLASS1  0x0100
#define OP_CLASS2  0x0200
#define OP_CLASS3  0x0300
// these ORd bits specify specific classes of opcodes and subgroups
#define OP_STD     (0x1000 | OP_CLASS1)
#define OP_ASL     (0x2000 | OP_CLASS2)
#define OP_STX     (0x3000 | OP_CLASS2)
#define OP_C0      (0x4000 | OP_CLASS0)

enum asmErrors
{
	errNone,
	errWarn,
	errIncomplete,
	errUnimplemented,
	errFatal,
	errBadAddressMode,
	errBadOpcode,
	errIncompatibleOpcode,
	errBadByteCount,
	errBadBranch,
	errForwardRef,
	errNoRedefinition,
	errDupSymbol,
	errBadDUMop,
	errOverflow,
	errRecursiveOp,
	errOpcodeNotStarted,
	errDuplicateFile,
	errFileNotFound,
	errFileNoAccess,
	errBadEvaluation,
	errIllegalCharOperand,
	errBadCharacter,
	errUnexpectedOp,
	errUnexpectedEOF,
	errBadLUPOperand,
	errBadLabel,
	errBadOperand,
	errMAX
};

#ifdef ADD_ERROR_STRINGS
const std::string errStrings[errMAX + 1] =
{
	"No Error",
	"Warning",
	"Unfinished Opcode",
	"Unimplemented Instruction",
	"Fatal",
	"Unsupported Addressing Mode",
	"Bad opcode",
	"Opcode not available under CPU mode",
	"Byte output differs between passes",
	"Relative branch offset too large",
	"Forward Reference to symbol",
	"Unable to redefine symbol",
	"Duplicate Symbol",
	"Invalid use of DUM/DEND",
	"Overflow detected",
	"Recursive Operand",
	"Opcode without start",
	"File already included",
	"File not found",
	"File no access",
	"Unable to evaluate",
	"Illegal char in operand",
	"Unexpected character in input",
	"Unexpected opcode",
	"Unexpected End of File",
	"LUP value must be 0 < VAL <= $8000",
	"Unknown label",
	"Bad operand",

	""
};
#else
extern const std::string errStrings[errMAX];
extern uint8_t opCodeCompatibility[256];

#endif

enum
{
	syn_err = -1,  	// error - not recognized
	syn_none = 0,   // should never be returned 0
	syn_implied,    // no operand               1
	syn_s, 			// expr,s                   2
	syn_sy,			// (expr,s),y               3
	syn_imm,		// #expr                    4
	syn_diix,		// (expr,x)                 5
	syn_diiy,		// (expr),y                 6
	syn_di,			// (expr)                   7
	syn_iyl,		// [expr],y                 8
	syn_dil,		// [expr]                   9
	syn_absx,		// expr,x                  10
	syn_absy,		// expr,y                  11
	syn_bm,			// block move              12
	syn_abs,		// expr                    13

	syn_MAX
};

class TOriginSection
{
// SGQ - if you do something unusual here, be aware of copy constructor
// may be needed
public:
	uint32_t origin;
	uint32_t currentpc;
	uint32_t totalbytes;
	uint32_t orgsave;
// leave this code here in case we ever need a copy/assignment constructor
#if 0
	TOriginSection(const TOriginSection &old)
	{
		origin = old.origin;
		currentpc = old.currentpc;
		orgsave = old.orgsave;
		totalbytes = old.totalbytes;
	};

	TOriginSection& operator=(const TOriginSection &other)
	{
		origin = other.origin;
		currentpc = other.currentpc;
		totalbytes = other.totalbytes;
		orgsave = other.orgsave;
		return (*this);
	};

	TOriginSection()
	{
		origin = 0;
		currentpc = 0;
		totalbytes = 0;
		orgsave = 0;
	};
	~TOriginSection()
	{
	}
#endif
};

class MerlinLine
{
public:

	uint8_t syntax;
	std::string lable;
	std::string printlable;
	std::string opcode;
	std::string opcodelower;
	std::string operand;
	std::string operand_expr;
	std::string operand_expr2;
	std::string comment;
	std::string addrtext;
	uint8_t linemx;
	uint16_t commentcol;
	bool showmx;
	uint32_t lineno;
	uint32_t flags;
	uint16_t opflags;
	int32_t startpc;
	uint32_t addressmode;
	uint32_t expr_value;
	uint8_t expr_shift;  // after an eval, this byte will reflect any shift code on expr (|^<>)
	uint32_t eval_result; // this is the error code from the evaluate routing (0 or neg)
	uint32_t errorcode;
	std::string errorText;

	uint16_t pass0bytect;
	uint16_t bytect;
	uint16_t datafillct;
	uint8_t  datafillbyte;
	uint16_t outbytect;
	std::vector<uint8_t> outbytes;

public:
	MerlinLine();
	MerlinLine(std::string line);
	void clear();
	void set(std::string line);
	void print(uint32_t lineno);
	void setError(uint32_t ecode);
};

class TFileProcessor
{
protected:
	std::string initialdir;
	std::vector<std::string> filenames;
	uint8_t syntax;
	uint64_t starttime;
	uint8_t tabs[10];

	uint32_t filecount; // how many files have been read in (because of included files from source
public:
	uint32_t errorct;
	std::string filename;

	TFileProcessor();
	virtual ~TFileProcessor();
	virtual std::string processFilename(std::string p, std::string currentdir, int level);
	virtual int processfile(std::string p, std::string &newfilename);
	virtual void init(void);
	virtual int doline(int lineno, std::string line);
	virtual void process(void);
	virtual void complete(void);
	virtual void errorOut(uint16_t code);
};

class TMerlinConverter : public TFileProcessor
{
protected:
	std::vector<MerlinLine> lines;

public:
	TMerlinConverter();
	virtual ~TMerlinConverter();
	virtual void init(void);
	virtual int doline(int lineno, std::string line);
	virtual void process(void);
	virtual void complete(void);
};

class TLUPstruct
{
public:
	TLUPstruct()
	{
		clear();
	}
	void clear(void) {
		lupct=0;
		lupoffset=0;
		luprunning=0;
		lupskip=false;
	}
	uint16_t lupct;
	bool lupskip;
	uint32_t lupoffset;
	uint16_t luprunning;
};

class TDOstruct
{
public:
	TDOstruct()
	{
		clear();
	}
	void clear(void) {
		doskip=false;
		value=0;
	}
	uint32_t value;
	bool doskip;
};

class TSymbol;
typedef int (*TOpCB)(MerlinLine &line, TSymbol &sym);
typedef std::function<int (MerlinLine &line, TSymbol &sym)> TOpCallback;

class TSymbol
{
public:
	std::string namelc;
	std::string name;
	std::string text;
	uint32_t value;
	uint16_t stype;
	uint8_t opcode;
	bool used;
	TOpCallback cb;
	Poco::HashMap<std::string, TSymbol>locals;

	TSymbol()
	{
		clear();
	};
	void clear(void)
	{
		value = 0;
		used = false;
		text = "";
		name = "";
		namelc = "";
		stype = 0;
		opcode = 0;
		locals.clear();
	}
};

class TPsuedoOp;

class T65816Asm : public TFileProcessor
{
public:
	// options
	bool casesen;
	bool listing;
	bool showmx;
	bool trackrep;
	bool merlincompat;
	bool allowdup;
	uint8_t mx;
	uint8_t cpumode; // 0=6502, 1=65C02, 2=65816

	bool passcomplete;
	bool relocatable;
	int dumstart; // must be signed
	uint32_t dumstartaddr;
	bool skiplist; // used if lst is on, but LST opcode turns it off
	uint32_t lineno;

	std::string savepath;
	TSymbol *currentsym;
	std::vector<MerlinLine> lines;
	Poco::HashMap<std::string, TSymbol>opcodes;
	Poco::HashMap<std::string, TSymbol> macros;
	Poco::HashMap<std::string, TSymbol> symbols;
	Poco::HashMap<std::string, TSymbol> variables;

	TOriginSection PC;
	TLUPstruct curLUP;
	TDOstruct curDO;

	std::stack<TOriginSection> PCstack;
	std::stack<TLUPstruct> LUPstack;
	std::stack<TDOstruct> DOstack;

	TPsuedoOp *psuedoops;

	uint16_t pass;

	T65816Asm();
	virtual ~T65816Asm();

	virtual void init(void);
	virtual int doline(int lineno, std::string line);
	virtual void process(void);
	virtual void complete(void);

	void insertOpcodes(void);
	void pushopcode(std::string op, uint8_t opcode, uint16_t flags, TOpCallback cb);

	int callOpCode(std::string op, MerlinLine &line);
	TSymbol *findSymbol(std::string sym);
	TSymbol *addSymbol(std::string sym, uint32_t val, bool replace);
	TSymbol *findVariable(std::string sym);
	TSymbol *addVariable(std::string sym, std::string val, bool replace);


	void initpass(void);
	void showSymbolTable(bool alpha);
	void showVariables(void);
	int evaluate(MerlinLine &line, std::string expr, int64_t &value);

	int substituteVariables(MerlinLine & line);
	bool codeSkipped(void);

	int parseOperand(MerlinLine &line);
	int  getAddrMode(MerlinLine &line);
	void setOpcode(MerlinLine &line, uint8_t op);


	int doPSEUDO(MerlinLine &line, TSymbol &sym);
	int doEND(MerlinLine &line, TSymbol &sym);
	int doBase6502(MerlinLine &line, TSymbol &sym);
	int doBRANCH(MerlinLine &line, TSymbol &sym);
	int doJMP(MerlinLine &line, TSymbol &sym);
	int doAddress(MerlinLine &line, TSymbol &sym);
	int doNoPattern(MerlinLine &line, TSymbol &sym);
	int doMVN(MerlinLine &line, TSymbol &sym);
	int doPER(MerlinLine &line, TSymbol &sym);

	int doEQU(MerlinLine &line, TSymbol &sym);
	int doXC(MerlinLine &line, TSymbol &sym);
	int doMX(MerlinLine &line, TSymbol &sym);

	int doBYTE(MerlinLine &line, TSymbol &sym);
	int doUNK(MerlinLine &line, TSymbol &sym);

};

class T65816Link : public TFileProcessor
{
public:
	T65816Link();
	virtual ~T65816Link();
	virtual void init(void);
	virtual int doline(int lineno, std::string line);
	virtual void process(void);

	virtual void complete(void);
};