FastLineRecognition.h

Go to the documentation of this file.

/****************************************************************************
*
* This is a part of TOTEM offline software.
* Authors: 
*   Jan Kašpar (jan.kaspar@gmail.com) 
*
****************************************************************************/

#ifndef RecoCTPPS_TotemRPLocal_FastLineRecognition
#define RecoCTPPS_TotemRPLocal_FastLineRecognition

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

#include "Geometry/VeryForwardGeometryBuilder/interface/CTPPSGeometry.h"
#include "DataFormats/CTPPSDetId/interface/TotemRPDetId.h"
#include "DataFormats/CTPPSReco/interface/TotemRPRecHit.h"
#include "DataFormats/CTPPSReco/interface/TotemRPUVPattern.h"


class FastLineRecognition
{
  public:
    FastLineRecognition(double cw_a = 0., double cw_b = 0.);

    ~FastLineRecognition();

    void resetGeometry(const CTPPSGeometry *_g)
    {
      geometry = _g;
      geometryMap.clear();
    }

    void getPatterns(const edm::DetSetVector<TotemRPRecHit> &input, double _z0, double threshold,
      edm::DetSet<TotemRPUVPattern> &patterns);

  protected:
    static const double sigma0;

    double z0;

    double chw_a, chw_b;

    double threshold;

    const CTPPSGeometry* geometry;

    struct GeomData
    {
      double z;   
      double s;   
    };

    std::map<unsigned int, GeomData> geometryMap;

    GeomData getGeomData(unsigned int id);

    struct Point
    {
      unsigned int detId;       
      const TotemRPRecHit* hit; 
      double h;                 
      double z;                 
      double w;                 
      bool usable;              
      Point(unsigned int _d=0, const TotemRPRecHit* _hit=nullptr, double _h=0., double _z=0., double _w=0.) :
        detId(_d), hit(_hit), h(_h), z(_z), w(_w), usable(true) {}
    };
    
    struct Cluster
    {
      double Saw, Sbw, Sw, S1;
      double weight;
      double min_a, max_a, min_b, max_b;

      std::vector<const Point *> contents;
      
      Cluster() : Saw(0.), Sbw(0.), Sw(0.), S1(0.), weight(0.) {}
      
      void add(const Point *p1, const Point *p2, double a, double b, double w);

      bool operator<(const Cluster &c) const
      {
        return (this->Sw > c.Sw) ? true : false;
      }
    };

    bool getOneLine(const std::vector<Point> &points, double threshold, Cluster &result);
};

#endif