[u/mrichter/AliRoot.git] / ITS / ITSSPDVertexDiamondda.cxx

/*
Contact: cvetan.cheshkov@cern.ch
Link: http://alisoft.cern.ch/viewvc/trunk/ITS/ITSSPDVertexDiamondda.cxx?root=AliRoot&view=log , /afs/cern.ch/user/c/cheshkov/public/ITS/VD_da_test.date , /afs/cern.ch/user/c/cheshkov/public/08000058338016.30.root.date.gz
Reference Run: 58338
Run Type: PHYSICS
DA Type: MON
Number of events needed: 100
Input Files: GRP/Geometry/Data , ITS/Align/Data , spd_noisy_ocdb , spd_dead_ocdb , TRIGGER/SPD/PITConditions (all the files are taken from SPD daqDetDB)
Output Files: SPDVertexDiamondDA.root
Trigger types used: PHYSICS, SPD-F0 
*/

#define OUTPUT_FILE "SPDVertexDiamondDA.root"
#define N_EVENTS_AUTOSAVE 50

extern "C" {
#include "daqDA.h"
}

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

#ifdef ALI_AMORE
#include <AmoreDA.h>
//int amore::da::Updated(char const*) {}
#endif

#include <TPluginManager.h>
#include <TROOT.h>
#include <TH1.h>
#include <TH2.h>
#include <TSystem.h>
#include <TGeoGlobalMagField.h>

#include "AliLog.h"
#include "AliMagF.h"
#include "AliRawReaderDate.h"
#include "AliCDBManager.h"
#include "AliITSMeanVertexer.h"

