ECLNeighbours.h

/**************************************************************************
 * 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.                  *
 **************************************************************************/
 
#pragma once
 
/* C++ headers. */
#include <string>
#include <vector>
 
namespace Belle2 {
  namespace ECL {
 
    class ECLNeighbours {
 
    public:
 
      ECLNeighbours(const std::string& neighbourDef, const double par, const bool sorted = false);
 
      ~ECLNeighbours();
 
      const std::vector<short int>& getNeighbours(short int cid) const;
 
      short int getCrystalsPerRing(const short int thetaid) const { return m_crystalsPerRing[thetaid]; }
 
    private:
      std::vector < std::vector < short int > > m_neighbourMap;
 
      std::vector < std::vector < short int > > m_neighbourMapTemp;
 
      const short m_crystalsPerRing[69] = {
        48, 48, 64, 64, 64, 96, 96, 96, 96, 96, 96, 144, 144, //FWD up to 13
        144, 144, 144, 144, 144, 144, 144,  // BARREL up to 20
        144, 144, 144, 144, 144, 144, 144, 144, 144, 144, // ...30
        144, 144, 144, 144, 144, 144, 144, 144, 144, 144, // ...40
        144, 144, 144, 144, 144, 144, 144, 144, 144, 144, // ...50
        144, 144, 144, 144, 144, 144, 144, 144, 144, // BARREL up to 59
        144, 144, 96, 96, 96, 96, 96, 64, 64, 64
      }; //BWD
 
 
      void initializeN(const int nneighbours, const bool sorted = false);
 
      void initializeNC(const int nneighbours);
 
      void initializeNLegacy(const int nneighbours);
 
      void initializeNCLegacy(const int nneighbours, const int corners);
 
      void initializeR(const double radius);
 
      void initializeF(const double fraction);
 
      short int decreasePhiId(const short int phiid, const short int thetaid, const short int n);
 
      short int increasePhiId(const short int phiid, const short int thetaid, const short int n);
 
      std::vector<short int> getPhiIdsInBetween(const short int phiInc, const short int phiDec, const short int theta);
 
      std::vector<short int> getPhiIdsInBetweenC(const short int phiInc, const short int phiDec, const short int theta,
                                                 const int corners);
 
      double getDistance(const double alpha, const double R);
 
    };
 
  } // end of namespace ECL
} // end of namespace Belle2