major dielectron update (included also the data and plotting macros for paper)
[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectronPairLegCuts.cxx
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
75   AliVParticle *leg1=pair->GetFirstDaughter();
76   AliVParticle *leg2=pair->GetSecondDaughter();
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);
84   Bool_t isLeg2selected=(fFilterLeg2.IsSelected(leg2)==selectedMaskLeg2);
85   
86   Bool_t isLeg1selectedMirror=(fFilterLeg1.IsSelected(leg2)==selectedMaskLeg1);
87   Bool_t isLeg2selectedMirror=(fFilterLeg2.IsSelected(leg1)==selectedMaskLeg2);
88   
89   Bool_t isSelected=isLeg1selected&&isLeg2selected;
90   if (fCutType==kAnyLeg)
91     isSelected=isLeg1selected||isLeg2selected;
92   
93   if (fCutType==kMixLegs)
94     isSelected=(isLeg1selected&&isLeg2selected)||(isLeg1selectedMirror&&isLeg2selectedMirror);
95   
96   SetSelected(isSelected);
97   return isSelected;
98 }
99
100
101