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:

   enum DebugLevel_pi0 { pDEBUG = 0, pINFO = 1, pERROR = 2 };

   typedef std::map<DetId, EcalRecHit> RecHitsMap;

   EndcapPiZeroDiscriminatorAlgo() :
   preshStripEnergyCut_(0.), preshSeededNstr_(5), debugLevel_(pINFO), pathToFiles_("")
   {}

   EndcapPiZeroDiscriminatorAlgo(double stripEnergyCut, int nStripCut, const std::string& path,
                                 DebugLevel_pi0 debugLevel );

   ~EndcapPiZeroDiscriminatorAlgo() {};

   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);

   void readWeightFile(const char *WFile);

   float Activation_fun(float SUM);

   float getNNoutput(int sel_wfile);

   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);  


   float* get_input_vector() {return input_var;};

 private:

   double preshStripEnergyCut_;
   int preshSeededNstr_;
   int debugLevel_;

   int inp_var;
   int barrelstart;  

   int Nfiles_EB;
   int Nfiles_EE;

   int Layers, Indim, Hidden, Outdim;
   int EE_Layers, EE_Indim, EE_Hidden, EE_Outdim;
   int EB_Layers, EB_Indim, EB_Hidden, EB_Outdim;

   float* I_H_Weight;
   float* H_O_Weight;
   float* H_Thresh;
   float* O_Thresh;

   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;

   float* input_var;
//   float input_var[25]; // array with the 25 variables to be used as input in NN


   // The map of hits
   RecHitsMap *rechits_map;

   // path to weight files
  std::string pathToFiles_;

};
#endif