EndcapPiZeroDiscriminatorAlgo Class Reference

#include <EndcapPiZeroDiscriminatorAlgo.h>

Public Types
typedef std::map< DetId, EcalRecHit >	RecHitsMap

Public Member Functions
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)
bool	calculateNNInputVariables (std::vector< float > &vph1, std::vector< float > &vph2, float pS1_max, float pS9_max, float pS25_max, int EScorr)
	EndcapPiZeroDiscriminatorAlgo ()
	EndcapPiZeroDiscriminatorAlgo (double stripEnergyCut, int nStripCut, const std::string &path)
void	findPi0Road (ESDetId strip, EcalPreshowerNavigator &theESNav, int plane, std::vector< ESDetId > &vout)
std::vector< float >	findPreshVector (ESDetId strip, RecHitsMap *rechits_map, CaloSubdetectorTopology *topology_p)
std::vector< float > const &	get_input_vector () const
float	GetBarrelNNOutput (float EB_Et)
float	GetNNOutput (float EE_Et)
bool	goodPi0Strip (RecHitsMap::iterator candidate_it, ESDetId lastID)

Private Member Functions
float	Activation_fun (float SUM) const
float	getNNoutput (int sel_wfile, int Layers, int Indim, int Hidden, int Outdim, int barrelstart) const
void	readWeightFile (const char *WFile, int &Layers, int &Indim, int &Hidden, int &Outdim)

Private Attributes
int	debugLevel_
int	EB_Hidden
int	EB_Indim
int	EB_Layers
int	EB_Outdim
int	EE_Hidden
int	EE_Indim
int	EE_Layers
int	EE_Outdim
std::vector< float >	H_O_Weight_all
std::vector< float >	H_Thresh_all
std::vector< float >	I_H_Weight_all
std::vector< float >	input_var
std::vector< float >	O_Thresh_all
int	preshSeededNstr_
double	preshStripEnergyCut_

Detailed Description

Definition at line 15 of file EndcapPiZeroDiscriminatorAlgo.h.

Member Typedef Documentation

RecHitsMap

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

Definition at line 17 of file EndcapPiZeroDiscriminatorAlgo.h.

Constructor & Destructor Documentation

EndcapPiZeroDiscriminatorAlgo() [1/2]

EndcapPiZeroDiscriminatorAlgo::EndcapPiZeroDiscriminatorAlgo ( )

inline

Definition at line 19 of file EndcapPiZeroDiscriminatorAlgo.h.

: preshStripEnergyCut_(0.), preshSeededNstr_(5) {}

EndcapPiZeroDiscriminatorAlgo() [2/2]

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

Member Function Documentation

Activation_fun()

float EndcapPiZeroDiscriminatorAlgo::Activation_fun ( float  SUM ) const

private

Definition at line 328 of file EndcapPiZeroDiscriminatorAlgo.cc.

References JetChargeProducer_cfi::exp, and SUM.

Referenced by getNNoutput().

{ return (1.0 / (1.0 + exp(-2.0 * SUM))); }

calculateBarrelNNInputVariables()

void EndcapPiZeroDiscriminatorAlgo::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
	)

Definition at line 467 of file EndcapPiZeroDiscriminatorAlgo.cc.

References EB_Indim, input_var, callgraph::m2, particleFlowDisplacedVertex_cfi::ratio, photons_cff::s4, and mathSSE::sqrt().

                                                                                 {
  input_var.resize(EB_Indim);

  double lam, lam1, lam2;

  if (xcog < 0.) {
    input_var[0] = -xcog / s25;
  } else {
    input_var[0] = xcog / s25;
  }

  input_var[1] = cee / 0.0004;

  if (cpp < .001) {
    input_var[2] = cpp / .001;
  } else {
    input_var[2] = 0.;
  }

  if (s9 != 0.) {
    input_var[3] = s1 / s9;
    input_var[8] = s6 / s9;
    input_var[10] = (m2 + s1) / s9;
  } else {
    input_var[3] = 0.;
    input_var[8] = 0.;
    input_var[10] = 0.;
  }

  if (s25 - s1 > 0.) {
    input_var[4] = (s9 - s1) / (s25 - s1);
  } else {
    input_var[4] = 0.;
  }

  if (s25 > 0.) {
    input_var[5] = s4 / s25;
  } else {
    input_var[5] = 0.;
  }

  if (ycog < 0.) {
    input_var[6] = -ycog / s25;
  } else {
    input_var[6] = ycog / s25;
  }

  input_var[7] = ratio;

  lam = sqrt((cee - cpp) * (cee - cpp) + 4 * cep * cep);
  lam1 = (cee + cpp + lam) / 2;
  lam2 = (cee + cpp - lam) / 2;

  if (lam1 == 0) {
    input_var[9] = .0;
  } else {
    input_var[9] = lam2 / lam1;
  }
  if (s4 != 0.) {
    input_var[11] = (m2 + s1) / s4;
  } else {
    input_var[11] = 0.;
  }
}

