SplineGenerator Class Reference

#include <SplineGenerator.h>

Classes
struct	BinInfo

Public Member Functions
	SplineGenerator (TRandom3 *rndm, const TSpline3 &spline)
double	integral () const
double	mean () const
double	gen_random () const
double	integral (double xmin, double xmax)
const TSpline3 &	spline () const

Private Member Functions
double	partial_integral (int i, double x)
double	integral (int i, double xmin, double xmax)
std::pair< double, double >	turning_points (int i) const
double	mean_part_integral (int i, double x)
double	mean (int i, double xmin, double xmax)

Private Attributes
TRandom3 *	m_rndm
TSpline3	m_spline
double	m_integral
double	m_boxintegral
double	m_mean
std::vector< BinInfo >	m_bins

Detailed Description

Definition at line 9 of file SplineGenerator.h.

Constructor & Destructor Documentation

SplineGenerator()

SplineGenerator::SplineGenerator	(	TRandom3 *	rndm,
		const TSpline3 &	spline
	)

Definition at line 7 of file SplineGenerator.cpp.

                                                                       :
  m_rndm(rndm),
  m_spline(spline),
  m_integral(0.),
  m_boxintegral(0.),
  m_mean(0.),
  m_bins(spline.GetNp())
{
  BinInfo ibin ;
  double xmin(0.), a(0.), b(0.), c(0.), d(0.) ;
  int i = m_spline.GetNp() - 1 ;
  // Get info on the highest knot in the spline.
  m_spline.GetCoeff(i, xmin, a, b, c, d) ;
  ibin.xmin = xmin ;
  ibin.xmax = 1e30 ;
  ibin.integral = 0. ;
  ibin.ymin = 0. ;
  ibin.ymax = 1e30 ;
  m_bins[i] = ibin ;
  --i ;
  // Loop backwards over knots, so we can use the x value of the knot above
  // to set the range.
  for( ; i >= 0 ; --i){
    m_spline.GetCoeff(i, xmin, a, b, c, d) ;
    ibin.xmin = xmin ;
    ibin.xmax = m_bins[i+1].xmin ;
    // cout << "ibin " << i << " xmin " << ibin.xmin << " xmax " << ibin.xmax << endl ;
    // cout << "a " << a << " b " << b << " c " << c << " d " << d << endl ;
    ibin.integral = integral(i, ibin.xmin, ibin.xmax) ;
    // cout << "integral " << ibin.integral << endl ;
    m_integral += ibin.integral ;
    pair<double, double> tps = turning_points(i) ;
    // cout << "tps.first " << tps.first << " tps.second " << tps.second << endl ;
    tps.first = max(min(tps.first, ibin.xmax), ibin.xmin) ;
    tps.second = max(min(tps.second, ibin.xmax), ibin.xmin) ;
    // cout << "tps.first " << tps.first << " tps.second " << tps.second << endl ;
    ibin.ymax = max(
            max(
            max(m_spline.Eval(ibin.xmin),
                m_spline.Eval(ibin.xmax)),
            m_spline.Eval(tps.first)),
            m_spline.Eval(tps.second)) ;
    ibin.ymin = min(
            min(
            min(m_spline.Eval(ibin.xmin),
                m_spline.Eval(ibin.xmax)),
            m_spline.Eval(tps.first)),
            m_spline.Eval(tps.second)) ;
    if(ibin.ymin < 0.){
      cerr << "SplineGenerator ERROR: ymin < 0. (" << ibin.ymin << ") for bin " << i << endl ;
    }
    ibin.boxintegral = ibin.ymax * (ibin.xmax - ibin.xmin) ;
    // cout << "ymax " << ibin.ymax << " ymin " << ibin.ymin << " boxintegral " << ibin.boxintegral << endl ;
    m_boxintegral += ibin.boxintegral ;
    m_bins[i] = ibin ;
    m_mean += mean(i, ibin.xmin, ibin.xmax) ;
  }
  if(m_integral != 0.)
    m_mean /= m_integral ;
}

Member Function Documentation

gen_random()

double SplineGenerator::gen_random

(

)

const

Definition at line 79 of file SplineGenerator.cpp.

                                         {
  while(true){
    double boxsel = m_rndm->Rndm() * m_boxintegral ;
    double boxsum(0.) ;
    vector<BinInfo>::const_iterator ibin = m_bins.begin() ;
    for(; ibin != m_bins.end() ; ++ibin){
      boxsum += ibin->boxintegral ;
      if(boxsum >= boxsel)
    break ;
    }
    double x = m_rndm->Rndm() * (ibin->xmax - ibin->xmin) + ibin->xmin ;
    if(m_rndm->Rndm() * ibin->ymax < m_spline.Eval(x))
      return x ;
  }
}

