[u/mrichter/AliRoot.git] / ITS / UpdateSSDGainOCDB.C

#if !defined(__CINT__) || defined(__MAKECINT__)
#include <Riostream.h>
#include <TFile.h>
#include <TString.H>
#include "AliITSGainSSDv2.h"
#include "AliCDBEntry.h"
#include "AliCDBManager.h"
#include "AliCDBMetaData.h"
#endif
Int_t GainCalibration(AliCDBManager * man, Float_t* gainP, Float_t* gainN) 
{
	const Int_t fgkSSDMODULES = 1698;
  	static const Int_t fgkDefaultNStripsSSD = 768;
  	AliITSGainSSDv2 *gainSSD = new AliITSGainSSDv2();
  	AliCDBEntry *entryGainSSD = man->Get("ITS/Calib/GainSSD");
  	TObject *empty = (TObject *)entryGainSSD->GetObject();
  	TString objectname = empty->GetName();
  	if(objectname=="AliITSGainSSDv2") 
	{
    		cout<<"Reading the new format of the calibration file..."<<endl;
    		gainSSD = (AliITSGainSSDv2 *)entryGainSSD->GetObject();
  	}
	else
	{
		cout<<"Now OCDB object"<<endl;
		return 0 ;
	}
  	for (Int_t i = 0; i < fgkSSDMODULES; i++) 
	{
  			//for (Int_t i = 0; i < 1; i++) {
    			//AliITSgeomTGeo::GetModuleId(i+500,layer,ladder,module);
   		for(Int_t j = 0; j < fgkDefaultNStripsSSD; j++) 
	 	{
     	 		gainP[i*fgkDefaultNStripsSSD+j] = gainSSD->GetGainP(i,j);
      			gainN[i*fgkDefaultNStripsSSD+j]= gainSSD->GetGainN(i,j);
     	
    		}//strip loop
  	}//module loop
	return 1;
}


