// this is simple offline processing example application
// it captures images from camera (using xiAPI) and process them using xiProcAPI		

#ifdef WIN32
#include "xiApi.h"
#else
#include <m3api/xiApi.h>
#endif

#include <stdio.h>

#include <string>
#include <iostream>
using namespace std;

#define throw_line(message) {char err[500]="";sprintf(err,"Fatal error: %s in:" __FILE__ "line:%d\n",message,__LINE__);throw err;}
#define XI_CARE(func) {XI_RETURN ret=func;char msg[200]="";if (XI_OK!=ret) {sprintf(msg,"Error on " #func " - expected XI_OK. Result:%d\n",ret);throw_line(msg);};}
#define THROW_IF_ZERO(val) {if (!val) throw_line("Value "#val"is zero");}

void TestOfflineProcessing_UsingAPI_WithCamera()
{
	cout << "TestOfflineProcessing_UsingAPI_WithCamera V3\n";
	
	// images buffer
	#define MAX_IMAGES 10
	unsigned char* src_data[MAX_IMAGES];
	
	XI_IMG image;
	memset(&image,0,sizeof(image));
	image.size = sizeof(XI_IMG);
	
	// get images from camera to RAM buffers
	cout << "Capturing images from camera\n";
	HANDLE camh = NULL;
	DWORD dwNumberOfDevices = 0;
	XI_CARE(xiGetNumberDevices(&dwNumberOfDevices));
	THROW_IF_ZERO(dwNumberOfDevices);
	XI_CARE(xiOpenDevice(0, &camh))
	XI_CARE(xiSetParamInt(camh, XI_PRM_IMAGE_DATA_FORMAT, XI_RAW8));
	XI_CARE(xiSetParamInt(camh, XI_PRM_BUFFER_POLICY, XI_BP_SAFE));
	// set the same parameters like when camera processing is running online
	// these parameters will be part of API context list
	XI_CARE(xiSetParamInt(camh, XI_PRM_EXPOSURE, 10000));
	XI_CARE(xiSetParamFloat(camh, XI_PRM_WB_KR,     1.05F));
	XI_CARE(xiSetParamFloat(camh, XI_PRM_WB_KG,     1.00F));
	XI_CARE(xiSetParamFloat(camh, XI_PRM_WB_KB,     2.10F));
	XI_CARE(xiSetParamFloat(camh, XI_PRM_GAMMAY,    1.0F));
	XI_CARE(xiSetParamFloat(camh, XI_PRM_SHARPNESS, 1.00F));

	// allocate buffers for input images
	int img_size_bytes=0;
	XI_CARE(xiGetParamInt(camh, XI_PRM_IMAGE_PAYLOAD_SIZE, &img_size_bytes));
	for (int i=0;i<MAX_IMAGES;i++)
	{
		src_data[i] = (unsigned char*)malloc(img_size_bytes);
		THROW_IF_ZERO(src_data[i]);
	}

	XI_CARE(xiStartAcquisition(camh))
	for (int i=0;i < MAX_IMAGES;i++)
	{
		// getting image from camera
		image.bp = src_data[i];
		image.bp_size = img_size_bytes;
		#define IMAGE_TIMEOUT_MS 3000
		XI_CARE(xiGetImage(camh, IMAGE_TIMEOUT_MS, &image)); // this will copy data into src_data[i] - because of XI_BP_SAFE policy set
		cout << "Image " << i << ": " << image.width << "x" << image.height << " received from camera. First pixel: " << (int)src_data[i][0] << endl;
	}
	char* cam_context=NULL;
	#define SIZE_OF_CONTEXT_BUFFER (10*1024*1024) // 10MiB
	cam_context = (char*)malloc(SIZE_OF_CONTEXT_BUFFER);
	XI_CARE(xiGetParamString(camh, XI_PRM_API_CONTEXT_LIST, cam_context, SIZE_OF_CONTEXT_BUFFER));
	XI_CARE(xiStopAcquisition(camh));
	XI_CARE(xiCloseDevice(camh));
	
	// offline processing
	cout << "Offline processing started\n";
	xiProcessingHandle_t proc;
	XI_CARE(xiProcOpen(&proc));
	int cam_context_len = (DWORD)strlen(cam_context);
	
	XI_CARE(xiProcSetParam(proc, XI_PRM_API_CONTEXT_LIST, cam_context, cam_context_len, xiTypeString));
	XI_IMG_FORMAT output_format = XI_RGB24;
	XI_CARE(xiProcSetParam(proc, XI_PRM_IMAGE_DATA_FORMAT, &output_format, sizeof(output_format), xiTypeInteger));
	for (int i=0;i < MAX_IMAGES;i++)
	{
		// image structure
		XI_IMG out_image;
		out_image.size = sizeof(XI_IMG);
		out_image.bp = NULL;
		out_image.bp_size = 0;
		// process
		XI_CARE(xiProcPushImage(proc, src_data[i]));
		XI_CARE(xiProcPullImage(proc, 0, &out_image)); // out_image.bp is NULL, so API allocate buffer for the output image. API will free the all allocated buffers on xiProcClose.
		cout << "Image " << i << ": " << out_image.width << "x" << out_image.height << " processed by offline processing in format: " << out_image.frm << ", First pixel: " << (unsigned short)((unsigned char*)out_image.bp)[0] << endl;
	}
	XI_CARE(xiProcClose(proc));
}

int main(int argc, char* argv[])
{
	try
	{
		TestOfflineProcessing_UsingAPI_WithCamera();
	}
	catch (char* err)
	{
		cout << "Error" << err << "\n";
	}
	cout << "Done\n";
	cout << "Press enter\n";
	cin.get();
}