| 1 | #ifndef CSPLINE_H_SEEN | 
|---|
| 2 | #define CSPLINE_H_SEEN | 
|---|
| 3 |  | 
|---|
| 4 | #include "machdefs.h" | 
|---|
| 5 | #include "pexceptions.h" | 
|---|
| 6 |  | 
|---|
| 7 | /////////////////////////////////////////////////////////////////// | 
|---|
| 8 | // Class CSpline et CSpline2 selon Numerical Receipes (cmv 17/02/97) | 
|---|
| 9 | /////////////////////////////////////////////////////////////////// | 
|---|
| 10 | class CSpline EXC_AWARE { | 
|---|
| 11 |  | 
|---|
| 12 | public: | 
|---|
| 13 |  | 
|---|
| 14 | friend class CSpline2; | 
|---|
| 15 |  | 
|---|
| 16 | enum { NaturalAll = 3, NoNatural = 0, Natural1 = 1, NaturalN = 2 }; | 
|---|
| 17 |  | 
|---|
| 18 | CSpline(int n,double* x,double* y,double yp1=0.,double ypn=0. | 
|---|
| 19 | ,int natural=NaturalAll,bool order=true); | 
|---|
| 20 |  | 
|---|
| 21 | CSpline(double yp1=0.,double ypn=0.,int natural=NaturalAll); | 
|---|
| 22 |  | 
|---|
| 23 | virtual ~CSpline(); | 
|---|
| 24 |  | 
|---|
| 25 | void SetNewTab(int n,double* x,double* y | 
|---|
| 26 | ,bool order=true,bool force=false); | 
|---|
| 27 |  | 
|---|
| 28 | void SetBound1er(double yp1 = 0.,double yp2 = 0.); | 
|---|
| 29 |  | 
|---|
| 30 | inline void SetNaturalCSpline(int type = NaturalAll) | 
|---|
| 31 | { Natural = type;} | 
|---|
| 32 |  | 
|---|
| 33 | inline void Free_Tmp() | 
|---|
| 34 | { if(tmp != NULL) delete [] tmp; tmp=NULL;} | 
|---|
| 35 |  | 
|---|
| 36 | void ComputeCSpline(); | 
|---|
| 37 |  | 
|---|
| 38 | double CSplineInt(double x); | 
|---|
| 39 |  | 
|---|
| 40 | protected: | 
|---|
| 41 |  | 
|---|
| 42 | void DelTab(); | 
|---|
| 43 |  | 
|---|
| 44 | // nombre d elements dans les tableaux X et Y | 
|---|
| 45 | int Nel; | 
|---|
| 46 | // true si les tableaux ont ete changes | 
|---|
| 47 | // et qu il faut recalculer ComputeCSpline() | 
|---|
| 48 | bool corrupt_Y2; | 
|---|
| 49 | // true si les tableaux X,Y ont ete alloues | 
|---|
| 50 | bool XY_Created; | 
|---|
| 51 | // type de contraintes sur la derivee 2sd | 
|---|
| 52 | int Natural; | 
|---|
| 53 | // valeurs imposees de la derivee 1ere aux limites | 
|---|
| 54 | double YP1, YPn; | 
|---|
| 55 |  | 
|---|
| 56 | // tableaux rellement alloues si "order=true" ou seulement | 
|---|
| 57 | // connectes aux tableaux externes si "order=false" | 
|---|
| 58 | double* X; | 
|---|
| 59 | double* Y; | 
|---|
| 60 |  | 
|---|
| 61 | // tableau des coeff permettant l interpolation, | 
|---|
| 62 | // remplis par ComputeCSpline() | 
|---|
| 63 | double* Y2; | 
|---|
| 64 |  | 
|---|
| 65 | // tableau tampon utilise dans ComputeCSpline() | 
|---|
| 66 | double* tmp; | 
|---|
| 67 | int_4*  ind; | 
|---|
| 68 |  | 
|---|
| 69 | }; | 
|---|
| 70 |  | 
|---|
| 71 |  | 
|---|
| 72 | /////////////////////////////////////////////////////////////////// | 
|---|
| 73 | class CSpline2 EXC_AWARE { | 
|---|
| 74 |  | 
|---|
| 75 | public: | 
|---|
| 76 | CSpline2(int n1,double* x1,int n2,double* x2,double* y | 
|---|
| 77 | ,int natural=CSpline::NaturalAll,bool order=true); | 
|---|
| 78 |  | 
|---|
| 79 | CSpline2(int natural=CSpline::NaturalAll); | 
|---|
| 80 |  | 
|---|
| 81 | virtual ~CSpline2(); | 
|---|
| 82 |  | 
|---|
| 83 | void SetNewTab(int n1,double* x1,int n2,double* x2,double* y | 
|---|
| 84 | ,bool order=true,bool force=false); | 
|---|
| 85 |  | 
|---|
| 86 | inline void SetNaturalCSpline(int type = CSpline::NaturalAll) | 
|---|
| 87 | { Natural = type;} | 
|---|
| 88 |  | 
|---|
| 89 | void ComputeCSpline(); | 
|---|
| 90 |  | 
|---|
| 91 | double CSplineInt(double x1,double x2); | 
|---|
| 92 |  | 
|---|
| 93 | protected: | 
|---|
| 94 | void DelTab(); | 
|---|
| 95 |  | 
|---|
| 96 | int Nel1, Nel2; | 
|---|
| 97 | bool corrupt_Y2; | 
|---|
| 98 | bool XY_Created; | 
|---|
| 99 | int Natural; | 
|---|
| 100 |  | 
|---|
| 101 | double* X1; | 
|---|
| 102 | double* X2; | 
|---|
| 103 | double* Y; | 
|---|
| 104 | double* Y2; | 
|---|
| 105 |  | 
|---|
| 106 | // nombre d elements alloues dans S | 
|---|
| 107 | int Nel_S; | 
|---|
| 108 | // tableau de CSpline pour interpolation selon X1 | 
|---|
| 109 | CSpline** S;       // S[0->n2] | 
|---|
| 110 | CSpline*  Sint; | 
|---|
| 111 |  | 
|---|
| 112 | // tableau tampon utilise dans CSplineInt() | 
|---|
| 113 | double* tmp;  // tmp[max(n1,n2)] | 
|---|
| 114 | int_4*  ind; | 
|---|
| 115 |  | 
|---|
| 116 | }; | 
|---|
| 117 |  | 
|---|
| 118 | #endif /* CSPLINE_H_SEEN */ | 
|---|