CbmFieldMapDistorted.h

Go to the documentation of this file.

/* Copyright (C) 2008-2020 Justus-Liebig-Universitaet Giessen, Giessen
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Elena Lebedeva [committer], Florian Uhlig */
 
// -------------------------------------------------------------------------
// -----               CbmFieldMapSym3Distorted header              -----
// -----                  Created 18/01/08  by E.Litvinenko            -----
// -------------------------------------------------------------------------
 
 
#ifndef CBMMAGFIELDMAPDISTORTED_H
#define CBMMAGFIELDMAPDISTORTED_H 1
 
#include "CbmFieldMap.h"  // for CbmFieldMap
 
#include <Rtypes.h>      // for THashConsistencyHolder, ClassDef
#include <RtypesCore.h>  // for Double_t, Int_t, Bool_t, Option_t
#include <TString.h>     // for TString
 
class CbmFieldPar;
class FairField;
class TFormula;
 
class CbmFieldMapDistorted : public CbmFieldMap {
 
 
 public:
  CbmFieldMapDistorted();

  CbmFieldMapDistorted(const char* mapName, const char* pfDistortionFilename, const char* parentName,
                       const char* fileTypeParent = "R", Int_t pfTypeOfParent = 3);

  CbmFieldMapDistorted(Double_t xMin, Double_t xMax, Double_t yMin, Double_t yMax, Double_t zMin, Double_t zMax,
                       Double_t bX, Double_t bY, Double_t bZ, const char* mapName, const char* pfDistortionFilename,
                       const char* parentName = "");

  CbmFieldMapDistorted(CbmFieldPar* fieldPar);
 

  virtual ~CbmFieldMapDistorted();
 

  virtual void Init() override;
 
  void ReadDistortionInformation(const char* filename = 0);
  void WriteDistortionInformation(const char* filename = 0);

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

  virtual void GetFieldValue(const Double_t point[3], Double_t* bField) override;
 
  //  virtual Double_t GetDistorted(Double_t b,Double_t *coords, TFormula *formula);
 
  void SetType(Int_t pfType = 0) { fType = pfType; }
 
  const char* GetDistortionFilename() { return fDistortionFilename.Data(); }
  Bool_t SetDistortionFilename(const char* filename)
  {
    fDistortionFilename = filename;
    return (fDistortionFilename.Length() != 0);
  }
 
  TFormula* GetDistortionFormula(const char* component_option = "y",
                                 const char* action_option    = "m");  // component_opt:"x","y","z";action_opt:"m","a"
  Bool_t SetDistortionFormula(TFormula* parDistortionFormula, const char* component_option = "y",
                              const char* action_option = "m");
  Bool_t SetDistortionFormula(const char* parDistortionFormulaText, const char* component_option = "y",
                              const char* action_option = "m");
 

  void SetFromParent(FairField* field);

  FairField* GetParent() { return fParentField; }

  virtual void Print(Option_t* = "") const override;

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

  virtual void SetScale(Double_t factor) override;

  void PlotBy(Int_t n = 250, Double_t zmin = -50, Double_t zmax = 450);
 
 protected:
  FairField* fParentField;  

  Int_t fTypeOfParent;

  TString fDistortionFilename;

  TFormula *fBxDistortionFormulaMult,
    *fBxDistortionFormulaAdd;  
  TFormula *fByDistortionFormulaMult,
    *fByDistortionFormulaAdd;  
  TFormula *fBzDistortionFormulaMult,
    *fBzDistortionFormulaAdd;  
 
 private:
  CbmFieldMapDistorted(const CbmFieldMapDistorted&);
  CbmFieldMapDistorted& operator=(const CbmFieldMapDistorted&);
 
  ClassDefOverride(CbmFieldMapDistorted, 1)
};
 
#endif