#include "branchobj.h"

#include "texteditor.h"

#include "mapeditor.h"

#include "mainwindow.h"

extern TextEditor *textEditor;

extern Main *mainWindow;

extern FlagRowObj *standardFlagsDefault;

extern QAction *actionEditOpenURL;

/////////////////////////////////////////////////////////////////

// BranchObj

/////////////////////////////////////////////////////////////////

BranchObj* BranchObj::itLast=NULL;

BranchObj::BranchObj () :OrnamentedObj()

{

//    cout << "Const BranchObj ()\n";

    setParObj (this);	

    init();

    depth=-1;

}

BranchObj::BranchObj (QCanvas* c):OrnamentedObj (c)

{

//    cout << "Const BranchObj (c)  called from MapCenterObj (c)\n";

	parObj=NULL;

    canvas=c;

}

BranchObj::BranchObj (QCanvas* c, LinkableMapObj* p):OrnamentedObj (c)

{

//    cout << "Const BranchObj (c,p)\n";

    canvas=c;

    setParObj (p);	

    depth=p->getDepth()+1;

	if (depth==1)

		// Calc angle to mapCenter if I am a mainbranch

		// needed for reordering the mainbranches clockwise 

		// around mapcenter 

		angle=getAngle (QPoint ((int)(x() - parObj->getChildPos().x() ), 

								(int)(y() - parObj->getChildPos().y() ) ) );

    init();

}

BranchObj::~BranchObj ()

{

//	cout << "Destr BranchObj of "<<this<<endl;

	// Check, if this branch was the last child to be deleted

	// If so, unset the scrolled flags

	BranchObj *po=(BranchObj*)(parObj);

	BranchObj *bo;

	if (po)

	{

		bo=((BranchObj*)(parObj))->getLastBranch();

		if (!bo) po->unScroll();

	}

	clear();

}

bool BranchObj::operator< ( const BranchObj & other )

{

    return  angle < other.angle;

}

bool BranchObj::operator== ( const BranchObj & other )

{

    return angle == other.angle;

}

int BranchObjPtrList::compareItems ( QPtrCollection::Item i, QPtrCollection::Item j)

{

	// Make sure PtrList::find works

	if (i==j) return 0;

	if ( ((BranchObj*)(i))->angle > ((BranchObj*)(j))->angle )

		return 1;

	else

		return -1;

}

void BranchObj::init () 

{

    branch.setAutoDelete (false);

    floatimage.setAutoDelete (true);

    xlink.setAutoDelete (false);

	if (parObj)

	{

		absPos=getRandPos();

		absPos+=parObj->getChildPos();

	}

    lastSelectedBranch=-1;

    setChildObj(this);

	scrolled=false;

	tmpUnscrolled=false;

	includeImagesVer=false;

	includeImagesHor=false;

	url="";

	vymLink="";

}

void BranchObj::copy (BranchObj* other)

{

    OrnamentedObj::copy(other);

	branch.clear();

    BranchObj* b;

    for (b=other->branch.first(); b;b=other->branch.next() ) 

		// Make deep copy of b

		// Because addBranch again calls copy for the childs,

		// Those will get a deep copy, too

		addBranch(b);	

	FloatImageObj *fi;

	for (fi=other->floatimage.first(); fi;fi=other->floatimage.next() )

		addFloatImage (fi);

	scrolled=other->scrolled;

	tmpUnscrolled=other->tmpUnscrolled;

	setVisibility (other->visible);

	url=other->url;

	vymLink=other->vymLink;

	angle=other->angle;

    positionBBox();

}

void BranchObj::clear() 

{

	floatimage.clear();

	while (!xlink.isEmpty())

		deleteXLink (xlink.first() );

	BranchObj *bo;

	while (!branch.isEmpty())

	{

		bo=branch.first();

		branch.removeFirst();

		delete (bo);

	}

}

int BranchObj::getNum()

{

	if (parObj)

		return ((BranchObj*)(parObj))->getNum ((BranchObj*)(this));

	else

		return 0;

}

int BranchObj::getNum(BranchObj *bo)

