/* Capture official GRAYSCALE output via ThermalSDK + CoreSDKLib.
 * ThermalSDK creates the channel; we find it and read the grayscale
 * buffer through MAG_GetOutputBMPdata.
 */
#define _WIN32_WINNT 0x0601
#include <windows.h>
#include <stdio.h>
#include <stdbool.h>
#include <string.h>

typedef void (CALLBACK *INewIRFrame)(const unsigned char *, int, int, int);
typedef bool (CALLBACK *fn_Start_t)(void);
typedef void (CALLBACK *fn_Stop_t)(void);
typedef void (CALLBACK *fn_SetNewIRFrameDelegate_t)(INewIRFrame);
typedef int (CALLBACK *MAG_GetOutputBMPdata_t)(int, int *, unsigned char **);
typedef int (CALLBACK *MAG_GetOutputBMPdataRGB24_t)(int, unsigned char *, int, int);
typedef int (CALLBACK *MAG_GetOutputColorBardata_t)(int, void *, int);

static volatile int g_frames;
static void CALLBACK onIR(const unsigned char *buf, int w, int h, int ch) {
    (void)buf; (void)w; (void)h; (void)ch;
    g_frames++;
}

int main(void) {
    SetDllDirectoryA("C:\\Project\\MAG160C\\IR_Camera_SDK-1.0.1\\windows\\windows\\app");
    HMODULE t = LoadLibraryA("ThermalSDK.dll");
    HMODULE c = LoadLibraryA("CoreSDKLib.dll");
    printf("T=%p C=%p\n", (void*)t, (void*)c);
    if (!t || !c) return 1;
    fn_Start_t T_Start = (fn_Start_t)GetProcAddress(t, "Start");
    fn_Stop_t T_Stop = (fn_Stop_t)GetProcAddress(t, "Stop");
    fn_SetNewIRFrameDelegate_t T_Set = (fn_SetNewIRFrameDelegate_t)GetProcAddress(t, "SetNewIRFrameDelegate");
    MAG_GetOutputBMPdata_t G_Gray = (MAG_GetOutputBMPdata_t)GetProcAddress(c, "MAG_GetOutputBMPdata");
    MAG_GetOutputBMPdataRGB24_t G_RGB = (MAG_GetOutputBMPdataRGB24_t)GetProcAddress(c, "MAG_GetOutputBMPdataRGB24");
    MAG_GetOutputColorBardata_t G_Bar = (MAG_GetOutputColorBardata_t)GetProcAddress(c, "MAG_GetOutputColorBardata");
    printf("Gray=%p RGB=%p Bar=%p\n", (void*)G_Gray, (void*)G_RGB, (void*)G_Bar);

    T_Set(onIR);
    printf("Start()...\n"); fflush(stdout);
    BOOL ok = T_Start();
    printf("Start=%d\n", ok); fflush(stdout);
    Sleep(3000);
    printf("frames=%d\n", g_frames);

    /* try channels 0..4 for the grayscale buffer */
    for (int chan = 0; chan < 5; ++chan) {
        int w = 0;
        unsigned char *buf = NULL;
        int rc = G_Gray(chan, &w, &buf);
        printf("chan %d: GetOutputBMPdata rc=%d w=%d buf=%p\n", chan, rc, w, (void*)buf);
        if (rc && buf) {
            /* grayscale is w*h bytes; save it */
            int n = w > 0 && w <= 40000 ? w : 19200;
            FILE *f = fopen("official_gray.bin", "wb");
            if (f) { fwrite(buf, 1, n, f); fclose(f); }
            printf("  saved official_gray.bin (%d bytes)\n", n);
            break;
        }
    }
    T_Stop();
    printf("done\n");
    return 0;
}