CheatTrack.h

Go to the documentation of this file.

#ifndef LAREXAMPLES_ALGORITHMS_TOTALLYCHEATTRACKS_CHEATTRACKDATA_CHEATTRACK_H
#define LAREXAMPLES_ALGORITHMS_TOTALLYCHEATTRACKS_CHEATTRACKDATA_CHEATTRACK_H


// LArSoft libraries
#include "lardataobj/RecoBase/Trajectory.h"

// ROOT libraries
#include "TDatabasePDG.h"

// C/C++ standard libraries
#include <ostream>
#include <string>
#include <utility> // std::move(), std::forward()


namespace lar {
  
  namespace example {
    // BEGIN TotallyCheatTracks group ------------------------------------------
    
    class CheatTrack {
      
        public:
      
      using PDGID_t = int; 
      
      static constexpr PDGID_t InvalidParticleID = 0;
      
      CheatTrack() = default;
      
      CheatTrack(recob::Trajectory&& traj, PDGID_t pid)
        : fTraj(std::move(traj))
        , fPDGID(pid)
        {}
      
      
      recob::Trajectory const& trajectory() const { return fTraj; }
      
      // --- BEGIN access to data ----------------------------------------------
      
      double momentum() const { return trajectory().StartMomentum(); }
      
      PDGID_t particleId() const { return fPDGID; }
      
      bool hasParticleId() const { return particleId() != InvalidParticleID; }
      
      // --- END access to data ------------------------------------------------
      
      // --- BEGIN printing data -----------------------------------------------
      
      static constexpr unsigned int DefaultDumpVerbosity = 1U;
      
      static constexpr unsigned int MaxDumpVerbosity
        = recob::Trajectory::MaxDumpVerbosity;
      

      template <typename Stream>
      void dump(
        Stream&& out, unsigned int verbosity,
        std::string indent, std::string firstIndent
        ) const;
      template <typename Stream>
      void dump(
        Stream&& out, unsigned int verbosity = DefaultDumpVerbosity,
        std::string indent = ""
        ) const
        { dump(std::forward<Stream>(out), verbosity, indent, indent); }
      
      
      // --- END printing data -------------------------------------------------
      
        private:
      
      recob::Trajectory fTraj;            
      PDGID_t fPDGID = InvalidParticleID; 
      
    }; // class CheatTrack
    
    
    inline std::ostream& operator<<
      (std::ostream& out, lar::example::CheatTrack const& track)
      { track.dump(out); return out; }
    
    // END TotallyCheatTracks group ------------------------------------------
    
  } // namespace example
  
} // namespace lar


//------------------------------------------------------------------------------
//---  template implementation
//------------------------------------------------------------------------------
template <typename Stream>
void lar::example::CheatTrack::dump(
  Stream&& out, unsigned int verbosity,
  std::string indent, std::string firstIndent
  ) const
{
  
  // we could use ROOT's TDatabasePDG to get the name of the ID, but we'd rather
  // not depend on ROOT here...
  out << firstIndent
    << "particle: ";
  auto const* pPDGinfo = TDatabasePDG::Instance()->GetParticle(particleId());
  if (pPDGinfo) out << pPDGinfo->GetName() << " (ID=" << particleId() << ")";
  else          out << "ID " << particleId();
  out << "; momentum: " << momentum() << " GeV/c; ";
  trajectory().Dump(std::forward<Stream>(out), verbosity, indent, "");
  
} // lar::example::CheatTrack::dump()

//------------------------------------------------------------------------------


#endif // LAREXAMPLES_ALGORITHMS_TOTALLYCHEATTRACKS_CHEATTRACKDATA_CHEATTRACK_H