[u/mrichter/AliRoot.git] / STEER / AliCheb3DCalc.h

#ifndef ALICHEB3DCALC_H
#define ALICHEB3DCALC_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
#include <TNamed.h>
#include <TSystem.h>
//
// Author: Ruben Shahoyan
// ruben.shahoyan@cern.ch   09/09/2006
// See Comments in AliCheb3D.h
//


// to decrease the compilable code size comment this define. This will exclude the routines 
// used for the calculation and saving of the coefficients. 
#define _INC_CREATION_ALICHEB3D_

// when _BRING_TO_BOUNDARY_ is defined, the point outside of the fitted folume is assumed
// to be on the surface 
#define _BRING_TO_BOUNDARY_
//


class AliCheb3DCalc: public TNamed
{
 public:
  AliCheb3DCalc();
  AliCheb3DCalc(const AliCheb3DCalc& src);
  AliCheb3DCalc(FILE* stream);
  ~AliCheb3DCalc()                                                           {Clear();}
  //
  AliCheb3DCalc& operator=(const AliCheb3DCalc& rhs);
  void       Print(Option_t* opt="")                                   const;
  void       LoadData(FILE* stream);
  Float_t    Eval(Float_t  *par)                                       const;
  Float_t    EvalDeriv(int dim, Float_t  *par)                         const;
  //
#ifdef _INC_CREATION_ALICHEB3D_
  void       SaveData(const char* outfile,Bool_t append=kFALSE)        const;
  void       SaveData(FILE* stream=stdout)                             const;
#endif
  //
  void       InitRows(int nr);
  void       InitCols(int nc);
  Int_t*     GetNColsAtRow()                                            const {return fNColsAtRow;}
  Int_t*     GetColAtRowBg()                                            const {return fColAtRowBg;}
  void       InitElemBound2D(int ne);
  Int_t*     GetCoefBound2D0()                                          const {return fCoefBound2D0;}
  Int_t*     GetCoefBound2D1()                                          const {return fCoefBound2D1;}
  void       Clear(Option_t* option = "");
  static Float_t    ChebEval1D(Float_t  x, const Float_t * array, int ncf);//     const;  
  static Float_t    ChebEval1Deriv(Float_t  x, const Float_t * array, int ncf);// const;  
  void       InitCoefs(int nc);
  Float_t *  GetCoefs()                                                 const {return fCoefs;}
  //
  static void ReadLine(TString& str,FILE* stream);
  //
 protected:
  Int_t      fNCoefs;            // total number of coeeficients
  Int_t      fNRows;             // number of significant rows in the 3D coeffs matrix
  Int_t      fNCols;             // max number of significant cols in the 3D coeffs matrix
  Int_t      fNElemBound2D;      // number of elements (fNRows*fNCols) to store for the 2D boundary of significant coeffs
  Int_t*     fNColsAtRow;        //[fNRows] number of sighificant columns (2nd dim) at each row of 3D coefs matrix
  Int_t*     fColAtRowBg;        //[fNRows] beginnig of significant columns (2nd dim) for row in the 2D boundary matrix
  Int_t*     fCoefBound2D0;      //[fNElemBound2D] 2D matrix defining the boundary of significance for 3D coeffs.matrix (Ncoefs for col/row)
  Int_t*     fCoefBound2D1;      //[fNElemBound2D] 2D matrix defining the start beginnig of significant coeffs for col/row
  Float_t *  fCoefs;             //[fNCoefs] array of Chebyshev coefficients
  //
  Float_t *  fTmpCf1;            //[fNCols] temp. coeffs for 2d summation
  Float_t *  fTmpCf0;            //[fNRows] temp. coeffs for 1d summation
  //
  ClassDef(AliCheb3DCalc,1)      // Class for interpolation of 3D->1 function by Chebyshev parametrization 
};

//__________________________________________________________________________________________
inline Float_t AliCheb3DCalc::ChebEval1D(Float_t  x, const Float_t * array, int ncf ) 
{
  // evaluate 1D Chebyshev parameterization. x is the argument mapped to [-1:1] interval
  Float_t b0, b1, b2, x2 = x+x;
  b0 = array[--ncf]; 
  b1 = b2 = 0;
  for (int i=ncf;i--;) {
    b2 = b1;
    b1 = b0;
    b0 = array[i] + x2*b1 -b2;
  }
  return b0 - x*b1;
  //
}

