swh:1:snp:af87cd67498ef4fe47c76ed3e7caffe5b61facaf
Tip revision: 1208869c69b3a1a6fbd33681369277163d599355 authored by Unknown Author on 06 September 2006, 07:11:38 UTC
This commit was manufactured by cvs2svn to create tag 'v5-12-00e'.
This commit was manufactured by cvs2svn to create tag 'v5-12-00e'.
Tip revision: 1208869
RootFinder.cxx
/**********************************************************************************
* Project: TMVA - a Root-integrated toolkit for multivariate data analysis *
* Package: TMVA *
* Class : TMVA::RootFinder *
* *
* Description: *
* Implementation (see header file for description) *
* *
* Authors (alphabetical): *
* Andreas Hoecker <Andreas.Hocker@cern.ch> - CERN, Switzerland *
* Xavier Prudent <prudent@lapp.in2p3.fr> - LAPP, France *
* Helge Voss <Helge.Voss@cern.ch> - MPI-KP Heidelberg, Germany *
* Kai Voss <Kai.Voss@cern.ch> - U. of Victoria, Canada *
* *
* Copyright (c) 2005: *
* CERN, Switzerland, *
* U. of Victoria, Canada, *
* MPI-KP Heidelberg, Germany, *
* LAPP, Annecy, France *
* *
* Redistribution and use in source and binary forms, with or without *
* modification, are permitted according to the terms listed in LICENSE *
* (http://mva.sourceforge.net/license.txt) *
* *
**********************************************************************************/
#include "TMVA/RootFinder.h"
#include "Riostream.h"
#include "TMath.h"
ClassImp(TMVA::RootFinder)
//_______________________________________________________________________
TMVA::RootFinder::RootFinder( Double_t (*rootVal)( Double_t ),
Double_t rootMin,
Double_t rootMax,
Int_t maxIterations,
Double_t absTolerance )
: fRootMin( rootMin ),
fRootMax( rootMax ),
fMaxIter( maxIterations ),
fAbsTol ( absTolerance )
{
// constructor
fGetRootVal = rootVal;
}
//_______________________________________________________________________
TMVA::RootFinder::~RootFinder( void )
{
// destructor
}
//_______________________________________________________________________
Double_t TMVA::RootFinder::Root( Double_t refValue )
{
// Root finding using Brents algorithm; taken from CERNLIB function RZERO
Double_t a = fRootMin, b = fRootMax;
Double_t fa = (*fGetRootVal)( a ) - refValue;
Double_t fb = (*fGetRootVal)( b ) - refValue;
if (fb*fa > 0) {
cout << "--- " << GetName() << "::Root: initial interval w/o root: "
<< "(a=" << a << ", b=" << b << "),"
<< " (Eff_a=" << (*fGetRootVal)( a )
<< ", Eff_b=" << (*fGetRootVal)( b ) << "), "
<< "(fa=" << fa << ", fb=" << fb << "), "
<< "refValue = " << refValue << endl;
return 1;
}
Bool_t ac_equal(kFALSE);
Double_t fc = fb;
Double_t c = 0, d = 0, e = 0;
for (Int_t iter= 0; iter <= fMaxIter; iter++) {
if ((fb < 0 && fc < 0) || (fb > 0 && fc > 0)) {
// Rename a,b,c and adjust bounding interval d
ac_equal = kTRUE;
c = a; fc = fa;
d = b - a; e = b - a;
}
if (TMath::Abs(fc) < TMath::Abs(fb)) {
ac_equal = kTRUE;
a = b; b = c; c = a;
fa = fb; fb = fc; fc = fa;
}
Double_t tol = 0.5 * 2.2204460492503131e-16 * TMath::Abs(b);
Double_t m = 0.5 * (c - b);
if (fb == 0 || TMath::Abs(m) <= tol || TMath::Abs(fb) < fAbsTol) return b;
// Bounds decreasing too slowly: use bisection
if (TMath::Abs (e) < tol || TMath::Abs (fa) <= TMath::Abs (fb)) { d = m; e = m; }
else {
// Attempt inverse cubic interpolation
Double_t p, q, r;
Double_t s = fb / fa;
if (ac_equal) { p = 2 * m * s; q = 1 - s; }
else {
q = fa / fc; r = fb / fc;
p = s * (2 * m * q * (q - r) - (b - a) * (r - 1));
q = (q - 1) * (r - 1) * (s - 1);
}
// Check whether we are in bounds
if (p > 0) q = -q;
else p = -p;
Double_t min1 = 3 * m * q - TMath::Abs (tol * q);
Double_t min2 = TMath::Abs (e * q);
if (2 * p < (min1 < min2 ? min1 : min2)) {
// Accept the interpolation
e = d; d = p / q;
}
else { d = m; e = m; } // Interpolation failed: use bisection.
}
// Move last best guess to a
a = b; fa = fb;
// Evaluate new trial root
if (TMath::Abs(d) > tol) b += d;
else b += (m > 0 ? +tol : -tol);
fb = (*fGetRootVal)( b ) - refValue;
}
// Return our best guess if we run out of iterations
cout << "--- " << GetName()
<< "::Root: maximum iterations (" << fMaxIter
<< ") reached before convergence" << endl;
return b;
}