ECLNeighbours.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Returns the list of neighbours for a given cell ID.                    *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Torben Ferber (ferber@physics.ubc.ca)                    *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#ifndef ECLNEIGHBOURS_H
#define ECLNEIGHBOURS_H
 
#include <vector>
#include <string>
 
namespace Belle2 {
  namespace ECL {
 
    class ECLNeighbours {
 
    public:
 
      ECLNeighbours(const std::string& neighbourDef, const double par);
 
      ~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);
 
      void initializeNC(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
 
#endif