1 #define AliAnalysisTaskMuonAODfromGeneral_cxx
4 // Class implementation for the specific dimuon AOD generation from a general AOD
13 #include "AliAODEvent.h"
14 #include "AliAnalysisTaskMuonAODfromGeneral.h"
15 #include "AliAODHandler.h"
16 #include "AliAnalysisManager.h"
17 #include "AliAODEventInfo.h"
18 #include "AliAODDimuon.h"
20 ClassImp(AliAnalysisTaskMuonAODfromGeneral)
22 //________________________________________________________________________
23 AliAnalysisTaskMuonAODfromGeneral::AliAnalysisTaskMuonAODfromGeneral(const char *name, Double_t BeamEnergy):
24 AliAnalysisTask(name, "AnalysisTaskMuonAODfromGeneral"),
32 // Input slot #0 works with an Ntuple
33 SetBeamEnergy(BeamEnergy);
34 DefineInput(0, TChain::Class());
35 // Output slot #0 writes into a TTree container
36 DefineOutput(0, TTree::Class());
39 //________________________________________________________________________
40 void AliAnalysisTaskMuonAODfromGeneral::ConnectInputData(Option_t *) {
41 printf(" ConnectInputData %s\n", GetName());
42 // New connection to the AOD
43 fChain = (TChain*)GetInputData(0);
44 fOrgAOD = new AliAODEvent();
45 fOrgAOD->ReadFromTree(fChain); // This checks also if the branch address is already set
48 //________________________________________________________________________
49 void AliAnalysisTaskMuonAODfromGeneral::CreateOutputObjects() {
50 printf("Creating output objects\n");
51 if(!fNewAOD)fNewAOD=new AliAODEvent();
52 fNewAOD->CreateStdContent();
55 ft = new TTree("AOD","Muon AOD tree",0);
57 ft->Branch(fNewAOD->GetList());
61 //Add new Objects: MuonInfos & Dimuons
62 fInfos = new AliAODEventInfo();
63 fNewAOD->AddObject(fInfos);
64 const char *nameInfos = "MuonInfos";
65 fInfos->SetName(nameInfos);
66 ft->Branch(nameInfos, &fInfos);
67 fDimuons = new TClonesArray("AliAODDimuon",0);
68 fNewAOD->AddObject(fDimuons);
69 const char *nameDimuons = "Dimuons";
70 fDimuons->SetName(nameDimuons);
71 ft->Branch(nameDimuons, &fDimuons);
75 //________________________________________________________________________
76 void AliAnalysisTaskMuonAODfromGeneral::Exec(Option_t *) {
78 // Task making a Muon AOD
80 TChain *chain = (TChain*)GetInputData(0);
81 Long64_t ientry = chain->GetReadEntry();
86 Int_t nTracks=fOrgAOD->GetNumberOfTracks();
89 Int_t mutrNumb[10]; for (Int_t i=0; i<10; mutrNumb[i++]=0);
90 for (Int_t iTrack=0; iTrack<nTracks; iTrack++){
91 const Double_t *trackpid=fOrgAOD->GetTrack(iTrack)->PID();
92 if (trackpid[AliAODTrack::kMuon]==1.) {
93 mutrNumb[nMuTracks]=iTrack;
96 if (fOrgAOD->GetTrack(iTrack)->Charge()> 0) nPosTracks++;
99 Bool_t ExistMuon=kFALSE;
100 if (nMuTracks>0) ExistMuon=kTRUE;
102 //--------------------------------------------------------------------
103 // set arrays and pointers
107 // Access to the header
108 AliAODHeader *header = fNewAOD->GetHeader();
111 header->SetRunNumber (fOrgAOD->GetRunNumber() );
112 header->SetBunchCrossNumber(fOrgAOD->GetBunchCrossNumber());
113 header->SetOrbitNumber (fOrgAOD->GetOrbitNumber() );
114 header->SetPeriodNumber (fOrgAOD->GetPeriodNumber() );
115 header->SetTriggerMask (fOrgAOD->GetTriggerMask() );
116 header->SetTriggerCluster (fOrgAOD->GetTriggerCluster() );
117 header->SetEventType (fOrgAOD->GetEventType() );
118 header->SetMagneticField (fOrgAOD->GetMagneticField() );
119 header->SetZDCN1Energy (fOrgAOD->GetZDCN1Energy() );
120 header->SetZDCP1Energy (fOrgAOD->GetZDCP1Energy() );
121 header->SetZDCN2Energy (fOrgAOD->GetZDCN2Energy() );
122 header->SetZDCP2Energy (fOrgAOD->GetZDCP2Energy() );
123 header->SetZDCEMEnergy (0,fOrgAOD->GetZDCEMEnergy(0) );
124 header->SetZDCEMEnergy (1,fOrgAOD->GetZDCEMEnergy(1) );
125 header->SetRefMultiplicity (nTracks);
126 header->SetRefMultiplicityPos(nPosTracks);
127 header->SetRefMultiplicityNeg(nTracks - nPosTracks);
128 header->SetMuonMagFieldScale(-999.); // FIXME
129 header->SetCentrality(-999.); // FIXME
132 const AliAODVertex *vtx = fOrgAOD->GetPrimaryVertex();
133 if(vtx)nVertices = 1;
135 Int_t nDimuons=nMuTracks*(nMuTracks-1)/2;
137 // Access to the AOD container of vertices
138 TClonesArray &vertices = *(fNewAOD->GetVertices());
141 // Access to the AOD container of tracks
142 TClonesArray &tracks = *(fNewAOD->GetTracks());
147 vtx->GetCovMatrix(covVtx); //covariance matrix
149 AliAODVertex * primary = new(vertices[jVertices++])
150 AliAODVertex(pos, covVtx, vtx->GetChi2perNDF(), NULL, AliAODVertex::kPrimary);
153 static int numtracks=0;
154 for (Int_t iTrack=0; iTrack<nMuTracks; iTrack++) {
155 AliAODTrack *aodTrack;
156 primary->AddDaughter( aodTrack=new(tracks[jTracks++])
157 AliAODTrack((*(fOrgAOD->GetTrack(mutrNumb[iTrack])))));
163 fDimuons->Expand(nDimuons);
164 TClonesArray &rDimuons = *fDimuons;
167 for(Int_t i=0; i<nMuTracks; i++){
168 for(Int_t j=i+1; j<nMuTracks; j++){
169 new(rDimuons[jDimuons++]) AliAODDimuon(tracks[i],tracks[j],fInfos);
173 fInfos->SetBeamEnergy(fBeamEnergy);
177 fInfos->tr=fNewAOD->GetTracks();
179 fInfos->fMUON_Single_LPt_L0=0;
180 fInfos->fMUON_Single_HPt_L0=1;
181 fInfos->fMUON_Like_LPt_L0=2;
182 fInfos->fMUON_Like_HPt_L0=3;
183 fInfos->fMUON_Unlike_LPt_L0=4;
184 fInfos->fMUON_Unlike_HPt_L0=5;
185 if(ExistMuon) ft->Fill();
188 // Post final data. It will be written to a file with option "RECREATE"
192 //________________________________________________________________________
193 void AliAnalysisTaskMuonAODfromGeneral::Terminate(Option_t *) {
194 // Draw some histogram at the end.
195 ft->GetUserInfo()->Add(fNewAOD);
196 ft->Write("",TObject::kOverwrite);
197 ft->ResetBranchAddresses();
198 if (!gROOT->IsBatch()) {
199 TCanvas *c1 = new TCanvas("c1","Pt",10,10,310,310);
200 c1->SetFillColor(10);
201 c1->SetHighLightColor(10);
203 c1->cd(1)->SetLeftMargin(0.15);
204 c1->cd(1)->SetBottomMargin(0.15);
205 c1->cd(1)->SetLogy();
206 ft->Draw("tracks.Pt()");
207 //ft->Draw("dimuons.M()");