CDCSZObservations.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/fitting/EFitVariance.h>
 
#include <Math/Vector2D.h>
 
#include <vector>
 
namespace Belle2 {
 
  namespace TrackingUtilities {
    class CDCTrack;
    class CDCRecoHit3D;
    class CDCSegment3D;
  }
 
  namespace TrackFindingCDC {

    class CDCSZObservations {
 
    public:
      explicit CDCSZObservations(EFitVariance fitVariance = EFitVariance::c_Proper,
                                 bool onlyStereo = false)
        : m_fitVariance(fitVariance)
        , m_onlyStereo(onlyStereo)
      {
      }
 
    public:
      std::size_t size() const
      {
        return m_szObservations.size() / 3;
      }

      double* data()
      {
        return m_szObservations.data();
      }

      bool empty() const
      {
        return m_szObservations.empty();
      }

      void clear()
      {
        m_szObservations.clear();
      }

      void reserve(std::size_t nObservations)
      {
        m_szObservations.reserve(nObservations * 3);
      }

      double getS(int iObservation) const
      {
        return m_szObservations[iObservation * 3];
      }

      double getZ(int iObservation) const
      {
        return m_szObservations[iObservation * 3 + 1];
      }

      double getWeight(int iObservation) const
      {
        return m_szObservations[iObservation * 3 + 2];
      }

      std::size_t fill(double s, double z, double weight = 1.0);

      std::size_t append(const TrackingUtilities::CDCRecoHit3D& recoHit3D);

      std::size_t appendRange(const std::vector<TrackingUtilities::CDCRecoHit3D>& recoHit3Ds);

      std::size_t appendRange(const TrackingUtilities::CDCSegment3D& segment3D);

      std::size_t appendRange(const TrackingUtilities::CDCTrack& track);

      ROOT::Math::XYVector getCentralPoint() const;

      void passiveMoveBy(const ROOT::Math::XYVector& origin);

      ROOT::Math::XYVector centralize();
 
    public:
      void setFitVariance(EFitVariance fitVariance)
      {
        m_fitVariance = fitVariance;
      }
 
    private:
      std::vector<double> m_szObservations;

      EFitVariance m_fitVariance;

      bool m_onlyStereo;
    };
  }
}