ConvBremSeedProducer.cc

Go to the documentation of this file.

#include "CommonTools/BaseParticlePropagator/interface/BaseParticlePropagator.h"
#include "DataFormats/GeometrySurface/interface/Surface.h"
#include "DataFormats/GeometrySurface/interface/TangentPlane.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "DataFormats/ParticleFlowReco/interface/ConvBremSeed.h"
#include "DataFormats/ParticleFlowReco/interface/ConvBremSeedFwd.h"
#include "DataFormats/ParticleFlowReco/interface/GsfPFRecTrack.h"
#include "DataFormats/ParticleFlowReco/interface/GsfPFRecTrackFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFBrem.h"
#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2DCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2DCollection.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHitFwd.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FastSimulation/ParticlePropagator/interface/MagneticFieldMapRecord.h"
#include "FastSimulation/ParticlePropagator/interface/ParticlePropagator.h"
#include "FastSimulation/TrackerSetup/interface/TrackerInteractionGeometry.h"
#include "FastSimulation/TrackerSetup/interface/TrackerInteractionGeometryRecord.h"
#include "FastSimulation/TrajectoryManager/interface/InsideBoundsMeasurementEstimator.h"
#include "FastSimulation/TrajectoryManager/interface/LocalMagneticField.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "RecoTracker/Record/interface/TrackerRecoGeometryRecord.h"
#include "RecoTracker/TkDetLayers/interface/GeometricSearchTracker.h"
#include "RecoTracker/TkSeedGenerator/interface/FastHelix.h"
#include "RecoTracker/TransientTrackingRecHit/interface/TkTransientTrackingRecHitBuilder.h"
#include "TrackingTools/DetLayers/interface/BarrelDetLayer.h"
#include "TrackingTools/DetLayers/interface/DetLayer.h"
#include "TrackingTools/DetLayers/interface/ForwardDetLayer.h"
#include "TrackingTools/GeomPropagators/interface/AnalyticalPropagator.h"
#include "TrackingTools/KalmanUpdators/interface/KFUpdator.h"
#include "TrackingTools/MaterialEffects/interface/PropagatorWithMaterial.h"
#include "TrackingTools/Records/interface/TransientRecHitRecord.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateTransform.h"

#include "TMath.h"

#include <memory>

class ConvBremSeedProducer : public edm::stream::EDProducer<> {
public:
  explicit ConvBremSeedProducer(const edm::ParameterSet&);

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

private:
  void beginRun(const edm::Run&, const edm::EventSetup&) override;
  void produce(edm::Event&, const edm::EventSetup&) override;
  void endRun(const edm::Run&, const edm::EventSetup&) override;
  void initializeLayerMap();
  std::vector<const DetLayer*> theLayerMap;
  TrajectoryStateOnSurface makeTrajectoryState(const DetLayer* layer,
                                               const ParticlePropagator& pp,
                                               const MagneticField* field) const;
  const DetLayer* detLayer(const TrackerLayer& layer, float zpos) const;

  bool isGsfTrack(const reco::Track&, const TrackingRecHit*);

  int GoodCluster(const BaseParticlePropagator& bpg,
                  const reco::PFClusterCollection& pfc,
                  float minep,
                  bool sec = false);

  std::vector<bool> sharedHits(const std::vector<std::pair<TrajectorySeed, std::pair<GlobalVector, float> > >&);

  const edm::EDGetTokenT<reco::PFClusterCollection> pfToken_;
  const edm::EDGetTokenT<SiPixelRecHitCollection> pixelHitsToken_;
  const edm::EDGetTokenT<SiStripRecHit2DCollection> rphirecHitsToken_;
  const edm::EDGetTokenT<SiStripMatchedRecHit2DCollection> matchedrecHitsToken_;
  const edm::EDGetTokenT<reco::GsfPFRecTrackCollection> thePfRecTrackToken_;

