XTCalibrationAlgorithm.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 "string"
#include "TH2D.h"
#include "TH1D.h"
#include "TF1.h"
#include <calibration/CalibrationAlgorithm.h>
#include <cdc/dbobjects/CDCGeometry.h>
#include <framework/database/DBObjPtr.h>
 
namespace Belle2 {
  namespace CDC {

    enum {c_Left = 0, c_Right = 1};

    enum {c_Polynomial = 0, c_Chebyshev = 1};

 
    enum FitStatus {c_lowStat = -1, c_fitFailure = 0, c_OK = 1,
                    c_errorOuter = 2, c_errorInner = 3
                   };

    class XTCalibrationAlgorithm :  public CalibrationAlgorithm {
    public:
      XTCalibrationAlgorithm();

      ~XTCalibrationAlgorithm() {}

      void setBField(bool bfield) {m_bField = bfield;}

      void setDebug(bool debug = false) {m_debug = debug; }

      void setMinimumNDF(double ndf) {m_minNdf = ndf;}

      void setMinimumPval(double pval) {m_minPval = pval;}

      void setXtMode(unsigned short mode = c_Chebyshev) {m_xtMode = mode;}

      void setStoreHisto(bool storeHist = false) {m_storeHisto = storeHist;}

      void enableTextOutput(bool output = true) {m_textOutput = output;}

      void setOutputFileName(std::string outputname) {m_outputFileName.assign(outputname);}

      void setHistFileName(const std::string& name) {m_histName = "histXT_" + name + ".root";}

      void setLRSeparate(bool lr = true) {m_LRseparate = lr;}

      void setThreshold(double th = 0.6) {m_threshold = th;}
 
 
    protected:

      EResult calibrate() override;

      void createHisto();

      void write();

      void storeHisto();

      void prepare();

      EResult checkConvergence();

      void sanitaryCheck();
    private:
      double m_minNdf = 5;    
      double m_minPval = 0.;  
      bool m_debug = false;   
      bool m_storeHisto = true;  
      bool m_LRseparate = true; 
      bool m_bField = true;  
      double m_threshold = 0.6;  
 
      TProfile* m_hProf[56][2][20][10];     
      TH2F* m_hist2d[56][2][20][10];        
      TH2F* m_hist2dDraw[56][20][10];       
      TH1F* m_hist2d_1[56][2][20][10];      
      TF1* m_xtFunc[56][2][20][10];         
 
      double m_xtPrior[56][2][18][7][8];     
 
      int m_fitStatus[56][2][20][10];       
      bool m_useSliceFit = false; 
      int m_minEntriesRequired = 5000; 
      int m_nAlphaBins; 
      int m_nThetaBins; 
      int m_xtMode = c_Chebyshev;  
      int m_xtModePrior;   
      float m_lowerAlpha[18];
      float m_upperAlpha[18];
      float m_iAlpha[18]; 
      float m_lowerTheta[7];
      float m_upperTheta[7];
      float m_iTheta[7]; 
      double m_par6[56] = {89, 91, 94, 99, 104, 107, 110, 117,
                           126, 144, 150, 157, 170, 180,
                           160, 167, 183, 205, 200, 194,
                           177, 189, 192, 206, 224, 234,
                           193, 206, 209, 215, 222, 239,
                           204, 212, 217, 227, 235, 240,
                           215, 222, 230, 239, 246, 253,
                           227, 232, 239, 243, 253, 258,
                           231, 243, 246, 256, 263, 300
                          };
 
      bool  m_textOutput = false; 
      std::string m_outputFileName = "xt_new.dat"; 
      std::string m_histName = "histXT.root"; 
      DBObjPtr<CDCGeometry> m_cdcGeo; 
    };
  }
}