[u/mrichter/AliRoot.git] / PMD / PMDGAINda.cxx

/*
PMD DA for online calibration

contact: basanta@phy.iitb.ac.in
Link:
Reference run:/afs/cern.ch/user/b/bnandi/public/gaindata/pythia100evts.date
Run Type: PHYSICS
DA Type: MON
Number of events needed: 1 million for PB+PB, 200 milion for p+p
Input Files: PMD_PED.root, PMD_GAIN_CONFIGFILE, pmd_gain_tempfile.dat
Output Files: PMDGAINS.root, to be exported to the DAQ FES
Trigger types used: PHYSICS_EVENT

*/
extern "C" {
#include <daqDA.h>
}

#include "event.h"
#include "monitor.h"

#include <Riostream.h>
#include <stdio.h>
#include <stdlib.h>

//AliRoot
#include "AliRawReaderDate.h"
#include "AliPMDCalibPedestal.h"
#include "AliPMDCalibGain.h"

//ROOT
#include "TFile.h"
#include "TH1F.h"
#include "TBenchmark.h"
#include "TTree.h"
#include "TROOT.h"
#include "TPluginManager.h"


/* Main routine
      Arguments: 
      1- monitoring data source
*/
int main(int argc, char **argv) {
  
    /* magic line from Rene */
    gROOT->GetPluginManager()->AddHandler("TVirtualStreamerInfo",
					  "*",
					  "TStreamerInfo",
					  "RIO",
					  "TStreamerInfo()");


    int status = 0;


    Int_t filestatus = -1, xvar = 5;
    Int_t totevt = -1, maxevt = -1;
    Int_t hotevtsize = -1;
    Bool_t hotfilestatus = false;

    // Reads the pedestal file and keep the values in memory for subtraction

    AliPMDCalibGain calibgain;

    // Fetch the pedestal file - PMD_PED.root 

    status = daqDA_DB_getFile("PMD_PED.root","PMD_PED.root");

    if(!status)
      {
	printf("*** Pedestal file retrieved from DB *** \n");
      }
    else
      {
	printf("*** Pedestal file NOT retrieved from DB *** \n");
	return -1;
      }
    
    Int_t pstatus = calibgain.ExtractPedestal("PMD_PED.root");

    if(pstatus == -3) return -3;

    TTree *ic    = NULL;
    TTree *meanc = NULL;

    // Retrieve the PMD_GAIN_CONFIGFILE
    status = daqDA_DB_getFile("PMD_GAIN_CONFIGFILE","PMD_GAIN_CONFIGFILE");

    FILE *fp1 = NULL;

    fp1 = fopen("PMD_GAIN_CONFIGFILE","r");

    if (fp1 == NULL)
      {
	printf("*** PMD GAIN Configfile doesn't exist,Provide one ***\n");
	return -1;
      }
    else
      {
	fscanf(fp1,"%d %d %d %d %d\n",&filestatus, &xvar, &totevt, &maxevt, &hotevtsize);
	//printf("%d %d %d %d %d\n",filestatus, xvar, totevt, maxevt, hotevtsize);
      }
    fclose(fp1);

    
    if (filestatus == 1)
      {
	// Retrieve the Temporray ascii file from DB
	status = daqDA_DB_getFile("pmd_gain_tempfile.dat","pmd_gain_tempfile.dat");
	if(!status)
	  {
	    calibgain.ReadTempFile("pmd_gain_tempfile.dat");
	  }
	else
	  {
	    printf("--- pmd_gain_tempfile.dat: not retrieved from DB ---\n");
	  }
	// Retrieve the hot cell file from DB - PMD_HOT.root
	status = daqDA_DB_getFile("PMD_HOT.root","PMD_HOT.root");
	if(!status)
	  {
	    calibgain.ExtractHotChannel("PMD_HOT.root");
	  }
	else
	  {
	    printf("--- pmd_gain_tempfile.dat: not retrieved from DB ---\n");
	  }
      }


    // decoding the events
    

    if (argc!=2) {
	printf("Wrong number of arguments\n");
	return -1;
    }
    
    
    /* define data source : this is argument 1 */  
    status=monitorSetDataSource( argv[1] );
    if (status!=0) {
	printf("monitorSetDataSource() failed : %s\n",monitorDecodeError(status));
	return -1;
    }
    
    /* declare monitoring program */
    status=monitorDeclareMp( __FILE__ );
    if (status!=0) {
	printf("monitorDeclareMp() failed : %s\n",monitorDecodeError(status));
	return -1;
    }
    
    /* define wait event timeout - 1s max */
    monitorSetNowait();
    monitorSetNoWaitNetworkTimeout(1000);
    
    /* log start of process */
    printf("PMD GAIN DA - strted generating the gain of a cell\n");  
    
    /* init some counters */
    int nevents_physics=0;
    int nevents_total=0;
    
    struct eventHeaderStruct *event;
    eventTypeType eventT = 0;

    Int_t iev=0;
    
    /* main loop (infinite) */
    for(;;) {
	
	/* check shutdown condition */
	if (daqDA_checkShutdown()) {break;}
	
	/* get next event (blocking call until timeout) */
	status=monitorGetEventDynamic((void **)&event);
	if (status==MON_ERR_EOF) {
	    printf ("End of File detected\n");
	    break; /* end of monitoring file has been reached */
	}
	
	if (status!=0) {
	    printf("monitorGetEventDynamic() failed : %s\n",monitorDecodeError(status));
	    break;
	}
	
	/* retry if got no event */
	if (event==NULL) {
	    continue;
	}
	
	iev++; 
	
	/* use event - here, just write event id to result file */
	nevents_total++;
	eventT=event->eventType;
	switch (event->eventType){
      
	    /* START OF RUN */
	    case START_OF_RUN:
		break;
		/* END START OF RUN */
		
		/* END OF RUN */
	    case END_OF_RUN:
		break;
		
	    case PHYSICS_EVENT:
		nevents_physics++;
		totevt++;
		//if(nevents_physics%100 == 0)printf("Physis Events = %d\n",nevents_physics);
		AliRawReader *rawReader = new AliRawReaderDate((void*)event);
		TObjArray *pmdddlcont = new TObjArray();
		calibgain.ProcessEvent(rawReader, pmdddlcont);

		if (totevt%hotevtsize == 0) hotfilestatus = true;
		delete pmdddlcont;
		pmdddlcont = 0x0;
		delete rawReader;
		rawReader = 0x0;
		
	}
       
	/* free resources */
	free(event);
	
    }

    /* exit when last event received, no need to wait for TERM signal */


    ic    = new TTree("ic","PMD Gain tree");
    meanc = new TTree("meanc","PMD Module mean tree");

    if (filestatus == 0)
      {
	TFile *hotRun = new TFile ("PMD_HOT.root","RECREATE");

	TTree *hot = new TTree("hot","PMD Hot cell tree");
	
	calibgain.FindHotCell(hot,xvar);
	
	hot->Write();
	hotRun->Close();

	// store the hot cell root file in the DB

	status = daqDA_DB_storeFile("PMD_HOT.root","PMD_HOT.root");

	// store the hot cell root file in the file exchange server

	printf("root file for hot cell is created and getting exported\n");
	status = daqDA_FES_storeFile("PMD_HOT.root","PMD_HOT.root");
      }

    if (hotfilestatus)
      {
	TFile *hotRun = new TFile ("PMD_HOT.root","RECREATE");

	TTree *hot = new TTree("hot","PMD Hot cell tree");
	
	calibgain.FindHotCell(hot,xvar);
	
	hot->Write();
	hotRun->Close();

	// store the hot cell root file in the DB

	status = daqDA_DB_storeFile("PMD_HOT.root","PMD_HOT.root");
	
	// store the hot cell root file in the file exchange server

	printf("root file for hot cell is created and getting exported\n");
	status = daqDA_FES_storeFile("PMD_HOT.root","PMD_HOT.root");
      }


    fp1 = fopen("PMD_GAIN_CONFIGFILE","w+");

    if (totevt < maxevt)
      {
	printf("-----------------------------------------------\n");
	printf("***  Required Number of Events not reached  ***\n");
	printf("***  Number of Events processed = %d        ***\n",totevt);
	printf("***  Writing the intermediate ASCII file    ***\n");
	printf("-----------------------------------------------\n");

	calibgain.WriteTempFile("pmd_gain_tempfile.dat");

	// Store the Intermediate ascii file in the DB
	status = daqDA_DB_storeFile("pmd_gain_tempfile.dat","pmd_gain_tempfile.dat");

	filestatus = 1;
	fprintf(fp1,"%d %d %d %d %d\n",filestatus,xvar,totevt,maxevt,hotevtsize);
	fclose(fp1);

	// Store the configfile in the DB
	status = daqDA_DB_storeFile("PMD_GAIN_CONFIGFILE","PMD_GAIN_CONFIGFILE");

      }
    else if (totevt >= maxevt)
      {
	printf("-----------------------------------------------\n");
	printf("***  Required Number of Events reached = %d ***\n",totevt);
	printf("***  Writing the PMDGAINS.root file           ***\n");
	printf("-----------------------------------------------\n");

	calibgain.Analyse(ic, meanc);

	TFile * gainRun = new TFile ("PMDGAINS.root","RECREATE"); 
	ic->Write();
	gainRun->Close();

	TFile * meanRun = new TFile ("PMD_MEAN_SM.root","RECREATE"); 
	meanc->Write();
	meanRun->Close();


	filestatus = 0;
	totevt     = 0;
	fprintf(fp1,"%d %d %d %d %d\n",filestatus,xvar,totevt,maxevt,hotevtsize);
	fclose(fp1);

	// Store the configfile in the DB
	status = daqDA_DB_storeFile("PMD_GAIN_CONFIGFILE","PMD_GAIN_CONFIGFILE");
      }
    
    delete ic;
    ic = 0;

    delete meanc;
    meanc = 0;
    

    /* store the result file on FES */
 
    if (filestatus == 0)
      {
	printf("root file for cell gain is created and getting exported\n");
	status = daqDA_FES_storeFile("PMDGAINS.root","PMDGAINS.root");
	printf("root file for normalised means of different modules\n");
	status = daqDA_FES_storeFile("PMD_MEAN_SM.root","PMD_MEAN_SM.root");
      }

    if (hotfilestatus)
      {
      }

    
    if (status) {
      status = -2;
    }


    return status;
}
Commit	Line	Data
e43e472f	1	/*
	2	PMD DA for online calibration
	3
	4	contact: basanta@phy.iitb.ac.in
d31b0352	5	Link:
d31b0352	6	Reference run:/afs/cern.ch/user/b/bnandi/public/gaindata/pythia100evts.date
e43e472f	7	Run Type: PHYSICS
	8	DA Type: MON
	9	Number of events needed: 1 million for PB+PB, 200 milion for p+p
d31b0352	10	Input Files: PMD_PED.root, PMD_GAIN_CONFIGFILE, pmd_gain_tempfile.dat
d31b0352	11	Output Files: PMDGAINS.root, to be exported to the DAQ FES
e43e472f	12	Trigger types used: PHYSICS_EVENT
	13
	14	*/
	15	extern "C" {
	16	#include <daqDA.h>
	17	}
	18
	19	#include "event.h"
	20	#include "monitor.h"
