#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <iostream>
#include <string>
#include <vector>
#include <cmath>
#include <boost/filesystem.hpp>
#include <boost/program_options.hpp>

#include "unicode.h"
#include "grammar.h"
#include "profile.h"
#include "derivReader.h"
#include "derivReader.cpp"

using namespace std;

void collectleaves(Grammar &g, tToken s, vector<tToken> &ancestors, 
	vector<tToken> &leaves);

void parse_options(int argc, char **argv, string *gfname, string *pname, 
	int *parsenum);

int main (int argc, char **argv)
{
	// setting option variables
	string gfname, pname;
	int parsenum;
	parse_options(argc, argv, &gfname, &pname, &parsenum);

	// UTF-8 encoder
	initialize_encoding_converter("utf-8");

	Grammar g(gfname);
	Profile p(pname);
	vector<tToken> leaves; //used in collectleaves()
	DerivReader<vector<tToken> > reader(g, leaves, NULL, NULL, &collectleaves);

	pair<pair<tIid,int>,string> result = p.getResult();
	tIid context = -1; // context is item id
	int event = -1; 	// event is parse id

	while (result.first.first >= 0) {//new item/context
		if (result.first.first != context) {//new item/context
			if (context != -1) {
				//finish last context
			}//finished last context
		}
		context = result.first.first;
		event = result.first.second;
		if (parsenum >= 0 && event != parsenum) {
			result = p.getResult();
			continue;
		}
		//do stuff with result
		leaves.clear();
		reader.readDeriv(result.second);
		result = p.getResult();

		for(vector<tToken>::iterator it = leaves.begin(); it != leaves.end(); 
			++it) {
			cout << context << "\t" << (p.isGold(context,event)?"Y":"N") << "\t"
				<< it->start() << "\t" << it->end() << "\t"
				<< (it->carg().empty()?it->surface():it->carg());
			vector<string> stags = it->supertags();
			vector<string> sprobs = it->superprobs();
			cout << "\t" << it->tag();
			if (stags.size() > 0) {
				for (int i=0; i <= stags.size()-1; ++i) {
					cout << "\t" << stags[i];
					if (i < sprobs.size()) {
						cout << "\t" << sprobs[i];
					}
				}
			}
			cout << endl;
		}
	}
	//finish last context

	return 0;
}

void parse_options(int argc, char **argv, string *gfname, string *pname, 
	int *parsenum)
{
	namespace po = boost::program_options;
	po::options_description visible("Options");
	visible.add_options()
		("help,h", "This usage information.")
		("analysis,a", po::value<int>(parsenum)->default_value(-1),
			"Select a specific analysis, default (-1): all.")
	;
	po::options_description hidden("Hidden options");
	hidden.add_options()
		("grammar-file", po::value<string>(gfname), "grammar .tdl file")
		("profile", po::value<string>(pname), "profile")
	;

	po::options_description cmd_line ("Command line options");
	cmd_line.add(visible).add(hidden);

	po::positional_options_description p;
	p.add("grammar-file",1).add("profile",1);

	po::variables_map vm;
	po::store(po::command_line_parser(argc, argv).
		options(cmd_line).positional(p).run(), vm);
	notify(vm);

	if (vm.count("help")) {
		cout << "Usage: " << argv[0] << " [options] "
			<< "grammar-file profile" << endl;
		cout << visible << endl;
		exit(0);
	}

	if (!vm.count("grammar-file") || !vm.count("profile")) {
		cerr << "Insufficient arguments given." << endl;
		cerr << "Usage: " << argv[0] << " [options] "
			<< "grammar-file profile" << endl;
		cerr << visible << endl;
		exit(1);
	}
}

void collectleaves(Grammar &g, tToken s, vector<tToken> &ancestors, 
	vector<tToken> &leaves)
{
	leaves.push_back(s);
	string tag(ancestors.back().surface());
	for (int x=ancestors.size()-2; x >= 0
		&& g.is_lexrule(ancestors[x].surface()); x--) {
		string infl = ancestors[x].surface();
		if (infl.compare(infl.length()-3, 3,"plr") != 0
			&& (infl.length() < 5 ||
				(infl.length() >=5 && infl.compare(0,5,"punct") != 0))) {
			tag+=":";
			tag+=infl;
		}
	}
	leaves.back().tag(tag);
	leaves.back().prob("1.0");
	leaves.back().start(ancestors.back().start());
	leaves.back().end(ancestors.back().end());
}