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

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

//
// Interface between the TVirtualMagField and AliMagWrapCheb: wrapper to
// the set of magnetic field data + Tosca parameterization by 
// Chebyshev polynomials
// 
// Author: ruben.shahoyan@cern.ch
//

//#include <TGeoGlobalMagField.h>
#include <TVirtualMagField.h>
class AliMagWrapCheb;

class AliMagF : public TVirtualMagField
{
 public:
  enum BMap_t      {k2kG, k5kG, k5kGUniform};
  enum BeamType_t  {kBeamTypeAA, kBeamTypepp, kNoBeamField};
  enum PolarityConvention_t {kConvLHC,kConvDCS2008,kConvMap2005};
  enum             {kOverrideGRP=BIT(14)}; // don't recreate from GRP if set
  //
  AliMagF();
  AliMagF(const char *name, const char* title, Int_t integ, 
	  Double_t factorSol=1., Double_t factorDip=1., 
	  Double_t fmax=15, BMap_t maptype = k5kG,
	  const char* path="$(ALICE_ROOT)/data/maps/mfchebKGI_sym.root",
	  BeamType_t btype=kBeamTypepp, Double_t benergy=7000.);
  AliMagF(const AliMagF& src);             
  AliMagF& operator=(const AliMagF& src);
  virtual ~AliMagF();
  //
  virtual void Field(const Double_t *x, Double_t *b);
  void       GetTPCInt(const Double_t *xyz, Double_t *b)        const;
  void       GetTPCIntCyl(const Double_t *rphiz, Double_t *b)   const;
  Double_t   GetBz(const Double_t *xyz)                         const;
  //
  AliMagWrapCheb* GetMeasuredMap()                              const {return fMeasuredMap;}
  //
  // former AliMagF methods or their aliases
  void         SetFactorSol(Float_t fc=1.);
  void         SetFactorDip(Float_t fc=1.);
  Double_t     GetFactorSol()                                   const;
  Double_t     GetFactorDip()                                   const;
  Double_t     Factor()                                         const {return GetFactorSol();}
  Bool_t       IsUniform()                                      const {return fMapType == k5kGUniform;}
  //
  void         MachineField(const Double_t  *x, Double_t *b)    const;
  BMap_t       GetMapType()                                     const {return fMapType;}
  BeamType_t   GetBeamType()                                    const {return fBeamType;}
  Double_t     GetBeamEnergy()                                  const {return fBeamEnergy;}
  Double_t     Max()                                            const {return fMax;}
  Int_t        Integ()                                          const {return fInteg;}
  Int_t        PrecInteg()                                      const {return fPrecInteg;}  
  Double_t     SolenoidField()                                  const {return fFactorSol*fSolenoid;}
  //
  Char_t*      GetDataFileName()                                const {return (Char_t*)fParNames.GetName();}
  Char_t*      GetParamName()                                   const {return (Char_t*)fParNames.GetTitle();}
  void         SetDataFileName(const Char_t* nm)                      {fParNames.SetName(nm);}
  void         SetParamName(const Char_t* nm)                         {fParNames.SetTitle(nm);}
  //
  Bool_t       LoadParameterization();
  static Int_t GetPolarityConvention()                                {return Int_t(fgkPolarityConvention);}
  //
 protected:
  // not supposed to be changed during the run, set only at the initialization via constructor
  void         InitMachineField(BeamType_t btype, Double_t benergy);
  void         SetBeamType(BeamType_t type)                           {fBeamType = type;}
  void         SetBeamEnergy(Float_t energy)                          {fBeamEnergy = energy;}
  //
 protected:
  AliMagWrapCheb*  fMeasuredMap;     //! Measured part of the field map
  BMap_t           fMapType;         // field map type
  Double_t         fSolenoid;        // Solenoid field setting
  BeamType_t       fBeamType;        // Beam type: A-A (fBeamType=0) or p-p (fBeamType=1)
  Double_t         fBeamEnergy;      // Beam energy in GeV
  // 
  Int_t            fInteg;           // Default integration method as indicated in Geant
  Int_t            fPrecInteg;       // Alternative integration method, e.g. for higher precision
  Double_t         fFactorSol;       // Multiplicative factor for solenoid
  Double_t         fFactorDip;       // Multiplicative factor for dipole
  Double_t         fMax;             // Max Field as indicated in Geant
  Bool_t           fDipoleOFF;       // Dipole ON/OFF flag
  //
  Double_t         fQuadGradient;    // Gradient field for inner triplet quadrupoles
  Double_t         fDipoleField;     // Field value for D1 and D2 dipoles
  Double_t         fCCorrField;      // Side C 2nd compensator field
  Double_t         fACorr1Field;     // Side A 1st compensator field 
  Double_t         fACorr2Field;     // Side A 2nd compensator field
  //
  TNamed           fParNames;        // file and parameterization loadad
  //
  static const Double_t  fgkSol2DipZ;    // conventional Z of transition from L3 to Dipole field
  static const UShort_t  fgkPolarityConvention; // convention for the mapping of the curr.sign on main component sign
  //   
  ClassDef(AliMagF, 2)           // Class for all Alice MagField wrapper for measured data + Tosca parameterization
};


#endif
Commit	Line	Data
aee8290b	1	#ifndef ALIMAGF_H
aee8290b	2	#define ALIMAGF_H
3da30618	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
db83d72f	6	//
	7	// Interface between the TVirtualMagField and AliMagWrapCheb: wrapper to
	8	// the set of magnetic field data + Tosca parameterization by
	9	// Chebyshev polynomials
	10	//
	11	// Author: ruben.shahoyan@cern.ch
	12	//
