EndcapPiZeroDiscriminatorAlgo.h

Go to the documentation of this file.

#ifndef RecoEcal_EgammaClusterAlgos_EndcapPiZeroDiscriminatorAlgo_h
#define RecoEcal_EgammaClusterAlgos_EndcapPiZeroDiscriminatorAlgo_h
 
#include "DataFormats/EcalDetId/interface/ESDetId.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "RecoCaloTools/Navigation/interface/EcalPreshowerNavigator.h"
 
// C/C++ headers
#include <string>
#include <vector>
#include <map>
 
// authors A. Kyriakis , D. Maletic
 
class EndcapPiZeroDiscriminatorAlgo {
public:
  typedef std::map<DetId, EcalRecHit> RecHitsMap;
 
  EndcapPiZeroDiscriminatorAlgo() : preshStripEnergyCut_(0.), preshSeededNstr_(5) {}
 
  EndcapPiZeroDiscriminatorAlgo(double stripEnergyCut, int nStripCut, const std::string& path);
 
  std::vector<float> findPreshVector(ESDetId strip, RecHitsMap* rechits_map, CaloSubdetectorTopology* topology_p);
 
  void findPi0Road(ESDetId strip, EcalPreshowerNavigator& theESNav, int plane, std::vector<ESDetId>& vout);
 
  bool goodPi0Strip(RecHitsMap::iterator candidate_it, ESDetId lastID);
 
  bool calculateNNInputVariables(
      std::vector<float>& vph1, std::vector<float>& vph2, float pS1_max, float pS9_max, float pS25_max, int EScorr);
 
  void calculateBarrelNNInputVariables(float et,
                                       double s1,
                                       double s9,
                                       double s25,
                                       double m2,
                                       double cee,
                                       double cep,
                                       double cpp,
                                       double s4,
                                       double s6,
                                       double ratio,
                                       double xcog,
                                       double ycog);
 
  float GetNNOutput(float EE_Et);
 
  float GetBarrelNNOutput(float EB_Et);
 
  std::vector<float> const& get_input_vector() const { return input_var; }
 
private:
  void readWeightFile(const char* WFile, int& Layers, int& Indim, int& Hidden, int& Outdim);
  float getNNoutput(int sel_wfile, int Layers, int Indim, int Hidden, int Outdim, int barrelstart) const;
  float Activation_fun(float SUM) const;
 
  double preshStripEnergyCut_;
  int preshSeededNstr_;
  int debugLevel_;
 
  int EE_Layers, EE_Indim, EE_Hidden, EE_Outdim;
  int EB_Layers, EB_Indim, EB_Hidden, EB_Outdim;
 
  std::vector<float> I_H_Weight_all;
  std::vector<float> H_O_Weight_all;
  std::vector<float> H_Thresh_all;
  std::vector<float> O_Thresh_all;
 
  std::vector<float> input_var;
  //   float input_var[25]; // array with the 25 variables to be used as input in NN
};
#endif