ROIPixelTranslator.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/pxdDataReductionClasses/ROIPixelTranslator.h>
#include <framework/logging/Logger.h>
#include <vxd/geometry/GeoCache.h>
#include <vxd/geometry/SensorInfoBase.h>
 
using namespace std;
using namespace Belle2;
 
ROIPixelTranslator::ROIPixelTranslator(double user_sigmaSystU, double user_sigmaSystV, double user_numSigmaTotU,
                                       double user_numSigmaTotV, double user_maxWidthU, double user_maxWidthV)
  : m_sigmaSystU(user_sigmaSystU)
  , m_sigmaSystV(user_sigmaSystV)
  , m_numSigmaTotU(user_numSigmaTotU)
  , m_numSigmaTotV(user_numSigmaTotV)
  , m_maxWidthU(user_maxWidthU)
  , m_maxWidthV(user_maxWidthV)
{}
 
 
ROIPixelTranslator::ROIPixelTranslator(const ROIinfo* theROIinfo)
  : m_sigmaSystU(theROIinfo->sigmaSystU)
  , m_sigmaSystV(theROIinfo->sigmaSystV)
  , m_numSigmaTotU(theROIinfo->numSigmaTotU)
  , m_numSigmaTotV(theROIinfo->numSigmaTotV)
  , m_maxWidthU(theROIinfo->maxWidthU)
  , m_maxWidthV(theROIinfo->maxWidthV)
{}
 
 
ROIPixelTranslator::~ROIPixelTranslator()
{}
 
 
void
ROIPixelTranslator::fillRoiIDList(StoreArray<PXDIntercept>* listOfIntercepts,
                                  StoreArray<ROIid>* ROIidList)
{
 
  VXD::GeoCache& aGeometry = VXD::GeoCache::getInstance();
 
  for (int i = 0; i < listOfIntercepts->getEntries(); i++) {
 
    B2DEBUG(10, "  --->> a NEW INTERCEPT!");
 
 
    const VXD::SensorInfoBase& aSensorInfo = aGeometry.getSensorInfo((*listOfIntercepts)[i]->getSensorID());
 
    double widthTotU = std::min(m_maxWidthU ,
                                sqrt((*listOfIntercepts)[i]->getSigmaU() * (*listOfIntercepts)[i]->getSigmaU() + m_sigmaSystU * m_sigmaSystU) * m_numSigmaTotU);
    double widthTotV = std::min(m_maxWidthV ,
                                sqrt((*listOfIntercepts)[i]->getSigmaV() * (*listOfIntercepts)[i]->getSigmaV() + m_sigmaSystV * m_sigmaSystV) * m_numSigmaTotV);
 
    double minU = (*listOfIntercepts)[i]->getCoorU() - widthTotU / 2 ;
    double maxU = (*listOfIntercepts)[i]->getCoorU() + widthTotU / 2 ;
    const int nPixelsU = aSensorInfo.getUCells() - 1;
 
    double minV = (*listOfIntercepts)[i]->getCoorV() - widthTotV / 2;
    double maxV = (*listOfIntercepts)[i]->getCoorV() + widthTotV / 2;
    const int nPixelsV = aSensorInfo.getVCells() - 1;
 
    const int firstPixelID = 0;
 
    double bottomLeft_uID = aSensorInfo.getUCellID(minU, minV, false);
    double bottomLeft_vID = aSensorInfo.getVCellID(minV, false);
    double topRight_uID = aSensorInfo.getUCellID(maxU, maxV, false);
    double topRight_vID = aSensorInfo.getVCellID(maxV, false);
 
    B2DEBUG(10, "  LAYER = " << VxdID((*listOfIntercepts)[i]->getSensorID()).getLayerNumber()
            << " LADDER = " << VxdID((*listOfIntercepts)[i]->getSensorID()).getLadderNumber()
            << " SENSOR = " << VxdID((*listOfIntercepts)[i]->getSensorID()).getSensorNumber()
           );
 
    B2DEBUG(10, "  nPixels (U,V) = (" << nPixelsU << "," << nPixelsV << ")");
 
    B2DEBUG(10, "  widthU = " << maxU - minU
            << "  minU = "  << minU
            << "  maxU = "  << maxU
            << "  lengthU = " << aSensorInfo.getUSize((*listOfIntercepts)[i]->getCoorV())
           );
 
    B2DEBUG(10, "  widthV = " << maxV - minV
            << "  minV = " << minV
            << "  maxV = " << maxV
            << "  lengthV = " << aSensorInfo.getVSize());
 
    B2DEBUG(10, "  bottom left pixel (U,V) = (" << bottomLeft_uID << "," << bottomLeft_vID << ")");
    B2DEBUG(10, "  top right pixel (U,V) = (" << topRight_uID << "," << topRight_vID << ")");
 
 
    //check that the pixel belong to the sensor
    bool inside = true;
    if (bottomLeft_uID > nPixelsU || topRight_uID < firstPixelID || bottomLeft_vID > nPixelsV || topRight_vID < firstPixelID) {
      B2DEBUG(5, "  OOOPS: this pixel does NOT belong to the sensor");
      inside = false;
    }
 
    ROIid tmpROIid;
 
    if (inside) {
      tmpROIid.setMinUid(aSensorInfo.getUCellID(minU, minV, true));
      tmpROIid.setMinVid(aSensorInfo.getVCellID(minV, true));
      tmpROIid.setMaxUid(aSensorInfo.getUCellID(maxU, maxV, true));
      tmpROIid.setMaxVid(aSensorInfo.getVCellID(maxV, true));
      tmpROIid.setSensorID((*listOfIntercepts)[i]->getSensorID()) ;
 
      ROIidList->appendNew(tmpROIid);
 
      // this is the pointer to the transient copy of tmpROIid
      ROIid* transientROIid = (*ROIidList)[ROIidList->getEntries() - 1];
 
      (*listOfIntercepts)[i]->addRelationTo(transientROIid);
 
    }
 
  }
 
}