1 /**************************************************************************
2 * Author: Panos Christakoglou. *
3 * Contributors are mentioned in the code where appropriate. *
5 * Permission to use, copy, modify and distribute this software and its *
6 * documentation strictly for non-commercial purposes is hereby granted *
7 * without fee, provided that the above copyright notice appears in all *
8 * copies and that both the copyright notice and this permission notice *
9 * appear in the supporting documentation. The authors make no claims *
10 * about the suitability of this software for any purpose. It is *
11 * provided "as is" without express or implied warranty. *
12 **************************************************************************/
16 //-----------------------------------------------------------------
17 // AliAODTagCreator class
18 // This is the class to deal with the tag creation (post process)
19 // Origin: Panos Christakoglou, UOA-CERN, Panos.Christakoglou@cern.ch
20 //-----------------------------------------------------------------
23 #include <Riostream.h>
29 #include <TLorentzVector.h>
30 #include <TRefArray.h>
34 #include <TGridResult.h>
37 #include "AliRunTag.h"
38 #include "AliEventTag.h"
40 #include "AliAODEvent.h"
41 #include "AliAODVertex.h"
44 #include "AliAODTagCreator.h"
47 ClassImp(AliAODTagCreator)
50 //______________________________________________________________________________
51 AliAODTagCreator::AliAODTagCreator() :
60 //==============Default constructor for a AliAODTagCreator================
63 //______________________________________________________________________________
64 AliAODTagCreator::~AliAODTagCreator() {
65 //================Default destructor for a AliAODTagCreator===================
70 //______________________________________________________________________________
71 Bool_t AliAODTagCreator::ReadGridCollection(TGridResult *fresult) {
72 // Reads the entry of the TGridResult and creates the tags
73 Int_t nEntries = fresult->GetEntries();
81 fChain = new TChain("aodTree");
83 for(Int_t i = 0; i < nEntries; i++) {
84 alienUrl = fresult->GetKey(i,"turl");
85 guid = fresult->GetKey(i,"guid");
86 if(fresult->GetKey(i,"size")) size = atol (fresult->GetKey(i,"size"));
87 md5 = fresult->GetKey(i,"md5");
88 turl = fresult->GetKey(i,"turl");
89 if(md5 && !strlen(guid)) md5 = 0;
90 if(guid && !strlen(guid)) guid = 0;
92 fChain->Add(alienUrl);
95 AliInfo(Form("AOD chain created......."));
96 AliInfo(Form("Chain entries: %l",fChain->GetEntries()));
98 CreateTag(fChain, "grid");
103 //______________________________________________________________________________
104 Bool_t AliAODTagCreator::ReadLocalCollection(const char *localpath, const char* pattern) {
105 // Checks the different subdirs of the given local path and in the
106 // case where it finds an AliAOD.root file it creates the tags
108 void *dira = gSystem->OpenDirectory(localpath);
110 const char * dirname = 0x0;
111 const char * filename = 0x0;
113 fChain = new TChain("aodTree");
115 while((dirname = gSystem->GetDirEntry(dira))) {
116 sprintf(fPath,"%s/%s",localpath,dirname);
117 void *dirb = gSystem->OpenDirectory(fPath);
118 while((filename = gSystem->GetDirEntry(dirb))) {
119 TString bstr = dirname;
120 if(bstr.Contains("..")) continue;
121 if(strstr(filename,pattern)) {
125 aodFileName += pattern;
126 fChain->Add(aodFileName);
128 } //child directory's entry loop
129 } //parent directory's entry loop
131 AliInfo(Form("AOD chain created......."));
132 AliInfo(Form("Chain entries: %l",fChain->GetEntries()));
134 CreateTag(fChain, "local");
139 //______________________________________________________________________________
140 Bool_t AliAODTagCreator::ReadCAFCollection(const char *filename) {
141 // Temporary solution for CAF: Takes as an input the ascii file that
142 // lists the AODs stored in the SE of the CAF and creates the tags.
144 // Open the input stream
150 fChain = new TChain("aodTree");
152 // Read the input list of files and add them to the chain
155 if (!aodfile.Contains("root")) continue; // protection
156 fChain->Add(aodfile);
159 AliInfo(Form("AOD chain created......."));
160 AliInfo(Form("Chain entries: %l",fChain->GetEntries()));
162 CreateTag(fChain, "proof");
167 //__________________________________________________________________________
168 void AliAODTagCreator::CreateAODTags(Int_t fFirstEvent, Int_t fLastEvent, TList */*grpList*/) {
169 // Creates tag files for AODs
171 AliInfo(Form("Creating the AOD tags......."));
173 TFile *file = TFile::Open("AliAOD.root");
174 if (!file || !file->IsOpen()) {
175 AliError(Form("opening failed"));
180 fChain = new TChain("aodTree");
181 fChain->Add("AliAOD.root");
183 fAODEvent = new AliAODEvent();
184 fAODEvent->ReadFromTree(fChain);
187 if(fLastEvent == -1) lastEvent = (Int_t)fChain->GetEntries();
188 else lastEvent = fLastEvent;
191 sprintf(fileName, "Run%d.Event%d_%d.AOD.tag.root",
192 fAODEvent->GetRunNumber(), fFirstEvent, lastEvent );
193 AliInfo(Form("writing tags to file %s", fileName));
194 AliDebug(1, Form("writing tags to file %s", fileName));
196 TFile* ftag = TFile::Open(fileName, "recreate");
198 fRunTag = new AliRunTag();
199 fTreeT = new TTree("T","A Tree with event tags");
200 TBranch * btag = fTreeT->Branch("AliTAG", &fRunTag);
201 btag->SetCompressionLevel(9);
214 //_____________________________________________________________________________
215 void AliAODTagCreator::CreateTag(TChain* chain, const char *type) {
217 // Private method that creates tag files
220 //reading the esd tag file
221 fTreeTEsd = new TChain("T");
222 const char * tagPattern = "ESD.tag";
223 // Open the working directory
224 void * dirp = gSystem->OpenDirectory(gSystem->pwd());
225 const char * name = 0x0;
226 // Add all files matching *pattern* to the chain
227 while((name = gSystem->GetDirEntry(dirp))) {
228 if (strstr(name,tagPattern)) fTreeTEsd->Add(name);
230 AliInfo(Form("Chained tag files: %l", fTreeTEsd->GetEntries()));
234 TString fSession = type;
235 TString fguid, fmd5, fturl;
236 fAODEvent = new AliAODEvent();
237 fAODEvent->ReadFromTree(fChain);
239 Int_t firstEvent = 0;
241 TString localFileName = "Run";
242 localFileName += fAODEvent->GetRunNumber();
243 localFileName += ".Event";
244 localFileName += firstEvent;
245 localFileName += "_";
246 localFileName += chain->GetEntries(); //localFileName += "."; localFileName += Counter;
247 localFileName += ".AOD.tag.root";
252 fileName = localFileName.Data();
253 AliInfo(Form("Writing tags to local file: %s",fileName.Data()));
255 else if(fStorage == 1) {
256 TString alienLocation = "/alien";
257 alienLocation += gGrid->Pwd();
258 alienLocation += fgridpath.Data();
259 alienLocation += "/";
260 alienLocation += localFileName;
261 alienLocation += "?se=";
262 alienLocation += fSE.Data();
263 fileName = alienLocation.Data();
264 AliInfo(Form("Writing tags to grid file: %s",fileName.Data()));
267 TFile* ftag = TFile::Open(fileName, "recreate");
269 fRunTag = new AliRunTag();
270 fTreeT = new TTree("T", "A Tree with event tags");
271 TBranch * btag = fTreeT->Branch("AliTAG", &fRunTag);
272 btag->SetCompressionLevel(9);
274 // Access information from esd tag
275 fRunTagEsd = new AliRunTag();
276 fTreeTEsd->SetBranchAddress("AliTAG",&fRunTagEsd);
277 // Creating new information of aod
278 AliInfo(Form("Creating the AOD tags......."));
286 void AliAODTagCreator::CreateTags(const char* type)
288 // Event loop for tag creation
293 TFile *f = fChain->GetFile();
294 TString ftempGuid = f->GetUUID().AsString();
296 Int_t nEvents = fChain->GetEntries();
299 const TClonesArray *evTagList = 0;
301 for (Int_t iEventNumber = 0; iEventNumber < nEvents; iEventNumber++) {
302 // Copy old tag information
303 if (iEventNumber >= ntags) {
304 fTreeTEsd->GetEntry(tagentry++);
305 fRunTag->CopyStandardContent(fRunTagEsd);
306 evTagList = fRunTagEsd->GetEventTags();
307 ntags += evTagList->GetEntries();
311 AliEventTag* evTag = new AliEventTag();
313 fChain->GetEntry(iEventNumber);
314 if (iEventNumber == 0) oldRun = fAODEvent->GetRunNumber();
315 // Reference to the input file
316 TFile *file = fChain->GetFile();
317 const TUrl *url = file->GetEndpointUrl();
318 fguid = file->GetUUID().AsString();
320 if (!strcmp(type,"grid")) {
321 TString fturltemp = "alien://"; fturltemp += url->GetFile();
322 fturl = fturltemp(0,fturltemp.Index(".root",5,0,TString::kExact)+5);
324 fturl = url->GetFile();
327 fAODEvent->GetStdContent();
329 // Fill the event tag from the aod informatiom
330 FillEventTag(fAODEvent, evTag);
331 // Set the event and input file references
332 //evTag->SetEventId(iEventNumber+1);
333 evTag->SetGUID(fguid);
334 if(!strcmp(type,"grid")) {
336 evTag->SetTURL(fturl);
339 else evTag->SetPath(fturl);
341 // Check if a new run has to be created
343 if(fguid != ftempGuid) {
349 if (oldRun != (fAODEvent->GetRunNumber()))
351 oldRun = fAODEvent->GetRunNumber();
357 fRunTag->AddEventTag(*evTag);
360 if(iEventNumber+1 == fChain->GetEntries()) {
369 void AliAODTagCreator::FillEventTag(AliAODEvent* aod, AliEventTag* evTag)
372 // Fill the event tag information
377 Float_t fLowPtCut = 1.0;
378 Float_t fHighPtCut = 3.0;
379 Float_t fVeryHighPtCut = 10.0;
381 Double_t partFrac[10] = {0.01, 0.01, 0.85, 0.10, 0.05, 0., 0., 0., 0., 0.};
383 // Creates the tags for all the events in a given AOD file
385 Int_t nPos = 0, nNeg = 0, nNeutr =0;
386 Int_t nKinks = 0, nV0s = 0, nCascades = 0;
387 Int_t nK0s = 0, nNeutrons = 0, nPi0s = 0, nGamas = 0;
388 Int_t nProtons = 0, nKaons = 0, nPions = 0, nMuons = 0, nElectrons = 0, nFWMuons = 0;
389 Int_t nCh1GeV = 0, nCh3GeV = 0, nCh10GeV = 0;
390 Int_t nMu1GeV = 0, nMu3GeV = 0, nMu10GeV = 0;
391 Int_t nEl1GeV = 0, nEl3GeV = 0, nEl10GeV = 0;
392 Float_t maxPt = .0, etamaxPt = -999., phimaxPt = -999., meanPt = .0, totalP = .0;
398 AliAODVertex *pVertex = fAODEvent->GetPrimaryVertex();
400 evTag->SetVertexX(pVertex->GetX());
401 evTag->SetVertexY(pVertex->GetY());
402 evTag->SetVertexZ(pVertex->GetZ());
403 Double_t covmatrix[6];
404 pVertex->GetCovarianceMatrix(covmatrix);
405 evTag->SetVertexZError(sqrt(covmatrix[5]));
407 // loop over vertices
408 Int_t nVtxs = fAODEvent->GetNumberOfVertices();
409 for (Int_t nVtx = 0; nVtx < nVtxs; nVtx++) {
410 AliAODVertex *vertex = fAODEvent->GetVertex(nVtx);
411 if(vertex->GetType() == 1) nKinks += 1;
412 if(vertex->GetType() == 2) nV0s += 1;
413 if(vertex->GetType() == 3) nCascades += 1;
415 Int_t nTracks = fAODEvent->GetNTracks();
416 for (Int_t nTr = 0; nTr < nTracks; nTr++) {
417 AliAODTrack *track = fAODEvent->GetTrack(nTr);
419 Double_t fPt = track->Pt();
422 etamaxPt = track->Eta();
423 phimaxPt = track->Phi();
426 if(track->Charge() > 0) {
428 if(fPt > fLowPtCut) nCh1GeV++;
429 if(fPt > fHighPtCut) nCh3GeV++;
430 if(fPt > fVeryHighPtCut) nCh10GeV++;
432 if(track->Charge() < 0) {
434 if(fPt > fLowPtCut) nCh1GeV++;
435 if(fPt > fHighPtCut) nCh3GeV++;
436 if(fPt > fVeryHighPtCut) nCh10GeV++;
438 if(track->Charge() == 0) nNeutr++;
440 const Double32_t *prob = track->PID();
442 for(Int_t i = 0; i < AliPID::kSPECIES; i++) rcc += prob[i]*partFrac[i];
443 if(rcc == 0.0) continue;
446 for(Int_t i = 0; i < AliPID::kSPECIES; i++) w[i] = prob[i]*partFrac[i]/rcc;
449 if ((w[4]>w[3])&&(w[4]>w[2])&&(w[4]>w[1])&&(w[4]>w[0])) nProtons++;
451 if ((w[3]>w[4])&&(w[3]>w[2])&&(w[3]>w[1])&&(w[3]>w[0])) nKaons++;
453 if ((w[2]>w[4])&&(w[2]>w[3])&&(w[2]>w[1])&&(w[2]>w[0])) nPions++;
455 if ((w[1]>w[4])&&(w[1]>w[3])&&(w[1]>w[2])&&(w[1]>w[0])) {
457 if(fPt > fLowPtCut) nMu1GeV++;
458 if(fPt > fHighPtCut) nMu3GeV++;
459 if(fPt > fVeryHighPtCut) nMu10GeV++;
462 if ((w[0]>w[4])&&(w[0]>w[3])&&(w[0]>w[2])&&(w[0]>w[1])) {
464 if(fPt > fLowPtCut) nEl1GeV++;
465 if(fPt > fHighPtCut) nEl3GeV++;
466 if(fPt > fVeryHighPtCut) nEl10GeV++;
468 totalP += track->P();
471 // forward muons (in the dimuon spectrometer)
472 if(track->IsMuonTrack()) nFWMuons++;
476 // Fill the event tags
478 meanPt = meanPt/ntrack;
481 evTag->SetNumOfTracks(nTracks);
482 evTag->SetNumOfPosTracks(nPos);
483 evTag->SetNumOfNegTracks(nNeg);
484 evTag->SetNumOfNeutrTracks(nNeutr);
486 evTag->SetNumOfV0s(nV0s);
487 evTag->SetNumOfCascades(nCascades);
488 evTag->SetNumOfKinks(nKinks);
490 evTag->SetNumOfProtons(nProtons);
491 evTag->SetNumOfKaons(nKaons);
492 evTag->SetNumOfPions(nPions);
493 evTag->SetNumOfMuons(nMuons);
494 evTag->SetNumOfFWMuons(nFWMuons);
495 evTag->SetNumOfElectrons(nElectrons);
496 evTag->SetNumOfPhotons(nGamas);
497 evTag->SetNumOfPi0s(nPi0s);
498 evTag->SetNumOfNeutrons(nNeutrons);
499 evTag->SetNumOfKaon0s(nK0s);
501 evTag->SetNumOfChargedAbove1GeV(nCh1GeV);
502 evTag->SetNumOfChargedAbove3GeV(nCh3GeV);
503 evTag->SetNumOfChargedAbove10GeV(nCh10GeV);
504 evTag->SetNumOfMuonsAbove1GeV(nMu1GeV);
505 evTag->SetNumOfMuonsAbove3GeV(nMu3GeV);
506 evTag->SetNumOfMuonsAbove10GeV(nMu10GeV);
507 evTag->SetNumOfElectronsAbove1GeV(nEl1GeV);
508 evTag->SetNumOfElectronsAbove3GeV(nEl3GeV);
509 evTag->SetNumOfElectronsAbove10GeV(nEl10GeV);
512 evTag->SetNumOfPHOSClusters(fAODEvent->GetPHOSClusters(&tmp));
514 evTag->SetNumOfEMCALClusters(fAODEvent->GetEMCALClusters(&tmp));
516 evTag->SetTotalMomentum(totalP);
517 evTag->SetMeanPt(meanPt);
518 evTag->SetMaxPt(maxPt);
519 evTag->SetEtaMaxPt(etamaxPt);
520 evTag->SetPhiMaxPt(phimaxPt);