directory restuctured

slsDetectorCalibration/moenchExecutables/moench03ZmqOnTheFlyInterpolation.cpp

@@ -0,0 +1,219 @@
+#include "ZmqSocket.h"
+
+#include "ansi.h"
+#include <iostream>
+
+#include "moench03T1ZmqData.h"
+
+#include "interpolatingDetector.h"
+#include "etaInterpolationPosXY.h"
+#include "linearInterpolation.h"
+#include "noInterpolation.h"
+#include "multiThreadedDetector.h"
+
+using namespace std;
+
+
+#define SLS_DETECTOR_JSON_HEADER_VERSION 0x2
+
+
+int main(int argc, char *argv[]) {
+/**
+ * trial.o [socket ip] [starting port number] [image size] [send_socket ip] [send port number]
+ *
+ */
+
+
+
+  int nthreads=3;
+  int nsubpix=5;
+  int etabins=550;
+  double etamin=-1, etamax=2;
+  int nx=400, ny=400;
+  int save=1;
+  int nsigma=5;
+  int nped=1000;
+  int ok;
+  //int* image = new int[(size/sizeof(int))]();
+  char* buff;
+
+  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);
+
+
+
+
+  filter->readPedestals("/scratch/ped_100.tiff");
+  interp->readFlatField("/scratch/eta_100.tiff",etamin,etamax);
+  cout << "filter "<< endl;
+  
+
+
+  filter->newDataSet();
+
+
+
+  multiThreadedDetector *mt=new multiThreadedDetector(filter,nthreads,100);
+
+
+
+  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: ./trial [receive socket ip] [receive starting port number] [image size]  [send_socket ip] [send starting port number]\n");
+		return EXIT_FAILURE;
+	}
+
+	// receive parameters
+	bool send = false;
+	char* socketip = argv[1];
+	uint32_t portnum = atoi(argv[2]);
+	int size = atoi(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;
+	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 imsize=nx*ny*nsubpix*nsubpix;
+	int i_image=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");
+
+
+
+			while (mt->isBusy()) {;}//wait until all data are processed from the queues
+			
+			
+			// stream dummy  to socket2 to signal end of acquisition
+			if (send) {	
+			  zmqsocket2->SendData((char*)(mt->getInterpolatedImage()),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) {
+			   char tit[10000];
+			   sprintf(tit,"/scratch/int_image_%d.tiff",i_image);
+			   mt->writeInterpolatedImage(tit);
+			   mt->clearInterpolatedImage(tit);
+			}
+			i_image++;
+			// dont get data
+			break; //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);
+
+
+
+
+
+
+
+
+
+
+
+
+
+		// //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;
+}
+