Line data    Source code

       1             : #include "crpropa/magneticField/MagneticFieldGrid.h"
       2             : 
       3             : namespace crpropa {
       4             : 
       5           0 : MagneticFieldGrid::MagneticFieldGrid(ref_ptr<Grid3f> grid) {
       6           0 :         setGrid(grid);
       7           0 : }
       8             : 
       9           0 : void MagneticFieldGrid::setGrid(ref_ptr<Grid3f> grid) {
      10           0 :         this->grid = grid;
      11           0 : }
      12             : 
      13           0 : ref_ptr<Grid3f> MagneticFieldGrid::getGrid() {
      14           0 :         return grid;
      15             : }
      16             : 
      17           0 : Vector3d MagneticFieldGrid::getField(const Vector3d &pos) const {
      18           0 :         return grid->interpolate(pos);
      19             : }
      20             : 
      21           0 : ModulatedMagneticFieldGrid::ModulatedMagneticFieldGrid(ref_ptr<Grid3f> grid,
      22           0 :                 ref_ptr<Grid1f> modGrid) {
      23             :         grid->setReflective(false);
      24             :         modGrid->setReflective(true);
      25           0 :         setGrid(grid);
      26           0 :         setModulationGrid(modGrid);
      27           0 : }
      28             : 
      29           0 : void ModulatedMagneticFieldGrid::setGrid(ref_ptr<Grid3f> g) {
      30           0 :         grid = g;
      31           0 : }
      32             : 
      33           0 : ref_ptr<Grid3f> ModulatedMagneticFieldGrid::getGrid() {
      34           0 :         return grid;
      35             : }
      36             : 
      37           0 : void ModulatedMagneticFieldGrid::setModulationGrid(ref_ptr<Grid1f> g) {
      38           0 :         modGrid = g;
      39           0 : }
      40             : 
      41           0 : ref_ptr<Grid1f> ModulatedMagneticFieldGrid::getModulationGrid() {
      42           0 :         return modGrid;
      43             : }
      44             : 
      45           0 : void ModulatedMagneticFieldGrid::setReflective(bool gridReflective,
      46             :                 bool modGridReflective) {
      47             :         grid->setReflective(gridReflective);
      48             :         modGrid->setReflective(modGridReflective);
      49           0 : }
      50             : 
      51           0 : Vector3d ModulatedMagneticFieldGrid::getField(const Vector3d &pos) const {
      52           0 :         float m = modGrid->interpolate(pos);
      53           0 :         Vector3d b = grid->interpolate(pos);
      54           0 :         return b * m;
      55             : }
      56             : 
      57             : } // namespace crpropa