PrepRawData.h

Go to the documentation of this file.

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

// PrepRawData.h, (c) ATLAS Detector software
 
#ifndef TRKPREPRAWDATA_PREPRAWDATA_H
#define TRKPREPRAWDATA_PREPRAWDATA_H
 
#include "Identifier/Identifier.h"
//
#include "AthLinks/tools/IdentContIndex.h"
 
// Eigen ATLAS helpers
#include "EventPrimitives/EventPrimitives.h"
#include "GeoPrimitives/GeoPrimitives.h"
#include "TrkEventPrimitives/TrkObjectCounter.h"
//
#include <atomic>
#include <iostream>
#include <memory>
#include <vector>
 
class MsgStream;
class PrepRawDataCnv_p1;
 
namespace Muon {
class RpcPrepDataContainerCnv_p1;
}
 
namespace Trk {
 
/* The various kind of PrepRawData
 * to avoid dynamic_cast via using the type method
 */
enum class PrepRawDataType
{
  SiCluster,
  PixelCluster,
  SCT_Cluster,
  TRT_DriftCircle,
  HGTD_Cluster,
  PlanarCluster,
  MdtPrepData,
  CscStripPrepData,
  CscPrepData,
  MMPrepData,
  RpcPrepData,
  TgcPrepData,
  sTgcPrepData
};
 
class TrkDetElementBase;

class PrepRawDataUndefinedVariable
{};
 
class PrepRawData : public Trk::ObjectCounter<Trk::PrepRawData>
{
  // Public methods:
public:
  PrepRawData();
  PrepRawData(const PrepRawData&) = default;
  PrepRawData(PrepRawData&&) noexcept = default;
  PrepRawData& operator=(const PrepRawData&) = default;
  PrepRawData& operator=(PrepRawData&&) noexcept = default;
  virtual ~PrepRawData() = default;

  PrepRawData(const Identifier& clusId,
              const Amg::Vector2D& locpos,
              const std::vector<Identifier>& rdoList,
              const Amg::MatrixX& locerr);

  PrepRawData(const Identifier& clusId,
              const Amg::Vector2D& locpos,
              std::vector<Identifier>&& rdoList,
              Amg::MatrixX&& locerr);

  PrepRawData(const Identifier& clusId,
              const Amg::Vector2D& locpos,
              const Amg::MatrixX& locerr);
 
  PrepRawData(const Identifier& clusId,
              const Amg::Vector2D& locpos,
              Amg::MatrixX&& locerr);

  Identifier identify() const;

  const Amg::Vector2D& localPosition() const;

  const std::vector<Identifier>& rdoList() const;

  const Amg::MatrixX& localCovariance() const;
  bool hasLocalCovariance() const;

  virtual const TrkDetElementBase* detectorElement() const = 0;

  virtual bool type(PrepRawDataType type) const {
      return prdType() == type;
  }

  virtual PrepRawDataType prdType() const = 0;

  virtual MsgStream& dump(MsgStream& stream) const;
  virtual std::ostream& dump(std::ostream& stream) const;

  void setHashAndIndex(unsigned short collHash, unsigned short objIndex);
  const IdentContIndex& getHashAndIndex() const;
 
private:
  friend class ::PrepRawDataCnv_p1;
  friend class Muon::RpcPrepDataContainerCnv_p1;

  Identifier m_clusId{ 0 };
  // Need to force proper alignment; otherwise cling gets it wrong.
  alignas(16) Amg::Vector2D m_localPos{};
  std::vector<Identifier> m_rdoList{};
  Amg::MatrixX m_localCovariance{};

  IdentContIndex m_indexAndHash;
};
 
MsgStream&
operator<<(MsgStream& stream, const PrepRawData& prd);
std::ostream&
operator<<(std::ostream& stream, const PrepRawData& prd);
}
 
#include "TrkPrepRawData/PrepRawData.icc"
#endif // TRKPREPRAWDATA_PREPRAWDATA_H