#ifndef CHECKBOX_H
#define CHECKBOX_H
#include <windows.h>
#include <button.hpp>
#include <string>

using namespace std;

class CheckBox : public Button {
	public:
    CheckBox(string text, int x, int y, int width, int height, HWND hParent);
    CheckBox(string text, int x, int y, int width, int height, HWND hParent, int id);
    void show();
    void show(int style);
    void show(int style, int exstyle);
    void set_checked(int style);
    int get_checked();
    };

#endif

#include "checkbox.hpp"

CheckBox::CheckBox(string text, int x, int y, int width, int height, HWND hParent) : Button(text, x, y, width, height, hParent) {}

CheckBox::CheckBox(string text, int x, int y, int width, int height, HWND hParent, int id) : Button(text, x, y, width, height, hParent, id) {}

void CheckBox::show() {
    Button::show();	
	LONG nNewStyle = GetWindowLong(this->get_handle(), GWL_STYLE) | BS_AUTOCHECKBOX;
	SetWindowLong(this->get_handle(), GWL_STYLE, nNewStyle);
}

void CheckBox::show(int style) {
	Button::show(style);
	LONG nNewStyle = GetWindowLong(this->get_handle(), GWL_STYLE) | BS_AUTOCHECKBOX;
	SetWindowLong(this->get_handle(), GWL_STYLE, nNewStyle);
}

void CheckBox::show(int style, int exstyle) {
	Button::show(style, exstyle);
	LONG newstyle = GetWindowLong(this->get_handle(), GWL_STYLE) | BS_AUTOCHECKBOX;
	SetWindowLong(this->get_handle(), GWL_STYLE, newstyle);
}

void CheckBox::set_checked(int style) {
	SendMessage(this->get_handle(), BM_SETCHECK, style, 0);
}

int CheckBox::get_checked() {
	int state = SendMessage(this->get_handle(), BM_GETCHECK, 0, 0);
	return state;
}

#include <windows.h>
#include "checkbox.hpp"

CheckBox *cb1;
CheckBox *cb2;
CheckBox *cb3;
/* This is where all the input to the window goes to */
LRESULT CALLBACK WndProc(HWND hwnd, UINT Message, WPARAM wParam, LPARAM lParam) {
	switch(Message) {
    case WM_CREATE:
    	cb1 = new CheckBox("Checkbox1", 10, 10, 100, 20, hwnd);	
    	cb1->show();
    	cb1->set_checked(BST_CHECKED);
    	
    	cb2 = new CheckBox("Checkbox2", 10, 40, 100, 20, hwnd);	
    	cb2->show();
    	cb2->set_checked(BST_INDETERMINATE);
    	
    	cb3 = new CheckBox("Checkbox3", 10, 70, 100, 20, hwnd);	
    	cb3->show();
    	cb3->set_checked(BST_UNCHECKED);    	
    break;
    /* Upon destruction, tell the main thread to stop */
    case WM_DESTROY: {
    	PostQuitMessage(0);
    	break;
    }
    
    /* All other messages (a lot of them) are processed using default procedures */
    default:
    	return DefWindowProc(hwnd, Message, wParam, lParam);
	}
	return 0;
}

/* The 'main' function of Win32 GUI programs: this is where execution starts */
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow) {
	WNDCLASSEX wc; /* A properties struct of our window */
	HWND hwnd; /* A 'HANDLE', hence the H, or a pointer to our window */
	MSG msg; /* A temporary location for all messages */

	/* zero out the struct and set the stuff we want to modify */
	memset(&wc,0,sizeof(wc));
	wc.cbSize     = sizeof(WNDCLASSEX);
	wc.lpfnWndProc	 = WndProc; /* This is where we will send messages to */
	wc.hInstance	 = hInstance;
	wc.hCursor     = LoadCursor(NULL, IDC_ARROW);
	
	/* White, COLOR_WINDOW is just a #define for a system color, try Ctrl+Clicking it */
	wc.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
	wc.lpszClassName = "WindowClass";
	wc.hIcon     = LoadIcon(NULL, IDI_APPLICATION); /* Load a standard icon */
	wc.hIconSm     = LoadIcon(NULL, IDI_APPLICATION); /* use the name "A" to use the project icon */

	if(!RegisterClassEx(&wc)) {
    MessageBox(NULL, "Window Registration Failed!","Error!",MB_ICONEXCLAMATION|MB_OK);
    return 0;
	}

	hwnd = CreateWindowEx(WS_EX_CLIENTEDGE,"WindowClass","My Control CheckBox Demo",WS_VISIBLE|WS_OVERLAPPEDWINDOW,
    CW_USEDEFAULT, /* x */
    CW_USEDEFAULT, /* y */
    350, /* width */
    200, /* height */
    NULL,NULL,hInstance,NULL);

	if(hwnd == NULL) {
    MessageBox(NULL, "Window Creation Failed!","Error!",MB_ICONEXCLAMATION|MB_OK);
    return 0;
	}

	/*
    This is the heart of our program where all input is processed and 
    sent to WndProc. Note that GetMessage blocks code flow until it receives something, so
    this loop will not produce unreasonably high CPU usage
	*/
	while(GetMessage(&msg, NULL, 0, 0) > 0) { /* If no error is received... */
    TranslateMessage(&msg); /* Translate key codes to chars if present */
    DispatchMessage(&msg); /* Send it to WndProc */
	}
	return msg.wParam;
}