Class to perform fitting for each xt function. More...

#include <XTFunction.h>

Public Member Functions
	XTFunction (TProfile *h1)
	Initialized with TProfile histogram.

	XTFunction (TProfile *h1, int mode)
	Initialized with TProfile histogram and mode.

	XTFunction (TH1F *h1, int mode)
	Initialized with TH1D histogram and mode.

	XTFunction (const XTFunction &x)
	Copy constructor.

XTFunction &	operator= (const XTFunction &x)
	Assignment operator.

void	setP6 (double p6)
	Set Parameter 6 for polynomia fit.

bool	isValid ()
	Is valid.

void	setMode (int mode)
	Set XT mode.

void	setBField (bool bfield)
	set to use BField

void	setXTParams (const double p[8])
	Set Parameters for fit.

void	setXTParams (double p0, double p1, double p2, double p3, double p4, double p5, double p6, double p7)
	Set Initial parameters for fitting.

void	setFitRange (double tmin, double tmax)
	Set Fit range.

void	setSmallestEntryRequired (int min)
	Set minimum number of entry required for fit.

void	setDebug (bool debug)
	Set Debug.

int	getFitStatus ()
	get fitted flag.

double	getProb ()
	Get the chi2 probability.

void	getFittedXTParams (double pa[8])
	get fit parameters.

TF1 *	getXTFunction ()
	Get XT function.

TProfile *	getFittedHisto ()
	Get histogram.

void	fitXT ()
	Do fitting.

void	FitPol5 ()
	Fit xt histogram incase 5th order polynomial is used.

void	FitChebyshev ()
	Fit xt histogram incase 5th order Chebeshev polynomial is used.

bool	validate ()
	Validate the xt has proper shape.

Private Attributes
TProfile *	m_h1
	Histogram of xt relation.

TF1 *	m_fitFunc
	Fit function.

int	m_mode = c_Chebyshev
	XT mode, 0 is for 5th order polynomial, 1 is Chebshev polynomial.

bool	m_debug = true
	Print debug durring fitting or not.

bool	m_draw = false
	Draw and store png plot of each histo or not.

bool	m_bField = true
	With magnetic field or not.

int	m_minRequiredEntry = 800
	Minimum entry required for each histo.

double	m_XTParam [8] = {}
	Parameter fo xt.

double	m_FittedXTParams [8] = {}
	Fitted parameters.

int	m_fitflag = 0
	Fit Flag =-1: low statitic =1: good =0: Fit failure =2: Error Outer =3: Error Inner part;.

double	m_Prob = 0
	Chi2 prob of fitting.

double	m_tmin = 20
	lower boundary of fit range

double	m_tmax = m_XTParam[6] + 50
	upper boundary of fit range

Detailed Description

Class to perform fitting for each xt function.

Definition at line 69 of file XTFunction.h.

Constructor & Destructor Documentation

◆ XTFunction() [1/4]

XTFunction ( TProfile * h1 )

inlineexplicit

Initialized with TProfile histogram.

Definition at line 74 of file XTFunction.h.

      {
        m_h1 = (TProfile*)h1->Clone();
        m_h1->SetDirectory(0);
        if (m_mode == c_Chebyshev) {
          m_fitFunc = new TF1("xtCheb5", cheby5pol1, 0.0, 700, 8);
        } else {
          m_fitFunc = new TF1("xtpol5", pol5pol1, 0.0, 700, 8);
        }
      }

◆ XTFunction() [2/4]

XTFunction	(	TProfile *	h1,
		int	mode
	)

inline

Initialized with TProfile histogram and mode.

Definition at line 87 of file XTFunction.h.

      {
        m_h1 = (TProfile*)h1->Clone();
        m_h1->SetDirectory(0);
        m_mode = mode;
        if (m_mode == c_Chebyshev) {
          m_fitFunc = new TF1("xtCheb5", cheby5pol1, 0.0, 700, 8);
        } else {
          m_fitFunc = new TF1("xtpol5", pol5pol1, 0.0, 700, 8);
        }
 
      }

◆ XTFunction() [3/4]

XTFunction	(	TH1F *	h1,
		int	mode
	)

inline

Initialized with TH1D histogram and mode.

