#include <config.h>

#ifdef __GNUG__
#pragma implementation
#endif

#include "debug.h"
#include "array.h"
#include "math_inset.h"
#include "math_scriptinset.h"
#include "math_parser.h"
#include "mathed/support.h"

using namespace std;


MathArray::MathArray()
{}


MathArray::~MathArray()
{
	for (int pos = 0; pos < size(); next(pos)) 
		if (MathIsInset(pos)) 
			delete GetInset(pos);
}


MathArray::MathArray(MathArray const & array)
	: bf_(array.bf_)
{
	for (int pos = 0; pos < size(); next(pos)) 
		if (isInset(pos)) 
			replace(pos, GetInset(pos)->Clone());
}


bool MathArray::next(int & pos) const
{
	if (pos >= size() - 1)
		return false;

	pos += item_size(pos);
	return true;
}


bool MathArray::prev(int & pos) const
{
	if (pos == 0)
		return false;

	pos -= item_size(pos - 1);
	return true;
}


bool MathArray::last(int & pos) const
{
	pos = bf_.size();
	return prev(pos);
}


int MathArray::item_size(int pos) const
{
	return 2 + (isInset(pos) ? sizeof(MathInset*) : 1);
}
		


void MathArray::substitute(MathMacro const & m)
{
	MathArray tmp;
	for (int pos = 0; pos < size(); next(pos)) {
		if (isInset(pos)) 
			GetInset(pos)->substitute(tmp, m);
		else 
			tmp.push_back(GetChar(pos), GetCode(pos));
	}
	swap(tmp);
}


MathArray & MathArray::operator=(MathArray const & array)
{
	MathArray tmp(array);
	swap(tmp);
	return *this;
}


MathInset * MathArray::GetInset(int pos) const
{
	if (!isInset(pos))
		return 0;
	MathInset * p;
	memcpy(&p, bf_.begin() + pos + 1, sizeof(p));
	return p;
}

byte MathArray::GetChar(int pos) const
{
	return pos < size() ? bf_[pos + 1] : '\0';
}

MathTextCodes MathArray::GetCode(int pos) const
{
	return pos < size() ? MathTextCodes(bf_[pos]) : LM_TC_MIN;
}

void MathArray::insert(int pos, MathInset * p)
{
	bf_.insert(bf_.begin() + pos, 2 + sizeof(p), LM_TC_INSET);
	memcpy(&bf_[pos + 1], &p, sizeof(p));
}


void MathArray::replace(int pos, MathInset * p)
{
	memcpy(&bf_[pos + 1], &p, sizeof(p));
}

void MathArray::insert(int pos, byte b, MathTextCodes t)
{
	bf_.insert(bf_.begin() + pos, 3, t);
	bf_[pos + 1] = b;
}


void MathArray::insert(int pos, MathArray const & array)
{
#ifdef WITH_WARNINGS
#warning quick and really dirty: make sure that we really own our insets
#endif
	MathArray a = array;
	bf_.insert(bf_.begin() + pos, a.bf_.begin(), a.bf_.end());
}


void MathArray::push_back(MathInset * p)
{	
	insert(size(), p);
}

void MathArray::push_back(byte b, MathTextCodes c)
{
	insert(size(), b, c);
}

void MathArray::push_back(MathArray const & array)
{
	insert(size(), array);
}



void MathArray::clear()
{
	bf_.clear();
}


void MathArray::swap(MathArray & array)
{
	if (this != &array) 
		bf_.swap(array.bf_);
}


bool MathArray::empty() const
{
	return bf_.empty();
}
   

int MathArray::size() const
{
	return bf_.size();
}


void MathArray::erase(int pos)
{
	if (pos < static_cast<int>(bf_.size()))
		erase(pos, pos + item_size(pos));
}


void MathArray::erase(int pos1, int pos2)
{
	bf_.erase(bf_.begin() + pos1, bf_.begin() + pos2);
}


bool MathArray::isInset(int pos) const
{
	if (pos >= size())
		return false;
	return MathIsInset(bf_[pos]);
}


MathInset * MathArray::back_inset() const
{
	if (!empty()) {
		int pos = size();
		prev(pos);
		if (isInset(pos))
			return GetInset(pos);
	}
	return 0;
}


MathScriptInset * MathArray::prevScriptInset(int pos) const
{
	if (!pos)
		return 0;
	prev(pos);

	MathInset * inset = GetInset(pos);
	if (inset && inset->isScriptInset()) 
		return static_cast<MathScriptInset *>(inset);

	return 0;
}

MathScriptInset * MathArray::nextScriptInset(int pos) const
{
	MathInset * inset = GetInset(pos);
	if (inset && inset->isScriptInset()) 
		return static_cast<MathScriptInset *>(inset);

	return 0;
}


void MathArray::dump2(ostream & os) const
{
	for (buffer_type::const_iterator it = bf_.begin(); it != bf_.end(); ++it)
		os << int(*it) << ' ';
	os << endl;
}



void MathArray::dump(ostream & os) const
{
	for (int pos = 0; pos < size(); next(pos)) {
		if (isInset(pos)) 
			os << "<inset: " << GetInset(pos) << ">";
		else 
			os << "<" << int(bf_[pos]) << " " << int(bf_[pos+1]) << ">";
	}
}


std::ostream & operator<<(std::ostream & os, MathArray const & ar)
{
	ar.dump2(os);
	return os;
}


void MathArray::Write(ostream & os, bool fragile) const
{
	if (empty())
		return;

	int brace = 0;
	
	for (int pos = 0; pos < size(); next(pos)) {
		if (isInset(pos)) {

			GetInset(pos)->Write(os, fragile);

		} else {

			MathTextCodes fcode = GetCode(pos);
			byte c = GetChar(pos);

			if (MathIsSymbol(fcode)) {
				latexkeys const * l = lm_get_key_by_id(c, LM_TK_SYM);

				if (l == 0) {
					l = lm_get_key_by_id(c, LM_TK_BIGSYM);
				}

				if (l) {
					os << '\\' << l->name << ' ';
				} else {
					lyxerr << "Could not find the LaTeX name for  " << c << " and fcode " << fcode << "!" << std::endl;
				}
			} else {
				if (fcode >= LM_TC_RM && fcode <= LM_TC_TEXTRM) 
					os << '\\' << math_font_name[fcode - LM_TC_RM] << '{';

				// Is there a standard logical XOR?
				if ((fcode == LM_TC_TEX && c != '{' && c != '}') ||
						(fcode == LM_TC_SPECIAL))
					os << '\\';
				else {
					if (c == '{')
						++brace;
					if (c == '}')
						--brace;
				}
				if (c == '}' && fcode == LM_TC_TEX && brace < 0) 
					lyxerr <<"Math warning: Unexpected closing brace.\n";
				else	       
					os << c;
			}

			if (fcode >= LM_TC_RM && fcode <= LM_TC_TEXTRM)
				os << '}';
			
		}
	}

	if (brace > 0)
		os << string(brace, '}');
}


void MathArray::WriteNormal(ostream & os) const
{
	if (empty()) {
		os << "[par] ";
		return;
	}

	Write(os, true);
}