/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/MagneticField/Interpolation/src/Grid3D.h

Go to the documentation of this file.

#ifndef Grid3D_h
#define Grid3D_h

#include "DataFormats/GeometryVector/interface/Basic3DVector.h"
// #include "DataFormats/Math/interface/SIMDVec.h"
#include "Grid1D.h"
#include <vector>
#include "FWCore/Utilities/interface/Visibility.h"

// the storage class
// needed just because legacy software used () constructor
struct BStorageArray {
  BStorageArray(){}
  BStorageArray(float x,float y, float z) : v{x,y,z}{}

  float const & operator[](int i) const { return v[i];}

  float v[3];
};

class dso_internal Grid3D {
public:

 // typedef double   Scalar;
  typedef float   Scalar;
  typedef Basic3DVector<Scalar>   ValueType;
  typedef ValueType ReturnType; 
 
  using BVector = BStorageArray;
  //using BVector =  ValueType;
  using Container = std::vector<BVector>;

  Grid3D() {}

  Grid3D( const Grid1D& ga, const Grid1D& gb, const Grid1D& gc,
          std::vector<BVector>& data) : 
    grida_(ga), gridb_(gb), gridc_(gc) {
     data_.swap(data);
     stride1_ = gridb_.nodes() * gridc_.nodes();
     stride2_ = gridc_.nodes();
  }


  //  Grid3D( const Grid1D& ga, const Grid1D& gb, const Grid1D& gc,
  //      std::vector<ValueType> const & data);


  int index(int i, int j, int k) const {return i*stride1_ + j*stride2_ + k;}
  int stride1() const { return stride1_;}
  int stride2() const { return stride2_;}
  int stride3() const { return 1;}
  ValueType operator()(int i) const {
    return ValueType(data_[i][0],data_[i][1],data_[i][2]);
  }

  ValueType operator()(int i, int j, int k) const {
    return (*this)(index(i,j,k));
  }

  const Grid1D& grida() const {return grida_;}
  const Grid1D& gridb() const {return gridb_;}
  const Grid1D& gridc() const {return gridc_;}

  const Container & data() const {return data_;}

  void dump() const;

private:

  Grid1D grida_;
  Grid1D gridb_;
  Grid1D gridc_;

  Container data_;

  int stride1_;
  int stride2_;


};

#endif