Definition at line 103 of file XTFunction.h.

      {
        m_h1 = (TProfile*)h1->Clone();
        m_h1->SetDirectory(0);
        m_mode = mode;
        if (m_mode == c_Chebyshev) {
          m_fitFunc = new TF1("xtCheb5", cheby5pol1, 0.0, 700, 8);
        } else {
          m_fitFunc = new TF1("xtpol5", pol5pol1, 0.0, 700, 8);
        }
      }

◆ XTFunction() [4/4]

XTFunction ( const XTFunction & x )

inline

Copy constructor.

Definition at line 118 of file XTFunction.h.

                                      :
        m_h1(x.m_h1),
        m_mode(x.m_mode),
        m_debug(x.m_debug),
        m_draw(x.m_draw),
        m_bField(x.m_bField),
        m_minRequiredEntry(x.m_minRequiredEntry),
        m_fitflag(x.m_fitflag),
        m_Prob(x.m_Prob),
        m_tmin(x.m_tmin),
        m_tmax(x.m_tmax)
      {
        m_fitFunc = (TF1*) x.m_fitFunc->Clone();
        for (int i = 0; i < 8; ++i) {
          m_XTParam[i] = x.m_XTParam[i];
          m_FittedXTParams[i] = x.m_XTParam[i];
        }
      }

Member Function Documentation

◆ FitChebyshev()

void FitChebyshev ( )

Fit xt histogram incase 5th order Chebeshev polynomial is used.

Definition at line 436 of file XTFunction.h.

    {
      if (m_mode != c_Chebyshev) {
        B2ERROR("Fitting function is wrong");
      }
 
      if (m_h1->GetEntries() < m_minRequiredEntry) {
        m_fitflag = -1;
        return;
      }
      //  m_tmax = m_XTParam[6] + 100;
      //xtCheb5->SetParameters(0.0, 0.005, 0., 0., 0., 0., m_XTParam[6], 0.001);
      double par[8] = {0.0};
      m_fitFunc->SetParLimits(7, 0., 0.001);
      int fitresult = m_h1->Fit("chebyshev5", "QME", "", m_tmin, m_XTParam[6]);
      if (fitresult >= 0) {
        m_h1->GetFunction("chebyshev5")->GetParameters(par);
        m_fitFunc->SetParameters(par[0], par[1], par[2], par[3], par[4], par[5], m_XTParam[6], 0.000);
      }
      double stat;
      for (int i = 0; i < 10; ++i) {
        stat = m_h1->Fit(m_fitFunc, "MQ", "0", m_tmin, m_tmax);
        if (stat == 0) {
          m_fitFunc->SetParameters(par[0], par[1], par[2], par[3], par[4], par[5], m_XTParam[6] - 20, 0.000);
          m_tmax -= 10;
          continue;
        }
        m_fitFunc->GetParameters(par);
        if (par[1] < 0) { // negative c1
          // std::cout << " neg c1 converted" << std::endl;
          par[1] *= -1.0;
          m_fitFunc->SetParLimits(1, 0., 0.01);
          m_tmin += 10.0;
          continue;
        }
 
        /*Eval outer region,*/
        double fp6 = m_fitFunc->Eval(par[6]);
        double fbehindp6 = m_fitFunc->Eval(par[6] - 10) - 0.005;
        if (fp6 < fbehindp6 || fp6 > 1) { /*may be change to good value*/
          m_fitflag = 2;
          //      out += 1;
          m_fitFunc->SetParameters(par[0], par[1], par[2], par[3], par[4], par[5], par[6] - 20, 0.000);
          m_fitFunc->SetParLimits(6, par[6] - 50,  par[6] - 10);
          m_tmax -= 10;
          //      if (m_tmax < p6default + 30) {
          //  m_tmax = p6default + 30;
          // }
        } else {
          break;
        }
        //    m_tmax +=10;
        //if (stat != 0) break;
      }
      if (par[1] > 0) {
        m_fitFunc->GetParameters(m_FittedXTParams);
        m_Prob = m_fitFunc->GetProb();
      }
      if (stat == 0)
        m_fitflag = 0;
      else
        m_fitflag = 1;
      //  m_fitFunc->DrawF1(0,400,"same");
      if (m_draw) {
        TString hname = m_h1->GetName();
        TString name = hname + ".pdf";
        TCanvas* c1 = new TCanvas("c1", "", 800, 600);
        m_h1->Draw();
        m_fitFunc->SetLineColor(kBlack);
        m_fitFunc->DrawF1(0, 400, "same");
        c1->SaveAs(name);
      }
    }

