BmnFieldMap.h

Go to the documentation of this file.

// ---------------------------------------------------------------------------------
// -----                    BmnFieldMap header file                    -------------
// -----                Created 03/02/2015  by P. Batyuk               -------------
// ---------------------------------------------------------------------------------
#ifndef BMNFIELDMAP_H
#define BMNFIELDMAP_H 1
 
#include "BmnFieldMapData.h"
#include "BmnFieldPar.h"
#include "FairField.h"
#include "TArrayF.h"
 
typedef struct
{
    Int_t N;
    Double_t min;
    Double_t max;
    Double_t step;
} coordinate_info_t;
 
class BmnFieldMap : public FairField
{
  public:
    BmnFieldMap();
    BmnFieldMap(const char* mapFileName);
    BmnFieldMap(BmnFieldPar* fieldPar);
    virtual ~BmnFieldMap();
 
    virtual void Init();
 
    virtual Double_t GetBx(Double_t x, Double_t y, Double_t z) = 0;
    virtual Double_t GetBy(Double_t x, Double_t y, Double_t z) = 0;
    virtual Double_t GetBz(Double_t x, Double_t y, Double_t z) = 0;
 
    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) = 0;
 
    void WriteAsciiFile(const char* asciiFileName);
 
    void WriteRootFile(const char* rootFileName, const char* mapFileName);
 
    void SetPosition(Double_t x, Double_t y, Double_t z);
 
    void SetScale(Double_t factor)
    {
        fScale = factor;
        fIsOff = (factor == 0 ? kTRUE : kFALSE);
    }
 
    void SetFieldOff(Bool_t is_off = kTRUE) { fIsOff = is_off; }
 
    void SetFileName(const char* map_file_name) { fName = map_file_name; }
 
    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; }
 
    Bool_t IsFieldOff() const { return fIsOff; }
 
    TArrayF* GetBx() const { return fBx; }
    TArrayF* GetBy() const { return fBy; }
    TArrayF* GetBz() const { return fBz; }
 
    const char* GetFileName() { return fName.Data(); }
 
    virtual void Print(Option_t*) const;
 
    virtual void FillParContainer() = 0;
 
    void Reset();
 
  protected:
    // void Reset();
 
    void ReadAsciiFile(const char* asciiFileName);
 
    void ReadRootFile(const char* rootFileName);
    void ReadRootFileNewFormat(const char* rootFileName);
    void ReadRootFileNewFormatExtrap(const char* rootFileName);
 
    void SetField(const BmnFieldMapData* data);
 
    Double_t Interpolate(Double_t dx, Double_t dy, Double_t dz);
 
    Double_t fScale;
 
    Double_t fPosX, fPosY, fPosZ;
    Double_t fPosBx, fPosBy, fPosBz;
 
    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];         
 
    Bool_t fDebugInfo;
    Bool_t fIsOff;
 
  private:
    BmnFieldMap(const BmnFieldMap&) = delete;
    BmnFieldMap& operator=(const BmnFieldMap&) = delete;
 
    ClassDef(BmnFieldMap, 3)
};
 
#endif