#ifdef __GNUG__
#pragma implementation
#endif

#include <vector>

#include "math_matrixinset.h"
#include "support.h"
#include "debug.h"
#include "support/LOstream.h"
#include "Painter.h"
#include "LaTeXFeatures.h"


namespace {


int getCols(MathInsetTypes type)
{
	switch (type) {
		case LM_OT_EQNARRAY:
			return 3;
		case LM_OT_ALIGN:
		case LM_OT_ALIGNAT:
		case LM_OT_XALIGNAT:
		case LM_OT_XXALIGNAT:
			return 2;
		default:;
	}
	return 1;
}


// returns position of first relation operator in the array
// used for "intelligent splitting"
int firstRelOp(MathArray const & array)
{
	for (MathArray::const_iterator it = array.begin(); it != array.end(); ++it)
		if (it->nucleus()->isRelOp())
			return it - array.begin();
	return array.size();
}


char const * star(bool numbered)
{
	return numbered ? "" : "*";
}

}


MathMatrixInset::MathMatrixInset()
	: MathGridInset(1, 1), objtype_(LM_OT_SIMPLE), nonum_(1), label_(1)
{
	setDefaults();
}


MathMatrixInset::MathMatrixInset(MathInsetTypes t)
	: MathGridInset(getCols(t), 1), objtype_(t), nonum_(1), label_(1)
{
	setDefaults();
}


MathMatrixInset::MathMatrixInset(MathInsetTypes t, unsigned int cols)
	: MathGridInset(cols, 1), objtype_(t), nonum_(1), label_(1)
{
	setDefaults();
}


MathInset * MathMatrixInset::clone() const
{
	return new MathMatrixInset(*this);
}


char MathMatrixInset::defaultColAlign(unsigned int col)
{
	switch (getType()) {
		case LM_OT_ALIGN:
		case LM_OT_ALIGNAT:
		case LM_OT_XALIGNAT:
		case LM_OT_XXALIGNAT:
			return "rl"[col & 1];
		case LM_OT_EQNARRAY:
			return "rcl"[col];
		default:;
	}
	return 'c';
}


int MathMatrixInset::defaultColSpace(unsigned int col)
{
	switch (getType()) {
		case LM_OT_ALIGN:
		case LM_OT_ALIGNAT:
			return 0;
		case LM_OT_XALIGNAT:
			return (col & 1) ? 20 : 0;
		case LM_OT_XXALIGNAT:
			return (col & 1) ? 40 : 0;
		default:;
	}
	return 10;
}


void MathMatrixInset::metrics(MathStyles) const
{
	size_ = (getType() == LM_OT_SIMPLE) ? LM_ST_TEXT : LM_ST_DISPLAY;

	// let the cells adjust themselves
	MathGridInset::metrics(size_);

	if (display()) {
		ascent_  += 12;
		descent_ += 12;
	}	

	if (numberedType()) {
		int l = 0;
		for (unsigned int row = 0; row < nrows(); ++row)
			l = std::max(l, mathed_string_width(LM_TC_BF, size(), nicelabel(row)));

		if (l)
			width_ += 30 + l;
	}

	// make it at least as high as the current font
	int asc = 0;
	int des = 0;
	math_font_max_dim(LM_TC_TEXTRM, LM_ST_TEXT, asc, des);
	ascent_  = std::max(ascent_,  asc);
	descent_ = std::max(descent_, des);
}


void MathMatrixInset::draw(Painter & pain, int x, int y) const
{
	xo(x);
	yo(y);

	MathGridInset::draw(pain, x, y);

	if (numberedType()) {
		int xx = x + colinfo_.back().offset_ + colinfo_.back().width_ + 20;
		for (unsigned int row = 0; row < nrows(); ++row) {
			int yy = y + rowinfo_[row].offset_;
			drawStr(pain, LM_TC_BF, size(), xx, yy, nicelabel(row));
		}
	}
}


