]>
Commit | Line | Data |
---|---|---|
b2a297fa | 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 | /////////////////////////////////////////////////////////////////////////// | |
17 | // Cut class providing cuts for both legs in the AliDielectronPair // | |
18 | // // | |
19 | // // | |
20 | /* | |
21 | Add any number of leg cuts using e.g. for leg 1 | |
22 | GetFilterLeg1().AddCuts(mycut) | |
23 | where mycut has to inherit from AliAnalysisCuts | |
24 | ||
25 | */ | |
26 | // // | |
27 | /////////////////////////////////////////////////////////////////////////// | |
28 | ||
29 | #include <TList.h> | |
30 | ||
31 | #include "AliDielectronPair.h" | |
32 | #include "AliVParticle.h" | |
33 | ||
34 | #include "AliDielectronPairLegCuts.h" | |
35 | ||
36 | ClassImp(AliDielectronPairLegCuts) | |
37 | ||
38 | ||
39 | AliDielectronPairLegCuts::AliDielectronPairLegCuts() : | |
40 | AliAnalysisCuts(), | |
41 | fFilterLeg1("PairFilterLeg1","PairFilterLeg1"), | |
42 | fFilterLeg2("PairFilterLeg2","PairFilterLeg2"), | |
43 | fCutType(kBothLegs) | |
44 | { | |
45 | // | |
46 | // Default contructor | |
47 | // | |
48 | } | |
49 | ||
50 | //________________________________________________________________________ | |
51 | AliDielectronPairLegCuts::AliDielectronPairLegCuts(const char* name, const char* title) : | |
52 | AliAnalysisCuts(name,title), | |
53 | fFilterLeg1("PairFilterLeg1","PairFilterLeg1"), | |
54 | fFilterLeg2("PairFilterLeg2","PairFilterLeg2"), | |
55 | fCutType(kBothLegs) | |
56 | { | |
57 | // | |
58 | // Named contructor | |
59 | // | |
60 | } | |
61 | ||
62 | ||
63 | //________________________________________________________________________ | |
64 | Bool_t AliDielectronPairLegCuts::IsSelected(TObject* track) | |
65 | { | |
66 | // | |
67 | // check if cuts are fulfilled | |
68 | // | |
69 | ||
70 | //check if we have a AliDielectronPair | |
71 | AliDielectronPair *pair=dynamic_cast<AliDielectronPair*>(track); | |
72 | if (!pair) return kFALSE; | |
73 | ||
74 | //get both legs | |
54ce9dc4 | 75 | AliVParticle *leg1=pair->GetFirstDaughterP(); |
76 | AliVParticle *leg2=pair->GetSecondDaughterP(); | |
b2a297fa | 77 | |
78 | //mask used to require that all cuts are fulfilled | |
79 | UInt_t selectedMaskLeg1=(1<<fFilterLeg1.GetCuts()->GetEntries())-1; | |
80 | UInt_t selectedMaskLeg2=(1<<fFilterLeg2.GetCuts()->GetEntries())-1; | |
81 | ||
82 | //test cuts | |
83 | Bool_t isLeg1selected=(fFilterLeg1.IsSelected(leg1)==selectedMaskLeg1); | |
42c76bea | 84 | if(fCutType==kBothLegs && !isLeg1selected) { |
85 | SetSelected(isLeg1selected); | |
86 | return isLeg1selected; | |
87 | } | |
b2a297fa | 88 | Bool_t isLeg2selected=(fFilterLeg2.IsSelected(leg2)==selectedMaskLeg2); |
89 | ||
8df8e382 | 90 | Bool_t isLeg1selectedMirror=(fFilterLeg1.IsSelected(leg2)==selectedMaskLeg1); |
91 | Bool_t isLeg2selectedMirror=(fFilterLeg2.IsSelected(leg1)==selectedMaskLeg2); | |
92 | ||
b2a297fa | 93 | Bool_t isSelected=isLeg1selected&&isLeg2selected; |
94 | if (fCutType==kAnyLeg) | |
95 | isSelected=isLeg1selected||isLeg2selected; | |
8df8e382 | 96 | |
97 | if (fCutType==kMixLegs) | |
98 | isSelected=(isLeg1selected&&isLeg2selected)||(isLeg1selectedMirror&&isLeg2selectedMirror); | |
99 | ||
b2a297fa | 100 | SetSelected(isSelected); |
101 | return isSelected; | |
102 | } | |
103 | ||
104 | ||
105 |