int main(int argc, char **argv) {

  gROOT->GetPluginManager()->AddHandler("TVirtualStreamerInfo",
					"*",
					"TStreamerInfo",
					"RIO",
					"TStreamerInfo()"); 

  int status;
  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("Vertex-Diamond SPD DA started\n");  

  /* init some counters */
  int nevents_with_vertex = 0;
  int nevents_physics=0;
  int nevents_total=0;

  struct eventHeaderStruct *event;
  eventTypeType eventT;

  // Get run number
  if (getenv("DATE_RUN_NUMBER")==0) {
    printf("DATE_RUN_NUMBER not properly set.\n");
    return -1;
  }
  int runNr = atoi(getenv("DATE_RUN_NUMBER"));

  // Get the necessary OCDB files from the DAQ detector DB
  if (gSystem->AccessPathName("localOCDB/GRP/Geometry/Data",kFileExists)) {
    if (gSystem->mkdir("localOCDB/GRP/Geometry/Data",kTRUE) != 0) {
      printf("Failed to create directory: localOCDB/GRP/Geometry/Data");
      return -1;
    }
  }
  status = daqDA_DB_getFile("GRP/Geometry/Data","localOCDB/GRP/Geometry/Data/Run0_999999999_v0_s0.root");
  if (status) {
    printf("Failed to get geometry file (GRP/Geometry/Data) from DAQdetDB, status=%d\n", status);
    return -1;
  }

  if (gSystem->AccessPathName("localOCDB/ITS/Align/Data",kFileExists)) {
    if (gSystem->mkdir("localOCDB/ITS/Align/Data",kTRUE) != 0) {
      printf("Failed to create directory: localOCDB/ITS/Align/Data");
      return -1;
    }
  }
  status = daqDA_DB_getFile("ITS/Align/Data","localOCDB/ITS/Align/Data/Run0_999999999_v0_s0.root");
  if (status) {
    printf("Failed to get its-alignment file (ITS/Align/Data) from DAQdetDB, status=%d\n", status);
    return -1;
  }

  if (gSystem->AccessPathName("localOCDB/ITS/Calib/SPDNoisy",kFileExists)) {
    if (gSystem->mkdir("localOCDB/ITS/Calib/SPDNoisy",kTRUE) != 0) {
      printf("Failed to create directory: localOCDB/ITS/Calib/SPDNoisy");
      return -1;
    }
  }
  status = daqDA_DB_getFile("spd_noisy_ocdb","localOCDB/ITS/Calib/SPDNoisy/Run0_999999999_v0_s0.root");
  if (status) {
    printf("Failed to get spd file (spd_noisy_ocdb) from DAQdetDB, status=%d\n", status);
    return -1;
  }

  if (gSystem->AccessPathName("localOCDB/ITS/Calib/SPDDead",kFileExists)) {
    if (gSystem->mkdir("localOCDB/ITS/Calib/SPDDead",kTRUE) != 0) {
      printf("Failed to create directory: localOCDB/ITS/Calib/SPDDead");
      return -1;
    }
  }
  status = daqDA_DB_getFile("spd_dead_ocdb","localOCDB/ITS/Calib/SPDDead/Run0_999999999_v0_s0.root");
  if (status) {
    printf("Failed to get spd file (spd_dead_ocdb) from DAQdetDB, status=%d\n", status);
    return -1;
  }

  if (gSystem->AccessPathName("localOCDB/TRIGGER/SPD/PITConditions",kFileExists)) {
    if (gSystem->mkdir("localOCDB/TRIGGER/SPD/PITConditions",kTRUE) != 0) {
      printf("Failed to create directory: localOCDB/TRIGGER/SPD/PITConditions");
      return -1;
    }
  }
  status = daqDA_DB_getFile("TRIGGER/SPD/PITConditions","localOCDB/TRIGGER/SPD/PITConditions/Run0_999999999_v0_s0.root");
  if (status) {
    printf("Failed to get spd trigger file (TRIGGER/SPD/PITConditions) from DAQdetDB, status=%d\n", status);
    return -1;
  }
 
 status = daqDA_DB_getFile("mfchebKGI_sym.root","localOCDB/mfchebKGI_sym.root");
  if (status) {
    printf("Failed to get spd file (mfchebKGI_sym.root) from DAQdetDB, status=%d\n", status);
    return -1;
  }

  // Global initializations

  // The B filed is required in AliITSClusterFinderV2SPD
  // for the Lorentz angle correction. B set to 0.      
  TGeoGlobalMagField::Instance()->SetField(new AliMagF("Maps","Maps", 0., 0., AliMagF::k5kGUniform,AliMagF::kBeamTypeAA,-1,2,15,"localOCDB/mfchebKGI_sym.root"));  
  AliLog::SetGlobalLogLevel(AliLog::kError);
  AliCDBManager *man = AliCDBManager::Instance();
  man->SetDefaultStorage("local://localOCDB");
  man->SetRun(runNr);

  // Init mean vertexer
  AliITSMeanVertexer *mv = new AliITSMeanVertexer();
  if (!mv->Init()) {
    printf("Initialization of mean vertexer object failed ! Check the log for details");
    return -1;
  }

  // Initialization of AMORE sender
#ifdef ALI_AMORE
  amore::da::AmoreDA vtxAmore(amore::da::AmoreDA::kSender);
#endif
  /* 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;
    }

    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++;
      AliRawReader *rawReader = new AliRawReaderDate((void*)event);

      // Run mean-vertexer reco
      if (mv->Reconstruct(rawReader)) nevents_with_vertex++;

      // Auto save
      if ((nevents_physics%N_EVENTS_AUTOSAVE) == 0) {

	((TH2F*)mv->GetVertexXY())->SetTitle(Form("%f events with vertex (%i out of %i processed events)",((Double_t)nevents_with_vertex)/((Double_t)nevents_physics),nevents_with_vertex,nevents_physics));

	mv->WriteVertices(OUTPUT_FILE);

#ifdef ALI_AMORE
      // send the histos to AMORE pool
	printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexXY()->GetName(),mv->GetVertexXY()));
	printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexZ()->GetName(),mv->GetVertexZ()));
#endif
      }

      delete rawReader;
    }
    
    /* free resources */
    free(event);
    
    /* exit when last event received, no need to wait for TERM signal */
    if (eventT==END_OF_RUN) {
      printf("EOR event detected\n");
      break;
    }
  }
  ((TH2F*)mv->GetVertexXY())->SetTitle(Form("%f events with vertex (%i out of %i processed events)",((Double_t)nevents_with_vertex)/((Double_t)nevents_physics),nevents_with_vertex,nevents_physics));

  mv->WriteVertices(OUTPUT_FILE);

#ifdef ALI_AMORE
  // send the histos to AMORE pool
  printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexXY()->GetName(),mv->GetVertexXY()));
  printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexZ()->GetName(),mv->GetVertexZ()));
