#include "ZmqSocket.h"

#include "ansi.h"
#include <iostream>

#include "moench03T1ZmqData.h"

// #include "interpolatingDetector.h"
// #include "etaInterpolationPosXY.h"
 #include "linearInterpolation.h"
// #include "noInterpolation.h"
#include "multiThreadedAnalogDetector.h"
//#include "singlePhotonDetector.h"
#include "interpolatingDetector.h"


using namespace std;
#define NC 400
#define NR 400


#define SLS_DETECTOR_JSON_HEADER_VERSION 0x2


int main(int argc, char *argv[]) {
/**
 * trial.o [socket ip] [starting port number] [outfname]
 *
 */

  int p=1000;
  int fifosize=1000;
  int nthreads=6;
  int nsubpix=2;
  // int etabins=550;
  // double etamin=-1, etamax=2;
  int csize=3;
  int nx=400, ny=400;
  int save=1;
  int nsigma=5;
  int nped=1000;
  int ndark=100;
  int ok;
  //int* image = new int[(size/sizeof(int))]();
  char* buff;
  int iprog=0;
  char outfname[10000];


  moench03T1ZmqData *decoder=new  moench03T1ZmqData();
  // cout << "decoder "<< endl;
  //etaInterpolationPosXY *interp=new etaInterpolationPosXY(nx, ny, nsubpix, etabins, etamin, etamax);
  // linearInterpolation *interp=new linearInterpolation(NC, NR, nsubpix);
  //noInterpolation *interp=new noInterpolation(NC, NR, nsubpix);
  // interpolatingDetector *filter=new interpolatingDetector(decoder,interp, nsigma, 1, 0, nped, 10);
    singlePhotonDetector *filter=new singlePhotonDetector(decoder,csize, nsigma, 1, 0, nped, 100);
  char tit[10000];




  // filter->readPedestals("/scratch/ped_100.tiff");
  // interp->readFlatField("/scratch/eta_100.tiff",etamin,etamax);
  // cout << "filter "<< endl;
  


  filter->newDataSet();



  multiThreadedAnalogDetector *mt=new multiThreadedAnalogDetector(filter,nthreads,fifosize);



  mt->setFrameMode(eFrame); //need to find a way to switch between flat and frames!
  // mt->prepareInterpolation(ok);
  mt->StartThreads();
  mt->popFree(buff);


	// help
	if (argc < 4 || (argc > 4 && argc != 6)) {
	  cprintf(RED, "Help: %s [receive socket ip] [receive starting port number] [fname]\n", argv[0]);
		return EXIT_FAILURE;
	}

	// receive parameters
	bool send = false;
	char* socketip = argv[1];
	uint32_t portnum = atoi(argv[2]);
	int size = 327680;////atoi(argv[3]);

	char fn[10000];
	strcpy(fn, argv[3]);
	// send parameters if any
	// char* socketip2 = 0;
	// uint32_t portnum2 = 0;
	// if (argc > 4) {
	// 	send = true;
	// 	socketip2 = argv[4];
	// 	portnum2 = atoi(argv[5]);
	// }
	cout << "\nrx socket ip : " << socketip <<
	  "\nrx port num  : " <<  portnum <<
	//	"\nsize         : " <<  size;
	  "\nfname  : " <<  fn ;
	// if (send) {
	// 	cout << "\nsd socket ip : " << socketip2 <<
	// 			"\nsd port num  : " <<  portnum2;
	// }
	cout << endl;






	// receive socket
	ZmqSocket* zmqsocket = new ZmqSocket(socketip,portnum);
	if (zmqsocket->IsError()) {
		cprintf(RED, "Error: Could not create Zmq socket on port %d with ip %s\n", portnum, socketip);
		delete zmqsocket;
		return EXIT_FAILURE;
	}
	printf("Zmq Client at %s\n", zmqsocket->GetZmqServerAddress());

	// send socket
	// ZmqSocket* zmqsocket2 = 0;
	// if (send) {
	// 	zmqsocket2 = new ZmqSocket(portnum2, socketip2);
	// 	if (zmqsocket2->IsError()) {
	// 		bprintf(RED, "Error: Could not create Zmq socket server on port %d and ip %s\n",  portnum2, socketip2);
	// 		delete zmqsocket2;
	// 		delete zmqsocket;
	// 		return EXIT_FAILURE;
	// 	}
	// 	printf("Zmq Server started at %s\n", zmqsocket2->GetZmqServerAddress());
	// }




	// header variables
	uint64_t acqIndex = -1;
	uint64_t frameIndex = -1;
	uint32_t subframeIndex = -1;
	string filename = "";
	int nnx, nny, nns;
	int imsize=filter->getImageSize(nnx,nny,nns);
	//int imsize=nx*ny;
	int i_image=0;

	int iframe=0;

	// infinite loop
	while(1) {


		// get header, (if dummy, fail is on parse error or end of acquisition)
		if (!zmqsocket->ReceiveHeader(0, acqIndex, frameIndex, subframeIndex, filename)) {
			cprintf(RED, "Acquisition finished\n");

			

			cout << "Recieved " << iframe << " frames " << endl;
			while (mt->isBusy()) {;}//wait until all data are processed from the queues
			cout << "threads done " << endl;
			
			// stream dummy  to socket2 to signal end of acquisition
			// if (send) {	
			//   zmqsocket2->SendData((char*)(mt->getImage()),imsize*sizeof(int));
			//   cprintf(BLUE, "Sent Interpolated image\n");
			// 	// zmqsocket2->SendHeaderData(0, true, SLS_DETECTOR_JSON_HEADER_VERSION);
			// 	// cprintf(RED, "Sent Dummy\n");
			// }
			//	if (save) {
			  sprintf(tit,"%s_%d.tiff",fn, i_image);
			  cout << tit << endl;
			   mt->writeImage(tit);
			  cout << "wrote" << endl;
			   mt->clearImage();
			   //	}
			i_image++;
			// dont get data
			continue; //continue to not get out
		}
		//	cprintf(GREEN, "Got Header \n");
		// get data
		int length = zmqsocket->ReceiveData(0, (int*)buff, size);
		//	cprintf(GREEN, "Got Data\n");

		//processing with image
		//...




	mt->pushData(buff);
	mt->nextThread();
	//	cout << " " << (void*)buff;
	mt->popFree(buff);
	
	
		iframe++;
		if (iframe%p==0) {

	while (mt->isBusy()) {;}//wait until all data are processed from the queues
	  sprintf(tit,"%s_tmp.tiff",fn);
			  mt->writeImage(tit);
			  //mt->clearImage();

			  cout <<"*"<< iprog++ << endl;

		}









		// //stream data from socket 2
		// if (send) {
		// 	zmqsocket2->SendHeaderData(0, false, SLS_DETECTOR_JSON_HEADER_VERSION,
		// 			0,0,0,acqIndex,frameIndex,(char*)"run", acqIndex, 0,0,0,0,0,0,0,0,0,0,0,1);
		// 	cprintf(GREEN, "Sent Header\n");

		// 	zmqsocket2->SendData((char*)image,length);
		// 	cprintf(GREEN, "Sent Data\n");
		// }


	}// exiting infinite loop



	delete zmqsocket;
	// if (send)
	// 	delete zmqsocket2;


	cout<<"Goodbye"<<  endl;
	return 0;
}