ERFieldMap.h

// -------------------------------------------------------------------------
// -----                      ERFieldMap header file                   -----
// -----                   Created 05/17  by V.Schetinin               -----
// -------------------------------------------------------------------------
#ifndef ERFieldMap_H
#define ERFieldMap_H 1

#include "TGeoNode.h"

#include "FairField.h"

class TArrayF;

class ERFieldPar;

class ERFieldMap : public FairField {
public:
  ERFieldMap();
  ERFieldMap(const char* mapName, const char* fileType = "R");
  ERFieldMap(ERFieldPar* fieldPar);
  virtual ~ERFieldMap();
  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);
  virtual void SetPosition(Double_t x, Double_t y, Double_t z);

//void SetLocalNode(TGeoNode* localMagFieldNode) {fLocalMagFieldNode = localMagFieldNode;}
  void SetVolume(TString localMagFieldVolName) {fLocalMagFieldVolName = localMagFieldVolName;}
  virtual 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 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(Option_t* = "") const;
        
        
 protected:
  void Reset();
  void ReadAsciiFile(const char* fileName);
  void ReadRootFile(const char* fileName, const char* mapName);
  /*void SetField(const ERFieldMapData* data);*/
  Double_t Interpolate(Double_t dx, Double_t dy, Double_t dz); 
  TString fFileName;
  Double_t fScale;             
  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];                  
  Double_t fBxOrigin;               
  Double_t fByOrigin;               
  Double_t fBzOrigin;               

  TGeoNode *fLocalMagFieldNode;
  TString  fLocalMagFieldVolName;
 private:
  ERFieldMap(const ERFieldMap&);
  ERFieldMap& operator=(const ERFieldMap&);
  ClassDef(ERFieldMap,2) 
};
#endif