PyArray.hpp

Go to the documentation of this file.

/*
 *~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 * Copyright (c) 2019, Lawrence Livermore National Security, LLC.
 *
 * Produced at the Lawrence Livermore National Laboratory
 *
 * LLNL-CODE-746361
 *
 * All rights reserved. See COPYRIGHT for details.
 *
 * This file is part of the GEOSX Simulation Framework.
 *
 * GEOSX is a free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License (as published by the
 * Free Software Foundation) version 2.1 dated February 1999.
 *~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 */

#pragma once

// Source includes
#include "pythonForwardDeclarations.hpp"
#include "numpyHelpers.hpp"
#include "../Array.hpp"
#include "../Macros.hpp"
#include "../limits.hpp"
#include "../output.hpp"
#include "pythonHelpers.hpp"

// System includes
#include <string>
#include <memory>
#include <typeindex>

namespace LvArray
{
namespace python
{
namespace internal
{

class PyArrayWrapperBase
{
public:

  virtual ~PyArrayWrapperBase() = default;

  virtual int getAccessLevel() const
  { return m_accessLevel; }

  virtual void setAccessLevel( int const accessLevel, int const memorySpace ) = 0;

  virtual std::string repr() const = 0;

  virtual int ndim() const = 0;

  virtual int getSingleParameterResizeIndex() const = 0;

  virtual void setSingleParameterResizeIndex( int const dim ) const = 0;

  virtual void resize( long long const newSize ) = 0;

  virtual void resize( long long const * const newSizes ) = 0;

  virtual PyObject * toNumPy() const = 0;

  virtual std::type_index valueType() const = 0;

protected:

  int m_accessLevel = static_cast< int >( LvArray::python::PyModify::READ_ONLY );
};

template< typename T,
          int NDIM,
          typename PERM,
          typename INDEX_TYPE,
          template< typename > class BUFFER_TYPE >
class PyArrayWrapper final : public PyArrayWrapperBase
{
public:

  PyArrayWrapper( Array< T, NDIM, PERM, INDEX_TYPE, BUFFER_TYPE > & array ):
    PyArrayWrapperBase( ),
    m_array( array )
  {}

  virtual ~PyArrayWrapper() = default;

  virtual std::string repr() const final override
  { return system::demangleType< Array< T, NDIM, PERM, INDEX_TYPE, BUFFER_TYPE > >(); }

  virtual int ndim() const final override
  { return NDIM; }

  virtual int getSingleParameterResizeIndex() const final override
  { return m_array.getSingleParameterResizeIndex(); }

  virtual void setSingleParameterResizeIndex( int const dim ) const final override
  { m_array.setSingleParameterResizeIndex( dim ); }

  virtual void resize( long long const newSize ) final override
  { m_array.resize( integerConversion< INDEX_TYPE >( newSize ) ); }

  virtual void resize( long long const * const newSizes ) final override
  {
    INDEX_TYPE dims[ NDIM ];

    for( int i = 0; i < NDIM; ++i )
    { dims[ i ] = integerConversion< INDEX_TYPE >( newSizes[ i ] ); }

    m_array.resize( NDIM, dims );
  }

  virtual void setAccessLevel( int accessLevel, int memorySpace ) final override
  {
    LVARRAY_UNUSED_VARIABLE( memorySpace );
    if( accessLevel >= static_cast< int >( LvArray::python::PyModify::RESIZEABLE ) )
    {
      // touch
    }
    m_accessLevel = accessLevel;
  }

  virtual PyObject * toNumPy() const final override
  {
    m_array.move( MemorySpace::host, m_accessLevel != static_cast< int >(PyModify::READ_ONLY) );
    return toNumPyImpl();
  };

  virtual std::type_index valueType() const final override
  {
    if( std::is_same< T, std::string >::value && NDIM == 1 )
    { return std::type_index( typeid( PyObject * ) ); }
    return std::type_index( typeid( T ) );
  }

private:

  template< typename _T=T >
  std::enable_if_t< !std::is_same< _T, std::string >::value, PyObject * >
  toNumPyImpl() const
  {
    return createNumPyArray( m_array.data(),
                             m_accessLevel >= static_cast< int >( LvArray::python::PyModify::MODIFIABLE ),
                             NDIM,
                             m_array.dims(),
                             m_array.strides() );
  }

  template< typename _T=T >
  std::enable_if_t< std::is_same< _T, std::string >::value && NDIM == 1, PyObject * >
  toNumPyImpl() const
  { return createPyListOfStrings( m_array.data(), integerConversion< long long >( m_array.size() ) ); }

  Array< T, NDIM, PERM, INDEX_TYPE, BUFFER_TYPE > & m_array;
};

PyObject * create( std::unique_ptr< internal::PyArrayWrapperBase > && array );

} // namespace internal

template< typename T,
          int NDIM,
          typename PERM,
          typename INDEX_TYPE,
          template< typename > class BUFFER_TYPE >
std::enable_if_t< internal::canExportToNumpy< T > || ( std::is_same< T, std::string >::value && NDIM == 1 ), PyObject * >
create( Array< T, NDIM, PERM, INDEX_TYPE, BUFFER_TYPE > & array )
{
  using WrapperType = internal::PyArrayWrapper< T, NDIM, PERM, INDEX_TYPE, BUFFER_TYPE >;
  return internal::create( std::make_unique< WrapperType >( array ) );
}

PyTypeObject * getPyArrayType();

} // namespace python
} // namespace LvArray