#include "DQM/Physics/src/EwkElecDQM.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/Handle.h"

#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"

#include "DQMServices/Core/interface/DQMStore.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

//#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"  // I guess this is the right one??
// also need Fwd.h file ???
#include "DataFormats/METReco/interface/MET.h"
#include "DataFormats/JetReco/interface/Jet.h"

#include "DataFormats/GeometryVector/interface/Phi.h"

#include "FWCore/Common/interface/TriggerNames.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/Common/interface/TriggerResults.h"

#include "DataFormats/Common/interface/View.h"

#include "HLTrigger/HLTcore/interface/HLTConfigProvider.h"

using namespace edm;
using namespace std;
using namespace reco;

EwkElecDQM::EwkElecDQM(const ParameterSet& cfg)
    :  // Input collections
      metTag_(cfg.getUntrackedParameter<edm::InputTag>("METTag", edm::InputTag("met"))),
      jetTag_(cfg.getUntrackedParameter<edm::InputTag>("JetTag", edm::InputTag("sisCone5CaloJets"))),
      //      trigTag_(consumes<edm::TriggerResults>(
      //        cfg.getUntrackedParameter<edm::InputTag>(
      //         "TrigTag", edm::InputTag("TriggerResults::HLT")))),
      elecTag_(consumes<edm::View<reco::GsfElectron> >(
          cfg.getUntrackedParameter<edm::InputTag>("ElecTag", edm::InputTag("gsfElectrons")))),
      metToken_(
          consumes<edm::View<reco::MET> >(cfg.getUntrackedParameter<edm::InputTag>("METTag", edm::InputTag("met")))),
      jetToken_(consumes<edm::View<reco::Jet> >(
          cfg.getUntrackedParameter<edm::InputTag>("JetTag", edm::InputTag("sisCone5CaloJets")))),
      vertexTag_(consumes<edm::View<reco::Vertex> >(
          cfg.getUntrackedParameter<edm::InputTag>("VertexTag", edm::InputTag("offlinePrimaryVertices")))),
      beamSpotTag_(
          consumes<reco::BeamSpot>(cfg.getUntrackedParameter<edm::InputTag>("BeamSpotTag", edm::InputTag("BeamSpot")))),

      // Main cuts
      //      muonTrig_(cfg.getUntrackedParameter<std::string> ("MuonTrig",
      // "HLT_Mu9")),
      // elecTrig_(cfg.getUntrackedParameter<std::vector< std::string >
      // >("ElecTrig", "HLT_Ele10_SW_L1R")),
      elecTrig_(cfg.getUntrackedParameter<std::vector<std::string> >("ElecTrig")),
      //      ptCut_(cfg.getUntrackedParameter<double>("PtCut", 25.)),
      ptCut_(cfg.getUntrackedParameter<double>("PtCut", 10.)),
      //      etaCut_(cfg.getUntrackedParameter<double>("EtaCut", 2.1)),
      etaCut_(cfg.getUntrackedParameter<double>("EtaCut", 2.4)),
      sieieCutBarrel_(cfg.getUntrackedParameter<double>("SieieBarrel", 0.01)),
      sieieCutEndcap_(cfg.getUntrackedParameter<double>("SieieEndcap", 0.028)),
      detainCutBarrel_(cfg.getUntrackedParameter<double>("DetainBarrel", 0.0071)),
      detainCutEndcap_(cfg.getUntrackedParameter<double>("DetainEndcap", 0.0066)),
      //      isRelativeIso_(cfg.getUntrackedParameter<bool>("IsRelativeIso",
      // true)),
      //      isCombinedIso_(cfg.getUntrackedParameter<bool>("IsCombinedIso",
      // false)),
      //      isoCut03_(cfg.getUntrackedParameter<double>("IsoCut03", 0.1)),
      ecalIsoCutBarrel_(cfg.getUntrackedParameter<double>("EcalIsoCutBarrel", 5.7)),
      ecalIsoCutEndcap_(cfg.getUntrackedParameter<double>("EcalIsoCutEndcap", 5.0)),
      hcalIsoCutBarrel_(cfg.getUntrackedParameter<double>("HcalIsoCutBarrel", 8.1)),
      hcalIsoCutEndcap_(cfg.getUntrackedParameter<double>("HcalIsoCutEndcap", 3.4)),
      trkIsoCutBarrel_(cfg.getUntrackedParameter<double>("TrkIsoCutBarrel", 7.2)),
      trkIsoCutEndcap_(cfg.getUntrackedParameter<double>("TrkIsoCutEndcap", 5.1)),
      mtMin_(cfg.getUntrackedParameter<double>("MtMin", -999999)),
      mtMax_(cfg.getUntrackedParameter<double>("MtMax", 999999.)),
      metMin_(cfg.getUntrackedParameter<double>("MetMin", -999999.)),
      metMax_(cfg.getUntrackedParameter<double>("MetMax", 999999.)),
      //      acopCut_(cfg.getUntrackedParameter<double>("AcopCut", 2.)),

      // Muon quality cuts
      //      dxyCut_(cfg.getUntrackedParameter<double>("DxyCut", 0.2)),
      //      normalizedChi2Cut_(cfg.getUntrackedParameter<double>("NormalizedChi2Cut",
      // 10.)),
      //      trackerHitsCut_(cfg.getUntrackedParameter<int>("TrackerHitsCut",
      // 11)),
      //      isAlsoTrackerMuon_(cfg.getUntrackedParameter<bool>("IsAlsoTrackerMuon",
      // true)),

      // Z rejection
      //      ptThrForZ1_(cfg.getUntrackedParameter<double>("PtThrForZ1", 20.)),
      //      ptThrForZ2_(cfg.getUntrackedParameter<double>("PtThrForZ2", 10.)),

      // Top rejection
      eJetMin_(cfg.getUntrackedParameter<double>("EJetMin", 999999.)),
      nJetMax_(cfg.getUntrackedParameter<int>("NJetMax", 999999)),
      PUMax_(cfg.getUntrackedParameter<unsigned int>("PUMax", 60)),
      PUBinCount_(cfg.getUntrackedParameter<unsigned int>("PUBinCount", 12)),
      hltPrescaleProvider_(cfg, consumesCollector(), *this)