#endif

  delete mv;

  /* write report */
  printf("Run #%s, received %d events with vertex, out of %d physics and out of %d total events\n",getenv("DATE_RUN_NUMBER"),nevents_with_vertex,nevents_physics,nevents_total);

  status=0;

  /* export file to FXS */
  if (daqDA_FES_storeFile(OUTPUT_FILE, "VertexDiamond")) {
    status=-2;
  }
  
  return status;
}
Commit	Line	Data
eb35e591	1	/*
eb35e591	2	Contact: cvetan.cheshkov@cern.ch
9141d382	3	Link: http://alisoft.cern.ch/viewvc/trunk/ITS/ITSSPDVertexDiamondda.cxx?root=AliRoot&view=log , /afs/cern.ch/user/c/cheshkov/public/ITS/VD_da_test.date , /afs/cern.ch/user/c/cheshkov/public/08000058338016.30.root.date.gz
6bf5b296	4	Reference Run: 58338
eb35e591	5	Run Type: PHYSICS
eb35e591	6	DA Type: MON
049f9cac	7	Number of events needed: 100
9141d382	8	Input Files: GRP/Geometry/Data , ITS/Align/Data , spd_noisy_ocdb , spd_dead_ocdb , TRIGGER/SPD/PITConditions (all the files are taken from SPD daqDetDB)
6bf5b296	9	Output Files: SPDVertexDiamondDA.root
9141d382	10	Trigger types used: PHYSICS, SPD-F0
eb35e591	11	*/
eb35e591	12
eb35e591	13	#define OUTPUT_FILE "SPDVertexDiamondDA.root"
eb35e591	14	#define N_EVENTS_AUTOSAVE 50
eb35e591	15
	16	extern "C" {
	17	#include "daqDA.h"
	18	}
	19
	20	#include "event.h"
	21	#include "monitor.h"
	22
