1 //#ifndef ALIANALYSISTASKMUONAODCREATION_CXX
2 //#define ALIANALYSISTASKMUONAODCREATION_CXX
8 #include <TClonesArray.h>
11 #include "AliAnalysisTaskMuonAODCreation.h"
12 #include "AliAnalysisTaskSE.h"
13 #include "AliAnalysisManager.h"
14 #include "AliAnalysisDataSlot.h"
15 #include "AliESDEvent.h"
17 #include "AliAODEvent.h"
18 #include "AliAODHeader.h"
19 #include "AliVEvent.h"
20 #include "AliAODHandler.h"
21 #include "AliAODInputHandler.h"
22 #include "AliMCEventHandler.h"
23 #include "AliInputEventHandler.h"
24 #include "AliMCEvent.h"
28 ClassImp(AliAnalysisTaskMuonAODCreation)
30 //__________________________________________________________________________
31 AliAnalysisTaskMuonAODCreation::AliAnalysisTaskMuonAODCreation() :
37 //___________________________________________________________________________
38 AliAnalysisTaskMuonAODCreation::AliAnalysisTaskMuonAODCreation(const Char_t* name) :
39 AliAnalysisTaskSE(name),
44 // Constructor. Initialization of Inputs and Outputs
47 DefineOutput(1,TList::Class());
51 //___________________________________________________________________________
52 AliAnalysisTaskMuonAODCreation& AliAnalysisTaskMuonAODCreation::operator=(const AliAnalysisTaskMuonAODCreation& c)
55 // Assignment operator
58 AliAnalysisTaskSE::operator=(c) ;
63 //___________________________________________________________________________
64 AliAnalysisTaskMuonAODCreation::AliAnalysisTaskMuonAODCreation(const AliAnalysisTaskMuonAODCreation& c) :
68 fOutputAOD(c.fOutputAOD)
75 //___________________________________________________________________________
76 AliAnalysisTaskMuonAODCreation::~AliAnalysisTaskMuonAODCreation() {
80 Info("~AliAnalysisTaskMuonAODCreation","Calling Destructor");
83 //___________________________________________________________________________
84 void AliAnalysisTaskMuonAODCreation::UserCreateOutputObjects(){
86 AliAODHandler* handler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
88 fOutputAOD = handler->GetAOD();
89 fTree = handler->GetTree();
91 fOutput = new TList();
94 TH1D *pt_alltracks = new TH1D("pt_alltracks","pt_alltracks",10,0,20);
95 TH1D *pt_muontracks = new TH1D("pt_muontracks","pt_muontracks",10,0,20);
97 fOutput->Add(pt_alltracks);
98 fOutput->Add(pt_muontracks);
104 //_________________________________________________
105 void AliAnalysisTaskMuonAODCreation::UserExec(Option_t *)
108 AliAODEvent *aod = dynamic_cast<AliAODEvent*> (InputEvent());
112 for (Int_t j = 0; j<aod->GetNumberOfTracks(); j++) {
113 AliAODTrack *track = aod->GetTrack(j);
114 ((TH1D*)(fOutput->FindObject("pt_alltracks")))->Fill(track->Pt());
115 if(track->IsMuonTrack()) {
117 ((TH1D*)(fOutput->FindObject("pt_muontracks")))->Fill(track->Pt());
121 AliAODHandler* outputHandler = (AliAODHandler*)
122 ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
124 outputHandler->SetFillAOD(kFALSE);
127 outputHandler->SetFillAOD(kTRUE);
137 //________________________________________________________________________
138 void AliAnalysisTaskMuonAODCreation::Terminate(Option_t *)
140 TCanvas *c = new TCanvas("c","plots",20,20,600,600);
143 TH1D *h_pt_all = dynamic_cast<TH1D*> (fOutput->FindObject("pt_alltracks"));
144 TH1D *h_pt_muons = dynamic_cast<TH1D*> (fOutput->FindObject("pt_muontracks"));
151 h_pt_muons->Draw("same");