TotemRPLocalTrackFitterAlgorithm.h

Go to the documentation of this file.

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

#ifndef RecoCTPPS_TotemRPLocal_TotemRPLocalTrackFitterAlgorithm
#define RecoCTPPS_TotemRPLocal_TotemRPLocalTrackFitterAlgorithm

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

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

#include "Geometry/VeryForwardGeometryBuilder/interface/CTPPSGeometry.h"
#include "Geometry/VeryForwardRPTopology/interface/RPTopology.h"

#include "TVector3.h"
#include "TVector2.h"

#include <unordered_map>

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

class TotemRPLocalTrackFitterAlgorithm
{
  public:
    TotemRPLocalTrackFitterAlgorithm(const edm::ParameterSet &conf);

    bool fitTrack(const edm::DetSetVector<TotemRPRecHit> &hits, double z_0, const CTPPSGeometry &tot_geom, TotemRPLocalTrack &fitted_track);

    void reset();

  private:
    struct RPDetCoordinateAlgebraObjs
    {
      TVector3 centre_of_det_global_position_;
      double rec_u_0_;              
      TVector2 readout_direction_;  
      bool available_;              
    };

    std::unordered_map<unsigned int, RPDetCoordinateAlgebraObjs> det_data_map_;

    RPTopology rp_topology_;

    RPDetCoordinateAlgebraObjs *getDetAlgebraData(unsigned int det_id, const CTPPSGeometry &tot_rp_geom);

    RPDetCoordinateAlgebraObjs prepareReconstAlgebraData(unsigned int det_id, const CTPPSGeometry &tot_rp_geom);

    void multiplyByDiagonalInPlace(TMatrixD &mt, const TVectorD &diag);
    
    static TVector3 convert3vector(const CLHEP::Hep3Vector & v)
    {
      return TVector3(v.x(),v.y(),v.z()) ;
    }
};

#endif