{

	// keep current pointer in branch, 

	// otherwise save might fail

	int cur=branch.at();

	int ind=branch.findRef (bo);

	branch.at(cur);

	return ind;

}

int BranchObj::getFloatImageNum(FloatImageObj *fio)

{

	return floatimage.findRef (fio);

}

int BranchObj::countBranches()

{

	return branch.count();

}

int BranchObj::countFloatImages()

{

	return floatimage.count();

}

int BranchObj::countXLinks()

{

	return xlink.count();

}

void BranchObj::setParObjTmp(LinkableMapObj* lmo, QPoint m, int off)

{

	// Temporary link to lmo

	// m is position of mouse pointer 

	// offset 0: default 1: below lmo   -1 above lmo  (if possible)

	BranchObj* o=(BranchObj*)(lmo);

	if (!parObjTmpBuf) 

		parObjTmpBuf=parObj;

	// ignore mapcenter and mainbranch

	if (lmo->getDepth()<2) off=0;

	if (off==0)

		link2ParPos=false;

	else

		link2ParPos=true;

	parObj=o;

	depth=parObj->getDepth()+1;

	// setLinkStyle calls updateLink, only set it once

	if (style!=getDefLinkStyle() ) setLinkStyle (getDefLinkStyle());

	// Move temporary to new position at destination

	// Usually the positioning would be done by reposition(),

	// but then also the destination branch would "Jump" around...

	// Better just do it approximately

	if (depth==1)

	{	// new parent is the mapcenter itself

		QPoint p= normalise ( QPoint (m.x() - o->getChildPos().x(),

									  m.y() - o->getChildPos().y() ));

		if (p.x()<0) p.setX( p.x()-bbox.width() );

		move2RelPos (p);

	} else

	{	

		int y;

		if (off==0)

		{

			// new parent is just a branch, link to it

			QRect t=o->getBBoxSizeWithChilds();

			if (o->getLastBranch())

				y=t.y() + t.height() ;

			else

				y=t.y();

		} else

		{

			if (off<0)

				// we want to link above lmo

				y=o->y() - height() + 5;

			else	

				// we want to link below lmo

				// Bottom of sel should be 5 pixels above

				// the bottom of the branch _below_ the target:

				// Don't try to find that branch, guess 12 pixels

				y=o->getChildPos().y()  -height() + 12; 

		}	

		if (o->getOrientation()==OrientLeftOfCenter)

			move ( o->getChildPos().x() - linkwidth, y );

		else	

			move (o->getChildPos().x() + linkwidth, y );

	}	

	// updateLink is called implicitly in move

	reposition();	// FIXME shouldn't be this a request?

}

void BranchObj::unsetParObjTmp()

{

	if (parObjTmpBuf) 

	{

		link2ParPos=false;

		parObj=parObjTmpBuf;

		parObjTmpBuf=NULL;

		depth=parObj->getDepth()+1;

		setLinkStyle (getDefLinkStyle() );

		updateLink();

	}		

}

void BranchObj::unScroll()

{

	if (tmpUnscrolled) resetTmpUnscroll();

	if (scrolled) toggleScroll();

}

void BranchObj::toggleScroll()

{

	BranchObj *bo;

	if (scrolled)

	{

		scrolled=false;

		systemFlags->deactivate("scrolledright");

		for (bo=branch.first(); bo; bo=branch.next() )

		{

			bo->setVisibility(true);

		}

	} else

	{

		scrolled=true;

		systemFlags->activate("scrolledright");

		for (bo=branch.first(); bo; bo=branch.next() )

		{

			bo->setVisibility(false);

		}

	}

	calcBBoxSize();

	positionBBox();	

	move (absPos.x(), absPos.y() );

	forceReposition();

}

bool BranchObj::isScrolled()

{

	return scrolled;

}

bool BranchObj::hasScrolledParent(BranchObj *start)

{

	// Calls parents recursivly to

	// find out, if we are scrolled at all.

	// But ignore myself, just look at parents.

	if (this !=start && scrolled) return true;

	BranchObj* bo=(BranchObj*)(parObj);

	if (bo) 

		return bo->hasScrolledParent(start);

	else

		return false;

}

void BranchObj::tmpUnscroll()