calculateNNInputVariables()

bool EndcapPiZeroDiscriminatorAlgo::calculateNNInputVariables	(	std::vector< float > &	vph1,
		std::vector< float > &	vph2,
		float	pS1_max,
		float	pS9_max,
		float	pS25_max,
		int	EScorr
	)

Definition at line 340 of file EndcapPiZeroDiscriminatorAlgo.cc.

References MillePedeFileConverter_cfg::e, EE_Indim, mps_fire::i, input_var, dqmdumpme::k, GetRecoTauVFromDQM_MC_cff::kk, and LogTrace.

                                                                                                        {
  input_var.resize(EE_Indim);
  bool valid_NNinput = true;

  /*   
   for(int i = 0; i<11;i++) {
   LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: Energies of the Preshower Strips in X plane = " << vph1[i] ;
   }
   
   for(int i = 0; i<11;i++) {
   LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: Energies of the Preshower Strips in Y plane = " << vph2[i] ;
   } 
   */

  // check if all Preshower info is availabla - If NOT use remaning info
  for (int k = 0; k < 11; k++) {
    if (vph1[k] < 0) {
      LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: Oops!!! Preshower Info for strip : " << k
                               << " of X plane Do not exists";

      vph1[k] = 0.0;
    }
    if (vph2[k] < 0) {
      LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: Oops!!! Preshower Info for strip : " << k
                               << " of Y plane Do not exists";

      vph2[k] = 0.0;
    }
  }

  /*
   for(int i = 0; i<11;i++) {  
     LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: After: Energies of the Preshower Strips in X plane = " << vph1[i] ;
   }

   for(int i = 0; i<11;i++) {
 
     LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: After: Energies of the Preshower Strips in Y plane = " << vph2[i] ;
   }
   */

  // FIRST : Produce the 22 NN variables related with the Preshower
  // --------------------------------------------------------------
  // New normalization of the preshower strip energies Aris 8/11/2004
  for (int kk = 0; kk < 11; kk++) {
    input_var[kk] = fabs(vph1[kk] / 0.01);
    input_var[kk + 11] = fabs(vph2[kk] / 0.02);
    if (input_var[kk] < 0.0001)
      input_var[kk] = 0.;
    if (input_var[kk + 11] < 0.0001)
      input_var[kk + 11] = 0.;
  }
  input_var[0] = fabs(input_var[0] / 2.);
  input_var[1] = fabs(input_var[1] / 2.);
  input_var[6] = fabs(input_var[6] / 2.);
  input_var[11] = fabs(input_var[11] / 2.);
  input_var[12] = fabs(input_var[12] / 2.);
  input_var[17] = fabs(input_var[17] / 2.);

  // correction for version > CMSSW_3_1_0_pre5 where extra enegry is given to the ES strips
  // Aris 18/5/2009
  if (EScorr == 1) {
    input_var[0] -= 0.05;
    input_var[1] -= 0.035;
    input_var[2] -= 0.035;
    input_var[3] -= 0.02;
    input_var[4] -= 0.015;
    input_var[5] -= 0.0075;
    input_var[6] -= 0.035;
    input_var[7] -= 0.035;
    input_var[8] -= 0.02;
    input_var[9] -= 0.015;
    input_var[10] -= 0.0075;

    input_var[11] -= 0.05;
    input_var[12] -= 0.035;
    input_var[13] -= 0.035;
    input_var[14] -= 0.02;
    input_var[15] -= 0.015;
    input_var[16] -= 0.0075;
    input_var[17] -= 0.035;
    input_var[18] -= 0.035;
    input_var[19] -= 0.02;
    input_var[20] -= 0.015;
    input_var[21] -= 0.0075;

    for (int kk1 = 0; kk1 < 22; kk1++) {
      if (input_var[kk1] < 0)
        input_var[kk1] = 0.0;
    }
  }
  // SECOND: Take the final NN variable related to the ECAL
  // -----------------------------------------------
  float ECAL_norm_factor = 500.;
  if (pS25_max > 500 && pS25_max <= 1000)
    ECAL_norm_factor = 1000;
  if (pS25_max > 1000)
    ECAL_norm_factor = 7000;

  input_var[22] = pS1_max / ECAL_norm_factor;
  input_var[23] = pS9_max / ECAL_norm_factor;
  input_var[24] = pS25_max / ECAL_norm_factor;

  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: S1/ECAL_norm_factor = " << input_var[22];
  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: S9/ECAL_norm_factor = " << input_var[23];
  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: S25/ECAL_norm_factor = " << input_var[24];

  for (int i = 0; i < EE_Indim; i++) {
    if (input_var[i] > 1.0e+00) {
      valid_NNinput = false;
      break;
    }
  }

  LogTrace("EcalClusters") << " valid_NNinput = " << valid_NNinput;

  return valid_NNinput;
}