//       caloJetCollection_(cfg.getUntrackedParameter<edm:InputTag>("CaloJetCollection","sisCone5CaloJets"))
{
  isValidHltConfig_ = false;
}

void EwkElecDQM::dqmBeginRun(const Run& iRun, const EventSetup& iSet) {
  nall = 0;
  nsel = 0;

  nrec = 0;
  neid = 0;
  niso = 0;
  //       nhlt = 0;
  //       nmet = 0;

  // passed as parameter to HLTConfigProvider::init(), not yet used
  bool isConfigChanged = false;
  // isValidHltConfig_ could be used to short-circuit analyze() in case of
  // problems
  isValidHltConfig_ = hltPrescaleProvider_.init(iRun, iSet, "HLT", isConfigChanged);

  LogTrace("") << "isValidHltConfig_=" << isValidHltConfig_ << "\n";
}

void EwkElecDQM::bookHistograms(DQMStore::IBooker& ibooker, edm::Run const&, edm::EventSetup const&) {
  ibooker.setCurrentFolder("Physics/EwkElecDQM");

  char chtitle[256] = "";

  pt_before_ = ibooker.book1D("PT_BEFORECUTS", "Electron transverse momentum [GeV]", 100, 0., 100.);
  pt_after_ = ibooker.book1D("PT_LASTCUT", "Electron transverse momentum [GeV]", 100, 0., 100.);

  eta_before_ = ibooker.book1D("ETA_BEFORECUTS", "Electron pseudo-rapidity", 50, -2.5, 2.5);
  eta_after_ = ibooker.book1D("ETA_LASTCUT", "Electron pseudo-rapidity", 50, -2.5, 2.5);

  sieiebarrel_before_ = ibooker.book1D("SIEIEBARREL_BEFORECUTS", "Electron #sigma_{i#etai#eta} (barrel)", 70, 0., 0.07);
  sieiebarrel_after_ = ibooker.book1D("SIEIEBARREL_LASTCUT", "Electron #sigma_{i#etai#eta} (barrel)", 70, 0., 0.07);

  sieieendcap_before_ = ibooker.book1D("SIEIEENDCAP_BEFORECUTS", "Electron #sigma_{i#etai#eta} (endcap)", 70, 0., 0.07);
  sieieendcap_after_ = ibooker.book1D("SIEIEENDCAP_LASTCUT", "Electron #sigma_{i#etai#eta} (endcap)", 70, 0., 0.07);

  detainbarrel_before_ =
      ibooker.book1D("DETAINBARREL_BEFORECUTS", "Electron #Delta#eta_{in} (barrel)", 40, -0.02, 0.02);
  detainbarrel_after_ = ibooker.book1D("DETAINBARREL_LASTCUT", "Electron #Delta#eta_{in} (barrel)", 40, -0.02, 0.02);

  detainendcap_before_ =
      ibooker.book1D("DETAINENDCAP_BEFORECUTS", "Electron #Delta#eta_{in} (endcap)", 40, -0.02, 0.02);
  detainendcap_after_ = ibooker.book1D("DETAINENDCAP_LASTCUT", "Electron #Delta#eta_{in} (endcap)", 40, -0.02, 0.02);

  ecalisobarrel_before_ = ibooker.book1D(
      "ECALISOBARREL_BEFORECUTS", "Absolute electron ECAL isolation variable (barrel) [GeV]", 50, 0., 50.);
  ecalisobarrel_after_ =
      ibooker.book1D("ECALISOBARREL_LASTCUT", "Absolute electron ECAL isolation variable (barrel) [GeV]", 50, 0., 50.);

  ecalisoendcap_before_ = ibooker.book1D(
      "ECALISOENDCAP_BEFORECUTS", "Absolute electron ECAL isolation variable (endcap) [GeV]", 50, 0., 50.);
  ecalisoendcap_after_ =
      ibooker.book1D("ECALISOENDCAP_LASTCUT", "Absolute electron ECAL isolation variable (endcap) [GeV]", 50, 0., 50.);

  hcalisobarrel_before_ = ibooker.book1D(
      "HCALISOBARREL_BEFORECUTS", "Absolute electron HCAL isolation variable (barrel) [GeV]", 50, 0., 50.);
  hcalisobarrel_after_ =
      ibooker.book1D("HCALISOBARREL_LASTCUT", "Absolute electron HCAL isolation variable (barrel) [GeV]", 50, 0., 50.);

  hcalisoendcap_before_ = ibooker.book1D(
      "HCALISOENDCAP_BEFORECUTS", "Absolute electron HCAL isolation variable (endcap) [GeV]", 50, 0., 50.);
  hcalisoendcap_after_ =
      ibooker.book1D("HCALISOENDCAP_LASTCUT", "Absolute electron HCAL isolation variable (endcap) [GeV]", 50, 0., 50.);

  trkisobarrel_before_ = ibooker.book1D(
      "TRKISOBARREL_BEFORECUTS", "Absolute electron track isolation variable (barrel) [GeV]", 50, 0., 50.);
  trkisobarrel_after_ =
      ibooker.book1D("TRKISOBARREL_LASTCUT", "Absolute electron track isolation variable (barrel) [GeV]", 50, 0., 50.);

  trkisoendcap_before_ = ibooker.book1D(
      "TRKISOENDCAP_BEFORECUTS", "Absolute electron track isolation variable (endcap) [GeV]", 50, 0., 50.);
  trkisoendcap_after_ =
      ibooker.book1D("TRKISOENDCAP_LASTCUT", "Absolute electron track isolation variable (endcap) [GeV]", 50, 0., 50.);

  // trig_before_ = ibooker.book1D("TRIG_BEFORECUTS", "Trigger response", 2, -0.5,
  /// 1.5);  // elecTrig_ is now a vector of strings!
  //  trig_after_ = ibooker.book1D("TRIG_LASTCUT", "Trigger response", 2, -0.5, 1.5);

  invmass_before_ = ibooker.book1D("INVMASS_BEFORECUTS", "Di-electron invariant mass [GeV]", 100, 0., 200.);
  invmass_after_ = ibooker.book1D("INVMASS_AFTERCUTS", "Di-electron invariant mass [GeV]", 100, 0., 200.);

  invmassPU_before_ = ibooker.book2D("INVMASS_PU_BEFORECUTS",
                                     "Di-electron invariant mass [GeV] vs PU; mass [GeV]; PU count",
                                     100,
                                     0.,
                                     200.,
                                     PUBinCount_,
                                     -0.5,
                                     PUMax_ + 0.5);
  invmassPU_afterZ_ = ibooker.book2D("INVMASS_PU_AFTERZCUTS",
                                     "Di-electron invariant mass [GeV] vs PU; mass [GeV]; PU count",
                                     100,
                                     0.,
                                     200.,
                                     PUBinCount_,
                                     -0.5,
                                     PUMax_ + 0.5);

  npvs_before_ =
      ibooker.book1D("NPVs_BEFORECUTS", "Number of Valid Primary Vertices; nGoodPVs", PUMax_ + 1, -0.5, PUMax_ + 0.5);

  npvs_afterZ_ =
      ibooker.book1D("NPVs_AFTERZCUTS", "Number of Valid Primary Vertices; nGoodPVs", PUMax_ + 1, -0.5, PUMax_ + 0.5);

  nelectrons_before_ = ibooker.book1D("NELECTRONS_BEFORECUTS", "Number of electrons in event", 10, -0.5, 9.5);
  nelectrons_after_ = ibooker.book1D("NELECTRONS_AFTERCUTS", "Number of electrons in event", 10, -0.5, 9.5);

  snprintf(chtitle, 255, "Transverse mass (%s) [GeV]", metTag_.label().data());
  mt_before_ = ibooker.book1D("MT_BEFORECUTS", chtitle, 150, 0., 300.);
  mt_after_ = ibooker.book1D("MT_LASTCUT", chtitle, 150, 0., 300.);

  snprintf(chtitle, 255, "Missing transverse energy (%s) [GeV]", metTag_.label().data());
  met_before_ = ibooker.book1D("MET_BEFORECUTS", chtitle, 100, 0., 200.);
  met_after_ = ibooker.book1D("MET_LASTCUT", chtitle, 100, 0., 200.);

  snprintf(chtitle, 255, "Number of jets (%s) above %.2f GeV", jetTag_.label().data(), eJetMin_);
  njets_before_ = ibooker.book1D("NJETS_BEFORECUTS", chtitle, 10, -0.5, 9.5);
  njets_after_ = ibooker.book1D("NJETS_LASTCUT", chtitle, 10, -0.5, 9.5);

  snprintf(chtitle, 255, "Jet with highest E_{T} (%s)", jetTag_.label().data());
  jet_et_before_ = ibooker.book1D("JETET1_BEFORECUTS", chtitle, 20, 0., 200.0);
  jet_et_after_ = ibooker.book1D("JETET1_AFTERCUTS", chtitle, 20, 0., 200.0);

  snprintf(chtitle, 255, "Eta of Jet with highest E_{T} (%s)", jetTag_.label().data());
  jet_eta_before_ = ibooker.book1D("JETETA1_BEFORECUTS", chtitle, 20, -5, 5);
  jet_eta_after_ = ibooker.book1D("JETETA1_AFTERCUTS", chtitle, 20, -5, 5);
}

