FeasibleRLFacetFilter.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/filters/facet/FeasibleRLFacetFilter.h>
 
#include <tracking/trackingUtilities/eventdata/hits/CDCFacet.h>
#include <tracking/trackingUtilities/eventdata/hits/CDCRLWireHitTriple.h>
 
#include <tracking/trackingUtilities/numerics/Modulo.h>
 
#include <tracking/trackingUtilities/utilities/StringManipulation.h>
 
#include <framework/core/ModuleParamList.templateDetails.h>
 
using namespace Belle2;
using namespace TrackFindingCDC;
using namespace TrackingUtilities;
 
FeasibleRLFacetFilter::FeasibleRLFacetFilter(bool hardRLCut)
  : m_param_hardRLCut(hardRLCut)
{
}
 
void FeasibleRLFacetFilter::exposeParameters(ModuleParamList* moduleParamList,
                                             const std::string& prefix)
{
  Super::exposeParameters(moduleParamList, prefix);
  moduleParamList->addParameter(prefixed(prefix, "hardRLCut"),
                                m_param_hardRLCut,
                                "Switch to disallow the boarderline possible hit and "
                                "right left passage information.",
                                m_param_hardRLCut);
}
 
Weight FeasibleRLFacetFilter::operator()(const CDCFacet& facet)
{
  if (this->isFeasible(facet)) {
    return 3;
  } else {
    return NAN;
  }
}
 
bool FeasibleRLFacetFilter::isFeasible(const CDCRLWireHitTriple& rlWireHitTriple) const
{
  return isFeasible(rlWireHitTriple.getShape(),
                    rlWireHitTriple.getStartRLInfo(),
                    rlWireHitTriple.getMiddleRLInfo(),
                    rlWireHitTriple.getEndRLInfo());
}
 
bool FeasibleRLFacetFilter::isFeasible(const CDCRLWireHitTriple::Shape& shape,
                                       const ERightLeft startRLInfo,
                                       const ERightLeft middleRLInfo,
                                       const ERightLeft endRLInfo) const
{
  const short oClockDelta = shape.getOClockDelta();
  const short absOClockDelta = std::abs(oClockDelta);
  const short cellExtend = shape.getCellExtend();
 
  if (cellExtend + absOClockDelta > 6) {
    // funny formula, but basically checks the triple to be a progressing forward and not turning back in itself.
    return false;
  }
 
  const short stableTwist = -sign(shape.getOClockDelta()) * middleRLInfo;
  const bool startToMiddleIsCrossing = startRLInfo != middleRLInfo;
  const bool middleToEndIsCrossing = middleRLInfo != endRLInfo;
 
  const bool bothAreCrossing = startToMiddleIsCrossing and middleToEndIsCrossing;
 
  const bool startToMiddleIsLong = shape.getStartToMiddleCellDistance() > shape.getMiddleToEndCellDistance();
  const bool shortArmIsCrossing = startToMiddleIsLong ? middleToEndIsCrossing : startToMiddleIsCrossing;
 
  const bool onlyOneShortArm = isOdd(cellExtend);
  const bool shortArmsAreCrossing = bothAreCrossing or (onlyOneShortArm and shortArmIsCrossing);
  const bool noneAreCrossing = not startToMiddleIsCrossing and not middleToEndIsCrossing;
 
  const bool orthoHard = stableTwist > 0 and shortArmsAreCrossing;
  const bool ortho = stableTwist > 0 and not noneAreCrossing;
  const bool meta = stableTwist > 0 or noneAreCrossing;
  const bool para = not bothAreCrossing;
 
  // Redundant check saves a bit of computation time.
  if (oClockDelta == 0) {
    return para;
  }
 
  switch (cellExtend) {
    case 2:
      switch (absOClockDelta) {
        case 0:
          return para;
          break;
 
        case 2:
          return ortho or (not m_param_hardRLCut and meta);
          break;
 
        case 4:
          return orthoHard or (not m_param_hardRLCut and ortho);
          break;
      }
      break;
 
    case 3:
      switch (absOClockDelta) {
        case 0:
          return para;
          break;
 
        case 1:
          return meta;
          break;
 
        case 2:
          return orthoHard or (not m_param_hardRLCut and ortho);
          break;
 
        case 3:
          return orthoHard;
          break;
      }
      break;
 
    case 4:
      switch (absOClockDelta) {
        case 0:
          return para;
          break;
 
        case 1:
          return ortho or (not m_param_hardRLCut and meta);
          break;
 
        case 2:
          return orthoHard;
          break;
      }
      break;
  }
  return false;
}