  const GeometricSearchTracker* geomSearchTracker_;
  const TrackerInteractionGeometry* geometry_;
  const TrackerGeometry* tracker_;
  const MagneticField* magfield_;
  const MagneticFieldMap* fieldMap_;
  const PropagatorWithMaterial* propagator_;
  const KFUpdator* kfUpdator_;
  const TransientTrackingRecHitBuilder* hitBuilder_;
  std::vector<const DetLayer*> layerMap_;
  int negLayerOffset_;
  math::XYZVector B_;

  // Event setup tokens
  const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magFieldToken_;
  const edm::ESGetToken<GeometricSearchTracker, TrackerRecoGeometryRecord> geomSearchTrackerToken_;
  const edm::ESGetToken<TrackerInteractionGeometry, TrackerInteractionGeometryRecord> geometryToken_;
  const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> trackerToken_;
  const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magFieldToken_beginRun_;
  const edm::ESGetToken<MagneticFieldMap, MagneticFieldMapRecord> magFieldMapToken_;
  const edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> hitBuilderToken_;
};

#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(ConvBremSeedProducer);

void ConvBremSeedProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  // convBremSeeds
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("pixelRecHits", edm::InputTag("gsPixelRecHits"));
  desc.add<edm::InputTag>("matchedrecHits", edm::InputTag("gsStripRecHits", "matchedRecHit"));
  desc.add<std::string>("TTRHBuilder", "WithTrackAngle");
  desc.add<edm::InputTag>("rphirecHits", edm::InputTag("gsStripRecHits", "rphiRecHit"));
  desc.add<edm::InputTag>("PFClusters", edm::InputTag("particleFlowClusterECAL"));
  desc.add<edm::InputTag>("PFRecTrackLabel", edm::InputTag("pfTrackElec"));
}

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

ConvBremSeedProducer::ConvBremSeedProducer(const ParameterSet& iConfig)
    : pfToken_(consumes<reco::PFClusterCollection>(iConfig.getParameter<InputTag>("PFClusters"))),
      pixelHitsToken_(consumes<SiPixelRecHitCollection>(iConfig.getParameter<InputTag>("pixelRecHits"))),
      rphirecHitsToken_(consumes<SiStripRecHit2DCollection>(iConfig.getParameter<InputTag>("rphirecHits"))),
      matchedrecHitsToken_(
          consumes<SiStripMatchedRecHit2DCollection>(iConfig.getParameter<InputTag>("matchedrecHits"))),
      thePfRecTrackToken_(consumes<reco::GsfPFRecTrackCollection>(iConfig.getParameter<InputTag>("PFRecTrackLabel"))),
      fieldMap_(nullptr),
      layerMap_(56, static_cast<const DetLayer*>(nullptr)),
      negLayerOffset_(27),
      magFieldToken_(esConsumes()),
      geomSearchTrackerToken_(esConsumes<edm::Transition::BeginRun>()),
      geometryToken_(esConsumes<edm::Transition::BeginRun>()),
      trackerToken_(esConsumes<edm::Transition::BeginRun>()),
      magFieldToken_beginRun_(esConsumes<edm::Transition::BeginRun>()),
      magFieldMapToken_(esConsumes<edm::Transition::BeginRun>()),
      hitBuilderToken_(
          esConsumes<edm::Transition::BeginRun>(edm::ESInputTag("", iConfig.getParameter<string>("TTRHBuilder")))) {
  produces<ConvBremSeedCollection>();
}