findPi0Road()

void EndcapPiZeroDiscriminatorAlgo::findPi0Road	(	ESDetId	strip,
		EcalPreshowerNavigator &	theESNav,
		int	plane,
		std::vector< ESDetId > &	vout
	)

Definition at line 141 of file EndcapPiZeroDiscriminatorAlgo.cc.

References CaloNavigator< T, TOPO >::east(), CaloNavigator< T, TOPO >::home(), LogTrace, GetRecoTauVFromDQM_MC_cff::next, CaloNavigator< T, TOPO >::north(), preshSeededNstr_, CaloNavigator< T, TOPO >::setHome(), CaloNavigator< T, TOPO >::south(), nano_mu_digi_cff::strip, and CaloNavigator< T, TOPO >::west().

Referenced by findPreshVector().

                                                                       {
  if (strip == ESDetId(0))
    return;
  ESDetId next;
  theESNav.setHome(strip);
  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: starts from strip " << strip;

  if (plane == 1) {
    // east road
    int n_east = 0;
    LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: Go to the East ";

    while (((next = theESNav.east()) != ESDetId(0) && next != strip)) {
      LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: East: " << n_east << " current strip is "
                               << next;

      vout.push_back(next);
      ++n_east;
      if (n_east == preshSeededNstr_)
        break;
    }
    // west road
    int n_west = 0;
    LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: Go to the West ";

    theESNav.home();
    while (((next = theESNav.west()) != ESDetId(0) && next != strip)) {
      LogTrace("EcalClusters") << "findPi0Road: West: " << n_west << " current strip is " << next;

      vout.push_back(next);
      ++n_west;
      if (n_west == preshSeededNstr_)
        break;
    }
    LogTrace("EcalClusters")
        << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: Total number of strips found in the road at 1-st plane is "
        << n_east + n_west;

  } else if (plane == 2) {
    // north road
    int n_north = 0;
    LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: Go to the North ";

    while (((next = theESNav.north()) != ESDetId(0) && next != strip)) {
      LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: North: " << n_north
                               << " current strip is " << next;

      vout.push_back(next);
      ++n_north;
      if (n_north == preshSeededNstr_)
        break;
    }
    // south road
    int n_south = 0;
    LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: Go to the South ";

    theESNav.home();
    while (((next = theESNav.south()) != ESDetId(0) && next != strip)) {
      LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: South: " << n_south
                               << " current strip is " << next;

      vout.push_back(next);
      ++n_south;
      if (n_south == preshSeededNstr_)
        break;
    }
    LogTrace("EcalClusters")
        << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: Total number of strips found in the road at 2-nd plane is "
        << n_south + n_north;

  } else {
    LogTrace("EcalClusters")
        << "EndcapPiZeroDiscriminatorAlgo: findPi0Road: Wrong plane number, null cluster will be returned! ";

  }  // end of if

  theESNav.home();
}

