MuonSelector.hpp

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// Updates: 25 Oct. 2005, 
//               - Supriya Jain (sjain@fnal.gov)
//                 Added options: 
//                            JetBranch, MinDeltaR, MaxDeltaR
//
//          20 Dec, 2005
//               - Mark Owen (markowen@fnal.gov)
//                 Added interface to Muon ID standard slection
//                 Added DCA calculation wrt beam spot
//                 Added possibility to correct pT wrt beam spot
//
//          8 Jan, 2006
//               - Mark Owen (markowen@fnal.gov)
//                 Moved pT correction to before function
//             
//
// ---------------------------------------------------------------------------
#ifndef CAFUTIL_MUONSELECTOR_HPP__
#define CAFUTIL_MUONSELECTOR_HPP__

#include "cafe/Stat.hpp"
#include "caf_util/TRefFinder.hpp"
#include "cafe/Event.hpp"
#include "cafe/SelectUserObjects.hpp"
#include "tmb_tree/TMBMuon.hpp"

namespace caf_util {
  
  class MuonSelector : public cafe::SelectUserObjects<TMBMuon>
  {
  public:
    MuonSelector(const char *name);
    ~MuonSelector() {}
    bool processEvent(cafe::Event &event);
    bool selectObject(const TMBMuon &muon);
  private:
    void before(cafe::Collection<TMBMuon>& from) ;    
    void after(cafe::Collection<TMBMuon>& accepted, cafe::Collection<TMBMuon>& rejected);
  private:
    enum Quality {Loose, Medium, Tight} ;

    cafe::StatPointer _stat ; //< pointer to the statistics collector class

    int _nmuons ; //< minimal number of muons to be selected
    int _nmuonsmax ; //< maximal number of muons to be selected
    int _nselected ; //< number of selected objects

    Quality _qual ; //< muon quality
    int _nseg ; //< minimal number of muon segments
    float _pTcut ; //< minimal muon pT
    float _eta ; //< the maximal absolute value of the detector eta
    float _chisq ; //< Chisquare from global matching (global muon fit) 
    float _dcaSignif ; //< maximal DCA significance for the central track
    float _dcaSMT; //< maximal DCA for tracks with SMT hits
    float _dcaNoSMT; //< maximal DCA for tracks with no SMT hits 
    int _nSMT; //< minimal number of SMT hits on the muon track
    float _etHalo ; //< maximal value on the calorimeter isolation
    float _etHaloMin ; //< minimal value on the calorimeter isolation
    float _etHaloScaled ; //< maximal value on the scaled calorimeter isolation
    float _etHaloScaledMin ; //< minimal value on the scaled calorimeter isolation
    float _etTrkCone ; //< maximal value on the track isolation
    float _etTrkConeMin ; //< minimal value on the track isolation
    float _etTrkConeScaled ; //< maximal value on the scaled track isolation
    float _etTrkConeScaledMin ; //< minimal value on the scaled track isolation
    bool _useCentralMatched ; //< use only muons with central track matched
    bool _cosmicVeto ; //< use timing cuts to remove cosmic muons
    bool _sort; 
    bool _use_beam_spot; //< use beam spot to calculate DCA, if false use vertex
    bool _reject_no_beam_spot; //< reject runs with no beam spot entry
    bool _corr_pT_SMT; //< correct pT with respect to beam spot for tracks with SMT hits
    bool _corr_pT_noSMT; //< correct pT with respect to beam spot for tracks with no SMT hits
    bool _mc; //< is sample mc - if find it is then don't need beam spot file
    bool _excludeHole; //< do you want to exclude muon that are in the bottom hole ?

    // Parameters for DeltaR cuts
    cafe::Collection<TMBJet> _jets;
    float _MinDeltaR; //< min. delta_R from nearest jet
    float _MaxDeltaR; //< max. delta_R from nearest jet
    std::string _JetBranch; //< for computing delta_R

    cafe::Event* _event ;
    cafe::Variables _jetvars ; // variables for optimize jets reading
    TRefFinder _finder ; //< workaround for the TRef problem 

    // Functions to calculate DCA, either from vertex or beamspot
    //double CalcDCAwrtVertex(double &ip[2], double &iperr[3], const TMBTrack* mutrk);

    // Additions to use MUID standard cuts
    bool _use_muid; //< switch to use standard muon cuts
    bool _cuts_set; //< bool to check muon id cuts were set correctly
    std::string _muqual_file; //< name of file containing muon quality definions
    std::string _muqual_name; //< name of muon quality definition
    int _muqual_version; //< muon quality version number
    std::string _trkqual_file; //< name of file containing track quality definions
    std::string _trkqual_name; //< name of track quality definition
    int _trkqual_version; //< track quality version number
    std::string _isoqual_file; //< name of file containing isolation quality definions
    std::string _isoqual_name; //< name of isolation quality definition
    int _isoqual_version; //< isolation quality version number    
    std::string _path; //< path of the definitions.

    //Beam spot info
    FILE *_beamspot_file; //< file for beam spot info
    std::string _beam; //< name of beam spot file
    bool _open_beam; //< is beam spot file open?
    float _bx, _by, _bsx, _bsy;
    int _lastrun;
    cafe::Variables _runvar;
    bool GetBeamSpot(FILE *fp, int run, float &bx, float &by,  float &bsx, float &bsy);
    cafe::Variables _chunkvar;
    double DCAfromBeamSpot(const TMBTrack* muon_trk);

    //Get Corrected Pt
    double GetCorrectedPt(const TMBMuon& muon, double dca);

    //Function to set up all the cuts based on the standard muIDs
    bool SetStdCuts(std::string path, std::string mu_file_name, std::string mu_name, int mu_ver, std::string trk_file_name, std::string trk_name, int trk_ver, std::string iso_file_name, std::string iso_name, int iso_ver);


    void DisplayMuonQualityCuts();
    void DisplayMuonTrackQualityCuts();
    void DisplayMuonIsolationQualityCuts();

    //Debug switch for development
    int _debuglevel;

    ClassDef(MuonSelector, 0) ;
  };
  
}
#endif // CAFUTIL_MUONSELECTOR_HPP__

---