PMD/AliPMDPreprocessor.cxx

   1 // --- ROOT system
   2 #include <TFile.h>
   3 #include <TTimeStamp.h>
   4
   5 #include "AliPMDPreprocessor.h"
   6 #include "AliPMDPedestal.h"
   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)
  18
  19 //______________________________________________________________________________________________
  20 AliPMDPreprocessor::AliPMDPreprocessor(AliShuttleInterface* shuttle) :
  21   AliPreprocessor("PMD", shuttle)
  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
  38     AliPreprocessor::Initialize(run, startTime, endTime);
  39
  40     AliInfo(Form("\n\tRun %d \n\tStartTime %s \n\tEndTime %s", run,
  41                  TTimeStamp(startTime).AsString(),
  42                  TTimeStamp(endTime).AsString()));
  43
  44     fRun = run;
  45     fStartTime = startTime;
  46     fEndTime = endTime;
  47
  48 }
  49
  50 //______________________________________________________________________________________________
  51 UInt_t AliPMDPreprocessor::Process(TMap* pdaqAliasMap)
  52 {
  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"){
 129
 130         AliPMDCalibData *calibda = new AliPMDCalibData();
 131
 132         TList* filesources = GetFileSources(kDAQ, "PMDGAINS");
 133
 134         if(!filesources) {
 135             Log(Form("No sources found for PMDGAINS!"));
 136             return 1;
 137         }
 138
 139         AliInfo("Here's the list of sources for PMDGAINS");
 140         filesources->Print();
 141
 142         TIter iter(filesources);
 143         TObjString* source;
 144         UInt_t result = 0;
 145         TString filename;
 146         while((source=dynamic_cast<TObjString*> (iter.Next()))){
 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;
 207 }
 208