two track study as function of pt

[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnCutPIDNSigma.cxx

//
// Class AliRsnCutPIDNSigma
//
// General implementation of a single cut strategy, which can be:
// - a value contained in a given interval  [--> IsBetween()   ]
// - a value equal to a given reference     [--> MatchesValue()]
//
// In all cases, the reference value(s) is (are) given as data members
// and each kind of cut requires a given value type (Int, UInt, Double),
// but the cut check procedure is then automatized and chosen thanks to
// an enumeration of the implemented cut types.
// At the end, the user (or any other point which uses this object) has
// to use the method IsSelected() to check if this cut has been passed.
//
// authors: Martin Vala (martin.vala@cern.ch)
//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
//

#include "AliPIDResponse.h"
#include "AliAnalysisManager.h"
#include "AliInputEventHandler.h"
#include "AliMultiInputEventHandler.h"

#include "AliRsnCutPIDNSigma.h"

ClassImp(AliRsnCutPIDNSigma)

//_________________________________________________________________________________________________
AliRsnCutPIDNSigma::AliRsnCutPIDNSigma
(const char *name, AliPID::EParticleType species, EDetector det, Double_t nsigma) :
   AliRsnCut(name, AliRsnCut::kDaughter, -nsigma, nsigma),
   fSpecies(species),
   fDetector(det),
   fMomMin(0.0),
   fMomMax(1E20),
   fRejectUnmatched(kFALSE)
{
//
// Main constructor.
//
}

//_________________________________________________________________________________________________
AliRsnCutPIDNSigma::AliRsnCutPIDNSigma
(const AliRsnCutPIDNSigma& copy) :
   AliRsnCut(copy),
   fSpecies(copy.fSpecies),
   fDetector(copy.fDetector),
   fMomMin(copy.fMomMin),
   fMomMax(copy.fMomMax),
   fRejectUnmatched(copy.fRejectUnmatched)
{
//
// Copy constructor.
//
}

//_________________________________________________________________________________________________
AliRsnCutPIDNSigma& AliRsnCutPIDNSigma::operator=(const AliRsnCutPIDNSigma& copy)
{
//
// Assignment operator
//

   AliRsnCut::operator=(copy);

   fSpecies = copy.fSpecies;
   fDetector = copy.fDetector;
   fMomMin = copy.fMomMin;
   fMomMax = copy.fMomMax;
   fRejectUnmatched = copy.fRejectUnmatched;

   return (*this);
}

//_________________________________________________________________________________________________
Bool_t AliRsnCutPIDNSigma::IsSelected(TObject *object)
{
//
// Cut checker.
//

   // coherence check
   if (!TargetOK(object)) return kFALSE;
   
   // check initialization of PID object
   AliPIDResponse *pid = fEvent->GetPIDResponse();
   if (!pid) {
      AliFatal("NULL PID response");
      return kFALSE;
   }

   // get reference momentum, for range cut
   Double_t momentum = -1.0;
   if (!fDaughter->Ref2Vtrack()) {
      AliDebugClass(2, "Referenced daughter is not a track");
      return kFALSE;
   }
   if (fDetector == kTPC)
      momentum = fDaughter->Ref2Vtrack()->GetTPCmomentum();
   else
      momentum = fDaughter->GetRef()->P();
      
   // check momentum range, if required
   if (momentum < fMomMin || momentum > fMomMax) {
      AliDebugClass(2, Form("Track momentum = %.5f, outside allowed range [%.2f - %.2f]", momentum, fMomMin, fMomMax));
      return kFALSE;
   }
   
   // matching check, if required
   if (fRejectUnmatched) {
      switch (fDetector) {
         case kITS:
            if (!IsITS(fDaughter->Ref2Vtrack())) {
               AliDebug(3, "Rejecting track not matched in ITS");
               return kFALSE;
            }
            break;
         case kTPC:
            if (!IsTPC(fDaughter->Ref2Vtrack())) {
               AliDebug(3, "Rejecting track not matched in TPC");
               return kFALSE;
            }
            break;
         case kTOF:
            if (!IsTOF(fDaughter->Ref2Vtrack())) {
               AliDebug(3, "Rejecting track not matched in TOF");
               return kFALSE;
            }
            break;
         default:
            AliWarning("Required to reject unmatched tracks, but no detector defined. Track rejected");
            return kFALSE;
      }
   }
   
   // check PID
   // the number of sigmas is set as cut value, which is then checked
   // using the basic functions available in AliRsnCut
   switch (fDetector) {
      case kITS:
         fCutValueD = pid->NumberOfSigmasITS(fDaughter->GetRef(), fSpecies);
         break;
      case kTPC:
         fCutValueD = pid->NumberOfSigmasTPC(fDaughter->GetRef(), fSpecies);
         break;
      case kTOF:
         fCutValueD = pid->NumberOfSigmasTOF(fDaughter->GetRef(), fSpecies);
         break;
      default:
         return kFALSE;
   }
   
   return OkRangeD();
}

//_________________________________________________________________________________________________
void AliRsnCutPIDNSigma::Print(const Option_t *) const
{
//
// Print information on this cut
//

   Char_t mom[200], det[100], match[200];
      
   if (fRejectUnmatched)
      snprintf(match, 200, "Unmatched tracks are rejected");
   else
      snprintf(match, 200, "No check on track matching");
      
   switch (fDetector) {
      case kITS: snprintf(det, 3, "ITS"); break;
      case kTPC: snprintf(det, 3, "TPC"); break;
      case kTOF: snprintf(det, 3, "TOF"); break;
      default  : snprintf(det, 3, "undefined");
   }

   AliInfo(Form("Cut name          : %s", GetName()));
   AliInfo(Form("--> PID detector  : %s", det));
   AliInfo(Form("--> match criteria: %s", match));
   AliInfo(Form("--> momentum range: %s", mom));   
}