#include <windows.h>
#include "resource.h"
#include <fstream>
#include "Graphics\Renderer.h"
#include <map>

std::map<int, Animation *> animations;
Renderer * renderer;
int texture;

LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
BOOL CALLBACK DlgProc(HWND, UINT, WPARAM, LPARAM);
char * openFileDialog(char * filename);
bool loadAnimationFile(const char * filename, HWND listBox);
bool loadFramesForAnimation(int animationID, HWND listBox);

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nShowCmd) {
	MSG msg;
    texture = 1;
	/*HWND hwnd;
	WNDCLASS wndClass;

	wndClass.style = CS_HREDRAW | CS_VREDRAW;
	wndClass.lpfnWndProc = WndProc;
	wndClass.cbClsExtra = 0;
	wndClass.cbWndExtra = 0;
	wndClass.hInstance = hInstance;
	wndClass.hIcon = 0;
	wndClass.hCursor = LoadCursor(0, IDC_ARROW);
	wndClass.hbrBackground = (HBRUSH) GetStockObject(WHITE_BRUSH);
	wndClass.lpszMenuName = MAKEINTRESOURCE(IDR_MENU1);
	wndClass.lpszClassName = TEXT("Program");

	RegisterClass(&wndClass);

	hwnd = CreateWindow(TEXT("Program"),
						TEXT("Demo"),
						WS_OVERLAPPEDWINDOW,
						CW_USEDEFAULT, CW_USEDEFAULT,
						CW_USEDEFAULT, CW_USEDEFAULT,
						0, 0, hInstance, 0);

	ShowWindow(hwnd, nShowCmd);
	UpdateWindow(hwnd);*/

	DialogBox(hInstance, MAKEINTRESOURCE(IDD_MYDIALOG), 0, DlgProc);

	while(GetMessage(&msg, 0, 0, 0)) {
		TranslateMessage(&msg);
		DispatchMessage(&msg);
	}
	return msg.wParam;
}

//LRESULT CALLBACK WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam) {
//	static HINSTANCE hInstance;
//	int j = 0;
//
//	switch(message) {
//		case WM_CREATE:
//			hInstance = ((LPCREATESTRUCT) lParam)->hInstance;
//			//DialogBox(hInstance, MAKEINTRESOURCE(IDD_MYDIALOG), hwnd, AboutDlgProc);
//			return 0;
//
//		case WM_COMMAND:
//			switch(LOWORD(wParam)) {
//				case ID_FILE_ABOUT:
//					MessageBox(hwnd, TEXT("Hello"), TEXT("Hello"), 0);
//					
//					return 0;
//
//				case ID_FILE_EXIT:
//					PostQuitMessage(0);
//					return 0;
//			};
//			break;
//
//		case WM_DESTROY:
//			PostQuitMessage(0);
//			return 0;
//	}
//	return DefWindowProc(hwnd, message, wParam, lParam);
//}

BOOL CALLBACK DlgProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam) {
	HWND listbox = 0;
    Object obj;

	switch(message) {
        case WM_PAINT:
            obj.setTexture(texture);
            //renderer->addObject(&obj, 0, 0);
            //renderer->Render();
            return true;
	    case WM_INITDIALOG:
            //renderer = new Renderer(GetDlgItem(hDlg, IDC_DRAWFRAME), 0);
			return true;

		case WM_COMMAND:
			switch(LOWORD(wParam)) {
				case IDOPENANIMATIONFILE:
					loadAnimationFile(openFileDialog(0), GetDlgItem(hDlg, IDC_ANIMATIONLIST));
					break;

				case IDEXIT:
					EndDialog(hDlg, 0);
					PostQuitMessage(0);
					return true;

                case IDC_ANIMATIONLIST:
                    switch(HIWORD(wParam)) {
                        case LBN_SELCHANGE:
                            loadFramesForAnimation(1, GetDlgItem(hDlg, IDC_FRAMELIST));
                            break;
                    }
                    break;

                case IDC_FRAMELIST:
                    switch(HIWORD(wParam)) {
                        case LBN_SELCHANGE:
                            texture = SendMessage(GetDlgItem(hDlg, IDC_FRAMELIST), LB_GETCURSEL, 0, 0);
                            break;
                    }
                    break;
			}
			break;

        case WM_CLOSE:
			PostQuitMessage(0);
			return 0;
	}
	return false;
}

bool loadFramesForAnimation(int animationID, HWND listBox) {
    SendMessage(listBox, LB_RESETCONTENT, 0, 0);
    Animation * anim = animations[animationID];

    for(int i = 0; i < anim->numFrames; i++) {
        char name[10];
        sprintf(name, "Frame%d", i);
        SendMessage(listBox, LB_ADDSTRING, 0, (LPARAM)name);
    }
    return true;
}

bool loadAnimationFile(const char * filename, HWND listBox) {
    SendMessage(listBox, LB_RESETCONTENT, 0, 0);
    std::ifstream afile;
    afile.open("anim.dat", std::ifstream::in, std::ifstream::binary);
    if(!afile.good()) {
        MessageBox(0, "ERROR: Could not open file", (LPCSTR)filename, 0);
        return false;
    }

    animFileHeader filehead;
    afile.read((char *)&filehead, sizeof(animFileHeader));

    if(memcmp(filehead.ident, "SS13", 4)) {
        MessageBox(0, "ERROR: Corrupted animation file.", "Fatal Error", 0);
        return false;
    }

    animHeader ahead;

    Animation *anim = new Animation;

    for(int i = 0; i < filehead.numAnimations; i++) {
        afile.read((char *)&ahead, sizeof(animHeader));
        int len = 0;
        afile.read((char *)&len, 1);
        char * animName = new char[len];
        afile.read(animName, len);
        anim->name = animName;
        anim->animationID = ahead.animID;
        anim->numFrames = ahead.numFrames;
        anim->frames = new animFrame[anim->numFrames];

        for(int j = 0; j < ahead.numFrames; j++) {
            afile.read((char *)&anim->frames[j], sizeof(animFrame));            
        }
        animations[anim->animationID] = anim;
        SendMessage(listBox, LB_ADDSTRING, 0, (LPARAM)anim->name);
    }
    
    return true;
}

char * openFileDialog(char * filename) {
	if(!filename) {
		filename = new char[256];
	}
	memset(filename, 0x00, 256);

	OPENFILENAME opf;
	opf.hwndOwner = 0;
	opf.hInstance = GetModuleHandle(0);
	opf.lpstrFilter = TEXT("Dat Files (*.dat)\0*.dat\0\0");
	opf.lpstrCustomFilter = 0;
	opf.nMaxCustFilter = 0L;
	opf.nFilterIndex = 1L;
	opf.lpstrFile = (LPSTR)filename;
	opf.nMaxFile = 255;
	opf.lpstrFileTitle = 0;
	opf.nMaxFileTitle=50;
	opf.lpstrInitialDir = 0;
	opf.lpstrTitle = 0;
	opf.nFileOffset = 0;
	opf.nFileExtension = 2;
	opf.lpstrDefExt = TEXT("*.dat");
	opf.lpfnHook = NULL;
	opf.lCustData = 0;
	opf.Flags = OFN_PATHMUSTEXIST;
	opf.lStructSize = sizeof(OPENFILENAMEW);
	GetOpenFileName(&opf);

	return filename;
}