- macro for filling histograms for tracking diagnostics (TPCTrackHisto component)
[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskFilterFriend.cxx
CommitLineData
6d3a7bbf 1/**************************************************************************
2 * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15
16/*$Id$*/
17
18/////////////////////////////////////////////////////////////
19//
20// Test task
21//
22// /////////////////////////////////////////////////////////////
23
24#include <TTree.h>
25#include <TChain.h>
26
27#include "AliLog.h"
28#include "AliESDInputHandler.h"
29#include "AliESDtrack.h"
30#include "AliESDEvent.h"
31#include "AliESDfriend.h"
32#include "AliAnalysisTaskFilter.h"
33#include "AliAnalysisManager.h"
34#include "AliAnalysisTaskFilterFriend.h"
35
36
37ClassImp(AliAnalysisTaskFilterFriend)
38
39
40//________________________________________________________________________
41AliAnalysisTaskFilterFriend::AliAnalysisTaskFilterFriend():
42AliAnalysisTaskFilter(),
43fESDInput(0),
44fESDfriendInput(0)
45{
46 // Constructor
47
48 // Define input and output slots here
49 // Input slot #0 works with a TChain
50 DefineInput(0, TChain::Class());
51 // Output slot #0 writes into a TTree
52 //DefineOutput(0,TTree::Class());
53}
54
55//________________________________________________________________________
56AliAnalysisTaskFilterFriend::AliAnalysisTaskFilterFriend(const char* name):
57AliAnalysisTaskFilter(name),
58fESDInput(0),
59fESDfriendInput(0)
60{
61 // Constructor
62
63 // Define input and output slots here
64 // Input slot #0 works with a TChain
65 DefineInput(0, TChain::Class());
66 // Output slot #0 writes into a TTree
67 //DefineOutput(0,TTree::Class());
68}
69
70//________________________________________________________________________
71AliAnalysisTaskFilterFriend::~AliAnalysisTaskFilterFriend()
72{
73
74 // dtor
75
76}
77
78//________________________________________________________________________
79void AliAnalysisTaskFilterFriend::Init()
80{
81 // Initialization
82
83 return;
84}
85
86//________________________________________________________________________
87void AliAnalysisTaskFilterFriend::UserCreateOutputObjects()
88{
89 //
90 // Create the output container
91 //
92
93 return;
94}
95
96//________________________________________________________________________
97void AliAnalysisTaskFilterFriend::UserExec(Option_t */*option*/)
98{
99
100 AliInfo("Filling Friends");
101
102 fESDInput = dynamic_cast<AliESDEvent*>(InputEvent()); // get the input ESD
103 fESDfriendInput = (AliESDfriend*)(fESDInput->FindListObject("AliESDfriend"));
104 if(!fESDInput) {
105 printf("AliAnalysisTaskFilterFriend::Exec(): no ESD \n");
106 return;
107 }
108 if(!fESDfriendInput) {
109 printf("AliAnalysisTaskFilterFriend::Exec(): no ESDfriend \n");
110 return;
111 }
112 // attach ESDfriend
6d3a7bbf 113
114 AliESDfriend* esdFriendOutput = (AliESDfriend*)ESDfriend();
250e5362 115 AliDebug(2,Form("Number of ESD tracks in input = %d ",fESDInput->GetNumberOfTracks()));
116 AliDebug(2,Form("Number of tracks in input friends = %d ",fESDfriendInput->GetNumberOfTracks()));
117 AliDebug(2,Form("Number of tracks in output friendsNew before filtering = %d ",esdFriendOutput->GetNumberOfTracks()));
6d3a7bbf 118
119 AliESDfriendTrack* tNull = new AliESDfriendTrack();
120
121 for (Int_t i = 0; i< fESDInput->GetNumberOfTracks(); i++){
122 if (i%2 ==0){
123 // keep friend
250e5362 124 AliDebug(2,Form("Keeping %d-th track",i));
6d3a7bbf 125 AliESDfriendTrack* tOld = (AliESDfriendTrack*)fESDfriendInput->GetTrack(i);
250e5362 126 AliDebug(3,Form("1P of the %d-th track = %f",i,tOld->Get1P()));
127 AliDebug(3,Form("MaxITScluster %d-th track = %d",i,tOld->GetMaxITScluster()));
6d3a7bbf 128 // tOld->Dump();
129 AddFriendTrackAt(tOld,i);
130
131 }
132 else {
133 //discard friend
134 AddFriendTrackAt(tNull,i);
135 }
136
137 }
250e5362 138 AliDebug(2,Form("Number of tracks in output friendsNew after filtering with GetEntries() = %d ",esdFriendOutput->GetEntriesInTracks()));
6d3a7bbf 139 return;
140}
141
142//________________________________________________________________________
143void AliAnalysisTaskFilterFriend::Terminate(Option_t */*option*/)
144{
145 // Terminate analysis
146 //
147 AliDebug(2,"AliAnalysisTaskFilterFriend: Terminate() \n");
148
149 return;
150}
151//________________________________________________________________________
152Bool_t AliAnalysisTaskFilterFriend::UserSelectESDfriendForCurrentEvent()
153{
154 //
155 // Selecting or discarding current event
156 //
157
158
159 fESDInput = dynamic_cast<AliESDEvent*>(InputEvent()); // get the input ESD
160 if ((fESDInput->GetNumberOfTracks())%2 == 0) {
250e5362 161 AliDebug(2,"Selecting event");
6d3a7bbf 162 return kTRUE;
163 }
250e5362 164 AliDebug(2,"Discarding event");
6d3a7bbf 165 return kFALSE;
166 /*
167 return kTRUE;
168 */
169}