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 analysation of conversion particle - track correlations
  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 //  fCutAxis(),
  72   fHistograms(NULL)
  73 {
  74   //Constructor
  75   //SetUpAxes();
  76 }
  77 //________________________________________________________________________
  78 AliConversionTrackCuts::AliConversionTrackCuts(TString name, TString title = "title") :
  79   AliAnalysisCuts(name, title),
  80   fFlagsOn(0x0),
  81   fFlagsOff(0x0),
  82   fRejectKinkDaughters(kTRUE),
  83   fDCARfixed(kTRUE),
  84   fDCARptFormula(""),
  85   fDCARmax(1E20),
  86   fDCAZfixed(kTRUE),
  87   fDCAZptFormula(""),
  88   fDCAZmax(1E20),
  89   fDCAXYmax(1E20),
  90   fSPDminNClusters(0),
  91   fITSminNClusters(0),
  92   fITSmaxChi2(1E20),
  93   fTPCminNClusters(0),
  94   fTPCmaxChi2(1E20),
  95   fAODTestFilterBit(-1),
  96   fRequireTPCRefit(kFALSE),
  97   fhPhi(NULL),
  98   fhPt(NULL),
  99   fhPhiPt(NULL),
 100   fhdcaxyPt(NULL),
 101   fHistograms(NULL)
 102 {
 103   //Constructor
 104 //  SetUpAxes();
 105 }
 106
 107
 108 //________________________________________________________________________________
 109  AliConversionTrackCuts::~AliConversionTrackCuts() {
 110    ///destructor
 111    // if(fHistograms)
 112    //    delete fHistograms;
 113    // fHistograms = NULL;
 114
 115 }
 116
 117 TList * AliConversionTrackCuts::CreateHistograms() {
 118   //Create the histograms
 119   if(!fHistograms) fHistograms = new TList();
 120
 121   fHistograms->SetOwner(kTRUE);
 122   fHistograms->SetName("trackCuts");
 123
 124   fhPhi = new TH2F("phi", "phi", kNCuts+2, kPreCut -0.5, kNCuts + 0.5,
 125                                    128, 0, TMath::TwoPi());
 126   TAxis * xax = fhPhi->GetXaxis();
 127   for(Int_t i = 0; i < kNCuts; i++){
 128         xax->SetBinLabel(xax->FindFixBin(i), fgkCutNames[i]);
 129   }
 130   fHistograms->Add(fhPhi);
 131
 132
 133
 134   fhPt = new TH2F("pt", "pt", kNCuts+2, kPreCut -0.5, kNCuts + 0.5,
 135                                   100, 0., 100.);
 136   xax = fhPt->GetXaxis();
 137   for(Int_t i = 0; i < kNCuts; i++){
 138         xax->SetBinLabel(xax->FindFixBin(i), fgkCutNames[i]);
 139   }
 140   fHistograms->Add(fhPt);
 141
 142   fhPhiPt = new TH2F("phipt", "phipt", 100, 0, 100, 64, 0, TMath::TwoPi());
 143   fHistograms->Add(fhPhiPt);
 144
 145   fhdcaxyPt = new TH2F("dcaxypt", "dcaxypt", 100, 0, 100, 100, 0, 100);
 146   fHistograms->Add(fhdcaxyPt);
 147
 148   fhdcazPt = new TH2F("dcazpt", "dcazpt", 100, 0, 100, 100, 0, 100);
 149   fHistograms->Add(fhdcazPt);
 150
 151   fhnclpt = new TH2F("nclstpcvspt", "nclstpcvspt", 100, 0, 100, 100, 0, 100);
 152   fHistograms->Add(fhnclpt);
 153
 154   fhnclsfpt = new TH2F("nclsfpt", "nclsfpt", 100, 0, 100, 100, 0, 1.2);
 155   fHistograms->Add(fhnclsfpt);
 156
 157   return fHistograms;
 158 }
 159
 160
 161 void AliConversionTrackCuts::FillHistograms(Int_t cutIndex, AliVTrack * track, Bool_t passed = kFALSE) {
 162   //Fill histograms
 163   fhPhi->Fill(cutIndex, track->Phi());
 164   fhPt->Fill(cutIndex, track->Pt());
 165   if(passed) fhPhiPt->Fill(track->Pt(), track->Phi());
 166
 167 }
 168
 169 Bool_t AliConversionTrackCuts::AcceptTrack(AliAODTrack * track) {
 170   //Check aod track
 171
 172   fhdcaxyPt->Fill(track->Pt(), TMath::Sqrt(track->XAtDCA()*track->XAtDCA() + track->YAtDCA()*track->YAtDCA()));
 173   fhdcazPt->Fill(track->Pt(), TMath::Abs(track->ZAtDCA()));
 174
 175   fhnclpt->Fill(track->Pt(), track->GetTPCNcls());
 176   if(track->GetTPCNclsF() > 0) fhnclsfpt->Fill(track->Pt(), ((Double_t) track->GetTPCNcls())/track->GetTPCNclsF());
 177
 178
 179   FillHistograms(kPreCut, track);
 180
 181   if (track->GetTPCNcls() < fTPCminNClusters) return kFALSE;
 182   FillHistograms(kCutNcls, track);
 183
 184   Double_t foundclusters = 0.0001;
 185   if(track->GetTPCNclsF() > 0) foundclusters = ( (Double_t) track->GetTPCNcls() )/track->GetTPCNclsF();
 186   if (foundclusters < fTPCClusOverFindable) return kFALSE;
 187   FillHistograms(kCutNclsFrac, track);
 188
 189   if (track->Chi2perNDF() > fTPCmaxChi2) return kFALSE;
 190   FillHistograms(kCutNDF, track);
 191
 192   AliAODVertex *vertex = track->GetProdVertex();
 193   if (vertex && fRejectKinkDaughters) {
 194         if (vertex->GetType() == AliAODVertex::kKink) {
 195           return kFALSE;
 196         }
 197   }
 198   FillHistograms(kCutKinc, track);
 199
 200   if(TMath::Abs(track->ZAtDCA()) > fDCAZmax) {
 201         return kFALSE;
 202   }
 203   FillHistograms(kCutDCAZ, track);
 204
 205
 206   Float_t xatdca = track->XAtDCA();
 207   Float_t yatdca = track->YAtDCA();
 208   Float_t xy = xatdca*xatdca + yatdca*yatdca;
 209   if(xy > fDCAXYmax) {
 210         AliDebug(AliLog::kDebug + 2, "Kink daughter. Rejected");
 211         return kFALSE;
 212   }
 213   FillHistograms(kCutDCAXY, track);
 214
 215   ULong_t status = track->GetStatus();
 216   if (fRequireTPCRefit && (status&AliESDtrack::kTPCrefit) == 0) {
 217         AliDebug(AliLog::kDebug + 2, "Kink daughter. Rejected");
 218         return kFALSE;
 219   }
 220   FillHistograms(kCutTPCRefit, track);
 221
 222   FillHistograms(kNCuts, track, kTRUE);
 223   return kTRUE;
 224 }
 225
 226 // void AliConversionTrackCuts::SetUpAxes() {
 227 //   //
 228 //   fCutAxis.Set(kNCuts + 1, -1.5, kNCuts -0.5);
 229 //   fCutAxis.SetAxisLabel(-1, "track in");
 230 //   for (Int_t i = 0; i < kNCuts; i++) {
 231 //      fCutAxis.SetAxisLabel(i, fgkCutNames[i]);
 232 //   }
 233 //   for(Int_t i = 0, i < kNCuts; i++) {
 234 //   }
 235 // }
 236
 237
 238 //_________________________________________________________________________________________________
 239 void AliConversionTrackCuts::Print(const Option_t *) const
 240 {
 241 //
 242 // Print information on this cut
 243 //
 244
 245   printf("Cut name                : %s \n", GetName());
 246   printf("Kink daughters are      : %s \n", (fRejectKinkDaughters ? "rejected" : "accepted"));
 247   printf("TPC requirements        : clusters/findable %f, min. cluster = %d, max chi2 = %f, %s require refit\n", fTPCClusOverFindable, fTPCminNClusters, fTPCmaxChi2, (fRequireTPCRefit) ? "" : "Don't");
 248   printf("ITS requirements        : min. cluster = %d (all), %d (SPD), max chi2 = %f \n", fITSminNClusters, fSPDminNClusters, fITSmaxChi2);
 249   printf("DCA z cut               : fixed to %f cm \n", fDCAZmax);
 250   printf("DCA xy cut              : fixed to %f cm \n", fDCAXYmax);
 251
 252 }
 253
 254
 255
 256
 257
 258