num_util.h

Go to the documentation of this file.

#ifndef NUM_UTIL_H__
#define NUM_UTIL_H__

#include <boost/python.hpp>

#ifdef HAVE_NUMERIC
#include <Numeric/arrayobject.h>
#else
#include <numarray/arrayobject.h>
#endif

#include <iostream>
#include <sstream>
#include <vector>
#include <map>


namespace num_util{

  boost::python::numeric::array makeNum(boost::python::object x);

  boost::python::numeric::array makeNum(int n, PyArray_TYPES t);

  boost::python::numeric::array makeNum(std::vector<int> dimens, 
                                        PyArray_TYPES t);
                                      
  boost::python::numeric::array makeNum(int * data, int n);

  boost::python::numeric::array makeNum(float * data, int n);

  boost::python::numeric::array makeNum(double * data, int n);

  boost::python::numeric::array makeNum(char * data, std::vector<int> dims);

  boost::python::numeric::array makeNum(short * data, std::vector<int> dims);

  boost::python::numeric::array makeNum(int * data, std::vector<int> dims);

  boost::python::numeric::array makeNum(float * data, std::vector<int> dims);

  boost::python::numeric::array makeNum(double * data, std::vector<int> dims);

  boost::python::numeric::array makeNum(const 
                                        boost::python::numeric::array& arr);

  PyArray_TYPES type(boost::python::numeric::array arr);

  void check_type(boost::python::numeric::array arr, 
                  PyArray_TYPES expected_type);

  int rank(boost::python::numeric::array arr);

  void check_rank(boost::python::numeric::array arr, int expected_rank);
  
  int size(boost::python::numeric::array arr);
  
  void check_size(boost::python::numeric::array arr, int expected_size);

  std::vector<int> shape(boost::python::numeric::array arr);

  int get_dim(boost::python::numeric::array arr, int dimnum);

  void check_shape(boost::python::numeric::array arr, 
                   std::vector<int> expected_dims);

  void check_dim(boost::python::numeric::array arr, int dimnum, int dimsize);

  bool iscontiguous(boost::python::numeric::array arr);

  void check_contiguous(boost::python::numeric::array arr);

  char* data(boost::python::numeric::array arr);

  void copy_data(boost::python::numeric::array arr, char* new_data);
  
  boost::python::numeric::array clone(boost::python::numeric::array arr);
  
  boost::python::numeric::array astype(boost::python::numeric::array arr, 
                                       PyArray_TYPES t);

  bool spacesaver(boost::python::numeric::array arr);
  
  void savespace(boost::python::numeric::array arr, bool set_savespace);

  int refcount(boost::python::numeric::array arr);

  std::vector<int> strides(boost::python::numeric::array arr);

  void check_PyArrayElementType(boost::python::object newo);

  typedef std::map<PyArray_TYPES, std::string> KindStringMap;

  typedef std::map<PyArray_TYPES, char> KindCharMap;

  typedef std::map<char, PyArray_TYPES> KindTypeMap;

  std::string type2string(PyArray_TYPES t_type);

  char type2char(PyArray_TYPES t_type);
  
  PyArray_TYPES char2type(char e_type);

  template <class T>
  inline std::string vector_str(const std::vector<T>& vec);

  inline void check_size_match(std::vector<int> dims, int n);

} //  namespace num_util

#endif

---