◆ FitPol5()

void FitPol5 ( )

Fit xt histogram incase 5th order polynomial is used.

Definition at line 331 of file XTFunction.h.

    {
      if (m_mode != c_Polynomial) {
        B2ERROR("Fitting function is wrong");
      }
      double max_dif = 0.12;
      double max_dif2 = 0.05;
      double par[8];
      m_h1->Fit("pol1", "MQ", "", m_tmin, 50);
      TF1* f1 = (TF1*)m_h1->GetFunction("pol1");
      double p0 = f1->GetParameter(0);
      double p1 = f1->GetParameter(1);
      double f10 = f1->Eval(10);
      /****************************/
      int in = 0; /*how many time inner part change fit limit*/
      int out = 0; /*how many time outer part change fit limit*/
      m_fitFunc->SetParameters(p0, p1, 0, 0, 0, 0, m_XTParam[6], 0);
      double p6default = m_XTParam[6];
      if (m_bField) {
        m_fitFunc->SetParLimits(7, 0.0, 0.001);
        m_fitFunc->SetParLimits(1, 0.0, 0.01);
      } else {
        m_fitFunc->SetParLimits(7, 0.0, 0.01);
        m_fitFunc->SetParLimits(1, 0.0, 0.01);
      }
 
      for (int i = 0; i < 10; ++i) {
        m_fitflag = 1;
        std::cout << "Fitting" << std::endl;
        double stat = m_h1->Fit(m_fitFunc, "M", "0", m_tmin, m_tmax);
        m_fitFunc->GetParameters(par);
 
        /*Eval outer region,*/
        double fp6 = m_fitFunc->Eval(par[6]);
        double fbehindp6 = m_fitFunc->Eval(par[6] - 12) - 0.01;
        if (fp6 < fbehindp6 || fp6 > 1) { /*may be change to good value*/
          m_fitflag = 2;
          out += 1;
          m_fitFunc->SetParameters(p0, p1, 0, 0, 0, 0, p6default, 0);
          m_fitFunc->SetParLimits(6, p6default - 10,  p6default + 10 - out / 2);
          m_tmax += 2;
          if (m_tmax < p6default + 30) {
            m_tmax = p6default + 30;
          }
        }
        /* EVal inner region*/
        /* compare p0 of linear fit vs p0 of xtfun fit and eval point t=10)*/
        if (fabs(par[0] - p0) > max_dif || fabs(f10 - m_fitFunc->Eval(10)) > max_dif2) {
          m_fitflag = 3;
          if (i == 9) std::cout << "ERROR XT FIT inner part" << std::endl;
          in += 1;
          m_fitFunc->SetParameters(p0, p1, 0, 0, 0, 0, p6default, 0);
          m_fitFunc->SetParLimits(1, 0, 0.08);
          m_tmin -= 0.5;
 
          if (m_tmin < 14) {
            m_tmin = 14; std::cout << "ERROR: tmin so small, fit iter: " << std::endl;
          }
        }
 
        if (stat == 0) {
          m_fitflag = 0;
          if (m_debug) {std::cout << "FIT Failure; Layer: " << std::endl;}
        }
 
        if (m_debug) {printf("P6default= %3.2f, p6 fit = %3.2f", p6default, par[6]);}
 
        if (m_fitflag == 1) {
          if (m_debug) std::cout << "Fit success" << std::endl;
          m_fitFunc->GetParameters(m_FittedXTParams);
          m_Prob = m_fitFunc->GetProb();
          break;
        }
 
      } //end loop of fitting
      if (m_debug) B2INFO("Number of failures due to inner (outer) regions " << in << "(" << out << ")");
 
      if (m_draw) {
        TString hname = m_h1->GetName();
        TString name = hname + ".pdf";
        TCanvas* c1 = new TCanvas("c1", "", 800, 600);
        m_h1->Draw();
        m_fitFunc->SetLineColor(kBlack);
        m_fitFunc->DrawF1(0, 400, "same");
        c1->SaveAs(name);
      }
    }