e43e472f	21
	22	#include <Riostream.h>
	23	#include <stdio.h>
	24	#include <stdlib.h>
	25
	26	//AliRoot
	27	#include "AliRawReaderDate.h"
	28	#include "AliPMDCalibPedestal.h"
	29	#include "AliPMDCalibGain.h"
	30
	31	//ROOT
	32	#include "TFile.h"
	33	#include "TH1F.h"
	34	#include "TBenchmark.h"
	35	#include "TTree.h"
	36	#include "TROOT.h"
	37	#include "TPluginManager.h"
	38
	39
	40
	41	/* Main routine
	42	Arguments:
	43	1- monitoring data source
	44	*/
	45	int main(int argc, char **argv) {
	46
	47	/* magic line from Rene */
	48	gROOT->GetPluginManager()->AddHandler("TVirtualStreamerInfo",
	49	"*",
	50	"TStreamerInfo",
	51	"RIO",
	52	"TStreamerInfo()");
	53
d31b0352	54
	55	int status = 0;
	56
	57
9d1486e3	58	Int_t filestatus = -1, xvar = 5;
9d1486e3	59	Int_t totevt = -1, maxevt = -1;
b2a5fcdc	60	Int_t hotevtsize = -1;
b2a5fcdc	61	Bool_t hotfilestatus = false;
1da50520	62
	63	// Reads the pedestal file and keep the values in memory for subtraction
	64
e43e472f	65	AliPMDCalibGain calibgain;
d31b0352	66
d31b0352	67	// Fetch the pedestal file - PMD_PED.root
9d1486e3	68
d31b0352	69	status = daqDA_DB_getFile("PMD_PED.root","PMD_PED.root");
	70
	71	if(!status)
	72	{
	73	printf("* Pedestal file retrieved from DB * \n");
	74	}
	75	else
	76	{
	77	printf("* Pedestal file NOT retrieved from DB * \n");
	78	return -1;
	79	}
9d1486e3	80
d31b0352	81	Int_t pstatus = calibgain.ExtractPedestal("PMD_PED.root");
1da50520	82
1da50520	83	if(pstatus == -3) return -3;
e43e472f	84
9d1486e3	85	TTree *ic = NULL;
9d1486e3	86	TTree *meanc = NULL;
e43e472f	87
d31b0352	88	// Retrieve the PMD_GAIN_CONFIGFILE
	89	status = daqDA_DB_getFile("PMD_GAIN_CONFIGFILE","PMD_GAIN_CONFIGFILE");
	90
1da50520	91	FILE *fp1 = NULL;
1da50520	92
d31b0352	93	fp1 = fopen("PMD_GAIN_CONFIGFILE","r");
1da50520	94
	95	if (fp1 == NULL)
	96	{
d31b0352	97	printf("* PMD GAIN Configfile doesn't exist,Provide one *\n");
d31b0352	98	return -1;
1da50520	99	}
	100	else
	101	{
b2a5fcdc	102	fscanf(fp1,"%d %d %d %d %d\n",&filestatus, &xvar, &totevt, &maxevt, &hotevtsize);
b2a5fcdc	103	//printf("%d %d %d %d %d\n",filestatus, xvar, totevt, maxevt, hotevtsize);
1da50520	104	}
1da50520	105	fclose(fp1);
9d1486e3	106
e43e472f	107
1da50520	108	if (filestatus == 1)
1da50520	109	{
d31b0352	110	// Retrieve the Temporray ascii file from DB
	111	status = daqDA_DB_getFile("pmd_gain_tempfile.dat","pmd_gain_tempfile.dat");
	112	if(!status)
	113	{
	114	calibgain.ReadTempFile("pmd_gain_tempfile.dat");
	115	}
	116	else
	117	{
	118	printf("--- pmd_gain_tempfile.dat: not retrieved from DB ---\n");
	119	}
9d1486e3	120	// Retrieve the hot cell file from DB - PMD_HOT.root
	121	status = daqDA_DB_getFile("PMD_HOT.root","PMD_HOT.root");
	122	if(!status)
	123	{
	124	calibgain.ExtractHotChannel("PMD_HOT.root");
	125	}
	126	else
	127	{
	128	printf("--- pmd_gain_tempfile.dat: not retrieved from DB ---\n");
	129	}
1da50520	130	}
1da50520	131
9d1486e3	132
e43e472f	133	// decoding the events
e43e472f	134
e43e472f	135
	136	if (argc!=2) {
	137	printf("Wrong number of arguments\n");
	138	return -1;
	139	}
	140
e43e472f	141
	142	/* define data source : this is argument 1 */
	143	status=monitorSetDataSource( argv[1] );
	144	if (status!=0) {
	145	printf("monitorSetDataSource() failed : %s\n",monitorDecodeError(status));
	146	return -1;
	147	}
	148
	149	/* declare monitoring program */
	150	status=monitorDeclareMp( __FILE__ );
	151	if (status!=0) {
	152	printf("monitorDeclareMp() failed : %s\n",monitorDecodeError(status));
	153	return -1;
	154	}
	155
	156	/* define wait event timeout - 1s max */
	157	monitorSetNowait();
	158	monitorSetNoWaitNetworkTimeout(1000);
	159
	160	/* log start of process */
1da50520	161	printf("PMD GAIN DA - strted generating the gain of a cell\n");
e43e472f	162
	163	/* init some counters */
	164	int nevents_physics=0;
	165	int nevents_total=0;
	166
	167	struct eventHeaderStruct *event;
338c285a	168	eventTypeType eventT = 0;
338c285a	169
e43e472f	170	Int_t iev=0;
	171
	172	/* main loop (infinite) */
	173	for(;;) {
	174
	175	/* check shutdown condition */
	176	if (daqDA_checkShutdown()) {break;}
	177
	178	/* get next event (blocking call until timeout) */
	179	status=monitorGetEventDynamic((void **)&event);
	180	if (status==MON_ERR_EOF) {
	181	printf ("End of File detected\n");
	182	break; /* end of monitoring file has been reached */
	183	}
	184
	185	if (status!=0) {
	186	printf("monitorGetEventDynamic() failed : %s\n",monitorDecodeError(status));
	187	break;
	188	}
	189
	190	/* retry if got no event */
	191	if (event==NULL) {
	192	continue;
	193	}
	194
	195	iev++;
	196
	197	/* use event - here, just write event id to result file */
	198	nevents_total++;
	199	eventT=event->eventType;
	200	switch (event->eventType){
	201
	202	/* START OF RUN */
	203	case START_OF_RUN:
	204	break;
	205	/* END START OF RUN */
	206
	207	/* END OF RUN */
	208	case END_OF_RUN:
	209	break;
	210
	211	case PHYSICS_EVENT:
	212	nevents_physics++;
b2a5fcdc	213	totevt++;
1da50520	214	//if(nevents_physics%100 == 0)printf("Physis Events = %d\n",nevents_physics);
e43e472f	215	AliRawReader rawReader = new AliRawReaderDate((void)event);
338c285a	216	TObjArray *pmdddlcont = new TObjArray();
338c285a	217	calibgain.ProcessEvent(rawReader, pmdddlcont);
e43e472f	218
b2a5fcdc	219	if (totevt%hotevtsize == 0) hotfilestatus = true;
338c285a	220	delete pmdddlcont;
338c285a	221	pmdddlcont = 0x0;
e43e472f	222	delete rawReader;
	223	rawReader = 0x0;
	224
	225	}
	226
	227	/* free resources */
	228	free(event);
	229
338c285a	230	}
	231
	232	/* exit when last event received, no need to wait for TERM signal */
40a42048	233
9d1486e3	234
	235	ic = new TTree("ic","PMD Gain tree");
	236	meanc = new TTree("meanc","PMD Module mean tree");
	237
	238	if (filestatus == 0)
	239	{
	240	TFile *hotRun = new TFile ("PMD_HOT.root","RECREATE");
	241
	242	TTree *hot = new TTree("hot","PMD Hot cell tree");
	243
	244	calibgain.FindHotCell(hot,xvar);
	245
	246	hot->Write();
	247	hotRun->Close();
	248
	249	// store the hot cell root file in the DB
	250
	251	status = daqDA_DB_storeFile("PMD_HOT.root","PMD_HOT.root");
b2a5fcdc	252
	253	// store the hot cell root file in the file exchange server
	254
	255	printf("root file for hot cell is created and getting exported\n");
	256	status = daqDA_FES_storeFile("PMD_HOT.root","PMD_HOT.root");
9d1486e3	257	}
9d1486e3	258
b2a5fcdc	259	if (hotfilestatus)
	260	{
	261	TFile *hotRun = new TFile ("PMD_HOT.root","RECREATE");
	262
	263	TTree *hot = new TTree("hot","PMD Hot cell tree");
	264
	265	calibgain.FindHotCell(hot,xvar);
	266
	267	hot->Write();
	268	hotRun->Close();
	269
	270	// store the hot cell root file in the DB
	271
	272	status = daqDA_DB_storeFile("PMD_HOT.root","PMD_HOT.root");
	273
	274	// store the hot cell root file in the file exchange server
	275
	276	printf("root file for hot cell is created and getting exported\n");
	277	status = daqDA_FES_storeFile("PMD_HOT.root","PMD_HOT.root");
	278	}
e43e472f	279
1da50520	280
d31b0352	281	fp1 = fopen("PMD_GAIN_CONFIGFILE","w+");
1da50520	282
	283	if (totevt < maxevt)
	284	{
d31b0352	285	printf("-----------------------------------------------\n");
	286	printf("* Required Number of Events not reached *\n");
	287	printf("* Number of Events processed = %d *\n",totevt);
	288	printf("* Writing the intermediate ASCII file *\n");
	289	printf("-----------------------------------------------\n");
	290
	291	calibgain.WriteTempFile("pmd_gain_tempfile.dat");
	292
	293	// Store the Intermediate ascii file in the DB
	294	status = daqDA_DB_storeFile("pmd_gain_tempfile.dat","pmd_gain_tempfile.dat");
1da50520	295
1da50520	296	filestatus = 1;
b2a5fcdc	297	fprintf(fp1,"%d %d %d %d %d\n",filestatus,xvar,totevt,maxevt,hotevtsize);
9d1486e3	298	fclose(fp1);
d31b0352	299
	300	// Store the configfile in the DB
	301	status = daqDA_DB_storeFile("PMD_GAIN_CONFIGFILE","PMD_GAIN_CONFIGFILE");
	302
1da50520	303	}
	304	else if (totevt >= maxevt)
	305	{
d31b0352	306	printf("-----------------------------------------------\n");
	307	printf("* Required Number of Events reached = %d *\n",totevt);
	308	printf("* Writing the PMDGAINS.root file *\n");
	309	printf("-----------------------------------------------\n");
	310
9d1486e3	311	calibgain.Analyse(ic, meanc);
1da50520	312
	313	TFile * gainRun = new TFile ("PMDGAINS.root","RECREATE");
	314	ic->Write();
	315	gainRun->Close();
	316
9d1486e3	317	TFile * meanRun = new TFile ("PMD_MEAN_SM.root","RECREATE");
	318	meanc->Write();
	319	meanRun->Close();
	320
	321
1da50520	322	filestatus = 0;
1da50520	323	totevt = 0;
b2a5fcdc	324	fprintf(fp1,"%d %d %d %d %d\n",filestatus,xvar,totevt,maxevt,hotevtsize);
9d1486e3	325	fclose(fp1);
d31b0352	326
	327	// Store the configfile in the DB
	328	status = daqDA_DB_storeFile("PMD_GAIN_CONFIGFILE","PMD_GAIN_CONFIGFILE");
1da50520	329	}
e43e472f	330
338c285a	331	delete ic;
338c285a	332	ic = 0;
9d1486e3	333
	334	delete meanc;
	335	meanc = 0;
1da50520	336
	337
	338	/* store the result file on FES */
	339
	340	if (filestatus == 0)
	341	{
9d1486e3	342	printf("root file for cell gain is created and getting exported\n");
a5f36c63	343	status = daqDA_FES_storeFile("PMDGAINS.root","PMDGAINS.root");
9d1486e3	344	printf("root file for normalised means of different modules\n");
9d1486e3	345	status = daqDA_FES_storeFile("PMD_MEAN_SM.root","PMD_MEAN_SM.root");
1da50520	346	}
b2a5fcdc	347
	348	if (hotfilestatus)
	349	{
	350	}
	351
9d1486e3	352
1da50520	353	if (status) {
	354	status = -2;
	355	}
338c285a	356
338c285a	357
9d1486e3	358
e43e472f	359	return status;
e43e472f	360	}