void ConvBremSeedProducer::produce(Event& iEvent, const EventSetup& iSetup) {
  LogDebug("ConvBremSeedProducerProducer") << "START event: " << iEvent.id().event() << " in run " << iEvent.id().run();

  constexpr float pfmass = 0.0005;


  const auto& PPP = iEvent.get(pfToken_);

  const auto& pixelHits = iEvent.getHandle(pixelHitsToken_);

  const auto& rphirecHits = iEvent.getHandle(rphirecHitsToken_);
  const auto& matchedrecHits = iEvent.getHandle(matchedrecHitsToken_);

  //GSFPFRECTRACKS
  const auto& thePfRecTrackCollection = iEvent.getHandle(thePfRecTrackToken_);
  const auto& PfRTkColl = *thePfRecTrackCollection;

  auto output = std::make_unique<ConvBremSeedCollection>();

  vector<pair<TrajectorySeed, pair<GlobalVector, float> > > unclean;
  //TRIPLET OF MODULES TO BE USED FOR SEEDING
  vector<vector<long int> > tripl;
  //LAYER MAP
  initializeLayerMap();


  for (unsigned int ipft = 0; ipft < PfRTkColl.size(); ipft++) {
    GsfPFRecTrackRef pft(thePfRecTrackCollection, ipft);
    LogDebug("ConvBremSeedProducerProducer") << "NEW GsfPFRecTRACK ";
    float eta_br = 0;
    unclean.clear();
    tripl.clear();
    vector<int> gc;
    auto const& gsfRecHits = *pft->gsfTrackRef();
    float pfoutenergy = sqrt((pfmass * pfmass) + pft->gsfTrackRef()->outerMomentum().Mag2());
    XYZTLorentzVector mom = XYZTLorentzVector(pft->gsfTrackRef()->outerMomentum().x(),
                                              pft->gsfTrackRef()->outerMomentum().y(),
                                              pft->gsfTrackRef()->outerMomentum().z(),
                                              pfoutenergy);
    XYZTLorentzVector pos = XYZTLorentzVector(pft->gsfTrackRef()->outerPosition().x(),
                                              pft->gsfTrackRef()->outerPosition().y(),
                                              pft->gsfTrackRef()->outerPosition().z(),
                                              0.);
    BaseParticlePropagator theOutParticle =
        BaseParticlePropagator(RawParticle(mom, pos, pft->gsfTrackRef()->charge()), 0, 0, B_.z());

    gc.push_back(GoodCluster(theOutParticle, PPP, 0.5));

    vector<PFBrem> brem = (*pft).PFRecBrem();
    vector<PFBrem>::iterator ib = brem.begin();
    vector<PFBrem>::iterator ib_end = brem.end();
    LogDebug("ConvBremSeedProducerProducer") << "NUMBER OF BREMS " << brem.size();

    for (; ib != ib_end; ++ib) {
      XYZTLorentzVector mom = pft->trajectoryPoint(ib->indTrajPoint()).momentum();
      XYZTLorentzVector pos = XYZTLorentzVector(pft->trajectoryPoint(ib->indTrajPoint()).position().x(),
                                                pft->trajectoryPoint(ib->indTrajPoint()).position().y(),
                                                pft->trajectoryPoint(ib->indTrajPoint()).position().z(),
                                                0);

      if (pos.Rho() > 80)
        continue;
      if ((pos.Rho() > 5) && (fabs(ib->SigmaDeltaP() / ib->DeltaP()) > 3))
        continue;
      if (fabs(ib->DeltaP()) < 3)
        continue;
      eta_br = mom.eta();
      vector<vector<long int> > Idd;

      BaseParticlePropagator p(RawParticle(mom, pos, 0.), 0, 0, B_.z());
      gc.push_back(GoodCluster(p, PPP, 0.2));

      ParticlePropagator PP(p, fieldMap_, nullptr);

      list<TrackerLayer>::const_iterator cyliter = geometry_->cylinderBegin();
      for (; cyliter != geometry_->cylinderEnd(); ++cyliter) {
        if (!(cyliter->sensitive()))
          continue;
        PP.setPropagationConditions(*cyliter);
        PP.propagate();
        if (PP.getSuccess() == 0)
          continue;

        LocalMagneticField mf(PP.getMagneticField());
        AnalyticalPropagator alongProp(&mf, anyDirection);
        InsideBoundsMeasurementEstimator est;
        const DetLayer* tkLayer = detLayer(*cyliter, PP.particle().Z());
        if (&(*tkLayer) == nullptr)
          continue;
        TrajectoryStateOnSurface trajState = makeTrajectoryState(tkLayer, PP, &mf);

        auto compat = tkLayer->compatibleDets(trajState, alongProp, est);
        vector<long int> temp;
        if (compat.empty())
          continue;

        for (auto i = compat.begin(); i != compat.end(); i++) {
          long int detid = i->first->geographicalId().rawId();

          if (!GeomDetEnumerators::isTrackerPixel(tkLayer->subDetector())) {
            auto DetMatch = (rphirecHits.product())->find((detid));
            auto MDetMatch = (matchedrecHits.product())->find((detid));

            long int DetID = (DetMatch != rphirecHits->end()) ? detid : 0;

            if ((MDetMatch != matchedrecHits->end()) && !MDetMatch->empty()) {
              long int pii = MDetMatch->begin()->monoId();
              auto CDetMatch = (rphirecHits.product())->find((pii));
              DetID = (CDetMatch != rphirecHits->end()) ? pii : 0;
            }

            temp.push_back(DetID);

          } else {
            auto DetMatch = (pixelHits.product())->find((detid));
            long int DetID = (DetMatch != pixelHits->end()) ? detid : 0;
            temp.push_back(DetID);
          }
        }

        Idd.push_back(temp);

      }  //END TRACKER LAYER LOOP
      if (Idd.size() < 2)
        continue;

      for (unsigned int i = 0; i < Idd.size() - 2; i++) {
        for (unsigned int i1 = 0; i1 < Idd[i].size(); i1++) {
          for (unsigned int i2 = 0; i2 < Idd[i + 1].size(); i2++) {
            for (unsigned int i3 = 0; i3 < Idd[i + 2].size(); i3++) {
              if ((Idd[i][i1] != 0) && (Idd[i + 1][i2] != 0) && (Idd[i + 2][i3] != 0)) {
                vector<long int> tmp;
                tmp.push_back(Idd[i][i1]);
                tmp.push_back(Idd[i + 1][i2]);
                tmp.push_back(Idd[i + 2][i3]);

                bool newTrip = true;
                for (unsigned int iv = 0; iv < tripl.size(); iv++) {
                  if ((tripl[iv][0] == tmp[0]) && (tripl[iv][1] == tmp[1]) && (tripl[iv][2] == tmp[2]))
                    newTrip = false;
                }
                if (newTrip) {
                  tripl.push_back(tmp);
                }
              }
            }
          }
        }
      }
    }  //END BREM LOOP

    float sineta_brem = sinh(eta_br);

    //OUTPUT COLLECTION
    auto bfield = iSetup.getHandle(magFieldToken_);
    float nomField = bfield->nominalValue();

    TransientTrackingRecHit::ConstRecHitContainer glob_hits;
    OwnVector<TrackingRecHit> loc_hits;
    for (unsigned int i = 0; i < tripl.size(); i++) {
      auto DetMatch1 = (rphirecHits.product())->find(tripl[i][0]);
      auto DetMatch2 = (rphirecHits.product())->find(tripl[i][1]);
      auto DetMatch3 = (rphirecHits.product())->find(tripl[i][2]);
      if ((DetMatch1 == rphirecHits->end()) || (DetMatch2 == rphirecHits->end()) || (DetMatch3 == rphirecHits->end()))
        continue;
      auto DetSet1 = *DetMatch1;
      auto DetSet2 = *DetMatch2;
      auto DetSet3 = *DetMatch3;

      for (auto it1 = DetSet1.begin(); it1 != DetSet1.end(); ++it1) {
        GlobalPoint gp1 = tracker_->idToDet(tripl[i][0])->surface().toGlobal(it1->localPosition());

        bool tak1 = isGsfTrack(gsfRecHits, &(*it1));

        for (auto it2 = DetSet2.begin(); it2 != DetSet2.end(); ++it2) {
          GlobalPoint gp2 = tracker_->idToDet(tripl[i][1])->surface().toGlobal(it2->localPosition());
          bool tak2 = isGsfTrack(gsfRecHits, &(*it2));

          for (auto it3 = DetSet3.begin(); it3 != DetSet3.end(); ++it3) {
            //  ips++;
            GlobalPoint gp3 = tracker_->idToDet(tripl[i][2])->surface().toGlobal(it3->localPosition());
            bool tak3 = isGsfTrack(gsfRecHits, &(*it3));

            FastHelix helix(gp3, gp2, gp1, nomField, &*bfield);
            GlobalVector gv = helix.stateAtVertex().momentum();
            GlobalVector gv_corr(gv.x(), gv.y(), gv.perp() * sineta_brem);
            float ene = sqrt(gv_corr.mag2() + (pfmass * pfmass));

            GlobalPoint gp = helix.stateAtVertex().position();
            float ch = helix.stateAtVertex().charge();

            XYZTLorentzVector mom = XYZTLorentzVector(gv.x(), gv.y(), gv_corr.z(), ene);
            XYZTLorentzVector pos = XYZTLorentzVector(gp.x(), gp.y(), gp.z(), 0.);
            BaseParticlePropagator theOutParticle(RawParticle(mom, pos, ch), 0, 0, B_.z());
            int bgc = GoodCluster(theOutParticle, PPP, 0.3, true);

            if (gv.perp() < 0.5)
              continue;

            if (tak1 + tak2 + tak3 > 2)
              continue;

            if (bgc == -1)
              continue;
            bool clTak = false;
            for (unsigned int igcc = 0; igcc < gc.size(); igcc++) {
              if (clTak)
                continue;
              if (bgc == gc[igcc])
                clTak = true;
            }
            if (clTak)
              continue;

            GlobalTrajectoryParameters Gtp(gp1, gv, int(ch), &(*magfield_));
            glob_hits.clear();
            loc_hits.clear();
            glob_hits.push_back(hitBuilder_->build(it1->clone()));
            glob_hits.push_back(hitBuilder_->build(it2->clone()));
            glob_hits.push_back(hitBuilder_->build(it3->clone()));


            FreeTrajectoryState CSeed(Gtp, CurvilinearTrajectoryError(AlgebraicSymMatrix55(AlgebraicMatrixID())));
            TrajectoryStateOnSurface updatedState;

            for (int ih = 0; ih < 3; ih++) {
              TrajectoryStateOnSurface state =
                  (ih == 0) ? propagator_->propagate(CSeed, tracker_->idToDet(tripl[i][ih])->surface())
                            : propagator_->propagate(updatedState, tracker_->idToDet(tripl[i][ih])->surface());

              if (!state.isValid()) {
                ih = 3;
                continue;
              }

              updatedState = kfUpdator_->update(state, *glob_hits[ih]);
              loc_hits.push_back(glob_hits[ih]->hit()->clone());
              if (ih == 2) {
                PTrajectoryStateOnDet const& PTraj =
                    trajectoryStateTransform::persistentState(updatedState, tripl[i][2]);
                //      output->push_back(Trajectoryseed(PTraj,loc_hits,alongMomentum));
                unclean.push_back(make_pair(TrajectorySeed(PTraj, loc_hits, alongMomentum), make_pair(gv_corr, ch)));
              }
              //    }
            }
          }
        }
      }
    }
    vector<bool> inPhot = sharedHits(unclean);
    for (unsigned int iu = 0; iu < unclean.size(); iu++) {
      if (inPhot[iu])
        output->push_back(ConvBremSeed(unclean[iu].first, pft));
    }

  }  //END GSF TRACK COLLECTION LOOP
  LogDebug("ConvBremSeedProducerProducer") << "END";
  iEvent.put(std::move(output));
}

