FillReco.h

Go to the documentation of this file.

#ifndef CAF_FILLRECO_H
#define CAF_FILLRECO_H

#include "StandardRecord/StandardRecord.h"
#include "FillTruth.h"

namespace rb{class Cluster; class Prong; class FitSum; class Shower; class Track; class Energy;}
namespace me{class TrkME; class SlcME;}
namespace bpfit{class BPFPId;}
namespace cosrej{class TrkCntObj;}
namespace ndreco{class NDRecoTrkObj;}
namespace ndreco{class NDRecoBpfObj;}
namespace ndreco{class NDRecoPngObj;}
namespace trackinfo{class TrackInfoObj;}

namespace caf
{
  void FillSliceVars(const rb::Cluster& slice,
                     caf::SRSlice& srslice,
                     bool allowEmpty = false);

  void AddTrkMEToVec(const me::TrkME& michel,
                     std::vector<SRTrkME>* vec,
                     SRTrack& srTrk,
                     const std::vector<rb::Cluster>& sliceList,
                     std::vector<cheat::TrackIDE>& allTracks,
                     std::vector<cheat::TrackIDE>& sliceTracks,
                     std::vector<cheat::TrackIDE>& allTracksBirks,
                     std::vector<cheat::TrackIDE>& sliceTracksBirks,
                     int sliceIdx);

  void AddSlcMEToVec(const me::SlcME& michel,
                     std::vector<SRSlcME>* vec,
                     const std::vector<rb::Cluster>& sliceList,
                     std::vector<cheat::TrackIDE>& allTracks,
                     std::vector<cheat::TrackIDE>& sliceTracks,
                     std::vector<cheat::TrackIDE>& allTracksBirks,
                     std::vector<cheat::TrackIDE>& sliceTracksBirks,
                     int sliceIdx);

  /// Helper for FillDiFVars and FillDiFShowerVars
  bool IsTrueDiF(const rb::Cluster& slice);

  void FillDiFVars(const rb::Cluster& slice,
                   const std::vector<rb::Cluster>& sliceList,
                   std::vector<cheat::TrackIDE>& allTracks,
                   std::vector<cheat::TrackIDE>& sliceTracks,
                   std::vector<cheat::TrackIDE>& allTracksBirks,
                   std::vector<cheat::TrackIDE>& sliceTracksBirks,
                   int sliceIdx,
                   SRMRProperties& srMR);

  void FillTrackVars(const rb::Track& trk,
                     SRTrack& srTrk,
                     const std::vector<rb::Cluster>& sliceList,
                     std::vector<cheat::TrackIDE>& allTracks,
                     std::vector<cheat::TrackIDE>& sliceTracks,
                     std::vector<cheat::TrackIDE>& allTracksBirks,
                     std::vector<cheat::TrackIDE>& sliceTracksBirks,
                     int sliceIdx);

  /// Function to fill SRTrack branch if overlapping energy object is present
  void FillTrackVarsWithOverlapE(const rb::Track& trk,
                                 const rb::Energy& energy,
                                 SRTrack& srTrk,
                                 const std::vector<rb::Cluster>& sliceList,
                                 std::vector<cheat::TrackIDE>& allTracks,
                                 std::vector<cheat::TrackIDE>& sliceTracks,
                                 std::vector<cheat::TrackIDE>& allTracksBirks,
                                 std::vector<cheat::TrackIDE>& sliceTracksBirks,
                                 int sliceIdx);

  /// Function to add rb::FitSum information to SRBpfTrack (inherits from
  /// SRTrack)
  void FillTrackVarsBpfFitSum(const rb::FitSum &fitsum,
                              SRBpfTrack& srBpfTrk);

  /// Function to add bpf::BPFPId information to SRBpfTrack (inherits from
  /// SRTrack)
  void FillTrackVarsBpfPid(const bpfit::BPFPId &bpfpid,
                           SRBpfTrack& srBpfTrk);

  /// Acts on any object that has a superset of the same fields as the old
  /// SRProng
  template<class T> void FillProngVars(const rb::Prong &trk,
                                       T& srPrng,
                                       const std::vector<rb::Cluster>& sliceList,
                                       std::vector<cheat::TrackIDE>& allTracks,
                                       std::vector<cheat::TrackIDE>& sliceTracks,
                                       std::vector<cheat::TrackIDE>& allTracksBirks,
                                       std::vector<cheat::TrackIDE>& sliceTracksBirks,
                                       int sliceIdx);

  void FillShowerVars(const rb::Shower& shw,
                      caf::SRFuzzyKProng& srshw,
                      const std::vector<rb::Cluster>& sliceList,
                      std::vector<cheat::TrackIDE>& allTracks,
                      std::vector<cheat::TrackIDE>& sliceTracks,
                      std::vector<cheat::TrackIDE>& allTracksBirks,
                      std::vector<cheat::TrackIDE>& sliceTracksBirks,
                      int sliceIdx);

  void FillDiFShowerVars(const rb::Shower& shw,
                         const std::vector<rb::Cluster>& sliceList,
                         std::vector<cheat::TrackIDE>& allTracks,
                         std::vector<cheat::TrackIDE>& sliceTracks,
                         std::vector<cheat::TrackIDE>& allTracksBirks,
                         std::vector<cheat::TrackIDE>& sliceTracksBirks,
                         int sliceIdx, SRMRProperties& srMR);

  /// Function to fill SRTrack containment information
  void FillTrackContainmentVars(const cosrej::TrkCntObj&, SRTrack&);

  /// Functions to fill SRKalmanTrack pion energy reconstruction information
  void FillNDRecoTrkVars(const ndreco::NDRecoTrkObj&, SRKalmanTrack&);
  void FillNDRecoBPFVars(const ndreco::NDRecoBpfObj&, SRBpfTrack&);
  void FillNDRecoPngVars(const ndreco::NDRecoPngObj&, SRFuzzyKProng&);
  
  /// Function to fill additional SRTrack information
  void FillTrackInfoVars(const trackinfo::TrackInfoObj&, SRTrack&);
}

#endif