CDCSegment3D.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/trackFindingCDC/eventdata/segments/CDCSegment3D.h>
 
#include <tracking/trackFindingCDC/eventdata/segments/CDCSegment2D.h>
#include <tracking/trackFindingCDC/eventdata/hits/CDCRecoHit3D.h>
#include <tracking/trackFindingCDC/eventdata/hits/CDCRecoHit2D.h>
#include <tracking/trackFindingCDC/eventdata/hits/CDCWireHit.h>
 
#include <tracking/trackFindingCDC/ca/AutomatonCell.h>
 
using namespace Belle2;
using namespace TrackFindingCDC;
 
CDCSegment3D CDCSegment3D::reconstruct(const CDCSegment2D& segment2D,
                                       const CDCTrajectory2D& trajectory2D)
{
  CDCSegment3D segment3D;
  for (const CDCRecoHit2D& recoHit2D : segment2D) {
    CDCRecoHit3D recoHit3D = CDCRecoHit3D::reconstruct(recoHit2D, trajectory2D);
    segment3D.push_back(recoHit3D);
  }
  segment3D.setAliasScore(segment2D.getAliasScore());
  return segment3D;
}
 
CDCSegment2D CDCSegment3D::stereoProjectToRef() const
{
  CDCSegment2D result;
  for (const CDCRecoHit3D&  recoHit3D : *this) {
    result.push_back(recoHit3D.stereoProjectToRef());
  }
  result.setAliasScore(getAliasScore());
  return result;
}
 
void CDCSegment3D::unsetAndForwardMaskedFlag(bool toHits) const
{
  getAutomatonCell().unsetMaskedFlag();
  if (not toHits) return;
  for (const CDCRecoHit3D& recoHit3D : *this) {
    const CDCWireHit& wireHit = recoHit3D.getWireHit();
    wireHit.getAutomatonCell().unsetMaskedFlag();
  }
}
 
void CDCSegment3D::setAndForwardMaskedFlag(bool toHits) const
{
  getAutomatonCell().setMaskedFlag();
  if (not toHits) return;
  for (const CDCRecoHit3D& recoHit3D : *this) {
    const CDCWireHit& wireHit = recoHit3D.getWireHit();
    wireHit.getAutomatonCell().setMaskedFlag();
  }
}
 
void CDCSegment3D::receiveMaskedFlag(bool fromHits) const
{
  if (not fromHits) return;
  int nMasked = 0 ;
  int nNotMasked = 0;
  for (const CDCRecoHit3D& recoHit3D : *this) {
    const CDCWireHit& wireHit = recoHit3D.getWireHit();
    if (wireHit.getAutomatonCell().hasMaskedFlag()) {
      ++nMasked;
    } else {
      ++nNotMasked;
    }
  }
 
  const int nTolerance = 2;
  if (nMasked > nNotMasked or nMasked >= nTolerance) {
    getAutomatonCell().setMaskedFlag();
  }
}