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

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

#include "AliPMDPreprocessor.h"
#include "AliPMDPedestal.h"
#include "AliPMDCalibData.h"
#include "AliLog.h"
#include "AliShuttleInterface.h"
#include "AliCDBMetaData.h"
#include <TTimeStamp.h>
#include <TObjString.h>
#include <TTree.h>
#include <TSystem.h>


ClassImp(AliPMDPreprocessor)
  
//______________________________________________________________________________________________
AliPMDPreprocessor::AliPMDPreprocessor(AliShuttleInterface* shuttle) :
  AliPreprocessor("PMD", shuttle)
{
  // constructor
}

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

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

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

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

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

}

//______________________________________________________________________________________________
UInt_t AliPMDPreprocessor::Process(TMap* pdaqAliasMap)
{
    
    if(!pdaqAliasMap) return 1;
    TString runType = GetRunType();
    if(runType == "PEDESTAL_RUN"){
	AliPMDPedestal *pedestal = new AliPMDPedestal();
	
        TList* filesources = GetFileSources(kDAQ, "PMD_PED");
	
        if(!filesources) {
	    Log(Form("No sources found for PMD_PED!"));
	    return 1;
	}
	
        AliInfo("Here's the list of sources for PMD_PED");
        filesources->Print();
	
        TIter iter(filesources);
        TObjString* source;
        UInt_t result = 0;
	TString filename;
        while((source=dynamic_cast<TObjString*> (iter.Next()))){
	    filename = GetFile(kDAQ, "PMD_PED", source->GetName());
	    if(filename.Length() == 0) {
		Log(Form("Error retrieving file from source %s failed!", source->GetName()));
		delete filesources;
		return 1;
	    }
	    
	    Log(Form("File with id PMD_PED got from %s", source->GetName()));
	    Int_t DET,SM,ROW,COL;
	    Float_t MEAN,RMS;
	    TFile *f= new TFile(filename.Data());
	    if(!f || !f->IsOpen()) 
	    {
		Log(Form("Error opening file with Id PMD_PED from source %s!", source->GetName()));
		return 1;
	    } 
	    TTree *tree = dynamic_cast<TTree *> (f->Get("ped"));
	    if (!tree) 
	    {
		Log("Could not find object \"ped\" in PED file!");
		return 1;
	    }
	    
	    tree->SetBranchAddress("DET",  &DET);
	    tree->SetBranchAddress("SM",   &SM);
	    tree->SetBranchAddress("ROW",  &ROW);
	    tree->SetBranchAddress("COL",  &COL);
	    tree->SetBranchAddress("MEAN", &MEAN);
	    tree->SetBranchAddress("RMS",  &RMS);
	    Int_t nEntries = (Int_t) tree->GetEntries();
	    for(Int_t i = 0; i < nEntries; i++)
	    {
		tree->GetEntry(i);
		pedestal->SetPedMeanRms(DET,SM,ROW,COL,MEAN,RMS);
	    }
	    f->Close();
	    delete f;
	}
	AliCDBMetaData metaData;
	metaData.SetBeamPeriod(0);
	metaData.SetComment("test PMD preprocessor");
	
	result = Store("Calib","Ped", pedestal, &metaData);
	delete pedestal;
	if(result==0)
	{
	    Log("Error storing");                        
	    return 1;
	}
	else
	{
	    return 0;
	}
	
    }else if (runType == "PHYSICS"){
	
	AliPMDCalibData *calibda = new AliPMDCalibData();
	
        TList* filesources = GetFileSources(kDAQ, "PMDGAINS");
	
        if(!filesources) {
	    Log(Form("No sources found for PMDGAINS!"));
	    return 1;
	}
	
        AliInfo("Here's the list of sources for PMDGAINS");
        filesources->Print();
	
        TIter iter(filesources);
        TObjString* source;
        UInt_t result = 0;
	TString filename;
        while((source=dynamic_cast<TObjString*> (iter.Next()))){
	    filename = GetFile(kDAQ, "PMDGAINS", source->GetName());
	    if(filename.Length() == 0) {
		Log(Form("Error retrieving file from source %s failed!", source->GetName()));
		delete filesources;
		return 1;
	    }
	    
	    Log(Form("File with id PMDGAINS got from %s", source->GetName()));
	    Int_t DET,SM,ROW,COL;
	    Float_t GAIN;
	    TFile *f1= new TFile(filename.Data());
	    if(!f1 || !f1->IsOpen()) 
	    {
		Log(Form("Error opening file with Id PMDGAINS from source %s!", source->GetName()));
		return 1;
	    } 
	    TTree *tree = dynamic_cast<TTree *> (f1->Get("ic"));
	    if (!tree) 
	    {
		Log("Could not find object \"ic\" in DAQ file!");
		return 1;
	    }
	    
	    tree->SetBranchAddress("DET",  &DET);
	    tree->SetBranchAddress("SM",   &SM);
	    tree->SetBranchAddress("ROW",  &ROW);
	    tree->SetBranchAddress("COL",  &COL);
	    tree->SetBranchAddress("GAIN", &GAIN);
	    Int_t nEntries = (Int_t) tree->GetEntries();
	    for(Int_t i = 0; i < nEntries; i++)
	    {
		tree->GetEntry(i);
//	  if(DET>1 || SM>23 || ROW>95 || COL>95) {
// 	printf("Error! gain[%d,%d,%d,%d] = %f\n",DET,SM,ROW,COL,GAIN);
//	continue;
		//    		 	}
		calibda->SetGainFact(DET,SM,ROW,COL,GAIN);
	    }
	    f1->Close();
	    delete f1;
	}

	AliCDBMetaData metaData;
	metaData.SetBeamPeriod(0);
	metaData.SetComment("test PMD preprocessor");
	result = Store("Calib","Gain", calibda, &metaData);
	delete calibda;
	if(result==0)
	{
	    Log("Error storing");                        
	    return 1;
	}
	else
	{
	    return 0;
	}
	
    }
    
    return 2;
}
Commit	Line	Data
ad6005fa	1	// --- ROOT system
	2	#include <TFile.h>
	3	#include <TTimeStamp.h>
	4
	5	#include "AliPMDPreprocessor.h"
