TotemRPUVPattern.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 DataFormats_CTPPSReco_TotemRPUVPattern
#define DataFormats_CTPPSReco_TotemRPUVPattern

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

class TotemRPUVPattern
{
  public:
    enum ProjectionType { projInvalid, projU, projV };

    TotemRPUVPattern() : projection(projInvalid), a(0.), b(0.), w(0.), fittable(false)
    {
    }

    ProjectionType getProjection() const { return projection; }
    void setProjection(ProjectionType p_) { projection = p_; }

    double getA() const { return a; }
    void setA(double a_) { a = a_; }

    double getB() const { return b; }
    void setB(double b_) { b = b_; }

    double getW() const { return w; }
    void setW(double w_) { w = w_; }

    bool getFittable() const { return fittable; }
    void setFittable(bool f_) { fittable = f_; }

    void addHit(edm::det_id_type detId, const TotemRPRecHit &hit)
    {
      hits.find_or_insert(detId).push_back(hit);
    }

    const edm::DetSetVector<TotemRPRecHit>& getHits() const { return hits; }

    friend bool operator< (const TotemRPUVPattern &l, const TotemRPUVPattern &r);

  private:
    ProjectionType projection;              
    double a;                               
    double b;                               
    double w;                               
    bool fittable;                          

    edm::DetSetVector<TotemRPRecHit> hits;  
};

//----------------------------------------------------------------------------------------------------

extern bool operator< (const TotemRPUVPattern &l, const TotemRPUVPattern &r);

#endif