[u/mrichter/AliRoot.git] / ZDC / AliZDCPreprocessor.cxx

// --- ROOT system
#include <TFile.h>
#include <TClonesArray.h>
#include <TList.h>
#include <TObjString.h>
#include <TTimeStamp.h>

#include "AliZDCPreprocessor.h"
#include "AliCDBManager.h"
#include "AliCDBEntry.h"
#include "AliCDBMetaData.h"
#include "AliDCSValue.h"
#include "AliAlignObj.h"
#include "AliAlignObjParams.h"
#include "AliLog.h"
#include "AliZDCDataDCS.h"
#include "AliZDCCalibData.h"

//
// Class implementing ZDC pre-processor.
// It takes data from DCS and passes it to the class AliZDCDataDCS.
// The class is then written to the CDB.
//

ClassImp(AliZDCPreprocessor)

//______________________________________________________________________________________________
AliZDCPreprocessor::AliZDCPreprocessor(AliShuttleInterface* shuttle) :
  AliPreprocessor("ZDC", shuttle),
  fData(0)
{
  // constructor
}

//______________________________________________________________________________________________
AliZDCPreprocessor::~AliZDCPreprocessor()
{
  // destructor
}


//______________________________________________________________________________________________
void AliZDCPreprocessor::Initialize(Int_t run, UInt_t startTime,
	UInt_t endTime)
{
  // Creates AliZDCDataDCS object

  AliPreprocessor::Initialize(run, startTime, endTime);

	Log(Form("\n\tRun %d \n\tStartTime %s \n\tEndTime %s", run,
		TTimeStamp(startTime).AsString(),
		TTimeStamp(endTime).AsString()));

 	fRun = run;
        fStartTime = startTime;
        fEndTime = endTime;

	fData = new AliZDCDataDCS(fRun, fStartTime, fEndTime);
}

