CompetingTRT_DriftCirclesOnTrack.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// CompetingTRT_DriftCirclesOnTrack.cxx, (c) ATLAS Detector software
 
// Trk
#include "GaudiKernel/MsgStream.h"
#include "TrkSurfaces/Surface.h"
// InDet
#include "InDetCompetingRIOsOnTrack/CompetingTRT_DriftCirclesOnTrack.h"
// std
#include <cmath>
#include <ostream>
 
// default constructor
InDet::CompetingTRT_DriftCirclesOnTrack::CompetingTRT_DriftCirclesOnTrack()
  : Trk::CompetingRIOsOnTrack()
  , Trk::SurfacePtrHolderDetEl(nullptr)
  , m_globalPosition{}
  , m_containedChildRots()
  , m_ROTsHaveCommonSurface(8)
{
}
 
// copy constructor
InDet::CompetingTRT_DriftCirclesOnTrack::CompetingTRT_DriftCirclesOnTrack(
  const InDet::CompetingTRT_DriftCirclesOnTrack& compROT)
  : Trk::CompetingRIOsOnTrack(compROT)
  , Trk::SurfacePtrHolderDetEl(compROT)
  , m_globalPosition{}
  , m_containedChildRots{}
  , m_ROTsHaveCommonSurface(compROT.m_ROTsHaveCommonSurface)
{
  std::vector<const InDet::TRT_DriftCircleOnTrack*>::const_iterator rotIter =
    compROT.m_containedChildRots.begin();
  for (; rotIter != compROT.m_containedChildRots.end(); ++rotIter) {
    m_containedChildRots.push_back((*rotIter)->clone());
  }
  if (compROT.m_globalPosition) {
    m_globalPosition.store(
      std::make_unique<const Amg::Vector3D>(*(compROT.m_globalPosition)));
  }
}
 
// explicit constructor
InDet::CompetingTRT_DriftCirclesOnTrack::CompetingTRT_DriftCirclesOnTrack(
  const Trk::Surface* sf,
  std::vector<const InDet::TRT_DriftCircleOnTrack*>&& childrots,
  std::vector<AssignmentProb>&& assgnProb,
  const Trk::LocalParameters& effecLocalPars,
  const Amg::MatrixX& effecLocalErrMat,
  int ROTsHaveComSrfc // meaning of the values are described in the definition
                      // of ROTsHaveCommonSurface()
  )
  : Trk::CompetingRIOsOnTrack(std::move(assgnProb))
  , Trk::SurfacePtrHolderDetEl(sf)
  , m_globalPosition{}
  , m_containedChildRots(std::move(childrots))
  , m_ROTsHaveCommonSurface(ROTsHaveComSrfc)
{
  m_localParams = effecLocalPars;
  m_localCovariance = effecLocalErrMat;
}
 
InDet::CompetingTRT_DriftCirclesOnTrack&
InDet::CompetingTRT_DriftCirclesOnTrack::operator=(
  const InDet::CompetingTRT_DriftCirclesOnTrack& compROT)
{
  if (this != &compROT) {
    // assignment operator of base class
    Trk::CompetingRIOsOnTrack::operator=(compROT);
    Trk::SurfacePtrHolderDetEl::operator=(compROT);
 
    // clear rots
    clearChildRotVector();
    m_containedChildRots.clear();
 
    if (compROT.m_globalPosition) {
      m_globalPosition.store(
        std::make_unique<const Amg::Vector3D>(*compROT.m_globalPosition));
    } else if (m_globalPosition) {
      m_globalPosition.release().reset();
    }
    //
    m_ROTsHaveCommonSurface = compROT.m_ROTsHaveCommonSurface;
    //
    std::vector<const InDet::TRT_DriftCircleOnTrack*>::const_iterator rotIter =
      compROT.m_containedChildRots.begin();
    for (; rotIter != compROT.m_containedChildRots.end(); ++rotIter)
      m_containedChildRots.push_back((*rotIter)->clone());
  }
  return (*this);
}
 
InDet::CompetingTRT_DriftCirclesOnTrack&
InDet::CompetingTRT_DriftCirclesOnTrack::operator=(
  InDet::CompetingTRT_DriftCirclesOnTrack&& compROT) noexcept
{
  if (this != &compROT) {
    // base class move
    Trk::CompetingRIOsOnTrack::operator=(compROT);
    Trk::SurfacePtrHolderDetEl::operator=(compROT);
    // rots
    clearChildRotVector();
    m_containedChildRots = std::move(compROT.m_containedChildRots);
    m_globalPosition = std::move(compROT.m_globalPosition);
    m_ROTsHaveCommonSurface = compROT.m_ROTsHaveCommonSurface;
 }
  return (*this);
}
 
