#pragma once

#ifndef TNZ_BUTTON_INCLUDED
#define TNZ_BUTTON_INCLUDED

#include "tw/tw.h"
#include "tw/action.h"
#include "traster.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 TButtonSet;

class DVAPI TButton : public TWidget, public TCommandSource
{
protected:
	bool m_pressed, m_active;
	int m_border;

	vector<TButtonSet *> *m_buttonSets;
	friend class TButtonSet;

public:
	TButton(TWidget *parent, string name = "button");
	~TButton();

	void repaint();
	void enter(const TPoint &p);
	void leave(const TPoint &p);
	void leftButtonDown(const TMouseEvent &);
	void leftButtonUp(const TMouseEvent &);

	virtual void setTitle(string title) { m_name = title; }
};

class DVAPI TIconButton : public TButton
{
	TRaster32P m_rasterUp, m_rasterDown;

public:
	TIconButton(TWidget *parent, TRaster32P raster, string name = "button");
	TIconButton(TWidget *parent, TRaster32P rasterUp, TRaster32P rasterDown, string name = "button");
	void repaint();
	void leftButtonDown(const TMouseEvent &);
	void leftButtonUp(const TMouseEvent &);
};

class DVAPI TIconToggle : public TButton
{
	TRaster32P m_up, m_down;

public:
	TIconToggle(TWidget *parent,
				TRaster32P up, TRaster32P down,
				string name = "button");
	void repaint();
	void setStatus(bool v);
	bool getStatus();
	void leftButtonDown(const TMouseEvent &e);
	void leftButtonUp(const TMouseEvent &e);
};

class TButtonSetAction
{
public:
	TButtonSetAction(){};
	virtual ~TButtonSetAction(){};
	virtual void sendCommand(string value) = 0;
};

template <class T>
class TButtonSetActionT : public TButtonSetAction
{
public:
	typedef void (T::*CommandMethod)(string);

	T *m_target;
	CommandMethod m_method;

	TButtonSetActionT<T>(T *target, CommandMethod method) : m_target(target), m_method(method){};
	void sendCommand(string value) { (m_target->*m_method)(value); };
};

class DVAPI TButtonSet
{
	std::map<TButton *, string> *m_buttons;
	string m_currentValue;
	TButtonSetAction *m_action;

public:
	TButtonSet();
	~TButtonSet();

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

public:
	void addButton(TButton *button, string value);
	void removeButton(TButton *button);
	void setValue(string value);
	void setValue(TButton *value);
	// Note: setValue fires the buttons actions as well as the buttonset action
	string getValue() { return m_currentValue; };

	void setAction(TButtonSetAction *action);

	void enable(bool on);
};

#endif