/*
For general Scribus (>=1.3.2) copyright and licensing information please refer
to the COPYING file provided with the program. Following this notice may exist
a copyright and/or license notice that predates the release of Scribus 1.3.2
for which a new license (GPL+exception) is in place.
*/
/***************************************************************************
    begin                : Jun 2005
    copyright            : (C) 2005 by Craig Bradney
    email                : cbradney@zip.com.au
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
#include "tocgenerator.h"

#include <QMap>
#include <QString>

#include "scribusdoc.h"
#include "pagestructs.h"
#include "pageitem.h"
#include "gtparagraphstyle.h"
#include "gtframestyle.h"
#include "gtwriter.h"
#include "page.h"

TOCGenerator::TOCGenerator(QObject *parent, ScribusDoc *doc) : QObject(parent)
{
	currDoc=doc;
}


TOCGenerator::~TOCGenerator()
{
}

void TOCGenerator::setDoc(ScribusDoc *doc)
{
	currDoc=doc;
}

PageItem* TOCGenerator::findTargetFrame(const QString &targetFrameName)
{
	PageItem* targetFrame=NULL;
	if (currDoc!=NULL)
	{
		for (int d = 0; d < currDoc->DocItems.count(); ++d)
		{
			if (currDoc->DocItems.at(d) !=NULL )
			{
				if (currDoc->DocItems.at(d)->itemType()==PageItem::TextFrame && currDoc->DocItems.at(d)->itemName()==targetFrameName)
				{
					targetFrame=currDoc->DocItems.at(d);
					break;
				}
			}
		}
	}
	return targetFrame;
}

void TOCGenerator::generateDefault()
{
	if (currDoc==NULL)
		return;
	Q_ASSERT(!currDoc->masterPageMode());
	for(ToCSetupVector::Iterator tocSetupIt = currDoc->docToCSetups.begin() ; tocSetupIt != currDoc->docToCSetups.end(); ++tocSetupIt )
	{
		PageItem* tocFrame=findTargetFrame((*tocSetupIt).frameName);
		if (tocFrame!=NULL)
		{
			PageItem *currentDocItem;
			QMap<QString, QString> tocMap;
			tocMap.clear();
			uint *pageCounter = new uint[currDoc->DocPages.count()];
			if (pageCounter==NULL)
				return;
			uint pageNumberWidth=QString("%1").arg(currDoc->DocPages.count()).length();
			for (int i=0;i<currDoc->DocPages.count();++i)
				pageCounter[i]=0;
			int maxDataWidth=0;
			for (int d = 0; d < currDoc->DocItems.count(); ++d)
			{
				currentDocItem = currDoc->DocItems.at(d);
				if (currentDocItem!=NULL)
				{
					//Item not on a page, continue
					if (currentDocItem->OwnPage==-1)
						continue;
					//If we dont want to list non printing frames and this one is set to not print, continue
					if (!(*tocSetupIt).listNonPrintingFrames && !currentDocItem->printEnabled())
						continue;
					ObjectAttribute objattr=currentDocItem->getObjectAttribute((*tocSetupIt).itemAttrName);
					if (!objattr.name.isNull())
					{
						//The key is generated to produce a sequence of numbers for the page numbers
						//First is the page of the item
						//Second is an incremented counter for the item so multiple per page works
						//Third is the section based page number which is actually used in the TOC.
						QString key=QString("%1,%2,%3").arg(currentDocItem->OwnPage + currDoc->FirstPnum, pageNumberWidth).arg(pageCounter[currentDocItem->OwnPage]++).arg(currDoc->getSectionPageNumberForPageIndex(currentDocItem->OwnPage));
						tocMap.insert(key, objattr.value);
						if (objattr.value.length()>maxDataWidth)
							maxDataWidth=objattr.value.length();
					}
				}
			}
			//Set up the gtWriter instance with the selected paragraph style
			gtWriter* writer = new gtWriter(false, tocFrame);
			if (writer!=NULL)
			{
				writer->setUpdateParagraphStyles(false);
				writer->setOverridePStyleFont(false);
				gtFrameStyle* fstyle = writer->getDefaultStyle();
				gtParagraphStyle* pstyle = new gtParagraphStyle(*fstyle);
				pstyle->setName((*tocSetupIt).textStyle);
				writer->setParagraphStyle(pstyle);
				
				QString oldTocPage=QString::null;
				for (QMap<QString, QString>::Iterator tocIt=tocMap.begin();tocIt!=tocMap.end();++tocIt)
				{
					QString tocPage(tocIt.key().section( ',', 2, 2 ).trimmed());
					QString tocLine;
					//Start with text or numbers
					if ((*tocSetupIt).pageLocation==End || (*tocSetupIt).pageLocation==NotShown)
						tocLine = tocIt.value();
					if ((*tocSetupIt).pageLocation==Beginning && oldTocPage!=tocPage)
						tocLine = tocPage;
					//Add in the tab for the leaders
					tocLine+="\t";
					//End with text or numbers
					if ((*tocSetupIt).pageLocation==Beginning)
						tocLine += tocIt.value();
					if ((*tocSetupIt).pageLocation==End && oldTocPage!=tocPage)
						tocLine += tocPage;
					tocLine += "\n";
					writer->append(tocLine);
				}
				delete writer;
			}
			delete[] pageCounter;
		}
	}
}