ana::muonid_classifier Namespace Reference

Enumerations
enum	EMuonIDReturnMode { kGetMuonID, kGetTrack }

Functions
float	GetMuonIDSR (const caf::SRProxy *sr, const EMuonIDReturnMode return_mode)
float	GetMuonIDKalmanTrack (const caf::SRKalmanTrack *kaltrk)
float	MuonID (const caf::SRProxy *sr)
float	BestTrack (const caf::SRProxy *sr)
float	InitTMVA ()
const Var	kMuonID (muonid_classifier::MuonID)
const Var	kBestMuonTrack (muonid_classifier::BestTrack)

Variables
float	TMVAvars [4]
TMVA::Reader *	readerBDT

Enumeration Type Documentation

enum ana::muonid_classifier::EMuonIDReturnMode

Enumerator
kGetMuonID
kGetTrack

Definition at line 18 of file MuonID.h.

{kGetMuonID, kGetTrack};

Function Documentation

float ana::muonid_classifier::BestTrack

(

const caf::SRProxy *

sr

)

Definition at line 79 of file MuonID.cxx.

References GetMuonIDSR(), and kGetTrack.

  {
    return ana::muonid_classifier::GetMuonIDSR(sr, ana::muonid_classifier::kGetTrack); // Check the MuonID and return highest scoring track
  }

float ana::muonid_classifier::GetMuonIDKalmanTrack

(

const caf::SRKalmanTrack *

kaltrk

)

Definition at line 51 of file MuonID.cxx.

References caf::SRTrack::avedEdxlast10cm, caf::SRTrack::avedEdxlast40cm, make_root_from_grid_output::dedxll, caf::SRKalmanTrack::dedxllh, InitTMVA(), std::min(), make_root_from_grid_output::scatll, and caf::SRKalmanTrack::scatllh.

  {
    if(!readerBDT)  InitTMVA();

    float scatll           = kaltrk->scatllh;
    float dedxll           = kaltrk->dedxllh;
    float avgdedxlast10cm  = -999.0;
    float avgdedxlast40cm  = -999.0;

    if (kaltrk->avedEdxlast10cm < 0 || kaltrk->avedEdxlast40cm < 0)
      return -99.0;

    avgdedxlast10cm = std::min(kaltrk->avedEdxlast10cm, float(30.));
    avgdedxlast40cm = std::min(kaltrk->avedEdxlast40cm, float(30.));

    TMVAvars[0]  = dedxll;
    TMVAvars[1]  = scatll;
    TMVAvars[2]  = avgdedxlast10cm;
    TMVAvars[3]  = avgdedxlast40cm;

    return readerBDT->EvaluateMVA("BDTG");
  }

float ana::muonid_classifier::GetMuonIDSR	(	const caf::SRProxy *	sr,
		const EMuonIDReturnMode	return_mode
	)

Definition at line 13 of file MuonID.cxx.

References make_root_from_grid_output::dedxll, InitTMVA(), caf::Proxy< caf::SRTrackBranch >::kalman, kGetTrack, std::min(), caf::Proxy< caf::SRKalman >::ntracks, make_root_from_grid_output::scatll, caf::Proxy< caf::SRKalman >::tracks, and caf::Proxy< caf::StandardRecord >::trk.

Referenced by BestTrack(), and MuonID().

  {
    if(!readerBDT)  InitTMVA();

    float best_track  = -999.0;
    float muonid     = -999.0;

    float dedxll = -999.0;
    float scatll = -999.0;
    float avgdedxlast10cm  = -999.0;
    float avgdedxlast40cm  = -999.0;

    for (unsigned int itrk = 0; itrk < sr->trk.kalman.ntracks; ++itrk){
      scatll    = sr->trk.kalman.tracks[itrk].scatllh;
      dedxll    = sr->trk.kalman.tracks[itrk].dedxllh;
      
      if (sr->trk.kalman.tracks[itrk].avedEdxlast10cm < 0 or sr->trk.kalman.tracks[itrk].avedEdxlast40cm < 0) continue;

      avgdedxlast10cm = std::min(sr->trk.kalman.tracks[itrk].avedEdxlast10cm, float(30.));
      avgdedxlast40cm = std::min(sr->trk.kalman.tracks[itrk].avedEdxlast40cm, float(30.));    
      
      TMVAvars[0]  = dedxll;
      TMVAvars[1]  = scatll;
      TMVAvars[2]  = avgdedxlast10cm;
      TMVAvars[3]  = avgdedxlast40cm;

      float newmuonid = readerBDT->EvaluateMVA("BDTG");
      if (newmuonid > muonid){
        best_track = itrk;
        muonid = newmuonid;
      }
    }
    // Check if we're returning the best track id, or the highest MuonID
    if (return_mode == EMuonIDReturnMode::kGetTrack)
      return best_track;
    return muonid;
  }

float ana::muonid_classifier::InitTMVA

(

)

Definition at line 85 of file MuonID.cxx.

References cet::getenv(), and string.

Referenced by GetMuonIDKalmanTrack(), and GetMuonIDSR().

  {
    // Initialize the static members
    readerBDT   = 0;
    TMVAvars[0] = -999;
    TMVAvars[1] = -999;
    TMVAvars[2] = -999;
    TMVAvars[3] = -999;

    if(!readerBDT) readerBDT = new TMVA::Reader();
    const std::string srt_public  = getenv("SRT_PUBLIC_CONTEXT");
    const std::string srt_private = getenv("SRT_PRIVATE_CONTEXT");
    const std::string weightfile_loc = "/NDAna/Classifiers/MuonID.weights.xml";

    // Default to SRT_PRIVATE_CONTEXT, then SRT_PUBLIC_CONTEXT
    std::string bdt_weight_path;
    if (!gSystem->AccessPathName((srt_private + weightfile_loc).c_str()))
      bdt_weight_path = srt_private + weightfile_loc;
    else
      bdt_weight_path = srt_public + weightfile_loc;
    
    readerBDT->AddVariable("DedxLL",                   &TMVAvars[0]);
    readerBDT->AddVariable("ScatLL",                   &TMVAvars[1]);
    readerBDT->AddVariable("Avededxlast10cm",          &TMVAvars[2]);
    readerBDT->AddVariable("Avededxlast40cm",          &TMVAvars[3]);
    readerBDT->BookMVA("BDTG", bdt_weight_path);
    return -5.0f;
  }

const Var ana::muonid_classifier::kBestMuonTrack

(

muonid_classifier::BestTrack

)

const Var ana::muonid_classifier::kMuonID

(

muonid_classifier::MuonID

)

Referenced by muonid_fhc_dist(), muonid_opt(), muonid_optimization(), and muonid_rhc_dist().

float ana::muonid_classifier::MuonID

(

const caf::SRProxy *

sr

)

Definition at line 74 of file MuonID.cxx.

References GetMuonIDSR().

Referenced by vf::TrackEva::analyze().

  {
    return ana::muonid_classifier::GetMuonIDSR(sr); // Check the MuonID and return it
  }

Variable Documentation

TMVA::Reader* ana::muonid_classifier::readerBDT

Definition at line 10 of file MuonID.cxx.

float ana::muonid_classifier::TMVAvars[4]

Definition at line 9 of file MuonID.cxx.