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

#ifndef ALIFIELDMAP_H
#define ALIFIELDMAP_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//
// Author: Andreas Morsch <andreas.morsch@cern.ch>
//
#include <TNamed.h>
#include <TVector.h>
#include <TBuffer.h>

class AliFieldMap : public TNamed
{
  //Alice Magnetic Field with constant mesh

public:
    AliFieldMap();
    AliFieldMap(const char *name, const char *title);
    AliFieldMap(const AliFieldMap &mag);
    virtual ~AliFieldMap();
    void Copy(AliFieldMap &map) const;
    virtual AliFieldMap & operator=(const AliFieldMap &map);

    virtual void Field(Float_t *x, Float_t *b);
    Float_t Bx(const Int_t ix, const Int_t iy, const Int_t iz) {
	return (*fB)(3*(ix*(fZn*fYn)+iy*fZn+iz));
    }
    Float_t By(const Int_t ix, const Int_t iy, const Int_t iz) {
	return (*fB)(3*(ix*(fZn*fYn)+iy*fZn+iz)+1);
    }
    Float_t Bz(const Int_t ix, const Int_t iy, const Int_t iz) {
	return (*fB)(3*(ix*(fZn*fYn)+iy*fZn+iz)+2);
    }

    Bool_t Inside(Float_t x, Float_t y, Float_t z) 
	{ return (x > fXbeg && x <= fXend &&
		  y > fYbeg && y <= fYend &&
		  z > fZbeg && z <= fZend);
	}
    Float_t Xmin()  {return fXbeg;}
    Float_t Xmax()  {return fXend;}
    Float_t DelX()  {return fXdel;}
    Float_t DeliX() {return fXdeli;}
    
    Float_t Ymin()  {return fYbeg;}
    Float_t Ymax()  {return fYend;}
    Float_t DelY()  {return fYdel;}
    Float_t DeliY() {return fYdeli;}
    
    Float_t Zmin()  {return fZbeg;}
    Float_t Zmax()  {return fZend;}
    Float_t DelZ()  {return fZdel;}
    Float_t DeliZ() {return fZdeli;}
    void    SetLimits(Float_t xmin, Float_t xmax, Float_t ymin, Float_t ymax,
		      Float_t zmin, Float_t zmax)
	{
	    fXbeg = xmin; fXend = xmax; fYbeg = ymin; fYend = ymax;
	    fZbeg = zmin; fZend = zmax;
	}
    void SetWriteEnable(Int_t flag = 1) {fWriteEnable = flag;}
 private:
    void    ReadField();
 protected:
    
    Float_t    fXbeg;         // Start of mesh in x
    Float_t    fYbeg;         // Start of mesh in y
    Float_t    fZbeg;         // Start of mesh in z
    Float_t    fXend;         // End of mesh in x
    Float_t    fYend;         // End of mesh in y
    Float_t    fZend;         // End of mesh in z
    Float_t    fXdel;         // Mesh step in x
    Float_t    fYdel;         // Mesh step in y
    Float_t    fZdel;         // Mesh step in z
    Double_t   fXdeli;        // Inverse of Mesh step in x
    Double_t   fYdeli;        // Inverse of Mesh step in y
    Double_t   fZdeli;        // Inverse of Mesh step in z
    Int_t      fXn;           // Number of mesh points in x
    Int_t      fYn;           // Number of mesh points in y
    Int_t      fZn;           // Number of mesh points in z
    Int_t      fWriteEnable;  // Enable flag for writing of field data.
    TVector*   fB;            // Field map
    
    ClassDef(AliFieldMap,3)  //Class for Field Map
};

#endif
Commit	Line	Data
84737f5e	1	#ifndef ALIFIELDMAP_H
	2	#define ALIFIELDMAP_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
	8	//
	9	// Author: Andreas Morsch <andreas.morsch@cern.ch>
	10	//
	11	#include <TNamed.h>
	12	#include <TVector.h>
7b6cddfa	13	#include <TBuffer.h>
84737f5e	14
	15	class AliFieldMap : public TNamed
	16	{
	17	//Alice Magnetic Field with constant mesh
	18
	19	public:
	20	AliFieldMap();
	21	AliFieldMap(const char name, const char title);
	22	AliFieldMap(const AliFieldMap &mag);
	23	virtual ~AliFieldMap();
	24	void Copy(AliFieldMap &map) const;
	25	virtual AliFieldMap & operator=(const AliFieldMap &map);
	26
	27	virtual void Field(Float_t x, Float_t b);
	28	Float_t Bx(const Int_t ix, const Int_t iy, const Int_t iz) {
	29	return (fB)(3(ix(fZnfYn)+iy*fZn+iz));
	30	}
	31	Float_t By(const Int_t ix, const Int_t iy, const Int_t iz) {
	32	return (fB)(3(ix(fZnfYn)+iy*fZn+iz)+1);
	33	}
	34	Float_t Bz(const Int_t ix, const Int_t iy, const Int_t iz) {
	35	return (fB)(3(ix(fZnfYn)+iy*fZn+iz)+2);
	36	}
	37
	38	Bool_t Inside(Float_t x, Float_t y, Float_t z)
	39	{ return (x > fXbeg && x <= fXend &&
	40	y > fYbeg && y <= fYend &&
	41	z > fZbeg && z <= fZend);
	42	}
	43	Float_t Xmin() {return fXbeg;}
	44	Float_t Xmax() {return fXend;}
	45	Float_t DelX() {return fXdel;}
	46	Float_t DeliX() {return fXdeli;}
	47
	48	Float_t Ymin() {return fYbeg;}
	49	Float_t Ymax() {return fYend;}
	50	Float_t DelY() {return fYdel;}
	51	Float_t DeliY() {return fYdeli;}
	52
	53	Float_t Zmin() {return fZbeg;}
	54	Float_t Zmax() {return fZend;}
	55	Float_t DelZ() {return fZdel;}
	56	Float_t DeliZ() {return fZdeli;}
0871c0cc	57	void SetLimits(Float_t xmin, Float_t xmax, Float_t ymin, Float_t ymax,
	58	Float_t zmin, Float_t zmax)
	59	{
	60	fXbeg = xmin; fXend = xmax; fYbeg = ymin; fYend = ymax;
	61	fZbeg = zmin; fZend = zmax;
	62	}
7b6cddfa	63	void SetWriteEnable(Int_t flag = 1) {fWriteEnable = flag;}
84737f5e	64	private:
	65	void ReadField();
	66	protected:
84737f5e	67
7b6cddfa	68	Float_t fXbeg; // Start of mesh in x
	69	Float_t fYbeg; // Start of mesh in y
	70	Float_t fZbeg; // Start of mesh in z
	71	Float_t fXend; // End of mesh in x
	72	Float_t fYend; // End of mesh in y
	73	Float_t fZend; // End of mesh in z
	74	Float_t fXdel; // Mesh step in x
	75	Float_t fYdel; // Mesh step in y
	76	Float_t fZdel; // Mesh step in z
	77	Double_t fXdeli; // Inverse of Mesh step in x
	78	Double_t fYdeli; // Inverse of Mesh step in y
	79	Double_t fZdeli; // Inverse of Mesh step in z
	80	Int_t fXn; // Number of mesh points in x
	81	Int_t fYn; // Number of mesh points in y
	82	Int_t fZn; // Number of mesh points in z
	83	Int_t fWriteEnable; // Enable flag for writing of field data.
	84	TVector* fB; // Field map
	85
	86	ClassDef(AliFieldMap,3) //Class for Field Map
84737f5e	87	};
	88
	89	#endif