HLTDoublet.cc

Go to the documentation of this file.

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "HLTrigger/HLTfilters/interface/HLTDoublet.h"

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

#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h"

#include "DataFormats/Candidate/interface/Particle.h"

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

#include<cmath>

//
// constructors and destructor
//
template<typename T1, int Tid1, typename T2, int Tid2>
HLTDoublet<T1,Tid1,T2,Tid2>::HLTDoublet(const edm::ParameterSet& iConfig) :
  inputTag1_(iConfig.template getParameter<edm::InputTag>("inputTag1")),
  inputTag2_(iConfig.template getParameter<edm::InputTag>("inputTag2")),
  saveTags_ (iConfig.template getUntrackedParameter<bool>("saveTags",false)),
  min_Dphi_ (iConfig.template getParameter<double>("MinDphi")),
  max_Dphi_ (iConfig.template getParameter<double>("MaxDphi")),
  min_Deta_ (iConfig.template getParameter<double>("MinDeta")),
  max_Deta_ (iConfig.template getParameter<double>("MaxDeta")),
  min_Minv_ (iConfig.template getParameter<double>("MinMinv")),
  max_Minv_ (iConfig.template getParameter<double>("MaxMinv")),
  min_DelR_ (iConfig.template getParameter<double>("MinDelR")),
  max_DelR_ (iConfig.template getParameter<double>("MaxDelR")),
  min_N_    (iConfig.template getParameter<int>("MinN")),
  coll1_(),
  coll2_()
{
   // same collections to be compared?
   same_ = (inputTag1_.encode()==inputTag2_.encode());

   cutdphi_ = (min_Dphi_ <= max_Dphi_); // cut active?
   cutdeta_ = (min_Deta_ <= max_Deta_); // cut active?
   cutminv_ = (min_Minv_ <= max_Minv_); // cut active?
   cutdelr_ = (min_DelR_ <= max_DelR_); // cut active?

   LogDebug("") << "InputTags and cuts : " 
        << inputTag1_.encode() << " " << inputTag2_.encode()
        << " Dphi [" << min_Dphi_ << " " << max_Dphi_ << "]"
                << " Deta [" << min_Deta_ << " " << max_Deta_ << "]"
                << " Minv [" << min_Minv_ << " " << max_Minv_ << "]"
                << " DelR [" << min_DelR_ << " " << max_DelR_ << "]"
                << " MinN =" << min_N_
        << " same/dphi/deta/minv/delr "
        << same_ << cutdphi_ << cutdeta_ << cutminv_ << cutdelr_;

   //register your products
   produces<trigger::TriggerFilterObjectWithRefs>();
}

template<typename T1, int Tid1, typename T2, int Tid2>
HLTDoublet<T1,Tid1,T2,Tid2>::~HLTDoublet()
{
}

//
// member functions
//

// ------------ method called to produce the data  ------------
template<typename T1, int Tid1, typename T2, int Tid2>
bool
HLTDoublet<T1,Tid1,T2,Tid2>::filter(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
   using namespace std;
   using namespace edm;
   using namespace reco;
   using namespace trigger;

   // All HLT filters must create and fill an HLT filter object,
   // recording any reconstructed physics objects satisfying (or not)
   // this HLT filter, and place it in the Event.

   // The filter object
   auto_ptr<TriggerFilterObjectWithRefs>
     filterobject (new TriggerFilterObjectWithRefs(path(),module()));
   // Don't saveTag the TFOWRs, but rather the collections pointed to!
   //   if (saveTags_) {
   //     filterobject->addCollectionTag(inputTag1_);
   //     filterobject->addCollectionTag(inputTag2_);
   //   }
   bool accept(false);

   // get hold of pre-filtered object collections
   Handle<TriggerFilterObjectWithRefs> coll1,coll2;
   if (iEvent.getByLabel (inputTag1_,coll1) && iEvent.getByLabel (inputTag2_,coll2)) {
     coll1_.clear();
     coll1->getObjects(Tid1,coll1_);
     const size_type n1(coll1_.size());
     coll2_.clear();
     coll2->getObjects(Tid2,coll2_);
     const size_type n2(coll2_.size());

     if (saveTags_) {
       InputTag tagOld;
       tagOld=InputTag();
       for (size_type i1=0; i1!=n1; ++i1) {
     const ProductID pid(coll1_[i1].id());
     const string&    label(iEvent.getProvenance(pid).moduleLabel());
     const string& instance(iEvent.getProvenance(pid).productInstanceName());
     const string&  process(iEvent.getProvenance(pid).processName());
     InputTag tagNew(InputTag(label,instance,process));
     if (tagOld.encode()!=tagNew.encode()) {
       filterobject->addCollectionTag(tagNew);
       tagOld=tagNew;
     }
       }
       tagOld=InputTag();
       for (size_type i2=0; i2!=n2; ++i2) {
     const ProductID pid(coll2_[i2].id());
     const string&    label(iEvent.getProvenance(pid).moduleLabel());
     const string& instance(iEvent.getProvenance(pid).productInstanceName());
     const string&  process(iEvent.getProvenance(pid).processName());
     InputTag tagNew(InputTag(label,instance,process));
     if (tagOld.encode()!=tagNew.encode()) {
       filterobject->addCollectionTag(tagNew);
       tagOld=tagNew;
     }
       }
     }

     int n(0);
     T1Ref r1;
     T2Ref r2;
     Particle::LorentzVector p1,p2,p;
     for (unsigned int i1=0; i1!=n1; i1++) {
       r1=coll1_[i1];
       p1=r1->p4();
       unsigned int I(0);
       if (same_) {I=i1+1;}
       for (unsigned int i2=I; i2!=n2; i2++) {
     r2=coll2_[i2];
     p2=r2->p4();

     double Dphi(std::abs(p1.phi()-p2.phi()));
     if (Dphi>M_PI) Dphi=2.0*M_PI-Dphi;
     
     double Deta(std::abs(p1.eta()-p2.eta()));
     
     p=p1+p2;
     double Minv(std::abs(p.mass()));

     double DelR(sqrt(Dphi*Dphi+Deta*Deta));
     
     if ( ( (!cutdphi_) || ((min_Dphi_<=Dphi) && (Dphi<=max_Dphi_)) ) &&
          ( (!cutdeta_) || ((min_Deta_<=Deta) && (Deta<=max_Deta_)) ) &&
          ( (!cutminv_) || ((min_Minv_<=Minv) && (Minv<=max_Minv_)) ) &&
              ( (!cutdelr_) || ((min_DelR_<=DelR) && (DelR<=max_DelR_)) ) ) {
       n++;
       filterobject->addObject(Tid1,r1);
       filterobject->addObject(Tid2,r2);
     }
     
       }
     }
     // filter decision
     accept = accept || (n>=min_N_);
   }

   iEvent.put(filterobject);
   return accept;
}