CDCSegment.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2013 - 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/eventdata/trajectories/CDCTrajectory2D.h>
 
#include <tracking/trackFindingCDC/topology/ISuperLayer.h>
#include <tracking/trackFindingCDC/topology/EStereoKind.h>
 
#include <vector>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    template <class T>
    class CDCSegment : public std::vector<T> {
 
    public:
      CDCSegment() = default;
 
      EStereoKind getStereoKind() const
      {
        return ISuperLayerUtil::getStereoKind(getISuperLayer());
      }
 
      bool isAxial() const
      {
        return getStereoKind() == EStereoKind::c_Axial;
      }
 
      ISuperLayer getISuperLayer() const
      {
        return ISuperLayerUtil::getFrom(this->front());
      }
 
      const std::vector<T>& items() const
      {
        return *this;
      }
 
      CDCTrajectory2D& getTrajectory2D() const
      {
        return m_trajectory2D;
      }
 
      void setTrajectory2D(const CDCTrajectory2D& trajectory2D) const
      {
        m_trajectory2D = trajectory2D;
      }
 
      void sortByArcLength2D()
      {
        std::stable_sort(this->begin(), this->end(), [](const T & recoHit, const T & otherRecoHit) {
          return recoHit.getArcLength2D() < otherRecoHit.getArcLength2D();
        });
      }
 
      double getAliasScore() const
      {
        return m_aliasScore;
      }
 
      void setAliasScore(double aliasScore)
      {
        m_aliasScore = aliasScore;
      }
 
    private:
      mutable CDCTrajectory2D m_trajectory2D;
 
      double m_aliasScore;
 
    };
 
  }
}