TestMuonCaloCleaner.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    TestMuonCaloCleaner
// Class:      TestMuonCaloCleaner
// 

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/EgammaReco/interface/ElectronSeed.h"
#include "DataFormats/PatCandidates/interface/Muon.h"

#include "DataFormats/Math/interface/deltaR.h"

#include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/EcalDetId/interface/EcalSubdetector.h"
#include "DataFormats/HcalDetId/interface/HcalSubdetector.h"

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

#include "TTree.h"
#include "TFile.h"
#include "TH2F.h"


#include <boost/foreach.hpp>

#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"


//
// class declaration
//

class TestMuonCaloCleaner : public edm::EDAnalyzer {
   public:
      explicit TestMuonCaloCleaner(const edm::ParameterSet&);
      ~TestMuonCaloCleaner();

      static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);


   private:
      virtual void beginJob() ;
      virtual void analyze(const edm::Event&, const edm::EventSetup&);
      virtual void endJob() ;

      virtual void beginRun(edm::Run const&, edm::EventSetup const&);
      virtual void endRun(edm::Run const&, edm::EventSetup const&);
      virtual void beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&);
      virtual void endLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&);
      edm::InputTag colLen_;
      edm::InputTag colDep_;
      edm::InputTag _inputCol;
      int charge_;
      
      std::string getKey(const DetId & det);
      std::vector<std::string> getAllKeys();
      
      
      typedef std::map<int, std::string > TMyMainMap;
      typedef std::map<int, std::map<int, std::string> > TMySubMap;
      typedef std::map<std::string, TH2F * > THistoMap;
      
      
      TMyMainMap detMap_;
      TMySubMap  subDetMap_;
      
      
      // ----------member data ---------------------------
};

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
TestMuonCaloCleaner::TestMuonCaloCleaner(const edm::ParameterSet& iConfig):
 colLen_(iConfig.getParameter<edm::InputTag>("colLen")),
 colDep_(iConfig.getParameter<edm::InputTag>("colDep")),
 _inputCol(iConfig.getParameter<edm::InputTag>("selectedMuons")),
 charge_(iConfig.getParameter<int>("charge"))
{
  
  detMap_[DetId::Hcal]="Hcal";
  detMap_[DetId::Ecal]="Ecal";
  
  subDetMap_[DetId::Ecal][EcalBarrel]="EcalBarrel";
  subDetMap_[DetId::Ecal][EcalEndcap]="EcalEndcap";
  subDetMap_[DetId::Ecal][EcalPreshower ]="EcalPreshower";
  subDetMap_[DetId::Ecal][EcalTriggerTower]="EcalTriggerTower";
  subDetMap_[DetId::Ecal][EcalLaserPnDiode]="EcalLaserPnDiode";
  
  subDetMap_[DetId::Hcal][HcalEmpty]="HcalEmpty";
  subDetMap_[DetId::Hcal][HcalBarrel]="HcalBarrel";
  subDetMap_[DetId::Hcal][HcalEndcap]="HcalEndcap";
  subDetMap_[DetId::Hcal][HcalOuter]="HcalOuter";
  subDetMap_[DetId::Hcal][HcalForward]="HcalForward";
  subDetMap_[DetId::Hcal][HcalTriggerTower]="HcalTriggerTower";
  subDetMap_[DetId::Hcal][HcalOther]="HcalOther";

  
}


