Draw method implemented for all Button variations (ugh)

1 files changed, 42 insertions, 2 deletions

diff --git a/src/c_fl_light_button.cpp b/src/c_fl_light_button.cpp
index daa99ef..5ef173e 100644
--- a/src/c_fl_light_button.cpp
+++ b/src/c_fl_light_button.cpp
@@ -4,13 +4,53 @@
 #include "c_fl_light_button.h"
 
 
+typedef void (hook)(void*);
+typedef hook* hook_p;
+
+
+
+
+class My_Light_Button : public Fl_Light_Button {
+    public:
+        using Fl_Light_Button::Fl_Light_Button;
+        friend void light_button_set_draw_hook(LIGHTBUTTON b, void * d);
+        friend void fl_light_button_draw(LIGHTBUTTON b);
+    protected:
+        void draw();
+        void real_draw();
+        hook_p draw_hook;
+};
+
+
+void My_Light_Button::draw() {
+    (*draw_hook)(this->user_data());
+}
+
+
+void My_Light_Button::real_draw() {
+    Fl_Light_Button::draw();
+}
+
+
+void light_button_set_draw_hook(LIGHTBUTTON b, void * d) {
+    reinterpret_cast<My_Light_Button*>(b)->draw_hook = reinterpret_cast<hook_p>(d);
+}
+
+
+void fl_light_button_draw(LIGHTBUTTON b) {
+    reinterpret_cast<My_Light_Button*>(b)->real_draw();
+}
+
+
+
+
 LIGHTBUTTON new_fl_light_button(int x, int y, int w, int h, char* label) {
-    Fl_Light_Button *b = new Fl_Light_Button(x, y, w, h, label);
+    My_Light_Button *b = new My_Light_Button(x, y, w, h, label);
     return b;
 }
 
 
 void free_fl_light_button(LIGHTBUTTON b) {
-    delete reinterpret_cast<Fl_Light_Button*>(b);
+    delete reinterpret_cast<My_Light_Button*>(b);
 }