Tracking/TrkEvent/TrkSegment/TrkSegment/Segment.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/

// Segment.h, (c) ATLAS Detector software
 
#ifndef TRKSEGMENT_SEGMENT_H
#define TRKSEGMENT_SEGMENT_H
 
// Trk
#include "AthContainers/DataVector.h"
#include "TrkEventPrimitives/TrkObjectCounter.h"
#include "TrkMeasurementBase/MeasurementBase.h"
#include <atomic>
#include <memory>
#include <string>
#include <utility>
#include <vector>
class SegmentCnv_p1;
 
namespace Trk {
 
class PrepRawData;
class RIO_OnTrack;
class TrkDetElementBase;
class FitQuality;

 
class Segment
  : public MeasurementBase
  , public Trk::ObjectCounter<Trk::Segment>
{
 
public:
  enum Author
  {
    AuthorUnknown = 0,
    MooMdtSegmentMakerTool = 1,
    MooCscSegmentMakerTool = 2,
    Muonboy = 3,
    DCMathSegmentMaker = 4,
    MDT_DHoughSegmentMakerTool = 5,
    CSC_DHoughSegmentMakerTool = 6,
    Csc2dSegmentMaker = 7,
    Csc4dSegmentMaker = 8,
    TRT_SegmentMaker = 9,
    CTBTracking = 10,
    DCMathSegmentMakerCurved = 11,
    NswStereoSeeded = 12,
    NswStgcSeeded = 13,
    NswPadSeeded = 14,
    NswQuadAlign = 15,
    NumberOfAuthors = 16
  };

  Segment();
  Segment(const Segment& seg);
  Segment(Segment&&) noexcept;
  Segment& operator=(const Segment& seg);
  Segment& operator=(Segment&&) noexcept;

  Segment(LocalParameters&& locpars,
          Amg::MatrixX&& locerr,
          DataVector<const MeasurementBase>&& measurements,
          FitQuality* fitq = nullptr,
          Author author = AuthorUnknown);

  virtual ~Segment();

  virtual Segment* clone() const override = 0;

  std::unique_ptr<Segment> uniqueClone() const
  {
    return std::unique_ptr<Segment>(clone());
  }

  virtual bool type(MeasurementBaseType::Type type) const override final
  {
    return (type == MeasurementBaseType::Segment);
  }

  const std::vector<const Trk::MeasurementBase*>& containedMeasurements() const;
 
  const DataVector<const Trk::MeasurementBase>&
  containedMeasurementsDataVector() const;
 
  bool hasContainedMeasurements() const;
  unsigned int numberOfMeasurementBases() const;

  const MeasurementBase* measurement(unsigned int) const;

  const FitQuality* fitQuality() const;

  void setAuthor(Author a);
  Author author() const;

  std::string dumpAuthor() const;
 
protected:
  friend class ::SegmentCnv_p1;

  std::unique_ptr<FitQuality> m_fitQuality;

  DataVector<const MeasurementBase> m_containedMeasBases;

  Author m_author;
};
 
inline const FitQuality*
Segment::fitQuality() const
{
  return m_fitQuality.get();
}
 
inline const std::vector<const MeasurementBase*>&
Segment::containedMeasurements() const
{
  return m_containedMeasBases.stdcont();
}
 
inline const DataVector<const MeasurementBase>&
Segment::containedMeasurementsDataVector() const
{
  return m_containedMeasBases;
}
 
inline bool
Segment::hasContainedMeasurements() const
{
  return !m_containedMeasBases.empty();
}
 
inline const MeasurementBase*
Segment::measurement(unsigned int indx) const
{
  if (!m_containedMeasBases.empty() && indx < m_containedMeasBases.size()) {
    return std::as_const(m_containedMeasBases)[indx];
  }
  return nullptr;
}
 
inline unsigned int
Segment::numberOfMeasurementBases() const
{
  return m_containedMeasBases.size();
}
 
inline Segment::Author
Segment::author() const
{
  return m_author;
}
}
 
#endif // TRKSEGMENT_SEGMENT_H