void
TestMuonCaloCleaner::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
{

  //std::cout << "------------------------------\n";
  edm::Handle< std::vector<reco::Muon> > muonsHandle;
  iEvent.getByLabel(_inputCol, muonsHandle); 
    
  const reco::Muon * myMu = 0;
  BOOST_FOREACH(const reco::Muon & mu, *muonsHandle){
      if (!mu.isGlobalMuon()) continue;
      if (mu.charge()!= charge_) continue;
      myMu = &mu;
      break;
  }
    
  if (myMu==0){
    // std::cout << " XXX TestMuonCaloCleaner says whoooops" << std::endl;
    return;
  }
  
  typedef std::map<unsigned int,float> TMyCol;
  edm::Handle< TMyCol > hLenghts;
  edm::Handle< TMyCol > hDeposits;
  

  iEvent.getByLabel(colLen_,hLenghts);
  iEvent.getByLabel(colDep_,hDeposits);
 
  unsigned int  selectedDet  = 0;

  BOOST_FOREACH(const TMyCol::value_type & entry, *hLenghts){
    DetId det(entry.first);
    float len = entry.second;
    float val = 0;
    std::string name =getKey(det);
    if (hDeposits->find(entry.first) != hDeposits->end())
      val = (hDeposits->find(entry.first))->second;
    else if (name == "H_Ecal_EcalEndcap") {
     //std::cout << " nm " << name << std::endl;
     selectedDet = entry.first;
    }

    const bool fDebug = false;
    if(fDebug)
      std::cout << "XX " << name << " " << det.rawId() << " " << len << " " << val << std::endl;
  }

  if (selectedDet != 0) {
     //std::cout << "TT"<<std::endl;
     edm::Handle< edm::SortedCollection<EcalRecHit> > hEErh;
     iEvent.getByLabel(edm::InputTag( "ecalRecHit", "EcalRecHitsEE"), hEErh );
 
     BOOST_FOREACH(const EcalRecHit & rh, *hEErh){
 
       //std::cout << " " <<   rh.energy() << std::endl;
       //DetId det(selectedDet);

       if ( selectedDet == rh.id().rawId() ) {
         std::cout << selectedDet <<  " " <<   rh.energy() << std::endl;
       }
     }

  }
}
   

std::string TestMuonCaloCleaner::getKey(const DetId & det){
  return "H_"+detMap_[det.det()]+"_"+subDetMap_[det.det()][det.subdetId()];
}

std::vector<std::string> TestMuonCaloCleaner::getAllKeys(){
  std::vector<std::string> ret;
  ret.push_back("H__");
  BOOST_FOREACH(TMyMainMap::value_type & entry, detMap_){
    BOOST_FOREACH(TMySubMap::mapped_type::value_type & subEntry, subDetMap_[entry.first]){
      std::string name = "H_"+entry.second+"_"+subEntry.second;
      //std::cout << "XX " << name << std::endl;
      ret.push_back(name);
    }
    
  }
  
  return ret;
}

TestMuonCaloCleaner::~TestMuonCaloCleaner()
{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}


//
// member functions
//

// ------------ method called for each event  ------------


// ------------ method called once each job just before starting event loop  ------------
void 
TestMuonCaloCleaner::beginJob()
{
}

// ------------ method called once each job just after ending the event loop  ------------
void 
TestMuonCaloCleaner::endJob() 
{

}

// ------------ method called when starting to processes a run  ------------
void 
TestMuonCaloCleaner::beginRun(edm::Run const&, edm::EventSetup const&)
{
}

// ------------ method called when ending the processing of a run  ------------
void 
TestMuonCaloCleaner::endRun(edm::Run const&, edm::EventSetup const&)
{
}

// ------------ method called when starting to processes a luminosity block  ------------
void 
TestMuonCaloCleaner::beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&)
{
}

// ------------ method called when ending the processing of a luminosity block  ------------
void 
TestMuonCaloCleaner::endLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&)
{
}

// ------------ method fills 'descriptions' with the allowed parameters for the module  ------------
void
TestMuonCaloCleaner::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  //The following says we do not know what parameters are allowed so do no validation
  // Please change this to state exactly what you do use, even if it is no parameters
  edm::ParameterSetDescription desc;
  desc.setUnknown();
  descriptions.addDefault(desc);
}

//define this as a plug-in
DEFINE_FWK_MODULE(TestMuonCaloCleaner);