RhoGoodTrackSelector.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#ifndef RHOGOODTRACKSELECTOR_H
#define RHOGOODTRACKSELECTOR_H
//                                                                      //
// RhoGoodTrackSelector                                                 //
//                                                                      //
// Selector classes for track quality selections                        //
//                                                                      //
// Author List:                                                         //
// Marcel Kunze,  RUB, Feb. 99                                          //
// Copyright (C) 1999-2001, Ruhr-University Bochum.                     //
// Ralf Kliemt, HIM/GSI Feb.2013 (Cleanup & Restructuring)              //
// Klaus Goetzen, GSI, May. 2020 (Updated/modified criteria)            //
//                                                                      //

#include "RhoParticleSelectorBase.h"

class RhoGoodTrackSelector : public RhoParticleSelectorBase {

 public:
  // Default constructor installs the currently accepted general setting
  RhoGoodTrackSelector(const char *name = "RhoGoodTrackSelector", Double_t pmin=0.0, Double_t ptmin=0.0, Double_t fitprob=-1, Int_t mvdhits=0, Double_t xydoca = 100, Double_t zdoca = 100);
  RhoGoodTrackSelector(const char *name, criterion crit);
  RhoGoodTrackSelector(const char *name, const char *crit);
  // Destructor
  virtual ~RhoGoodTrackSelector() {};
  virtual void PrintOn(std::ostream &o = std::cout) const;

  //    void SetNHitsCut ( Int_t min, Int_t max=65535 ) {
  //      fNHitsMin = min;
  //      fNHitsMax = max;
  //      fCutHits=kTRUE;
  //    }
  void SetMomentumCut(Double_t pmin, Double_t pmax = 1.E8)
  {
    fPMin = pmin;
    fPMax = pmax;
  }
  
  void SetPtCut(Double_t ptmin, Double_t ptmax = 1.E8)
  {
    fPtMin = ptmin;
    fPtMax = ptmax;
  }
  
  void SetFitProbCut(Double_t probmin)
  {
    fFitProbMin = probmin;
  }

  void SetMvdHitsCut(Int_t hitsmin, Int_t hitsmax=100000)
  {
    fMvdHitsMin = hitsmin;
    fMvdHitsMax = hitsmax;
  }
  
  void SetPrimaryVertexCut(Double_t d, Double_t zmin, Double_t zmax)
  {
    fXYDocaMax = d;
    fZDocaMin = zmin;
    fZDocaMax = zmax;
  }
  
  // backward compatibility for PndGoodTrackSelector; values are ignored!
  void SetFitCut(Double_t chi = 1e9, Double_t chipm = 0.0)
  {
    fChisqProbMin = chipm;
    fChisqMax = chi;
  } 
  
  // operations
  virtual void SetCriterion(const char *crit);
  virtual void SetCriterion(criterion crit);
  virtual Bool_t Accept(RhoCandidate *b);
  virtual Bool_t Accept(PndPidCandidate *b);

 private:
  //    Int_t fNHitsMin;  // Number of hits: total
  //    Int_t fNHitsMax;  // Number of hits: total
  Double_t fPMin, fPMax, fPtMin, fPtMax; // Momentum cuts (abs. value)
  Double_t fFitProbMin;               // Fit probability
  Int_t    fMvdHitsMin, fMvdHitsMax;  // Mvd hits
  Double_t fXYDocaMax;                // Distance of closest approach
  Double_t fZDocaMin, fZDocaMax;      // Closest z approach to beam spot
  Double_t fChisqProbMin, fChisqMax;     // Fit chi-square probability 
  
 public:
  ClassDef(RhoGoodTrackSelector, 1) // Track selector
};

// standalone print
std::ostream &operator<<(std::ostream &o, const RhoGoodTrackSelector &);

#endif