//______________________________________________________________________________________________
UInt_t AliZDCPreprocessor::Process(TMap* dcsAliasMap)
{
  // *************** From DCS ******************
  // Fills data into a AliZDCDataDCS object
  if(!dcsAliasMap) return 1;

  // The processing of the DCS input data is forwarded to AliZDCDataDCS
  Float_t DCSValues[28]; // DCSAliases=28
  fData->ProcessData(*dcsAliasMap, DCSValues);
  //dcsAliasMap->Print("");
  //
  // --- Writing ZDC table positions into alignment object
  TClonesArray *array = new TClonesArray("AliAlignObjParams",10);
  TClonesArray &alobj = *array;
  AliAlignObjParams a;
  Double_t dx=0., dz=0., dpsi=0., dtheta=0., dphi=0.;
  // Vertical table position in mm from DCS
  Double_t dyZN1 = (Double_t) (DCSValues[0]/10.);
  Double_t dyZP1 = (Double_t) (DCSValues[1]/10.);
  Double_t dyZN2 = (Double_t) (DCSValues[2]/10.);
  Double_t dyZP2 = (Double_t) (DCSValues[3]/10.);
  const char *ZDCn1="ZDC/NeutronZDC1";
  const char *ZDCp1="ZDC/ProtonZDC1";
  const char *ZDCn2="ZDC/NeutronZDC2";
  const char *ZDCp2="ZDC/ProtonZDC2";
  UShort_t iIndex=0;
  AliGeomManager::ELayerID iLayer = AliGeomManager::kInvalidLayer;
  UShort_t volid = AliGeomManager::LayerToVolUID(iLayer,iIndex);
  //
  new(alobj[0]) AliAlignObjParams(ZDCn1, volid, dx, dyZN1, dz, dpsi, dtheta, dphi, kTRUE);
  new(alobj[1]) AliAlignObjParams(ZDCp1, volid, dx, dyZP1, dz, dpsi, dtheta, dphi, kTRUE);
  new(alobj[2]) AliAlignObjParams(ZDCn2, volid, dx, dyZN2, dz, dpsi, dtheta, dphi, kTRUE);
  new(alobj[3]) AliAlignObjParams(ZDCp2, volid, dx, dyZP2, dz, dpsi, dtheta, dphi, kTRUE);
  // save in CDB storage
  AliCDBMetaData md;
  md.SetResponsible("Chiara Oppedisano");
  md.SetComment("Alignment object for ZDC");
  Bool_t resultAl = kFALSE;
  resultAl = Store("Align","Data", array, &md, 0, 0);
  
  AliZDCCalibData *calibdata = new AliZDCCalibData("ZDC");
  
// *************** From DAQ ******************
// *****************************************************
// [a] PEDESTALS -> Pedestal subtraction
// *****************************************************
TString runType = GetRunType();
printf("\n\t AliZDCPreprocessor -> runType detected %s\n\n",runType.Data());
if(runType == "PEDESTAL_RUN") {
  TList* daqSources = GetFileSources(kDAQ, "PEDESTALS");
  if(!daqSources){
    Log(Form("No source for PEDESTALS run %d !", fRun));
    return 1;
  }
  Log("\t List of sources for PEDESTALS");
  daqSources->Print();
  //
  TIter iter(daqSources);
  TObjString* source = 0;
  Int_t i=0;
  while((source = dynamic_cast<TObjString*> (iter.Next()))){
       Log(Form("\n\t Getting file #%d\n",++i));
       TString stringPedFileName = GetFile(kDAQ, "PEDESTALS", source->GetName());
       if(stringPedFileName.Length() <= 0){
          Log(Form("No PEDESTAL file from source %s!", source->GetName()));
	  return 1;
       }
       const char* PedFileName = stringPedFileName.Data();
       // no. ADCch = (22 signal ch. + 2 reference PMs) * 2 gain chain = 48
       const Int_t NZDCch = 48;
       if(PedFileName){
         FILE *file;
         if((file = fopen(PedFileName,"r")) == NULL){
           printf("Cannot open file %s \n",PedFileName);
	   return 1;
         }
         Log(Form("File %s connected to analyze pedestal events", PedFileName));
         Float_t PedVal[(3*NZDCch)][2];
         for(Int_t i=0; i<(3*NZDCch); i++){
            for(Int_t j=0; j<2; j++){
               fscanf(file,"%f",&PedVal[i][j]);
	       //if(j==1) printf("PedVal[%d] -> %f, %f \n",i,PedVal[i][0],PedVal[i][1]);
            }
            if(i<NZDCch){
              calibdata->SetMeanPed(i,PedVal[i][0]);
              calibdata->SetMeanPedWidth(i,PedVal[i][1]);
            }
            else if(i>=NZDCch && i<(2*NZDCch)){
              calibdata->SetOOTPed(i-NZDCch,PedVal[i][0]);
              calibdata->SetOOTPedWidth(i-NZDCch,PedVal[i][1]);
            }
            else if(i>=(2*NZDCch) && i<(3*NZDCch)){
              calibdata->SetPedCorrCoeff(i-(2*NZDCch),PedVal[i][0],PedVal[i][1]);
            }
         }
       }
       else{
          Log(Form("File %s not found", PedFileName));
          return 1;
       }
       //
      //calibdata->Print("");
  }
  delete daqSources; daqSources = 0;
}
// *****************************************************
// [b] EMD EVENTS -> Energy calibration and equalization
// *****************************************************
else if(runType == "PULSER_RUN"){
  TList* daqSources = GetFileSources(kDAQ, "EMDCALIB");
  if(!daqSources){
    AliError(Form("No sources for PULSER_RUN run %d !", fRun));
    return 1;
  }
  Log("\t List of sources for PULSER_RUN");
  daqSources->Print();
  //
  TIter iter2(daqSources);
  TObjString* source = 0;
  Int_t j=0;
  while((source = dynamic_cast<TObjString*> (iter2.Next()))){
       Log(Form("\n\t Getting file #%d\n",++j));
       TString stringEMDFileName = GetFile(kDAQ, "EMDCALIB", source->GetName());
       if(stringEMDFileName.Length() <= 0){
         Log(Form("No EMDCALIB file from source %s!", source->GetName()));
	 return 1;
       }
       const char* EMDFileName = stringEMDFileName.Data();
       if(EMDFileName){
    	 FILE *file;
    	 if((file = fopen(EMDFileName,"r")) == NULL){
    	   printf("Cannot open file %s \n",EMDFileName);
	   return 1;
    	 }
    	 Log(Form("File %s connected to analyze EM dissociation events", EMDFileName));
    	 Float_t fitValEMD[6]; Float_t equalCoeff[5][4];
	 Float_t CalibVal[4];
    	 for(Int_t j=0; j<10; j++){	    
    	   if(j<6){
	     fscanf(file,"%f",&fitValEMD[j]);
             if(j<4){
	       CalibVal[j] = fitValEMD[j]/2.76;
	       calibdata->SetEnCalib(j,CalibVal[j]);
	     }
	     else calibdata->SetEnCalib(j,fitValEMD[j]);
	   }
	   else{
	     for(Int_t k=0; k<5; k++){
	        fscanf(file,"%f",&equalCoeff[j][k]);
	        if(j==6) calibdata->SetZN1EqualCoeff(k, equalCoeff[j][k]);
	 	else if(j==7) calibdata->SetZP1EqualCoeff(k, equalCoeff[j][k]);
	 	else if(j==8) calibdata->SetZN2EqualCoeff(k, equalCoeff[j][k]);
	 	else if(j==9) calibdata->SetZP2EqualCoeff(k, equalCoeff[j][k]);	 
             }
	   }
         }
       }
       else{
         Log(Form("File %s not found", EMDFileName));
         return 1;
       }
       //calibdata->Print("");
  }
} 
else {
  Log(Form("Nothing to do: run type is %s", runType.Data()));
  return 0;
} 

  // note that the parameters are returned as character strings!
  const char* nEvents = GetRunParameter("totalEvents");
  if(nEvents) Log(Form("Number of events for run %d: %s",fRun, nEvents));
  else Log(Form("Number of events not put in logbook!"));

  // Storing the final CDB file
  AliCDBMetaData metaData;
  metaData.SetBeamPeriod(0);
  metaData.SetResponsible("Chiara");
  metaData.SetComment("Filling AliZDCCalibData object");

  Bool_t resultCal = kFALSE;
  resultCal = Store("Calib","Data",calibdata, &metaData, 0, 1);
 
  UInt_t result = 0;
  if(resultAl == kFALSE || resultCal == kFALSE){
    if(resultAl == kFALSE  && resultCal == kFALSE ) result = 3;
    else result = 2;
  }
  
  return result;
  
}
Commit	Line	Data
31af5828	1	// --- ROOT system
31af5828	2	#include <TFile.h>
79563ba1	3	#include <TClonesArray.h>
	4	#include <TList.h>
	5	#include <TObjString.h>
