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 RecoPPS_Local_TotemRPLocalTrackFitterAlgorithm
#define RecoPPS_Local_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 CTPPSGeometry::Vector &v) { return TVector3(v.x(), v.y(), v.z()); }
};
 
#endif