/data/refman/pasoursint/CMSSW_6_1_2_SLHC4_patch1/src/DPGAnalysis/Skims/src/BeamSplash.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:   BeamSplash
// Class:     BeamSPlash
//
//
// Original Author:  Luca Malgeri

#include <memory>
#include <vector>
#include <map>
#include <set>

// user include files
#include "DPGAnalysis/Skims/interface/BeamSplash.h"

#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDFilter.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EEDetId.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h"
#include "DataFormats/HcalDetId/interface/HcalDetId.h"

using namespace edm;
using namespace std;

BeamSplash::BeamSplash(const edm::ParameterSet& iConfig)
{
  EBRecHitCollection_ = iConfig.getParameter<edm::InputTag>("ebrechitcollection");
  EERecHitCollection_ = iConfig.getParameter<edm::InputTag>("eerechitcollection");
  HBHERecHitCollection_ = iConfig.getParameter<edm::InputTag>("hbherechitcollection");

  EnergyCutTot = iConfig.getUntrackedParameter<double>("energycuttot");
  EnergyCutEcal = iConfig.getUntrackedParameter<double>("energycutecal");
  EnergyCutHcal = iConfig.getUntrackedParameter<double>("energycuthcal");
  applyfilter = iConfig.getUntrackedParameter<bool>("applyfilter",true);
}

BeamSplash::~BeamSplash()
{
}

bool BeamSplash::filter( edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  bool accepted = false;

  bool acceptedtot = false;
  bool acceptedEcal = false;
  bool acceptedHcal = false;

  int ievt = iEvent.id().event();
  int irun = iEvent.id().run();
  int ils = iEvent.luminosityBlock();
  int ibx = iEvent.bunchCrossing();

  double totene=0;
  double ecalene=0;
  double hcalene=0;


  Handle< EBRecHitCollection > pEBRecHits;
  Handle< EERecHitCollection > pEERecHits;
  Handle< HBHERecHitCollection > pHBHERecHits;
 
  const EBRecHitCollection*  EBRecHits = 0;
  const EERecHitCollection*  EERecHits = 0; 
  const HBHERecHitCollection*  HBHERecHits = 0; 
  
  if ( EBRecHitCollection_.label() != "" && EBRecHitCollection_.instance() != "" ) 
    {
      iEvent.getByLabel( EBRecHitCollection_, pEBRecHits);
      if ( pEBRecHits.isValid() ) 
        {
          EBRecHits = pEBRecHits.product(); // get a ptr to the product
        } 
      else 
        {
          edm::LogError("EcalRecHitError") << "Error! can't get the product " << EBRecHitCollection_.label() ;
        }
    }

  if ( EERecHitCollection_.label() != "" && EERecHitCollection_.instance() != "" ) 
    {
      iEvent.getByLabel( EERecHitCollection_, pEERecHits);
      
      if ( pEERecHits.isValid() ) 
        {
          EERecHits = pEERecHits.product(); // get a ptr to the product
        } 
      else 
        {
          edm::LogError("EcalRecHitError") << "Error! can't get the product " << EERecHitCollection_.label() ;
      }
    }

  if ( HBHERecHitCollection_.label() != "" )
    {
      iEvent.getByLabel( HBHERecHitCollection_, pHBHERecHits);
      
      if ( pHBHERecHits.isValid() ) 
        {
          HBHERecHits = pHBHERecHits.product(); // get a ptr to the product
        } 
      else 
        {
          edm::LogError("HcalRecHitError") << "Error! can't get the product " << HBHERecHitCollection_.label() ;
      }
    }

  // now sum over them
  if (EBRecHits) 
    {
      for(EBRecHitCollection::const_iterator it  = EBRecHits->begin(); it != EBRecHits->end(); ++it) 
        {
          totene+=it->energy();
          ecalene+=it->energy();
        }
    }
  if (EERecHits) 
    {
      for(EERecHitCollection::const_iterator it  = EERecHits->begin(); it != EERecHits->end(); ++it) 
        {
          totene+=it->energy();
          ecalene+=it->energy();
        }
    }
  if (HBHERecHits) 
    {
      for(HBHERecHitCollection::const_iterator it  = HBHERecHits->begin(); it != HBHERecHits->end(); ++it) 
        {
          totene+=it->energy();
          hcalene+=it->energy();
        }
    }

  if(totene>EnergyCutTot) acceptedtot=true;
  if(ecalene>EnergyCutEcal) acceptedEcal=true;
  if(hcalene>EnergyCutHcal) acceptedHcal=true;

  accepted = acceptedtot|acceptedEcal|acceptedHcal;
  
  
  if (accepted)
        {
    edm::LogVerbatim("BeamSplash") << "!!!!!!!BeamSplash!!!!!!!: run:" << irun << " event:" << ievt << " ls:"<< ils << " bx= " << ibx  <<" totene=" << totene << " ecalene=" << ecalene << " hcalene=" << hcalene ; 
    std::cout << "!!!!!!!BeamSplash!!!!!!!: run:" << irun << " event:" << ievt << " ls:"<< ils << " bx= " << ibx  <<" totene=" <<totene << " ecalene=" << ecalene << " hcalene=" << hcalene << std::endl; 
 } 

  if (applyfilter)
    return accepted;
  else
    return true;

}

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