31af5828	6	#include <TTimeStamp.h>
64a7c78d	7
31af5828	8	#include "AliZDCPreprocessor.h"
	9	#include "AliCDBManager.h"
	10	#include "AliCDBEntry.h"
64a7c78d	11	#include "AliCDBMetaData.h"
64a7c78d	12	#include "AliDCSValue.h"
79563ba1	13	#include "AliAlignObj.h"
90dbf5fb	14	#include "AliAlignObjParams.h"
64a7c78d	15	#include "AliLog.h"
64a7c78d	16	#include "AliZDCDataDCS.h"
31af5828	17	#include "AliZDCCalibData.h"
64a7c78d	18
64a7c78d	19	//
7a78280f	20	// Class implementing ZDC pre-processor.
	21	// It takes data from DCS and passes it to the class AliZDCDataDCS.
	22	// The class is then written to the CDB.
64a7c78d	23	//
	24
	25	ClassImp(AliZDCPreprocessor)
	26
	27	//______________________________________________________________________________________________
857ece97	28	AliZDCPreprocessor::AliZDCPreprocessor(AliShuttleInterface* shuttle) :
857ece97	29	AliPreprocessor("ZDC", shuttle),
64a7c78d	30	fData(0)
	31	{
	32	// constructor
	33	}
	34
	35	//______________________________________________________________________________________________
	36	AliZDCPreprocessor::~AliZDCPreprocessor()
	37	{
	38	// destructor
	39	}
	40
