TruthHitProducer.cxx

 
#include "TrigScint/TruthHitProducer.h"
 
#include "SimCore/Event/SimCalorimeterHit.h"
#include "SimCore/Event/SimParticle.h"
 
namespace trigscint {
 
TruthHitProducer::TruthHitProducer(const std::string& name,
                                   framework::Process& process)
    : Producer(name, process) {}
 
void TruthHitProducer::configure(framework::config::Parameters& parameters) {
  input_collection_ = parameters.getParameter<std::string>("input_collection");
  input_pass_name_ = parameters.getParameter<std::string>("input_pass_name");
  output_collection_ =
      parameters.getParameter<std::string>("output_collection");
  sim_particles_pass_name_ =
      parameters.getParameter<std::string>("sim_particles_pass_name");
 
  ldmx_log(info) << "In TruthHitProducer: configure done!";
  ldmx_log(info) << "Got parameters:  " << "\nInput collection:     "
                 << input_collection_
                 << "\nInput pass name:     " << input_pass_name_
                 << "\nOutput collection:    " << output_collection_;
}
 
void TruthHitProducer::produce(framework::Event& event) {
  // Check if the collection exists.  If not, don't bother processing the event.
  if (!event.exists(input_collection_, input_pass_name_)) {
    ldmx_log(error) << "No input collection called " << input_collection_ << "_"
                    << input_pass_name_ << " found; skipping!";
    return;
  }
  if (!event.exists("SimParticles", sim_particles_pass_name_)) {
    ldmx_log(error) << "No input SimParticle collection with pass "
                    << sim_particles_pass_name_ << " found; skipping!";
    return;
  }
 
  // looper over sim hits and aggregate energy depositions for each detID
  const auto sim_hits{event.getCollection<ldmx::SimCalorimeterHit>(
      input_collection_, input_pass_name_)};
  auto particle_map{event.getMap<int, ldmx::SimParticle>(
      "SimParticles", sim_particles_pass_name_)};
 
  std::vector<ldmx::SimCalorimeterHit> truth_beam_electrons;
 
  // TODO: Convert this to using a for_each and lambda
  for (const auto& sim_hit : sim_hits) {
    bool keep{false};
    // check if hit is from beam electron and, if so, add to output collection
    for (int i = 0; i < sim_hit.getNumberOfContribs(); i++) {
      auto contrib = sim_hit.getContrib(i);
      ldmx_log(trace) << "contrib " << i << " track_id: " << contrib.track_id_
                      << " pdgID: " << contrib.pdg_code_
                      << " edep: " << contrib.edep_;
      ldmx_log(trace) << "\t particle id: "
                      << particle_map[contrib.track_id_].getPdgID()
                      << " particle status: "
                      << particle_map[contrib.track_id_].getGenStatus();
 
      // if the trackID is in the map
      if (particle_map.find(contrib.track_id_) != particle_map.end()) {
        // beam electron (PDGID = 11, genStatus == 1)
        if (particle_map[contrib.track_id_].getPdgID() == 11 &&
            particle_map[contrib.track_id_].getGenStatus() == 1) {
          keep = true;
        }
      }
      if (keep) truth_beam_electrons.push_back(sim_hit);
    }  // over simhit contribs
  }  // over simhits
  event.add(output_collection_, truth_beam_electrons);
}
}  // namespace trigscint
 
DECLARE_PRODUCER(trigscint::TruthHitProducer)