NuclearDQM.h

#ifndef DQM_NUCLEARDQM_H
#define DQM_NUCLEARDQM_H
 
#include <map>
#include <string>
#include <vector>
 
#include "Framework/Configure/Parameters.h"
#include "Framework/EventProcessor.h"
#include "SimCore/Event/SimParticle.h"
 
namespace dqm {
 
class NuclearDQM : public framework::Analyzer {
 public:
  enum class EventType {
    nothing_hard = 0,
    single_neutron = 1,
    two_neutrons = 2,
    three_or_more_neutrons = 3,
    single_charged_pion = 4,
    two_charged_pions = 5,
    single_neutral_pion = 6,
    single_charged_pion_and_nucleon = 7,
    single_charged_pion_and_two_nucleons = 8,
    two_charged_pions_and_nucleon = 9,
    single_neutral_pion_and_nucleon = 10,
    single_neutral_pion_and_two_nucleons = 11,
    single_neutral_pion_charged_pion_and_nucleon = 12,
    single_proton = 13,
    two_protons = 14,
    proton_neutron = 15,
    klong = 16,
    charged_kaon = 17,
    kshort = 18,
    exotics = 19,
    multibody = 20,
  };
 
  enum class CompactEventType {
    single_neutron = 0,
    single_charged_kaon = 1,
    single_neutral_kaon = 2,
    two_neutrons = 3,
    soft = 4,
    other = 5,
  };
 
  NuclearDQM(const std::string& name, framework::Process& process);
  virtual ~NuclearDQM() = default;
 
  void configure(framework::config::Parameters& parameters) override;
 
 protected:
  std::vector<const ldmx::SimParticle*> findDaughters(
      const std::map<int, ldmx::SimParticle>& particleMap,
      const ldmx::SimParticle* parent, int require_process_type = -1) const;
 
  void findParticleKinematics(
      const std::vector<const ldmx::SimParticle*>& daughters,
      const std::string& prefix);
 
  void findExtendedKinematics(
      const std::vector<const ldmx::SimParticle*>& daughters,
      const std::string& prefix);
 
  void findSubleadingKinematics(
      const ldmx::SimParticle* initiator,
      const std::vector<const ldmx::SimParticle*>& daughters,
      EventType eventType);
 
  EventType classifyEvent(
      const std::vector<const ldmx::SimParticle*>& daughters, double threshold);
 
  CompactEventType classifyCompactEvent(
      const ldmx::SimParticle* initiator,
      const std::vector<const ldmx::SimParticle*>& daughters, double threshold);
 
  constexpr bool isLightIon(int pdgCode) const {
    if (pdgCode > 1000000000) {
      return ((pdgCode / 10) % 1000) <= 4;
    }
    return false;
  }
 
  std::string sim_particles_coll_name_;
  std::string sim_particles_passname_;
  bool count_light_ions_{true};
};
 
}  // namespace dqm
 
#endif  // DQM_NUCLEARDQM_H