7a78280f	41
64a7c78d	42	//______________________________________________________________________________________________
	43	void AliZDCPreprocessor::Initialize(Int_t run, UInt_t startTime,
	44	UInt_t endTime)
	45	{
	46	// Creates AliZDCDataDCS object
	47
	48	AliPreprocessor::Initialize(run, startTime, endTime);
	49
79563ba1	50	Log(Form("\n\tRun %d \n\tStartTime %s \n\tEndTime %s", run,
64a7c78d	51	TTimeStamp(startTime).AsString(),
	52	TTimeStamp(endTime).AsString()));
	53
79563ba1	54	fRun = run;
	55	fStartTime = startTime;
	56	fEndTime = endTime;
	57
64a7c78d	58	fData = new AliZDCDataDCS(fRun, fStartTime, fEndTime);
	59	}
	60
	61	//______________________________________________________________________________________________
	62	UInt_t AliZDCPreprocessor::Process(TMap* dcsAliasMap)
	63	{
79563ba1	64	// ************* From DCS ****************
64a7c78d	65	// Fills data into a AliZDCDataDCS object
f181c9b8	66	if(!dcsAliasMap) return 1;
64a7c78d	67
64a7c78d	68	// The processing of the DCS input data is forwarded to AliZDCDataDCS
c0aec6f6	69	Float_t DCSValues[28]; // DCSAliases=28
31af5828	70	fData->ProcessData(*dcsAliasMap, DCSValues);
e1ea0c63	71	//dcsAliasMap->Print("");
31af5828	72	//
79563ba1	73	// --- Writing ZDC table positions into alignment object
90dbf5fb	74	TClonesArray *array = new TClonesArray("AliAlignObjParams",10);
79563ba1	75	TClonesArray &alobj = *array;
90dbf5fb	76	AliAlignObjParams a;
79563ba1	77	Double_t dx=0., dz=0., dpsi=0., dtheta=0., dphi=0.;
cb50c7c8	78	// Vertical table position in mm from DCS
	79	Double_t dyZN1 = (Double_t) (DCSValues[0]/10.);
	80	Double_t dyZP1 = (Double_t) (DCSValues[1]/10.);
	81	Double_t dyZN2 = (Double_t) (DCSValues[2]/10.);
	82	Double_t dyZP2 = (Double_t) (DCSValues[3]/10.);
79563ba1	83	const char *ZDCn1="ZDC/NeutronZDC1";
	84	const char *ZDCp1="ZDC/ProtonZDC1";
	85	const char *ZDCn2="ZDC/NeutronZDC2";
	86	const char *ZDCp2="ZDC/ProtonZDC2";
	87	UShort_t iIndex=0;
ae079791	88	AliGeomManager::ELayerID iLayer = AliGeomManager::kInvalidLayer;
ae079791	89	UShort_t volid = AliGeomManager::LayerToVolUID(iLayer,iIndex);
79563ba1	90	//
90dbf5fb	91	new(alobj[0]) AliAlignObjParams(ZDCn1, volid, dx, dyZN1, dz, dpsi, dtheta, dphi, kTRUE);
	92	new(alobj[1]) AliAlignObjParams(ZDCp1, volid, dx, dyZP1, dz, dpsi, dtheta, dphi, kTRUE);
	93	new(alobj[2]) AliAlignObjParams(ZDCn2, volid, dx, dyZN2, dz, dpsi, dtheta, dphi, kTRUE);
	94	new(alobj[3]) AliAlignObjParams(ZDCp2, volid, dx, dyZP2, dz, dpsi, dtheta, dphi, kTRUE);
79563ba1	95	// save in CDB storage
	96	AliCDBMetaData md;
	97	md.SetResponsible("Chiara Oppedisano");
	98	md.SetComment("Alignment object for ZDC");
f181c9b8	99	Bool_t resultAl = kFALSE;
cb50c7c8	100	resultAl = Store("Align","Data", array, &md, 0, 0);
79563ba1	101
e1ea0c63	102	AliZDCCalibData *calibdata = new AliZDCCalibData("ZDC");
79563ba1	103
028eb4a1	104	// ************* From DAQ ****************
	105	// *****************************************************
	106	// [a] PEDESTALS -> Pedestal subtraction
	107	// *****************************************************
06db4c54	108	TString runType = GetRunType();
028eb4a1	109	printf("\n\t AliZDCPreprocessor -> runType detected %s\n\n",runType.Data());
028eb4a1	110	if(runType == "PEDESTAL_RUN") {
cb50c7c8	111	TList* daqSources = GetFileSources(kDAQ, "PEDESTALS");
	112	if(!daqSources){
	113	Log(Form("No source for PEDESTALS run %d !", fRun));
c58c7e82	114	return 1;
79563ba1	115	}
79563ba1	116	Log("\t List of sources for PEDESTALS");
cb50c7c8	117	daqSources->Print();
79563ba1	118	//
cb50c7c8	119	TIter iter(daqSources);
cb50c7c8	120	TObjString* source = 0;
79563ba1	121	Int_t i=0;
cb50c7c8	122	while((source = dynamic_cast<TObjString*> (iter.Next()))){
79563ba1	123	Log(Form("\n\t Getting file #%d\n",++i));
cb50c7c8	124	TString stringPedFileName = GetFile(kDAQ, "PEDESTALS", source->GetName());
	125	if(stringPedFileName.Length() <= 0){
	126	Log(Form("No PEDESTAL file from source %s!", source->GetName()));
c58c7e82	127	return 1;
cb50c7c8	128	}
cb50c7c8	129	const char* PedFileName = stringPedFileName.Data();
c0aec6f6	130	// no. ADCch = (22 signal ch. + 2 reference PMs) * 2 gain chain = 48
c0aec6f6	131	const Int_t NZDCch = 48;
79563ba1	132	if(PedFileName){
	133	FILE *file;
	134	if((file = fopen(PedFileName,"r")) == NULL){
	135	printf("Cannot open file %s \n",PedFileName);
c58c7e82	136	return 1;
79563ba1	137	}
	138	Log(Form("File %s connected to analyze pedestal events", PedFileName));
	139	Float_t PedVal[(3*NZDCch)][2];
	140	for(Int_t i=0; i<(3*NZDCch); i++){
	141	for(Int_t j=0; j<2; j++){
	142	fscanf(file,"%f",&PedVal[i][j]);
	143	//if(j==1) printf("PedVal[%d] -> %f, %f \n",i,PedVal[i][0],PedVal[i][1]);
	144	}
	145	if(i<NZDCch){
	146	calibdata->SetMeanPed(i,PedVal[i][0]);
	147	calibdata->SetMeanPedWidth(i,PedVal[i][1]);
	148	}
	149	else if(i>=NZDCch && i<(2*NZDCch)){
	150	calibdata->SetOOTPed(i-NZDCch,PedVal[i][0]);
	151	calibdata->SetOOTPedWidth(i-NZDCch,PedVal[i][1]);
	152	}
	153	else if(i>=(2NZDCch) && i<(3NZDCch)){
	154	calibdata->SetPedCorrCoeff(i-(2*NZDCch),PedVal[i][0],PedVal[i][1]);
	155	}
	156	}
e1ea0c63	157	}
79563ba1	158	else{
79563ba1	159	Log(Form("File %s not found", PedFileName));
c58c7e82	160	return 1;
e1ea0c63	161	}
79563ba1	162	//
79563ba1	163	//calibdata->Print("");
64a7c78d	164	}
cb50c7c8	165	delete daqSources; daqSources = 0;
04bd7caa	166	}
028eb4a1	167	// *****************************************************
	168	// [b] EMD EVENTS -> Energy calibration and equalization
	169	// *****************************************************
	170	else if(runType == "PULSER_RUN"){
	171	TList* daqSources = GetFileSources(kDAQ, "EMDCALIB");
cb50c7c8	172	if(!daqSources){
028eb4a1	173	AliError(Form("No sources for PULSER_RUN run %d !", fRun));
c58c7e82	174	return 1;
64a7c78d	175	}
028eb4a1	176	Log("\t List of sources for PULSER_RUN");
cb50c7c8	177	daqSources->Print();
e1ea0c63	178	//
cb50c7c8	179	TIter iter2(daqSources);
06db4c54	180	TObjString* source = 0;
79563ba1	181	Int_t j=0;
cb50c7c8	182	while((source = dynamic_cast<TObjString*> (iter2.Next()))){
79563ba1	183	Log(Form("\n\t Getting file #%d\n",++j));
028eb4a1	184	TString stringEMDFileName = GetFile(kDAQ, "EMDCALIB", source->GetName());
cb50c7c8	185	if(stringEMDFileName.Length() <= 0){
028eb4a1	186	Log(Form("No EMDCALIB file from source %s!", source->GetName()));
c58c7e82	187	return 1;
cb50c7c8	188	}
cb50c7c8	189	const char* EMDFileName = stringEMDFileName.Data();
79563ba1	190	if(EMDFileName){
	191	FILE *file;
	192	if((file = fopen(EMDFileName,"r")) == NULL){
	193	printf("Cannot open file %s \n",EMDFileName);
c58c7e82	194	return 1;
79563ba1	195	}
79563ba1	196	Log(Form("File %s connected to analyze EM dissociation events", EMDFileName));
028eb4a1	197	Float_t fitValEMD[6]; Float_t equalCoeff[5][4];
	198	Float_t CalibVal[4];
	199	for(Int_t j=0; j<10; j++){
	200	if(j<6){
	201	fscanf(file,"%f",&fitValEMD[j]);
	202	if(j<4){
	203	CalibVal[j] = fitValEMD[j]/2.76;
	204	calibdata->SetEnCalib(j,CalibVal[j]);
	205	}
	206	else calibdata->SetEnCalib(j,fitValEMD[j]);
	207	}
	208	else{
	209	for(Int_t k=0; k<5; k++){
	210	fscanf(file,"%f",&equalCoeff[j][k]);
	211	if(j==6) calibdata->SetZN1EqualCoeff(k, equalCoeff[j][k]);
	212	else if(j==7) calibdata->SetZP1EqualCoeff(k, equalCoeff[j][k]);
	213	else if(j==8) calibdata->SetZN2EqualCoeff(k, equalCoeff[j][k]);
	214	else if(j==9) calibdata->SetZP2EqualCoeff(k, equalCoeff[j][k]);
	215	}
	216	}
79563ba1	217	}
79563ba1	218	}
	219	else{
	220	Log(Form("File %s not found", EMDFileName));
c58c7e82	221	return 1;
79563ba1	222	}
	223	//calibdata->Print("");
	224	}
c58c7e82	225	}
c58c7e82	226	else {
bff168f8	227	Log(Form("Nothing to do: run type is %s", runType.Data()));
bff168f8	228	return 0;
c58c7e82	229	}
028eb4a1	230
e1ea0c63	231	// note that the parameters are returned as character strings!
e1ea0c63	232	const char* nEvents = GetRunParameter("totalEvents");
79563ba1	233	if(nEvents) Log(Form("Number of events for run %d: %s",fRun, nEvents));
79563ba1	234	else Log(Form("Number of events not put in logbook!"));
e1ea0c63	235
79563ba1	236	// Storing the final CDB file
64a7c78d	237	AliCDBMetaData metaData;
31af5828	238	metaData.SetBeamPeriod(0);
31af5828	239	metaData.SetResponsible("Chiara");
79563ba1	240	metaData.SetComment("Filling AliZDCCalibData object");
64a7c78d	241
f181c9b8	242	Bool_t resultCal = kFALSE;
b2324386	243	resultCal = Store("Calib","Data",calibdata, &metaData, 0, 1);
79563ba1	244
cb50c7c8	245	UInt_t result = 0;
04bd7caa	246	if(resultAl == kFALSE \|\| resultCal == kFALSE){
c58c7e82	247	if(resultAl == kFALSE && resultCal == kFALSE ) result = 3;
c58c7e82	248	else result = 2;
cb50c7c8	249	}
cb50c7c8	250
64a7c78d	251	return result;
cb50c7c8	252
64a7c78d	253	}