CDCCKFState.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 <genfit/MeasuredStateOnPlane.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <tracking/trackFindingCDC/eventdata/trajectories/CDCTrajectory3D.h>
#include <tracking/trackFindingCDC/geometry/Vector3D.h>
#include <tracking/trackFindingCDC/eventdata/hits/CDCWireHit.h>
#include <tracking/trackFindingCDC/numerics/ERightLeft.h>
 
namespace Belle2 {

  class CDCCKFState {
  public:
    CDCCKFState(const RecoTrack* seed, const genfit::MeasuredStateOnPlane& trackState) :
      m_seed(seed), m_trackState(trackState) {}

    explicit CDCCKFState(const TrackFindingCDC::CDCWireHit* wireHit) : m_cdcWireHit(wireHit) {}

    const TrackFindingCDC::CDCWireHit* getWireHit() const
    {
      B2ASSERT("State does not represent a wire hit", static_cast<bool>(m_cdcWireHit));
      return *m_cdcWireHit;
    }

    const RecoTrack* getSeed() const
    {
      B2ASSERT("State does not represent a seed", static_cast<bool>(m_seed));
      return *m_seed;
    }

    const RecoTrack* getMCRecoTrack(const std::string& mcRecoTrackStoreArrayName = "MCRecoTracks") const
    {
      const RecoTrack* seed = getSeed();
      return seed->getRelated<RecoTrack>(mcRecoTrackStoreArrayName);
    }

    bool isSeed() const
    {
      return static_cast<bool>(m_seed);
    }

    const genfit::MeasuredStateOnPlane& getTrackState() const
    {
      B2ASSERT("State does not have a track state (yet)", static_cast<bool>(m_trackState));
      return *m_trackState;
    }

    void setTrackState(const genfit::MeasuredStateOnPlane& trackState)
    {
      m_trackState = trackState;
    }

    void setRLinfo(const TrackFindingCDC::ERightLeft& rl)
    {
      m_rl = rl;
    }

    TrackFindingCDC::ERightLeft getRLinfo() const
    {
      B2ASSERT("LR info is not present yet", m_rl != TrackFindingCDC::ERightLeft::c_Unknown);
      return m_rl;
    }

    void setArcLength(double arcLength)
    {
      m_arcLength = arcLength;
    }

    double getArcLength() const
    {
      return m_arcLength;
    }

    void setHitDistance(double hitDistance)
    {
      m_hitDistance = hitDistance;
    }

    double getHitDistance() const
    {
      return m_hitDistance;
    }

    void setWeight(double weight)
    {
      m_weight = weight;
    }

    double getWeight() const
    {
      return m_weight;
    }

    void setChi2(double chi2)
    {
      m_chi2 = chi2;
    }

    double getChi2() const
    {
      return m_chi2;
    }

    void setReconstructedZ(double reconstructedZ)
    {
      m_reconstructedZ = reconstructedZ;
    }

    double getReconstructedZ() const
    {
      return m_reconstructedZ;
    }

    TrackFindingCDC::CDCTrajectory3D getTrajectory() const
    {
      const auto& trackState = getTrackState();
      const TrackFindingCDC::Vector3D trackPosition(trackState.getPos());
      const TrackFindingCDC::Vector3D trackMomentum(trackState.getMom());
      return TrackFindingCDC::CDCTrajectory3D(trackPosition, trackState.getTime(),
                                              trackMomentum, trackState.getCharge());
    }
 
  private:
    std::optional<const RecoTrack*> m_seed;
    std::optional<const TrackFindingCDC::CDCWireHit*> m_cdcWireHit;

    std::optional<genfit::MeasuredStateOnPlane> m_trackState;

    double m_arcLength = 0;

    double m_hitDistance = 0;

    double m_weight = 0;

    double m_chi2 = 0;

    double m_reconstructedZ = 0;

    TrackFindingCDC::ERightLeft m_rl = TrackFindingCDC::ERightLeft::c_Unknown;
  };

  std::ostream& operator<<(std::ostream& output, const CDCCKFState& state);
}