#ifndef TNZ_TREEVIEW_INCLUDED

#define TNZ_TREEVIEW_INCLUDED

#include "tw/tw.h"

#include "tw/dragdrop.h"

#include <set></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;</ttreeviewitem>

	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;</visibleitem>

#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;</int>

	TScrollbar *m_vScb, *m_hScb;

	TTextField *m_renameTextField;

	bool m_multiSelectionEnabled;

	int m_dropIndex;

	vector<listener *=""> m_listeners;</listener>

	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