SimplePi0DiscAnalyzer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    RecoEgamma/Examples
// Class:      SimplePi0DiscAnalyzer
//
//
// Original Author:  Aristotelis Kyriakis NCSR "Demokritos" Athens
//                    D Maletic, "Vinca" Belgrade
//         Created:  May 26 13:22:06 CEST 2009
//
//

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/EcalDetId/interface/EcalSubdetector.h"
#include "DataFormats/EgammaCandidates/interface/Photon.h"
#include "DataFormats/EgammaCandidates/interface/PhotonFwd.h"
#include "DataFormats/EgammaCandidates/interface/PhotonPi0DiscriminatorAssociation.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "CLHEP/Units/PhysicalConstants.h"

#include "TFile.h"
#include "TH1F.h"
#include "TH1I.h"
#include "TH2F.h"
#include "TProfile.h"
#include "TTree.h"

#include <iostream>

class SimplePi0DiscAnalyzer : public edm::one::EDAnalyzer<> {
public:
  explicit SimplePi0DiscAnalyzer(const edm::ParameterSet& conf);

  ~SimplePi0DiscAnalyzer() override;

  void beginJob() override;
  void endJob() override;
  void analyze(const edm::Event& e, const edm::EventSetup& c) override;

private:
  // ----------member data ---------------------------

  std::string photonCollectionProducer_;
  std::string photonCollection_;

  std::string outputFile_;
  TFile* rootFile_;

  TH1F* hConv_ntracks_;

  TH1F* hAll_nnout_Assoc_;
  TH1F* hAll_nnout_NoConv_Assoc_;
  TH1F* hBarrel_nnout_Assoc_;
  TH1F* hBarrel_nnout_NoConv_Assoc_;
  TH1F* hEndcNoPresh_nnout_Assoc_;
  TH1F* hEndcNoPresh_nnout_NoConv_Assoc_;
  TH1F* hEndcWithPresh_nnout_Assoc_;
  TH1F* hEndcWithPresh_nnout_NoConv_Assoc_;
  TH1F* hAll_nnout_NoConv_Assoc_R9_;
  TH1F* hBarrel_nnout_NoConv_Assoc_R9_;
  TH1F* hEndcNoPresh_nnout_NoConv_Assoc_R9_;
  TH1F* hEndcWithPresh_nnout_NoConv_Assoc_R9_;
};

#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(SimplePi0DiscAnalyzer);

using namespace reco;

SimplePi0DiscAnalyzer::SimplePi0DiscAnalyzer(const edm::ParameterSet& conf) {
  outputFile_ = conf.getParameter<std::string>("outputFile");
  rootFile_ = new TFile(outputFile_.c_str(), "RECREATE");

  photonCollectionProducer_ = conf.getParameter<std::string>("phoProducer");
  photonCollection_ = conf.getParameter<std::string>("photonCollection");
}

SimplePi0DiscAnalyzer::~SimplePi0DiscAnalyzer() {
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
  rootFile_->Write();
  rootFile_->Close();
}

void SimplePi0DiscAnalyzer::beginJob() {
  rootFile_->cd();
  std::cout << "beginJob() ->  Book the Histograms" << std::endl;

  hConv_ntracks_ = new TH1F("nConvTracks", "Number of tracks of converted Photons ", 10, 0., 10);
  hAll_nnout_Assoc_ = new TH1F("All_nnout_Assoc", "NNout for All Photons(AssociationMap)", 100, 0., 1.);
  hAll_nnout_NoConv_Assoc_ =
      new TH1F("All_nnout_NoConv_Assoc", "NNout for Unconverted Photons(AssociationMap)", 100, 0., 1.);
  hAll_nnout_NoConv_Assoc_R9_ =
      new TH1F("All_nnout_NoConv_Assoc_R9", "NNout for Unconverted Photons with R9>0.93 (AssociationMap)", 100, 0., 1.);
  hBarrel_nnout_Assoc_ = new TH1F("barrel_nnout_Assoc", "NNout for Barrel Photons(AssociationMap)", 100, 0., 1.);
  hBarrel_nnout_NoConv_Assoc_ =
      new TH1F("barrel_nnout_NoConv_Assoc", "NNout for Barrel Unconverted Photons(AssociationMap)", 100, 0., 1.);
  hBarrel_nnout_NoConv_Assoc_R9_ = new TH1F(
      "barrel_nnout_NoConv_Assoc_R9", "NNout for Barrel Unconverted Photons with R9>0.93 (AssociationMap)", 100, 0., 1.);
  hEndcNoPresh_nnout_Assoc_ =
      new TH1F("endcNoPresh_nnout_Assoc", "NNout for Endcap NoPresh Photons(AssociationMap)", 100, 0., 1.);
  hEndcNoPresh_nnout_NoConv_Assoc_ = new TH1F(
      "endcNoPresh_nnout_NoConv_Assoc", "NNout for Endcap Unconverted NoPresh Photons(AssociationMap)", 100, 0., 1.);
  hEndcNoPresh_nnout_NoConv_Assoc_R9_ =
      new TH1F("endcNoPresh_nnout_NoConv_Assoc_R9",
               "NNout for Endcap Unconverted NoPresh Photons with R9>0.93 (AssociationMap)",
               100,
               0.,
               1.);
  hEndcWithPresh_nnout_Assoc_ =
      new TH1F("endcWithPresh_nnout_Assoc", "NNout for Endcap WithPresh Photons(AssociationMap)", 100, 0., 1.);
  hEndcWithPresh_nnout_NoConv_Assoc_ = new TH1F(
      "endcWithPresh_nnout_NoConv_Assoc", "NNout for Endcap Unconverted WithPresh Photons(AssociationMap)", 100, 0., 1.);
  hEndcWithPresh_nnout_NoConv_Assoc_R9_ =
      new TH1F("endcWithPresh_nnout_NoConv_Assoc_R9",
               "NNout for Endcap Unconverted WithPresh Photons with R9>0.93 (AssociationMap)",
               100,
               0.,
               1.);
}

