#pragma once

#include "uMod_IDirect3DTexture9.h"
#include "uMod_IDirect3DDevice9.h"
#include "uMod_Error.h"
#include "uMod_ArrayHandler.h"


class uMod_TextureServer;

/*
 *  An object of this class is owned by each d3d9 device.
 *  functions called by the Server are called from the server thread instance.
 *  All other functions are called from the render thread instance of the game itself.
 */

class uMod_TextureClient {
public:
    uMod_TextureClient(uMod_TextureServer* server, IDirect3DDevice9* device);
    ~uMod_TextureClient();

    int AddTexture(uMod_IDirect3DTexture9* tex); //called from uMod_IDirect3DDevice9::CreateTexture(...) or uMod_IDirect3DDevice9::BeginScene()
    int AddTexture(uMod_IDirect3DVolumeTexture9* tex); //called from uMod_IDirect3DVolumeTexture9::CreateTexture(...) or uMod_IDirect3DDevice9::BeginScene()
    int AddTexture(uMod_IDirect3DCubeTexture9* tex); //called from uMod_IDirect3DCubeTexture9::CreateTexture(...) or uMod_IDirect3DDevice9::BeginScene()

    int RemoveTexture(uMod_IDirect3DTexture9* tex); //called from  uMod_IDirect3DTexture9::Release()
    int RemoveTexture(uMod_IDirect3DVolumeTexture9* tex); //called from  uMod_IDirect3DVolumeTexture9::Release()
    int RemoveTexture(uMod_IDirect3DCubeTexture9* tex); //called from  uMod_IDirect3DCubeTexture9::Release()

    int SaveAllTextures(bool val); //called from the Server
    int SaveSingleTexture(bool val); //called from the Server

    int SetSaveDirectory(wchar_t* dir); //called from the Server
    int SetGameName(wchar_t* dir); //called from the Server

    int SaveTexture(uMod_IDirect3DTexture9* pTexture); //called from uMod_IDirect3DDevice9::BeginScene() (save button) or from AddTexture(...) (SaveAllTextures)
    int SaveTexture(uMod_IDirect3DVolumeTexture9* pTexture); //called from uMod_IDirect3DDevice9::BeginScene() (save button) or from AddTexture(...) (SaveAllTextures)
    int SaveTexture(uMod_IDirect3DCubeTexture9* pTexture); //called from uMod_IDirect3DDevice9::BeginScene() (save button) or from AddTexture(...) (SaveAllTextures)



    int SetKeyBack(int key)
    {
        if (key > 0) {
            KeyBack = key;
        }
        return RETURN_OK;
    } //called from the Server
    int SetKeySave(int key)
    {
        if (key > 0) {
            KeySave = key;
        }
        return RETURN_OK;
    } //called from the Server
    int SetKeyNext(int key)
    {
        if (key > 0) {
            KeyNext = key;
        }
        return RETURN_OK;
    } //called from the Server

    int SetFontColour(DWORD r, DWORD g, DWORD b)
    {
        FontColour = D3DCOLOR_ARGB(255, r, g, b);
        return RETURN_OK;
    } //called from the Server
    int SetTextureColour(DWORD r, DWORD g, DWORD b)
    {
        TextureColour = D3DCOLOR_ARGB(255, r, g, b);
        return RETURN_OK;
    } //called from the Server


    int AddUpdate(TextureFileStruct* update, int number);  //called from the Server, client object must delete update array
    int MergeUpdate(); //called from uMod_IDirect3DDevice9::BeginScene()

    int LookUpToMod(uMod_IDirect3DTexture9* pTexture, int num_index_list = 0, int* index_list = nullptr); // called at the end AddTexture(...) and from Device->UpdateTexture(...)
    int LookUpToMod(uMod_IDirect3DVolumeTexture9* pTexture, int num_index_list = 0, int* index_list = nullptr); // called at the end AddTexture(...) and from Device->UpdateTexture(...)
    int LookUpToMod(uMod_IDirect3DCubeTexture9* pTexture, int num_index_list = 0, int* index_list = nullptr); // called at the end AddTexture(...) and from Device->UpdateTexture(...)

    uMod_TextureHandler<uMod_IDirect3DTexture9> OriginalTextures; // stores the pointer to the uMod_IDirect3DTexture9 objects created by the game
    uMod_TextureHandler<uMod_IDirect3DVolumeTexture9> OriginalVolumeTextures; // stores the pointer to the uMod_IDirect3DVolumeTexture9 objects created by the game
    uMod_TextureHandler<uMod_IDirect3DCubeTexture9> OriginalCubeTextures; // stores the pointer to the uMod_IDirect3DCubeTexture9 objects created by the game

    bool BoolSaveAllTextures;
    bool BoolSaveSingleTexture;
    int KeyBack;
    int KeySave;
    int KeyNext;

    D3DCOLOR FontColour;
    D3DCOLOR TextureColour;

private:
    uMod_TextureServer* Server;
    IDirect3DDevice9* D3D9Device;
    wchar_t SavePath[MAX_PATH];
    wchar_t GameName[MAX_PATH];

    TextureFileStruct* Update;
    int NumberOfUpdate;

    int LockMutex();
    int UnlockMutex();
    HANDLE Mutex;

    int NumberToMod; // number of texture to be modded
    TextureFileStruct* FileToMod; // array which stores the file in memory and the hash of each texture to be modded


    int LookUpToMod(MyTypeHash hash, int num_index_list, int* index_list); // called from LookUpToMod(...);
    int LoadTexture(TextureFileStruct* file_in_memory, uMod_IDirect3DTexture9** ppTexture); // called if a target texture is found
    int LoadTexture(TextureFileStruct* file_in_memory, uMod_IDirect3DVolumeTexture9** ppTexture); // called if a target texture is found
    int LoadTexture(TextureFileStruct* file_in_memory, uMod_IDirect3DCubeTexture9** ppTexture); // called if a target texture is found

    // and the corresponding fake texture should be loaded

    //MyTypeHash GetHash(unsigned char *str, int len);
    //unsigned int GetCRC32(char *pcDatabuf, unsigned int ulDatalen);
};