CMSMidpointAlgorithm.h

Go to the documentation of this file.

#ifndef JetAlgorithms_CMSMidpointAlgorithm_h
#define JetAlgorithms_CMSMidpointAlgorithm_h

#include "RecoParticleFlow/PFRootEvent/interface/JetRecoTypes.h"

class CMSMidpointAlgorithm 
{
 public:
  typedef std::vector<ProtoJet*> InternalCollection;

  CMSMidpointAlgorithm () :
    theSeedThreshold(3.0),
    theConeRadius(0.5),
    theConeAreaFraction(1.0),
    theMaxPairSize(2),
    theMaxIterations(100),
    theOverlapThreshold(0.75),
    theDebugLevel(0)
  { }

  CMSMidpointAlgorithm(double fSeedThreshold, double fConeRadius, double fConeAreaFraction, 
               int fMaxPairSize, int fMaxIterations, double fOverlapThreshold, int fDebugLevel) : 
    theSeedThreshold(fSeedThreshold),
    theConeRadius(fConeRadius),
    theConeAreaFraction(fConeAreaFraction),
    theMaxPairSize(fMaxPairSize),
    theMaxIterations(fMaxIterations),
    theOverlapThreshold(fOverlapThreshold),
    theDebugLevel(fDebugLevel)
  { }

  void run(const JetReco::InputCollection& fInput, JetReco::OutputCollection* fOutput);

 private:
  void findStableConesFromSeeds(const JetReco::InputCollection& fInput, InternalCollection* fOutput);

  void iterateCone(const JetReco::InputCollection& fInput,
           double startRapidity, double startPhi, double startPt, bool reduceConeSize, 
           InternalCollection* fOutput);

  void findStableConesFromMidPoints(const JetReco::InputCollection& fInput, InternalCollection* fOutput);

  void addClustersToPairs(const JetReco::InputCollection& fInput,
              std::vector<int>& testPair, std::vector<std::vector<int> >& pairs,
              std::vector<std::vector<bool> >& distanceOK, int maxClustersInPair);

  void splitAndMerge(const JetReco::InputCollection& fInput,
             InternalCollection* fProtoJets, JetReco::OutputCollection* fFinalJets);


  double theSeedThreshold;
  double theConeRadius;
  double theConeAreaFraction;
  int    theMaxPairSize;
  int    theMaxIterations;
  double theOverlapThreshold;
  int    theDebugLevel;
};

#endif