void MathMatrixInset::write(std::ostream & os, bool fragile) const
{
  header_write(os);

	bool n = numberedType();

	for (unsigned int row = 0; row < nrows(); ++row) {
		for (unsigned int col = 0; col < ncols(); ++col) {
			cell(index(row, col)).write(os, fragile);
			os << eocString(col);
		}
		if (n) {
			if (!label_[row].empty())
				os << "\\label{" << label_[row] << "}";
			if (nonum_[row])
				os << "\\nonumber ";
		}
		os << eolString(row);
	}

  footer_write(os);
}


string MathMatrixInset::label(unsigned int row) const
{
	return label_[row];
}


void MathMatrixInset::label(unsigned int row, string const & label)
{
	label_[row] = label; 
}


void MathMatrixInset::numbered(unsigned int row, bool num)
{
	nonum_[row] = !num; 
}


bool MathMatrixInset::numbered(unsigned int row) const
{
	return !nonum_[row];
}


bool MathMatrixInset::ams() const
{
	return true;
}


bool MathMatrixInset::display() const
{
	return getType() != LM_OT_SIMPLE;
}


std::vector<string> const MathMatrixInset::getLabelList() const
{
	std::vector<string> res;
	for (unsigned int row = 0; row < nrows(); ++row)
		if (!label_[row].empty() && nonum_[row] != 1)
			res.push_back(label_[row]);
	return res;
}


bool MathMatrixInset::numberedType() const
{
	if (getType() == LM_OT_SIMPLE || getType() == LM_OT_XXALIGNAT)
		return false;
	for (unsigned int row = 0; row < nrows(); ++row)
		if (!nonum_[row])
			return true;
	return false;
}


void MathMatrixInset::validate(LaTeXFeatures & features) const
{
	features.amsstyle = ams();

	// Validation is necessary only if not using AMS math.
	// To be safe, we will always run mathedvalidate.
	//if (features.amsstyle)
	//  return;

	features.boldsymbol = true;
	//features.binom      = true;

	MathNestInset::validate(features);
}


void MathMatrixInset::header_write(std::ostream & os) const
{
	bool n = numberedType();

	switch (getType()) {
		case LM_OT_SIMPLE:
			os << '$';
			if (cell(0).empty())
				os << ' ';
			break;

		case LM_OT_EQUATION:
			if (n)
				os << "\\begin{equation" << star(n) << "}\n"; 
			else
				os << "\\[\n"; 
			break;

		case LM_OT_EQNARRAY:
			os << "\\begin{eqnarray" << star(n) << "}\n";
			break;

		case LM_OT_ALIGN:
			os << "\\begin{align" << star(n) << "}";
			break;

		case LM_OT_ALIGNAT:
			os << "\\begin{alignat" << star(n) << "}" << "{" << ncols()/2 << "}\n";
			break;

		case LM_OT_XALIGNAT:
			os << "\\begin{xalignat" << star(n) << "}" << "{" << ncols()/2 << "}\n";
			break;

		case LM_OT_XXALIGNAT:
			os << "\\begin{xxalignat}" << "{" << ncols()/2 << "}\n";
			break;

		case LM_OT_MULTLINE:
			os << "\\begin{multline}\n";
			break;

		case LM_OT_GATHER:
			os << "\\begin{gather}\n";
			break;

		default:
			os << "\\begin{unknown" << star(n) << "}";
	}
}


void MathMatrixInset::footer_write(std::ostream & os) const
{
	bool n = numberedType();

	switch (getType()) {
		case LM_OT_SIMPLE:
			os << '$';
			break;

		case LM_OT_EQUATION:
			if (n)
				os << "\\end{equation" << star(n) << "}\n"; 
			else
				os << "\\]\n"; 
			break;

		case LM_OT_EQNARRAY:
			os << "\\end{eqnarray" << star(n) << "}\n";
			break;

		case LM_OT_ALIGN:
			os << "\\end{align" << star(n) << "}\n";
			break;

		case LM_OT_ALIGNAT:
			os << "\\end{alignat" << star(n) << "}\n";
			break;

		case LM_OT_XALIGNAT:
			os << "\\end{xalignat" << star(n) << "}\n";
			break;

		case LM_OT_XXALIGNAT:
			os << "\\end{xxalignat}\n";
			break;

		case LM_OT_MULTLINE:
			os << "\\end{multline}\n";
			break;

		case LM_OT_GATHER:
			os << "\\end{gather}\n";
			break;

		default:
			os << "\\end{unknown" << star(n) << "}";
	}
}