findPreshVector()

vector< float > EndcapPiZeroDiscriminatorAlgo::findPreshVector	(	ESDetId	strip,
		RecHitsMap *	rechits_map,
		CaloSubdetectorTopology *	topology_p
	)

Definition at line 40 of file EndcapPiZeroDiscriminatorAlgo.cc.

References findPi0Road(), goodPi0Strip(), mps_fire::i, LogTrace, HLT_2023v12_cff::navigator, and nano_mu_digi_cff::strip.

                                                                                                  {
  vector<float> vout_stripE;

  // skip if rechits_map contains no hits
  if (rechits_map->empty()) {
    edm::LogWarning("EndcapPiZeroDiscriminatorAlgo") << "RecHitsMap has size 0.";
    return vout_stripE;
  }

  vout_stripE.clear();

  vector<ESDetId> road_2d;
  road_2d.clear();

  int plane = strip.plane();

  LogTrace("EcalClusters")
      << "EndcapPiZeroDiscriminatorAlgo: findPreshVectors: Preshower Seeded Algorithm - looking for clusters"
      << "n"
      << "findPreshVectors: Preshower is intersected at strip " << strip.strip() << ", at plane " << plane;

  if (strip == ESDetId(0)) {  //works in case of no intersected strip found
    for (int i = 0; i < 11; i++) {
      vout_stripE.push_back(-100.);
    }
  }

  // Add to the road the central strip
  road_2d.push_back(strip);

  //Make a navigator, and set it to the strip cell.
  EcalPreshowerNavigator navigator(strip, topology_p);
  navigator.setHome(strip);
  //search for neighbours in the central road
  findPi0Road(strip, navigator, plane, road_2d);

  LogTrace("EcalClusters")
      << "EndcapPiZeroDiscriminatorAlgo:findPreshVectors: Total number of strips in the central road: "
      << road_2d.size();

  // Find the energy of each strip
  RecHitsMap::iterator final_strip = rechits_map->end();
  // very dangerous, added a protection on the rechits_map->size()
  // at the beginning of the method
  final_strip--;
  ESDetId last_stripID = final_strip->first;

  vector<ESDetId>::iterator itID;
  for (itID = road_2d.begin(); itID != road_2d.end(); itID++) {
    LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPreshVectors: ID = " << *itID;

    float E = 0.;
    RecHitsMap::iterator strip_it = rechits_map->find(*itID);
    if (goodPi0Strip(strip_it, last_stripID)) {  // continue if strip not found in rechit_map
      E = strip_it->second.energy();
    }
    vout_stripE.push_back(E);
    LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: findPreshVectors: E = " << E;
  }

  // ***ML beg***
  // vector of size=11, content of vout_stripE is copied into vout_ElevenStrips_Energy
  // to avoid problem in case number of strips is less than 11
  vector<float> vout_ElevenStrips_Energy;
  vout_ElevenStrips_Energy.reserve(11);
  for (int i = 0; i < 11; i++) {
    vout_ElevenStrips_Energy.push_back(0.);
  }

  for (unsigned int i = 0; i < vout_stripE.size(); i++) {
    vout_ElevenStrips_Energy[i] = vout_stripE.at(i);
  }

  //return vout_stripE;
  return vout_ElevenStrips_Energy;
  // ***ML end***
}

get_input_vector()

std::vector<float> const& EndcapPiZeroDiscriminatorAlgo::get_input_vector ( ) const

inline

Definition at line 50 of file EndcapPiZeroDiscriminatorAlgo.h.

References input_var.

{ return input_var; }

GetBarrelNNOutput()

float EndcapPiZeroDiscriminatorAlgo::GetBarrelNNOutput

(

float

EB_Et

)

Definition at line 592 of file EndcapPiZeroDiscriminatorAlgo.cc.

