CDCFitter2D.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2014 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Oliver Frost                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/fitting/EFitVariance.h>
 
#include <vector>
 
namespace Belle2 {
  namespace TrackFindingCDC {
    class CDCTrack;
    class CDCSegment3D;
    class CDCAxialSegmentPair;
    class CDCSegment2D;
    class CDCWireHitSegment;
    class CDCWireHit;
    class CDCTrajectory2D;
    class CDCObservations2D;
 
    class CDCWire;
 
    template<class AFitMethod>
    class CDCFitter2D : public AFitMethod {
 
    public:
      CDCFitter2D();
 
      ~CDCFitter2D();
 
      CDCTrajectory2D fit(const CDCObservations2D& observations2D) const;
 
      CDCTrajectory2D fit(CDCObservations2D&& observations2D) const;
 
      void update(CDCTrajectory2D& trajectory2D, const CDCObservations2D& observations2D) const;
 
      void update(CDCTrajectory2D& trajectory2D, CDCObservations2D&& observations2D) const;
 
      CDCTrajectory2D fit(const CDCTrack& track) const;
 
      CDCTrajectory2D fit(const CDCSegment3D& segment) const;
 
      CDCTrajectory2D fit(const CDCSegment2D& fromSegment, const CDCSegment2D& toSegment) const;
 
      CDCTrajectory2D fit(const CDCSegment2D& segment) const;
 
      CDCTrajectory2D fit(const std::vector<const CDCWireHit*>& wireHit) const;
 
      CDCTrajectory2D fit(const std::vector<const CDCWire*>& wires) const;
 
      CDCTrajectory2D fit(const CDCWireHitSegment& wireHits) const;
 
      void update(CDCTrajectory2D& trajectory2D, const CDCSegment2D& segment) const;
 
      void update(CDCTrajectory2D& trajectory2D, const CDCAxialSegmentPair& axialSegmentPair) const;
 
    private:
      template<class AHits>
      CDCTrajectory2D fitGeneric(const AHits& hits) const;
 
      template<class AStartHits, class AEndHits>
      CDCTrajectory2D fitGeneric(const AStartHits& startHits, const AEndHits& endHits) const;
 
      template <class AHits>
      void updateGeneric(CDCTrajectory2D& trajectory2D, const AHits& hits) const;
 
      template <class AStartHits, class AEndHits>
      void updateGeneric(CDCTrajectory2D& trajectory2D,
                         const AStartHits& startHits,
                         const AEndHits& endHits) const;
 
    public:
      //set which information should be used from the recohits
      //useOnlyPosition is standard
 
      void useOnlyPosition();
 
      void useOnlyOrientation();
 
      void usePositionAndOrientation();
 
      void setFitVariance(EFitVariance fitVariance);
 
    private:
      bool m_usePosition = true;
 
      bool m_useOrientation = false;
 
      EFitVariance m_fitVariance = EFitVariance::c_Proper;
    };
  }
}