void MathMatrixInset::addRow(unsigned int row) 
{
	nonum_.insert(nonum_.begin() + row + 1, !numberedType());
	label_.insert(label_.begin() + row + 1, string());
	MathGridInset::addRow(row);
}


void MathMatrixInset::appendRow()
{
	nonum_.push_back(!numberedType());
	label_.push_back(string());
	MathGridInset::appendRow();
}


void MathMatrixInset::delRow(unsigned int row) 
{
	MathGridInset::delRow(row);
	nonum_.erase(nonum_.begin() + row);
	label_.erase(label_.begin() + row);
}


void MathMatrixInset::addCol(unsigned int col)
{
	switch (getType()) {
		case LM_OT_EQUATION:
			mutate(LM_OT_EQNARRAY);
			break;

		case LM_OT_EQNARRAY:
			mutate(LM_OT_ALIGN);
			addCol(col);
			break;

		case LM_OT_ALIGN:
			mutate(LM_OT_ALIGNAT);
			addCol(col);
			break;

		case LM_OT_ALIGNAT:
		case LM_OT_XALIGNAT:
		case LM_OT_XXALIGNAT:
			MathGridInset::addCol(col);
			MathGridInset::addCol(col + 1);
			break;

		default:
			break;
	}
}


void MathMatrixInset::delCol(unsigned int col)
{
	switch (getType()) {
		case LM_OT_ALIGNAT:
		case LM_OT_XALIGNAT:
		case LM_OT_XXALIGNAT:
			MathGridInset::delCol(col + 1);
			MathGridInset::delCol(col);
			break;
		default:
			break;
	}
}


string MathMatrixInset::nicelabel(unsigned int row) const
{
	if (nonum_[row])
		return string();
	if (label_[row].empty())
		return string("(#)");
	return "(" + label_[row] + ")";
}


namespace {
	MathInsetTypes typecode(string const & s)
	{
		if (s == "equation")
			return LM_OT_EQUATION;
		if (s == "display")
			return LM_OT_EQUATION;
		if (s == "eqnarray")
			return LM_OT_EQNARRAY;
		if (s == "align")
			return LM_OT_ALIGN;
		if (s == "alignat")
			return LM_OT_ALIGN;
		if (s == "xalignat")
			return LM_OT_XALIGNAT;
		if (s == "xxalignat")
			return LM_OT_XXALIGNAT;
		if (s == "multline")
			return LM_OT_MULTLINE;
		if (s == "gather")
			return LM_OT_GATHER;
		return LM_OT_SIMPLE;
	}	
}


void MathMatrixInset::mutate(string const & newtype)
{
	if (newtype == "dump") {
		dump();
		return;
	}
	//lyxerr << "mutating from '" << getType() << "' to '" << newtype << "'\n";
	mutate(typecode(newtype));
}


void MathMatrixInset::glueall()
{
	MathArray ar;
	for (unsigned int i = 0; i < nargs(); ++i)
		ar.push_back(cell(i));
	*this = MathMatrixInset(LM_OT_SIMPLE);
	cell(0) = ar;
}


MathInsetTypes MathMatrixInset::getType() const
{
	return objtype_;
}


void MathMatrixInset::setType(MathInsetTypes t)
{
	objtype_ = t;
	setDefaults();
}



