SEvent/StationSimData.cc

Go to the documentation of this file.

#include <sevt/StationSimData.h>
#include <sevt/StationTriggerData.h>
#include <sevt/StationGPSData.h>
#include <sdet/Station.h>
#include <utl/Particle.h>
#include <utl/ShadowPtr.h>
#include <utl/ErrorLogger.h>
#include <fwk/RandomEngineRegistry.h>
#include <sstream>

using namespace utl;
using namespace fwk;
using namespace std;


namespace sevt {

  void
  StationSimData::AddParticle(const utl::Particle& particle)
  {
    /*
     * Add one to the counter for all injection candidate particles
     * regardless of whether they are truly injected in the end or
     * removed during a Shrink() (station-level particle thinning).
     */
    ++fTotalSimCandidateParticleCount;

    // keep this decision inline
    if (fThinning == 1)
      AlwaysInsertParticle(particle);
    else
      SometimesInsertParticle(particle);
    if (fParticles.size() >= fMaxNParticles)
      Shrink();
  }


  void
  StationSimData::CountParticle(const utl::Particle& particle)
  {
    ++fTotalParticleCount;
    /*
     * Should return enum, not int. Perhaps modification of GetType() member
     * function of Particle class warranted
     */
    int type = particle.GetType();
    if (type == utl::Particle::eMuon || type == utl::Particle::eAntiMuon)
      ++fNumberOfMuons;
    else if (type == utl::Particle::eElectron || type == utl::Particle::ePositron)
      ++fNumberOfElectrons;
    else if (type == utl::Particle::ePhoton)
      ++fNumberOfPhotons;
  }


  void
  StationSimData::SometimesInsertParticle(const Particle& particle)
  {
    const double sometimes =
      RandomEngineRegistry::GetInstance().Get(RandomEngineRegistry::eDetector).GetEngine().flat();
    if (sometimes < fThinning)
      AlwaysInsertParticle(particle);
  }


  void
  StationSimData::Shrink()
  {
    ParticleVector newList;
    auto& rnd = RandomEngineRegistry::GetInstance().Get(RandomEngineRegistry::eDetector).GetEngine();
    for (const auto& p : fParticles)
      if (rnd.flat() < fThinningFactor)
        newList.push_back(p);
    fParticles.swap(newList);
    fThinning *= fThinningFactor;
  }


  void
  StationSimData::ClearParticleList()
  {
    decltype(fParticles)().swap(fParticles);
  }


  void
  StationSimData::MakeTriggerAndGPSData(const TimeStamp& time)
  {
    if (HasTriggerData(time)) {
      WARNING("Trigger and GPS data already present!");
      return;
    }

    fTriggerAndGPSData.emplace(time, TriggerGPSPair(StationTriggerData(), StationGPSData()));
  }


  const std::string&
  StationSimData::GetSimulatorSignature()
    const
  {
    if (fSimulatorSignature.empty()) {
      const string err = "Requesting an SD simulator signature which was never set.";
      ERROR(err);
      throw NonExistentComponentException(err);
    }
    return fSimulatorSignature;
  }


  bool
  StationSimData::HasGPSData(const TimeStamp& time)
    const
  {
    return fTriggerAndGPSData.find(time) != fTriggerAndGPSData.end();
  }


  StationGPSData&
  StationSimData::GetGPSData(const TimeStamp& time)
  {
    if (!HasTriggerData(time)) {
      ostringstream err;
      err << "Tried to get unavailable GPS data at time " << time;
      ERROR(err);
      throw NonExistentComponentException(err.str());
    }

    return fTriggerAndGPSData.find(time)->second.second;
  }


  const StationGPSData&
  StationSimData::GetGPSData(const TimeStamp& time)
    const
  {
    if (!HasTriggerData(time)) {
      ostringstream err;
      err << "Tried to get unavailable GPS data at time " << time;
      ERROR(err);
      throw NonExistentComponentException(err.str());
    }

    return fTriggerAndGPSData.find(time)->second.second;
  }


  StationTriggerData&
  StationSimData::GetTriggerData(const TimeStamp& time)
  {
    if (!HasTriggerData(time)) {
      ostringstream err;
      err << "Tried to get unavailable trigger data at time " << time;
      ERROR(err);
      throw NonExistentComponentException(err.str());
    }

    return fTriggerAndGPSData.find(time)->second.first;
  }


  const StationTriggerData&
  StationSimData::GetTriggerData(const TimeStamp& time)
    const
  {
    if (!HasTriggerData(time)) {
      ostringstream err;
      err << "Tried to get unavailable trigger data at time " << time;
      ERROR(err);
      throw NonExistentComponentException(err.str());
    }

    return fTriggerAndGPSData.find(time)->second.first;
  }

}