#endif
Commit	Line	Data
	1	#ifndef ALICHEB3DCALC_H
	2	#define ALICHEB3DCALC_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5	#include <TNamed.h>
	6	#include <TSystem.h>
	7	//
	8	// Author: Ruben Shahoyan
	9	// ruben.shahoyan@cern.ch 09/09/2006
	10	// See Comments in AliCheb3D.h
	11	//
	12
	13
	14	// to decrease the compilable code size comment this define. This will exclude the routines
	15	// used for the calculation and saving of the coefficients.
	16	#define _INC_CREATION_ALICHEB3D_
	17
	18	// when _BRING_TO_BOUNDARY_ is defined, the point outside of the fitted folume is assumed
	19	// to be on the surface
	20	#define _BRING_TO_BOUNDARY_
	21	//
	22
	23
	24	class AliCheb3DCalc: public TNamed
	25	{
	26	public:
	27	AliCheb3DCalc();
	28	AliCheb3DCalc(const AliCheb3DCalc& src);
	29	AliCheb3DCalc(FILE* stream);
	30	~AliCheb3DCalc() {Clear();}
	31	//
	32	AliCheb3DCalc& operator=(const AliCheb3DCalc& rhs);
	33	void Print(Option_t* opt="") const;
	34	void LoadData(FILE* stream);
	35	Float_t Eval(Float_t *par) const;
	36	Float_t EvalDeriv(int dim, Float_t *par) const;
	37	//
	38	#ifdef _INC_CREATION_ALICHEB3D_
	39	void SaveData(const char* outfile,Bool_t append=kFALSE) const;
	40	void SaveData(FILE* stream=stdout) const;
	41	#endif
	42	//
	43	void InitRows(int nr);
	44	void InitCols(int nc);
	45	Int_t* GetNColsAtRow() const {return fNColsAtRow;}
	46	Int_t* GetColAtRowBg() const {return fColAtRowBg;}
	47	void InitElemBound2D(int ne);
	48	Int_t* GetCoefBound2D0() const {return fCoefBound2D0;}
	49	Int_t* GetCoefBound2D1() const {return fCoefBound2D1;}
	50	void Clear(Option_t* option = "");
	51	static Float_t ChebEval1D(Float_t x, const Float_t * array, int ncf);// const;
	52	static Float_t ChebEval1Deriv(Float_t x, const Float_t * array, int ncf);// const;
	53	void InitCoefs(int nc);
	54	Float_t * GetCoefs() const {return fCoefs;}
	55	//
	56	static void ReadLine(TString& str,FILE* stream);
	57	//
	58	protected:
	59	Int_t fNCoefs; // total number of coeeficients
	60	Int_t fNRows; // number of significant rows in the 3D coeffs matrix
	61	Int_t fNCols; // max number of significant cols in the 3D coeffs matrix
	62	Int_t fNElemBound2D; // number of elements (fNRows*fNCols) to store for the 2D boundary of significant coeffs
	63	Int_t* fNColsAtRow; //[fNRows] number of sighificant columns (2nd dim) at each row of 3D coefs matrix
	64	Int_t* fColAtRowBg; //[fNRows] beginnig of significant columns (2nd dim) for row in the 2D boundary matrix
	65	Int_t* fCoefBound2D0; //[fNElemBound2D] 2D matrix defining the boundary of significance for 3D coeffs.matrix (Ncoefs for col/row)
	66	Int_t* fCoefBound2D1; //[fNElemBound2D] 2D matrix defining the start beginnig of significant coeffs for col/row
	67	Float_t * fCoefs; //[fNCoefs] array of Chebyshev coefficients
	68	//
	69	Float_t * fTmpCf1; //[fNCols] temp. coeffs for 2d summation
	70	Float_t * fTmpCf0; //[fNRows] temp. coeffs for 1d summation
	71	//
	72	ClassDef(AliCheb3DCalc,1) // Class for interpolation of 3D->1 function by Chebyshev parametrization
	73	};
	74
	75	//__________________________________________________________________________________________
	76	inline Float_t AliCheb3DCalc::ChebEval1D(Float_t x, const Float_t * array, int ncf )
	77	{
	78	// evaluate 1D Chebyshev parameterization. x is the argument mapped to [-1:1] interval
	79	Float_t b0, b1, b2, x2 = x+x;
	80	b0 = array[--ncf];
	81	b1 = b2 = 0;
	82	for (int i=ncf;i--;) {
	83	b2 = b1;
	84	b1 = b0;
	85	b0 = array[i] + x2*b1 -b2;
	86	}
	87	return b0 - x*b1;
	88	//
	89	}
	90
	91	#endif