1 files changed, 14 insertions, 1 deletions
diff --git a/aocharbutton.h b/aocharbutton.h
index e91a120..f715416 100644
--- a/aocharbutton.h
+++ b/aocharbutton.h
@@ -6,20 +6,33 @@
 #include <QPushButton>
 #include <QString>
 #include <QWidget>
+#include "aoimage.h"
 
 class AOCharButton : public QPushButton
 {
   Q_OBJECT
 
 public:
-  AOCharButton(QWidget *parent, AOApplication *p_ao_app);
+  AOCharButton(QWidget *parent, AOApplication *p_ao_app, int x_pos, int y_pos);
 
   AOApplication *ao_app;
 
+  void reset();
+  void set_taken();
+  void set_passworded();
+
   void set_image(QString p_character);
 
 private:
   QWidget *m_parent;
+
+  AOImage *ui_taken;
+  AOImage *ui_passworded;
+  AOImage *ui_selector;
+
+protected:
+  void enterEvent(QEvent *e);
+  void leaveEvent(QEvent *e);
 };
 
 #endif // AOCHARBUTTON_H