CVNProngVars.h

Go to the documentation of this file.

#pragma once

// Definitions for variables related prongs or choosing prongs with CVN score.

#include "CAFAna/Core/Var.h"
#include "CAFAna/Core/MultiVar.h"

namespace ana
{

  //=============================//
  // CVN-prong variables:
  //=============================//

  /// \brief:   Prong index of best muon prong by CVN score & length
  ///
  /// Prong index for the most muon like prong. Determined by highest CVN-prong score iff all prongs
  /// are <= 500 cm long and by longest prong if any prongs are > 500 cm.
  extern const Var kCVNMuonIdx;


  /// \brief:   Muon score for best muon prong by CVN score & length
  ///
  /// CVN-Prong score for most muon-like prong in the slice.
  /// NOTE:   This score assumes tracks longer than 500 cm to have a muon score of 1.0. This is necessary since
  ///         CVN-prong does NOT compute a score for prongs longe than 500 cm.
  extern const Var kCVNBestMuonScore;


  /// \brief:   Returns "adjusted" CVN prong muon score for a given prong. Here "adjusted" means that the
  ///           the score defaults to 1.0 if the prong is longer than 5.0 m. Not that in the extremely
  ///           rare event that a slice has more than one prong longer than 5.0 m, then all of those prongs
  ///           will get a score of 1.0 with this function.
  ///
  /// ProngIdx - index of prong to be considered
  /// sr       - the standard record
  double GetCVNProngMuonScore(unsigned int ProngIdx, caf::SRProxy *sr);


  /// \brief:   Raw muon score for best muon prong by CVN score & length
  ///
  /// Raw CVN-Prong score for most muon-like prong in the slice (unadjusted for prongs longer than 500 cm.)
  extern const Var kCVNBestMuonRawScore;



  //
  // Define functions for calculating various combo PIDs
  //
  double EMScore(unsigned int ProngIdx, const caf::SRProxy* sr);
  double HadScore(unsigned int ProngIdx, const caf::SRProxy* sr);
  double HadScoreWithMuon(unsigned int ProngIdx, const caf::SRProxy* sr);



  // ==================== //
  //
  // Helper functions and variables for choosing prongs based on CVN score:
  //
  // ==================== //

  bool IsNumuCCPionByCVN(unsigned int ProngIdx, const caf::SRProxy* sr);
  bool IsNumuCCProtonByCVN(unsigned int ProngIdx, const caf::SRProxy* sr);
  bool IsNumuCCEMShowerByCVN(unsigned int ProngIdx, const caf::SRProxy* sr);

  /// kNumuCCPionIndices
  /// \brief:   Returns a vector of indices for prongs selected as pions
  ///
  extern const MultiVar kNumuCCPionIndices;

  /// kNumuCCProtonIndices
  /// \brief:   Returns a vector of indices for prongs selected as protons
  ///
  extern const MultiVar kNumuCCProtonIndices;

  /// kNumuCCEMShowerIndices
  /// \brief:   Returns a vector of indices for prongs selected as EM showers
  ///
  extern const MultiVar kNumuCCEMShowerIndices;

} // end namespace ana