ParticleHistory.cxx

Go to the documentation of this file.

#include "nutools/ParticleNavigation/ParticleList.h"
#include "nutools/ParticleNavigation/ParticleHistory.h"

#include <cmath>

namespace sim {

  //----------------------------------------------------------------------------
  // Nothing special need be done for the constructor.
  ParticleHistory::ParticleHistory( const sim::ParticleList* list, const int trackID ) 
    : m_particleList(list)
    , m_trackID(trackID)
  {
    // Look for the track in the particle list.
    ParticleList::const_iterator search = m_particleList->find( m_trackID );

    // While we're still finding particles in the chain...
    while ( search != m_particleList->end() ){
      const simb::MCParticle* particle = (*search).second;
      push_front( particle );
      
      // If this is a primary particle, we're done.
      int trackID = particle->TrackId();
      if ( m_particleList->IsPrimary( trackID ) ) break;
      
      // Now look for the parent of this particle.
      int parentID = particle->Mother();
      search = m_particleList->find( parentID );
      
    } // while we're finding particles in the chain
  }

  //----------------------------------------------------------------------------
  // Nothing special for the destructor.
  ParticleHistory::~ParticleHistory()
  {}

  //----------------------------------------------------------------------------
  std::ostream& operator<< ( std::ostream& output, const ParticleHistory& list )
  {
    // Determine a field width for the particle number.
    ParticleHistory::size_type numberOfParticles = list.size();
    int numberOfDigits = (int) std::log10( (double) numberOfParticles ) + 1;

    // A simple header.
    output.width( numberOfDigits );
    output << "#" << ": < ID, particle >" << "\n"; 

    // Write each particle on a separate line.
    ParticleHistory::size_type nParticle = 0;
    for ( ParticleHistory::const_iterator particle = list.begin(); 
          particle != list.end(); ++particle, ++nParticle ){
      output.width( numberOfDigits );
      output << nParticle << ": " 
             << (*particle)
             << "\n";
    }
    
    return output;
  }
  
} // namespace sim