Cylindrical2DMagneticField.h

#ifndef CYLINDRICAL2DMAGNETICFIELD_H
#define CYLINDRICAL2DMAGNETICFIELD_H 1

// stl
#include <string>
#include <vector>

// CLHEP
#include <CLHEP/Matrix/Vector.h>
#include <CLHEP/Vector/ThreeVector.h>

// MarlinTPC
#include "MagneticField.h"

namespace EVENT
{

    class LCCollection;
}

namespace marlintpc
{

    class Cylindrical2DMagneticField: public MagneticField
    {

        public:
            Cylindrical2DMagneticField();

            virtual ~Cylindrical2DMagneticField();

 void init(double, double, double, double, double);


            virtual Field *new_field(void);

            virtual bool construct(EVENT::LCCollection *);

            struct FieldMapEntry
            {

                float _Br;
                float _Bz;
            };

        protected:

            CLHEP::Hep3Vector get_field_value_coil_coordinates(const CLHEP::Hep3Vector &);


        private:

            double _r_min;
            double _r_max;
            double _z_min;
            double _z_max;

            int _r_bins;
            int _z_bins;

            double _dr;
            double _dz;

            std::vector<std::vector<FieldMapEntry> > _field_map;
    };

} // namespace marlintpc

#endif // CYLINDRICAL2DMAGNETICFIELD_H