void ConvBremSeedProducer::beginRun(const edm::Run& run, const EventSetup& iSetup) {
  geomSearchTracker_ = &iSetup.getData(geomSearchTrackerToken_);

  geometry_ = &iSetup.getData(geometryToken_);

  tracker_ = &iSetup.getData(trackerToken_);

  magfield_ = &iSetup.getData(magFieldToken_beginRun_);
  B_ = magfield_->inTesla(GlobalPoint(0, 0, 0));

  fieldMap_ = &iSetup.getData(magFieldMapToken_);

  hitBuilder_ = &iSetup.getData(hitBuilderToken_);

  propagator_ = new PropagatorWithMaterial(alongMomentum, 0.0005, magfield_);
  kfUpdator_ = new KFUpdator();
}

void ConvBremSeedProducer::endRun(const edm::Run& run, const EventSetup& iSetup) {
  delete propagator_;
  delete kfUpdator_;
}

void ConvBremSeedProducer::initializeLayerMap() {
  // These are the BoundSurface&, the BoundDisk* and the BoundCylinder* for that layer
  //   const BoundSurface& theSurface = layer.surface();
  //   BoundDisk* theDisk = layer.disk();  // non zero for endcaps
  //   BoundCylinder* theCylinder = layer.cylinder(); // non zero for barrel
  //   int theLayer = layer.layerNumber(); // 1->3 PixB, 4->5 PixD,
  //                                       // 6->9 TIB, 10->12 TID,
  //                                       // 13->18 TOB, 19->27 TEC


  const std::vector<const BarrelDetLayer*>& barrelLayers = geomSearchTracker_->barrelLayers();
  LogDebug("FastTracker") << "Barrel DetLayer dump: ";
  for (auto bl = barrelLayers.begin(); bl != barrelLayers.end(); ++bl) {
    LogDebug("FastTracker") << "radius " << (**bl).specificSurface().radius();
  }

  const std::vector<const ForwardDetLayer*>& posForwardLayers = geomSearchTracker_->posForwardLayers();
  LogDebug("FastTracker") << "Positive Forward DetLayer dump: ";
  for (auto fl = posForwardLayers.begin(); fl != posForwardLayers.end(); ++fl) {
    LogDebug("FastTracker") << "Z pos " << (**fl).surface().position().z() << " radii "
                            << (**fl).specificSurface().innerRadius() << ", " << (**fl).specificSurface().outerRadius();
  }

  const float rTolerance = 1.5;
  const float zTolerance = 3.;

  LogDebug("FastTracker") << "Dump of TrackerInteractionGeometry cylinders:";
  for (std::list<TrackerLayer>::const_iterator i = geometry_->cylinderBegin(); i != geometry_->cylinderEnd(); ++i) {
    const BoundCylinder* cyl = i->cylinder();
    const BoundDisk* disk = i->disk();

    LogDebug("FastTracker") << "Famos Layer no " << i->layerNumber() << " is sensitive? " << i->sensitive() << " pos "
                            << i->surface().position();
    if (!i->sensitive())
      continue;

    if (cyl != nullptr) {
      LogDebug("FastTracker") << " cylinder radius " << cyl->radius();
      bool found = false;

      for (auto bl = barrelLayers.begin(); bl != barrelLayers.end(); ++bl) {
        if (fabs(cyl->radius() - (**bl).specificSurface().radius()) < rTolerance) {
          layerMap_[i->layerNumber()] = *bl;
          found = true;
          LogDebug("FastTracker") << "Corresponding DetLayer found with radius " << (**bl).specificSurface().radius();

          break;
        }
      }
      if (!found) {
        LogError("FastTracker") << "FAILED to find a corresponding DetLayer!";
      }
    } else {
      LogDebug("FastTracker") << " disk radii " << disk->innerRadius() << ", " << disk->outerRadius();

      bool found = false;

      for (auto fl = posForwardLayers.begin(); fl != posForwardLayers.end(); ++fl) {
        if (fabs(disk->position().z() - (**fl).surface().position().z()) < zTolerance) {
          layerMap_[i->layerNumber()] = *fl;
          found = true;
          LogDebug("FastTracker") << "Corresponding DetLayer found with Z pos " << (**fl).surface().position().z()
                                  << " and radii " << (**fl).specificSurface().innerRadius() << ", "
                                  << (**fl).specificSurface().outerRadius();
          break;
        }
      }
      if (!found) {
        LogError("FastTracker") << "FAILED to find a corresponding DetLayer!";
      }
    }
  }
}
const DetLayer* ConvBremSeedProducer::detLayer(const TrackerLayer& layer, float zpos) const {
  if (zpos > 0 || !layer.forward())
    return layerMap_[layer.layerNumber()];
  else
    return layerMap_[layer.layerNumber() + negLayerOffset_];
}

