#include <TList.h>
#include <TObjString.h>
#include <TLorentzVector.h>
+#include <TMap.h>
+#include <TTimeStamp.h>
+#include <TRefArray.h>
//ROOT-AliEn
#include <TGrid.h>
#include "AliESDEvent.h"
#include "AliESDVertex.h"
#include "AliLog.h"
+#include "AliGRPObject.h"
#include "AliESDTagCreator.h"
AliESDTagCreator::AliESDTagCreator() :
AliTagCreator(),
fChain(new TChain("esdTree")), fGUIDList(new TList()),
- fMD5List(new TList()), fTURLList(new TList()),
+ fMD5List(new TList()), fTURLList(new TList()), fBranches(""),
meminfo(new MemInfo_t) {
//==============Default constructor for a AliESDTagCreator================
}
counter += 1;
}//grid result loop
- AliInfo(Form("ESD chain created......."));
- AliInfo(Form("Chain entries: %d",fChain->GetEntries()));
+ if (fChain->GetEntries() > 0) {
+ AliInfo(Form("ESD chain created......."));
+ AliInfo(Form("Chain entries: %d",fChain->GetEntries()));
+ } else {
+ AliWarning(Form("No ESD files found !"));
+ return kFALSE;
+ }
+
+ // Switch of branches on user request
+ SwitchOffBranches();
CreateTag(fChain,"grid");
return kTRUE;
AliInfo(Form("ESD chain created......."));
AliInfo(Form("Chain entries: %d",fChain->GetEntries()));
+ // Switch of branches on user request
+ SwitchOffBranches();
CreateTag(fChain,"local");
return kTRUE;
AliInfo(Form("ESD chain created......."));
AliInfo(Form("Chain entries: %d",fChain->GetEntries()));
+ // Switch of branches on user request
+ SwitchOffBranches();
CreateTag(fChain,"proof");
return kTRUE;
}
//_____________________________________________________________________________
-void AliESDTagCreator::CreateTag(TChain* fChain, const char *type) {
+void AliESDTagCreator::CreateTag(TChain* chain, const char *type) {
//private method that creates tag files
TString fSession = type;
- //Int_t iCounter = 0;
TString fguid, fmd5, fturl;
- TString fTempGuid = 0;
+ TString fTempGuid;
/////////////
//muon code//
Int_t fCharge;
TLorentzVector fEPvector;
- Float_t fZVertexCut = 10.0;
- Float_t fRhoVertexCut = 2.0;
-
- Float_t fLowPtCut = 1.0;
- Float_t fHighPtCut = 3.0;
+ Float_t fLowPtCut = 1.0;
+ Float_t fHighPtCut = 3.0;
Float_t fVeryHighPtCut = 10.0;
////////////
// Creates the tags for all the events in a given ESD file
Bool_t fIsSim = kTRUE;
Int_t ntrack;
- Int_t nProtons, nKaons, nPions, nMuons, nElectrons;
+ Int_t nProtons, nKaons, nPions, nMuons, nElectrons, nFWMuons, nFWMatchedMuons;
Int_t nPos, nNeg, nNeutr;
Int_t nK0s, nNeutrons, nPi0s, nGamas;
Int_t nCh1GeV, nCh3GeV, nCh10GeV;
Int_t nMu1GeV, nMu3GeV, nMu10GeV;
Int_t nEl1GeV, nEl3GeV, nEl10GeV;
- Float_t maxPt = .0, meanPt = .0, totalP = .0;
+ Float_t maxPt = .0, etamaxPt = -999., phimaxPt = -999., meanPt = .0, totalP = .0;
Int_t fVertexflag;
Int_t iRunNumber = 0;
TString fVertexName;
- AliRunTag *tag = new AliRunTag();
- AliEventTag *evTag = new AliEventTag();
- TTree ttag("T","A Tree with event tags");
- TBranch * btag = ttag.Branch("AliTAG", &tag);
- btag->SetCompressionLevel(9);
//gSystem->GetMemInfo(meminfo);
//AliInfo(Form("After the tag initialization - Memory used: %d MB",meminfo->fMemUsed));
//Int_t tempmem = meminfo->fMemUsed;
Int_t firstEvent = 0,lastEvent = 0;
AliESDEvent *esd = new AliESDEvent();
- esd->ReadFromTree(fChain);
+ esd->ReadFromTree(chain);
+ AliESD *esdold = 0x0;
//gSystem->GetMemInfo(meminfo);
//AliInfo(Form("After the esd initialization - Memory used: %d MB - Increase: %d MB",meminfo->fMemUsed,meminfo->fMemUsed - tempmem));
//tempmem = meminfo->fMemUsed;
Int_t iInitRunNumber = -1;
- for(Int_t iEventNumber = 0; iEventNumber < fChain->GetEntries(); iEventNumber++) {
+ chain->GetEntry(0);
+ TFile *f = chain->GetFile();
+ fTempGuid = f->GetUUID().AsString();
+
+ TString localFileName = "Run"; localFileName += esd->GetRunNumber();
+ localFileName += ".Event"; localFileName += firstEvent; localFileName += "_"; localFileName += chain->GetEntries(); //localFileName += "."; localFileName += Counter;
+ localFileName += ".ESD.tag.root";
+
+ TString fileName;
+ TRefArray tmp;
+
+ if(fStorage == 0) {
+ fileName = localFileName.Data();
+ AliInfo(Form("Writing tags to local file: %s",fileName.Data()));
+ }
+ else if(fStorage == 1) {
+ TString alienLocation = "/alien";
+ alienLocation += gGrid->Pwd();
+ alienLocation += fgridpath.Data();
+ alienLocation += "/";
+ alienLocation += localFileName;
+ alienLocation += "?se=";
+ alienLocation += fSE.Data();
+ fileName = alienLocation.Data();
+ AliInfo(Form("Writing tags to grid file: %s",fileName.Data()));
+ }
+
+ TFile* ftag = TFile::Open(fileName, "recreate");
+
+ AliRunTag *tag = new AliRunTag();
+ AliEventTag *evTag = new AliEventTag();
+ TTree * ttag = new TTree("T","A Tree with event tags");
+ TBranch * btag = ttag->Branch("AliTAG", &tag);
+ btag->SetCompressionLevel(9);
+ // Run related information
+ tag->SetMagneticField(esd->GetMagneticField());
+ tag->SetBeamEnergy(esd->GetBeamEnergy());
+ tag->SetBeamType(TString(esd->GetBeamType()));
+
+ for(Int_t iEventNumber = 0; iEventNumber < chain->GetEntries(); iEventNumber++) {
ntrack = 0; nPos = 0; nNeg = 0; nNeutr =0;
nK0s = 0; nNeutrons = 0; nPi0s = 0;
nGamas = 0; nProtons = 0; nKaons = 0;
- nPions = 0; nMuons = 0; nElectrons = 0;
+ nPions = 0; nMuons = 0; nElectrons = 0; nFWMuons = 0; nFWMatchedMuons = 0;
nCh1GeV = 0; nCh3GeV = 0; nCh10GeV = 0;
nMu1GeV = 0; nMu3GeV = 0; nMu10GeV = 0;
nEl1GeV = 0; nEl3GeV = 0; nEl10GeV = 0;
- maxPt = .0; meanPt = .0; totalP = .0;
+ maxPt = .0; etamaxPt = -999.; phimaxPt = -999.; meanPt = .0; totalP = .0;
fVertexflag = 1;
- fChain->GetEntry(iEventNumber);
- TFile *f = fChain->GetFile();
- const TUrl *url = f->GetEndpointUrl();
- fguid = f->GetUUID().AsString();
+ chain->GetEntry(iEventNumber);
+ esdold = esd->GetAliESDOld();
+ if(esdold) esd->CopyFromOldESD();
+
+ TFile *file = chain->GetFile();
+ const TUrl *url = file->GetEndpointUrl();
+ fguid = file->GetUUID().AsString();
if(fSession == "grid") {
TString fturltemp = "alien://"; fturltemp += url->GetFile();
fturl = fturltemp(0,fturltemp.Index(".root",5,0,TString::kExact)+5);
if ((status&AliESDtrack::kESDpid)==0) continue;
Double_t p[3];
esdTrack->GetPxPyPz(p);
- Double_t momentum = sqrt(pow(p[0],2) + pow(p[1],2) + pow(p[2],2));
- Double_t fPt = sqrt(pow(p[0],2) + pow(p[1],2));
+ Double_t pt2 = p[0]*p[0]+p[1]*p[1];
+ Double_t momentum = TMath::Sqrt(pt2+p[2]*p[2]);
+ Double_t fPt = TMath::Sqrt(pt2);
totalP += momentum;
meanPt += fPt;
- if(fPt > maxPt) maxPt = fPt;
+ if(fPt > maxPt) {
+ maxPt = fPt;
+ phimaxPt = esdTrack->Phi();
+ etamaxPt = esdTrack->Eta();
+ }
if(esdTrack->GetSign() > 0) {
nPos++;
fEnergy = TMath::Sqrt(fMUONMASS * fMUONMASS + fPxRec * fPxRec + fPyRec * fPyRec + fPzRec * fPzRec);
fEPvector.SetPxPyPzE(fPxRec, fPyRec, fPzRec, fEnergy);
- // total number of muons inside a vertex cut
- if((TMath::Abs(fZ)<fZVertexCut) && (TMath::Sqrt(fY*fY+fX*fX)<fRhoVertexCut)) {
- nMuons++;
- if(fEPvector.Pt() > fLowPtCut) {
- nMu1GeV++;
- if(fEPvector.Pt() > fHighPtCut) {
- nMu3GeV++;
- if (fEPvector.Pt() > fVeryHighPtCut) {
- nMu10GeV++;
- }
- }
- }
+ if (muonTrack->GetMatchTrigger()>0) nFWMatchedMuons++;
+
+ nMuons++;
+ nFWMuons++;
+ if(fEPvector.Pt() > fLowPtCut) {
+ nMu1GeV++;
+ if(fEPvector.Pt() > fHighPtCut) {
+ nMu3GeV++;
+ if (fEPvector.Pt() > fVeryHighPtCut) {
+ nMu10GeV++;
+ }
+ }
}
}//muon track loop
// Fill the event tags
if(ntrack != 0) meanPt = meanPt/ntrack;
- AliInfo(Form("====================================="));
- AliInfo(Form("URL: %s - GUID: %s",fturl.Data(),fguid.Data()));
- AliInfo(Form("====================================="));
-
- evTag->SetEventId(iEventNumber+1);
+ //AliInfo(Form("====================================="));
+ //AliInfo(Form("URL: %s - GUID: %s",fturl.Data(),fguid.Data()));
+ //AliInfo(Form("====================================="));
+
+ //First physics data
+ const AliMultiplicity *spdMult = esd->GetMultiplicity();
+ evTag->SetNumberOfFiredChipsLayer1(spdMult->GetNumberOfFiredChips(0));
+ evTag->SetNumberOfFiredChipsLayer2(spdMult->GetNumberOfFiredChips(1));
+ evTag->SetNumberOfSPDTracklets(spdMult->GetNumberOfTracklets());
+
+ AliESDVZERO *vzeroData = esd->GetVZEROData();
+ evTag->SetMTotV0A(vzeroData->GetMTotV0A());
+ evTag->SetMTotV0C(vzeroData->GetMTotV0C());
+ evTag->SetNbPMV0A(vzeroData->GetNbPMV0A());
+ evTag->SetNbPMV0C(vzeroData->GetNbPMV0C());
+
+ //evTag->SetEventId(iEventNumber+1);
+ evTag->SetPeriodNumber(esd->GetPeriodNumber());
+ evTag->SetOrbitNumber(esd->GetOrbitNumber());
+ evTag->SetBunchCrossNumber(esd->GetBunchCrossNumber());
evTag->SetGUID(fguid);
if(fSession == "grid") {
- evTag->SetMD5(0);
+ evTag->SetMD5("");
evTag->SetTURL(fturl);
evTag->SetSize(0);
}
evTag->SetTriggerMask(esd->GetTriggerMask());
evTag->SetTriggerCluster(esd->GetTriggerCluster());
+ evTag->SetEventType(esd->GetEventType());
+ evTag->SetFiredTriggerClasses(esd->GetFiredTriggerClasses());
+
evTag->SetZDCNeutron1Energy(esd->GetZDCN1Energy());
evTag->SetZDCProton1Energy(esd->GetZDCP1Energy());
- evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy());
+ evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy(0),esd->GetZDCEMEnergy(1));
evTag->SetZDCNeutron1Energy(esd->GetZDCN2Energy());
evTag->SetZDCProton1Energy(esd->GetZDCP2Energy());
evTag->SetNumOfParticipants(esd->GetZDCParticipants());
evTag->SetNumOfKaons(nKaons);
evTag->SetNumOfPions(nPions);
evTag->SetNumOfMuons(nMuons);
+ evTag->SetNumOfFWMuons(nFWMuons);
+ evTag->SetNumOfFWMatchedMuons(nFWMatchedMuons);
evTag->SetNumOfElectrons(nElectrons);
evTag->SetNumOfPhotons(nGamas);
evTag->SetNumOfPi0s(nPi0s);
evTag->SetNumOfElectronsAbove3GeV(nEl3GeV);
evTag->SetNumOfElectronsAbove10GeV(nEl10GeV);
- evTag->SetNumOfPHOSClusters(esd->GetNumberOfPHOSClusters());
- evTag->SetNumOfEMCALClusters(esd->GetNumberOfEMCALClusters());
+ tmp.Clear();
+ evTag->SetNumOfPHOSClusters(esd->GetPHOSClusters(&tmp));
+ tmp.Clear();
+ evTag->SetNumOfEMCALClusters(esd->GetEMCALClusters(&tmp));
evTag->SetTotalMomentum(totalP);
evTag->SetMeanPt(meanPt);
evTag->SetMaxPt(maxPt);
+ evTag->SetEtaMaxPt(etamaxPt);
+ evTag->SetPhiMaxPt(phimaxPt);
tag->SetRunId(iInitRunNumber);
if(fIsSim) tag->SetDataType(0);
else tag->SetDataType(1);
- tag->AddEventTag(*evTag);
if(fguid != fTempGuid) {
fTempGuid = fguid;
- ttag.Fill();
+ ttag->Fill();
tag->Clear("");
}
+ tag->AddEventTag(*evTag);
+ if(iEventNumber+1 == chain->GetEntries()) {
+ //AliInfo(Form("File: %s",fturl.Data()));
+ ttag->Fill();
+ tag->Clear("");
+ }
}//event loop
- lastEvent = fChain->GetEntries();
+ lastEvent = chain->GetEntries();
//gSystem->GetMemInfo(meminfo);
//AliInfo(Form("After the event and track loop - Memory used: %d MB - Increase: %d MB",meminfo->fMemUsed,meminfo->fMemUsed - tempmem));
//tempmem = meminfo->fMemUsed;
- //fChain->Delete("");
+ //chain->Delete("");
//gSystem->GetMemInfo(meminfo);
//AliInfo(Form("After the t->Delete - Memory used: %d MB - Increase: %d MB",meminfo->fMemUsed,meminfo->fMemUsed - tempmem));
//tempmem = meminfo->fMemUsed;
- TString localFileName = "Run"; localFileName += tag->GetRunId();
- localFileName += ".Event"; localFileName += firstEvent; localFileName += "_"; localFileName += lastEvent; //localFileName += "."; localFileName += Counter;
- localFileName += ".ESD.tag.root";
-
- TString fileName;
-
- if(fStorage == 0) {
- fileName = localFileName.Data();
- AliInfo(Form("Writing tags to local file: %s",fileName.Data()));
- }
- else if(fStorage == 1) {
- TString alienLocation = "/alien";
- alienLocation += gGrid->Pwd();
- alienLocation += fgridpath.Data();
- alienLocation += "/";
- alienLocation += localFileName;
- alienLocation += "?se=";
- alienLocation += fSE.Data();
- fileName = alienLocation.Data();
- AliInfo(Form("Writing tags to grid file: %s",fileName.Data()));
- }
-
- TFile* ftag = TFile::Open(fileName, "recreate");
ftag->cd();
- //ttag.Fill();
tag->Clear();
- ttag.Write();
+ ttag->Write();
ftag->Close();
//gSystem->GetMemInfo(meminfo);
//AliInfo(Form("After the file closing - Memory used: %d MB - Increase: %d MB",meminfo->fMemUsed,meminfo->fMemUsed - tempmem));
//tempmem = meminfo->fMemUsed;
- delete ftag;
delete esd;
-
delete tag;
+
//gSystem->GetMemInfo(meminfo);
//AliInfo(Form("After the delete objects - Memory used: %d MB - Increase: %d MB",meminfo->fMemUsed,meminfo->fMemUsed - tempmem));
}
Int_t fCharge;
TLorentzVector fEPvector;
- Float_t fZVertexCut = 10.0;
- Float_t fRhoVertexCut = 2.0;
-
Float_t fLowPtCut = 1.0;
Float_t fHighPtCut = 3.0;
Float_t fVeryHighPtCut = 10.0;
// Creates the tags for all the events in a given ESD file
Bool_t fIsSim = kTRUE;
Int_t ntrack;
- Int_t nProtons, nKaons, nPions, nMuons, nElectrons;
+ Int_t nProtons, nKaons, nPions, nMuons, nElectrons, nFWMuons, nFWMatchedMuons;
Int_t nPos, nNeg, nNeutr;
Int_t nK0s, nNeutrons, nPi0s, nGamas;
Int_t nCh1GeV, nCh3GeV, nCh10GeV;
Int_t nMu1GeV, nMu3GeV, nMu10GeV;
Int_t nEl1GeV, nEl3GeV, nEl10GeV;
- Float_t maxPt = .0, meanPt = .0, totalP = .0;
+ Float_t maxPt = .0, etamaxPt = -999., phimaxPt = -999., meanPt = .0, totalP = .0;
Int_t fVertexflag;
Int_t iRunNumber = 0;
TString fVertexName;
+ TRefArray tmp;
AliRunTag *tag = new AliRunTag();
AliEventTag *evTag = new AliEventTag();
- TTree ttag("T","A Tree with event tags");
- TBranch * btag = ttag.Branch("AliTAG", &tag);
+ TTree * ttag = new TTree("T","A Tree with event tags");
+ TBranch * btag = ttag->Branch("AliTAG", &tag);
btag->SetCompressionLevel(9);
gSystem->GetMemInfo(meminfo);
AliInfo(Form("After the tag initialization - Memory used: %d MB",meminfo->fMemUsed));
nKaons = 0;
nPions = 0;
nMuons = 0;
+ nFWMuons = 0;
+ nFWMatchedMuons = 0;
nElectrons = 0;
nCh1GeV = 0;
nCh3GeV = 0;
nEl3GeV = 0;
nEl10GeV = 0;
maxPt = .0;
+ etamaxPt = -999.;
+ phimaxPt = -999.;
meanPt = .0;
totalP = .0;
fVertexflag = 1;
if ((status&AliESDtrack::kESDpid)==0) continue;
Double_t p[3];
esdTrack->GetPxPyPz(p);
- Double_t momentum = sqrt(pow(p[0],2) + pow(p[1],2) + pow(p[2],2));
- Double_t fPt = sqrt(pow(p[0],2) + pow(p[1],2));
+ Double_t pt2 = p[0]*p[0]+p[1]*p[1];
+ Double_t momentum = TMath::Sqrt(pt2+p[2]*p[2]);
+ Double_t fPt = TMath::Sqrt(pt2);
totalP += momentum;
meanPt += fPt;
- if(fPt > maxPt) maxPt = fPt;
+ if(fPt > maxPt) {
+ maxPt = fPt;
+ etamaxPt = esdTrack->Eta();
+ phimaxPt = esdTrack->Phi();
+ }
+
if(esdTrack->GetSign() > 0) {
nPos++;
if(fPt > fLowPtCut) nCh1GeV++;
fEnergy = TMath::Sqrt(fMUONMASS * fMUONMASS + fPxRec * fPxRec + fPyRec * fPyRec + fPzRec * fPzRec);
fEPvector.SetPxPyPzE(fPxRec, fPyRec, fPzRec, fEnergy);
- // total number of muons inside a vertex cut
- if((TMath::Abs(fZ)<fZVertexCut) && (TMath::Sqrt(fY*fY+fX*fX)<fRhoVertexCut)) {
- nMuons++;
- if(fEPvector.Pt() > fLowPtCut) {
- nMu1GeV++;
- if(fEPvector.Pt() > fHighPtCut) {
- nMu3GeV++;
- if (fEPvector.Pt() > fVeryHighPtCut) {
- nMu10GeV++;
- }
- }
- }
+ if (muonTrack->GetMatchTrigger()>0) nFWMatchedMuons++;
+
+ nMuons++;
+ nFWMuons++;
+ if(fEPvector.Pt() > fLowPtCut) {
+ nMu1GeV++;
+ if(fEPvector.Pt() > fHighPtCut) {
+ nMu3GeV++;
+ if (fEPvector.Pt() > fVeryHighPtCut) {
+ nMu10GeV++;
+ }
+ }
}
}//muon track loop
// Fill the event tags
if(ntrack != 0) meanPt = meanPt/ntrack;
- evTag->SetEventId(iEventNumber+1);
+ //First physics data
+ const AliMultiplicity *spdMult = esd->GetMultiplicity();
+ evTag->SetNumberOfFiredChipsLayer1(spdMult->GetNumberOfFiredChips(0));
+ evTag->SetNumberOfFiredChipsLayer2(spdMult->GetNumberOfFiredChips(1));
+ evTag->SetNumberOfSPDTracklets(spdMult->GetNumberOfTracklets());
+
+ AliESDVZERO *vzeroData = esd->GetVZEROData();
+ evTag->SetMTotV0A(vzeroData->GetMTotV0A());
+ evTag->SetMTotV0C(vzeroData->GetMTotV0C());
+ evTag->SetNbPMV0A(vzeroData->GetNbPMV0A());
+ evTag->SetNbPMV0C(vzeroData->GetNbPMV0C());
+
+ //evTag->SetEventId(iEventNumber+1);
+ evTag->SetPeriodNumber(esd->GetPeriodNumber());
+ evTag->SetOrbitNumber(esd->GetOrbitNumber());
+ evTag->SetBunchCrossNumber(esd->GetBunchCrossNumber());
+
evTag->SetGUID(fguid);
evTag->SetMD5(fmd5);
evTag->SetTURL(fturl);
evTag->SetTriggerMask(esd->GetTriggerMask());
evTag->SetTriggerCluster(esd->GetTriggerCluster());
+ evTag->SetEventType(esd->GetEventType());
+ evTag->SetFiredTriggerClasses(esd->GetFiredTriggerClasses());
+
evTag->SetZDCNeutron1Energy(esd->GetZDCN1Energy());
evTag->SetZDCProton1Energy(esd->GetZDCP1Energy());
- evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy());
+ evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy(0),esd->GetZDCEMEnergy(1));
evTag->SetZDCNeutron1Energy(esd->GetZDCN2Energy());
evTag->SetZDCProton1Energy(esd->GetZDCP2Energy());
evTag->SetNumOfParticipants(esd->GetZDCParticipants());
evTag->SetNumOfKaons(nKaons);
evTag->SetNumOfPions(nPions);
evTag->SetNumOfMuons(nMuons);
+ evTag->SetNumOfFWMuons(nFWMuons);
+ evTag->SetNumOfFWMatchedMuons(nFWMatchedMuons);
evTag->SetNumOfElectrons(nElectrons);
evTag->SetNumOfPhotons(nGamas);
evTag->SetNumOfPi0s(nPi0s);
evTag->SetNumOfElectronsAbove3GeV(nEl3GeV);
evTag->SetNumOfElectronsAbove10GeV(nEl10GeV);
- evTag->SetNumOfPHOSClusters(esd->GetNumberOfPHOSClusters());
- evTag->SetNumOfEMCALClusters(esd->GetNumberOfEMCALClusters());
+ tmp.Clear();
+ evTag->SetNumOfPHOSClusters(esd->GetPHOSClusters(&tmp));
+ tmp.Clear();
+ evTag->SetNumOfEMCALClusters(esd->GetEMCALClusters(&tmp));
evTag->SetTotalMomentum(totalP);
evTag->SetMeanPt(meanPt);
evTag->SetMaxPt(maxPt);
+ evTag->SetEtaMaxPt(etamaxPt);
+ evTag->SetPhiMaxPt(phimaxPt);
tag->SetRunId(iInitRunNumber);
if(fIsSim) tag->SetDataType(0);
TFile* ftag = TFile::Open(fileName, "recreate");
ftag->cd();
- ttag.Fill();
+ ttag->Fill();
tag->Clear();
- ttag.Write();
+ ttag->Write();
ftag->Close();
gSystem->GetMemInfo(meminfo);
Int_t fCharge;
TLorentzVector fEPvector;
- Float_t fZVertexCut = 10.0;
- Float_t fRhoVertexCut = 2.0;
-
Float_t fLowPtCut = 1.0;
Float_t fHighPtCut = 3.0;
Float_t fVeryHighPtCut = 10.0;
// Creates the tags for all the events in a given ESD file
Bool_t fIsSim = kTRUE;
Int_t ntrack;
- Int_t nProtons, nKaons, nPions, nMuons, nElectrons;
+ Int_t nProtons, nKaons, nPions, nMuons, nElectrons, nFWMuons, nFWMatchedMuons;
Int_t nPos, nNeg, nNeutr;
Int_t nK0s, nNeutrons, nPi0s, nGamas;
Int_t nCh1GeV, nCh3GeV, nCh10GeV;
Int_t nMu1GeV, nMu3GeV, nMu10GeV;
Int_t nEl1GeV, nEl3GeV, nEl10GeV;
- Float_t maxPt = .0, meanPt = .0, totalP = .0;
+ Float_t maxPt = .0, etamaxPt = -999, phimaxPt = -999., meanPt = .0, totalP = .0;
Int_t fVertexflag;
Int_t iRunNumber = 0;
TString fVertexName;
+ TRefArray tmp;
AliRunTag *tag = new AliRunTag();
AliEventTag *evTag = new AliEventTag();
- TTree ttag("T","A Tree with event tags");
- TBranch * btag = ttag.Branch("AliTAG", &tag);
+ TTree * ttag = new TTree("T","A Tree with event tags");
+ TBranch * btag = ttag->Branch("AliTAG", &tag);
btag->SetCompressionLevel(9);
AliInfo(Form("Creating the ESD tags......."));
nKaons = 0;
nPions = 0;
nMuons = 0;
+ nFWMuons = 0;
+ nFWMatchedMuons = 0;
nElectrons = 0;
nCh1GeV = 0;
nCh3GeV = 0;
nEl3GeV = 0;
nEl10GeV = 0;
maxPt = .0;
+ etamaxPt = -999.;
+ phimaxPt = -999.;
meanPt = .0;
totalP = .0;
fVertexflag = 1;
if ((status&AliESDtrack::kESDpid)==0) continue;
Double_t p[3];
esdTrack->GetPxPyPz(p);
- Double_t momentum = sqrt(pow(p[0],2) + pow(p[1],2) + pow(p[2],2));
- Double_t fPt = sqrt(pow(p[0],2) + pow(p[1],2));
+ Double_t pt2 = p[0]*p[0]+p[1]*p[1];
+ Double_t momentum = TMath::Sqrt(pt2+p[2]*p[2]);
+ Double_t fPt = TMath::Sqrt(pt2);
totalP += momentum;
meanPt += fPt;
- if(fPt > maxPt) maxPt = fPt;
+ if(fPt > maxPt) {
+ maxPt = fPt;
+ etamaxPt = esdTrack->Eta();
+ phimaxPt = esdTrack->Phi();
+ }
+
if(esdTrack->GetSign() > 0) {
nPos++;
fEnergy = TMath::Sqrt(fMUONMASS * fMUONMASS + fPxRec * fPxRec + fPyRec * fPyRec + fPzRec * fPzRec);
fEPvector.SetPxPyPzE(fPxRec, fPyRec, fPzRec, fEnergy);
- // total number of muons inside a vertex cut
- if((TMath::Abs(fZ)<fZVertexCut) && (TMath::Sqrt(fY*fY+fX*fX)<fRhoVertexCut)) {
- nMuons++;
- if(fEPvector.Pt() > fLowPtCut) {
- nMu1GeV++;
- if(fEPvector.Pt() > fHighPtCut) {
- nMu3GeV++;
- if (fEPvector.Pt() > fVeryHighPtCut) {
- nMu10GeV++;
- }
- }
- }
+ if (muonTrack->GetMatchTrigger()>0) nFWMatchedMuons++;
+
+ nMuons++;
+ nFWMuons++;
+ if(fEPvector.Pt() > fLowPtCut) {
+ nMu1GeV++;
+ if(fEPvector.Pt() > fHighPtCut) {
+ nMu3GeV++;
+ if (fEPvector.Pt() > fVeryHighPtCut) {
+ nMu10GeV++;
+ }
+ }
}
}//muon track loop
// Fill the event tags
if(ntrack != 0) meanPt = meanPt/ntrack;
- evTag->SetEventId(iEventNumber+1);
+ //First physics data
+ const AliMultiplicity *spdMult = esd->GetMultiplicity();
+ evTag->SetNumberOfFiredChipsLayer1(spdMult->GetNumberOfFiredChips(0));
+ evTag->SetNumberOfFiredChipsLayer2(spdMult->GetNumberOfFiredChips(1));
+ evTag->SetNumberOfSPDTracklets(spdMult->GetNumberOfTracklets());
+
+ AliESDVZERO *vzeroData = esd->GetVZEROData();
+ evTag->SetMTotV0A(vzeroData->GetMTotV0A());
+ evTag->SetMTotV0C(vzeroData->GetMTotV0C());
+ evTag->SetNbPMV0A(vzeroData->GetNbPMV0A());
+ evTag->SetNbPMV0C(vzeroData->GetNbPMV0C());
+
+ //evTag->SetEventId(iEventNumber+1);
+ evTag->SetPeriodNumber(esd->GetPeriodNumber());
+ evTag->SetOrbitNumber(esd->GetOrbitNumber());
+ evTag->SetBunchCrossNumber(esd->GetBunchCrossNumber());
evTag->SetPath(filepath);
evTag->SetVertexX(vertexIn->GetXv());
evTag->SetTriggerMask(esd->GetTriggerMask());
evTag->SetTriggerCluster(esd->GetTriggerCluster());
+ evTag->SetEventType(esd->GetEventType());
+ evTag->SetFiredTriggerClasses(esd->GetFiredTriggerClasses());
+
evTag->SetZDCNeutron1Energy(esd->GetZDCN1Energy());
evTag->SetZDCProton1Energy(esd->GetZDCP1Energy());
- evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy());
+ evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy(0),esd->GetZDCEMEnergy(1));
evTag->SetZDCNeutron1Energy(esd->GetZDCN2Energy());
evTag->SetZDCProton1Energy(esd->GetZDCP2Energy());
evTag->SetNumOfParticipants(esd->GetZDCParticipants());
evTag->SetNumOfKaons(nKaons);
evTag->SetNumOfPions(nPions);
evTag->SetNumOfMuons(nMuons);
+ evTag->SetNumOfFWMuons(nFWMuons);
+ evTag->SetNumOfFWMatchedMuons(nFWMatchedMuons);
evTag->SetNumOfElectrons(nElectrons);
evTag->SetNumOfPhotons(nGamas);
evTag->SetNumOfPi0s(nPi0s);
evTag->SetNumOfElectronsAbove3GeV(nEl3GeV);
evTag->SetNumOfElectronsAbove10GeV(nEl10GeV);
- evTag->SetNumOfPHOSClusters(esd->GetNumberOfPHOSClusters());
- evTag->SetNumOfEMCALClusters(esd->GetNumberOfEMCALClusters());
+ tmp.Clear();
+ evTag->SetNumOfPHOSClusters(esd->GetPHOSClusters(&tmp));
+ tmp.Clear();
+ evTag->SetNumOfEMCALClusters(esd->GetEMCALClusters(&tmp));
evTag->SetTotalMomentum(totalP);
evTag->SetMeanPt(meanPt);
evTag->SetMaxPt(maxPt);
+ evTag->SetEtaMaxPt(etamaxPt);
+ evTag->SetPhiMaxPt(phimaxPt);
tag->SetRunId(iInitRunNumber);
if(fIsSim) tag->SetDataType(0);
TFile* ftag = TFile::Open(fileName, "recreate");
ftag->cd();
- ttag.Fill();
+ ttag->Fill();
tag->Clear();
- ttag.Write();
+ ttag->Write();
ftag->Close();
delete ftag;
}
//_____________________________________________________________________________
-void AliESDTagCreator::CreateESDTags(Int_t fFirstEvent, Int_t fLastEvent) {
+void AliESDTagCreator::CreateESDTags(Int_t fFirstEvent, Int_t fLastEvent, AliGRPObject *grpData, ULong_t * qa, Bool_t * es, Int_t qalength, Int_t eslength) {
//GRP
Float_t lhcLuminosity = 0.0;
TString lhcState = "test";
- UInt_t detectorMask = 0;
+ //UInt_t detectorMask = 0;
+ Int_t detectorMask = 0;
+
+ detectorMask = grpData->GetDetectorMask();
+ time_t startTime = grpData->GetTimeStart();
+ TTimeStamp t1(startTime);
+ time_t endTime = grpData->GetTimeEnd();
+ TTimeStamp t2(endTime);
+ const char* beamtype = grpData->GetBeamType();
+ Float_t beamenergy = grpData->GetBeamEnergy();
+
/////////////
//muon code//
Int_t fCharge;
TLorentzVector fEPvector;
- Float_t fZVertexCut = 10.0;
- Float_t fRhoVertexCut = 2.0;
-
Float_t fLowPtCut = 1.0;
Float_t fHighPtCut = 3.0;
Float_t fVeryHighPtCut = 10.0;
// Creates the tags for all the events in a given ESD file
Int_t ntrack;
- Int_t nProtons, nKaons, nPions, nMuons, nElectrons;
+ Int_t nProtons, nKaons, nPions, nMuons, nElectrons, nFWMuons, nFWMatchedMuons;
Int_t nPos, nNeg, nNeutr;
Int_t nK0s, nNeutrons, nPi0s, nGamas;
Int_t nCh1GeV, nCh3GeV, nCh10GeV;
Int_t nMu1GeV, nMu3GeV, nMu10GeV;
Int_t nEl1GeV, nEl3GeV, nEl10GeV;
- Float_t maxPt = .0, meanPt = .0, totalP = .0;
+ Float_t maxPt = .0, etamaxPt = -999., phimaxPt = -999., meanPt = .0, totalP = .0;
Int_t fVertexflag;
Int_t iRunNumber = 0;
TString fVertexName("default");
-
- AliRunTag *tag = new AliRunTag();
- AliEventTag *evTag = new AliEventTag();
- TTree ttag("T","A Tree with event tags");
- TBranch * btag = ttag.Branch("AliTAG", &tag);
- btag->SetCompressionLevel(9);
+ TRefArray tmp;
AliInfo(Form("Creating the ESD tags......."));
Int_t iInitRunNumber = esd->GetRunNumber();
Int_t iNumberOfEvents = (Int_t)b->GetEntries();
- if(fLastEvent != -1) iNumberOfEvents = fLastEvent + 1;
+ if ((fLastEvent == -1) || ((Int_t) b->GetEntries() < fLastEvent))
+ lastEvent = (Int_t)b->GetEntries();
+ else lastEvent = fLastEvent;
+
+ char fileName[256];
+ sprintf(fileName, "Run%d.Event%d_%d.ESD.tag.root",
+ iInitRunNumber,fFirstEvent,lastEvent);
+ AliInfo(Form("writing tags to file %s", fileName));
+ AliDebug(1, Form("writing tags to file %s", fileName));
+
+ TFile* ftag = TFile::Open(fileName, "recreate");
+
+ AliRunTag *tag = new AliRunTag();
+ AliEventTag *evTag = new AliEventTag();
+ TTree * ttag = new TTree("T","A Tree with event tags");
+ TBranch * btag = ttag->Branch("AliTAG", &tag);
+ btag->SetCompressionLevel(9);
+
+ if ((fLastEvent != -1) && ((Int_t) b->GetEntries() > fLastEvent))
+ iNumberOfEvents = fLastEvent + 1;
for (Int_t iEventNumber = fFirstEvent; iEventNumber < iNumberOfEvents; iEventNumber++) {
ntrack = 0;
nPos = 0;
nKaons = 0;
nPions = 0;
nMuons = 0;
+ nFWMuons = 0;
+ nFWMatchedMuons = 0;
nElectrons = 0;
nCh1GeV = 0;
nCh3GeV = 0;
nEl3GeV = 0;
nEl10GeV = 0;
maxPt = .0;
+ etamaxPt = -999.;
+ phimaxPt = -999.;
meanPt = .0;
totalP = .0;
fVertexflag = 0;
if ((status&AliESDtrack::kESDpid)==0) continue;
Double_t p[3];
esdTrack->GetPxPyPz(p);
- Double_t momentum = sqrt(pow(p[0],2) + pow(p[1],2) + pow(p[2],2));
- Double_t fPt = sqrt(pow(p[0],2) + pow(p[1],2));
+ Double_t pt2 = p[0]*p[0]+p[1]*p[1];
+ Double_t momentum = TMath::Sqrt(pt2+p[2]*p[2]);
+ Double_t fPt = TMath::Sqrt(pt2);
totalP += momentum;
meanPt += fPt;
- if(fPt > maxPt) maxPt = fPt;
+ if(fPt > maxPt) {
+ maxPt = fPt;
+ phimaxPt = esdTrack->Eta();
+ etamaxPt = esdTrack->Phi();
+ }
if(esdTrack->GetSign() > 0) {
nPos++;
fEnergy = TMath::Sqrt(fMUONMASS * fMUONMASS + fPxRec * fPxRec + fPyRec * fPyRec + fPzRec * fPzRec);
fEPvector.SetPxPyPzE(fPxRec, fPyRec, fPzRec, fEnergy);
- // total number of muons inside a vertex cut
- if((TMath::Abs(fZ)<fZVertexCut) && (TMath::Sqrt(fY*fY+fX*fX)<fRhoVertexCut)) {
- nMuons++;
- if(fEPvector.Pt() > fLowPtCut) {
- nMu1GeV++;
- if(fEPvector.Pt() > fHighPtCut) {
- nMu3GeV++;
- if (fEPvector.Pt() > fVeryHighPtCut) {
- nMu10GeV++;
- }
- }
- }
+ if (muonTrack->GetMatchTrigger()>0) nFWMatchedMuons++;
+
+ nMuons++;
+ nFWMuons++;
+ if(fEPvector.Pt() > fLowPtCut) {
+ nMu1GeV++;
+ if(fEPvector.Pt() > fHighPtCut) {
+ nMu3GeV++;
+ if (fEPvector.Pt() > fVeryHighPtCut) {
+ nMu10GeV++;
+ }
+ }
}
}//muon track loop
+
// Fill the event tags
if(ntrack != 0)
meanPt = meanPt/ntrack;
-
- evTag->SetEventId(iEventNumber+1);
+
+ //First physics data
+ const AliMultiplicity *spdMult = esd->GetMultiplicity();
+ evTag->SetNumberOfFiredChipsLayer1(spdMult->GetNumberOfFiredChips(0));
+ evTag->SetNumberOfFiredChipsLayer2(spdMult->GetNumberOfFiredChips(1));
+ evTag->SetNumberOfSPDTracklets(spdMult->GetNumberOfTracklets());
+
+ AliESDVZERO *vzeroData = esd->GetVZEROData();
+ evTag->SetMTotV0A(vzeroData->GetMTotV0A());
+ evTag->SetMTotV0C(vzeroData->GetMTotV0C());
+ evTag->SetNbPMV0A(vzeroData->GetNbPMV0A());
+ evTag->SetNbPMV0C(vzeroData->GetNbPMV0C());
+
+ //evTag->SetEventId(iEventNumber+1);
+ evTag->SetPeriodNumber(esd->GetPeriodNumber());
+ evTag->SetOrbitNumber(esd->GetOrbitNumber());
+ evTag->SetBunchCrossNumber(esd->GetBunchCrossNumber());
if (vertexIn) {
evTag->SetVertexX(vertexIn->GetXv());
evTag->SetVertexY(vertexIn->GetYv());
evTag->SetTriggerMask(esd->GetTriggerMask());
evTag->SetTriggerCluster(esd->GetTriggerCluster());
+ evTag->SetEventType(esd->GetEventType());
+ evTag->SetFiredTriggerClasses(esd->GetFiredTriggerClasses());
+
evTag->SetZDCNeutron1Energy(esd->GetZDCN1Energy());
evTag->SetZDCProton1Energy(esd->GetZDCP1Energy());
evTag->SetZDCNeutron2Energy(esd->GetZDCN2Energy());
evTag->SetZDCProton2Energy(esd->GetZDCP2Energy());
- evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy());
+ evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy(0),esd->GetZDCEMEnergy(1));
evTag->SetNumOfParticipants(esd->GetZDCParticipants());
evTag->SetNumOfKaons(nKaons);
evTag->SetNumOfPions(nPions);
evTag->SetNumOfMuons(nMuons);
+ evTag->SetNumOfFWMuons(nFWMuons);
+ evTag->SetNumOfFWMatchedMuons(nFWMatchedMuons);
evTag->SetNumOfElectrons(nElectrons);
evTag->SetNumOfPhotons(nGamas);
evTag->SetNumOfPi0s(nPi0s);
evTag->SetNumOfElectronsAbove3GeV(nEl3GeV);
evTag->SetNumOfElectronsAbove10GeV(nEl10GeV);
- evTag->SetNumOfPHOSClusters(esd->GetNumberOfPHOSClusters());
- evTag->SetNumOfEMCALClusters(esd->GetNumberOfEMCALClusters());
+ tmp.Clear();
+ evTag->SetNumOfPHOSClusters(esd->GetPHOSClusters(&tmp));
+ tmp.Clear();
+ evTag->SetNumOfEMCALClusters(esd->GetEMCALClusters(&tmp));
evTag->SetTotalMomentum(totalP);
evTag->SetMeanPt(meanPt);
evTag->SetMaxPt(maxPt);
+ evTag->SetEtaMaxPt(etamaxPt);
+ evTag->SetPhiMaxPt(phimaxPt);
tag->SetLHCTag(lhcLuminosity,lhcState);
tag->SetDetectorTag(detectorMask);
tag->SetRunId(iInitRunNumber);
+ tag->SetRunStartTime(t1.GetDate());
+ tag->SetRunStopTime(t2.GetDate());
+ tag->SetBeamEnergy(beamenergy);
+ tag->SetBeamType(beamtype);
+
+ //QA setting
+ tag->SetQAArray(qa, qalength) ;
+ tag->SetEventSpecies(es, eslength) ;
+
tag->AddEventTag(*evTag);
}
- if(fLastEvent == -1) lastEvent = (Int_t)b->GetEntries();
- else lastEvent = fLastEvent;
- char fileName[256];
- sprintf(fileName, "Run%d.Event%d_%d.ESD.tag.root",
- tag->GetRunId(),fFirstEvent,lastEvent );
- AliInfo(Form("writing tags to file %s", fileName));
- AliDebug(1, Form("writing tags to file %s", fileName));
-
- TFile* ftag = TFile::Open(fileName, "recreate");
ftag->cd();
- ttag.Fill();
+ ttag->Fill();
tag->Clear();
- ttag.Write();
+ ttag->Write();
ftag->Close();
file->cd();
+ delete file;
+ delete ftag;
+ delete esd;
delete tag;
delete evTag;
}
+//_____________________________________________________________________________
+void AliESDTagCreator::SwitchOffBranches() const {
+ //
+ // Switch of branches on user request
+ TObjArray * tokens = fBranches.Tokenize(" ");
+ Int_t ntok = tokens->GetEntries();
+ for (Int_t i = 0; i < ntok; i++) {
+ TString str = ((TObjString*) tokens->At(i))->GetString();
+ fChain->SetBranchStatus(Form("%s%s%s","*", str.Data(), "*"), 0);
+ AliInfo(Form("Branch %s switched off \n", str.Data()));
+ }
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff