]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGHF/hfe/AliHFEpidITS.cxx
New task for Lc->V0bachelor + update on cut class (Annalisa)
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEpidITS.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, 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 // ITS PID class
17 // checks ITS PID based on ITS dE/dx truncated mean
18 //
19 // Authors: Matus Kalisky <matus.kalisky@cern.ch>
20 //          Markus Fasel <M.Fasel@gsi.de>
21 //
22 #include <TClass.h>
23 #include <TH2F.h>
24 #include <TList.h>
25 #include <TMath.h>
26 #include <TString.h>
27
28 //#include "AliAODTrack.h"
29 //#include "AliAODMCParticle.h"
30 #include "AliESDtrack.h"
31 #include "AliLog.h"
32 #include "AliPID.h"
33 #include "AliVParticle.h"
34
35 #include "AliHFEpidITS.h"
36
37 ClassImp(AliHFEpidITS)
38
39 //___________________________________________________________________
40 AliHFEpidITS::AliHFEpidITS(const Char_t *name):
41     AliHFEpidBase(name)
42 {
43   //
44   // Default constructor
45   //
46 }
47
48 //___________________________________________________________________
49 AliHFEpidITS::AliHFEpidITS(const AliHFEpidITS &ref):
50     AliHFEpidBase("")
51 {
52   //
53   // Copy constructor
54   //
55   ref.Copy(*this);
56 }
57
58 //___________________________________________________________________
59 AliHFEpidITS &AliHFEpidITS::operator=(const AliHFEpidITS &ref){
60   //
61   // Assignment operator
62   //
63   if(this != &ref) ref.Copy(*this);
64   return *this;
65 }
66
67 //___________________________________________________________________
68 AliHFEpidITS::~AliHFEpidITS(){
69   //
70   // Destructor
71   //
72 }
73
74 //___________________________________________________________________
75 void AliHFEpidITS::Copy(TObject &o) const {
76   //
77   // Copy function
78   // Provides a deep copy
79   //
80   AliHFEpidBase::Copy(o);
81 }
82
83 //___________________________________________________________________
84 Bool_t AliHFEpidITS::InitializePID(Int_t /*run*/){
85   //
86   // ITS PID initialization
87   //
88   return kTRUE;
89 }
90
91
92 //___________________________________________________________________
93 Int_t AliHFEpidITS::IsSelected(const AliHFEpidObject* /*track*/, AliHFEpidQAmanager* /*pidqa*/) const {
94   //
95   // Does PID decision for ITS
96   // 
97   return 11;  // @TODO: Implement ITS PID decision
98 }
99
100 //___________________________________________________________________
101 Double_t AliHFEpidITS::GetITSSignalV1(AliVParticle *vtrack){
102   //
103   // Calculate the ITS signal according to the mean charge of the clusters
104   //
105   if(!TString(vtrack->IsA()->GetName()).CompareTo("AliAODTrack")){
106     AliError("PID for AODs not implemented yet");
107     return 0.;
108   }
109   AliESDtrack *track = dynamic_cast<AliESDtrack *>(vtrack);
110   if(!track) return 0.;
111   Double_t signal = 0.;
112 #ifdef TRUNK
113   Double_t dedx[4];
114   track->GetITSdEdxSamples(dedx);
115   signal = TMath::Mean(4, dedx);
116 #else
117   signal = track->GetITSsignal();
118 #endif
119   Double_t p = track->GetTPCInnerParam() ? track->GetTPCInnerParam()->P() : track->P();
120   AliDebug(1, Form("Momentum: %f, ITS Signal: %f", p, signal));
121   return signal;
122 }
123
124 //___________________________________________________________________
125 Double_t AliHFEpidITS::GetITSSignalV2(AliVParticle *vtrack){
126   //
127   // Calculates the ITS signal. Truncated mean is used.
128   //
129   if(!TString(vtrack->IsA()->GetName()).CompareTo("AliAODTrack")){
130     AliError("PID for AODs not implemented yet");
131     return 0.;
132   }
133   AliESDtrack *track = dynamic_cast<AliESDtrack *>(vtrack);
134   if(!track) return 0.;
135   Double_t dedx[4], tmp[4];
136   Int_t indices[4];
137   track->GetITSdEdxSamples(tmp);
138   TMath::Sort(4, tmp, indices);
139   for(Int_t ien = 0; ien < 4; ien++) dedx[ien] = tmp[indices[ien]];
140   Double_t signal = TMath::Mean(3, dedx); 
141   Double_t p = track->GetTPCInnerParam() ? track->GetTPCInnerParam()->P() : track->P();
142   AliDebug(1, Form("Momentum: %f, ITS Signal: %f", p, signal));
143   return signal;
144 }
145