InPhi0ImpactBox.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Oliver Frost                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/hough/boxes/Box.h>
#include <tracking/trackFindingCDC/topology/ILayer.h>
 
#include <array>
#include <cmath>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    class InPhi0ImpactBox {
 
    public:
      explicit InPhi0ImpactBox(float curlCurv = NAN)
        : m_curlCurv(curlCurv)
      {}
 
    public:
      using HoughBox = Box<DiscretePhi0, ContinuousImpact>;
 
    public:
      ESign getDistanceSign(const HoughBox& houghBox,
                            float x,
                            float y,
                            float l,
                            float /*dxdz*/ = 0,
                            float /*dydz*/ = 0,
                            ILayer /*iCLayer*/ = -1) const
      {
        const std::array<DiscretePhi0, 2>& phi0Vec = houghBox.getBounds<DiscretePhi0>();
        const std::array<ContinuousImpact, 2>& impact = houghBox.getBounds<ContinuousImpact>();
 
        std::array<float, 2> xRot;
        xRot[0] = x * phi0Vec[0]->x() + y * phi0Vec[0]->y();
        xRot[1] = x * phi0Vec[1]->x() + y * phi0Vec[1]->y();
 
        const bool onlyPositiveArm = 0 < m_curlCurv;
        if (onlyPositiveArm) {
          // Reject hit if it is on the inward going branch but the curvature suggest it is no curler
          if (xRot[0] < 0 and xRot[1] < 0) return ESign::c_Invalid;
        }
 
        std::array<float, 2> yRotPlusL;
        yRotPlusL[0] = -x * phi0Vec[0]->y() + y * phi0Vec[0]->x() + l;
        yRotPlusL[1] = -x * phi0Vec[1]->y() + y * phi0Vec[1]->x() + l;
 
        // Using binary notation encoding lower and upper box bounds to fill the flat array.
        std::array<float, 4> dist;
        dist[0b00] = yRotPlusL[0] - static_cast<float>(impact[0]);
        dist[0b10] = yRotPlusL[1] - static_cast<float>(impact[0]);
        dist[0b01] = yRotPlusL[0] - static_cast<float>(impact[1]);
        dist[0b11] = yRotPlusL[1] - static_cast<float>(impact[1]);
 
        return ESignUtil::common(dist);
      }
 
    private:
      float m_curlCurv;
    };
  }
}