| [658] | 1 | //
 | 
|---|
 | 2 | // histogrammes 2D     cmv 2/8/96
 | 
|---|
 | 3 | //
 | 
|---|
 | 4 | 
 | 
|---|
 | 5 | #ifndef HISTOS2_SEEN
 | 
|---|
 | 6 | #define HISTOS2_SEEN
 | 
|---|
 | 7 | 
 | 
|---|
 | 8 | #include "machdefs.h"
 | 
|---|
 | 9 | #include <string>
 | 
|---|
 | 10 | 
 | 
|---|
 | 11 | #include <list>
 | 
|---|
 | 12 | #if defined(__KCC__)
 | 
|---|
 | 13 | #include <list.h>
 | 
|---|
 | 14 | #endif
 | 
|---|
 | 15 | 
 | 
|---|
 | 16 | #include "peida.h"
 | 
|---|
 | 17 | #include "utils.h"
 | 
|---|
 | 18 | #include "histos.h"
 | 
|---|
 | 19 | 
 | 
|---|
 | 20 | namespace SOPHYA {
 | 
|---|
 | 21 | 
 | 
|---|
 | 22 | class GeneralFit;
 | 
|---|
 | 23 | 
 | 
|---|
 | 24 | class Histo2D : public AnyDataObj {
 | 
|---|
 | 25 |   friend class ObjFileIO<Histo2D>;
 | 
|---|
 | 26 | public:
 | 
|---|
 | 27 | 
 | 
|---|
 | 28 |   // CREATOR / DESTRUCTOR
 | 
|---|
 | 29 |   Histo2D(float xMin, float xMax, int nxBin
 | 
|---|
 | 30 |          ,float yMin, float yMax, int nyBin);
 | 
|---|
 | 31 |   Histo2D(const Histo2D& h);
 | 
|---|
 | 32 |   Histo2D();
 | 
|---|
 | 33 |   virtual  ~Histo2D();
 | 
|---|
 | 34 | 
 | 
|---|
 | 35 |   // OPTIONS
 | 
|---|
 | 36 |   void   Errors();  
 | 
|---|
 | 37 |   
 | 
|---|
 | 38 |   // UPDATING
 | 
|---|
 | 39 |   void   Zero();
 | 
|---|
 | 40 |   void   Add(float x, float y, float w = 1.);
 | 
|---|
 | 41 | 
 | 
|---|
 | 42 |   // Operators
 | 
|---|
 | 43 |   Histo2D& operator = (const Histo2D& h);
 | 
|---|
 | 44 |   Histo2D& operator *= (double b);
 | 
|---|
 | 45 |   Histo2D& operator /= (double b);
 | 
|---|
 | 46 |   Histo2D& operator += (double b);
 | 
|---|
 | 47 |   Histo2D& operator -= (double b);
 | 
|---|
 | 48 |   friend Histo2D operator * (const Histo2D& a, double b);
 | 
|---|
 | 49 |   friend Histo2D operator * (double b, const Histo2D& a);
 | 
|---|
 | 50 |   friend Histo2D operator / (const Histo2D& a, double b);
 | 
|---|
 | 51 |   friend Histo2D operator + (const Histo2D& a, double b);
 | 
|---|
 | 52 |   friend Histo2D operator + (double b, const Histo2D& a);
 | 
|---|
 | 53 |   friend Histo2D operator - (const Histo2D& a, double b);
 | 
|---|
 | 54 |   friend Histo2D operator - (double b, const Histo2D& a);
 | 
|---|
 | 55 |   Histo2D& operator += (const Histo2D& a);
 | 
|---|
 | 56 |   Histo2D& operator -= (const Histo2D& a);
 | 
|---|
 | 57 |   Histo2D& operator *= (const Histo2D& a);
 | 
|---|
 | 58 |   Histo2D& operator /= (const Histo2D& a);
 | 
|---|
 | 59 |   friend Histo2D operator + (const Histo2D& a, const Histo2D& b);
 | 
|---|
 | 60 |   friend Histo2D operator - (const Histo2D& a, const Histo2D& b);
 | 
|---|
 | 61 |   friend Histo2D operator * (const Histo2D& a, const Histo2D& b);
 | 
|---|
 | 62 |   friend Histo2D operator / (const Histo2D& a, const Histo2D& b);
 | 
|---|
 | 63 | 
 | 
|---|
 | 64 |   // get/put dans/depuis une matrice / vector
 | 
|---|
 | 65 |   void GetXCoor(Vector& v);
 | 
|---|
 | 66 |   void GetValue(Matrix &v);
 | 
|---|
 | 67 |   void GetYCoor(Vector& v);
 | 
|---|
 | 68 |   void GetError2(Matrix& v);
 | 
|---|
 | 69 |   void GetError(Matrix& v);
 | 
|---|
 | 70 |   void PutValue(Matrix& v, int ierr=0);
 | 
|---|
 | 71 |   void PutValueAdd(Matrix& v, int ierr=0);
 | 
|---|
 | 72 |   void PutError2(Matrix& v);
 | 
|---|
 | 73 |   void PutError2Add(Matrix& v);
 | 
|---|
 | 74 |   void PutError(Matrix& v);
 | 
|---|
 | 75 | 
 | 
|---|
 | 76 |   // INLINES
 | 
|---|
 | 77 |   inline float   XMin() const {return xmin;}
 | 
|---|
 | 78 |   inline float   XMax() const {return xmax;}
 | 
|---|
 | 79 |   inline float   YMin() const {return ymin;}
 | 
|---|
 | 80 |   inline float   YMax() const {return ymax;}
 | 
|---|
 | 81 |   inline int_4   NBinX() const {return nx;}
 | 
|---|
 | 82 |   inline int_4   NBinY() const {return ny;}
 | 
|---|
 | 83 |   inline float   WBinX() const {return wbinx;}
 | 
|---|
 | 84 |   inline float   WBinY() const {return wbiny;}
 | 
|---|
 | 85 |   inline float*  Bins() const  {return data;}
 | 
|---|
 | 86 |   inline float   operator()(int i,int j) const {return data[j*nx+i];}
 | 
|---|
 | 87 |   inline float&  operator()(int i,int j)       {return data[j*nx+i];}
 | 
|---|
 | 88 |   inline bool    HasErrors() { if(err2) return true; else return false;}
 | 
|---|
 | 89 |   inline float   Error(int i,int j)  const
 | 
|---|
 | 90 |                    {if(err2)
 | 
|---|
 | 91 |                       {if(err2[j*nx+i]>0.) return sqrt(err2[j*nx+i]); else return 0.;}
 | 
|---|
 | 92 |                     else return 0.;}
 | 
|---|
 | 93 |   inline double  Error2(int i,int j) const
 | 
|---|
 | 94 |                    {if(err2) return err2[j*nx+i]; else return 0.;}
 | 
|---|
 | 95 |   inline double& Error2(int i,int j) {return err2[j*nx+i];}
 | 
|---|
 | 96 |   inline float   NData() const    {return (float) nHist;}
 | 
|---|
 | 97 |   inline int_4   NEntries() const {return nEntries;}
 | 
|---|
 | 98 |   inline void BinLowEdge(int i,int j,float& x,float& y)
 | 
|---|
 | 99 |                  {x = xmin + i*wbinx; y = ymin + j*wbiny;}
 | 
|---|
 | 100 |   inline void BinCenter(int i,int j,float& x,float& y)
 | 
|---|
 | 101 |                  {x = xmin + (i+0.5)*wbinx; y = ymin + (j+0.5)*wbiny;}
 | 
|---|
 | 102 |   inline void BinHighEdge(int i,int j,float& x,float& y)
 | 
|---|
 | 103 |                  {x = xmin + (i+1)*wbinx; y = ymin + (j+1)*wbiny;}
 | 
|---|
 | 104 |   inline void FindBin(float x,float y,int& i,int& j)
 | 
|---|
 | 105 |     { i = (int) floorf((x - xmin)/wbinx); j = (int) floorf((y - ymin)/wbiny);}
 | 
|---|
 | 106 | 
 | 
|---|
 | 107 |   // Info, statistique et calculs sur les histogrammes
 | 
|---|
 | 108 |   float   NOver(int i=-1,int j=-1) const;
 | 
|---|
 | 109 |   int     BinNonNul() const;
 | 
|---|
 | 110 |   int     ErrNonNul() const;
 | 
|---|
 | 111 |   void    IJMax(int& imax,int& jmax,int il=1,int ih= -1,int jl=1,int jh= -1);
 | 
|---|
 | 112 |   void    IJMin(int& imax,int& jmax,int il=1,int ih= -1,int jl=1,int jh= -1);
 | 
|---|
 | 113 |   float   VMax(int il=1,int ih= -1,int jl=1,int jh= -1) const;
 | 
|---|
 | 114 |   float   VMin(int il=1,int ih= -1,int jl=1,int jh= -1) const;
 | 
|---|
 | 115 |   int     EstimeMax(float& xm,float& ym,int SzPav = 3
 | 
|---|
 | 116 |                    ,int il=1,int ih= -1,int jl=1,int jh= -1);
 | 
|---|
 | 117 |   int     EstimeMax(int im,int jm,float& xm,float& ym,int SzPav = 3);
 | 
|---|
 | 118 |   int     FindMax(int& im,int& jm,int SzPav = 3,float Dz = 0.
 | 
|---|
 | 119 |                  ,int il=1,int ih= -1,int jl=1,int jh= -1);
 | 
|---|
 | 120 | 
 | 
|---|
 | 121 |   // Fit
 | 
|---|
 | 122 |   int      Fit(GeneralFit& gfit,unsigned short typ_err=0);
 | 
|---|
 | 123 |   Histo2D  FitResidus(GeneralFit& gfit);
 | 
|---|
 | 124 |   Histo2D  FitFunction(GeneralFit& gfit);
 | 
|---|
 | 125 | 
 | 
|---|
 | 126 |   // Print et Display ASCII
 | 
|---|
 | 127 |   void    PrintStatus();
 | 
|---|
 | 128 |   void    Print(float min=1.,float max=-1.
 | 
|---|
 | 129 |                ,int il=1,int ih= -1,int jl=1,int jh= -1);
 | 
|---|
 | 130 | 
 | 
|---|
 | 131 |   // PROJECTIONS
 | 
|---|
 | 132 |   void           SetProjX();
 | 
|---|
 | 133 |   void           SetProjY();
 | 
|---|
 | 134 |   void           SetProj();
 | 
|---|
 | 135 |   void           DelProjX();
 | 
|---|
 | 136 |   void           DelProjY();
 | 
|---|
 | 137 |   void           DelProj();
 | 
|---|
 | 138 |   void           ZeroProjX();
 | 
|---|
 | 139 |   void           ZeroProjY();
 | 
|---|
 | 140 |   void           ZeroProj();
 | 
|---|
 | 141 |   inline Histo*  HProjX() const {return hprojx;}
 | 
|---|
 | 142 |   inline Histo*  HProjY() const {return hprojy;}
 | 
|---|
 | 143 |   void           ShowProj();
 | 
|---|
 | 144 | 
 | 
|---|
 | 145 |   // BANDES
 | 
|---|
 | 146 |   inline int     NBandX() const {return lbandx.size();}
 | 
|---|
 | 147 |   inline int     NBandY() const {return lbandy.size();}
 | 
|---|
 | 148 |   int            SetBandX(float ybmin,float ybmax);
 | 
|---|
 | 149 |   int            SetBandY(float xbmin,float xbmax);
 | 
|---|
 | 150 |   void           DelBandX();
 | 
|---|
 | 151 |   void           DelBandY();
 | 
|---|
 | 152 |   void           ZeroBandX();
 | 
|---|
 | 153 |   void           ZeroBandY();
 | 
|---|
 | 154 |   Histo*         HBandX(int n) const;
 | 
|---|
 | 155 |   Histo*         HBandY(int n) const;
 | 
|---|
 | 156 |   void           GetBandX(int n,float& ybmin,float& ybmax) const;
 | 
|---|
 | 157 |   void           GetBandY(int n,float& xbmin,float& xbmax) const;
 | 
|---|
 | 158 |   void           ShowBand(int lp = 0);
 | 
|---|
 | 159 | 
 | 
|---|
 | 160 |   // SLICES
 | 
|---|
 | 161 |   inline int     NSliX() const {return lslix.size();}
 | 
|---|
 | 162 |   inline int     NSliY() const {return lsliy.size();}
 | 
|---|
 | 163 |   int            SetSliX(int nsli);
 | 
|---|
 | 164 |   int            SetSliY(int nsli);
 | 
|---|
 | 165 |   void           DelSliX();
 | 
|---|
 | 166 |   void           DelSliY();
 | 
|---|
 | 167 |   void           ZeroSliX();
 | 
|---|
 | 168 |   void           ZeroSliY();
 | 
|---|
 | 169 |   Histo*         HSliX(int n) const;
 | 
|---|
 | 170 |   Histo*         HSliY(int n) const;
 | 
|---|
 | 171 |   void           ShowSli(int lp = 0);
 | 
|---|
 | 172 | 
 | 
|---|
 | 173 | #ifndef __DECCXX
 | 
|---|
 | 174 | protected:
 | 
|---|
 | 175 | #endif
 | 
|---|
 | 176 |   struct bande_slice {
 | 
|---|
 | 177 |     int num;
 | 
|---|
 | 178 |     float min, max;
 | 
|---|
 | 179 |     Histo* H;
 | 
|---|
 | 180 |     STRUCTCOMP(bande_slice)
 | 
|---|
 | 181 |   };
 | 
|---|
 | 182 | #ifdef __DECCXX
 | 
|---|
 | 183 | protected:
 | 
|---|
 | 184 | #endif
 | 
|---|
 | 185 | 
 | 
|---|
 | 186 |   void Delete();
 | 
|---|
 | 187 | 
 | 
|---|
 | 188 |   float*         data;
 | 
|---|
 | 189 |   double*        err2;
 | 
|---|
 | 190 | 
 | 
|---|
 | 191 |   float          over[3][3];
 | 
|---|
 | 192 |   double         nHist;
 | 
|---|
 | 193 |   int_4          nEntries;
 | 
|---|
 | 194 | 
 | 
|---|
 | 195 |   int_4          nx,ny,nxy;
 | 
|---|
 | 196 |   float          xmin;
 | 
|---|
 | 197 |   float          xmax;
 | 
|---|
 | 198 |   float          ymin;
 | 
|---|
 | 199 |   float          ymax;
 | 
|---|
 | 200 |   float          wbinx;
 | 
|---|
 | 201 |   float          wbiny;
 | 
|---|
 | 202 | 
 | 
|---|
 | 203 |   bande_slice    b_s;
 | 
|---|
 | 204 | 
 | 
|---|
 | 205 |   Histo*         hprojx;
 | 
|---|
 | 206 |   Histo*         hprojy;
 | 
|---|
 | 207 | 
 | 
|---|
 | 208 |   list<bande_slice>  lbandx;
 | 
|---|
 | 209 |   list<bande_slice>  lbandy;
 | 
|---|
 | 210 |   
 | 
|---|
 | 211 |   list<bande_slice>  lslix;
 | 
|---|
 | 212 |   list<bande_slice>  lsliy;
 | 
|---|
 | 213 | 
 | 
|---|
 | 214 | };
 | 
|---|
 | 215 | 
 | 
|---|
 | 216 | /////////////////////////////////////////////////////////////////////////
 | 
|---|
 | 217 | // Classe pour la gestion de persistance
 | 
|---|
 | 218 | 
 | 
|---|
 | 219 | inline POutPersist& operator << (POutPersist& os, Histo2D & obj)
 | 
|---|
 | 220 | { ObjFileIO<Histo2D> fio(&obj);  fio.Write(os);  return(os); }
 | 
|---|
 | 221 | inline PInPersist& operator >> (PInPersist& is, Histo2D & obj)
 | 
|---|
 | 222 | { ObjFileIO<Histo2D> fio(&obj);  fio.Read(is);  return(is); }
 | 
|---|
 | 223 | 
 | 
|---|
 | 224 | // Classe pour la gestion de persistance
 | 
|---|
 | 225 | // ObjFileIO<Histo2D>
 | 
|---|
 | 226 | 
 | 
|---|
 | 227 | } // Fin du namespace
 | 
|---|
 | 228 | 
 | 
|---|
 | 229 | #endif // HISTOS2_SEEN
 | 
|---|