References EB_Hidden, EB_Indim, EB_Layers, EB_Outdim, getNNoutput(), input_var, LogTrace, and findQualityFiles::v.

                                                                  {
  float nnout = -1;
  // Print the  NN input variables that are related to the ECAL Barrel
  // ------------------------------------------------------------------------
  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo::GetBarrelNNoutput :nn_invar_presh = ";

  LogTrace("EcalCluster").log([&](auto& lt) {
    for (auto const v : input_var) {
      lt << v << " ";
    }
  });
  LogTrace("EcalClusters") << " ";

  // select the appropriate Weigth file
  int sel_wfile;
  if (EB_Et < 25.0) {
    sel_wfile = 0;
  } else if (EB_Et >= 25.0 && EB_Et < 35.0) {
    sel_wfile = 1;
  } else if (EB_Et >= 35.0 && EB_Et < 45.0) {
    sel_wfile = 2;
  } else if (EB_Et >= 45.0 && EB_Et < 55.0) {
    sel_wfile = 3;
  } else {
    sel_wfile = 4;
  }
  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: E_SC = " << EB_Et
                           << " and I select Weight file Number = " << sel_wfile;

  nnout = getNNoutput(
      sel_wfile, EB_Layers, EB_Indim, EB_Hidden, EB_Outdim, 1);  // calculate the nnoutput for the given ECAL object

  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: ===================> GetNNOutput : NNout = " << nnout;

  return nnout;
}

GetNNOutput()

float EndcapPiZeroDiscriminatorAlgo::GetNNOutput

(

float

EE_Et

)

Definition at line 549 of file EndcapPiZeroDiscriminatorAlgo.cc.

References EE_Hidden, EE_Indim, EE_Layers, EE_Outdim, getNNoutput(), input_var, LogTrace, and findQualityFiles::v.

                                                            {
  float nnout = -1;
  // Print the  NN input variables that are related to the Preshower + ECAL
  // ------------------------------------------------------------------------
  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo::GetNNoutput :nn_invar_presh = ";

  LogTrace("EcalClusters").log([&](auto& lt) {
    for (auto const v : input_var) {
      lt << v << " ";
    }
  });
  LogTrace("EcalClusters") << " ";

  // select the appropriate Weigth file
  int sel_wfile;
  if (EE_Et < 25.0) {
    sel_wfile = 0;
  } else if (EE_Et >= 25.0 && EE_Et < 35.0) {
    sel_wfile = 1;
  } else if (EE_Et >= 35.0 && EE_Et < 45.0) {
    sel_wfile = 2;
  } else if (EE_Et >= 45.0 && EE_Et < 55.0) {
    sel_wfile = 3;
  } else {
    sel_wfile = 4;
  }

  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: Et_SC = " << EE_Et
                           << " and I select Weight file Number = " << sel_wfile;

  nnout = getNNoutput(
      sel_wfile, EE_Layers, EE_Indim, EE_Hidden, EE_Outdim, 0);  // calculate the nnoutput for the given ECAL object

  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: ===================> GetNNOutput : NNout = " << nnout;

  return nnout;
}

getNNoutput()

float EndcapPiZeroDiscriminatorAlgo::getNNoutput ( int  sel_wfile, int  Layers, int  Indim, int  Hidden, int  Outdim, int  barrelstart  ) const

private

Definition at line 297 of file EndcapPiZeroDiscriminatorAlgo.cc.

References Activation_fun(), EE_Hidden, EE_Indim, EE_Outdim, h, H_O_Weight_all, H_Thresh_all, mps_fire::i, I_H_Weight_all, input_var, LogTrace, CastorDigiReco::o1, O_Thresh_all, and CfgNavigationSchool_cfi::OUT.

