#pragma once

#ifndef TNZ_POPUPMENU_INCLUDED

#define TNZ_POPUPMENU_INCLUDED

#include "tw/action.h"

#include "tw/popup.h"

#include "tw/button.h"

#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

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

class TPopupMenu;

class TPopupMenuItem;

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

class DVAPI TPopupMenuListener

{

public:

	TPopupMenuListener() {}

	virtual ~TPopupMenuListener() {}

	virtual void onMenuSelect(TPopupMenuItem *item) = 0;

};

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

class DVAPI TPopupMenuItem : public TButton

{

	wstring m_title;

	bool m_selected;

public:

	TPopupMenuItem(TPopupMenu *parent, string name);

	void repaint();

	void enter(const TPoint &);

	void leave(const TPoint &);

	void leftButtonDown(const TMouseEvent &e);

	void leftButtonUp(const TMouseEvent &e);

	void setTitle(wstring title);

	TPopupMenu *getPopup();

	void select(bool on);

	bool isSelected() const { return m_selected; }

	virtual int getWidth();

};

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

class DVAPI TPopupMenuSeparator : public TWidget

{

public:

	TPopupMenuSeparator(TPopupMenu *parent);

	void repaint();

};

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

//

// Note: to implement right-mouse-bottom menu use

// the class TContextMenu (see below)

//

class DVAPI TPopupMenu : public TPopup

{

	TPopupMenuItem *m_currentItem;

	TPopupMenuListener *m_listener;

	int m_firstVisibleItem;

	int m_visibleItemCount;

	bool m_leftAlignment;

public:

	TPopupMenu(TWidget *parent, string name = "popupMenu");

	void draw();

	void configureNotify(const TDimension &);

	void leftButtonDrag(const TPoint &, UCHAR);

	void leftButtonUp(const TPoint &);

	void popup(const TPoint &pos, int width = 120);

	void close();

	TPopupMenuItem *getCurrentItem() { return m_currentItem; }

	void setListener(TPopupMenuListener *);

	static TPopupMenu *getCurrentMenu();

	void setLeftAlignment(bool a)

	{

		m_leftAlignment = a;

	}

private:

	void updateItemsVisibility();

	void updateSize();

	void computeSize(TPoint &pos, int maxY);

	friend class TPopupMenuArrow;

};

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

//

// TContextMenu :

//  a class specifically dedicated to

//  right-mouse-button popup menu (aka Context Menu)

//

// Note: DON'T use the method popup() (inherited from TPopupMenu)

//

// Note: the context menu will be automatically destroyed

// DON'T keep a reference to the context menu

//

// see examples below

//

/*---- Example 1 (position indipendent) ----

   void Pluto::rightButtonUp(const TMouseEvent &e) {

     TContextMenu *menu = new TContextMenu(this);

     menu->addCommand("MI_Cut");

     menu->addCommand("MI_Copy");

     menu->addCommand("MI_Paste");

     menu->addCommand("MI_Clear");

     TContextMenu::open(menu, e.m_pos);

   }

  ---- Example 2 (position dependent) ----

   class MyContextMenu : public TContextMenu {

     int m_myValue;

   public:

     MyContextMenu(TWidget*parent, int myValue) 

     : TContextMenu(parent), m_myValue(myValue), m_col(col) {

        tconnect(this, "foo",  this, &CellContextMenu::onFoo);

        tconnect(this, "bar",  this, &CellContextMenu::onBar);

     }

     void onFoo() { ... }

     void onBar() { ... }

   };

   void Pluto::rightButtonUp(const TMouseEvent &e) {

     int row = ....

     int col = ....

     TContextMenu::open(new CellContextMenu(this, row, col), e.m_pos);

   }

  -----------------------------------------*/

class DVAPI TContextMenu : public TPopupMenu

{

public:

	class CommandFilter

	{

	public:

		virtual bool check(string cmdName) const = 0;

		virtual ~CommandFilter() {}

	};

	TContextMenu(TWidget *parent);

	void add(string name, TGenericCommandAction *action);

	void addCommand(string cmdName);

	void addCommand(string cmdName, TGenericCommandAction *action);

	void addSeparator() { new TPopupMenuSeparator(this); }

	/*

  template <class t=""></class>

  inline void addCommand(string cmdName, T*target, void (T::*method)()) {

    //addCommand(cmdName, new TCommandAction<t>(target, method));</t>

  }

*/

	static void open(TContextMenu *menu, const TPoint &pos);

	static void setCommandFilter(CommandFilter *filter);

	void onTimer(int);

};

template <class t=""></class>

inline void tconnect(TContextMenu *menu, string cmdName, T *target, void (T::*method)())

{

	menu->addCommand(cmdName, new TCommandAction<t>(target, method));</t>

}

template <class arg="" class="" t,=""></class>

inline void tconnect(

	TContextMenu *menu, string cmdName,

	T *target, void (T::*method)(Arg), Arg arg)

{

	menu->addCommand(cmdName, new TCommandAction1<t, arg="">(target, method, arg));</t,>

}

#endif