CompetingRIOsOnTrack.h

Go to the documentation of this file.

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

// CompetingRIOsOnTrack.h, (c) ATLAS Detector software
 
#ifndef COMPETINGRIOSONTRACK_H
#define COMPETINGRIOSONTRACK_H
 
// Amg
#include "EventPrimitives/EventPrimitives.h"
#include "GeoPrimitives/GeoPrimitives.h"
 
// Trk
#include "TrkEventPrimitives/LocalParameters.h"
#include "TrkMeasurementBase/MeasurementBase.h"
#include <atomic>
#include <cassert>
#include <iosfwd>
#include <memory>
 
class MsgStream;
class CompetingRIOsOnTrackCnv_p1;
 
namespace Trk {
 
class PrepRawData;
class RIO_OnTrack;
class TrkDetElementBase;

 
class CompetingRIOsOnTrack : public MeasurementBase
{
 
public:
  typedef double AssignmentProb;

  CompetingRIOsOnTrack();
  CompetingRIOsOnTrack(std::vector<AssignmentProb>&& assgnProb);
  CompetingRIOsOnTrack(const CompetingRIOsOnTrack& compROT) = default;
  CompetingRIOsOnTrack(CompetingRIOsOnTrack&& compROT) noexcept = default;
  CompetingRIOsOnTrack& operator=(const CompetingRIOsOnTrack& compROT) = default;
  CompetingRIOsOnTrack& operator=(CompetingRIOsOnTrack&& compROT) noexcept = default;
  virtual ~CompetingRIOsOnTrack() = default;

  virtual CompetingRIOsOnTrack* clone() const override = 0;

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

  virtual unsigned int numberOfContainedROTs() const = 0;

  unsigned int indexOfMaxAssignProb() const;

  virtual const RIO_OnTrack& rioOnTrack(unsigned int) const = 0;

  AssignmentProb assignmentProbability(unsigned int indx) const;

  virtual void setLocalParametersAndErrorMatrix();

  virtual bool type(MeasurementBaseType::Type type) const override final
  {
    return (type == MeasurementBaseType::CompetingRIOsOnTrack);
  }
 
protected:
  friend class ::CompetingRIOsOnTrackCnv_p1;

  std::vector<AssignmentProb> m_assignProb;

  virtual bool ROTsHaveCommonSurface(const bool withNonVanishingAssignProb = true) const = 0;

  virtual MsgStream& dump(MsgStream& out) const override;
  virtual std::ostream& dump(std::ostream& out) const override;
};
 
inline CompetingRIOsOnTrack::AssignmentProb
CompetingRIOsOnTrack::assignmentProbability(unsigned int indx) const
{
  assert(indx < numberOfContainedROTs());
  if (indx < numberOfContainedROTs()) {
    return m_assignProb[indx];
  }
  return 0; // could consider throwing an exception here - EJWM
}
 
}
 
#endif // COMPETINGRIOSONTRACK_H