void UpdateSSDGainOCDB(TString gainfile,Float_t globalMPV=84.0,const char* type = "alien", Int_t runNumber = 0, Int_t runRangemin=0 ,Int_t runRangemax= AliCDBRunRange::Infinity()) 
{
	if((runRangemax<runRangemin))
	{
		cout<<"wrong run range "<<endl;
		return;
	}	
  	const Int_t fgkPNStripsPerModule = 768;
  	const Int_t fgkNumberOfSSDModules = 1698;
	Float_t gainPold[fgkPNStripsPerModule* fgkNumberOfSSDModules];
	Float_t gainNold[fgkPNStripsPerModule* fgkNumberOfSSDModules];
	
	for(int i=0;i<fgkPNStripsPerModule* fgkNumberOfSSDModules;i++)
	{
			gainPold[i]=0.0;
			gainNold[i]=0.0;
	}
	
	TString gType = type;
  	AliCDBManager * man = AliCDBManager::Instance();
	if(gType == "alien") 
   		 man->SetDefaultStorage("alien://folder=/alice/data/2010/OCDB/");
  	else if(gType == "local") 
    		man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
  	else 
	{
    		cout<<"Allowed types: local or alien!!!"<<endl;
    		return;
  	}
	  man->SetRun(runNumber);
	if(!GainCalibration(man,gainPold,gainNold))
		return;
		
	char buffer[256];
	ifstream ingainfile(gainfile);
	if(!ingainfile.good())
		return;
	
	
	 ingainfile.getline(buffer,256);
	cout<<buffer<<endl;
	
	
	Int_t module=0;
	Int_t gainflag=0;
	Float_t gainP=0;
	Float_t gainN=0;
	Float_t fMPV=0;
	Float_t corrP[1698];
	Float_t corrN[1698];
	
	for(int jj=0;jj<1698;jj++)
    	{
		ingainfile>>module;
		ingainfile>>gainflag;
		ingainfile>>gainP;
		ingainfile>>gainN;
		ingainfile>>fMPV;
		if(gainflag==1)
		{
			corrP[module]=gainP*globalMPV/fMPV;
			corrN[module]=gainN*globalMPV/fMPV;
			
		}
		else
		{
			corrP[module]=1.0;
			corrN[module]=1.0;
		}
			
	}
	
	
  	TObjArray *array = new TObjArray();
  	array->SetName("Gain");
  	AliITSGainSSDv2  *mc;
	mc = new AliITSGainSSDv2();
  	for(Int_t mind = 0; mind < fgkNumberOfSSDModules; mind++) 
	{
    		for (Int_t i = 0; i < fgkPNStripsPerModule; i++) 
		{
			mc->AddGainP(mind,i, gainPold[mind* fgkPNStripsPerModule+i]*corrP[mind]);
     			mc->AddGainN(mind,i,gainNold[mind* fgkPNStripsPerModule+i]*corrN[mind]);			
    		}
  	}

	man = AliCDBManager::Instance();
	man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
	man->SetRun(0);
	//AliCDBId id("ITS/Calib/GainSSD",0,AliCDBRunRange::Infinity());
	AliCDBId id("ITS/Calib/GainSSD",runRangemin,runRangemax);
	AliCDBMetaData *metadata= new AliCDBMetaData();
	metadata->SetResponsible("Marek.Chojnacki@cern.ch");
	metadata->SetComment("Default values for the SSD gain calibration");
	//man->Put(array,id,metadata);
	man->Put(mc,id,metadata);
	 TFile* fout=TFile::Open("AliITSGainSSD_new.root","recreate");
	mc ->Write();	
	 	fout->Close();
	
}
Commit	Line	Data
2d9ed4be	1	#if !defined(__CINT__) \|\| defined(__MAKECINT__)
	2	#include <Riostream.h>
	3	#include <TFile.h>
	4	#include <TString.H>
	5	#include "AliITSGainSSDv2.h"
	6	#include "AliCDBEntry.h"
	7	#include "AliCDBManager.h"
	8	#include "AliCDBMetaData.h"
	9	#endif
	10	Int_t GainCalibration(AliCDBManager * man, Float_t* gainP, Float_t* gainN)
	11	{
	12	const Int_t fgkSSDMODULES = 1698;
	13	static const Int_t fgkDefaultNStripsSSD = 768;
	14	AliITSGainSSDv2 *gainSSD = new AliITSGainSSDv2();
	15	AliCDBEntry *entryGainSSD = man->Get("ITS/Calib/GainSSD");
	16	TObject empty = (TObject )entryGainSSD->GetObject();
	17	TString objectname = empty->GetName();
	18	if(objectname=="AliITSGainSSDv2")
	19	{
	20	cout<<"Reading the new format of the calibration file..."<<endl;
	21	gainSSD = (AliITSGainSSDv2 *)entryGainSSD->GetObject();
	22	}
	23	else
	24	{
	25	cout<<"Now OCDB object"<<endl;
	26	return 0 ;
	27	}
	28	for (Int_t i = 0; i < fgkSSDMODULES; i++)
	29	{
	30	//for (Int_t i = 0; i < 1; i++) {
	31	//AliITSgeomTGeo::GetModuleId(i+500,layer,ladder,module);
	32	for(Int_t j = 0; j < fgkDefaultNStripsSSD; j++)
	33	{
	34	gainP[i*fgkDefaultNStripsSSD+j] = gainSSD->GetGainP(i,j);
	35	gainN[i*fgkDefaultNStripsSSD+j]= gainSSD->GetGainN(i,j);
	36
	37	}//strip loop
	38	}//module loop
	39	return 1;
	40	}
	41
	42
	43	void UpdateSSDGainOCDB(TString gainfile,Float_t globalMPV=84.0,const char* type = "alien", Int_t runNumber = 0, Int_t runRangemin=0 ,Int_t runRangemax= AliCDBRunRange::Infinity())
	44	{
	45	if((runRangemax<runRangemin))
	46	{
	47	cout<<"wrong run range "<<endl;
	48	return;
	49	}
	50	const Int_t fgkPNStripsPerModule = 768;
	51	const Int_t fgkNumberOfSSDModules = 1698;
	52	Float_t gainPold[fgkPNStripsPerModule* fgkNumberOfSSDModules];
	53	Float_t gainNold[fgkPNStripsPerModule* fgkNumberOfSSDModules];
	54
	55	for(int i=0;i<fgkPNStripsPerModule* fgkNumberOfSSDModules;i++)
	56	{
	57	gainPold[i]=0.0;
	58	gainNold[i]=0.0;
	59	}
	60
	61	TString gType = type;
	62	AliCDBManager * man = AliCDBManager::Instance();
	63	if(gType == "alien")
	64	man->SetDefaultStorage("alien://folder=/alice/data/2010/OCDB/");
65	else if(gType == "local")
66	man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
67	else
68	{
69	cout<<"Allowed types: local or alien!!!"<<endl;
70	return;
71	}
72	man->SetRun(runNumber);
73	if(!GainCalibration(man,gainPold,gainNold))
74	return;
75
76	char buffer[256];
77	ifstream ingainfile(gainfile);
78	if(!ingainfile.good())
79	return;
80
81
82	ingainfile.getline(buffer,256);
83	cout<<buffer<<endl;
84
85
86	Int_t module=0;
87	Int_t gainflag=0;
88	Float_t gainP=0;
89	Float_t gainN=0;
90	Float_t fMPV=0;
91	Float_t corrP[1698];
92	Float_t corrN[1698];
93
94	for(int jj=0;jj<1698;jj++)
95	{
96	ingainfile>>module;
97	ingainfile>>gainflag;
98	ingainfile>>gainP;
99	ingainfile>>gainN;
100	ingainfile>>fMPV;
101	if(gainflag==1)
102	{
103	corrP[module]=gainP*globalMPV/fMPV;
104	corrN[module]=gainN*globalMPV/fMPV;
105
106	}
107	else
108	{
109	corrP[module]=1.0;
110	corrN[module]=1.0;
111	}
112
113	}
114
115
116	TObjArray *array = new TObjArray();
117	array->SetName("Gain");
118	AliITSGainSSDv2 *mc;
119	mc = new AliITSGainSSDv2();
120	for(Int_t mind = 0; mind < fgkNumberOfSSDModules; mind++)
121	{
122	for (Int_t i = 0; i < fgkPNStripsPerModule; i++)
123	{
124	mc->AddGainP(mind,i, gainPold[mind* fgkPNStripsPerModule+i]*corrP[mind]);
125	mc->AddGainN(mind,i,gainNold[mind* fgkPNStripsPerModule+i]*corrN[mind]);
126	}
127	}
128
129	man = AliCDBManager::Instance();
130	man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
131	man->SetRun(0);
132	//AliCDBId id("ITS/Calib/GainSSD",0,AliCDBRunRange::Infinity());
133	AliCDBId id("ITS/Calib/GainSSD",runRangemin,runRangemax);
134	AliCDBMetaData *metadata= new AliCDBMetaData();
135	metadata->SetResponsible("Marek.Chojnacki@cern.ch");
136	metadata->SetComment("Default values for the SSD gain calibration");
137	//man->Put(array,id,metadata);
138	man->Put(mc,id,metadata);
139	TFile* fout=TFile::Open("AliITSGainSSD_new.root","recreate");
140	mc ->Write();
141	fout->Close();
142
143	}