cda4fe9e	6	#include "AliPMDPedestal.h"
ad6005fa	7	#include "AliPMDCalibData.h"
	8	#include "AliLog.h"
	9	#include "AliShuttleInterface.h"
	10	#include "AliCDBMetaData.h"
	11	#include <TTimeStamp.h>
	12	#include <TObjString.h>
	13	#include <TTree.h>
	14	#include <TSystem.h>
	15
	16
	17	ClassImp(AliPMDPreprocessor)
cda4fe9e	18
ad6005fa	19	//______________________________________________________________________________________________
a80b0ff4	20	AliPMDPreprocessor::AliPMDPreprocessor(AliShuttleInterface* shuttle) :
a80b0ff4	21	AliPreprocessor("PMD", shuttle)
ad6005fa	22	{
	23	// constructor
	24	}
	25
	26	//______________________________________________________________________________________________
	27	AliPMDPreprocessor::~AliPMDPreprocessor()
	28	{
	29	// destructor
	30	}
	31
	32	//______________________________________________________________________________________________
	33	void AliPMDPreprocessor::Initialize(Int_t run, UInt_t startTime,
	34	UInt_t endTime)
	35	{
	36	// Creates AliPMDDataDAQ object
	37
cda4fe9e	38	AliPreprocessor::Initialize(run, startTime, endTime);
ad6005fa	39
cda4fe9e	40	AliInfo(Form("\n\tRun %d \n\tStartTime %s \n\tEndTime %s", run,
	41	TTimeStamp(startTime).AsString(),
	42	TTimeStamp(endTime).AsString()));
ad6005fa	43
cda4fe9e	44	fRun = run;
	45	fStartTime = startTime;
	46	fEndTime = endTime;
ad6005fa	47
	48	}
	49
	50	//______________________________________________________________________________________________
	51	UInt_t AliPMDPreprocessor::Process(TMap* pdaqAliasMap)
	52	{
cda4fe9e	53
	54	if(!pdaqAliasMap) return 1;
	55	TString runType = GetRunType();
	56	if(runType == "PEDESTAL_RUN"){
	57	AliPMDPedestal *pedestal = new AliPMDPedestal();
	58
	59	TList* filesources = GetFileSources(kDAQ, "PMD_PED");
	60
	61	if(!filesources) {
	62	Log(Form("No sources found for PMD_PED!"));
	63	return 1;
	64	}
	65
	66	AliInfo("Here's the list of sources for PMD_PED");
	67	filesources->Print();
	68
	69	TIter iter(filesources);
	70	TObjString* source;
	71	UInt_t result = 0;
	72	TString filename;
	73	while((source=dynamic_cast<TObjString*> (iter.Next()))){
	74	filename = GetFile(kDAQ, "PMD_PED", source->GetName());
	75	if(filename.Length() == 0) {
	76	Log(Form("Error retrieving file from source %s failed!", source->GetName()));
	77	delete filesources;
	78	return 1;
	79	}
	80
	81	Log(Form("File with id PMD_PED got from %s", source->GetName()));
	82	Int_t DET,SM,ROW,COL;
	83	Float_t MEAN,RMS;
	84	TFile *f= new TFile(filename.Data());
	85	if(!f \|\| !f->IsOpen())
	86	{
	87	Log(Form("Error opening file with Id PMD_PED from source %s!", source->GetName()));
	88	return 1;
	89	}
	90	TTree tree = dynamic_cast<TTree > (f->Get("ped"));
	91	if (!tree)
	92	{
	93	Log("Could not find object \"ped\" in PED file!");
	94	return 1;
	95	}
	96
	97	tree->SetBranchAddress("DET", &DET);
	98	tree->SetBranchAddress("SM", &SM);
	99	tree->SetBranchAddress("ROW", &ROW);
	100	tree->SetBranchAddress("COL", &COL);
	101	tree->SetBranchAddress("MEAN", &MEAN);
	102	tree->SetBranchAddress("RMS", &RMS);
	103	Int_t nEntries = (Int_t) tree->GetEntries();
	104	for(Int_t i = 0; i < nEntries; i++)
	105	{
	106	tree->GetEntry(i);
	107	pedestal->SetPedMeanRms(DET,SM,ROW,COL,MEAN,RMS);
	108	}
	109	f->Close();
	110	delete f;
	111	}
	112	AliCDBMetaData metaData;
	113	metaData.SetBeamPeriod(0);
	114	metaData.SetComment("test PMD preprocessor");
	115
	116	result = Store("Calib","Ped", pedestal, &metaData);
117	delete pedestal;
118	if(result==0)
119	{
120	Log("Error storing");
121	return 1;
122	}
123	else
124	{
125	return 0;
126	}
127
128	}else if (runType == "PHYSICS"){
ad6005fa	129
ad6005fa	130	AliPMDCalibData *calibda = new AliPMDCalibData();
cda4fe9e	131
ad6005fa	132	TList* filesources = GetFileSources(kDAQ, "PMDGAINS");
cda4fe9e	133
ad6005fa	134	if(!filesources) {
cda4fe9e	135	Log(Form("No sources found for PMDGAINS!"));
	136	return 1;
	137	}
	138
ad6005fa	139	AliInfo("Here's the list of sources for PMDGAINS");
ad6005fa	140	filesources->Print();
cda4fe9e	141
ad6005fa	142	TIter iter(filesources);
ad6005fa	143	TObjString* source;
ad6005fa	144	UInt_t result = 0;
	145	TString filename;
	146	while((source=dynamic_cast<TObjString*> (iter.Next()))){
cda4fe9e	147	filename = GetFile(kDAQ, "PMDGAINS", source->GetName());
	148	if(filename.Length() == 0) {
	149	Log(Form("Error retrieving file from source %s failed!", source->GetName()));
	150	delete filesources;
	151	return 1;
	152	}
	153
	154	Log(Form("File with id PMDGAINS got from %s", source->GetName()));
	155	Int_t DET,SM,ROW,COL;
	156	Float_t GAIN;
	157	TFile *f1= new TFile(filename.Data());
	158	if(!f1 \|\| !f1->IsOpen())
	159	{
	160	Log(Form("Error opening file with Id PMDGAINS from source %s!", source->GetName()));
	161	return 1;
	162	}
	163	TTree tree = dynamic_cast<TTree > (f1->Get("ic"));
	164	if (!tree)
	165	{
	166	Log("Could not find object \"ic\" in DAQ file!");
	167	return 1;
	168	}
	169
	170	tree->SetBranchAddress("DET", &DET);
	171	tree->SetBranchAddress("SM", &SM);
	172	tree->SetBranchAddress("ROW", &ROW);
	173	tree->SetBranchAddress("COL", &COL);
	174	tree->SetBranchAddress("GAIN", &GAIN);
	175	Int_t nEntries = (Int_t) tree->GetEntries();
	176	for(Int_t i = 0; i < nEntries; i++)
	177	{
	178	tree->GetEntry(i);
	179	// if(DET>1 \|\| SM>23 \|\| ROW>95 \|\| COL>95) {
	180	// printf("Error! gain[%d,%d,%d,%d] = %f\n",DET,SM,ROW,COL,GAIN);
	181	// continue;
	182	// }
	183	calibda->SetGainFact(DET,SM,ROW,COL,GAIN);
	184	}
	185	f1->Close();
	186	delete f1;
	187	}
	188
	189	AliCDBMetaData metaData;
	190	metaData.SetBeamPeriod(0);
	191	metaData.SetComment("test PMD preprocessor");
	192	result = Store("Calib","Gain", calibda, &metaData);
	193	delete calibda;
	194	if(result==0)
	195	{
	196	Log("Error storing");
	197	return 1;
	198	}
	199	else
	200	{
	201	return 0;
	202	}
	203
	204	}
	205
	206	return 2;
ad6005fa	207	}
ad6005fa	208