Fixed bug in the computation of dip angle value
[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnCutDaughterType.cxx
1 //
2 // Class AliRsnCutDaughterType
3 //
4 // General implementation of a single cut strategy, which can be:
5 // - a value contained in a given interval  [--> IsBetween()   ]
6 // - a value equal to a given reference     [--> MatchesValue()]
7 //
8 // In all cases, the reference value(s) is (are) given as data members
9 // and each kind of cut requires a given value type (Int, UInt, Double),
10 // but the cut check procedure is then automatized and chosen thanks to
11 // an enumeration of the implemented cut types.
12 // At the end, the user (or any other point which uses this object) has
13 // to use the method IsSelected() to check if this cut has been passed.
14 //
15 // authors: Martin Vala (martin.vala@cern.ch)
16 //          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
17 //
18
19 #include "AliRsnDaughter.h"
20 #include "AliRsnCutDaughterType.h"
21
22 ClassImp(AliRsnCutDaughterType)
23
24 //_________________________________________________________________________________________________
25 AliRsnCutDaughterType::AliRsnCutDaughterType() :
26   AliRsnCut(),
27   fRefType(kTypes)
28 {
29 //
30 // Default constructor.
31 //
32
33   SetTargetType(AliRsnTarget::kDaughter);
34 }
35
36 //_________________________________________________________________________________________________
37 AliRsnCutDaughterType::AliRsnCutDaughterType
38 (const char *name, EType type) :
39   AliRsnCut(name, AliRsnCut::kDaughter, 0.0, 0.0),
40   fRefType(type)
41 {
42 //
43 // Main constructor.
44 //
45 }
46
47 //_________________________________________________________________________________________________
48 Bool_t AliRsnCutDaughterType::IsSelected(TObject *object)
49 {
50 //
51 // Cut checker.
52 //
53
54   // coherence check
55   if (!TargetOK(object)) return kFALSE;
56   
57   // check the daughter according to the selected type
58   // in some cases this means to retrieve the track status
59   AliRsnDaughter *daughter = dynamic_cast<AliRsnDaughter*>(object);
60   AliVTrack   *track  = dynamic_cast<AliVTrack*>(daughter->GetRef());
61   AliESDtrack *esdT   = dynamic_cast<AliESDtrack*>(daughter->GetRef());
62   ULong_t      status = 0x0;
63   if (track) status = (ULong_t)track->GetStatus();
64   
65   switch (fRefType)
66   {
67     case kTrackTPC:
68       return ((status & AliESDtrack::kTPCin)  != 0);
69     case kTrackITSSA:
70       if (esdT && track)
71       {
72         UChar_t itsCluMap = track->GetITSClusterMap();
73         Int_t   k, nITS   = 0;
74         for(k = 2; k < 6; k++) if(itsCluMap & (1 << k)) ++nITS;
75         if (nITS < 3) return kFALSE;
76       }
77       return ((status & AliESDtrack::kTPCin)  == 0 && (status & AliESDtrack::kITSrefit) != 0 && (status & AliESDtrack::kITSpureSA) == 0 && (status & AliESDtrack::kITSpid) != 0);
78     case kV0:
79       return daughter->IsV0();
80     default:
81       AliError("No good reference type is chosen. Cut skipped");
82       return kTRUE;
83   }
84 }