ElectronSeedMerger.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    PFTracking
// Class:      ElectronSeedMerger
//
// Original Author:  Michele Pioppi

#include "RecoParticleFlow/PFTracking/interface/ElectronSeedMerger.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "DataFormats/EgammaReco/interface/ElectronSeed.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"
#include "TrackingTools/Records/interface/TransientRecHitRecord.h"
#include <string>

using namespace edm;
using namespace std;
using namespace reco;

ElectronSeedMerger::ElectronSeedMerger(const ParameterSet& iConfig) : conf_(iConfig) {
  LogInfo("ElectronSeedMerger") << "Electron SeedMerger  started  ";

  ecalSeedToken_ = consumes<ElectronSeedCollection>(iConfig.getParameter<InputTag>("EcalBasedSeeds"));
  edm::InputTag tkSeedLabel_ = iConfig.getParameter<InputTag>("TkBasedSeeds");
  if (!tkSeedLabel_.label().empty())
    tkSeedToken_ = consumes<ElectronSeedCollection>(tkSeedLabel_);

  produces<ElectronSeedCollection>();
}

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

//
// member functions
//

// ------------ method called to produce the data  ------------
void ElectronSeedMerger::produce(Event& iEvent, const EventSetup& iSetup) {
  //CREATE OUTPUT COLLECTION
  auto output = std::make_unique<ElectronSeedCollection>();

  //HANDLE THE INPUT SEED COLLECTIONS
  Handle<ElectronSeedCollection> EcalBasedSeeds;
  iEvent.getByToken(ecalSeedToken_, EcalBasedSeeds);
  ElectronSeedCollection ESeed = *(EcalBasedSeeds.product());

  Handle<ElectronSeedCollection> TkBasedSeeds;
  ElectronSeedCollection TSeed;
  if (!tkSeedToken_.isUninitialized()) {
    iEvent.getByToken(tkSeedToken_, TkBasedSeeds);
    TSeed = *(TkBasedSeeds.product());
  }

  //VECTOR FOR MATCHED SEEDS
  vector<bool> TSeedMatched;
  for (unsigned int it = 0; it < TSeed.size(); it++) {
    TSeedMatched.push_back(false);
  }

  //LOOP OVER THE ECAL SEED COLLECTION
  ElectronSeedCollection::const_iterator e_beg = ESeed.begin();
  ElectronSeedCollection::const_iterator e_end = ESeed.end();
  for (; e_beg != e_end; ++e_beg) {
    ElectronSeed NewSeed = *(e_beg);
    bool AlreadyMatched = false;

    //LOOP OVER THE TK SEED COLLECTION
    for (unsigned int it = 0; it < TSeed.size(); it++) {
      if (AlreadyMatched)
        continue;

      //HITS FOR ECAL SEED
      TrajectorySeed::const_iterator eh = e_beg->recHits().first;
      TrajectorySeed::const_iterator eh_end = e_beg->recHits().second;

      //HITS FOR TK SEED
      unsigned int hitShared = 0;
      unsigned int hitSeed = 0;
      for (; eh != eh_end; ++eh) {
        if (!eh->isValid())
          continue;
        hitSeed++;
        bool Shared = false;
        TrajectorySeed::const_iterator th = TSeed[it].recHits().first;
        TrajectorySeed::const_iterator th_end = TSeed[it].recHits().second;
        for (; th != th_end; ++th) {
          if (!th->isValid())
            continue;
          //CHECK THE HIT COMPATIBILITY: put back sharesInput
          // as soon Egamma solves the bug on the seed collection
          if (eh->sharesInput(&(*th), TrackingRecHit::all))
            Shared = true;
          //   if(eh->geographicalId() == th->geographicalId() &&
          //         (eh->localPosition() - th->localPosition()).mag() < 0.001) Shared=true;
        }
        if (Shared)
          hitShared++;
      }
      if (hitShared == hitSeed) {
        AlreadyMatched = true;
        TSeedMatched[it] = true;
        NewSeed.setCtfTrack(TSeed[it].ctfTrack());
      }
      if (hitShared == (hitSeed - 1)) {
        NewSeed.setCtfTrack(TSeed[it].ctfTrack());
      }
    }

    output->push_back(NewSeed);
  }

  //FILL THE COLLECTION WITH UNMATCHED TK-BASED SEED
  for (unsigned int it = 0; it < TSeed.size(); it++) {
    if (!TSeedMatched[it])
      output->push_back(TSeed[it]);
  }

  //PUT THE MERGED COLLECTION IN THE EVENT
  iEvent.put(std::move(output));
}