integral() [1/3]

double SplineGenerator::integral

(

)

const

Definition at line 69 of file SplineGenerator.cpp.

                                       {
  return m_integral ;
}

integral() [2/3]

double SplineGenerator::integral	(	double	xmin,
		double	xmax
	)

Definition at line 96 of file SplineGenerator.cpp.

                                                         {
  int istart = m_spline.FindX(xmin) ;
  int iend = m_spline.FindX(xmax) ;
  double _integral = integral(istart, xmin, m_bins[istart].xmax)
    + integral(iend, m_bins[iend].xmin, xmax) ;
  for(int ibin = istart + 1 ; ibin != iend ; ++ibin)
    _integral += m_bins[ibin].integral ;
  return _integral ;
}

integral() [3/3]

double SplineGenerator::integral ( int  i, double  xmin, double  xmax  )

private

Definition at line 120 of file SplineGenerator.cpp.

                                                                {
  return partial_integral(i, xmax) - partial_integral(i, xmin) ;
}

mean() [1/2]

double SplineGenerator::mean

(

)

const

Definition at line 74 of file SplineGenerator.cpp.

                                   {
  return m_mean ;
}

mean() [2/2]

double SplineGenerator::mean ( int  i, double  xmin, double  xmax  )

private

Definition at line 149 of file SplineGenerator.cpp.

                                                            {
  return mean_part_integral(i, xmax) - mean_part_integral(i, xmin) ;
}

mean_part_integral()

double SplineGenerator::mean_part_integral ( int  i, double  x  )

private

Definition at line 140 of file SplineGenerator.cpp.

                                                          {
  double xmin(0.), a(0.), b(0.), c(0.), d(0.) ;
  m_spline.GetCoeff(i, xmin, a, b, c, d) ;
  double xshift = x - xmin ;
  return x * xshift * (a + xshift * (b/2. + xshift * (c/3. + xshift * d/4.)))
    - xshift * xshift * (a/2. + xshift * (b/6. + xshift * (c/12. + xshift * d/20.))) ;
}

partial_integral()

double SplineGenerator::partial_integral ( int  i, double  x  )

private

Definition at line 112 of file SplineGenerator.cpp.

                                                        {
  double xmin(0.), a(0.), b(0.), c(0.), d(0.) ;
  m_spline.GetCoeff(i, xmin, a, b, c, d) ;
  x -= xmin ;
  return x * (a + x * (b/2. + x * (c/3. + x * d/4.))) ;
}

spline()

const TSpline3 & SplineGenerator::spline

(

)

const

Definition at line 107 of file SplineGenerator.cpp.

                                              {
  return m_spline ;
}

turning_points()

pair< double, double > SplineGenerator::turning_points ( int  i ) const

private

Definition at line 125 of file SplineGenerator.cpp.

                                                                {
  // Note the reversal of coefficient labelling, as the standard
  // quadratic formula is a x^2 + b x + c while TSplinePoly
  // uses a + b x + c x^2 + d x^3.
  double xmin(0.), d(0.), c(0.), b(0.), a(0.) ;
  m_spline.GetCoeff(i, xmin, d, c, b, a) ;
  b *= 2. ;
  a *= 3. ;
  double arg = b*b - 4. * a * c ;
  if(arg < 0.)
    return pair<double, double>(xmin-1e30, xmin-1e30) ;
  return pair<double, double>(xmin + (-b - sqrt(arg))/2./a, xmin + (-b + sqrt(arg))/2./a) ;
}

Member Data Documentation

m_bins

std::vector<BinInfo> SplineGenerator::m_bins

private

Definition at line 25 of file SplineGenerator.h.

m_boxintegral

double SplineGenerator::m_boxintegral

private

Definition at line 23 of file SplineGenerator.h.

m_integral

double SplineGenerator::m_integral

private

Definition at line 22 of file SplineGenerator.h.

m_mean

double SplineGenerator::m_mean

private

Definition at line 24 of file SplineGenerator.h.

m_rndm

TRandom3* SplineGenerator::m_rndm

private

Definition at line 20 of file SplineGenerator.h.

m_spline

TSpline3 SplineGenerator::m_spline

mutableprivate

Definition at line 21 of file SplineGenerator.h.

---

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

• Mint/SplineGenerator.h
• src/Utils/SplineGenerator.cpp