55c5e86d	23	#ifdef ALI_AMORE
55c5e86d	24	#include <AmoreDA.h>
308c2f7c	25	//int amore::da::Updated(char const*) {}
55c5e86d	26	#endif
55c5e86d	27
eb35e591	28	#include <TPluginManager.h>
eb35e591	29	#include <TROOT.h>
cf0b1cea	30	#include <TH1.h>
cf0b1cea	31	#include <TH2.h>
6bf5b296	32	#include <TSystem.h>
ee0a8262	33	#include <TGeoGlobalMagField.h>
eb35e591	34
e1f6be44	35	#include "AliLog.h"
ee0a8262	36	#include "AliMagF.h"
eb35e591	37	#include "AliRawReaderDate.h"
eb35e591	38	#include "AliCDBManager.h"
cf0b1cea	39	#include "AliITSMeanVertexer.h"
eb35e591	40
	41	int main(int argc, char **argv) {
	42
	43	gROOT->GetPluginManager()->AddHandler("TVirtualStreamerInfo",
	44	"*",
	45	"TStreamerInfo",
	46	"RIO",
	47	"TStreamerInfo()");
eb35e591	48
	49	int status;
	50	if (argc<2) {
	51	printf("Wrong number of arguments\n");
	52	return -1;
	53	}
	54
	55	/* define data source : this is argument 1 */
	56	status=monitorSetDataSource( argv[1] );
	57	if (status!=0) {
	58	printf("monitorSetDataSource() failed : %s\n",monitorDecodeError(status));
	59	return -1;
	60	}
	61
	62	/* declare monitoring program */
	63	status=monitorDeclareMp( __FILE__ );
	64	if (status!=0) {
	65	printf("monitorDeclareMp() failed : %s\n",monitorDecodeError(status));
	66	return -1;
	67	}
	68
	69	/* define wait event timeout - 1s max */
	70	monitorSetNowait();
	71	monitorSetNoWaitNetworkTimeout(1000);
	72
	73	/* log start of process */
	74	printf("Vertex-Diamond SPD DA started\n");
	75
	76	/* init some counters */
	77	int nevents_with_vertex = 0;
	78	int nevents_physics=0;
	79	int nevents_total=0;
	80
	81	struct eventHeaderStruct *event;
	82	eventTypeType eventT;
	83
	84	// Get run number
	85	if (getenv("DATE_RUN_NUMBER")==0) {
	86	printf("DATE_RUN_NUMBER not properly set.\n");
	87	return -1;
	88	}
	89	int runNr = atoi(getenv("DATE_RUN_NUMBER"));
	90
6bf5b296	91	// Get the necessary OCDB files from the DAQ detector DB
	92	if (gSystem->AccessPathName("localOCDB/GRP/Geometry/Data",kFileExists)) {
	93	if (gSystem->mkdir("localOCDB/GRP/Geometry/Data",kTRUE) != 0) {
	94	printf("Failed to create directory: localOCDB/GRP/Geometry/Data");
	95	return -1;
	96	}
	97	}
	98	status = daqDA_DB_getFile("GRP/Geometry/Data","localOCDB/GRP/Geometry/Data/Run0_999999999_v0_s0.root");
	99	if (status) {
	100	printf("Failed to get geometry file (GRP/Geometry/Data) from DAQdetDB, status=%d\n", status);
	101	return -1;
	102	}
	103
	104	if (gSystem->AccessPathName("localOCDB/ITS/Align/Data",kFileExists)) {
	105	if (gSystem->mkdir("localOCDB/ITS/Align/Data",kTRUE) != 0) {
	106	printf("Failed to create directory: localOCDB/ITS/Align/Data");
	107	return -1;
	108	}
	109	}
	110	status = daqDA_DB_getFile("ITS/Align/Data","localOCDB/ITS/Align/Data/Run0_999999999_v0_s0.root");
	111	if (status) {
	112	printf("Failed to get its-alignment file (ITS/Align/Data) from DAQdetDB, status=%d\n", status);
	113	return -1;
	114	}
	115
	116	if (gSystem->AccessPathName("localOCDB/ITS/Calib/SPDNoisy",kFileExists)) {
	117	if (gSystem->mkdir("localOCDB/ITS/Calib/SPDNoisy",kTRUE) != 0) {
	118	printf("Failed to create directory: localOCDB/ITS/Calib/SPDNoisy");
	119	return -1;
	120	}
	121	}
	122	status = daqDA_DB_getFile("spd_noisy_ocdb","localOCDB/ITS/Calib/SPDNoisy/Run0_999999999_v0_s0.root");
	123	if (status) {
	124	printf("Failed to get spd file (spd_noisy_ocdb) from DAQdetDB, status=%d\n", status);
	125	return -1;
	126	}
	127
	128	if (gSystem->AccessPathName("localOCDB/ITS/Calib/SPDDead",kFileExists)) {
	129	if (gSystem->mkdir("localOCDB/ITS/Calib/SPDDead",kTRUE) != 0) {
	130	printf("Failed to create directory: localOCDB/ITS/Calib/SPDDead");
	131	return -1;
	132	}
	133	}
	134	status = daqDA_DB_getFile("spd_dead_ocdb","localOCDB/ITS/Calib/SPDDead/Run0_999999999_v0_s0.root");
	135	if (status) {
	136	printf("Failed to get spd file (spd_dead_ocdb) from DAQdetDB, status=%d\n", status);
	137	return -1;
	138	}
	139
9141d382	140	if (gSystem->AccessPathName("localOCDB/TRIGGER/SPD/PITConditions",kFileExists)) {
	141	if (gSystem->mkdir("localOCDB/TRIGGER/SPD/PITConditions",kTRUE) != 0) {
	142	printf("Failed to create directory: localOCDB/TRIGGER/SPD/PITConditions");
	143	return -1;
	144	}
	145	}
	146	status = daqDA_DB_getFile("TRIGGER/SPD/PITConditions","localOCDB/TRIGGER/SPD/PITConditions/Run0_999999999_v0_s0.root");
	147	if (status) {
	148	printf("Failed to get spd trigger file (TRIGGER/SPD/PITConditions) from DAQdetDB, status=%d\n", status);
	149	return -1;
	150	}
2a16d0d6	151
	152	status = daqDA_DB_getFile("mfchebKGI_sym.root","localOCDB/mfchebKGI_sym.root");
	153	if (status) {
	154	printf("Failed to get spd file (mfchebKGI_sym.root) from DAQdetDB, status=%d\n", status);
	155	return -1;
	156	}
bdffed96	157
eb35e591	158	// Global initializations
ee0a8262	159
	160	// The B filed is required in AliITSClusterFinderV2SPD
	161	// for the Lorentz angle correction. B set to 0.
2a16d0d6	162	TGeoGlobalMagField::Instance()->SetField(new AliMagF("Maps","Maps", 0., 0., AliMagF::k5kGUniform,AliMagF::kBeamTypeAA,-1,2,15,"localOCDB/mfchebKGI_sym.root"));
e1f6be44	163	AliLog::SetGlobalLogLevel(AliLog::kError);
eb35e591	164	AliCDBManager *man = AliCDBManager::Instance();
9141d382	165	man->SetDefaultStorage("local://localOCDB");
eb35e591	166	man->SetRun(runNr);
eb35e591	167
37add1d1	168	// Init mean vertexer
	169	AliITSMeanVertexer *mv = new AliITSMeanVertexer();
	170	if (!mv->Init()) {
	171	printf("Initialization of mean vertexer object failed ! Check the log for details");
	172	return -1;
	173	}
eb35e591	174
55c5e86d	175	// Initialization of AMORE sender
	176	#ifdef ALI_AMORE
	177	amore::da::AmoreDA vtxAmore(amore::da::AmoreDA::kSender);
	178	#endif
eb35e591	179	/* main loop (infinite) */
	180	for(;;) {
	181
	182	/* check shutdown condition */
	183	if (daqDA_checkShutdown()) {break;}
	184
	185	/* get next event (blocking call until timeout) */
	186	status=monitorGetEventDynamic((void **)&event);
	187	if (status==MON_ERR_EOF) {
	188	printf ("End of File detected\n");
	189	break; /* end of monitoring file has been reached */
	190	}
	191
	192	if (status!=0) {
	193	printf("monitorGetEventDynamic() failed : %s\n",monitorDecodeError(status));
	194	break;
	195	}
	196
	197	/* retry if got no event */
	198	if (event==NULL) {
	199	continue;
	200	}
	201
	202	nevents_total++;
	203	eventT=event->eventType;
	204	switch (event->eventType){
	205
	206	/* START OF RUN */
	207	case START_OF_RUN:
	208	break;
	209	/* END START OF RUN */
	210
	211	/* END OF RUN */
	212	case END_OF_RUN:
	213	break;
	214
	215	case PHYSICS_EVENT:
	216	nevents_physics++;
	217	AliRawReader rawReader = new AliRawReaderDate((void)event);
	218
37add1d1	219	// Run mean-vertexer reco
37add1d1	220	if (mv->Reconstruct(rawReader)) nevents_with_vertex++;
eb35e591	221
37add1d1	222	// Auto save
6bf5b296	223	if ((nevents_physics%N_EVENTS_AUTOSAVE) == 0) {
5dc5a279	224
	225	((TH2F*)mv->GetVertexXY())->SetTitle(Form("%f events with vertex (%i out of %i processed events)",((Double_t)nevents_with_vertex)/((Double_t)nevents_physics),nevents_with_vertex,nevents_physics));
	226
37add1d1	227	mv->WriteVertices(OUTPUT_FILE);
eb35e591	228
6bf5b296	229	#ifdef ALI_AMORE
	230	// send the histos to AMORE pool
	231	printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexXY()->GetName(),mv->GetVertexXY()));
	232	printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexZ()->GetName(),mv->GetVertexZ()));
	233	#endif
	234	}
	235