TrajectoryStateOnSurface ConvBremSeedProducer::makeTrajectoryState(const DetLayer* layer,
                                                                   const ParticlePropagator& pp,
                                                                   const MagneticField* field) const {
  GlobalPoint pos(pp.particle().X(), pp.particle().Y(), pp.particle().Z());
  GlobalVector mom(pp.particle().Px(), pp.particle().Py(), pp.particle().Pz());

  auto plane = layer->surface().tangentPlane(pos);

  return TrajectoryStateOnSurface(GlobalTrajectoryParameters(pos, mom, TrackCharge(pp.particle().charge()), field),
                                  *plane);
}
bool ConvBremSeedProducer::isGsfTrack(const reco::Track& tkv, const TrackingRecHit* h) {
  bool istaken = false;
  for (auto const& hit : tkv.recHits()) {
    if (istaken || !hit->isValid())
      continue;
    istaken = hit->sharesInput(h, TrackingRecHit::all);
  }
  return istaken;
}
vector<bool> ConvBremSeedProducer::sharedHits(const vector<pair<TrajectorySeed, pair<GlobalVector, float> > >& unclean) {
  vector<bool> goodseed;
  goodseed.clear();
  if (unclean.size() < 2) {
    for (unsigned int i = 0; i < unclean.size(); i++)
      goodseed.push_back(true);
  } else {
    for (unsigned int i = 0; i < unclean.size(); i++)
      goodseed.push_back(true);

    for (unsigned int iu = 0; iu < unclean.size() - 1; iu++) {
      if (!goodseed[iu])
        continue;
      for (unsigned int iu2 = iu + 1; iu2 < unclean.size(); iu2++) {
        if (!goodseed[iu])
          continue;
        if (!goodseed[iu2])
          continue;
        //    if (unclean[iu].second.second *unclean[iu2].second.second >0)continue;

        unsigned int shar = 0;
        for (auto const& sh : unclean[iu].first.recHits()) {
          for (auto const& sh2 : unclean[iu2].first.recHits()) {
            if (sh.sharesInput(&sh2, TrackingRecHit::all))

              shar++;
          }
        }
        if (shar >= 2) {
          if (unclean[iu].second.first.perp() < unclean[iu2].second.first.perp())
            goodseed[iu] = false;
          else
            goodseed[iu2] = false;
        }
      }
    }
  }
  return goodseed;
}

