CTPPSDiamondTrackRecognition.h

Go to the documentation of this file.

/****************************************************************************
 *
 * This is a part of CTPPS offline software.
 * Authors:
 *   Laurent Forthomme (laurent.forthomme@cern.ch)
 *   Nicola Minafra (nicola.minafra@cern.ch)
 *
 ****************************************************************************/

#ifndef RecoCTPPS_TotemRPLocal_CTPPSDiamondTrackRecognition
#define RecoCTPPS_TotemRPLocal_CTPPSDiamondTrackRecognition

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/Common/interface/DetSet.h"
#include "DataFormats/Common/interface/DetSetVector.h"

#include "DataFormats/CTPPSReco/interface/CTPPSDiamondRecHit.h"
#include "DataFormats/CTPPSReco/interface/CTPPSDiamondLocalTrack.h"

#include <vector>
#include <unordered_map>
#include "TF1.h"

class CTPPSDiamondTrackRecognition
{
  public:
    CTPPSDiamondTrackRecognition( const edm::ParameterSet& );
    ~CTPPSDiamondTrackRecognition();

    void clear();

    void addHit( const CTPPSDiamondRecHit& recHit );

    int produceTracks( edm::DetSet<CTPPSDiamondLocalTrack>& tracks );

  private:
    struct HitParameters {
      HitParameters( const float center, const float width ) :
        center( center ), width( width ) {}
      float center;
      float width;
    };
    typedef std::vector<HitParameters> HitParametersVector;
    typedef std::unordered_map<int,HitParametersVector> HitParametersVectorMap;

    static const std::string pixelEfficiencyDefaultFunction_;

    const float threshold_;
    const float thresholdFromMaximum_;
    const float resolution_;
    const float sigma_;
    const float startFromX_;
    const float stopAtX_;

    float yPosition_;
    float yWidth_;
    float yPositionInitial_;
    float yWidthInitial_;

    TF1 hit_f_;
    HitParametersVectorMap hitParametersVectorMap_;
    std::unordered_map<int,int> mhMap_;
};

#endif