#include <windows.h>
#include <winbase.h>

#include <captura.h>

#define DWM_EC_DISABLECOMPOSITION   0
#define DWM_EC_ENABLECOMPOSITION    1

typedef UINT (CALLBACK* DLLFUNC)(UINT);
DLLFUNC DwmEnableComposition;

bool init_dll();
void fix_win_vis_7(bool enable);

bool is_win_vis_7;

int main()
{
    if(init_dll()) {
        is_win_vis_7 = true;
        MessageBox(NULL, "Dwmapi.dll OK", "test", MB_OK | MB_ICONEXCLAMATION);
    }
    
    Sleep(5000);
    
    if(is_win_vis_7)
        fix_win_vis_7(true);
        
    cap("test.bmp");
    
    if(is_win_vis_7)
        fix_win_vis_7(false);
    
    MessageBox(NULL, "Screenshot sacada", "test", MB_OK | MB_ICONEXCLAMATION);
    
    return 0;
}

void fix_win_vis_7(bool enable)
{
     DwmEnableComposition(enable?DWM_EC_ENABLECOMPOSITION:DWM_EC_DISABLECOMPOSITION);
}

bool init_dll()
{
     HINSTANCE hDLL;
     
     hDLL = LoadLibrary("Dwmapi");
     if(hDLL == NULL)
          return false;

     DwmEnableComposition = (DLLFUNC)GetProcAddress(hDLL, "DwmEnableComposition");

     if(!DwmEnableComposition)
     {
          FreeLibrary(hDLL);
          return false;
     }
     return true;
}