PyFunc.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 "PySortedArray.hpp"
#include "pythonHelpers.hpp"
#include "../typeManipulation.hpp"

namespace LvArray
{
namespace python
{

namespace internal
{

bool err();

void callPyFunc( PyObject * func, PyObjectRef<> * args, long long const argc );

} // namespace internal

template< typename ... ARGS >
class PythonFunction
{
public:

  PythonFunction( PyObject * pyfunc ):
    m_function( pyfunc )
  { internal::xincref( pyfunc ); }

  void operator()( ARGS ... args )
  {
    constexpr long long ARGC = sizeof ... (args);
    PyObjectRef<> pyArgs[ ARGC ];

    long long i = 0;
    typeManipulation::forEachArg( [&i, &pyArgs]( auto & arg )
    {
      pyArgs[ i ] = create( arg );
      ++i;
    }, args ... );

    for( i = 0; i < ARGC; ++i )
    {
      if( pyArgs[ i ] == nullptr )
      { throw PythonError(); }
    }

    // callPyFunc steals all of the pyArgs references
    internal::callPyFunc( m_function, pyArgs, ARGC );

    if( internal::err() )
    { throw PythonError(); }
  }

private:
  PyObjectRef<> m_function;
};


} // namespace python
} // namespace LvArray