EcalPnetVetoProcessor.h

Go to the documentation of this file.

 
#ifndef EVENTPROC_ECALPNETVETOPROCESSOR_H_
#define EVENTPROC_ECALPNETVETOPROCESSOR_H_
 
// LDMX
#include "DetDescr/EcalGeometry.h"
#include "Ecal/Event/EcalHit.h"
#include "Framework/Configure/Parameters.h"
#include "Framework/EventProcessor.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;
 
  double disc_cut_logit_{0};
  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_coll_name_;
  std::string ecal_sp_hits_passname_;
  std::string track_pass_name_;
  std::string track_collection_;
  bool recoil_from_tracking_;
};
 
}  // namespace ecal
 
#endif