CDCAlphaScaleFactorForAsymmetry.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
 
#include <map>
#include <iostream>
#include <fstream>
#include <iomanip>
#include <TObject.h>
 
namespace Belle2 {

  class CDCAlphaScaleFactorForAsymmetry: public TObject {
  public:
    enum {c_nLayers    = 56,  
          c_nAlphaBins = 150, 
         };
 

    CDCAlphaScaleFactorForAsymmetry() {}
 

    void setScaleFactors(const std::array<std::array<float, c_nAlphaBins>, c_nLayers>& inputScales)
    {
      m_Scales = inputScales;
    }

    unsigned short getEntries() const
    {
      return c_nLayers * c_nAlphaBins ;
    }

    std::array<std::array<float, c_nAlphaBins>, c_nLayers> getScaleFactors() const
    {
      return m_Scales;
    }

    std::array<float, c_nAlphaBins> getScaleFactors(unsigned short iCLayer) const
    {
      if (iCLayer >= c_nLayers)
        B2FATAL("Required iCLayer is invalid ! Should be 0 to 55 .");
      return m_Scales[iCLayer] ;
    }

    double getScaleFactor(unsigned short iCLayer, double alpha) const
    {
      if (alpha > (M_PI / 2)) alpha = alpha - M_PI;
      if (alpha < -(M_PI / 2)) alpha = alpha + M_PI;
      int alpha_bin = std::trunc(std::abs(alpha) / c_AlphaBinWidth);
      if (alpha_bin > c_nAlphaBins) alpha_bin = c_nAlphaBins - 1 ;
      return m_Scales[iCLayer][alpha_bin];
    }
 

    void dump() const
    {
      std::cout << " " << std::endl;
      std::cout << "Scale factor list" << std::endl;
      std::cout << "#entries= " << c_nLayers* c_nAlphaBins << std::endl;
      std::cout << "the ratio of data to MC for the hit efficiency of positively charged tracks comparing to negitively charged tracks" <<
                std::endl;
 
      for (unsigned short iLayer = 0; iLayer < c_nLayers; iLayer++) {
        std::cout << "Scale factors for Layer " << iLayer << " : " << std::endl;
        for (unsigned short i = 0; i < c_nAlphaBins; ++i) {
          std::cout << " Alpha in ( " << i * 0.01 << " , " << (i + 1) * 0.01 << " ), Scale factor: "  << m_Scales[iLayer][i] << std::endl;
        }
      }
    }
 
  private:
    std::array<std::array<float, c_nAlphaBins>, c_nLayers> m_Scales; 
    double c_AlphaBinWidth = 0.01; 
    ClassDef(CDCAlphaScaleFactorForAsymmetry, 1); 
  };

} // end namespace Belle2