Add operator<< for ParticleType that accepts an rvalue

Harvard-Neutrino · Mar 4, 2024 · 98a8517 · 98a8517
1 parent 069eafe
commit 98a8517
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/projects/dataclasses/private/ParticleType.cxx b/projects/dataclasses/private/ParticleType.cxx
@@ -5,6 +5,11 @@
 #include <iostream>
 #include <inttypes.h>
 
+std::ostream & operator<<(std::ostream & os, LI::dataclasses::ParticleType && p) {
+    LI::dataclasses::ParticleType pp = p;
+    return operator<<(os, pp);
+}
+
 std::ostream & operator<<(std::ostream & os, LI::dataclasses::ParticleType const & p) {
     if (LI::dataclasses::ParticleTypeNames.find(p) != LI::dataclasses::ParticleTypeNames.end())
         os << LI::dataclasses::ParticleTypeNames.at(p);

diff --git a/projects/dataclasses/public/LeptonInjector/dataclasses/ParticleType.h b/projects/dataclasses/public/LeptonInjector/dataclasses/ParticleType.h
@@ -46,6 +46,7 @@ static const std::map<int32_t, std::string> ParticleTypeIntNames = {
 } // namespace LI
 } // namespace dataclasses
 
+std::ostream & operator<<(std::ostream & os, LI::dataclasses::ParticleType && p);
 std::ostream & operator<<(std::ostream & os, LI::dataclasses::ParticleType const & p);
 
 namespace cereal {