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