void EwkElecDQM::dqmEndRun(const Run& r, const EventSetup&) {
  // overall
  double all = nall;
  double esel = nsel / all;
  LogVerbatim("") << "\n>>>>>> SELECTION SUMMARY BEGIN >>>>>>>>>>>>>>>";
  LogVerbatim("") << "Total number of events analyzed: " << nall << " [events]";
  LogVerbatim("") << "Total number of events selected: " << nsel << " [events]";
  LogVerbatim("") << "Overall efficiency:             "
                  << "(" << setprecision(4) << esel * 100. << " +/- " << setprecision(2)
                  << sqrt(esel * (1 - esel) / all) * 100. << ")%";

  double erec = nrec / all;
  double eeid = neid / all;
  double eiso = niso / all;
  //   double ehlt = nhlt/all;
  //   double emet = nmet/all;

  // general reconstruction step??
  double num = nrec;
  double eff = erec;
  double err = sqrt(eff * (1 - eff) / all);
  LogVerbatim("") << "Passing Pt/Eta/Quality cuts:    " << num << " [events], (" << setprecision(4) << eff * 100.
                  << " +/- " << setprecision(2) << err * 100. << ")%";

  // electron ID step
  num = neid;
  eff = eeid;
  err = sqrt(eff * (1 - eff) / all);
  double effstep = 0.;
  double errstep = 0.;
  if (nrec > 0)
    effstep = eeid / erec;
  if (nrec > 0)
    errstep = sqrt(effstep * (1 - effstep) / nrec);
  LogVerbatim("") << "Passing eID cuts:         " << num << " [events], (" << setprecision(4) << eff * 100. << " +/- "
                  << setprecision(2) << err * 100. << ")%, to previous step: (" << setprecision(4) << effstep * 100.
                  << " +/- " << setprecision(2) << errstep * 100. << ")%";

  // isolation step
  num = niso;
  eff = eiso;
  err = sqrt(eff * (1 - eff) / all);
  effstep = 0.;
  errstep = 0.;
  if (neid > 0)
    effstep = eiso / eeid;
  if (neid > 0)
    errstep = sqrt(effstep * (1 - effstep) / neid);
  LogVerbatim("") << "Passing isolation cuts:         " << num << " [events], (" << setprecision(4) << eff * 100.
                  << " +/- " << setprecision(2) << err * 100. << ")%, to previous step: (" << setprecision(4)
                  << effstep * 100. << " +/- " << setprecision(2) << errstep * 100. << ")%";

  //   // trigger step
  //   num = nhlt;
  //   eff = ehlt;
  //   err = sqrt(eff*(1-eff)/all);
  //   effstep = 0.;
  //   errstep = 0.;
  //   if (niso>0) effstep = ehlt/eiso;
  //   if (niso>0) errstep = sqrt(effstep*(1-effstep)/niso);
  //   LogVerbatim("") << "Passing HLT criteria:           " << num << "
  // [events], (" << setprecision(4) << eff*100. <<" +/- "<< setprecision(2) <<
  // err*100. << ")%, to previous step: (" <<  setprecision(4) << effstep*100.
  // << " +/- "<< setprecision(2) << errstep*100. <<")%";

  // trigger step
  num = nsel;
  eff = esel;
  err = sqrt(eff * (1 - eff) / all);
  effstep = 0.;
  errstep = 0.;
  if (niso > 0)
    effstep = esel / eiso;
  if (niso > 0)
    errstep = sqrt(effstep * (1 - effstep) / niso);
  LogVerbatim("") << "Passing HLT criteria:           " << num << " [events], (" << setprecision(4) << eff * 100.
                  << " +/- " << setprecision(2) << err * 100. << ")%, to previous step: (" << setprecision(4)
                  << effstep * 100. << " +/- " << setprecision(2) << errstep * 100. << ")%";

  //   // met/acoplanarity cuts
  //   num = nmet;
  //   eff = emet;
  //   err = sqrt(eff*(1-eff)/all);
  //   effstep = 0.;
  //   errstep = 0.;
  //   if (nhlt>0) effstep = emet/ehlt;
  //   if (nhlt>0) errstep = sqrt(effstep*(1-effstep)/nhlt);
  //   LogVerbatim("") << "Passing MET/acoplanarity cuts:  " << num << "
  // [events], (" << setprecision(4) << eff*100. <<" +/- "<< setprecision(2) <<
  // err*100. << ")%, to previous step: (" <<  setprecision(4) << effstep*100.
  // << " +/- "<< setprecision(2) << errstep*100. <<")%";

  //   // Z/top selection cuts ALSO LAST STEP so "sel" for "selection"
  //   num = nsel;
  //   eff = esel;
  //   err = sqrt(eff*(1-eff)/all);
  //   effstep = 0.;
  //   errstep = 0.;
  //   if (nmet>0) effstep = esel/emet;
  //   if (nmet>0) errstep = sqrt(effstep*(1-effstep)/nmet);
  //   LogVerbatim("") << "Passing Z/top rejection cuts:   " << num << "
  // [events], (" << setprecision(4) << eff*100. <<" +/- "<< setprecision(2) <<
  // err*100. << ")%, to previous step: (" <<  setprecision(4) << effstep*100.
  // << " +/- "<< setprecision(2) << errstep*100. <<")%";

  LogVerbatim("") << ">>>>>> SELECTION SUMMARY END   >>>>>>>>>>>>>>>\n";
}

