SimParticle.cxx

#include "SimCore/Event/SimParticle.h"
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <iostream>
 
ClassImp(ldmx::SimParticle)
 
    namespace ldmx {
  SimParticle::ProcessTypeMap SimParticle::createProcessTypeMap() {
    ProcessTypeMap procMap;
    procMap["eBrem"] = ProcessType::eBrem;
    procMap["conv"] = ProcessType::conv;
    procMap["annihil"] = ProcessType::annihil;
    procMap["compt"] = ProcessType::compt;
    procMap["phot"] = ProcessType::phot;
    procMap["eIoni"] = ProcessType::eIoni;
    procMap["msc"] = ProcessType::msc;
    procMap["photonNuclear"] = ProcessType::photonNuclear;
    procMap["muonNuclear"] = ProcessType::muonNuclear;
    procMap["electronNuclear"] = ProcessType::electronNuclear;
    procMap["GammaToMuPair"] = ProcessType::GammaToMuPair;
    procMap["DarkBrem"] = ProcessType::eDarkBrem;
 
    // Inelastic interactions
    procMap["neutronInelastic"] = ProcessType::neutronInelastic;
    procMap["neutronCapture"] = ProcessType::neutronCapture;
    procMap["kaon-Inelastic"] = ProcessType::kaonInelastic;
    procMap["kaon+Inelastic"] = ProcessType::kaonInelastic;
    procMap["kaon0LInelastic"] = ProcessType::kaonInelastic;
    procMap["kaon0SInelastic"] = ProcessType::kaonInelastic;
    procMap["pi-Inelastic"] = ProcessType::pionInelastic;
    procMap["pi+Inelastic"] = ProcessType::pionInelastic;
    procMap["protonInelastic"] = ProcessType::protonInelastic;
 
    procMap["Primary"] = ProcessType::Primary;
    // Decay
    procMap["Decay"] = ProcessType::Decay;
    return procMap;
  }
 
  SimParticle::ProcessTypeMap SimParticle::PROCESS_MAP =
      SimParticle::createProcessTypeMap();
 
  void SimParticle::Clear() {
    daughters_.clear();
    parents_.clear();
 
    energy_ = 0;
    pdgID_ = 0;
    genStatus_ = -1;
    time_ = 0;
    x_ = 0;
    y_ = 0;
    z_ = 0;
    endX_ = 0;
    endY_ = 0;
    endZ_ = 0;
    px_ = 0;
    py_ = 0;
    pz_ = 0;
    endpx_ = 0;
    endpy_ = 0;
    endpz_ = 0;
    mass_ = 0;
    charge_ = 0;
    processType_ = ProcessType::unknown;
    vertexVolume_ = "";
  }
 
  void SimParticle::Print() const {
    std::cout << "SimParticle { "
              << "energy: " << energy_ << ", "
              << "PDG ID: " << pdgID_ << ", "
              << "genStatus: " << genStatus_ << ", "
              << "time: " << time_ << ", "
              << "vertex: ( " << x_ << ", " << y_ << ", " << z_ << " ), "
              << "endPoint: ( " << endX_ << ", " << endY_ << ", " << endZ_
              << " ), "
              << "momentum: ( " << px_ << ", " << py_ << ", " << pz_ << " ), "
              << "endPointMomentum: ( " << endpx_ << ", " << endpy_ << ", "
              << endpz_ << " ), "
              << "mass: " << mass_ << ", "
              << "nDaughters: " << daughters_.size() << ", "
              << "nParents: " << parents_.size() << ", "
              << "processType: " << processType_ << ", "
              << "vertex volume: " << vertexVolume_ << " }" << std::endl;
  }
 
  SimParticle::ProcessType SimParticle::findProcessType(
      std::string processName) {
    if (processName.find("biasWrapper") != std::string::npos) {
      std::size_t pos = processName.find_first_of("(") + 1;
      processName = processName.substr(pos, processName.size() - pos - 1);
    }
 
    if (PROCESS_MAP.find(processName) != PROCESS_MAP.end()) {
      return PROCESS_MAP[processName];
    } else {
      return ProcessType::unknown;
    }
  }
}  // namespace ldmx