steve-s

# Code taken from article: http://www.tylermw.com/throwing-shade/
# Author: Tyler Morgan-Wall
# License: GPLv3

volcanoshadow = matrix(1, ncol = ncol(volcano), nrow = nrow(volcano))

volc = list(x=1:nrow(volcano), y=1:ncol(volcano), z=volcano)
sunangle = 45 / 180*pi
angle = -90 / 180 * pi
diffangle = 90 / 180 * pi

steve-s

#include <Python.h>
#include <structmember.h>

typedef struct {
    PyHeapTypeObject super;
    int meta_magic;
    int meta_member;
    char some_more[64];
} DummyMeta;

steve-s

#include "Python.h"

static PyObject *myabs(PyObject *self, PyObject *arg) {
    return Py_Absolute(arg);
}

static PyMethodDef methods[] = {
    {"myabs", myabs, METH_O, ""},
    {NULL, NULL, 0, NULL}
};

steve-s

typedef struct {
  // ...
  HPy (*ctx_Absolute)(HPyContext *ctx, HPy h1);
  int (*ctx_IsTrue)(HPyContext *ctx, HPy h);
  HPy (*ctx_Type_FromSpec)(HPyContext *ctx, HPyType_Spec *spec, HPyType_SpecParam *params);
  // ...
} HPyContext;

// ...

steve-s

python setup.py --hpy-abi=universal build_ext

steve-s

def __bootstrap__():
  from sys import modules
  from hpy.universal import load
  from pkg_resources import resource_filename
  ext_filepath = resource_filename(__name__, 'kiwisolver.hpy.so')
  m = load('kiwisolver', ext_filepath, debug=is_debug)
  modules[__name__] = m

steve-s

HPyAPI_FUNC int HPy_IsTrue(HPyContext *ctx, HPy h) {
     return ctx->ctx_IsTrue ( ctx, h ); 
}

steve-s

static inline int HPy_IsTrue(HPyContext *ctx, HPy h)
{
    return PyObject_IsTrue(_h2py(h));
}

// In CPython ABI mode, HPy handles == PyObject*
static inline PyObject* _h2py(HPy h) {
    return (PyObject*) h._i;
}

steve-s

template<> inline
HPy BinaryAdd::operator()( HPyContext *ctx, Variable* first, double second, HPy h_first, HPy h_second )
{
  HPy temp = BinaryMul()( ctx, first, 1.0, h_first, HPy_NULL );
  if( HPy_IsNull(temp) )
    return HPy_NULL;
  return operator()( ctx, Term_AsStruct( ctx, temp ), second, temp, h_second );
}

steve-s

template<> inline
HPy BinaryAdd::operator()( HPyContext *ctx, Variable* first, double second, HPy h_first, HPy h_second )
{
  HPy temp = BinaryMul()( ctx, first, 1.0, h_first, HPy_NULL );
  if( HPy_IsNull(temp) )
    return HPy_NULL;
  HPy result = operator()( ctx, Term_AsStruct( ctx, temp ), second, temp, h_second );
  HPy_Close(ctx, temp);
  return result;
}