PlasmaMLPALLASMagneticField.hh

Go to the documentation of this file.

#ifndef PlasmaMLPALLASMagneticField_h
#define PlasmaMLPALLASMagneticField_h 1
 
// --- Includes ---
#include "G4MagneticField.hh"       
#include <array>                    
#include <CLHEP/Units/SystemOfUnits.h> 
 
constexpr size_t NumQuadrupoles = 4;
 
class PlasmaMLPALLASMagneticField : public G4MagneticField
{
public:
    PlasmaMLPALLASMagneticField();
 
    ~PlasmaMLPALLASMagneticField() override;
 
    void GetFieldValue(const G4double point[4], double *bField) const override;
 
 
    G4bool GetMapBFieldStatus() const { return StatusMapBField; }
 
    void SetDipoleField(G4double val);
 
    void SetGradient(size_t index, G4double gradient);
 
    G4double GetGradient(size_t index) const;
 
    void SetQLength(size_t index, G4double length);
 
    G4double GetQLength(size_t index) const;
 
    void SetQDrift(size_t index, G4double drift);
 
    G4double GetQDrift(size_t index) const;
 
    void SetMapBFieldStatus(G4bool val);
 
private:
    void DefineCommands();
 
    static void InitializeFieldToZero(G4double *bField);
 
    G4double CalculateConstantDipoleField(G4double posY, G4double posZ) const;
 
    static G4double CalculateFieldMapMode(G4double posY, G4double posZ);
 
    static bool InQuadrupoleRange(G4double posX, G4double posZ);
 
    void ApplyQuadrupoleContributions(G4double posY, G4double posX, G4double posZ, G4double *bField) const;
 
    G4double GetQuadrupoleBegin(int index) const;
 
    G4double ConstantDipoleBField = 0.0 * CLHEP::tesla; 
    std::array<G4double, NumQuadrupoles> gradients = {}; 
    std::array<G4double, NumQuadrupoles> qlength = {};   
    std::array<G4double, NumQuadrupoles> qdrift = {};    
    std::array<G4double, NumQuadrupoles> qbegin = {};    
    std::array<G4double, NumQuadrupoles> qend = {};    
    G4bool StatusMapBField = false;                      
};
 
G4double fitFunction(G4double *x, G4double *par);
 
G4double symmetrizedFunctionY(G4double *x, G4double *par);
 
G4double symmetrizedFunctionS(G4double *x, G4double *par);
 
#endif // PlasmaMLPALLASMagneticField_h