fe4da5cc	13
db83d72f	14	//#include <TGeoGlobalMagField.h>
	15	#include <TVirtualMagField.h>
	16	class AliMagWrapCheb;
0742d588	17
db83d72f	18	class AliMagF : public TVirtualMagField
	19	{
	20	public:
	21	enum BMap_t {k2kG, k5kG, k5kGUniform};
	22	enum BeamType_t {kBeamTypeAA, kBeamTypepp, kNoBeamField};
99c7d495	23	enum PolarityConvention_t {kConvLHC,kConvDCS2008,kConvMap2005};
5a004fb4	24	enum {kOverrideGRP=BIT(14)}; // don't recreate from GRP if set
db83d72f	25	//
e2afb3b6	26	AliMagF();
db83d72f	27	AliMagF(const char name, const char title, Int_t integ,
	28	Double_t factorSol=1., Double_t factorDip=1.,
	29	Double_t fmax=15, BMap_t maptype = k5kG,
	30	const char* path="$(ALICE_ROOT)/data/maps/mfchebKGI_sym.root",
9251fceb	31	BeamType_t btype=kBeamTypepp, Double_t benergy=7000.);
db83d72f	32	AliMagF(const AliMagF& src);
	33	AliMagF& operator=(const AliMagF& src);
	34	virtual ~AliMagF();
	35	//
	36	virtual void Field(const Double_t x, Double_t b);
	37	void GetTPCInt(const Double_t xyz, Double_t b) const;
	38	void GetTPCIntCyl(const Double_t rphiz, Double_t b) const;
	39	Double_t GetBz(const Double_t *xyz) const;
	40	//
	41	AliMagWrapCheb* GetMeasuredMap() const {return fMeasuredMap;}
	42	//
	43	// former AliMagF methods or their aliases
1dd3d90e	44	void SetFactorSol(Float_t fc=1.);
	45	void SetFactorDip(Float_t fc=1.);
	46	Double_t GetFactorSol() const;
	47	Double_t GetFactorDip() const;
db83d72f	48	Double_t Factor() const {return GetFactorSol();}
	49	Bool_t IsUniform() const {return fMapType == k5kGUniform;}
	50	//
	51	void MachineField(const Double_t x, Double_t b) const;
	52	BMap_t GetMapType() const {return fMapType;}
db83d72f	53	BeamType_t GetBeamType() const {return fBeamType;}
	54	Double_t GetBeamEnergy() const {return fBeamEnergy;}
	55	Double_t Max() const {return fMax;}
	56	Int_t Integ() const {return fInteg;}
	57	Int_t PrecInteg() const {return fPrecInteg;}
f04e7f5f	58	Double_t SolenoidField() const {return fFactorSol*fSolenoid;}
db83d72f	59	//
	60	Char_t* GetDataFileName() const {return (Char_t*)fParNames.GetName();}
	61	Char_t* GetParamName() const {return (Char_t*)fParNames.GetTitle();}
	62	void SetDataFileName(const Char_t* nm) {fParNames.SetName(nm);}
	63	void SetParamName(const Char_t* nm) {fParNames.SetTitle(nm);}
	64	//
	65	Bool_t LoadParameterization();
1dd3d90e	66	static Int_t GetPolarityConvention() {return Int_t(fgkPolarityConvention);}
db83d72f	67	//
a0201063	68	protected:
db83d72f	69	// not supposed to be changed during the run, set only at the initialization via constructor
	70	void InitMachineField(BeamType_t btype, Double_t benergy);
	71	void SetBeamType(BeamType_t type) {fBeamType = type;}
	72	void SetBeamEnergy(Float_t energy) {fBeamEnergy = energy;}
db83d72f	73	//
	74	protected:
	75	AliMagWrapCheb* fMeasuredMap; //! Measured part of the field map
	76	BMap_t fMapType; // field map type
	77	Double_t fSolenoid; // Solenoid field setting
	78	BeamType_t fBeamType; // Beam type: A-A (fBeamType=0) or p-p (fBeamType=1)
	79	Double_t fBeamEnergy; // Beam energy in GeV
db83d72f	80	//
	81	Int_t fInteg; // Default integration method as indicated in Geant
	82	Int_t fPrecInteg; // Alternative integration method, e.g. for higher precision
	83	Double_t fFactorSol; // Multiplicative factor for solenoid
	84	Double_t fFactorDip; // Multiplicative factor for dipole
	85	Double_t fMax; // Max Field as indicated in Geant
	86	Bool_t fDipoleOFF; // Dipole ON/OFF flag
	87	//
	88	Double_t fQuadGradient; // Gradient field for inner triplet quadrupoles
	89	Double_t fDipoleField; // Field value for D1 and D2 dipoles
	90	Double_t fCCorrField; // Side C 2nd compensator field
	91	Double_t fACorr1Field; // Side A 1st compensator field
	92	Double_t fACorr2Field; // Side A 2nd compensator field
	93	//
	94	TNamed fParNames; // file and parameterization loadad
	95	//
1dd3d90e	96	static const Double_t fgkSol2DipZ; // conventional Z of transition from L3 to Dipole field
1dd3d90e	97	static const UShort_t fgkPolarityConvention; // convention for the mapping of the curr.sign on main component sign
db83d72f	98	//
9251fceb	99	ClassDef(AliMagF, 2) // Class for all Alice MagField wrapper for measured data + Tosca parameterization
fe4da5cc	100	};
7a15f6b8	101
db83d72f	102
fe4da5cc	103	#endif