PndFieldMap.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      PndFieldMap header file                  -----
// -----          Created 12/01/04  by M. Al/Turany                    -----
// -------------------------------------------------------------------------

#ifndef PNDFIELDMAP_H
#define PNDFIELDMAP_H 1

#include "FairField.h"

class TArrayF;
class PndFieldMapData;
class PndFieldPar;

class PndFieldMap : public FairField {

 public:
  PndFieldMap();

  PndFieldMap(const char *mapName, const char *fileType = "R");

  PndFieldMap(PndFieldPar *fieldPar);

  virtual ~PndFieldMap();

  virtual void Init();

  virtual Double_t GetBx(Double_t x, Double_t y, Double_t z);
  virtual Double_t GetBy(Double_t x, Double_t y, Double_t z);
  virtual Double_t GetBz(Double_t x, Double_t y, Double_t z);

  virtual Bool_t IsInside(Double_t x, Double_t y, Double_t z, Int_t &ix, Int_t &iy, Int_t &iz, Double_t &dx, Double_t &dy, Double_t &dz);

  void WriteAsciiFile(const char *fileName);

  void WriteRootFile(const char *fileName, const char *mapName);

  void SetPosition(Double_t x, Double_t y, Double_t z);

  void SetScale(Double_t factor) { fScale = factor; }

  Double_t GetXmin() const { return fXmin; }
  Double_t GetYmin() const { return fYmin; }
  Double_t GetZmin() const { return fZmin; }
  Double_t GetXmax() const { return fXmax; }
  Double_t GetYmax() const { return fYmax; }
  Double_t GetZmax() const { return fZmax; }
  Double_t GetXstep() const { return fXstep; }
  Double_t GetYstep() const { return fYstep; }
  Double_t GetZstep() const { return fZstep; }
  Int_t GetNx() const { return fNx; }
  Int_t GetNy() const { return fNy; }
  Int_t GetNz() const { return fNz; }

  Double_t GetUnit() const { return funit; }

  Double_t GetPositionX() const { return fPosX; }
  Double_t GetPositionY() const { return fPosY; }
  Double_t GetPositionZ() const { return fPosZ; }

  Double_t GetScale() const { return fScale; }

  TArrayF *GetBx() const { return fBx; }
  TArrayF *GetBy() const { return fBy; }
  TArrayF *GetBz() const { return fBz; }

  const char *GetFileName() { return fFileName.Data(); }

  virtual void Print();

 private:
  PndFieldMap(const PndFieldMap &L);
  PndFieldMap &operator=(const PndFieldMap &) { return *this; };

 protected:
  void Reset();

  void ReadAsciiFile(const char *fileName);

  void ReadRootFile(const char *fileName, const char *mapName);

  void SetField(const PndFieldMapData *data);

  Double_t Interpolate(Double_t dx, Double_t dy, Double_t dz);

  TString fFileName;

  Double_t fScale;

  Double_t funit;

  Double_t fPosX, fPosY, fPosZ;

  Double_t fXmin, fXmax, fXstep;
  Double_t fYmin, fYmax, fYstep;
  Double_t fZmin, fZmax, fZstep;

  Int_t fNx, fNy, fNz;

  TArrayF *fBx;
  TArrayF *fBy;
  TArrayF *fBz;

  Double_t fHa[2][2][2]; 
  Double_t fHb[2][2];    
  Double_t fHc[2];       
  ClassDef(PndFieldMap, 1)
};

#endif