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 RecoPPS_Local_FastLineRecognition
#define RecoPPS_Local_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);

private:
  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;

    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