#include "tToken.h"
#include <iostream>
#include <utility>
#include <string>
#include <iomanip>
#include <sstream>

using namespace std;

void tToken::print(tTokenFormat format)
{
	switch (format) {
		case STRING:
			printStringToken(pair<string,double>());
			break;
		case CANDC:
			printCandCToken(pair<string,double>());
			break;
		case CONLL:
			printConllToken(pair<string,double>());
			break;
		case TNT:
			printTnTToken(pair<string,double>());
			break;
		case PET:
			printPETToken();
			break;
		case FSC:
			printFSCToken(pair<string,double>());
			break;
	}
}

const char *tToken::separator(tTokenFormat format)
{
	switch (format) {
		case STRING:
			return " ";
			break;
		case CANDC:
			return " ";
			break;
		case TNT:
			return "\n";
			break;
		case CONLL:
			return "\n";
			break;
		case PET:
			return " ";
			break;
		case FSC:
			return "\n";
			break;
	}
	return " ";
}

void tToken::print(tTokenFormat format, pair<string,double> stag)
{
	switch (format) {
		case STRING:
			printStringToken(stag);
			break;
		case CANDC:
			printCandCToken(stag);
			break;
		case CONLL:
			printConllToken(stag);
			break;
		case TNT:
			printTnTToken(stag);
			break;
		case PET:
			printPETToken();
			break;
		case FSC:
			printFSCToken(stag);
			break;
	}
}

void tToken::printStringToken(pair<string,double> stag)
{
	cout << _surface;
	if (!stag.first.empty()) {
		cout << "|" << stag.first << "|" << stag.second;
	}
}
void tToken::printCandCToken(pair<string,double> stag)
{
	cout << _surface;
	if (!_tags.empty()) {
		cout << "|" << _tags[0];
	}
	if (!stag.first.empty()) {
		cout << "|" << stag.first;
	}
}
void tToken::printConllToken(pair<string,double> stag)
{
	cout << _surface;
	if (!_tags.empty()) {
		cout << "\t";
		for (int i=0; i < _tags.size(); ++i) {
			double prob;
			istringstream(_probs[i]) >> prob;
			cout << " \"" << _tags[i] << "\" " << fixed << setprecision(4) 
				<< prob;
		}
	}
	if (!stag.first.empty()) {
		cout << "\t" << stag.first;
	}
}
void tToken::printTnTToken(pair<string,double> stag)
{
	cout << _surface;
	if (!_tags.empty() && stag.first.empty()) {
		cout << "\t" << _tags[0];
	}
	if (!stag.first.empty()) {
		cout << "\t" << stag.first;
	}
	//special
	if (!_tids.empty()) {
		for (vector<int>::iterator it = _tids.begin(); it != _tids.end();
			++it) {
			cout << " " << *it;
		}
	}
}
void tToken::printPETToken()
{
	cout << "(" << _id+42 << ", " << _id << ", " << _id+1 << ", <" 
		<< _start << ":" 
		<< _end << ">, 1, \"" << _surface << "\", 0, \"null\"";
	if (!_tags.empty()) {
		cout << ",";
		for (int i=0; i < _tags.size(); ++i) {
			double prob;
			istringstream(_probs[i]) >> prob;
			cout << " \"" << _tags[i] << "\" " << fixed << setprecision(4) 
				<< prob;
		}
	}
	cout << ")";
}
void tToken::printFSCToken(pair<string,double> stag)
{
	cout << "<edge source=\"v" << _id << "\" target=\"v"
		<< _id+1 << "\">" << endl;
	cout << "<fs type=\"token\">" << endl;
	cout << "<f name=\"+FORM\"><str><![CDATA[" << _surface 
		<< "]]></str></f>" << endl;
	cout << "<f name=\"+FROM\"><str>" << _start << "</str></f>" << endl;
	cout << "<f name=\"+TO\"><str>" << _end-1 << "</str></f>" << endl;
	if (!_tags.empty()) {
		cout << "<f name=\"+TNT\">" << endl;
		cout << "<fs type=\"tnt\">" << endl;
		cout << "<f name=\"+TAGS\" org=\"list\">";
		for (std::vector<std::string>::iterator i1 = _tags.begin();
			i1 != _tags.end(); ++i1) {
			cout << "<str>" << *i1 << "</str>";
		}
		cout << "</f>" << endl;
		cout << "<f name=\"+PRBS\" org=\"list\">";
		for (std::vector<std::string>::iterator i2 = _probs.begin();
			i2 != _probs.end(); ++i2) {
			double prob;
			istringstream(*i2) >> prob;
			cout << "<str>" << fixed << setprecision(4) << prob << "</str>";
		}
		cout << "</f>" << endl;
		cout << "</fs>" << endl << "</f>" << endl;
	}
	if (!stag.first.empty()) {
		cout << "<f name=\"+STAG\">" << endl;
		cout << "<fs type=\"stag\">" << endl;
		cout << "<f name=\"+TAG\" org=\"list\"><str>" << stag.first 
			<< "</str></f>" << endl;
		cout << "<f name=\"+PRB\" org=\"list\"><str>" << fixed << setprecision(6) 
			<< stag.second << "</str></f>" << endl;
		cout << "</fs>" << endl << "</f>" << "</fs>" << endl;
	}
	cout << "</fs>" << endl;
	cout << "</edge>";
}

void print_header(tTokenFormat format, string &itemno, string &sent, int n)
{
   if (format == STRING)
      cout << itemno << endl;
   if (format == FSC) {
      cout << "<?xml version='1.0' encoding='utf-8'?>" << endl;
      cout << "<fsc version=\"1.0\" >" << endl;
      cout << "<chart id=\"" << itemno << "\" >" << endl;
      cout << "<text><![CDATA[" << sent << "]]></text>" << endl;
      cout << "<lattice init=\"v0\" final=\"v" << n << "\">" << endl;
   }
	if (format == TNT) {
		cout << "%%" << itemno << endl;
	}
}

void print_footer(tTokenFormat format)
{
	if (format == STRING)
		cout << endl;
	if (format == PET)
		cout << endl;
	if (format == TNT)
		cout << endl << endl;
	if (format == CONLL)
		cout << endl << endl;
	if (format == CANDC)
		cout << endl;
   if (format == FSC) {
      cout << "</lattice>" << endl;
      cout << "</chart>" << endl;
      cout << "</fsc>" << endl;
      cout << endl;
   }
}

std::string tToken::tag()
{
	if (!_tags.empty()) {
		return _tags[0];
	} else {
		return std::string();
	}
}

std::string tToken::supertag()
{
	if (!_supertags.empty()) {
		return _supertags[0];
	} else {
		return std::string();
	}
}