source: Sophya/trunk/Poubelle/DPC:FitsIOServer/NTools/generaldata.h

Last change on this file was 658, checked in by ansari, 26 years ago

no message
File size: 6.7 KB
Line 
1// This may look like C code, but it is really -*- C++ -*-
2// C.Magneville 04/99
3#ifndef GENERALDATA_SEEN
4#define GENERALDATA_SEEN
5
6#include "objfio.h"
7#include <iostream.h>
8#include "pexceptions.h"
9#include "ppersist.h"
10#include "ntupintf.h"
11#include "poly.h"
12
13namespace SOPHYA {
14
15class GeneralFit;
16
17//================================================================
18// GENERALFITDATA
19//================================================================
20
21class GeneralFitData : public AnyDataObj , public NTupleInterface {
22 friend class GeneralFit;
23 friend class ObjFileIO<GeneralFitData>;
24public:
25 enum {Def_ErrF = 1, Def_ErrX = 0};
26
27 GeneralFitData(unsigned int nVar, unsigned int nDataAlloc, uint_2 errx=0);
28 GeneralFitData(const GeneralFitData& data, bool clean=false);
29 GeneralFitData();
30 virtual ~GeneralFitData();
31
32 void Alloc(unsigned int nVar, unsigned int nDataAlloc, int_2 errx=-1);
33 void SetDataPtr(int ptr = 0);
34
35 void KillData(int i);
36 void KillData(int i1,int i2);
37
38 void ValidData(int i);
39 void ValidData(int i1,int i2);
40 void ValidData();
41
42 void RedefineData1(int i,double x,double f,double err=Def_ErrF,double errx=Def_ErrX);
43 void RedefineData2(int i,double x,double y,double f,double err=Def_ErrF
44 ,double errx=Def_ErrX,double erry=Def_ErrX);
45 void RedefineData(int i,double* xp,double f,double err=Def_ErrF,double* errxp=NULL);
46
47 void AddData1(double x, double f, double err=Def_ErrF,double errx=Def_ErrX);
48 void AddData2(double x, double y, double f, double err=Def_ErrF
49 ,double errx=Def_ErrX,double erry=Def_ErrX);
50 void AddData(double* xp, double f, double err=Def_ErrF,double* errxp=NULL);
51 void AddData(float* xp, float f, float err=Def_ErrF,float* errxp=NULL);
52
53 void SetData1(int nData,double* x,double* f,double *err=NULL,double *errx=NULL);
54 void SetData1(int nData,float* x,float* f,float* err=NULL,float *errx=NULL);
55 void SetData2(int nData,double* x,double* y,double* f,double *err=NULL
56 ,double *errx=NULL,double *erry=NULL);
57 void SetData2(int nData,float* x,float* y,float* f,float* err=NULL
58 ,float *errx=NULL,float *erry=NULL);
59 void SetData(int nData,double** xp,double *f,double *err=NULL,double** errxp=NULL);
60 void SetData(int nData,float** xp,float* f,float* err=NULL,float** errxp=NULL);
61
62 void PrintStatus();
63 void PrintData(int i);
64 void PrintData(int i1,int i2);
65 void PrintData();
66 void Show(ostream& os) const;
67 inline void Show() const {Show(cout);}
68
69 inline int GetSpaceFree() const { return mNDataAlloc - mNData; }
70 inline int NVar() const {return mNVar;}
71 inline int NData() const {return mNData;}
72 inline int NDataGood() const {return mNDataGood;}
73 inline int NDataAlloc() const {return mNDataAlloc;}
74
75 inline unsigned short int IsValid(int i) const
76 {if(i>=0 && i<mNData) return mOK[i]; else return 0;}
77 inline bool HasXErrors() {if(mErrXP) return true; else return false;}
78
79 inline double X1(int i) const
80 {if(i>=0 && i<mNData) return mXP[i]; else return 0.;}
81 inline double X(int i) const
82 {if(i>=0 && i<mNData) return mXP[i*mNVar]; else return 0.;}
83 inline double Y(int i) const
84 {if(i>=0 && i<mNData && 1<mNVar) return mXP[i*mNVar+1]; else return 0.;}
85 inline double Z(int i) const
86 {if(i>=0 && i<mNData && 2<mNVar) return mXP[i*mNVar+2]; else return 0.;}
87 inline double Absc(int j,int i) const
88 {if(i>=0 && i<mNData && j<mNVar)return mXP[i*mNVar+j]; else return 0.;}
89 inline double Val(int i) const
90 {if(i>=0 && i<mNData) return mF[i]; else return 0.;}
91
92 inline double EX1(int i) const
93 {if(mErrXP && i>=0 && i<mNData) return mErrXP[i]; else return 0.;}
94 inline double EX(int i) const
95 {if(mErrXP && i>=0 && i<mNData) return mErrXP[i*mNVar]; else return 0.;}
96 inline double EY(int i) const
97 {if(mErrXP && i>=0 && i<mNData && 1<mNVar) return mErrXP[i*mNVar+1];
98 else return 0.;}
99 inline double EZ(int i) const
100 {if(mErrXP && i>=0 && i<mNData && 2<mNVar) return mErrXP[i*mNVar+2];
101 else return 0.;}
102 inline double EAbsc(int j,int i) const
103 {if(mErrXP && i>=0 && i<mNData && j<mNVar) return mErrXP[i*mNVar+j];
104 else return 0.;}
105 inline double EVal(int i) const
106 {if(i>=0 && i<mNData) return mErr[i]; else return 0.;}
107
108 r_8* GetVec(int n, r_8* ret=NULL) const;
109 r_4* GetVecR4(int n, r_4* ret=NULL) const;
110 int GetMnMx(int var,int& imin,int& imax) const;
111 int GetMnMx(int var,double& min,double& max) const;
112 int GetMeanSigma(int var,double& mean,double& sigma,double min=1.,double max=-1.);
113 int GetMoMeMed(int var,double& mode,double& mean,double& median,
114 double min=1.,double max=-1.,double coeff=0.8);
115 int GetMode(int var,double& mode,double min=1.,double max=-1.,double coeff=0.8);
116 double PolFit(int varx,Poly& pol,int degre,bool ey=true);
117 double PolFit(int varx,int vary,Poly2& pol,int degre1,int degre2=-1,bool ez=true);
118 GeneralFitData FitResidus(GeneralFit& gfit);
119 GeneralFitData FitFunction(GeneralFit& gfit);
120
121// Declaration de l interface NTuple
122 virtual uint_4 NbLines() const;
123 virtual uint_4 NbColumns() const;
124 virtual r_8 * GetLineD(int n) const;
125 virtual r_8 GetCell(int n, int k) const;
126 virtual r_8 GetCell(int n, string const & nom) const;
127 virtual void GetMinMax(int k, double& min, double& max) const;
128 virtual void GetMinMax(string const & nom, double& min, double& max) const;
129 virtual int ColumnIndex(string const & nom) const;
130 virtual string ColumnName(int k) const;
131 virtual string VarList_C(const char* nomx=NULL) const;
132
133protected:
134 void Delete();
135
136 int_4 mNVar;
137 int_4 mNDataAlloc;
138 int_4 mNData;
139 int_4 mNDataGood;
140 uint_2 mOk_EXP;
141 r_8* mXP; // x0 y0 z0, x1 y1 z1, ..., xn yn zn
142 r_8* mErrXP; // Ex0 Ey0 Ez0, Ex1 Ey1 Ez1, ..., Exn Eyn Ezn
143 r_8* mF; // F0, F1, ...., Fn
144 r_8* mErr; // EF0, EF1, ...., EFn
145 uint_2* mOK; // 1 si pt valid, 0 sinon
146 r_8* BuffVar;
147 r_4* BuffVarR4;
148};
149
150inline ostream& operator << (ostream& s, GeneralFitData const & g)
151 {g.Show(s); return(s);}
152
153/////////////////////////////////////////////////////////////////////////
154// Classe pour la gestion de persistance
155
156inline POutPersist& operator << (POutPersist& os, GeneralFitData & obj)
157{ ObjFileIO<GeneralFitData> fio(&obj); fio.Write(os); return(os); }
158inline PInPersist& operator >> (PInPersist& is, GeneralFitData & obj)
159{ ObjFileIO<GeneralFitData> fio(&obj); fio.Read(is); return(is); }
160
161// Classe pour la gestion de persistance
162// ObjFileIO<GeneralFitData>
163
164} // Fin du namespace
165
166#endif
Note: See TracBrowser for help on using the repository browser.