{

	// Unscroll parent (recursivly)

	BranchObj* bo=(BranchObj*)(parObj);

	if (bo) bo->tmpUnscroll();

	// Unscroll myself

	if (scrolled)

	{

		tmpUnscrolled=true;

		systemFlags->activate("tmpUnscrolledright");

		toggleScroll();

	}	

}

void BranchObj::resetTmpUnscroll()

{

	// Unscroll parent (recursivly)

	BranchObj* bo=(BranchObj*)(parObj);

	if (bo)

		bo->resetTmpUnscroll();

	// Unscroll myself

	if (tmpUnscrolled)

	{

		tmpUnscrolled=false;

		systemFlags->deactivate("tmpUnscrolledright");

		toggleScroll();

	}	

}

void BranchObj::setVisibility(bool v, int toDepth)

{

    if (depth <= toDepth)

    {

		frame->setVisibility(v);

		heading->setVisibility(v);

		systemFlags->setVisibility(v);

		standardFlags->setVisibility(v);

		LinkableMapObj::setVisibility (v);

		if (!scrolled && (depth < toDepth))

		{

			// Now go recursivly through all childs

			BranchObj* b;

			for (b=branch.first(); b;b=branch.next() ) 

				b->setVisibility (v,toDepth);	

			FloatImageObj *fio;

			for (fio=floatimage.first(); fio; fio=floatimage.next())

				fio->setVisibility (v);

			XLinkObj* xlo;

			for (xlo=xlink.first(); xlo;xlo=xlink.next() ) 

				xlo->setVisibility ();	

		}

    } // depth <= toDepth	

	requestReposition();

}	

void BranchObj::setVisibility(bool v)

{

    setVisibility (v,MAX_DEPTH);

}

void BranchObj::setLinkColor ()

{

	// Overloaded from LinkableMapObj

	// BranchObj can use color of heading

	if (mapEditor)

		if (mapEditor->getLinkColorHint()==HeadingColor)

			LinkableMapObj::setLinkColor (heading->getColor() );

		else	

			LinkableMapObj::setLinkColor ();

}

void BranchObj::setColor (QColor col, bool colorChilds)

{

    heading->setColor(col);

	setLinkColor();

    if (colorChilds) 

    {

		BranchObj *bo;

		for (bo=branch.first(); bo; bo=branch.next() )

			bo->setColor(col,colorChilds);

    }	

}

QColor BranchObj::getColor()

{

	return heading->getColor();

}

BranchObj* BranchObj::first()

{

	itLast=NULL;	

	return this; 

}

BranchObj* BranchObj::next()

{

	BranchObj *lmo;

	BranchObj *bo=branch.first();

	BranchObj *po=(BranchObj*)(parObj);

	if (!itLast)

	{	// We are just beginning at the mapCenter

		if (bo) 

		{

			itLast=this;

			return bo;

		}	

		else

		{

			itLast=NULL;

			return NULL;

		}	

	}

	if (itLast==parObj)

	{	// We come from above

		if (bo)

		{

			// there are childs, go there

			itLast=this;

			return bo;

		}	

		else

		{	// no childs, try to go up again

			if (po)

			{

				// go up

				itLast=this;

				lmo=po->next();

				itLast=this;

				return lmo;

			}	

			else

			{

				// can't go up, I am mapCenter

				itLast=NULL;

				return NULL;

			}	

		}

	}

	// Try to find last child, we came from, in my own childs

	bool searching=true;

	while (bo && searching)

	{

		if (itLast==bo) searching=false;

		bo=branch.next();

	}

	if (!searching)

	{	// found lastLMO in my childs

		if (bo)

		{

			// found a brother of lastLMO 

			itLast=this;

			return bo;

		}	

		else

		{

			if (po)

			{

				// go up

				itLast=this;

				lmo=po->next();

				itLast=this;

				return lmo;

			}

			else

			{

				// can't go up, I am mapCenter

				itLast=NULL;

				return NULL;

			}	

		}

	}

	// couldn't find last child, it must be a nephew of mine

	bo=branch.first();

	if (bo)

	{

		// proceed with my first child

		itLast=this;	

		return bo;

	}	

	else

	{

		// or go back to my parents

		if (po)

		{

			// go up

			itLast=this;

			lmo=po->next();

			itLast=this;

			return lmo;

		}	

		else

		{

			// can't go up, I am mapCenter

			itLast=NULL;

			return NULL;

		}	

	}	

}

