pyppc_compat.h

Go to the documentation of this file.

/* --------------------------------------------------------------------
Copyright (C) 2019 Swedish Meteorological and Hydrological Institute, SMHI,
 
This file is part of baltrad-ppc.
 
baltrad-ppc is 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, either version 3 of the License, or
(at your option) any later version.
 
baltrad-ppc is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Lesser General Public License for more details.
 
You should have received a copy of the GNU Lesser General Public License
along with baltrad-ppc.  If not, see <http://www.gnu.org/licenses/>.
------------------------------------------------------------------------*/
/* Define this to ensure that we are not using old API:s from Numpy. However, for now we are more interested in getting the
 * code to build on both 2.7 and 3.x
 */
/*
 * #define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
 */
#include "Python.h"
 
#ifndef PYPPCCOMPAT_H
#define PYPPCCOMPAT_H
 
 
#ifndef PyInt_Check
#define PyInt_Check             PyLong_Check
#define PyInt_FromLong          PyLong_FromLong
#define PyInt_AsLong            PyLong_AsLong
#define PyInt_Type              PyLong_Type
#endif
 
#ifndef PyString_Check
#define PyString_Check          PyUnicode_Check
#define PyString_AsString       PyUnicode_AsUTF8
#define PyString_FromString     PyUnicode_FromString
#define PyString_FromFormat     PyUnicode_FromFormat
#endif
 
#if PY_MAJOR_VERSION >= 3
#define PY_RAVE_ATTRO_NAME_TO_STRING PyUnicode_AsUTF8
 
#else
#define PY_RAVE_ATTRO_NAME_TO_STRING PyString_AsString
 
#endif
 
#define PY_COMPARE_ATTRO_NAME_WITH_STRING(ptr, name) PyPpcAPI_CompareWithASCIIString(ptr, name)
 
#define PY_COMPARE_STRING_WITH_ATTRO_NAME(name, ptr) PyPpcAPI_CompareWithASCIIString(ptr, name)
 
#if PY_MAJOR_VERSION >= 3
#define MOD_INIT_ERROR        NULL
#define MOD_INIT_SUCCESS(val) val
#define MOD_INIT(name)        PyMODINIT_FUNC PyInit_##name(void)
#define MOD_INIT_DEF(ob, name, doc, methods) \
  static struct PyModuleDef moduledef = { \
    PyModuleDef_HEAD_INIT, name, doc, -1, methods, }; \
    ob = PyModule_Create(&moduledef);
 
#if PY_VERSION_HEX < 0x030900A4 && !defined(Py_SET_TYPE)
static inline void _Py_SET_TYPE(PyObject *ob, PyTypeObject *type)
{ ob->ob_type = type; }
#define Py_SET_TYPE(ob, type) _Py_SET_TYPE((PyObject*)(ob), type)
#endif
 
#define MOD_INIT_CREATE_CAPI(ptr, name) PyCapsule_New(ptr, name, NULL)
#define MOD_INIT_IS_CAPI(ptr) PyCapsule_CheckExact(ptr)
#define MOD_INIT_GET_CAPI(ptr, name) PyCapsule_GetPointer(ptr, name)
#define MOD_INIT_SETUP_TYPE(itype, otype) Py_SET_TYPE(&itype, otype)
#define MOD_INIT_VERIFY_TYPE_READY(type) if (PyType_Ready(type) < 0) return MOD_INIT_ERROR
 
#else
#define MOD_INIT_ERROR
#define MOD_INIT_SUCCESS(val)
#define MOD_INIT(name) void init##name(void)
#define MOD_INIT_DEF(ob, name, doc, methods) \
  ob = Py_InitModule3(name, methods, doc);
#define MOD_INIT_CREATE_CAPI(ptr, name) PyCObject_FromVoidPtr(ptr, name)
#define MOD_INIT_IS_CAPI(ptr) PyCObject_Check(ptr)
#define MOD_INIT_GET_CAPI(ptr, name) PyCObject_AsVoidPtr(ptr)
#define MOD_INIT_SETUP_TYPE(itype, otype) itype.ob_type = otype
#define MOD_INIT_VERIFY_TYPE_READY(type)
#endif
 
#define MOD_DIR_FORWARD_DECLARE(name) static PyObject* _##name##__dir__(name *self)
 
#define MOD_DIR_REFERENCE(name) _##name##__dir__
 
#define MOD_DIR_APPEND(list, str)                               \
  do {                                            \
    PyObject *o = PyUnicode_FromString(str);        \
    if (o != NULL)                          \
      PyList_Append(list, o);         \
    Py_XDECREF(o);                          \
  } while (0)
 
#define MOD_DIR_FUNCTION(name, method_listing) static PyObject * _##name##__dir__(name *self) { \
  int i=0; \
  PyObject* rc = PyList_New(0); \
  if (!rc) \
    return NULL; \
\
  while (method_listing[i].ml_name != NULL) { \
    MOD_DIR_APPEND(rc, method_listing[i++].ml_name); \
  } \
\
  return rc; \
}
 
static int PyPpcAPI_CompareWithASCIIString(PyObject* ptr, const char* name) {
  int result = -1;
  if (!PyString_Check(ptr)){
#ifdef Py_USING_UNICODE
    if (PyUnicode_Check(ptr)) {
      PyObject* tmp = PyUnicode_AsEncodedString(ptr, NULL, NULL);
      if (tmp != NULL) {
        result = strcmp(PyString_AsString(tmp), name);
        Py_DecRef(tmp);
      }
    }
#endif
  } else {
        result = strcmp(PyString_AsString(ptr), name);
  }
  return result;
}
 
#endif