int ConvBremSeedProducer::GoodCluster(const BaseParticlePropagator& ubpg,
                                      const PFClusterCollection& pfc,
                                      float minep,
                                      bool sec) {
  BaseParticlePropagator bpg = ubpg;
  bpg.propagateToEcalEntrance(false);
  float dr = 1000;
  float de = 1000;
  float df = 1000;
  int ibest = -1;

  if (bpg.getSuccess() != 0) {
    for (unsigned int i = 0; i < pfc.size(); i++) {
      float tmp_ep = pfc[i].energy() / bpg.particle().momentum().e();
      float tmp_phi = fabs(pfc[i].position().phi() - bpg.particle().vertex().phi());
      if (tmp_phi > TMath::TwoPi())
        tmp_phi -= TMath::TwoPi();
      float tmp_eta = fabs(pfc[i].position().eta() - bpg.particle().vertex().eta());
      float tmp_dr = sqrt(pow(tmp_phi, 2) + pow(tmp_eta, 2));
      bool isBet = (tmp_dr < dr);
      if (sec)
        isBet = (tmp_phi < df);
      if ((isBet) && (tmp_ep > minep) && (tmp_ep < 10)) {
        dr = tmp_dr;
        de = tmp_eta;
        df = tmp_phi;
        ibest = i;
      }
    }
    bool isBad = (dr > 0.1);
    if (sec)
      isBad = ((df > 0.25) || (de > 0.5));

    if (isBad)
      ibest = -1;
  }
  return ibest;
}