BranchObj* BranchObj::getLastIterator()

{

	return itLast;

}

void BranchObj::setLastIterator(BranchObj* it)

{

	itLast=it;

}

void BranchObj::move (double x, double y)

{

	OrnamentedObj::move (x,y);

	FloatImageObj *fio;

    for (fio=floatimage.first(); fio; fio=floatimage.next() )

		fio->reposition();

    positionBBox();

}

void BranchObj::move (QPoint p)

{

	move (p.x(), p.y());

}

void BranchObj::moveBy (double x, double y)

{

	OrnamentedObj::moveBy (x,y);

    positionBBox();

    BranchObj* b;

    for (b=branch.first(); b;b=branch.next() ) 

		b->moveBy (x,y);

}

void BranchObj::moveBy (QPoint p)

{

	moveBy (p.x(), p.y());

}

void BranchObj::positionBBox()

{

	/*// TODO testing (optimization)

	QString h=getHeading();

	if (!h.isEmpty())

		cout << "BO::positionBBox("<<h<<")\n";

	else	

		cout << "BO::positionBBox (noHeading)\n";

*/		

	positionContents();

	int d=frame->getBorder()/2;

	bbox.moveTopLeft (QPoint (absPos.x(), absPos.y() - topPad));

	clickBox.moveTopLeft(QPoint (absPos.x()+d, absPos.y()+d ));

	setSelBox();

	// set the frame

	frame->setRect(QRect(bbox.x(),bbox.y(),bbox.width(),bbox.height() ) );

	// Update links to other branches

	XLinkObj *xlo;

    for (xlo=xlink.first(); xlo; xlo=xlink.next() )

		xlo->updateXLink();

}

void BranchObj::calcBBoxSize()

{

    QSize heading_r=heading->getSize();

    int heading_w=static_cast <int> (heading_r.width() );

    int heading_h=static_cast <int> (heading_r.height() );

    QSize sysflags_r=systemFlags->getSize();

	int sysflags_h=sysflags_r.height();

	int sysflags_w=sysflags_r.width();

    QSize stanflags_r=standardFlags->getSize();

	int stanflags_h=stanflags_r.height();

	int stanflags_w=stanflags_r.width();

    int w;

    int h;

	// set width to sum of all widths

	w=heading_w + sysflags_w + stanflags_w;

	// set height to maximum needed height

	h=max (sysflags_h,stanflags_h);

	h=max (h,heading_h);

    clickBox.setSize (QSize (w,h));

	// Floatimages 

	QPoint rp;

	FloatImageObj *foi;

	topPad=botPad=leftPad=rightPad=0;

	if (includeImagesVer || includeImagesHor)

	{

		if (countFloatImages()>0)

		{

			for (foi=floatimage.first(); foi; foi=floatimage.next() )

			{

				rp=foi->getRelPos();

				if (includeImagesVer)

				{

					if (rp.y() < 0) 

						topPad=max (topPad,-rp.y());

					if (rp.y()+foi->height() > h)

						botPad=max (botPad,rp.y()+foi->height()-h);

				}		

			}	

		}	

		h+=topPad+botPad;

	}

	// Frame thickness

    w+=frame->getBorder();

    h+=frame->getBorder();

	// Finally set size

    bbox.setSize (QSize (w,h));

}

LinkableMapObj* BranchObj::findMapObj(QPoint p, LinkableMapObj* excludeLMO)

{

	// Search branches

    BranchObj *b;

    LinkableMapObj *lmo;

    for (b=branch.first(); b; b=branch.next() )

    {	

		lmo=b->findMapObj(p, excludeLMO);

		if (lmo != NULL) return lmo;

    }

	// Search myself

    if (inBox (p) && (this != excludeLMO) && isVisibleObj() ) 

		return this;

	// Search float images

	FloatImageObj *foi;

    for (foi=floatimage.first(); foi; foi=floatimage.next() )

		if (foi->inBox(p) && 

			(foi != excludeLMO) && 

			foi->getParObj()!= excludeLMO &&

			foi->isVisibleObj() 

		) return foi;

    return NULL;

}

