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