//
// All cuts for single pions in Sigma* analysis 2010,
// based on track quality and particle identification
// with TPC and TOF.
// Author: Massimo Venaruzzo.
//
//

#include <Riostream.h>

#include "AliPID.h"
#include "AliPIDResponse.h"
#include "AliRsnCutDaughterSigmaStar2010PP.h"

ClassImp(AliRsnCutDaughterSigmaStar2010PP)

//__________________________________________________________________________________________________
AliRsnCutDaughterSigmaStar2010PP::AliRsnCutDaughterSigmaStar2010PP(const char *name, AliPID::EParticleType pid) :
   AliRsnCut(name, AliRsnTarget::kDaughter),
   fPID(pid),
   fCutQuality(Form("%sQuality", name)),
   fPIDCut(3.0)
{
//
// Constructor
// Initialize track quality cuts to 2010 defaults
//

   fCutQuality.SetPtRange(0.15, 1E+20);
   fCutQuality.SetEtaRange(-0.8, 0.8);
   fCutQuality.SetDCARmax(0.05);
   //fCutQuality.SetDCARPtFormula("0.0182+0.0350/pt^1.01");
   fCutQuality.SetDCAZmax(2.0);
   fCutQuality.SetSPDminNClusters(1);
   fCutQuality.SetITSminNClusters(0);
   fCutQuality.SetITSmaxChi2(1E+20);
   fCutQuality.SetTPCminNClusters(70);
   fCutQuality.SetTPCmaxChi2(4.0);
   fCutQuality.SetRejectKinkDaughters();
   fCutQuality.SetAODTestFilterBit(5);
}

//__________________________________________________________________________________________________
Bool_t AliRsnCutDaughterSigmaStar2010PP::IsSelected(TObject *obj)
{
//
// Global check
//

   // coherence check
   if (!TargetOK(obj)) return kFALSE;

   // check track
   AliVTrack *track = fDaughter->Ref2Vtrack();
   if (!track) {
      if (!fDaughter->GetRef()) AliWarning("NULL ref");
      return kFALSE;
   }

   AliDebugClass(2, "Checking status");


   // check flags
   if ((track->GetStatus() & AliESDtrack::kTPCin   ) == 0) return kFALSE;
   if ((track->GetStatus() & AliESDtrack::kTPCrefit) == 0) return kFALSE;
   if ((track->GetStatus() & AliESDtrack::kITSrefit) == 0) return kFALSE;

   AliDebugClass(2, "Checking quality cuts");

   // quality
   if (!fCutQuality.IsSelected(obj)) return kFALSE;

   AliDebugClass(2, "...passed");


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

   // check if TOF is matched
   // and computes all values used in the PID cut
   //Bool_t   isTOF  = MatchTOF(track);
   //Double_t pTPC   = track->GetTPCmomentum();
   //Double_t p      = track->P();
   Double_t nsTPC  = TMath::Abs(pid->NumberOfSigmasTPC(track, fPID));
   //Double_t nsTOF  = TMath::Abs(pid->NumberOfSigmasTOF(track, fPID));
   Double_t maxTPC = 1E20;
   //Double_t maxTOF = 1E20;

   // applies the cut differently depending on the PID and the momentum
   //if (isTOF) {
   // TPC: 5sigma cut for all
   //if (nsTPC > 5.0) return kFALSE;
   // TOF: 3sigma below 1.5 GeV, 2sigma above
   //if (p < 1.5) maxTOF = 3.0; else maxTOF = 2.0;
   //return (nsTOF <= maxTOF);
   //} else {

   //For the moment TOF is not used - PID ONLY WITH TPC - 3 sigmas in the whole range

   // TPC:
   // below 350 MeV: 5sigma
   // between 350 and 500 MeV: 3sigma
   // pions above 500 MeV: 2sigma
   // kaons between 500 and 700 MeV: 2sigma
   // kaons above 700 MeV: rejected
   /*if (pTPC <= 0.35)
      maxTPC = 5.0;
   else if (pTPC <= 0.5)
      maxTPC = 3.0;
   else if (pTPC > 0.5 && fPID == AliPID::kPion)
      maxTPC = 2.0;
   else if (pTPC > 0.5 && pTPC <= 0.7 && fPID == AliPID::kKaon)
      maxTPC = 2.0;
   else
      return kFALSE;*/
   //maxTPC = 3.0;
   maxTPC = fPIDCut;
   return (nsTPC <= maxTPC);

   AliDebugClass(2, "Good Pion (hallelujah)");
   return kTRUE;
}