void BranchObj::setHeading(QString s)

{

    heading->setText(s);	// set new heading

	calcBBoxSize();			// recalculate bbox

    positionBBox();			// rearrange contents

	requestReposition();

}

void BranchObj::setURL(QString s)

{

	url=s;

	if (!url.isEmpty())

		systemFlags->activate("url");

	else	

		systemFlags->deactivate("url");

	calcBBoxSize();			// recalculate bbox

    positionBBox();			// rearrange contents

	forceReposition();

}

QString BranchObj::getURL()

{

	return url;

}

void BranchObj::setVymLink(QString s)

{

	if (!s.isEmpty())

	{

		// We need the relative (from loading) 

		// or absolute path (from User event)

		// and build the absolute path.

		// Note: If we have relative, use path of

		// current map to build absolute path

		QDir d(s);

		if (!d.path().startsWith ("/"))

		{

			QString p=mapEditor->getDestPath();

			int i=p.findRev("/",-1);

			d.setPath(p.left(i)+"/"+s);

			d.convertToAbs();

		}

		vymLink=d.path();

		systemFlags->activate("vymLink");

	}	

	else	

	{

		systemFlags->deactivate("vymLink");

		vymLink="";

	}	

	calcBBoxSize();			// recalculate bbox

    positionBBox();			// rearrange contents

	forceReposition();

}

QString BranchObj::getVymLink()

{

	return vymLink;

}

QString BranchObj::saveToDir (const QString &tmpdir,const QString &prefix, const QPoint& offset)

{

    QString s,a;

	QString scrolledAttr;

	if (scrolled) 

		scrolledAttr=attribut ("scrolled","yes");

	else

		scrolledAttr="";

	QString posAttr;

	if (depth<2) posAttr=

		attribut("absPosX",QString().setNum(absPos.x(),10)) +

		attribut("absPosY",QString().setNum(absPos.y(),10)); 

	else

		posAttr="";

	QString linkAttr=getLinkAttr()+" "+getIncludeImageAttr();

	QString urlAttr;

	if (!url.isEmpty())

		urlAttr=attribut ("url",url);

	QString vymLinkAttr;

	if (!vymLink.isEmpty())

		vymLinkAttr=attribut ("vymLink",convertToRel(mapEditor->getDestPath(),vymLink) );

	QString frameAttr;

	if (frame->getFrameType()!=NoFrame)

		frameAttr=attribut ("frameType",frame->getFrameTypeName());

	else

		frameAttr="";

	// save area, if not scrolled

	QString areaAttr;

	if (!((BranchObj*)(parObj))->isScrolled() )

	{

		areaAttr=

			attribut("x1",QString().setNum(absPos.x()-offset.x(),10)) +

			attribut("y1",QString().setNum(absPos.y()-offset.y(),10)) +

			attribut("x2",QString().setNum(absPos.x()+width()-offset.x(),10)) +

			attribut("y2",QString().setNum(absPos.y()+height()-offset.y(),10));

	} else

		areaAttr="";

	// Providing an ID for a branch makes export to XHTML easier

	QString idAttr;

	if (countXLinks()>0)

		idAttr=attribut ("id",getSelectString());

	else

		idAttr="";

    s=beginElement ("branch" +scrolledAttr +posAttr +linkAttr +urlAttr +vymLinkAttr +frameAttr +areaAttr +idAttr);

    incIndent();

	// save heading

    s+=valueElement("heading", getHeading(),

		attribut ("textColor",QColor(heading->getColor()).name()));

	// save names of flags set

	s+=standardFlags->saveToDir(tmpdir,prefix,0);

	// save note

	if (!note.isEmpty() )

		s+=note.saveToDir();

	// Save branches

    BranchObj *bo;

    for (bo=branch.first(); bo; bo=branch.next() )

		s+=bo->saveToDir(tmpdir,prefix,offset);

	// Save FloatImages

	FloatImageObj *fio;

	for (fio=floatimage.first(); fio; fio=floatimage.next() )

		s+=fio->saveToDir (tmpdir,prefix,offset);

	// Save XLinks

	XLinkObj *xlo;

    for (xlo=xlink.first(); xlo; xlo=xlink.next() )

		s+=xlo->saveToDir();

    decIndent();

    s+=endElement   ("branch");

    return s;

}

