[8901] | 1 | /// \ingroup newmat
|
---|
| 2 | ///@{
|
---|
| 3 |
|
---|
| 4 | /// \file solution.h
|
---|
| 5 | /// One dimensional solve routine.
|
---|
| 6 |
|
---|
| 7 | #include "myexcept.h"
|
---|
| 8 |
|
---|
| 9 | #ifdef use_namespace
|
---|
| 10 | namespace RBD_COMMON {
|
---|
| 11 | #endif
|
---|
| 12 |
|
---|
| 13 |
|
---|
| 14 | // Solve the equation f(x)=y for x where f is a monotone continuous
|
---|
| 15 | // function of x
|
---|
| 16 | // Essentially Brent s method
|
---|
| 17 |
|
---|
| 18 | // You need to derive a class from R1_R1 and override "operator()"
|
---|
| 19 | // with the function you want to solve.
|
---|
| 20 | // Use an object from this class in OneDimSolve
|
---|
| 21 |
|
---|
| 22 | class R1_R1
|
---|
| 23 | {
|
---|
| 24 | // the prototype for a Real function of a Real variable
|
---|
| 25 | // you need to derive your function from this one and put in your
|
---|
| 26 | // function for operator() at least. You probably also want to set up a
|
---|
| 27 | // constructor to put in additional parameter values (e.g. that will not
|
---|
| 28 | // vary during a solve)
|
---|
| 29 |
|
---|
| 30 | protected:
|
---|
| 31 | Real x; // Current x value
|
---|
| 32 | bool xSet; // true if a value assigned to x
|
---|
| 33 |
|
---|
| 34 | public:
|
---|
| 35 | Real minX, maxX; // range of value x
|
---|
| 36 | bool minXinf, maxXinf; // true if these are infinite
|
---|
| 37 | R1_R1() : xSet(false), minXinf(true), maxXinf(true) {}
|
---|
| 38 | virtual Real operator()() = 0; // function value at current x
|
---|
| 39 | // set current x
|
---|
| 40 | virtual void Set(Real X); // set x, check OK
|
---|
| 41 | Real operator()(Real X) { Set(X); return operator()(); }
|
---|
| 42 | // set x, return value
|
---|
| 43 | virtual bool IsValid(Real X);
|
---|
| 44 | operator Real(); // implicit conversion
|
---|
| 45 | virtual ~R1_R1() {} // keep gnu happy
|
---|
| 46 | };
|
---|
| 47 |
|
---|
| 48 | class SolutionException : public BaseException
|
---|
| 49 | {
|
---|
| 50 | public:
|
---|
| 51 | static unsigned long Select;
|
---|
| 52 | SolutionException(const char* a_what = 0);
|
---|
| 53 | };
|
---|
| 54 |
|
---|
| 55 | class OneDimSolve
|
---|
| 56 | {
|
---|
| 57 | R1_R1& function; // reference to the function
|
---|
| 58 | Real accX; // accuracy in X direction
|
---|
| 59 | Real accY; // accuracy in Y direction
|
---|
| 60 | int lim; // maximum number of iterations
|
---|
| 61 |
|
---|
| 62 | public:
|
---|
| 63 | OneDimSolve(R1_R1& f, Real AccY = 0.0001, Real AccX = 0.0)
|
---|
| 64 | : function(f), accX(AccX), accY(AccY) {}
|
---|
| 65 | // f is an R1_R1 function
|
---|
| 66 | Real Solve(Real Y, Real X, Real Dev, int Lim=100);
|
---|
| 67 | // Solve for x in Y=f(x)
|
---|
| 68 | // X is the initial trial value of x
|
---|
| 69 | // X+Dev is the second trial value
|
---|
| 70 | // program returns a value of x such that
|
---|
| 71 | // |Y-f(x)| <= accY or |f.inv(Y)-x| <= accX
|
---|
| 72 |
|
---|
| 73 | private:
|
---|
| 74 | Real x[3], y[3]; // Trial values of X and Y
|
---|
| 75 | int L,C,U,Last; // Locations of trial values
|
---|
| 76 | int vpol, hpol; // polarities
|
---|
| 77 | Real YY; // target value
|
---|
| 78 | int i;
|
---|
| 79 | void LookAt(int); // get new value of function
|
---|
| 80 | bool Finish; // true if LookAt finds conv.
|
---|
| 81 | bool Captured; // true when target surrounded
|
---|
| 82 | void VFlip();
|
---|
| 83 | void HFlip();
|
---|
| 84 | void Flip();
|
---|
| 85 | void State(int I, int J, int K);
|
---|
| 86 | void Linear(int, int, int);
|
---|
| 87 | void Quadratic(int, int, int);
|
---|
| 88 | };
|
---|
| 89 |
|
---|
| 90 |
|
---|
| 91 | #ifdef use_namespace
|
---|
| 92 | }
|
---|
| 93 | #endif
|
---|
| 94 |
|
---|
| 95 | // body file: solution.cpp
|
---|
| 96 |
|
---|
| 97 |
|
---|
| 98 | ///@}
|
---|
| 99 |
|
---|