◆ fitXT()

void fitXT ( )

inline

Do fitting.

Definition at line 277 of file XTFunction.h.

      {
        if (m_mode == c_Polynomial) {
          FitPol5();
        } else if (m_mode == c_Chebyshev) {
          FitChebyshev();
        } else {
          B2ERROR("Undefined fitting function");
        }
      }

◆ getFitStatus()

int getFitStatus ( )

inline

get fitted flag.

Definition at line 243 of file XTFunction.h.

      {
        return m_fitflag;
      }

◆ getFittedHisto()

TProfile * getFittedHisto ( )

inline

Get histogram.

Definition at line 272 of file XTFunction.h.

272{return m_h1;}

◆ getFittedXTParams()

void getFittedXTParams ( double pa[8] )

inline

get fit parameters.

Definition at line 258 of file XTFunction.h.

      {
        for (int i = 0; i < 8; ++i) {pa[i] = m_FittedXTParams[i];}
      }

◆ getProb()

double getProb ( )

inline

Get the chi2 probability.

Definition at line 251 of file XTFunction.h.

      {
        return m_Prob;
      }

◆ getXTFunction()

TF1 * getXTFunction ( )

inline

Get XT function.

Definition at line 265 of file XTFunction.h.

      {
        return m_fitFunc;
      }

◆ isValid()

bool isValid ( )

inline

Is valid.

Definition at line 175 of file XTFunction.h.

      {
        if (m_fitFunc->IsValid() == true) {
          return true;
        } else {
          return false;
        }
      }

◆ operator=()

XTFunction & operator= ( const XTFunction & x )

inline

Assignment operator.

Definition at line 140 of file XTFunction.h.

      {
        if (this != &x) {
          m_h1 = x.m_h1;
          m_fitFunc = x.m_fitFunc;
          m_mode = x.m_mode;
          m_debug = x.m_debug;
          m_draw = x.m_draw;
          m_bField = x.m_bField;
          m_minRequiredEntry = x.m_minRequiredEntry;
          m_fitflag = x.m_fitflag;
          m_Prob = x.m_Prob;
          m_tmin = x.m_tmin;
          m_tmax = x.m_tmax;
 
          for (int i = 0; i < 8; ++i) {
            m_XTParam[i] = x.m_XTParam[i];
            m_FittedXTParams[i] = x.m_XTParam[i];
          }
        }
        return *this;
      }

◆ setBField()

void setBField ( bool bfield )

inline

set to use BField

Definition at line 196 of file XTFunction.h.

196{m_bField = bfield;}

◆ setDebug()

void setDebug ( bool debug )

inline

Set Debug.

Definition at line 236 of file XTFunction.h.

      {
        m_debug = debug;
      }

◆ setFitRange()

void setFitRange	(	double	tmin,
		double	tmax
	)

inline

Set Fit range.

Definition at line 221 of file XTFunction.h.

      {
        m_tmin = tmin;
        m_tmax = tmax;
      }

◆ setMode()

void setMode ( int mode )

inline

Set XT mode.

1 is 5th order Chebshev polynomial. 0 is 5th order polynomial.

Definition at line 189 of file XTFunction.h.

      {
        m_mode = mode;
      }

◆ setP6()

void setP6 ( double p6 )

inline

Set Parameter 6 for polynomia fit.

Definition at line 167 of file XTFunction.h.

      {
        m_XTParam[6] = p6;
      }

◆ setSmallestEntryRequired()

void setSmallestEntryRequired ( int min )

inline

Set minimum number of entry required for fit.

Definition at line 229 of file XTFunction.h.

      {
        m_minRequiredEntry = min;
      }

◆ setXTParams() [1/2]

void setXTParams ( const double p[8] )

inline

Set Parameters for fit.

Definition at line 201 of file XTFunction.h.

      {
        for (int i = 0; i < 8; ++i) {m_XTParam[i] = p[i];}
        m_tmax = p[6] + 50;
      }

◆ setXTParams() [2/2]