void BranchObj::addXLink (XLinkObj *xlo)

{

	xlink.append (xlo);

}

void BranchObj::removeXLinkRef (XLinkObj *xlo)

{

	xlink.remove (xlo);

}

void BranchObj::deleteXLink(XLinkObj *xlo)

{

	xlo->deactivate();

	if (!xlo->isUsed()) delete (xlo);

}

void BranchObj::deleteXLinkAt (int i)

{

	XLinkObj *xlo=xlink.at(i);

	xlo->deactivate();

	if (!xlo->isUsed()) delete(xlo);

}

XLinkObj* BranchObj::XLinkAt (int i)

{

	return xlink.at(i);

}

int BranchObj::countXLink()

{

	return xlink.count();

}

BranchObj* BranchObj::XLinkTargetAt (int i)

{

	if (xlink.at(i))

		return xlink.at(i)->otherBranch (this);

	else

		return NULL;

}

void BranchObj::setIncludeImagesVer(bool b)

{

	includeImagesVer=b;

	calcBBoxSize();

	positionBBox();

	requestReposition();

	// FIMXE undo needed

}

bool BranchObj::getIncludeImagesVer()

{

	return includeImagesVer;

}

void BranchObj::setIncludeImagesHor(bool b)

{

	includeImagesHor=b;

	calcBBoxSize();

	positionBBox();

	requestReposition();

	// FIMXE undo needed

}

bool BranchObj::getIncludeImagesHor()

{

	return includeImagesHor;

}

QString BranchObj::getIncludeImageAttr()

{

	QString a;

	if (includeImagesVer)

		a=attribut ("incImgV","true");

	else

		a=attribut ("incImgV","false");

	if (includeImagesHor)

		a+=" "+attribut ("incImgH","true");

	else

		a+=" "+attribut ("incImgH","false");

	return a;	

}

LinkableMapObj* BranchObj::addFloatImage ()

{

	FloatImageObj *newfi=new FloatImageObj (canvas,this);

	floatimage.append (newfi);

	if (hasScrolledParent(this) )

		newfi->setVisibility (false);

	else	

		newfi->setVisibility(visible);

	calcBBoxSize();

	positionBBox();

	requestReposition();

	return newfi;

	// FIMXE undo needed

}

LinkableMapObj* BranchObj::addFloatImage (FloatImageObj *fio)

{

	FloatImageObj *newfi=new FloatImageObj (canvas,this);

	floatimage.append (newfi);

	newfi->copy (fio);

	if (hasScrolledParent(this) )

		newfi->setVisibility (false);

	else	

		newfi->setVisibility(visible);

	calcBBoxSize();

	positionBBox();

	requestReposition();

	return newfi;

	// FIMXE undo needed

}

FloatImageObj* BranchObj::getFirstFloatImage ()

{

    return floatimage.first();

}

FloatImageObj* BranchObj::getLastFloatImage ()

{

    return floatimage.last();

}

FloatImageObj* BranchObj::getFloatImageNum (const uint &i)

{

    return floatimage.at(i);

}

void BranchObj::removeFloatImage (FloatImageObj *fio)

{

	floatimage.remove (fio);

	calcBBoxSize();

	positionBBox();

	requestReposition();

	// FIMXE undo needed

}

void BranchObj::savePosInAngle ()

{

	// Save position in angle

    BranchObj *b;

	int i=0;

    for (b=branch.first(); b; b=branch.next() )

	{

		b->angle=i;

		i++;

	}

}

void BranchObj::setDefAttr (BranchModification mod)

{

	int fontsize;

	switch (depth)

	{

		case 0: fontsize=16; break;

		case 1: fontsize=12; break;

		default: fontsize=10; break;

	}