eclNOptimalCollectorModule.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
 
/* ECL headers. */
#include <ecl/dataobjects/ECLElementNumbers.h>
 
/* Basf2 headers. */
#include <calibration/CalibrationCollectorModule.h>
#include <framework/datastore/StoreArray.h>
 
namespace Belle2 {
  class ECLCluster;
  class ECLShower;
  class ECLCalDigit;
  class MCParticle;
 
  namespace ECL {
    class ECLNeighbours;
  }


  class eclNOptimalCollectorModule : public CalibrationCollectorModule {
 
  public:

    eclNOptimalCollectorModule();

    void prepare() override;

    void collect() override;
 
  private:

    int m_numberEnergies; 
    std::vector<double> m_energiesForward; 
    std::vector<double> m_energiesBarrel; 
    std::vector<double> m_energiesBackward; 
 
    std::string m_digitArrayName;  
    std::string m_showerArrayName;  
 
    int m_nGroupPerThetaID; 

    std::vector< std::vector<int> > iEnergies;
    const int iFirstCellId = 161; 
    const int iLastCellId = 8608; 
    const int nLeakReg = 3; 
    const int nCrysMax = 21; 
 
    int nCrystalGroups; 
    int iGroupOfCrystal[ECLElementNumbers::c_NCrystals]; 
 
    ECL::ECLNeighbours* neighbours{nullptr}; 
    std::vector<int> thetaIDofCrysID; 
 
    bool storeParameters = true; 

    StoreArray<ECLShower> m_eclShowerArray; 
    StoreArray<ECLCluster> m_eclClusterArray; 
    StoreArray<ECLCalDigit> m_eclCalDigitArray; 
    StoreArray<MCParticle> m_mcParticleArray; 
  };

}