#include "vymview.h"

#include "branchitem.h"

#include "mainwindow.h"

#include "mapeditor.h"

#include "treeeditor.h"

extern Main *mainWindow;

VymView::VymView(VymModel *m)

{

	model=m;

	// Create TreeView

	treeEditor=new TreeEditor (model);

	//treeEditor->setModel ((QAbstractItemModel*)model);

	//treeEditor->setMinimumWidth (50);

	treeEditor->setColumnWidth (0,150);

	treeEditor->setAnimated (true);

	selModel=new QItemSelectionModel (model);

	model->setSelectionModel (selModel);

	treeEditor->setSelectionModel (selModel);

	// Create good old MapEditor

	mapEditor=model->getMapEditor();

	if (!mapEditor) mapEditor=new MapEditor (model);

	// Create Layout 

	QVBoxLayout* mainLayout = new QVBoxLayout (this); //FIXME-4 not needed

	QSplitter *splitter= new QSplitter (this);

	QSizePolicy sizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

    splitter->setSizePolicy(sizePolicy);

	mainLayout->addWidget (splitter);

	// Connect selections

		// Selection in Model changed	

		connect (

			selModel, SIGNAL (selectionChanged(const QItemSelection &, const QItemSelection &)), 

			this,SLOT (changeSelection(const QItemSelection &,const QItemSelection &)));

		// Tell MapEditor to update selection

		/* FIXME-3 done implicit here in VymView

		connect (

			selModel, SIGNAL (selectionChanged(const QItemSelection &, const QItemSelection &)), 

			mapEditor,SLOT (updateSelection(const QItemSelection &,const QItemSelection &)));

			*/

		// Needed to update selbox during animation

		connect (

			model, SIGNAL (selectionChanged(const QItemSelection &, const QItemSelection &)), 

			mapEditor,SLOT (updateSelection(const QItemSelection &,const QItemSelection &)));

	// Connect data changed signals	

	connect (

		model, SIGNAL (dataChanged(const QModelIndex &, const QModelIndex &)), 

		mapEditor,SLOT (updateData(const QModelIndex &) ) );

	connect (

		model, SIGNAL (sortFilterChanged (const QString &)),

		treeEditor, SLOT (setSortFilter (const QString &) ) );

	connect (

		model, SIGNAL (noteHasChanged(QModelIndex) ),

		mainWindow, SLOT (updateNoteEditor (QModelIndex) ) );

	connect (

		model, SIGNAL (expandAll() ),

		this, SLOT (expandAll () ) );

	connect (

		model, SIGNAL (expandOneLevel() ),

		this, SLOT (expandOneLevel() ) );

	connect (

		model, SIGNAL (collapseOneLevel() ),

		this, SLOT (collapseOneLevel() ) );

	connect (

		model, SIGNAL (showSelection() ),

		this, SLOT (showSelection() ) );

	connect (

		model, SIGNAL (updateLayout() ),

		mapEditor, SLOT (autoLayout() ) );

	mapEditor->setAntiAlias (mainWindow->isAliased());

	mapEditor->setSmoothPixmap(mainWindow->hasSmoothPixmapTransform());

	splitter->addWidget (treeEditor);

	splitter->addWidget (mapEditor);

	// Set geometry

	QList <int> widths;

	widths<<200;

	widths<<600;

	splitter->setSizes(widths);

}

VymView::~VymView()

{

	//cout << "Destructor VymView\n";

}

VymModel* VymView::getModel()

{

	return model;

}

MapEditor* VymView::getMapEditor()

{

	return mapEditor;

}

void VymView::initFocus()

{

	mapEditor->setFocus();

}

void VymView::changeSelection (const QItemSelection &newsel, const QItemSelection &oldsel)	

{

	//cout << "VV::changeSelection   newsel.count="<<newsel.indexes().count()<<endl;

	mainWindow->changeSelection (model,newsel,oldsel);	

	mapEditor->updateSelection (newsel,oldsel);

	if (newsel.indexes().count()>0)

	{

		QModelIndex ix=newsel.indexes().first();

		selModel->setCurrentIndex (ix,QItemSelectionModel::ClearAndSelect  );

		treeEditor->setCurrentIndex (ix);

		showSelection();

	}

}

void VymView::expandAll()

{

	treeEditor->expandAll();

}

void VymView::expandOneLevel()

{

	int level=999999;

	int d;

	BranchItem *cur=NULL;

	BranchItem *prev=NULL;

	QModelIndex pix;

	// Find level to expand

	model->nextBranch(cur,prev);

	while (cur) 

	{

		pix=model->index (cur);

		d=cur->depth();

		if (!treeEditor->isExpanded(pix) && d < level)

			level=d;

		model->nextBranch(cur,prev);	

	}

	// Expand all to level

	cur=NULL;

	prev=NULL;

	model->nextBranch(cur,prev);

	while (cur) 

	{

		pix=model->index (cur);

		d=cur->depth();

		if (!treeEditor->isExpanded(pix) && d <= level && cur->branchCount()>0)

			treeEditor->setExpanded(pix,true);

		model->nextBranch(cur,prev);	

	}

	/* FIXME-3 optimize expanding by using flat version of next

	model->nextBranch(cur,prev,false);

	while (cur) 

	{

		cout << "ok: "<<cur->getHeadingStd()<<endl./re/videochristinaprison1_wmvl.wmv

;

		model->nextBranch(cur,prev,false);	

	}

	*/

}

void VymView::collapseOneLevel()

{

	int level=-1;

	int d;

	BranchItem *cur=NULL;

	BranchItem *prev=NULL;

	QModelIndex pix;

	// Find level to collapse

	model->nextBranch(cur,prev);

	while (cur) 

	{

		pix=model->index (cur);

		d=cur->depth();

		if (treeEditor->isExpanded(pix) && d > level)

			level=d;

		model->nextBranch(cur,prev);	

	}

	// collapse all to level

	cur=NULL;

	prev=NULL;

	model->nextBranch(cur,prev);

	while (cur) 

	{

		pix=model->index (cur);

		d=cur->depth();

		if (treeEditor->isExpanded(pix) && d >= level)

			treeEditor->setExpanded(pix,false);

		model->nextBranch(cur,prev);	

	}

}

void VymView::showSelection()

{

	QModelIndex ix=model->getSelectedIndex();

	treeEditor->scrollTo( ix, QAbstractItemView::EnsureVisible);

	mapEditor->scrollTo ( ix);	// FIXME-3 also called from MapEditor::updateSelection...

}

void VymView::toggleTreeEditor()

{

	if (treeEditor->isVisible() )

		treeEditor->hide();

	else

		treeEditor->show();

}