#pragma once

#ifndef TNZ_TREEVIEW_INCLUDED
#define TNZ_TREEVIEW_INCLUDED

#include "tw/tw.h"
#include "tw/dragdrop.h"

#include <set>

#undef DVAPI
#undef DVVAR
#ifdef TWIN_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

#ifdef WIN32
#pragma warning(disable : 4251)
#endif

//-------------------------------------------------------------------

class TTreeView;
class TTreeViewItem;
class TTreeViewItemParent;
class TScrollbar;
class TTextField;
class TContextMenu;

//-------------------------------------------------------------------

class DVAPI TTreeViewItemUpdater
{
public:
	virtual bool done() const = 0;
	virtual void next() = 0;
	virtual bool matchItem(const TTreeViewItem *) const = 0;
	virtual TTreeViewItem *createItem(TTreeViewItemParent *parent) const = 0;
	virtual ~TTreeViewItemUpdater() {}
};

//-------------------------------------------------------------------

class DVAPI TTreeViewItemParent
{
	TTreeView *m_treeView;

protected:
	bool m_valid;

	typedef vector<TTreeViewItem *> ItemContainer;
	ItemContainer m_children;

public:
	TTreeViewItemParent()
		: m_treeView(0), m_valid(true)
	{
	}
	virtual ~TTreeViewItemParent();

	void clearItems();
	int getItemCount() const;
	TTreeViewItem *getItem(int index) const;
	TTreeViewItem *getItem(wstring name) const;
	TTreeViewItem *getItem(string name) const;

	void addItem(TTreeViewItem *item);
	void removeItem(TTreeViewItem *item);

	void updateItems(TTreeViewItemUpdater &updater);

	bool hasChild(TTreeViewItem *item) const;

	virtual void onItemChange();

	void setTreeView(TTreeView *treeView) { m_treeView = treeView; }
	TTreeView *getTreeView() const { return m_treeView; }

private:
	// not implemented
	TTreeViewItemParent(const TTreeViewItemParent &);
	TTreeViewItemParent &operator=(const TTreeViewItemParent &);
};

//-------------------------------------------------------------------

class DVAPI TTreeViewItem : public TTreeViewItemParent
{

	unsigned char m_status;
	TTreeViewItemParent *m_parent;

public:
	TTreeViewItem(TTreeViewItemParent *parent);
	virtual ~TTreeViewItem();

	TTreeViewItemParent *getParent() const { return m_parent; }
	void setParent(TTreeViewItemParent *parent);

	void clearParent() { m_parent = 0; }

	virtual void onOpen() {}
	virtual void onClose() {}
	virtual void onDoubleClick();

	virtual TTreeViewItem *getSon(string sonName) { return 0; }
	virtual TTreeViewItem *getSon(wstring sonName) { return 0; }

	bool isLeaf() const;
	bool isEmpty() const;
	bool isOpen() const;
	bool isSelected() const;

	void open();
	void close();
	virtual void refresh();

	void setIsLeaf(bool isLeaf);
	void select(bool on);

	bool unselectDescendents(); // ritorna true se ce n'era uno selezionato

	// "origin" sono le coordinate del punto in basso a sinistra dell'icona
	virtual void drawIcon(TTreeView *w, const TPoint &origin);
	virtual void drawName(TTreeView *w, const TPoint &origin);
	virtual void draw(TTreeView *w, const TPoint &origin);
	virtual int getWidth(TTreeView *w);

	virtual wstring getName() const { return L"noname"; }

	virtual TDimension getIconSize() const;
	virtual wstring getTooltipString() { return wstring(); }
	virtual string getContextHelpReference() { return string(); }

	virtual bool isRenameEnabled() const { return false; }
	virtual void rename(wstring name){};

	virtual TContextMenu *createContextMenu()
	{
		return 0;
	}

	virtual bool acceptDrop(const TData *data) const { return false; }
	virtual bool drop(const TData *data) { return false; }
	virtual bool click(const TPoint &p) { return false; }
	virtual bool drag(const TPoint &p) { return false; }
	virtual bool rightClick(const TPoint &p);

private:
	// not implemented
	TTreeViewItem(const TTreeViewItem &);
	TTreeViewItem &operator=(const TTreeViewItem &);
};

//-------------------------------------------------------------------

