#pragma once

#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