eb35e591	236	delete rawReader;
eb35e591	237	}
	238
	239	/* free resources */
	240	free(event);
	241
	242	/* exit when last event received, no need to wait for TERM signal */
	243	if (eventT==END_OF_RUN) {
	244	printf("EOR event detected\n");
	245	break;
	246	}
	247	}
5dc5a279	248	((TH2F*)mv->GetVertexXY())->SetTitle(Form("%f events with vertex (%i out of %i processed events)",((Double_t)nevents_with_vertex)/((Double_t)nevents_physics),nevents_with_vertex,nevents_physics));
eb35e591	249
37add1d1	250	mv->WriteVertices(OUTPUT_FILE);
eb35e591	251
55c5e86d	252	#ifdef ALI_AMORE
55c5e86d	253	// send the histos to AMORE pool
cf0b1cea	254	printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexXY()->GetName(),mv->GetVertexXY()));
cf0b1cea	255	printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexZ()->GetName(),mv->GetVertexZ()));
55c5e86d	256	#endif
55c5e86d	257
37add1d1	258	delete mv;
eb35e591	259
	260	/* write report */
	261	printf("Run #%s, received %d events with vertex, out of %d physics and out of %d total events\n",getenv("DATE_RUN_NUMBER"),nevents_with_vertex,nevents_physics,nevents_total);
	262
	263	status=0;
	264
	265	/* export file to FXS */
	266	if (daqDA_FES_storeFile(OUTPUT_FILE, "VertexDiamond")) {
	267	status=-2;
	268	}
	269
	270	return status;
	271	}