InDet::CompetingTRT_DriftCirclesOnTrack::~CompetingTRT_DriftCirclesOnTrack()
{
  clearChildRotVector();
}
 
void
InDet::CompetingTRT_DriftCirclesOnTrack::clearChildRotVector()
{
  std::vector<const InDet::TRT_DriftCircleOnTrack*>::const_iterator rotIter =
    m_containedChildRots.begin();
  for (; rotIter != m_containedChildRots.end(); ++rotIter) {
    delete (*rotIter);
  }
}
 
MsgStream&
InDet::CompetingTRT_DriftCirclesOnTrack::dump(MsgStream& out) const
{
  using std::ios;
  out << "Trk::CompetingTRT_DriftCirclesOnTrack with ["
      << numberOfContainedROTs() << "] competing RIO_OnTrack objects"
      << std::endl;
  out << "  - "
      << (this->ROTsHaveCommonSurface(true) ? "on common surface"
                                            : "over different surfaces")
      << "  (given prob>cut)" << std::endl;
  Trk::CompetingRIOsOnTrack::dump(out);
  out << "  - GlobalPosition        : ";
  if (not m_globalPosition)
    out << "null pointer" << endmsg;
  else
    out << *m_globalPosition << endmsg;
  return out;
}
 
std::ostream&
InDet::CompetingTRT_DriftCirclesOnTrack::dump(std::ostream& out) const
{
  using std::ios;
  out << "Trk::CompetingTRT_DriftCirclesOnTrack with ["
      << numberOfContainedROTs() << "] competing RIO_OnTrack objects"
      << std::endl;
  out << "  - "
      << (this->ROTsHaveCommonSurface(true) ? "on common surface"
                                            : "over different surfaces")
      << "  (given prob>cut)" << std::endl;
  Trk::CompetingRIOsOnTrack::dump(out);
  out << "  - GlobalPosition        : ";
  if (not m_globalPosition)
    out << "null pointer" << std::endl;
  else
    out << *m_globalPosition << std::endl;
  return out;
}
 
// Have all the contained ROTs a common associated surface?
bool
InDet::CompetingTRT_DriftCirclesOnTrack::ROTsHaveCommonSurface(
  const bool withNonVanishingAssignProb) const
{
  // if (m_ROTsHaveCommonSurface==0) {
  // values of int m_ROTsHaveCommonSurface:
  //  no  yes  NA    (ROTs on a common surface)  NA means not checked yet
  //  0    1   2
  //     for all    | just ROTs with non-vanishing assgnProb
  //      * 1       |   * 3
  if (withNonVanishingAssignProb) {
    return (m_ROTsHaveCommonSurface / 3);
  } else {
    return (m_ROTsHaveCommonSurface % 3);
  }
}
 
const Amg::Vector3D&
InDet::CompetingTRT_DriftCirclesOnTrack::globalPosition() const
{
  if (m_globalPosition) {
    return *m_globalPosition;
  }
  // cannot use the localToGlobal transformation, because the local z-coordinate
  // along the wire is not known here. The contained TRT_DriftCircleOnTrack use
  // the full transformation => use the weighted mean of their GlobalPositions
  // FIXME: introduce a special function in base class, which returns the sum of
  // assignment probabilities
  double assgnProbSum = 0.;
  std::vector<AssignmentProb>::const_iterator assgnProbIter =
    m_assignProb.begin();
  for (; assgnProbIter != m_assignProb.end(); ++assgnProbIter) {
    assgnProbSum += (*assgnProbIter);
  }
 
  Amg::Vector3D globalPos(0., 0., 0.);
  if (assgnProbSum > 0.) {
    std::vector<const InDet::TRT_DriftCircleOnTrack*>::const_iterator rotIter =
      std::as_const(m_containedChildRots).begin();
    assgnProbIter = m_assignProb.begin();
    for (; rotIter != std::as_const(m_containedChildRots).end();
         ++rotIter, ++assgnProbIter) {
      globalPos +=
        (((*assgnProbIter) / assgnProbSum) * ((*rotIter)->globalPosition()));
    }
 
  } else {
    globalPos = std::as_const(m_containedChildRots)[0]->globalPosition();
  }
  m_globalPosition.set(std::make_unique<const Amg::Vector3D>(globalPos));
  return *m_globalPosition;
}
 
void
InDet::CompetingTRT_DriftCirclesOnTrack::setLocalParametersAndErrorMatrix()
{
  std::cout << "InDet::CompetingTRT_DriftCirclesOnTrack: can not handle ROTs "
               "in different surfaces without detector specific knowledge "
            << std::endl;
}