SimMuonCounter.cc

Go to the documentation of this file.

#include "SimMuonCounter.h"

#include <evt/Event.h>
#include <evt/ShowerSimData.h>

#include <utl/ErrorLogger.h>
#include <utl/ShowerParticleIterator.h>
#include <utl/Particle.h>
#include <utl/UTMPoint.h>


namespace SimMuonCounter {

  fwk::VModule::ResultFlag
  SimMuonCounter::Run(evt::Event& event)
  {
    if (!event.HasSimShower())
      ERROR("Event has no sim shower");

    auto& simShower = event.GetSimShower();

    double totalWeight = 0;

    for (auto pIt = simShower.GroundParticlesBegin(), end = simShower.GroundParticlesEnd();
         pIt != end; ++pIt) {

      const auto type = pIt->GetType();

      if (!(type == utl::Particle::eMuon || type == utl::Particle::eAntiMuon))
        continue;

      totalWeight += pIt->GetWeight();
    }

    std::ostringstream info;
    info << "total weight of (anti)muons is " << totalWeight;
    INFO(info);
    simShower.SetMuonNumber(totalWeight);

    return eSuccess;
  }

}