eclFillCellIdMappingModule.cc

/**************************************************************************
* BASF2 (Belle Analysis Framework 2)                                     *
* Copyright(C) 2018 - Belle II Collaboration                             *
*                                                                        *
* Author: The Belle II Collaboration                                     *
* Contributors: Torben Ferber (torben.ferber@desy.de)                    *
*                                                                        *
* This software is provided "as is" without any warranty.                *
**************************************************************************/
 
#include <ecl/modules/eclFillCellIdMapping/eclFillCellIdMappingModule.h>
 
#include <ecl/dataobjects/ECLCellIdMapping.h>
#include <ecl/dataobjects/ECLCalDigit.h>
 
#include <ecl/geometry/ECLNeighbours.h>
#include <ecl/geometry/ECLGeometryPar.h>
 
#include <framework/geometry/B2Vector3.h>
 
using namespace Belle2;
using namespace ECL;
 
//-----------------------------------------------------------------
//                 Register the Module
//-----------------------------------------------------------------
REG_MODULE(ECLFillCellIdMapping)
 
//-----------------------------------------------------------------
//                 Implementation
//-----------------------------------------------------------------
 
ECLFillCellIdMappingModule::ECLFillCellIdMappingModule() : Module()
{
  // Set module properties
  setDescription("Fills a dataobject that provides maping between cell id and store arrays and neighbour maps");
}
 
void ECLFillCellIdMappingModule::initialize()
{
  m_eclCalDigits.isRequired();
  m_eclCellIdMapping.registerInDataStore();
 
  // make neighbourmap
  m_NeighbourMap5 = new ECLNeighbours("N", 2);
  m_NeighbourMap7 = new ECLNeighbours("N", 3);
 
  // get phi, theta, phiid, thetaid values
  m_CellIdToPhi.resize(Belle2::ECLCellIdMapping::c_nECLCellIds + 1);
  m_CellIdToTheta.resize(Belle2::ECLCellIdMapping::c_nECLCellIds + 1);
  m_CellIdToPhiId.resize(Belle2::ECLCellIdMapping::c_nECLCellIds + 1);
  m_CellIdToThetaId.resize(Belle2::ECLCellIdMapping::c_nECLCellIds + 1);
 
  // Geometry instance.
  m_geom = ECLGeometryPar::Instance();
 
  for (int idx = 1; idx <= Belle2::ECLCellIdMapping::c_nECLCellIds; idx++) {
    B2Vector3D vectorPosition = m_geom->GetCrystalPos(idx - 1);
 
    m_CellIdToPhi[idx] = vectorPosition.Phi();
    m_CellIdToTheta[idx] = vectorPosition.Theta();
 
    m_geom->Mapping(idx - 1);
    m_CellIdToPhiId[idx] = m_geom->GetPhiID();
    m_CellIdToThetaId[idx] = m_geom->GetThetaID();
  }
}
 
void ECLFillCellIdMappingModule::event()
{
  if (!m_eclCellIdMapping) {
    m_eclCellIdMapping.create();
  }
 
  for (int idx = 1; idx <= Belle2::ECLCellIdMapping::c_nECLCellIds; idx++) {
    m_eclCellIdMapping->setCellIdToNeighbour5(idx, m_NeighbourMap5->getNeighbours(idx));
    m_eclCellIdMapping->setCellIdToNeighbour7(idx, m_NeighbourMap7->getNeighbours(idx));
    m_eclCellIdMapping->setCellIdToPhi(idx, m_CellIdToPhi[idx]);
    m_eclCellIdMapping->setCellIdToTheta(idx, m_CellIdToTheta[idx]);
    m_eclCellIdMapping->setCellIdToPhiId(idx, m_CellIdToPhiId[idx]);
    m_eclCellIdMapping->setCellIdToThetaId(idx, m_CellIdToThetaId[idx]);
  }
 
  // Fill a vector that can be used to map cellid -> store array position for eclCalDigits.
  m_eclCellIdMapping->reset();
 
  for (int i = 0; i < m_eclCalDigits.getEntries(); i++) {
    m_eclCellIdMapping->setCellIdToStoreArray(m_eclCalDigits[i]->getCellId(), i);
  }
}
 
void ECLFillCellIdMappingModule::terminate()
{
  if (m_NeighbourMap5) delete m_NeighbourMap5;
  if (m_NeighbourMap7) delete m_NeighbourMap7;
}