#ifndef _MRS_H_
#define _MRS_H_

#include <map>
#include <string>
#include <vector>
#include <utility>
#include <iomanip>
#include <functional>
		
enum RelType {REAL, QUANT};
typedef std::string VarType; //e, x, h, handle, ref-ind
typedef std::string VarID; //numeric bit of Var
typedef std::pair<VarType, VarID> Var;
typedef std::string Feature; //SF, TENSE, PERS
typedef std::string Value; // UNTENSED, SG, 3
typedef std::string Role; //ARG0, ARG1, RESTR, BODY, R-INDEX etc
typedef std::map<Feature, Value> Props;
typedef std::map<Role, Var> Roles;
typedef std::pair<int, std::pair<std::string, int> > ATriple;
typedef std::pair<int, std::pair<std::string, std::string> > PTriple;
struct Rel {
	std::string name;
	std::string span;
	std::string carg;
	RelType type;
	Var lbl;
	Var arg0;
	Roles *roles;
};

class Unmatched {
	public:
		std::string dir;
		std::string name1;
		std::string span1;
		std::string pred;
		std::string name2;
		std::string span2;

		Unmatched(std::string d, std::string n1, std::string s1, std::string p, 
			std::string n2, std::string s2)
			: dir(d), name1(n1), span1(s1), pred(p), name2(n2), span2(s2) {}
		Unmatched(std::string d, std::string n1, std::string s1, std::string p, 
			std::string n2)
			: dir(d), name1(n1), span1(s1), pred(p), name2(n2) {}
		~Unmatched() {}

		bool operator>(const Unmatched &other) const {return span1 < other.span1
			|| (span1 == other.span1 && name1 < other.name1)
			|| (span1 == other.span1 && name1 == other.name1 && pred < other.pred);
		}
};

struct unmatched_cmp  
{
	bool operator()(const Unmatched &t1, const Unmatched &t2) {
		return t1.span1 < t2.span1 
		|| (t1.span1 == t2.span1 && t1.name1 < t2.name1)
		|| (t1.span1 == t2.span1 && t1.name1 == t2.name1 && t1.pred < t2.pred);
	}
};

class tMRS
{
	bool debug;
	std::vector<Rel *> rels; 
	std::map<Var, Var> hcons;
	std::map<Var, Props *> arg0s;
	std::map<Var, int> arg0ToRel; //real rels, not quants
	std::map<Var, Var> lblToArg0;
	std::vector<ATriple> argtriples;
	std::vector<PTriple> proptriples;
	std::map<std::string,std::vector<int> > spanToRel;
	std::map<std::string,std::vector<int> > spanToATriple;
	std::map<std::string,std::vector<int> > spanToPTriple;
	std::map<std::string, int> counts; 
	Var index;
	Var ltop;
	void parseChar(char, std::string &);
	void parseID(const std::string, std::string &);
	Var parseVar(std::string &);
	Props *parseProps(std::string &);
	void parseRel(std::string &);
	bool parseHCONS(std::string &);
	std::string readFeature(std::string &);
	std::pair<std::string,std::string> parseName(std::string &);
	std::string parseString(std::string &);


public:
	tMRS(std::string input, bool db=false);
	~tMRS();


	std::map<std::string, int> &getCounts() {return counts;};
	void printTriples();
	void printCounts();
	std::pair<std::map<std::string, int>, std::vector<Unmatched> >
		compareTriples(tMRS &test);

};

#endif