lyx_mirror/src/toc.C

// -*- C++ -*-
/* This file is part of
 * ======================================================
 *
 *           LyX, The Document Processor
 *
 *           Copyright 2002 The LyX Team.
 *
 * ======================================================
 *
 * \file toc.C
 * \author Angus Leeming <a.leeming@ic.ac.uk>
 * \author Jean-Marc Lasgouttes <lasgouttes@freesurf.fr>
 */

#include <config.h>

#ifdef __GNUG__
#pragma implementation
#endif

#include "support/lstrings.h"
#include "toc.h"
#include "buffer.h"
#include "frontends/LyXView.h"
#include "lyxfunc.h"
#include "LyXAction.h"
#include "paragraph.h"
#include "lyxtextclasslist.h"
#include "insets/insetfloat.h"
#include "debug.h"

using std::vector;

extern LyXAction lyxaction;

namespace toc
{

string const TocItem::asString() const
{
	return string(4 * depth, ' ') + str;
}

	
void TocItem::goTo(LyXView & lv_) const
{
	string const tmp = tostr(par->id());
	lv_.getLyXFunc()->dispatch(LFUN_GOTO_PARAGRAPH, tmp, false);
}


int TocItem::action() const
{
	return lyxaction.getPseudoAction(LFUN_GOTO_PARAGRAPH,
					 tostr(par->id()));
}


string const getType(string const & cmdName)
{
	// special case
	if (cmdName == "tableofcontents")
		return "TOC";
	else
		return cmdName;
}


TocList const getTocList(Buffer const * buf)
{
	TocList toclist;
	Paragraph * par = buf->paragraph;

	LyXTextClass const & textclass = textclasslist[buf->params.textclass];
	bool found = textclass.hasLayout("Caption");
	string const layout("Caption");

	while (par) {
#ifdef WITH_WARNINGS
#warning bogus type (Lgb)
#endif
		char const labeltype = par->layout()->labeltype;

		if (labeltype >= LABEL_COUNTER_CHAPTER
		    && labeltype <= LABEL_COUNTER_CHAPTER + buf->params.tocdepth) {
				// insert this into the table of contents
			const int depth = max(0, labeltype - textclass.maxcounter());
			TocItem const item(par, depth,
					   par->asString(buf, true));
			toclist["TOC"].push_back(item);
		}
		// For each paragraph, traverse its insets and look for
		// FLOAT_CODE

		if (found) {
			Paragraph::inset_iterator it =
				par->inset_iterator_begin();
			Paragraph::inset_iterator end =
				par->inset_iterator_end();

			for (; it != end; ++it) {
				if ((*it)->lyxCode() == Inset::FLOAT_CODE) {
					InsetFloat * il =
						static_cast<InsetFloat*>(*it);

					string const type = il->type();

					// Now find the caption in the float...
					// We now tranverse the paragraphs of
					// the inset...
					Paragraph * tmp = il->inset.paragraph();
					while (tmp) {
						if (tmp->layout()->name() == layout) {
							string const str =
								tostr(toclist[type].size()+1) + ". " + tmp->asString(buf, false);
							TocItem const item(tmp, 0 , str);
							toclist[type].push_back(item);
						}
						tmp = tmp->next();
					}
				}
			}
		} else {
			lyxerr << "Caption not found" << endl;
		}

		par = par->next();
	}
	return toclist;
}


vector<string> const getTypes(Buffer const * buffer)
{
	vector<string> types;

	TocList const tmp = getTocList(buffer);

	TocList::const_iterator cit = tmp.begin();
	TocList::const_iterator end = tmp.end();

	for (; cit != end; ++cit) {
		types.push_back(cit->first);
	}

	return types;
}


void asciiTocList(string const & type, Buffer const * buffer, ostream & os)
{
	TocList const toc_list = getTocList(buffer);
	TocList::const_iterator cit = toc_list.find(type);
	if (cit != toc_list.end()) {
		Toc::const_iterator ccit = cit->second.begin();
		Toc::const_iterator end = cit->second.end();
		for (; ccit != end; ++ccit)
			os << ccit->asString() << '\n';
	}
}


} // namespace toc