blitzutil.h

Go to the documentation of this file.

// include guard
#ifndef TF_BLITZUTIL_H_VERSION

#define TF_BLITZUTIL_H_VERSION \
  "TF_BLITZUTIL_H   V1.0   "

#include<blitz/array.h>
#include<tfxx/error.h>

namespace tfxx {

namespace blitzutil {

  template<class T1, class T2, int N1, int N2>
  bool sameShape(const blitz::Array<T1,N1>& a1,
                 const blitz::Array<T2,N2>& a2)
  { return(blitz::areShapesConformable(a1.shape(),a2.shape())); }

  template<class T1, class T2, int N1, int N2>
  bool sameDomain(const blitz::Array<T1,N1>& a1,
                  const blitz::Array<T2,N2>& a2)
  { return(blitz::areShapesConformable(a1.lbound(),a2.lbound())&&
           blitz::areShapesConformable(a1.ubound(),a2.ubound())); }

  template<class T, int N>
    void setToSameDomain(blitz::Array<T,N>& array,
                         const blitz::RectDomain<N>& domain,
                         const blitz::GeneralArrayStorage<N>& 
                           storage=blitz::fortranArray)
  { 
    blitz::TinyVector<int, N> extent;
    for (int i=0; i<N; i++)
    { extent[i]=domain.ubound()[i]-domain.lbound()[i]+1; }
    blitz::Array<T, N> result(domain.lbound(),extent,storage);
    array.reference(result);
  }

} // namespace blitzutil

} // namespace tfxx

#endif // TF_BLITZUTIL_H_VERSION (includeguard)

/* ----- END OF blitzutil.h ----- */