SeedForPhotonConversionFromQuadruplets.h

Go to the documentation of this file.

#ifndef SeedForPhotonConversionFromQuadruplets_H
#define SeedForPhotonConversionFromQuadruplets_H

#include "RecoTracker/TkTrackingRegions/interface/GlobalTrackingRegion.h"
#include "RecoTracker/TkSeedingLayers/interface/SeedingHitSet.h"
#include "FWCore/Framework/interface/FrameworkfwdMostUsed.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"

#include "PrintRecoObjects.h"
#include "Quad.h"

#include "RecoTracker/TransientTrackingRecHit/interface/TkTransientTrackingRecHitBuilder.h"

class FreeTrajectoryState;
class SeedComparitor;
class IdealMagneticFieldRecord;
class TransientRecHitRecord;
class TrackerDigiGeometryRecord;
class TrackingComponentsRecord;

class SeedForPhotonConversionFromQuadruplets {
public:
  static const int cotTheta_Max = 99999;

  SeedForPhotonConversionFromQuadruplets(const edm::ParameterSet& cfg,
                                         edm::ConsumesCollector& iC,
                                         const edm::ParameterSet& SeedComparitorPSet);

  SeedForPhotonConversionFromQuadruplets(edm::ConsumesCollector& iC,
                                         const edm::ParameterSet& SeedComparitorPSet,
                                         const std::string& propagator = "PropagatorWithMaterial",
                                         double seedMomentumForBOFF = -5.0);

  //dtor
  ~SeedForPhotonConversionFromQuadruplets();

  const TrajectorySeed* trajectorySeed(TrajectorySeedCollection& seedCollection,
                                       const SeedingHitSet& phits,
                                       const SeedingHitSet& mhits,
                                       const TrackingRegion& region,
                                       const edm::Event& ev,
                                       const edm::EventSetup& es,
                                       std::stringstream& ss,
                                       std::vector<Quad>& quadV,
                                       edm::ParameterSet& QuadCutPSet);

  double simpleGetSlope(const SeedingHitSet::ConstRecHitPointer& ohit,
                        const SeedingHitSet::ConstRecHitPointer& nohit,
                        const SeedingHitSet::ConstRecHitPointer& ihit,
                        const SeedingHitSet::ConstRecHitPointer& nihit,
                        const TrackingRegion& region,
                        double& cotTheta,
                        double& z0);
  double verySimpleFit(int size, double* ax, double* ay, double* e2y, double& p0, double& e2p0, double& p1);
  double getSqrEffectiveErrorOnZ(const SeedingHitSet::ConstRecHitPointer& hit, const TrackingRegion& region);

  //
  // Some utility methods added by sguazz
  void stupidPrint(std::string s, float* d);
  void stupidPrint(std::string s, double* d);
  void stupidPrint(const char* s, GlobalPoint* d);
  void stupidPrint(const char* s, GlobalPoint* d, int n);
  void bubbleSortVsPhi(GlobalPoint arr[], int n, GlobalPoint vtx);
  void bubbleReverseSortVsPhi(GlobalPoint arr[], int n, GlobalPoint vtx);
  //
  //

protected:
  bool checkHit(const TrajectoryStateOnSurface&,
                const SeedingHitSet::ConstRecHitPointer& hit,
                const edm::EventSetup& es) const {
    return true;
  }

  GlobalTrajectoryParameters initialKinematic(const SeedingHitSet& hits,
                                              const GlobalPoint& vertexPos,
                                              const edm::EventSetup& es,
                                              const float cotTheta) const;

  CurvilinearTrajectoryError initialError(const GlobalVector& vertexBounds, float ptMin, float sinTheta) const;

  const TrajectorySeed* buildSeed(TrajectorySeedCollection& seedCollection,
                                  const SeedingHitSet& hits,
                                  const FreeTrajectoryState& fts,
                                  const edm::EventSetup& es,
                                  bool apply_dzCut,
                                  const TrackingRegion& region) const;

  bool buildSeedBool(TrajectorySeedCollection& seedCollection,
                     const SeedingHitSet& hits,
                     const FreeTrajectoryState& fts,
                     const edm::EventSetup& es,
                     bool apply_dzCut,
                     const TrackingRegion& region,
                     double dzcut) const;

  SeedingHitSet::RecHitPointer refitHit(SeedingHitSet::ConstRecHitPointer hit,
                                        const TrajectoryStateOnSurface& state) const;

  bool similarQuadExist(Quad& thisQuad, std::vector<Quad>& quadV);

  double DeltaPhiManual(const math::XYZVector& v1, const math::XYZVector& v2);

protected:
  edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> theBfieldToken;
  edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> theTTRHBuilderToken;
  edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> theTrackerToken;
  edm::ESGetToken<Propagator, TrackingComponentsRecord> thePropagatorToken;
  double theBOFFMomentum;
  double kPI_;

  // FIXME (well the whole class needs to be fixed!)
  TkClonerImpl cloner;

  std::stringstream* pss;
  PrintRecoObjects po;
  std::unique_ptr<SeedComparitor> theComparitor;
};
#endif