class DVAPI TTreeView
	: public TWidget,
	  public TTreeViewItemParent,
	  public TDragDropListener
{
public:
	class Listener
	{
	public:
		virtual void onTreeViewSelect(TTreeView *treeView) = 0;
		virtual ~Listener() {}
	};

protected:
	struct VisibleItem {
		TPoint m_pos;
		// m_pos e' la coordinata del punto in basso a sinistra dell'icona
		// (l'origine e' il punto in alto a sinistra della prima icona)
		int m_height;
		// m_height == getIconSize().ly

		TTreeViewItem *m_item;

		enum {
			LastSiblingFlag = 0x1,
			RootFlag = 0x2,
			SonFlag = 0x4
		};
		int m_status;
		//int m_siblingLinkLength;
		//int m_childLinkLength;

		VisibleItem();
	};

	int m_xMargin,
		m_yMargin,
		m_iconMargin; // distanza orizzontale fra il bottone "+/-" e l'icona

	vector<VisibleItem> m_visibleItems;

#ifndef MACOSX
	//togliere l'ifdef con il gcc3.3.2
	bool m_valid;
#endif
	int m_height;
	int m_yoffset;
	int m_xoffset;
	//int m_selectedItemIndex;
	std::set<int> m_selectedItemIndices;

	TScrollbar *m_vScb, *m_hScb;
	TTextField *m_renameTextField;
	bool m_multiSelectionEnabled;
	int m_dropIndex;
	vector<Listener *> m_listeners;
	TPoint m_lastMousePos;

public:
	// costruttore/distruttore
	TTreeView(TWidget *parent, string name = "treeView");
	~TTreeView();

	// listeners
	void addListener(Listener *listener);
	void removeListener(Listener *listener);
	void notifyListeners();

	// configure
	void setScrollbars(TScrollbar *h, TScrollbar *v);
	void enableMultiSelection(bool on);
	bool isMultiSelectionEnabled() const
	{
		return m_multiSelectionEnabled;
	}

	// resize
	void configureNotify(const TDimension &s);

	// draw
	void repaint();
	virtual void drawButton(const TPoint &p, bool open);

	// event handlers
	void leftButtonDown(const TMouseEvent &e);
	void leftButtonDrag(const TMouseEvent &e);
	void leftButtonUp(const TMouseEvent &e);

	void rightButtonDown(const TMouseEvent &e);
	void rightButtonDrag(const TMouseEvent &e);
	void rightButtonUp(const TMouseEvent &e);

	void middleButtonDown(const TMouseEvent &e);
	void middleButtonDrag(const TMouseEvent &e);
	void middleButtonUp(const TMouseEvent &e);

	void mouseWheel(const TMouseEvent &, int wheel);

	void leftButtonDoubleClick(const TMouseEvent &);

	TDropSource::DropEffect onEnter(const TDragDropListener::Event &event);
	TDropSource::DropEffect onOver(const TDragDropListener::Event &event);
	TDropSource::DropEffect onDrop(const TDragDropListener::Event &event);
	void onLeave();

	// to override
	virtual void onSelect(TTreeViewItem *item){};
	virtual bool startDragDrop(TTreeViewItem *item) { return false; }
	virtual void onExpand(TTreeViewItem *item){};
	virtual void onCollapse(TTreeViewItem *item){};

	// queries
	int getSelectedItemCount() const;
	TTreeViewItem *getSelectedItem(int index = 0) const;

	// selection
	void selectNone();
	void select(TTreeViewItem *item);
	void selectParent();
	void selectSon(string sonName);
	void selectItem(int index); // provvisorio

	// refresh
	void refreshCurrentItem();
	void refresh();

	// tooltips &helps
	wstring getTooltipString(const TPoint &);
	string getContextHelpReference(const TPoint &p);

#ifndef MACOSX
	//togliere l'ifdef con il gcc3.3.2
	void onItemChange();
#endif

	// misc
	void scrollToX(int x);
	void scrollToY(int y);
	void renameCurrentItem(wstring s);
	void updateVisibleItems();

protected:
	void getPlacement(int index, TRect &rect);
	int getIndexFromY(int y);
	void addItemAndSons(TTreeViewItem *item, int indentation);
	void updateScrollbars();
};

#endif