#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