Referenced by GetBarrelNNOutput(), and GetNNOutput().

                                                                                         {
  float nnout = 0.0;
  int mij;

  std::vector<float> I_SUM(size_t(Hidden), 0.0);
  std::vector<float> OUT(size_t(Outdim), 0.0);

  for (int h = 0; h < Hidden; h++) {
    mij = h - Hidden;
    for (int i = 0; i < Indim; i++) {
      mij = mij + Hidden;
      I_SUM[h] += I_H_Weight_all[mij + sel_wfile * Indim * Hidden + barrelstart * Nfiles_EE * EE_Indim * EE_Hidden] *
                  input_var[i];
    }
    I_SUM[h] += H_Thresh_all[h + sel_wfile * Hidden + barrelstart * Nfiles_EE * EE_Hidden];
    for (int o1 = 0; o1 < Outdim; o1++) {
      OUT[o1] += H_O_Weight_all[barrelstart * Nfiles_EE * EE_Outdim * EE_Hidden + h * Outdim + o1 +
                                sel_wfile * Outdim * Hidden] *
                 Activation_fun(I_SUM[h]);
    }
  }
  for (int o2 = 0; o2 < Outdim; o2++) {
    OUT[o2] += O_Thresh_all[barrelstart * Nfiles_EE * EE_Outdim + o2 + sel_wfile * Outdim];
  }
  nnout = Activation_fun(OUT[0]);
  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: getNNoutput :: -> NNout = " << nnout;

  return (nnout);
}

goodPi0Strip()

bool EndcapPiZeroDiscriminatorAlgo::goodPi0Strip	(	RecHitsMap::iterator	candidate_it,
		ESDetId	lastID
	)

Definition at line 121 of file EndcapPiZeroDiscriminatorAlgo.cc.

References LogTrace, and preshStripEnergyCut_.

Referenced by findPreshVector().

                                                                                                {
  RecHitsMap::iterator candidate_tmp = candidate_it;
  candidate_tmp--;

  // crystal should not be included...
  if (candidate_tmp->first == lastID)  // ...if it corresponds to a hit
  {
    LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: goodPi0Strip No such a strip in rechits_map ";
    return false;
  } else if (candidate_it->second.energy() <= preshStripEnergyCut_)  // ...if it has a negative or zero energy
  {
    LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: goodPi0Strip Strip energy "
                             << candidate_it->second.energy() << " is below threshold ";
    return false;
  }

  return true;
}

readWeightFile()

void EndcapPiZeroDiscriminatorAlgo::readWeightFile ( const char *  WFile, int &  Layers, int &  Indim, int &  Hidden, int &  Outdim  )

private

Definition at line 228 of file EndcapPiZeroDiscriminatorAlgo.cc.

References cms::cuda::assert(), Exception, H_O_Weight_all, H_Thresh_all, mps_fire::i, I_H_Weight_all, dqmiolumiharvest::j, mps_splice::line, LogTrace, O_Thresh_all, and hltDeepSecondaryVertexTagInfosPFPuppi_cfi::weights.

                                                                                 {
  FILE* weights = nullptr;

  char line[80];

  bool checkinit = false;
  // Open the weights file, generated by jetnet, and read
  // in the nodes and weights
  //*******************************************************
  weights = fopen(Weights_file, "r");
  LogTrace("EcalClusters") << "EndcapPiZeroDiscriminatorAlgo: I opeded the Weights file  = " << Weights_file;
  if (weights == nullptr) {
    throw cms::Exception("MissingWeightFile") << "Could not open the weights file: " << Weights_file;
  }

  const auto I_H_W_offset = I_H_Weight_all.size();
  const auto H_O_W_offset = H_O_Weight_all.size();
  const auto H_T_offset = H_Thresh_all.size();
  const auto O_T_offset = O_Thresh_all.size();

  while (!feof(weights)) {
    fscanf(weights, "%s", line);
    if (line[0] == 'A') {              //Read in ANN nodes: Layers, input , Hidden, Output
      fscanf(weights, "%d", &Layers);  // # of NN Layers  used
      fscanf(weights, "%d", &Indim);   // # of Inputs actually used
      fscanf(weights, "%d", &Hidden);  // # of hidden nodes
      fscanf(weights, "%d", &Outdim);  // # of output nodes

      I_H_Weight_all.resize(I_H_W_offset + Indim * Hidden);
      H_Thresh_all.resize(H_T_offset + Hidden);
      H_O_Weight_all.resize(H_O_W_offset + Hidden * Outdim);
      O_Thresh_all.resize(O_T_offset + Outdim);
      checkinit = true;
    } else if (line[0] == 'B') {  // read in weights between hidden and intput nodes
      assert(checkinit);
      for (int i = 0; i < Indim; i++) {
        for (int j = 0; j < Hidden; j++) {
          fscanf(weights, "%f", &I_H_Weight_all[I_H_W_offset + i * Hidden + j]);
        }
      }
    } else if (line[0] == 'C') {  // Read in the thresholds for hidden nodes
      assert(checkinit);
      for (int i = 0; i < Hidden; i++) {
        fscanf(weights, "%f", &H_Thresh_all[H_T_offset + i]);
      }
    } else if (line[0] == 'D') {  // read in weights between hidden and output nodes
      assert(checkinit);
      for (int i = 0; i < Hidden * Outdim; i++) {
        fscanf(weights, "%f", &H_O_Weight_all[H_O_W_offset + i]);
      }
    } else if (line[0] == 'E') {  // read in the threshold for the output nodes
      assert(checkinit);
      for (int i = 0; i < Outdim; i++) {
        fscanf(weights, "%f", &O_Thresh_all[O_T_offset + i]);
      }
    } else {
      edm::LogError("EEPi0Discrim") << "EndcapPiZeroDiscriminatorAlgo: Not a Net file of Corrupted Net file " << endl;
    }
  }
  fclose(weights);
}

