TrkrClusterSourceLink.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file TrkrClusterSourceLink.hpp

#pragma once

#include "Acts/EventData/Measurement.hpp"
#include "Acts/EventData/MeasurementHelpers.hpp"
#include "Acts/EventData/SourceLinkConcept.hpp"
#include "Acts/EventData/detail/fittable_type_generator.hpp"
#include "Acts/Geometry/GeometryIdentifier.hpp"

#include "ActsExamples/EventData/GeometryContainers.hpp"
#include <boost/container/flat_map.hpp>
#include <boost/container/flat_set.hpp>

namespace ActsExamples {

class TrkrClusterSourceLink
{
 public:

  TrkrClusterSourceLink(uint64_t cluskey,
      std::shared_ptr<const Acts::Surface> surface,
      Acts::BoundVector loc,
      Acts::BoundMatrix cov)
    : m_cluskey(cluskey)
    , m_surface(surface)
    , m_geoId(surface->geometryId())
    , m_loc(loc)
    , m_cov(cov)
{
}

  TrkrClusterSourceLink()                             = default;
  TrkrClusterSourceLink(TrkrClusterSourceLink&&)      = default;
  TrkrClusterSourceLink(const TrkrClusterSourceLink&) = default;

  TrkrClusterSourceLink& operator=(TrkrClusterSourceLink&&)      = default;
  TrkrClusterSourceLink& operator=(const TrkrClusterSourceLink&) = default;

  const Acts::BoundVector location() const
  {
    return m_loc;
  }
  const Acts::BoundMatrix covariance() const
  {
    return m_cov;
  }

  const Acts::GeometryIdentifier geoId() const 
  {
    return m_geoId;
  }

  const Acts::Surface& referenceSurface() const 
  {
    return *m_surface;
  }

  Acts::FittableMeasurement<TrkrClusterSourceLink> operator*() const
  {

    return Acts::Measurement<TrkrClusterSourceLink, 
           Acts::BoundIndices,
           Acts::eBoundLoc0,
           Acts::eBoundLoc1>
      {m_surface,
    *this,
    m_cov.topLeftCorner<2, 2>(),
    m_loc[0],
    m_loc[1]
    };
  }

  uint64_t cluskey() const
  {
    return m_cluskey;
  }


private:

  uint64_t m_cluskey;
  std::shared_ptr<const Acts::Surface> m_surface;
  Acts::GeometryIdentifier m_geoId;

  Acts::BoundVector m_loc;
  Acts::BoundMatrix m_cov;

  friend constexpr bool
  operator==(const TrkrClusterSourceLink& lhs, const TrkrClusterSourceLink& rhs)
  {
    return lhs.m_cluskey == rhs.m_cluskey;
  }

};

    static_assert(Acts::SourceLinkConcept<TrkrClusterSourceLink>, 
      "TrkrClusterSourceLink does not fulfill SourceLinkConcept");

  // Construct a container for TrkrSourceLinks
  using TrkrClusterSourceLinkContainer = GeometryIdMultiset<TrkrClusterSourceLink>;

}