void setXTParams	(	double	p0,
		double	p1,
		double	p2,
		double	p3,
		double	p4,
		double	p5,
		double	p6,
		double	p7
	)

inline

Set Initial parameters for fitting.

Definition at line 209 of file XTFunction.h.

      {
        m_XTParam[0] = p0;     m_XTParam[1] = p1;     m_XTParam[2] = p2;
        m_XTParam[3] = p3;     m_XTParam[4] = p4;     m_XTParam[5] = p5;
        m_XTParam[6] = p6;     m_XTParam[7] = p7;
        m_tmax = p6 + 50;
      }

◆ validate()

bool validate ( )

Validate the xt has proper shape.

Suppose to be bad xt if |xt(0)| > 0.2.

Definition at line 419 of file XTFunction.h.

    {
 
      const double p6 = m_fitFunc->GetParameter(6);
      if (fabs(m_fitFunc->Eval(0))  > 0.3) {
        B2WARNING("Bad xt function");
        m_fitflag = 0;
        return false;
      } else if (p6 < 50.0) {
        B2WARNING("Unrealistic p6");
        m_fitflag = 0;
        return false;
      } else {
        return true;
      }
    }

Member Data Documentation

◆ m_bField

bool m_bField = true

private

With magnetic field or not.

Definition at line 312 of file XTFunction.h.

◆ m_debug

bool m_debug = true

private

Print debug durring fitting or not.

Definition at line 310 of file XTFunction.h.

◆ m_draw

bool m_draw = false

private

Draw and store png plot of each histo or not.

Definition at line 311 of file XTFunction.h.

◆ m_fitflag

int m_fitflag = 0

private

Fit Flag =-1: low statitic =1: good =0: Fit failure =2: Error Outer =3: Error Inner part;.

Definition at line 325 of file XTFunction.h.

◆ m_fitFunc

TF1* m_fitFunc

private

Fit function.

Definition at line 304 of file XTFunction.h.

◆ m_FittedXTParams

double m_FittedXTParams[8] = {}

private

Fitted parameters.

Definition at line 315 of file XTFunction.h.

◆ m_h1

TProfile* m_h1

private

Histogram of xt relation.

Definition at line 303 of file XTFunction.h.

◆ m_minRequiredEntry

int m_minRequiredEntry = 800

private

Minimum entry required for each histo.

Definition at line 313 of file XTFunction.h.

◆ m_mode

int m_mode = c_Chebyshev

private

XT mode, 0 is for 5th order polynomial, 1 is Chebshev polynomial.

Definition at line 309 of file XTFunction.h.

◆ m_Prob

double m_Prob = 0

private

Chi2 prob of fitting.

Definition at line 326 of file XTFunction.h.

◆ m_tmax

double m_tmax = m_XTParam[6] + 50

private

upper boundary of fit range

Definition at line 328 of file XTFunction.h.

◆ m_tmin

double m_tmin = 20

private

lower boundary of fit range

Definition at line 327 of file XTFunction.h.

◆ m_XTParam

double m_XTParam[8] = {}

private

Parameter fo xt.

Definition at line 314 of file XTFunction.h.

The documentation for this class was generated from the following file:

cdc/calibration/include/XTFunction.h

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ XTFunction() [1/4]

◆ XTFunction() [2/4]

◆ XTFunction() [3/4]

◆ XTFunction() [4/4]

Member Function Documentation

◆ FitChebyshev()

◆ FitPol5()

◆ fitXT()

◆ getFitStatus()

◆ getFittedHisto()

◆ getFittedXTParams()

◆ getProb()

◆ getXTFunction()

◆ isValid()

◆ operator=()

◆ setBField()

◆ setDebug()

◆ setFitRange()

◆ setMode()

◆ setP6()

◆ setSmallestEntryRequired()

◆ setXTParams() [1/2]

◆ setXTParams() [2/2]

◆ validate()

Member Data Documentation

◆ m_bField

◆ m_debug

◆ m_draw

◆ m_fitflag

◆ m_fitFunc

◆ m_FittedXTParams

◆ m_h1

◆ m_minRequiredEntry

◆ m_mode

◆ m_Prob

◆ m_tmax

◆ m_tmin

◆ m_XTParam