X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=STEER%2FCreateAODfromESD.C;h=6ca0e34adc6e0b491895ae944a271484bf2c8099;hb=55739e9e30060ffc10a60a2ecdf96e7681303b2a;hp=4a84dad02d443126e017f7d829d2895a40731e7b;hpb=cf22b3fca15de52a52232b8bba84b3019a8a6283;p=u%2Fmrichter%2FAliRoot.git diff --git a/STEER/CreateAODfromESD.C b/STEER/CreateAODfromESD.C index 4a84dad02d4..6ca0e34adc6 100644 --- a/STEER/CreateAODfromESD.C +++ b/STEER/CreateAODfromESD.C @@ -1,698 +1,84 @@ -#include -#include -#include -#include - -#include "AliAODEvent.h" -#include "AliAODVertex.h" -#include "AliAODTrack.h" -#include "AliAODNeutral.h" - -#include "AliESD.h" -#include "AliESDtrack.h" -#include "AliESDVertex.h" -#include "AliESDv0.h" -#include "AliESDCascade.h" -#include "AliESDCaloCluster.h" +#if !defined(__CINT__) || defined(__MAKECINT__) +#include +#include +#include "AliAnalysisManager.h" +#include "AliESDInputHandler.h" +#include "AliAODHandler.h" +#include "AliAnalysisTaskESDfilter.h" +#include "AliAnalysisDataContainer.h" +#endif void CreateAODfromESD(const char *inFileName = "AliESDs.root", const char *outFileName = "AliAOD.root") { - - // create an AliAOD object - AliAODEvent *aod = new AliAODEvent(); - aod->CreateStdContent(); - - // open the file - TFile *outFile = TFile::Open(outFileName, "RECREATE"); - - // create the tree - TTree *aodTree = new TTree("AOD", "AliAOD tree"); - aodTree->Branch(aod->GetList()); - - // connect to ESD - TFile *inFile = TFile::Open(inFileName, "READ"); - TTree *t = (TTree*) inFile->Get("esdTree"); - TBranch *b = t->GetBranch("ESD"); - AliESD *esd = 0; - b->SetAddress(&esd); - - Int_t nEvents = b->GetEntries(); - - // loop over events and fill them - for (Int_t iEvent = 0; iEvent < nEvents; ++iEvent) { - b->GetEntry(iEvent); - - // Multiplicity information needed by the header (to be revised!) - Int_t nTracks = esd->GetNumberOfTracks(); - Int_t nPosTracks = 0; - for (Int_t iTrack=0; iTrackGetTrack(iTrack)->GetSign()> 0) nPosTracks++; - - // create the header - aod->AddHeader(new AliAODHeader(esd ->GetEventNumber(), - esd->GetRunNumber(), - nTracks, - nPosTracks, - nTracks-nPosTracks, - esd->GetMagneticField(), - -999., // centrality; to be filled, still - esd->GetTriggerMask(), - esd->GetTriggerCluster(), - esd->GetEventType())); - - Int_t nV0s = esd->GetNumberOfV0s(); - Int_t nCascades = esd->GetNumberOfCascades(); - Int_t nKinks = esd->GetNumberOfKinks(); - Int_t nVertices = nV0s + nCascades + nKinks; - - aod->ResetStd(nTracks, nVertices); - - - // Array to take into account the tracks already added to the AOD - Bool_t * usedTrack = NULL; - if (nTracks>0) { - usedTrack = new Bool_t[nTracks]; - for (Int_t iTrack=0; iTrack0) { - usedV0 = new Bool_t[nV0s]; - for (Int_t iV0=0; iV00) { - usedKink = new Bool_t[nKinks]; - for (Int_t iKink=0; iKinkGetVertices()); - Int_t jVertices=0; - - // Access to the AOD container of tracks - TClonesArray &tracks = *(aod->GetTracks()); - Int_t jTracks=0; - // Add primary vertex. The primary tracks will be defined - // after the loops on the composite objects (V0, cascades, kinks) - const AliESDVertex *vtx = esd->GetPrimaryVertex(); - - Double_t pos[3]; - vtx->GetXYZ(pos); // position - Double_t cov[6]; - vtx->GetCovMatrix(cov); //covariance matrix - - AliAODVertex * primary = new(vertices[jVertices++]) - AliAODVertex(pos, cov, vtx->GetChi2(), NULL, AliAODVertex::kPrimary); - - // Create vertices starting from the most complex objects - - // Cascades - for (Int_t nCascade = 0; nCascade < nCascades; ++nCascade) { - AliESDcascade *cascade = esd->GetCascade(nCascade); - - Double_t posXi[3]; - cascade->GetXYZ(posXi[0], posXi[1], posXi[2]); - Double_t covXi[6]; - cascade->GetPosCovXi(covXi); - - // Add the cascade vertex - AliAODVertex * vcascade = new(vertices[jVertices++]) AliAODVertex(posXi, - covXi, - cascade->GetChi2Xi(), - primary, - AliAODVertex::kCascade); - - primary->AddDaughter(vcascade); - - // Add the V0 from the cascade. The ESD class have to be optimized... - // Now we have to search for the corresponding Vo in the list of V0s - // using the indeces of the positive and negative tracks - - Int_t posFromV0 = cascade->GetPindex(); - Int_t negFromV0 = cascade->GetNindex(); - - - AliESDv0 * v0 = 0x0; - Int_t indV0 = -1; - - for (Int_t iV0=0; iV0GetV0(iV0); - Int_t pos = v0->GetPindex(); - Int_t neg = v0->GetNindex(); - - if (pos==posFromV0 && neg==negFromV0) { - indV0 = iV0; - break; - } - } - - AliAODVertex * vV0FromCascade = 0x0; - - if (indV0>-1 && !usedV0[indV0] ) { - - // the V0 exists in the array of V0s and is not used - - usedV0[indV0] = kTRUE; - - Double_t posV0[3]; - v0->GetXYZ(posV0[0], posV0[1], posV0[2]); - Double_t covV0[6]; - v0->GetPosCov(covV0); - - vV0FromCascade = new(vertices[jVertices++]) AliAODVertex(posV0, - covV0, - v0->GetChi2V0(), - vcascade, - AliAODVertex::kV0); - } else { - - // the V0 doesn't exist in the array of V0s or was used - cerr << "Error: event " << iEvent << " cascade " << nCascade - << " The V0 " << indV0 - << " doesn't exist in the array of V0s or was used!" << endl; - - Double_t posV0[3]; - cascade->GetXYZ(posV0[0], posV0[1], posV0[2]); - Double_t covV0[6]; - cascade->GetPosCov(covV0); - - vV0FromCascade = new(vertices[jVertices++]) AliAODVertex(posV0, - covV0, - v0->GetChi2V0(), - vcascade, - AliAODVertex::kV0); - vcascade->AddDaughter(vV0FromCascade); - } - - // Add the positive tracks from the V0 - - if (! usedTrack[posFromV0]) { - - usedTrack[posFromV0] = kTRUE; - - AliESDtrack *esdTrack = esd->GetTrack(posFromV0); - - Double_t p[3]; - esdTrack->GetPxPyPz(p); - - Double_t x[3]; - esdTrack->GetXYZ(x); - - Double_t cov[21]; - esdTrack->GetCovarianceXYZPxPyPz(cov); - - Double_t pid[10]; - esdTrack->GetESDpid(pid); - - vV0FromCascade->AddDaughter( - new(tracks[jTracks++]) AliAODTrack(esdTrack->GetID(), - esdTrack->GetLabel(), - p, kTRUE, - x, - kFALSE, - cov, - (Short_t)esdTrack->GetSign(), - esdTrack->GetITSClusterMap(), - pid, - vV0FromCascade, - AliAODTrack::kSecondary) - ); - } - else { - cerr << "Error: event " << iEvent << " cascade " << nCascade - << " track " << posFromV0 << " has already been used!" << endl; - } - - // Add the negative tracks from the V0 - - if (!usedTrack[negFromV0]) { - - usedTrack[negFromV0] = kTRUE; - - AliESDtrack *esdTrack = esd->GetTrack(negFromV0); - - Double_t p[3]; - esdTrack->GetPxPyPz(p); - - Double_t x[3]; - esdTrack->GetXYZ(x); - - Double_t cov[21]; - esdTrack->GetCovarianceXYZPxPyPz(cov); - - Double_t pid[10]; - esdTrack->GetESDpid(pid); - - vV0FromCascade->AddDaughter( - new(tracks[jTracks++]) AliAODTrack(esdTrack->GetID(), - esdTrack->GetLabel(), - p, - kTRUE, - x, - kFALSE, - cov, - (Short_t)esdTrack->GetSign(), - esdTrack->GetITSClusterMap(), - pid, - vV0FromCascade, - AliAODTrack::kSecondary) - ); - } - else { - cerr << "Error: event " << iEvent << " cascade " << nCascade - << " track " << negFromV0 << " has already been used!" << endl; - } - - // Add the bachelor track from the cascade - - Int_t bachelor = cascade->GetBindex(); - - if(!usedTrack[bachelor]) { - - usedTrack[bachelor] = kTRUE; - - AliESDtrack *esdTrack = esd->GetTrack(bachelor); - - Double_t p[3]; - esdTrack->GetPxPyPz(p); - - Double_t x[3]; - esdTrack->GetXYZ(x); - - Double_t cov[21]; - esdTrack->GetCovarianceXYZPxPyPz(cov); - - Double_t pid[10]; - esdTrack->GetESDpid(pid); - - vcascade->AddDaughter( - new(tracks[jTracks++]) AliAODTrack(esdTrack->GetID(), - esdTrack->GetLabel(), - p, - kTRUE, - x, - kFALSE, - cov, - (Short_t)esdTrack->GetSign(), - esdTrack->GetITSClusterMap(), - pid, - vcascade, - AliAODTrack::kSecondary) - ); - } - else { - cerr << "Error: event " << iEvent << " cascade " << nCascade - << " track " << bachelor << " has already been used!" << endl; - } - - // Add the primary track of the cascade (if any) - - } // end of the loop on cascades - - // V0s - - for (Int_t nV0 = 0; nV0 < nV0s; ++nV0) { - - if (usedV0[nV0]) continue; // skip if aready added to the AOD - - AliESDv0 *v0 = esd->GetV0(nV0); - - Double_t posV0[3]; - v0->GetXYZ(posV0[0], posV0[1], posV0[2]); - Double_t covV0[6]; - v0->GetPosCov(covV0); - - AliAODVertex * vV0 = - new(vertices[jVertices++]) AliAODVertex(posV0, - covV0, - v0->GetChi2V0(), - primary, - AliAODVertex::kV0); - primary->AddDaughter(vV0); - - Int_t posFromV0 = v0->GetPindex(); - Int_t negFromV0 = v0->GetNindex(); - - // Add the positive tracks from the V0 - - if (!usedTrack[posFromV0]) { - - usedTrack[posFromV0] = kTRUE; - - AliESDtrack *esdTrack = esd->GetTrack(posFromV0); - - Double_t p[3]; - esdTrack->GetPxPyPz(p); - - Double_t x[3]; - esdTrack->GetXYZ(x); - - Double_t cov[21]; - esdTrack->GetCovarianceXYZPxPyPz(cov); - - Double_t pid[10]; - esdTrack->GetESDpid(pid); - - vV0->AddDaughter( - new(tracks[jTracks++]) AliAODTrack(esdTrack->GetID(), - esdTrack->GetLabel(), - p, kTRUE, - x, - kFALSE, - cov, - (Short_t)esdTrack->GetSign(), - esdTrack->GetITSClusterMap(), - pid, - vV0, - AliAODTrack::kSecondary) - ); - } - else { - cerr << "Error: event " << iEvent << " V0 " << nV0 - << " track " << posFromV0 << " has already been used!" << endl; - } - - // Add the negative tracks from the V0 - - if (!usedTrack[negFromV0]) { - - usedTrack[negFromV0] = kTRUE; - - AliESDtrack *esdTrack = esd->GetTrack(negFromV0); - - Double_t p[3]; - esdTrack->GetPxPyPz(p); - - Double_t x[3]; - esdTrack->GetXYZ(x); - - Double_t cov[21]; - esdTrack->GetCovarianceXYZPxPyPz(cov); - - Double_t pid[10]; - esdTrack->GetESDpid(pid); - - vV0->AddDaughter( - new(tracks[jTracks++]) AliAODTrack(esdTrack->GetID(), - esdTrack->GetLabel(), - p, - kTRUE, - x, - kFALSE, - cov, - (Short_t)esdTrack->GetSign(), - esdTrack->GetITSClusterMap(), - pid, - vV0, - AliAODTrack::kSecondary) - ); - } - else { - cerr << "Error: event " << iEvent << " V0 " << nV0 - << " track " << negFromV0 << " has already been used!" << endl; - } - - } // end of the loop on V0s + gSystem->Load("libTree"); + gSystem->Load("libGeom"); + gSystem->Load("libPhysics"); + gSystem->Load("libVMC"); + gSystem->Load("libSTEERBase"); + gSystem->Load("libESD"); + gSystem->Load("libAOD"); - // Kinks: it is a big mess the access to the information in the kinks - // The loop is on the tracks in order to find the mother and daugther of each kink - - - for (Int_t iTrack=0; iTrackGetTrack(iTrack); - - - Int_t ikink = track->GetKinkIndex(0); - - if (ikink) { - // Negative kink index: mother, positive: daughter - - // Search for the second track of the kink - - for (Int_t jTrack = iTrack+1; jTrackGetTrack(jTrack); - - Int_t jkink = track1->GetKinkIndex(0); - - if ( TMath::Abs(ikink)==TMath::Abs(jkink) ) { - - // The two tracks are from the same kink - - if (usedKink[TMath::Abs(ikink)-1]) continue; // skip used kinks - - Int_t imother = -1; - Int_t idaughter = -1; - - if (ikink<0 && jkink>0) { - - imother = iTrack; - idaughter = jTrack; - } - else if (ikink>0 && jkink<0) { - - imother = jTrack; - idaughter = iTrack; - } - else { - cerr << "Error: Wrong combination of kink indexes: " - << ikink << " " << jkink << endl; - continue; - } - - // Add the mother track - - AliAODTrack * mother = NULL; - - if (!usedTrack[imother]) { - - usedTrack[imother] = kTRUE; - - AliESDtrack *esdTrack = esd->GetTrack(imother); - - Double_t p[3]; - esdTrack->GetPxPyPz(p); - - Double_t x[3]; - esdTrack->GetXYZ(x); - - Double_t cov[21]; - esdTrack->GetCovarianceXYZPxPyPz(cov); - - Double_t pid[10]; - esdTrack->GetESDpid(pid); - - - mother = - new(tracks[jTracks++]) AliAODTrack(esdTrack->GetID(), - esdTrack->GetLabel(), - p, - kTRUE, - x, - kFALSE, - cov, - (Short_t)esdTrack->GetSign(), - esdTrack->GetITSClusterMap(), - pid, - primary, - AliAODTrack::kPrimary); - primary->AddDaughter(mother); - } - else { - cerr << "Error: event " << iEvent << " kink " << TMath::Abs(ikink)-1 - << " track " << imother << " has already been used!" << endl; - } - - // Add the kink vertex - AliESDkink * kink = esd->GetKink(TMath::Abs(ikink)-1); - - AliAODVertex * vkink = - new(vertices[jVertices++]) AliAODVertex(kink->GetPosition(), - NULL, - 0., - mother, - AliAODVertex::kKink); - // Add the daughter track - - AliAODTrack * daughter = NULL; - - if (!usedTrack[idaughter]) { - - usedTrack[idaughter] = kTRUE; - - AliESDtrack *esdTrack = esd->GetTrack(idaughter); - - Double_t p[3]; - esdTrack->GetPxPyPz(p); - - Double_t x[3]; - esdTrack->GetXYZ(x); - - Double_t cov[21]; - esdTrack->GetCovarianceXYZPxPyPz(cov); - - Double_t pid[10]; - esdTrack->GetESDpid(pid); - - - daughter = - new(tracks[jTracks++]) AliAODTrack(esdTrack->GetID(), - esdTrack->GetLabel(), - p, - kTRUE, - x, - kFALSE, - cov, - (Short_t)esdTrack->GetSign(), - esdTrack->GetITSClusterMap(), - pid, - vkink, - AliAODTrack::kPrimary); - vkink->AddDaughter(daughter); - } - else { - cerr << "Error: event " << iEvent << " kink " << TMath::Abs(ikink)-1 - << " track " << idaughter << " has already been used!" << endl; - } - - - } - } - - } - - } - + gSystem->Load("libANALYSIS"); + gSystem->Load("libANALYSISalice"); + gSystem->Load("libPWG3muon"); + + TChain *chain = new TChain("esdTree"); + // Steering input chain + chain->Add(inFileName); + AliAnalysisManager *mgr = new AliAnalysisManager("ESD to AOD", "Analysis Manager"); + + // Input + AliESDInputHandler* inpHandler = new AliESDInputHandler(); + mgr->SetInputEventHandler (inpHandler); + + // Output + AliAODHandler* aodHandler = new AliAODHandler(); + aodHandler->SetOutputFileName(outFileName); + mgr->SetOutputEventHandler(aodHandler); + + // Task + // Barrel Tracks + AliAnalysisTaskESDfilter *filter = new AliAnalysisTaskESDfilter("Filter"); + mgr->AddTask(filter); + // Muons + AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter"); + mgr->AddTask(esdmuonfilter); + + AliESDtrackCuts* esdTrackCutsL = new AliESDtrackCuts("AliESDtrackCuts", "Standard"); + esdTrackCutsL->SetMinNClustersTPC(50); + esdTrackCutsL->SetMaxChi2PerClusterTPC(3.5); + esdTrackCutsL->SetMaxCovDiagonalElements(2,2,0.5,0.5,2); + esdTrackCutsL->SetRequireTPCRefit(kTRUE); + esdTrackCutsL->SetMinNsigmaToVertex(3); + esdTrackCutsL->SetDCAToVertexXY(3.); + esdTrackCutsL->SetRequireSigmaToVertex(kTRUE); + esdTrackCutsL->SetAcceptKingDaughters(kFALSE); + + AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter"); + trackFilter->AddCuts(esdTrackCutsL); + + filter->SetTrackFilter(trackFilter); + + // Pipelining + AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("cchain", TChain::Class(), + AliAnalysisManager::kInputContainer); - // Tracks (primary and orphan) - - for (Int_t nTrack = 0; nTrack < nTracks; ++nTrack) { - - - if (usedTrack[nTrack]) continue; - - AliESDtrack *esdTrack = esd->GetTrack(nTrack); - - Double_t p[3]; - esdTrack->GetPxPyPz(p); - - Double_t x[3]; - esdTrack->GetXYZ(x); - - Double_t cov[21]; - esdTrack->GetCovarianceXYZPxPyPz(cov); - - Double_t pid[10]; - esdTrack->GetESDpid(pid); - - Float_t impactXY, impactZ; - - esdTrack->GetImpactParameters(impactXY,impactZ); - - if (impactXY<3) { - // track inside the beam pipe - - primary->AddDaughter( - new(tracks[jTracks++]) AliAODTrack(esdTrack->GetID(), - esdTrack->GetLabel(), - p, - kTRUE, - x, - kFALSE, - cov, - (Short_t)esdTrack->GetSign(), - esdTrack->GetITSClusterMap(), - pid, - primary, - AliAODTrack::kPrimary) - ); - } - else { - // outside the beam pipe: orphan track - new(tracks[jTracks++]) AliAODTrack(esdTrack->GetID(), - esdTrack->GetLabel(), - p, - kTRUE, - x, - kFALSE, - cov, - (Short_t)esdTrack->GetSign(), - esdTrack->GetITSClusterMap(), - pid, - NULL, - AliAODTrack::kOrphan); - } - } // end of loop on tracks - - - // Access to the AOD container of vertices - TClonesArray &clusters = *(aod->GetNeutrals()); - Int_t jClusters=0; - - // Clusters - Int_t nClusters = esd->GetNumberOfCaloClusters(); - - for (Int_t iClust=0; iClustGetCaloCluster(iClust); - - Int_t id = cluster->GetID(); - Int_t label = -1; - Float_t energy = cluster->GetClusterEnergy(); - Float_t x[3]; - cluster->GetGlobalPosition(x); - Float_t * covMatrix = NULL; - Float_t * pid = NULL; - AliAODVertex *prodVertex = primary; - AliAODTrack *primTrack = NULL; - Char_t ttype=AliAODNeutral::kUndef; - - if (cluster->IsPHOS()) ttype=AliAODNeutral::kPHOSCluster; - else if (cluster->IsEMCAL()) { - - if (cluster->GetClusterType() == AliESDCaloCluster::kPseudoCluster) - ttype = AliAODNeutral::kEMCALPseudoCluster; - else - ttype = AliAODNeutral::kEMCALClusterv1; - - } - - new(clusters[jClusters++]) AliAODNeutral(id, - label, - energy, - x, - covMatrix, - pid, - prodVertex, - primTrack, - ttype); - - } // end of loop on clusters - - - delete [] usedTrack; - delete [] usedV0; - delete [] usedKink; - - // fill the tree for this event - aodTree->Fill(); - } // end of event loop + AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("tree", TTree::Class(), + AliAnalysisManager::kOutputContainer, + "default"); - aodTree->GetUserInfo()->Add(aod); - // close ESD file - inFile->Close(); - - // write the tree to the specified file - outFile = aodTree->GetCurrentFile(); - outFile->cd(); - aodTree->Write(); - outFile->Close(); + mgr->ConnectInput (filter, 0, cinput1 ); + mgr->ConnectOutput(filter, 0, coutput1); + + mgr->ConnectInput (esdmuonfilter, 0, cinput1 ); + mgr->ConnectOutput(esdmuonfilter, 0, coutput1); + // + // Run the analysis + // + mgr->InitAnalysis(); + mgr->PrintStatus(); + mgr->StartAnalysis("local", chain); }