CDCSZObservations.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 Vector2D;
    class CDCRecoHit3D;
    class CDCSegment3D;
    class CDCTrack;
 
    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 CDCRecoHit3D& recoHit3D);
 
      std::size_t appendRange(const std::vector<CDCRecoHit3D>& recoHit3Ds);
 
      std::size_t appendRange(const CDCSegment3D& segment3D);
 
      std::size_t appendRange(const CDCTrack& track);
 
      Vector2D getCentralPoint() const;
 
      void passiveMoveBy(const Vector2D& origin);
 
      Vector2D centralize();
 
    public:
      void setFitVariance(EFitVariance fitVariance)
      {
        m_fitVariance = fitVariance;
      }
 
    private:
      std::vector<double> m_szObservations;
 
      EFitVariance m_fitVariance;
 
      bool m_onlyStereo;
    };
  }
}