ldmx-sw/EcalPnetVetoProcessor_8h_source.html

#ifndef EVENTPROC_ECALPNETVETOPROCESSOR_H_

#define EVENTPROC_ECALPNETVETOPROCESSOR_H_


// LDMX

#include <algorithm>

#include <numeric>


#include "DetDescr/EcalGeometry.h"

#include "Ecal/EcalHelper.h"

#include "Ecal/Event/EcalHit.h"

#include "Ecal/Event/EcalVetoResult.h"

#include "Framework/Configure/Parameters.h"

#include "Framework/EventProcessor.h"

#include "SimCore/Event/SimTrackerHit.h"

#include "Tools/ONNXRuntime.h"


namespace ecal {


class EcalPnetVetoProcessor : public framework::Producer {

 public:

  EcalPnetVetoProcessor(const std::string& name, framework::Process& process);

  virtual ~EcalPnetVetoProcessor() = default;

  void configure(framework::config::Parameters& parameters) override;

  void produce(framework::Event& event) override;


 private:

  void makeInputs(const ldmx::EcalGeometry& geom,

                  const std::vector<ldmx::EcalHit>& ecalRecHits,

                  std::array<double, 3> etraj, std::array<double, 3> enorm);


  std::vector<float> logSoftmax(const std::vector<float>& logits);


 private:

  constexpr static unsigned int MAX_NUM_HITS = 300;


  constexpr static unsigned int N_COORDINATE_DIM = 3;

  constexpr static unsigned int COORDINATE_X_OFFSET = 0;

  constexpr static unsigned int COORDINATE_Y_OFFSET = MAX_NUM_HITS;

  constexpr static unsigned int COORDINATE_Z_OFFSET = 2 * MAX_NUM_HITS;


  constexpr static unsigned int N_FEATURE_DIM = 5;

  constexpr static unsigned int FEATURE_X_OFFSET = 0;

  constexpr static unsigned int FEATURE_Y_OFFSET = MAX_NUM_HITS;

  constexpr static unsigned int FEATURE_Z_OFFSET = 2 * MAX_NUM_HITS;

  constexpr static unsigned int FEATURE_ENERGY_OFFSET = 3 * MAX_NUM_HITS;

  constexpr static unsigned int FEATURE_LAYER_ID_OFFSET = 4 * MAX_NUM_HITS;


  const static std::vector<std::string> INPUT_NAMES;

  const static std::vector<unsigned int> INPUT_SIZES;


  float disc_cut_ = -99;

  std::vector<std::vector<float>> data_;

  std::unique_ptr<ldmx::ort::ONNXRuntime> rt_;


  std::string collection_name_{"EcalPnetVeto"};


  std::string rec_coll_name_;

  std::string ecal_rec_hits_passname_;

  std::string ecal_sp_hits_passname_;

  std::string track_pass_name_;

  std::string track_collection_;

  bool recoil_from_tracking_;

};


}  // namespace ecal


#endif

EcalGeometry.h
Class that translates raw positions of ECal module hits into cells in a hexagonal readout.

EcalHelper.h
Class that propagates tracks to the ECAL face.

EcalVetoResult.h
Class used to encapsulate the results obtained from EcalVetoProcessor.

EventProcessor.h
Base classes for all user event processing components to extend.

SimTrackerHit.h
Class which encapsulates information from a hit in a simulated tracking detector.

ecal::EcalPnetVetoProcessor
Determines if event is vetoable using ECAL hit information w/ a deep neural network.
Definition EcalPnetVetoProcessor.h:31

ecal::EcalPnetVetoProcessor::produce
void produce(framework::Event &event) override
Process the event and put new data products into it.
Definition EcalPnetVetoProcessor.cxx:36

ecal::EcalPnetVetoProcessor::collection_name_
std::string collection_name_
Name of the collection which will containt the results.
Definition EcalPnetVetoProcessor.h:79

ecal::EcalPnetVetoProcessor::logSoftmax
std::vector< float > logSoftmax(const std::vector< float > &logits)
Transform logits to a probability.
Definition EcalPnetVetoProcessor.cxx:197

ecal::EcalPnetVetoProcessor::configure
void configure(framework::config::Parameters &parameters) override
Callback for the EventProcessor to configure itself from the given set of parameters.
Definition EcalPnetVetoProcessor.cxx:18

ecal::EcalPnetVetoProcessor::MAX_NUM_HITS
static constexpr unsigned int MAX_NUM_HITS
Maximum number of hits allowed in ECAL.
Definition EcalPnetVetoProcessor.h:57

ecal::EcalPnetVetoProcessor::makeInputs
void makeInputs(const ldmx::EcalGeometry &geom, const std::vector< ldmx::EcalHit > &ecalRecHits, std::array< double, 3 > etraj, std::array< double, 3 > enorm)
Make inputs to the DNN from ECAL RecHits.
Definition EcalPnetVetoProcessor.cxx:148

framework::Event
Implements an event buffer system for storing event data.
Definition Event.h:42

framework::Process
Class which represents the process under execution.
Definition Process.h:36

framework::Producer
Base class for a module which produces a data product.
Definition EventProcessor.h:257

framework::Producer::process
virtual void process(Event &event) final
Processing an event for a Producer is calling produce.
Definition EventProcessor.h:278

framework::config::Parameters
Class encapsulating parameters for configuring a processor.
Definition Parameters.h:29

ldmx::EcalGeometry
Translation between real-space positions and cell IDs within the ECal.
Definition EcalGeometry.h:108