CDCAxialSegmentPair.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#include <tracking/trackingUtilities/eventdata/tracks/CDCAxialSegmentPair.h>
 
#include <tracking/trackingUtilities/eventdata/segments/CDCSegment2D.h>
#include <tracking/trackingUtilities/eventdata/segments/CDCAxialSegment2D.h>
 
#include <tracking/trackingUtilities/eventdata/trajectories/CDCTrajectory2D.h>
 
#include <cdc/topology/ISuperLayer.h>
 
#include <tracking/trackingUtilities/ca/AutomatonCell.h>
 
#include <tracking/trackingUtilities/numerics/EForwardBackward.h>
 
#include <framework/logging/Logger.h>
 
using namespace Belle2;
using namespace CDC;
using namespace TrackingUtilities;
 
CDCAxialSegmentPair::CDCAxialSegmentPair(const CDCAxialSegment2D* startSegment,
                                         const CDCAxialSegment2D* endSegment)
  : m_startSegment(startSegment)
  , m_endSegment(endSegment)
{
  B2ASSERT("CDCAxialSegmentPair initialized with nullptr as start segment", startSegment);
  B2ASSERT("CDCAxialSegmentPair initialized with nullptr as end segment", endSegment);
}
 
CDCAxialSegmentPair::CDCAxialSegmentPair(const CDCAxialSegment2D* startSegment,
                                         const CDCAxialSegment2D* endSegment,
                                         const CDCTrajectory2D& trajectory2D)
  : m_startSegment(startSegment)
  , m_endSegment(endSegment)
  , m_trajectory2D(trajectory2D)
{
  B2ASSERT("CDCAxialSegmentPair initialized with nullptr as start segment", startSegment);
  B2ASSERT("CDCAxialSegmentPair initialized with nullptr as end segment", endSegment);
}
 
ISuperLayer CDCAxialSegmentPair::getStartISuperLayer() const
{
  return ISuperLayerUtil::getFrom(getStartSegment());
}
 
ISuperLayer CDCAxialSegmentPair::getEndISuperLayer() const
{
  return ISuperLayerUtil::getFrom(getEndSegment());
}
 
void CDCAxialSegmentPair::unsetAndForwardMaskedFlag() const
{
  getAutomatonCell().unsetMaskedFlag();
  const bool toHits = true;
  getStartSegment()->unsetAndForwardMaskedFlag(toHits);
  getEndSegment()->unsetAndForwardMaskedFlag(toHits);
}
 
void CDCAxialSegmentPair::setAndForwardMaskedFlag() const
{
  getAutomatonCell().setMaskedFlag();
  const bool toHits = true;
  getStartSegment()->setAndForwardMaskedFlag(toHits);
  getEndSegment()->setAndForwardMaskedFlag(toHits);
}
 
void CDCAxialSegmentPair::receiveMaskedFlag() const
{
  const bool fromHits = true;
  getStartSegment()->receiveMaskedFlag(fromHits);
  getEndSegment()->receiveMaskedFlag(fromHits);
 
  if (getStartSegment()->getAutomatonCell().hasMaskedFlag() or
      getEndSegment()->getAutomatonCell().hasMaskedFlag()) {
    getAutomatonCell().setMaskedFlag();
  }
}
 
EForwardBackward CDCAxialSegmentPair::isCoaligned(const CDCTrajectory2D& trajectory2D) const
{
  EForwardBackward startIsCoaligned = trajectory2D.isForwardOrBackwardTo(*(getStartSegment()));
  EForwardBackward endIsCoaligned = trajectory2D.isForwardOrBackwardTo(*(getEndSegment()));
  if (startIsCoaligned == EForwardBackward::c_Forward and
      endIsCoaligned == EForwardBackward::c_Forward) {
    return EForwardBackward::c_Forward;
  } else if (startIsCoaligned == EForwardBackward::c_Backward and
             endIsCoaligned == EForwardBackward::c_Backward) {
    return EForwardBackward::c_Backward;
  } else {
    return EForwardBackward::c_Invalid;
  }
}