inline void HERE(const char* msg) { std::cout << msg << "\n"; }

void EwkElecDQM::analyze(const Event& ev, const EventSetup& iSet) {
  // Reset global event selection flags
  bool rec_sel = false;
  bool eid_sel = false;
  bool iso_sel = false;
  bool all_sel = false;

  // Electron collection
  Handle<View<GsfElectron> > electronCollection;
  if (!ev.getByToken(elecTag_, electronCollection)) {
    // LogWarning("") << ">>> Electron collection does not exist !!!";
    return;
  }
  unsigned int electronCollectionSize = electronCollection->size();

  // Beam spot
  Handle<reco::BeamSpot> beamSpotHandle;
  if (!ev.getByToken(beamSpotTag_, beamSpotHandle)) {
    // LogWarning("") << ">>> No beam spot found !!!";
    return;
  }

  // MET
  double met_px = 0.;
  double met_py = 0.;
  Handle<View<MET> > metCollection;
  if (!ev.getByToken(metToken_, metCollection)) {
    // LogWarning("") << ">>> MET collection does not exist !!!";
    return;
  }

  const MET& met = metCollection->at(0);
  met_px = met.px();
  met_py = met.py();
  //       if (!metIncludesMuons_) {
  //             for (unsigned int i=0; i<muonCollectionSize; i++) {
  //                   const Muon& mu = muonCollection->at(i);
  //                   if (!mu.isGlobalMuon()) continue;
  //                   met_px -= mu.px();
  //                   met_py -= mu.py();
  //             }
  //       }
  double met_et = sqrt(met_px * met_px + met_py * met_py);
  LogTrace("") << ">>> MET, MET_px, MET_py: " << met_et << ", " << met_px << ", " << met_py << " [GeV]";
  met_before_->Fill(met_et);

  // Vertices in the event
  int npvCount = 0;
  Handle<View<reco::Vertex> > vertexCollection;
  if (!ev.getByToken(vertexTag_, vertexCollection)) {
    LogError("") << ">>> Vertex collection does not exist !!!";
    return;
  }

  for (unsigned int i = 0; i < vertexCollection->size(); i++) {
    const Vertex& vertex = vertexCollection->at(i);
    if (vertex.isValid())
      npvCount++;
  }
  npvs_before_->Fill(npvCount);

  // Trigger
  // Handle<TriggerResults> triggerResults;
  // if (!ev.getByToken(trigTag_, triggerResults)) {
  // LogWarning("") << ">>> TRIGGER collection does not exist !!!";
  return;
  // }
  // const edm::TriggerNames& trigNames = ev.triggerNames(*triggerResults);
  // bool trigger_fired = false;

  // HLTConfigProvider const&  hltConfigProvider = hltPrescaleProvider_.hltConfigProvider();

  /* very old code
  for (unsigned int i=0; i<triggerResults->size(); i++) {
        if (triggerResults->accept(i)) {
              LogTrace("") << "Accept by: " << i << ", Trigger: " <<
  trigNames.triggerName(i);
        }
  }
  */

  // the following gives error on CRAFT08 data where itrig1=19 (vector index out
  // of range)
  /*
  int itrig1 = trigNames.triggerIndex(muonTrig_);
  if (triggerResults->accept(itrig1)) trigger_fired = true;
  */
  // suggested replacement: lm250909
  /*  Fix buggy trigger logic
  for (unsigned int i=0; i<triggerResults->size(); i++)
    {
      std::string trigName = trigNames.triggerName(i);
      bool found=false;

// 	  for (unsigned int j = 0; j < elecTrig_.size(); j++)
// 	    {
// 	      if ( trigName == elecTrig_.at(j) && triggerResults->accept(i))
// 		{
// 		  trigger_fired = true;
// 		}
// 	    }


     for(unsigned int index=0; index<elecTrig_.size() && found==false; index++)
{
       size_t trigPath = trigName.find(elecTrig_.at(index)); // 0 if found, pos
if not
               if (trigPath==0) found=true;
          }
          if(!found) continue;

          bool prescaled=false;
          for (unsigned int ps= 0; ps<  hltConfigProvider.prescaleSize();
ps++){
              const unsigned int prescaleValue =
hltConfigProvider.prescaleValue(ps, trigName) ;
              if (prescaleValue != 1) prescaled =true;
          }

          if(triggerResults->accept(i) && !prescaled) trigger_fired=true;

    }
  */

  // get the prescale set for this event
  const int prescaleSet = hltPrescaleProvider_.prescaleSet(ev, iSet);
  if (prescaleSet == -1) {
    LogTrace("") << "Failed to determine prescaleSet\n";
    // std::cout << "Failed to determine prescaleSet. Check cmsRun GlobalTag\n";
    return;
  }

  // for (unsigned int i = 0;
  //    (i < triggerResults->size()) && (trigger_fired == false); i++) {
  // skip trigger, if it did not fire
  //  if (!triggerResults->accept(i)) continue;

  // skip trigger, if it is not on our list
  // bool found = false;
  // const std::string trigName = trigNames.triggerName(i);
  // for (unsigned int index = 0; index < elecTrig_.size() && found == false;
  //    index++) {
  // if (trigName.find(elecTrig_.at(index)) == 0) found = true;
  // }
  // if (!found) continue;

  // skip trigger, if it is prescaled
  // if (hltConfigProvider.prescaleValue(prescaleSet, trigName) != 1) continue;

  // std::cout << "found unprescaled trigger that fired: " << trigName <<
  // "\n";
  // trigger_fired = true;
  // }

  /*  LogTrace("") << ">>> Trigger bit: " << trigger_fired << " for one of ( ";
  for (unsigned int k = 0; k < elecTrig_.size(); k++) {
    LogTrace("") << elecTrig_.at(k) << " ";
  }
  LogTrace("") << ")";
  trig_before_->Fill(trigger_fired);
*/
  // Jet collection
  Handle<View<Jet> > jetCollection;
  if (!ev.getByToken(jetToken_, jetCollection)) {
    // LogError("") << ">>> JET collection does not exist !!!";
    return;
  }
  float electron_et = -8.0;
  float electron_eta = -8.0;
  float electron_phi = -8.0;
  float electron2_et = -9.0;
  float electron2_eta = -9.0;
  float electron2_phi = -9.0;
  // need to get some electron info so jets can be cleaned of them
  for (unsigned int i = 0; i < electronCollectionSize; i++) {
    const GsfElectron& elec = electronCollection->at(i);

    if (i < 1) {
      electron_et = elec.pt();
      electron_eta = elec.eta();
      electron_phi = elec.phi();
    }
    if (i == 2) {
      electron2_et = elec.pt();
      electron2_eta = elec.eta();
      electron2_phi = elec.phi();
    }
  }

  float jet_et = -8.0;
  float jet_eta = -8.0;
  float jet2_et = -9.0;
  unsigned int jetCollectionSize = jetCollection->size();
  int njets = 0;
  for (unsigned int i = 0; i < jetCollectionSize; i++) {
    const Jet& jet = jetCollection->at(i);

    float jet_current_et = jet.et();
    // 	cout << "jet_current_et " << jet_current_et << endl;
    // if it overlaps with electron, it is not a jet
    if (electron_et > 0.0 && fabs(jet.eta() - electron_eta) < 0.2 && calcDeltaPhi(jet.phi(), electron_phi) < 0.2)
      continue;
    if (electron2_et > 0.0 && fabs(jet.eta() - electron2_eta) < 0.2 && calcDeltaPhi(jet.phi(), electron2_phi) < 0.2)
      continue;

    // if it has too low Et, throw away
    // 	if (jet_current_et < eJetMin_) continue; //Keep if only want to plot
    // above jet cut

    if (jet.et() > eJetMin_) {
      njets++;
    }
    if (jet_current_et > jet_et) {
      jet2_et = jet_et;   // 2nd highest jet get's et from current highest
      jet_et = jet.et();  // current highest jet gets et from the new highest
      jet_eta = jet.eta();
    } else if (jet_current_et > jet2_et) {
      jet2_et = jet.et();
    }
  }

  // Fill After all electron cuts (or both before and after)
  if (jet_et > 10)  // don't want low energy "jets"
  {
    jet_et_before_->Fill(jet_et);
    // 	  jet2_et_before_  ->Fill(jet2_et);
    jet_eta_before_->Fill(jet_eta);
  }

  LogTrace("") << ">>> Total number of jets: " << jetCollectionSize;
  LogTrace("") << ">>> Number of jets above " << eJetMin_ << " [GeV]: " << njets;
  njets_before_->Fill(njets);

  // Start counting
  nall++;

  // Histograms per event should be done only once, so keep track of them
  //bool hlt_hist_done = false;
  // bool minv_hist_done = false;
  bool met_hist_done = false;
  //       bool nz1_hist_done = false;
  //       bool nz2_hist_done = false;
  bool njets_hist_done = false;

  // Central selection criteria
  // const int NFLAGS = 13; // number of individual selection criteria
  const int NFLAGS = 10;  // number of individual selection criteria
  // 0: pt cut           | rec
  // 1: eta cut          | rec
  // 2: sieie            | eid
  // 3: detain           | eid
  // 4: ecal iso         | iso
  // 5: hcal iso         | iso
  // 6: trk iso          | iso
  // 7: trigger fired    | hlt/all
  bool electron_sel[NFLAGS];

  // for invariant mass calculation
  // keep track of highest-pt electrons for initial (RECO) electrons
  // and "good" electrons (passing all cuts)
  // first array dimension is for first or second good electron
  // second array dimension is for relevant quantities of good electron
  //    [0]: 1 for electron found or 0 for not found (if 0, no other quantities
  // filled)
  //    [1]: mSqr
  //    [2]: E
  //    [3]: px
  //    [4]: py
  //    [5]: pz
  // inv mass = sqrt(m_1^2 + m_2^2 + 2*(E_1*E_2 - (px1*px2 + py1*py2 + pz1+pz2)
  // ) )
  double electron[2][6];
  double goodElectron[2][6];
  nGoodElectrons = 0;
  for (unsigned int i = 0; i < 2; i++) {
    for (unsigned int j = 0; j < 6; j++) {
      electron[i][j] = 0.;
      goodElectron[i][j] = 0.;
    }
  }

  for (unsigned int i = 0; i < electronCollectionSize; i++) {
    for (int j = 0; j < NFLAGS; ++j) {
      electron_sel[j] = false;
    }

    const GsfElectron& elec = electronCollection->at(i);
    // if (!mu.isGlobalMuon()) continue;
    // if (mu.globalTrack().isNull()) continue;
    // if (mu.innerTrack().isNull()) continue;

    LogTrace("") << "> elec: processing electron number " << i << "...";
    // reco::TrackRef gm = mu.globalTrack();
    // reco::TrackRef tk = mu.innerTrack();
    // should have stuff for electron track?

    if (i < 2) {
      electron[i][0] = 1.;
      electron[i][1] = elec.massSqr();
      electron[i][2] = elec.energy();
      electron[i][3] = elec.px();
      electron[i][4] = elec.py();
      electron[i][5] = elec.pz();
    }

    // Pt,eta cuts
    double pt = elec.pt();
    double px = elec.px();
    double py = elec.py();
    double eta = elec.eta();
    LogTrace("") << "\t... pt, eta: " << pt << " [GeV], " << eta;
    ;
    if (pt > ptCut_)
      electron_sel[0] = true;
    if (fabs(eta) < etaCut_)
      electron_sel[1] = true;

    bool isBarrel = false;
    bool isEndcap = false;
    if (eta < 1.4442 && eta > -1.4442) {
      isBarrel = true;
    } else if ((eta > 1.56 && eta < 2.4) || (eta < -1.56 && eta > -2.4)) {
      isEndcap = true;
    }

    //             // d0, chi2, nhits quality cuts
    //             double dxy = tk->dxy(beamSpotHandle->position());
    //             double normalizedChi2 = gm->normalizedChi2();
    //             double trackerHits = tk->numberOfValidHits();
    //             LogTrace("") << "\t... dxy, normalizedChi2, trackerHits,
    // isTrackerMuon?: " << dxy << " [cm], " << normalizedChi2 << ", " <<
    // trackerHits << ", " << mu.isTrackerMuon();
    //             if (fabs(dxy)<dxyCut_) muon_sel[2] = true;
    //             if (normalizedChi2<normalizedChi2Cut_) muon_sel[3] = true;
    //             if (trackerHits>=trackerHitsCut_) muon_sel[4] = true;
    //             if (mu.isTrackerMuon()) muon_sel[5] = true;

    pt_before_->Fill(pt);
    eta_before_->Fill(eta);
    //             dxy_before_->Fill(dxy);
    //             chi2_before_->Fill(normalizedChi2);
    //             nhits_before_->Fill(trackerHits);
    //             tkmu_before_->Fill(mu.isTrackerMuon());

    // Electron ID cuts
    double sieie = (double)elec.sigmaIetaIeta();
    double detain = (double)elec.deltaEtaSuperClusterTrackAtVtx();  // think this is detain
    if (sieie < sieieCutBarrel_ && isBarrel)
      electron_sel[2] = true;
    if (sieie < sieieCutEndcap_ && isEndcap)
      electron_sel[2] = true;
    if (detain < detainCutBarrel_ && isBarrel)
      electron_sel[3] = true;
    if (detain < detainCutEndcap_ && isEndcap)
      electron_sel[3] = true;
    if (isBarrel) {
      LogTrace("") << "\t... sieie value " << sieie << " (barrel), pass? " << electron_sel[2];
      LogTrace("") << "\t... detain value " << detain << " (barrel), pass? " << electron_sel[3];
    } else if (isEndcap) {
      LogTrace("") << "\t... sieie value " << sieie << " (endcap), pass? " << electron_sel[2];
      LogTrace("") << "\t... detain value " << detain << " (endcap), pass? " << electron_sel[2];
    }

    if (isBarrel) {
      sieiebarrel_before_->Fill(sieie);
      detainbarrel_before_->Fill(detain);
    } else if (isEndcap) {
      sieieendcap_before_->Fill(sieie);
      detainendcap_before_->Fill(detain);
    }

    // Isolation cuts
    // double isovar = mu.isolationR03().sumPt;
    double ecalisovar = elec.dr03EcalRecHitSumEt();  // picked one set!
    double hcalisovar = elec.dr03HcalTowerSumEt();   // try others if
    double trkisovar = elec.dr04TkSumPt();           // doesn't work
    // if (isCombinedIso_) {
    // isovar += mu.isolationR03().emEt;
    // isovar += mu.isolationR03().hadEt;
    //}
    // if (isRelativeIso_) isovar /= pt;
    if (ecalisovar < ecalIsoCutBarrel_ && isBarrel)
      electron_sel[4] = true;
    if (ecalisovar < ecalIsoCutEndcap_ && isEndcap)
      electron_sel[4] = true;
    if (hcalisovar < hcalIsoCutBarrel_ && isBarrel)
      electron_sel[5] = true;
    if (hcalisovar < hcalIsoCutEndcap_ && isEndcap)
      electron_sel[5] = true;
    if (trkisovar < trkIsoCutBarrel_ && isBarrel)
      electron_sel[6] = true;
    if (trkisovar < trkIsoCutEndcap_ && isEndcap)
      electron_sel[6] = true;
    if (isBarrel) {
      LogTrace("") << "\t... ecal isolation value " << ecalisovar << " (barrel), pass? " << electron_sel[4];
      LogTrace("") << "\t... hcal isolation value " << hcalisovar << " (barrel), pass? " << electron_sel[5];
      LogTrace("") << "\t... trk isolation value " << trkisovar << " (barrel), pass? " << electron_sel[6];
    } else if (isEndcap) {
      LogTrace("") << "\t... ecal isolation value " << ecalisovar << " (endcap), pass? " << electron_sel[4];
      LogTrace("") << "\t... hcal isolation value " << hcalisovar << " (endcap), pass? " << electron_sel[5];
      LogTrace("") << "\t... trk isolation value " << trkisovar << " (endcap), pass? " << electron_sel[6];
    }

    // iso_before_->Fill(isovar);
    if (isBarrel) {
      ecalisobarrel_before_->Fill(ecalisovar);
      hcalisobarrel_before_->Fill(hcalisovar);
      trkisobarrel_before_->Fill(trkisovar);
    } else if (isEndcap) {
      ecalisoendcap_before_->Fill(ecalisovar);
      hcalisoendcap_before_->Fill(hcalisovar);
      trkisoendcap_before_->Fill(trkisovar);
    }

    // HLT
    // if (trigger_fired) electron_sel[7] = true;

    //             // MET/MT cuts
    double w_et = met_et + pt;
    double w_px = met_px + px;
    double w_py = met_py + py;

    double massT = w_et * w_et - w_px * w_px - w_py * w_py;
    massT = (massT > 0) ? sqrt(massT) : 0;

    LogTrace("") << "\t... W mass, W_et, W_px, W_py: " << massT << ", " << w_et << ", " << w_px << ", " << w_py
                 << " [GeV]";
    if (massT > mtMin_ && massT < mtMax_)
      electron_sel[7] = true;
    mt_before_->Fill(massT);
    if (met_et > metMin_ && met_et < metMax_)
      electron_sel[8] = true;

    //             // Acoplanarity cuts
    //             Geom::Phi<double> deltaphi(mu.phi()-atan2(met_py,met_px));
    //             double acop = deltaphi.value();
    //             if (acop<0) acop = - acop;
    //             acop = M_PI - acop;
    //             LogTrace("") << "\t... acoplanarity: " << acop;
    //             if (acop<acopCut_) muon_sel[10] = true;
    //             acop_before_->Fill(acop);

    //             // Remaining flags (from global event information)
    //             if (nmuonsForZ1<1 || nmuonsForZ2<2) muon_sel[11] = true;
    if (njets <= nJetMax_)
      electron_sel[9] = true;

    // Collect necessary flags "per electron"
    int flags_passed = 0;
    bool rec_sel_this = true;
    bool eid_sel_this = true;
    bool iso_sel_this = true;
    bool all_sel_this = true;
    for (int j = 0; j < NFLAGS; ++j) {
      if (electron_sel[j])
        flags_passed += 1;
      if (j < 2 && !electron_sel[j])
        rec_sel_this = false;
      if (j < 4 && !electron_sel[j])
        eid_sel_this = false;
      if (j < 7 && !electron_sel[j])
        iso_sel_this = false;
      if (!electron_sel[j])
        all_sel_this = false;
    }

    if (all_sel_this) {
      if (nGoodElectrons < 2) {
        goodElectron[nGoodElectrons][0] = 1.;
        goodElectron[nGoodElectrons][1] = elec.massSqr();
        goodElectron[nGoodElectrons][2] = elec.energy();
        goodElectron[nGoodElectrons][3] = elec.px();
        goodElectron[nGoodElectrons][4] = elec.py();
        goodElectron[nGoodElectrons][5] = elec.pz();
      }
      nGoodElectrons++;
    }

    //             // "rec" => pt,eta and quality cuts are satisfied
    //             if (rec_sel_this) rec_sel = true;
    //             // "iso" => "rec" AND "muon is isolated"
    //             if (iso_sel_this) iso_sel = true;
    //             // "hlt" => "iso" AND "event is triggered"
    //             if (hlt_sel_this) hlt_sel = true;
    // 	         // "all" => "met" AND "Z/top rejection cuts"
    //             if (all_sel_this) all_sel = true;

    // "rec" => pt,eta cuts are satisfied
    if (rec_sel_this)
      rec_sel = true;
    // "eid" => "rec" AND "electron passes ID"
    if (eid_sel_this)
      iso_sel = true;
    // "iso" => "eid" AND "electron is isolated"
    if (iso_sel_this)
      iso_sel = true;
    // "met" => "iso" AND "MET/MT"
    // "all" => "met" AND "event is triggered"
    if (all_sel_this)
      all_sel = true;

    // Do N-1 histograms now (and only once for global event quantities)
    if (flags_passed >= (NFLAGS - 1)) {
      if (!electron_sel[0] || flags_passed == NFLAGS) {
        pt_after_->Fill(pt);
      }
      if (!electron_sel[1] || flags_passed == NFLAGS) {
        eta_after_->Fill(eta);
      }
      if (!electron_sel[2] || flags_passed == NFLAGS) {
        if (isBarrel) {
          sieiebarrel_after_->Fill(sieie);
        } else if (isEndcap) {
          sieieendcap_after_->Fill(sieie);
        }
      }
      if (!electron_sel[3] || flags_passed == NFLAGS) {
        if (isBarrel) {
          detainbarrel_after_->Fill(detain);
        } else if (isEndcap) {
          detainendcap_after_->Fill(detain);
        }
      }
      if (!electron_sel[4] || flags_passed == NFLAGS) {
        if (isBarrel) {
          ecalisobarrel_after_->Fill(ecalisovar);
        } else if (isEndcap) {
          ecalisoendcap_after_->Fill(ecalisovar);
        }
      }
      if (!electron_sel[5] || flags_passed == NFLAGS) {
        if (isBarrel) {
          hcalisobarrel_after_->Fill(hcalisovar);
        } else if (isEndcap) {
          hcalisoendcap_after_->Fill(hcalisovar);
        }
      }
      if (!electron_sel[6] || flags_passed == NFLAGS) {
        if (isBarrel) {
          trkisobarrel_after_->Fill(trkisovar);
        } else if (isEndcap) {
          trkisoendcap_after_->Fill(trkisovar);
        }
      }
      // 		if (!electron_sel[3] || flags_passed==NFLAGS)
      // 		  {
      // 		    detain_after_->Fill(detain);
      // 		  }
      // 		if (!electron_sel[4] || flags_passed==NFLAGS)
      // 		  {
      // 		    ecaliso_after_->Fill(trackerHits);
      // 		  }
      // 		if (!electron_sel[5] || flags_passed==NFLAGS)
      // 		  {
      // 		    tkelectr_after_->Fill(electr.isTrackerElectron());
      // 		  }
      // 		if (!electron_sel[6] || flags_passed==NFLAGS)
      // 		  {
      // 		    iso_after_->Fill(isovar);
      // 		  }
      /*  if (!electron_sel[7] || flags_passed == NFLAGS) {
        if (!hlt_hist_done) {
          trig_after_->Fill(trigger_fired);
        }
      }*/
      //hlt_hist_done = true;
      if (!electron_sel[7] || flags_passed == NFLAGS) {
        mt_after_->Fill(massT);
      }
      if (!electron_sel[8] || flags_passed == NFLAGS) {
        if (!met_hist_done) {
          met_after_->Fill(met_et);
        }
      }
      met_hist_done = true;
      //                   if (!muon_sel[10] || flags_passed==NFLAGS)
      //                         acop_after_->Fill(acop);
      //                   if (!muon_sel[11] || flags_passed==NFLAGS)
      //                         if (!nz1_hist_done)
      // nz1_after_->Fill(nmuonsForZ1);
      //                         nz1_hist_done = true;
      //                   if (!muon_sel[11] || flags_passed==NFLAGS)
      //                         if (!nz2_hist_done)
      // nz2_after_->Fill(nmuonsForZ2);
      //                         nz2_hist_done = true;
      if (!electron_sel[9] || flags_passed == NFLAGS) {
        if (!njets_hist_done) {
          njets_after_->Fill(njets);
          if (jet_et > 10)  // don't want low energy "jets"
          {
            jet_et_after_->Fill(jet_et);
            jet_eta_after_->Fill(jet_eta);
          }
        }
      }
      njets_hist_done = true;

    }  // end N-1 histos block

  }  // end loop through electrons

  // inv mass = sqrt(m_1^2 + m_2^2 + 2*(E_1*E_2 - (px1*px2 + py1*py2 + pz1+pz2)
  // ) )
  double invMass = 0;

  nelectrons_before_->Fill(electronCollectionSize);
  if (electronCollectionSize > 1) {
    invMass =
        sqrt(electron[0][1] + electron[1][1] +
             2 * (electron[0][2] * electron[1][2] - (electron[0][3] * electron[1][3] + electron[0][4] * electron[1][4] +
                                                     electron[0][5] * electron[1][5])));
    invmass_before_->Fill(invMass);
    invmassPU_before_->Fill(invMass, npvCount);
  }

  nelectrons_after_->Fill(nGoodElectrons);
  if (nGoodElectrons > 1) {
    invMass = sqrt(goodElectron[0][1] + goodElectron[1][1] +
                   2 * (goodElectron[0][2] * goodElectron[1][2] -
                        (goodElectron[0][3] * goodElectron[1][3] + goodElectron[0][4] * goodElectron[1][4] +
                         goodElectron[0][5] * goodElectron[1][5])));
    invmass_after_->Fill(invMass);
    invmassPU_afterZ_->Fill(invMass, npvCount);
    npvs_afterZ_->Fill(npvCount);
  }

  // Collect final flags
  if (rec_sel)
    nrec++;
  if (eid_sel)
    neid++;
  if (iso_sel)
    niso++;
  //      if (hlt_sel) nhlt++;
  //      if (met_sel) nmet++;

  if (all_sel) {
    nsel++;
    LogTrace("") << ">>>> Event ACCEPTED";
  } else {
    LogTrace("") << ">>>> Event REJECTED";
  }

  return;
}

// This always returns only a positive deltaPhi
double EwkElecDQM::calcDeltaPhi(double phi1, double phi2) {
  double deltaPhi = phi1 - phi2;

  if (deltaPhi < 0)
    deltaPhi = -deltaPhi;

  if (deltaPhi > 3.1415926) {
    deltaPhi = 2 * 3.1415926 - deltaPhi;
  }

  return deltaPhi;
}

// Local Variables:
// show-trailing-whitespace: t
// truncate-lines: t
// End: