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