ZMuMuOverlapExclusion.cc

Go to the documentation of this file.

#include "CommonTools/UtilAlgos/interface/OverlapExclusionSelector.h"
#include "CommonTools/UtilAlgos/interface/SingleObjectSelector.h"
#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"
#include <iostream>

struct ZMuMuOverlap {
  ZMuMuOverlap(const edm::ParameterSet&) {}
  bool operator()(const reco::Candidate& zMuMu, const reco::Candidate& z) const {
    using namespace std;
    using namespace reco;
    // check if a candidate z is different from zMuMu
    // (for example a Z can be done with two global muons, or with a global muon plus a standalone muon.
    // if the standalone muon is part of the second global muon in fact this is the same Z)

    unsigned int nd1 = zMuMu.numberOfDaughters();
    unsigned int nd2 = z.numberOfDaughters();

    assert(nd1 == 2 && nd2 == 2);
    const int maxd = 2;
    const Candidate* daughters1[maxd];
    const Candidate* daughters2[maxd];
    TrackRef trackerTrack1[maxd];
    TrackRef stAloneTrack1[maxd];
    TrackRef globalTrack1[maxd];
    TrackRef trackerTrack2[maxd];
    TrackRef stAloneTrack2[maxd];
    TrackRef globalTrack2[maxd];
    bool flag;
    unsigned int matched = 0;

    for (unsigned int i = 0; i < nd1; ++i) {
      daughters1[i] = zMuMu.daughter(i);
      trackerTrack1[i] = daughters1[i]->get<TrackRef>();
      stAloneTrack1[i] = daughters1[i]->get<TrackRef, reco::StandAloneMuonTag>();
      globalTrack1[i] = daughters1[i]->get<TrackRef, reco::CombinedMuonTag>();

      /*********************************************** just used for debug ********************
    if (trackerTrack1[i].isNull()) 
      cout << "in ZMuMu daughter " << i << " tracker ref non found " << endl;
    else
      cout << "in ZMuMu daughter " << i << " tracker ref FOUND" 
       << " id: " << trackerTrack1[i].id() << ", index: " << trackerTrack1[i].key() 
       << endl;
    if (stAloneTrack1[i].isNull()) 
      cout << "in ZMuMu daughter " << i << " stalone ref non found " << endl;
    else
      cout << "in ZMuMu daughter " << i << " stalone ref FOUND" 
       << " id: " << stAloneTrack1[i].id() << ", index: " << stAloneTrack1[i].key() 
       << endl;
    
    if (globalTrack1[i].isNull()) 
      cout << "in ZMuMu daughter " << i << " global ref non found " << endl;
    else
      cout << "in ZMuMu daughter " << i << " global ref FOUND"  
       << " id: " << globalTrack1[i].id() << ", index: " << globalTrack1[i].key() 
       << endl;
      */
    }
    for (unsigned int i = 0; i < nd2; ++i) {
      daughters2[i] = z.daughter(i);
      trackerTrack2[i] = daughters2[i]->get<TrackRef>();
      stAloneTrack2[i] = daughters2[i]->get<TrackRef, reco::StandAloneMuonTag>();
      globalTrack2[i] = daughters2[i]->get<TrackRef, reco::CombinedMuonTag>();

      /******************************************** just used for debug ************
    if (trackerTrack2[i].isNull()) 
      cout << "in ZMuSta daughter " << i << " tracker ref non found " << endl;
    else
      cout << "in ZMuSta daughter " << i << " tracker ref FOUND"  
       << " id: " << trackerTrack2[i].id() << ", index: " << trackerTrack2[i].key() 
       << endl;
    if (stAloneTrack2[i].isNull()) 
      cout << "in ZMuSta daughter " << i << " standalone ref non found " << endl;
    else
      cout << "in ZMuSta daughter " << i << " standalone ref FOUND" 
       << " id: " << stAloneTrack2[i].id() << ", index: " << stAloneTrack2[i].key() 
       << endl;
    
    if (globalTrack2[i].isNull()) 
      cout << "in ZMuSta daughter " << i << " global ref non found " << endl;
    else
      cout << "in ZMuSta daughter " << i << " global ref FOUND" 
       << " id: " << globalTrack2[i].id() << ", index: " << globalTrack2[i].key() 
       << endl;
       
      */
    }
    for (unsigned int i = 0; i < nd1; i++) {
      flag = false;
      for (unsigned int j = 0; j < nd2;
           j++) {  // if the obj2 is a standalone the trackref is alwais in the trackerTRack position
        if (((trackerTrack2[i].id() == trackerTrack1[j].id()) && (trackerTrack2[i].key() == trackerTrack1[j].key())) ||
            ((trackerTrack2[i].id() == stAloneTrack1[j].id()) && (trackerTrack2[i].key() == stAloneTrack1[j].key()))) {
          flag = true;
        }
      }
      if (flag)
        matched++;
    }
    if (matched == nd1)  // return true if all the childrens of the ZMuMu have a children matched in ZMuXX
      return true;
    else
      return false;
  }
};

typedef SingleObjectSelector<edm::View<reco::Candidate>,
                             OverlapExclusionSelector<reco::CandidateView, reco::Candidate, ZMuMuOverlap> >
    ZMuMuOverlapExclusionSelector;

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

DEFINE_FWK_MODULE(ZMuMuOverlapExclusionSelector);