void SimplePi0DiscAnalyzer::endJob() {
  rootFile_->cd();
  std::cout << "endJob() ->  Write the Histograms" << std::endl;
  hConv_ntracks_->Write();

  hAll_nnout_Assoc_->Write();
  hAll_nnout_NoConv_Assoc_->Write();
  hAll_nnout_NoConv_Assoc_R9_->Write();
  hBarrel_nnout_Assoc_->Write();
  hBarrel_nnout_NoConv_Assoc_->Write();
  hBarrel_nnout_NoConv_Assoc_R9_->Write();
  hEndcNoPresh_nnout_Assoc_->Write();
  hEndcNoPresh_nnout_NoConv_Assoc_->Write();
  hEndcNoPresh_nnout_NoConv_Assoc_R9_->Write();
  hEndcWithPresh_nnout_Assoc_->Write();
  hEndcWithPresh_nnout_NoConv_Assoc_->Write();
  hEndcWithPresh_nnout_NoConv_Assoc_R9_->Write();
}

void SimplePi0DiscAnalyzer::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
  std::cout << std::endl;
  std::cout << " -------------- NEW EVENT : Run, Event =  " << iEvent.id() << std::endl;

  edm::Handle<reco::PhotonCollection> PhotonHandle;
  iEvent.getByLabel(photonCollectionProducer_, photonCollection_, PhotonHandle);
  const reco::PhotonCollection photons = *(PhotonHandle.product());

  std::cout << "----> Photons size: " << photons.size() << std::endl;

  edm::Handle<reco::PhotonPi0DiscriminatorAssociationMap> map;
  iEvent.getByLabel("piZeroDiscriminators", "PhotonPi0DiscriminatorAssociationMap", map);
  reco::PhotonPi0DiscriminatorAssociationMap::const_iterator mapIter;

  //  int PhoInd = 0;

  for (reco::PhotonCollection::const_iterator iPho = photons.begin(); iPho != photons.end();
       iPho++) {  // Loop over Photons

    reco::Photon localPho(*iPho);

    float Photon_et = localPho.et();
    float Photon_eta = localPho.eta();
    float Photon_phi = localPho.phi();
    float Photon_r9 = localPho.r9();
    bool isPhotConv = localPho.hasConversionTracks();
    //    std::cout << "Photon Id = " << PhoInd
    std::cout << "Photon Id = " << iPho - photons.begin() << " with Et = " << Photon_et << " Eta = " << Photon_eta
              << " Phi = " << Photon_phi << " R9 = " << Photon_r9 << " and conv_id = " << isPhotConv << std::endl;

    auto it_super = localPho.superCluster();  // get the SC related to the Photon candidate

    //    hConv_ntracks_->Fill(Ntrk_conv);

    float nn = -10;
    //    mapIter = map->find(edm::Ref<reco::PhotonCollection>(PhotonHandle,PhoInd));
    mapIter = map->find(edm::Ref<reco::PhotonCollection>(PhotonHandle, iPho - photons.begin()));
    if (mapIter != map->end()) {
      nn = mapIter->val;
    }
    if (fabs(it_super->eta()) <= 1.442) {
      hBarrel_nnout_Assoc_->Fill(nn);
      hAll_nnout_Assoc_->Fill(nn);
      std::cout << "AssociationMap Barrel NN = " << nn << std::endl;
      if (!isPhotConv) {
        hBarrel_nnout_NoConv_Assoc_->Fill(nn);
        hAll_nnout_NoConv_Assoc_->Fill(nn);
      }
      if (Photon_r9 > 0.93) {
        hBarrel_nnout_NoConv_Assoc_R9_->Fill(nn);
        hAll_nnout_NoConv_Assoc_R9_->Fill(nn);
      }
    } else if ((fabs(it_super->eta()) >= 1.556 && fabs(it_super->eta()) < 1.65) || fabs(it_super->eta()) > 2.5) {
      hEndcNoPresh_nnout_Assoc_->Fill(nn);
      hAll_nnout_Assoc_->Fill(nn);
      std::cout << "AssociationMap EndcNoPresh NN = " << nn << std::endl;
      if (!isPhotConv) {
        hEndcNoPresh_nnout_NoConv_Assoc_->Fill(nn);
        hAll_nnout_NoConv_Assoc_->Fill(nn);
      }
      if (Photon_r9 > 0.93) {
        hEndcNoPresh_nnout_NoConv_Assoc_R9_->Fill(nn);
        hAll_nnout_NoConv_Assoc_R9_->Fill(nn);
      }
    } else if (fabs(it_super->eta()) >= 1.65 && fabs(it_super->eta()) <= 2.5) {
      hEndcWithPresh_nnout_Assoc_->Fill(nn);
      hAll_nnout_Assoc_->Fill(nn);
      std::cout << "AssociationMap EndcWithPresh NN = " << nn << std::endl;
      if (!isPhotConv) {
        hEndcWithPresh_nnout_NoConv_Assoc_->Fill(nn);
        hAll_nnout_NoConv_Assoc_->Fill(nn);
      }
      if (Photon_r9 > 0.93) {
        hEndcWithPresh_nnout_NoConv_Assoc_R9_->Fill(nn);
        hAll_nnout_NoConv_Assoc_R9_->Fill(nn);
      }
    }

    //    PhoInd++;
  }  // End Loop over Photons
}