Member Data Documentation

debugLevel_

int EndcapPiZeroDiscriminatorAlgo::debugLevel_

private

Definition at line 59 of file EndcapPiZeroDiscriminatorAlgo.h.

EB_Hidden

int EndcapPiZeroDiscriminatorAlgo::EB_Hidden

private

Definition at line 62 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by GetBarrelNNOutput().

EB_Indim

int EndcapPiZeroDiscriminatorAlgo::EB_Indim

private

Definition at line 62 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by calculateBarrelNNInputVariables(), and GetBarrelNNOutput().

EB_Layers

int EndcapPiZeroDiscriminatorAlgo::EB_Layers

private

Definition at line 62 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by GetBarrelNNOutput().

EB_Outdim

int EndcapPiZeroDiscriminatorAlgo::EB_Outdim

private

Definition at line 62 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by GetBarrelNNOutput().

EE_Hidden

int EndcapPiZeroDiscriminatorAlgo::EE_Hidden

private

Definition at line 61 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by GetNNOutput(), and getNNoutput().

EE_Indim

int EndcapPiZeroDiscriminatorAlgo::EE_Indim

private

Definition at line 61 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by calculateNNInputVariables(), GetNNOutput(), and getNNoutput().

EE_Layers

int EndcapPiZeroDiscriminatorAlgo::EE_Layers

private

Definition at line 61 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by GetNNOutput().

EE_Outdim

int EndcapPiZeroDiscriminatorAlgo::EE_Outdim

private

Definition at line 61 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by GetNNOutput(), and getNNoutput().

H_O_Weight_all

std::vector<float> EndcapPiZeroDiscriminatorAlgo::H_O_Weight_all

private

Definition at line 65 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by getNNoutput(), and readWeightFile().

H_Thresh_all

std::vector<float> EndcapPiZeroDiscriminatorAlgo::H_Thresh_all

private

Definition at line 66 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by getNNoutput(), and readWeightFile().

I_H_Weight_all

std::vector<float> EndcapPiZeroDiscriminatorAlgo::I_H_Weight_all

private

Definition at line 64 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by getNNoutput(), and readWeightFile().

input_var

std::vector<float> EndcapPiZeroDiscriminatorAlgo::input_var

private

Definition at line 69 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by calculateBarrelNNInputVariables(), calculateNNInputVariables(), get_input_vector(), GetBarrelNNOutput(), GetNNOutput(), and getNNoutput().

O_Thresh_all

std::vector<float> EndcapPiZeroDiscriminatorAlgo::O_Thresh_all

private

Definition at line 67 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by getNNoutput(), and readWeightFile().

preshSeededNstr_

int EndcapPiZeroDiscriminatorAlgo::preshSeededNstr_

private

Definition at line 58 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by findPi0Road().

preshStripEnergyCut_

double EndcapPiZeroDiscriminatorAlgo::preshStripEnergyCut_

private

Definition at line 57 of file EndcapPiZeroDiscriminatorAlgo.h.

Referenced by goodPi0Strip().