WrappedArray.h

Go to the documentation of this file.

/*******************************************************
*
* Copyright (c) 2003-2010 by University of Queensland
* Earth Systems Science Computational Center (ESSCC)
* http://www.uq.edu.au/esscc
*
* Primary Business: Queensland, Australia
* Licensed under the Open Software License version 3.0
* http://www.opensource.org/licenses/osl-3.0.php
*
*******************************************************/


#ifndef WrappedArray_20081202_H
#define WrappedArray_20081202_H
#include "system_dep.h"
#include "DataTypes.h"
#include "boost/python/extract.hpp"

namespace escript
{

class WrappedArray
{
public:
    WrappedArray(const boost::python::object& obj_in);
    ~WrappedArray();
    unsigned int getRank() const;
    const DataTypes::ShapeType& getShape() const;
    double getElt() const;
    double getElt(unsigned int i) const;
    double getElt(unsigned int i, unsigned int j) const;
    double getElt(unsigned int i, unsigned int j, unsigned int k) const;
    double getElt(unsigned int i, unsigned int j, unsigned int k, unsigned int m) const;
    void convertArray() const;
private:
    template<typename T> void convertNumpyArray(const T* array, const std::vector<int>& strides) const;
    const boost::python::object& obj;
    int rank;
    escript::DataTypes::ShapeType shape;
    double m_scalar;
    mutable double* dat;
};

inline unsigned int 
WrappedArray::getRank() const
{
    return rank;
}

inline const DataTypes::ShapeType& 
WrappedArray::getShape() const
{
    return shape;
}

inline double
WrappedArray::getElt() const
{
    return m_scalar;
}


inline double
WrappedArray::getElt(unsigned int i) const
{  // __float__ added to deal with numpy. If this causes problems we may have to register a custom converter
        return (dat!=0)?dat[i]:(boost::python::extract<double>(obj[i].attr("__float__")()));    
}

inline
double 
WrappedArray::getElt(unsigned int i, unsigned int j) const
{
    return (dat!=0)?dat[DataTypes::getRelIndex(shape,i,j)]:(boost::python::extract<double>(obj[i][j].attr("__float__")()));
}

inline
double 
WrappedArray::getElt(unsigned int i, unsigned int j, unsigned int k) const
{
    return (dat!=0)?dat[DataTypes::getRelIndex(shape,i,j,k)]:(boost::python::extract<double>(obj[i][j][k].attr("__float__")()));
}

inline
double 
WrappedArray::getElt(unsigned int i, unsigned int j, unsigned int k, unsigned int m) const
{
    return (dat!=0)?dat[DataTypes::getRelIndex(shape,i,j,k,m)]:(boost::python::extract<double>(obj[i][j][k][m].attr("__float__")()));
}

}

#endif