void MathMatrixInset::mutate(MathInsetTypes newtype)
{
	//lyxerr << "mutating from '" << getType() << "' to '" << newtype << "'\n";

	if (newtype == getType())
		return;

	switch (getType()) {
		case LM_OT_SIMPLE:
			setType(LM_OT_EQUATION);
			numbered(0, false);
			mutate(newtype);
			break;

		case LM_OT_EQUATION:
			switch (newtype) {
				case LM_OT_SIMPLE:
					setType(LM_OT_SIMPLE);
					break;

				case LM_OT_ALIGN: 
				case LM_OT_ALIGNAT:
				case LM_OT_XALIGNAT:
				case LM_OT_XXALIGNAT: {

					MathGridInset::addCol(1);

					// split it "nicely"
					unsigned int pos = firstRelOp(cell(0));	
					cell(1) = cell(0);
					cell(0).erase(pos, cell(0).size());
					cell(1).erase(0, pos);
					setType(LM_OT_ALIGN);
					mutate(newtype);
					break;
				}

				case LM_OT_EQNARRAY:
				default:
					MathGridInset::addCol(1);
					MathGridInset::addCol(1);

					// split it "nicely" on the firest relop
					unsigned int pos = firstRelOp(cell(0));	
					cell(1) = cell(0);
					cell(0).erase(pos, cell(0).size());
					cell(1).erase(0, pos);

					if (cell(1).size()) {
						cell(2) = cell(1);
						cell(1).erase(1, cell(1).size());
						cell(2).erase(0);
					}

					setType(LM_OT_EQNARRAY);
					mutate(newtype);
					break;
				}
			break;

		case LM_OT_EQNARRAY:
			switch (newtype) {
				case LM_OT_SIMPLE:
				case LM_OT_EQUATION: {
					// set correct (no)numbering
					bool allnonum = true;
					for (unsigned int row = 0; row < nrows(); ++row) {
						if (!nonum_[row])
							allnonum = false;
					}

					// set first non-empty label
					string label;
					for (unsigned int row = 0; row < nrows(); ++row) {
						if (!label_[row].empty()) {
							label = label_[row];
							break;
						}
					}

					glueall();

					nonum_[0] = allnonum;
					label_[0] = label;
					mutate(newtype);
					break;
				}

				case LM_OT_ALIGN:
				case LM_OT_ALIGNAT:
				case LM_OT_XALIGNAT:
				case LM_OT_XXALIGNAT:
				default: {
					for (unsigned int row = 0; row < nrows(); ++row) {
						unsigned int c = 3 * row + 1;
						cell(c).push_back(cell(c + 1));
					}
					MathGridInset::delCol(2);
					setType(LM_OT_ALIGN);
					mutate(newtype);
					break;
				}
			}
			break;

		case LM_OT_ALIGN:
			switch (newtype) {
				case LM_OT_SIMPLE:
				case LM_OT_EQUATION:
				case LM_OT_EQNARRAY:
					MathGridInset::addCol(1);
					setType(LM_OT_EQNARRAY);
					mutate(newtype);
					break;
				
				case LM_OT_ALIGNAT:
				case LM_OT_XALIGNAT:
				case LM_OT_XXALIGNAT:
					setType(newtype);
					break;

				default:
					lyxerr << "mutation from '" << getType()
						<< "' to '" << newtype << "' not implemented\n";
					break;
			}
			break;

		case LM_OT_MULTLINE:
			switch (newtype) {
				case LM_OT_GATHER:
					setType(LM_OT_GATHER);
					break;
				default:
					lyxerr << "mutation from '" << getType()
						<< "' to '" << newtype << "' not implemented\n";
					break;
			}

		case LM_OT_GATHER:
			switch (newtype) {
				case LM_OT_MULTLINE:
					setType(LM_OT_MULTLINE);
					break;
				default:
					lyxerr << "mutation from '" << getType()
						<< "' to '" << newtype << "' not implemented\n";
					break;
			}

		default:
			lyxerr << "mutation from '" << getType()
				<< "' to '" << newtype << "' not implemented\n";
	}
}