PWGGA/GammaConv/AliConversionTrackCuts.cxx

   1 /**************************************************************************
   2  * This file is property of and copyright by the ALICE HLT Project        *
   3  * ALICE Experiment at CERN, All rights reserved.                         *
   4  *                                                                        *
   5  * Primary Author: Svein Lindal <slindal@fys.uio.no>                      *
   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 /// @file   AliConversionTrackCuts.cxx
  17 /// @author Svein Lindal
  18 /// @brief  Base class for aod track cuts. An adapted copy of Alirsntrackcuts
  19
  20
  21 #include "AliConversionTrackCuts.h"
  22 //#include "AliAODTrack.h"
  23 #include "AliAODEvent.h"
  24 #include <TFormula.h>
  25 #include <iostream>
  26 #include "TH2F.h"
  27
  28
  29 using namespace std;
  30 ClassImp(AliConversionTrackCuts)
  31
  32
  33 const char* AliConversionTrackCuts::fgkCutNames[AliConversionTrackCuts::kNCuts] = {
  34   "nClusTPC",
  35   "FoundFindable",
  36   "Chi2PerNDF",
  37   "Kink",
  38   "DCA_Z",
  39   "DCA_XY",
  40   "TPCRefit"
  41   "kAccTracks"
  42 };
  43
  44
  45
  46 //________________________________________________________________________
  47 AliConversionTrackCuts::AliConversionTrackCuts() :
  48   AliAnalysisCuts(),
  49   fFlagsOn(0x0),
  50   fFlagsOff(0x0),
  51   fRejectKinkDaughters(kTRUE),
  52   fDCARfixed(kTRUE),
  53   fDCARptFormula(""),
  54   fDCARmax(1E20),
  55   fDCAZfixed(kTRUE),
  56   fDCAZptFormula(""),
  57   fDCAZmax(1E20),
  58   fDCAXYmax(1E20),
  59   fSPDminNClusters(0),
  60   fITSminNClusters(0),
  61   fITSmaxChi2(1E20),
  62   fTPCminNClusters(0),
  63   fTPCClusOverFindable(0.0),
  64   fTPCmaxChi2(1E20),
  65   fAODTestFilterBit(-1),
  66   fRequireTPCRefit(kFALSE),
  67   fhPhi(NULL),
  68   fhPt(NULL),
  69   fhPhiPt(NULL),
  70   fhdcaxyPt(NULL),
  71   fhdcazPt(NULL),
  72   fhnclpt(NULL),
  73   fhnclsfpt(NULL),
  74   fHistograms(NULL)
  75 {
  76   //Constructor
  77 }
  78 //________________________________________________________________________
  79 AliConversionTrackCuts::AliConversionTrackCuts(TString name, TString title = "title") :
  80   AliAnalysisCuts(name, title),
  81   fFlagsOn(0x0),
  82   fFlagsOff(0x0),
  83   fRejectKinkDaughters(kTRUE),
  84   fDCARfixed(kTRUE),
  85   fDCARptFormula(""),
  86   fDCARmax(1E20),
  87   fDCAZfixed(kTRUE),
  88   fDCAZptFormula(""),
  89   fDCAZmax(1E20),
  90   fDCAXYmax(1E20),
  91   fSPDminNClusters(0),
  92   fITSminNClusters(0),
  93   fITSmaxChi2(1E20),
  94   fTPCminNClusters(0),
  95   fTPCClusOverFindable(0.0),
  96   fTPCmaxChi2(1E20),
  97   fAODTestFilterBit(-1),
  98   fRequireTPCRefit(kFALSE),
  99   fhPhi(NULL),
 100   fhPt(NULL),
 101   fhPhiPt(NULL),
 102   fhdcaxyPt(NULL),
 103   fhdcazPt(NULL),
 104   fhnclpt(NULL),
 105   fhnclsfpt(NULL),
 106   fHistograms(NULL)
 107 {
 108   //Constructor
 109 }
 110
 111
 112 //________________________________________________________________________________
 113  AliConversionTrackCuts::~AliConversionTrackCuts() {
 114    ///destructor
 115    // if(fHistograms)
 116    //    delete fHistograms;
 117    // fHistograms = NULL;
 118
 119 }
 120
 121 TList * AliConversionTrackCuts::CreateHistograms() {
 122   //Create the histograms
 123   if(!fHistograms) fHistograms = new TList();
 124
 125   fHistograms->SetOwner(kTRUE);
 126   fHistograms->SetName("trackCuts");
 127
 128   fhPhi = new TH2F("phi", "phi", kNCuts+2, kPreCut -0.5, kNCuts + 0.5,
 129                                    128, 0, TMath::TwoPi());
 130   TAxis * xax = fhPhi->GetXaxis();
 131   for(Int_t i = 0; i < kNCuts; i++){
 132         xax->SetBinLabel(xax->FindFixBin(i), fgkCutNames[i]);
 133   }
 134   fHistograms->Add(fhPhi);
 135
 136
 137
 138   fhPt = new TH2F("pt", "pt", kNCuts+2, kPreCut -0.5, kNCuts + 0.5,
 139                                   100, 0., 100.);
 140   xax = fhPt->GetXaxis();
 141   for(Int_t i = 0; i < kNCuts; i++){
 142         xax->SetBinLabel(xax->FindFixBin(i), fgkCutNames[i]);
 143   }
 144   fHistograms->Add(fhPt);
 145
 146   fhPhiPt = new TH2F("phipt", "phipt", 100, 0, 100, 64, 0, TMath::TwoPi());
 147   fHistograms->Add(fhPhiPt);
 148
 149   fhdcaxyPt = new TH2F("dcaxypt", "dcaxypt", 100, 0, 100, 100, 0, 100);
 150   fHistograms->Add(fhdcaxyPt);
 151
 152   fhdcazPt = new TH2F("dcazpt", "dcazpt", 100, 0, 100, 100, 0, 100);
 153   fHistograms->Add(fhdcazPt);
 154
 155   fhnclpt = new TH2F("nclstpcvspt", "nclstpcvspt", 100, 0, 100, 100, 0, 100);
 156   fHistograms->Add(fhnclpt);
 157
 158   fhnclsfpt = new TH2F("nclsfpt", "nclsfpt", 100, 0, 100, 100, 0, 1.2);
 159   fHistograms->Add(fhnclsfpt);
 160
 161   return fHistograms;
 162 }
 163
 164
 165 void AliConversionTrackCuts::FillHistograms(Int_t cutIndex, AliVTrack * track, Bool_t passed = kFALSE) {
 166   //Fill histograms
 167   fhPhi->Fill(cutIndex, track->Phi());
 168   fhPt->Fill(cutIndex, track->Pt());
 169   if(passed) fhPhiPt->Fill(track->Pt(), track->Phi());
 170
 171 }
 172
 173 Bool_t AliConversionTrackCuts::AcceptTrack(AliAODTrack * track) {
 174   //Check aod track
 175
 176   fhdcaxyPt->Fill(track->Pt(), TMath::Sqrt(track->XAtDCA()*track->XAtDCA() + track->YAtDCA()*track->YAtDCA()));
 177   fhdcazPt->Fill(track->Pt(), TMath::Abs(track->ZAtDCA()));
 178
 179   fhnclpt->Fill(track->Pt(), track->GetTPCNcls());
 180   if(track->GetTPCNclsF() > 0) fhnclsfpt->Fill(track->Pt(), ((Double_t) track->GetTPCNcls())/track->GetTPCNclsF());
 181
 182
 183   FillHistograms(kPreCut, track);
 184
 185   if (track->GetTPCNcls() < fTPCminNClusters) return kFALSE;
 186   FillHistograms(kCutNcls, track);
 187
 188   Double_t foundclusters = 0.0001;
 189   if(track->GetTPCNclsF() > 0) foundclusters = ( (Double_t) track->GetTPCNcls() )/track->GetTPCNclsF();
 190   if (foundclusters < fTPCClusOverFindable) return kFALSE;
 191   FillHistograms(kCutNclsFrac, track);
 192
 193   if (track->Chi2perNDF() > fTPCmaxChi2) return kFALSE;
 194   FillHistograms(kCutNDF, track);
 195
 196   AliAODVertex *vertex = track->GetProdVertex();
 197   if (vertex && fRejectKinkDaughters) {
 198         if (vertex->GetType() == AliAODVertex::kKink) {
 199           return kFALSE;
 200         }
 201   }
 202   FillHistograms(kCutKinc, track);
 203
 204   if(TMath::Abs(track->ZAtDCA()) > fDCAZmax) {
 205         return kFALSE;
 206   }
 207   FillHistograms(kCutDCAZ, track);
 208
 209
 210   Float_t xatdca = track->XAtDCA();
 211   Float_t yatdca = track->YAtDCA();
 212   Float_t xy = xatdca*xatdca + yatdca*yatdca;
 213   if(xy > fDCAXYmax) {
 214         AliDebug(AliLog::kDebug + 2, "Kink daughter. Rejected");
 215         return kFALSE;
 216   }
 217   FillHistograms(kCutDCAXY, track);
 218
 219   ULong_t status = track->GetStatus();
 220   if (fRequireTPCRefit && (status&AliESDtrack::kTPCrefit) == 0) {
 221         AliDebug(AliLog::kDebug + 2, "Kink daughter. Rejected");
 222         return kFALSE;
 223   }
 224   FillHistograms(kCutTPCRefit, track);
 225
 226   FillHistograms(kNCuts, track, kTRUE);
 227   return kTRUE;
 228 }
 229
 230 // void AliConversionTrackCuts::SetUpAxes() {
 231 //   //
 232 //   fCutAxis.Set(kNCuts + 1, -1.5, kNCuts -0.5);
 233 //   fCutAxis.SetAxisLabel(-1, "track in");
 234 //   for (Int_t i = 0; i < kNCuts; i++) {
 235 //      fCutAxis.SetAxisLabel(i, fgkCutNames[i]);
 236 //   }
 237 //   for(Int_t i = 0, i < kNCuts; i++) {
 238 //   }
 239 // }
 240
 241
 242 //_________________________________________________________________________________________________
 243 void AliConversionTrackCuts::Print(const Option_t *) const
 244 {
 245 //
 246 // Print information on this cut
 247 //
 248
 249   printf("Cut name                : %s \n", GetName());
 250   printf("Kink daughters are      : %s \n", (fRejectKinkDaughters ? "rejected" : "accepted"));
 251   printf("TPC requirements        : clusters/findable %f, min. cluster = %d, max chi2 = %f, %s require refit\n", fTPCClusOverFindable, fTPCminNClusters, fTPCmaxChi2, (fRequireTPCRefit) ? "" : "Don't");
 252   printf("ITS requirements        : min. cluster = %d (all), %d (SPD), max chi2 = %f \n", fITSminNClusters, fSPDminNClusters, fITSmaxChi2);
 253   printf("DCA z cut               : fixed to %f cm \n", fDCAZmax);
 254   printf("DCA xy cut              : fixed to %f cm \n", fDCAXYmax);
 255
 256 }
 257
 258
 259
 260
 261
 262