#include "py4yap.h"

/**
 *
 * @section Python Built-Ins
 *
 * The Python engine includes a large number of Python built-ins. Some
 * of them are interfaced here.
 */

//@{

#include "python.h"

static PyObject *finalLookup(PyObject *i, const char *s) {
  PyObject *rc;
  if (i == NULL)
    return NULL;
  if (strcmp(s,"none") == 0)
    return Py_None;
  if (PyDict_Check(i)) {
    if ((rc = PyDict_GetItemString(i, s)))
      return rc;
  }
  if (PyModule_Check(i)) {
    if ((rc = PyDict_GetItemString(i, s)))
      return rc;
  }
  if (PyObject_HasAttrString(i, s)) {
    return PyObject_GetAttrString(i, s);
  }
  return NULL;
}

PyObject *PythonLookupSpecial(const char *s) {
  if (s == NULL)
    return NULL;
  if (strcmp(s, "true") == 0) {
    return Py_True;
  }
  if (strcmp(s, "false") == 0) {
    return Py_False;
  }
  if (strcmp(s, "[]") == 0) {
    return PyList_New(0);
  }
  if (strcmp(s, "{}") == 0) {
    return PyDict_New();
    /* return __main__,s */
  }
  return NULL;
}

PyObject *lookupPySymbol(const char *sp, PyObject *pContext, PyObject **duc) {
  PyObject *out = NULL;
  if (!sp)
    return NULL;
  if ((out = finalLookup(pContext, sp))) {
    return out;
  }
  if ((out = finalLookup(py_Context, sp))) {
    return out;
  }
  PyObject *py_Builtin = PyEval_GetBuiltins();
  if ((out = finalLookup(py_Builtin, sp))) {
    return out;
  }
  PyObject *py_Local = PyEval_GetLocals();
  if ((out = finalLookup(py_Local, sp)) && out != Py_None) {
    return out;
  }
  PyObject *py_Global = PyEval_GetGlobals();
  if ((out = finalLookup(py_Global, sp)) ) {
      return out;
  }
  if ((out = finalLookup(py_ModDict, sp))) {
      return out;
  }
  if ((out = finalLookup(py_Main, sp)) ) {
      return out;
  }
  return NULL;
}

PyObject *PythonLookup(const char *s, PyObject *oo) {
  PyObject *o;
  if ((o = PythonLookupSpecial(s)))
    return o;
  if ((o = lookupPySymbol(s, oo, NULL)) == NULL)
    return NULL;
  else {
    Py_INCREF(o);
    return o;
  }
}

PyObject *find_obj(PyObject *ob, term_t l, bool eval) {
  YAP_Term hd, yt;

  py_Context = NULL;
  yt = YAP_GetFromSlot(l);
  while (YAP_IsPairTerm(yt)) {
    hd = YAP_HeadOfTerm(yt);
    ob = yap_to_python(hd, eval, ob);
    ob = CHECKNULL(yt, ob);
    if (!ob){
      return Py_None;
    }
    yt = YAP_TailOfTerm(yt);

   }
  YAP_PutInSlot(l, yt);
  return ob;
}

/**
 * Python abs
 *
 * @param t Prolog term with a number
 *
 * @return a Python object with the number's absolute value
 */

static PyObject *bip_abs(term_t t) {
  PyObject *pVal, *nVal;

  AOK( PL_get_arg(1, t, t), NULL);
  pVal = term_to_python(t, true, NULL);
  pVal = CHECKNULL( t, pVal );
  nVal = PyNumber_Absolute(pVal);
  Py_DecRef(pVal);
  return nVal;
}

/**
 * Python all
 *
 * @param t Prolog term with a previously constructed Python iterator
> *
 * @return the Python boolean `True` if all elements of the iterator are `True`,
 *    `False`  otherwise.
 */
static PyObject *bip_all(term_t t) {
  PyObject *it, *item, *v;
  PyObject *(*iternext)(PyObject *);
  int cmp;

  AOK( PL_get_arg(1, t, t),  NULL );
  (v = term_to_python(t, true, NULL) );
  v = CHECKNULL( t, v);
  it = PyObject_GetIter(v);
  if (CHECKNULL( t, it) == NULL)
    return Py_None;
  iternext = *Py_TYPE(it)->tp_iternext;

  if (PyErr_Occurred()) {
    if (PyErr_ExceptionMatches(PyExc_StopIteration))
      PyErr_Clear();
    else
      return Py_None;
  }
  //  PyObject_Print(v, stderr, 0);
  for (;;) {
    item = iternext(it);
    if (CHECKNULL(t,item) == NULL)
      break;
    cmp = PyObject_IsTrue(item);
    Py_DECREF(item);
    if (cmp < 0) {
      Py_DECREF(it);
      return Py_None;
    }
    if (cmp == 0) {
      Py_DECREF(it);
      return Py_False;
    }
  }
  Py_DECREF(it);
  return Py_True;
}

/**
 * Python any
 *
 * @param t Prolog term with a previously constructed Python iterator
 *
 * @return the Python boolean `True` if any element of the iterator is `True`,
 *    `False`  if all of them are false.
 */
static PyObject *bip_any(term_t t) {
  PyObject *it, *item, *v;
  PyObject *(*iternext)(PyObject *);
  int cmp;

  AOK( PL_get_arg(1, t, t), NULL  );
  v = term_to_python(t, true, NULL);
  it = PyObject_GetIter(v);
  if (CHECKNULL(t,it) == NULL)
    return Py_None;
  iternext = *Py_TYPE(it)->tp_iternext;

  for (;;) {
    item = iternext(it);
    if (CHECKNULL(t,item) == NULL)
      break;
    cmp = PyObject_IsTrue(item);
    Py_DECREF(item);
    if (cmp < 0) {
      Py_DECREF(it);
      return Py_None;
    }
    if (cmp == 1) {
      Py_DECREF(it);
      Py_RETURN_TRUE;
    }
  }
  Py_DECREF(it);
  if (PyErr_Occurred()) {
    if (PyErr_ExceptionMatches(PyExc_StopIteration))
      PyErr_Clear();
    else
      return CHECKNULL(t,NULL);
  }
  Py_RETURN_FALSE;
}

static PyObject *bip_bin(term_t t) {
  PyObject *v;

  AOK ( PL_get_arg(1, t, t), NULL);
  v = term_to_python(t, true, NULL);
  return PyNumber_ToBase(v, 2);
}

static PyObject *bip_float(term_t t, bool eval) {
  PyObject *pVal, *o;

  AOK( PL_get_arg(1, t, t), NULL );
  pVal = term_to_python(t, eval, NULL);
  if (PyLong_Check(pVal)) {
    o = PyFloat_FromDouble(PyLong_AsLong(pVal));
#if PY_MAJOR_VERSION < 3
  } else if (PyInt_Check(pVal)) {
    o = PyFloat_FromDouble(PyInt_AsLong(pVal));
#endif
  } else if (PyFloat_Check(pVal)) {
    return pVal;
  } else
    return Py_None;
  Py_DECREF(pVal);
  return o;
}

static PyObject *bip_int(term_t t) {
  PyObject *pVal, *o;

  AOK( PL_get_arg(1, t, t), NULL );
  pVal = term_to_python(t, true, NULL);
#if PY_MAJOR_VERSION < 3
  if (PyLong_Check(pVal)) {
    o = PyInt_FromLong(PyLong_AsLong(pVal));
  } else if (PyInt_Check(pVal)) {
    return pVal;
#else
  if (PyLong_Check(pVal)) {
    return pVal;
#endif
  } else if (PyFloat_Check(pVal)) {
#if PY_MAJOR_VERSION < 3
    o = PyInt_FromLong(PyFloat_AsDouble(pVal));
#else
    o = PyLong_FromDouble(PyFloat_AsDouble(pVal));
#endif
  } else
    return Py_None;
  Py_DECREF(pVal);
  return o;
}

static PyObject *bip_long(term_t t) {
  PyObject *pVal, *o;

  AOK( PL_get_arg(1, t, t), NULL );
  pVal = term_to_python(t, true, NULL);
  if (PyLong_Check(pVal)) {
    return pVal;
#if PY_MAJOR_VERSION < 3
  } else if (PyInt_Check(pVal)) {
    o = PyLong_FromLong(PyInt_AsLong(pVal));
#endif
  } else if (PyFloat_Check(pVal)) {
    o = pVal;
  } else
    return Py_None;
  Py_DECREF(pVal);
  return o;
}

static PyObject *bip_iter(term_t t) {
  PyObject *v;

  AOK( PL_get_arg(1, t, t), NULL );
   v = term_to_python(t, true, NULL);
  return PyObject_GetIter(v);
}

static PyObject *bip_ord(term_t t) {
  PyObject *pVal;
  Py_ssize_t size;

  AOK( PL_get_arg(1, t, t), NULL );
 pVal = term_to_python(t, true, NULL);
  if (PyUnicode_Check(pVal)) {
#if PY_MAJOR_VERSION < 3
    size = PyUnicode_GET_SIZE(pVal);
#else
    size = PyUnicode_GetLength(pVal);
#endif
    if (size == 1) {
#if PY_MAJOR_VERSION < 3
      long ord = (long)*PyUnicode_AS_UNICODE(pVal);
      return PyInt_FromLong(ord);
#else
      Py_UCS4 ord = PyUnicode_ReadChar(pVal, 0);
      return PyLong_FromLong(ord);
#endif
    }
    return Py_None;
  } else if (PyByteArray_Check(pVal)) {
    char *s = PyByteArray_AsString(pVal);

    if (s[1])
      return Py_None;
#if PY_MAJOR_VERSION < 3
    return PyInt_FromLong(s[0]);
  } else if (PyString_Check(pVal)) {
    char *s = PyString_AsString(pVal);

    if (s[1])
      return Py_None;
    return PyInt_FromLong(s[0]);
#else
    return PyLong_FromLong(s[0]);
#endif
  } else
    return NULL;
}

static PyObject *bip_sum(term_t t) {
  PyObject *seq;
  PyObject *result = NULL;
  PyObject *temp, *item, *iter;

  AOK( PL_get_arg(1, t, t), NULL );
  seq = term_to_python(t, true, NULL);
  iter = PyObject_GetIter(seq);
  if (iter == NULL)
    return NULL;

  if (result == NULL) {
#if PY_MAJOR_VERSION < 3
    result = PyInt_FromLong(0);
#else
    result = PyLong_FromLong(0);
#endif
    if (result == NULL) {
      Py_DECREF(iter);
      return NULL;
    }
  } else {
#if PY_MAJOR_VERSION < 3
    /* reject string values for 'start' parameter */
    if (PyObject_TypeCheck(result, &PyBaseString_Type)) {
      PyErr_SetString(PyExc_TypeError,
                      "sum() can't sum strings [use ''.join(seq) instead]");
      Py_DECREF(iter);
      return NULL;
    }
    Py_INCREF(result);
#endif
  }

#ifndef SLOW_SUM
/* Fast addition by keeping temporary sums in C instead of new Python objects.
   Assumes all inputs are the same type.  If the assumption fails, default
   to the more general routine.
*/
#if PY_MAJOR_VERSION < 3
  if (PyInt_CheckExact(result)) {
    long i_result = PyInt_AS_LONG(result);
#else
  if (PyLong_CheckExact(result)) {
    long i_result = PyLong_AS_LONG(result);
#endif
    Py_DECREF(result);
    result = NULL;
    while (result == NULL) {
      item = PyIter_Next(iter);
      if (item == NULL) {
        Py_DECREF(iter);
        if (PyErr_Occurred())
          return NULL;
#if PY_MAJOR_VERSION < 3
        return PyInt_FromLong(i_result);
#else
        return PyLong_FromLong(i_result);
#endif
      }
#if PY_MAJOR_VERSION < 3
      if (PyInt_CheckExact(item)) {
        long b = PyInt_AS_LONG(item);
#else
      if (PyLong_CheckExact(item)) {
        long b = PyLong_AS_LONG(item);
#endif
        long x = i_result + b;
        if ((x ^ i_result) >= 0 || (x ^ b) >= 0) {
          i_result = x;
          Py_DECREF(item);
          continue;
        }
      }
/* Either overflowed or is not an int. Restore real objects and process normally
 */
#if PY_MAJOR_VERSION < 3
      result = PyInt_FromLong(i_result);
#else
      result = PyLong_FromLong(i_result);
#endif
      temp = PyNumber_Add(result, item);
      Py_DECREF(result);
      Py_DECREF(item);
      result = temp;
      if (result == NULL) {
        Py_DECREF(iter);
        return NULL;
      }
    }
  }

  if (PyFloat_CheckExact(result)) {
    double f_result = PyFloat_AS_DOUBLE(result);
    Py_DECREF(result);
    result = NULL;
    while (result == NULL) {
      item = PyIter_Next(iter);
      if (item == NULL) {
        Py_DECREF(iter);
        if (PyErr_Occurred())
          return NULL;
        return PyFloat_FromDouble(f_result);
      }
      if (PyFloat_CheckExact(item)) {
        PyFPE_START_PROTECT("add", Py_DECREF(item); Py_DECREF(iter); return 0)
            f_result += PyFloat_AS_DOUBLE(item);
        PyFPE_END_PROTECT(f_result) Py_DECREF(item);
        continue;
      }
#if PY_MAJOR_VERSION < 3
      if (PyInt_CheckExact(item)) {
        764PyFPE_START_PROTECT("add", Py_DECREF(item); Py_DECREF(iter); return 0)
            f_result += (double)PyInt_AS_LONG(item);
        PyFPE_END_PROTECT(f_result) Py_DECREF(item);
        continue;
      }
#else
      if (PyLong_CheckExact(item)) {
        PyFPE_START_PROTECT("add", Py_DECREF(item); Py_DECREF(iter); return 0)
            f_result += PyLong_AsDouble(item);
        PyFPE_END_PROTECT(f_result) Py_DECREF(item);
        continue;
      }
#endif
      result = PyFloat_FromDouble(f_result);
      temp = PyNumber_Add(result, item);
      Py_DECREF(result);
      Py_DECREF(item);
      result = temp;
      if (result == NULL) {
        Py_DECREF(iter);
        return NULL;
      }
    }
#endif
  }

  for (;;) {
    item = PyIter_Next(iter);
    if (item == NULL) {
      /* error, or end-of-sequence */
      if (PyErr_Occurred()) {
        Py_DECREF(result);
        result = NULL;
      }
      break;
    }
    /* It's tempting to use PyNumber_InPlaceAdd instead of
       PyNumber_Add here, to avoid quadratic running time
       when doing 'sum(list_of_lists, [])'.  However, this
       would produce a change in behaviour: a snippet like

       empty = []
       sum([[x] for x in range(10)], empty)

       would change the value of empty. */
    temp = PyNumber_Add(result, item);
    Py_DECREF(result);
    Py_DECREF(item);
    result = temp;
    if (result == NULL)
      break;
  }
  Py_DECREF(iter);
  return result;
}

//@}

static long get_int(term_t arg, bool eval) {
  long low;

  if (!PL_get_long(arg, &low)) {
    PyObject *low = term_to_python(arg, eval, NULL);
    if (PyLong_Check(low)) {
      return PyLong_AsLong(low);
#if PY_MAJOR_VERSION < 3
    } else if (PyInt_Check(low)) {
      return PyInt_AsLong(low);
#endif
    } else {
      return 0;
    }
  }
  return low;
}

/* Return number of items in range/xrange (lo, hi, step).  step > 0
 * required.  Return a value < 0 if & only if the true value is too
 * large to fit in a signed long.
 */
static long get_len_of_range(long lo, long hi, long step) {
  /* -------------------------------------------------------------
     If lo >= hi, the range is empty.
     Else if n values are in the range, the last one is
     lo + (n-1)*step, which must be <= hi-1.  Rearranging,
     n <= (hi - lo - 1)/step + 1, so taking the floor of the RHS gives
     the proper value.  Since lo < hi in this case, hi-lo-1 >= 0, so
     the RHS is non-negative and so truncation is the same as the
     floor.  Letting M be the largest positive long, the worst case
     for the RHS numerator is hi=M, lo=-M-1, and then
     hi-lo-1 = M-(-M-1)-1 = 2*M.  Therefore unsigned long has enough
     precision to compute the RHS exactly.
     ---------------------------------------------------------------*/
  long n = 0;
  if (lo < hi) {
    unsigned long uhi = (unsigned long)hi;
    unsigned long ulo = (unsigned long)lo;
    unsigned long diff = uhi - ulo - 1;
    n = (long)(diff / (unsigned long)step + 1);
  }
  return n;
}

#if PY_MAJOR_VERSION >= 3
static PyStructSequence_Field pnull[] = {
    {"A1", NULL},  {"A2", NULL},  {"A3", NULL},  {"A4", NULL},
    {"A5", NULL},  {"A6", NULL},  {"A7", NULL},  {"A8", NULL},
    {"A9", NULL},  {"A9", NULL},  {"A10", NULL}, {"A11", NULL},
    {"A12", NULL}, {"A13", NULL}, {"A14", NULL}, {"A15", NULL},
    {"A16", NULL}, {"A17", NULL}, {"A18", NULL}, {"A19", NULL},
    {"A19", NULL}, {"A20", NULL}, {"A21", NULL}, {"A22", NULL},
    {"A23", NULL}, {"A24", NULL}, {"A25", NULL}, {"A26", NULL},
    {"A27", NULL}, {"A28", NULL}, {"A29", NULL}, {"A29", NULL},
    {"A30", NULL}, {"A31", NULL}, {"A32", NULL}, {NULL,NULL}};

static PyObject *structseq_str(PyObject *iobj) {

/* buffer and type size were chosen well considered. */
#define REPR_BUFFER_SIZE 512
#define TYPE_MAXSIZE 100

  PyStructSequence *obj = (PyStructSequence *)iobj;
  PyTypeObject *typ = Py_TYPE(obj);
  bool removelast = false;
  Py_ssize_t len, i;
  char buf[REPR_BUFFER_SIZE];
  char *endofbuf, *pbuf = buf;
  /* pointer to end of writeable buffer; safes space for "...)\0" */
  endofbuf = &buf[REPR_BUFFER_SIZE - 5];

  /* "typename(", limited to  TYPE_MAXSIZE */
  len =
      strlen(typ->tp_name) > TYPE_MAXSIZE ? TYPE_MAXSIZE : strlen(typ->tp_name);
  strncpy(pbuf, typ->tp_name, len);
  pbuf += len;
  *pbuf++ = '(';

  for (i = 0; i < ((PyStructSequence *)obj)->ob_base.ob_size; i++) {
    PyObject *val, *repr;
    char *crepr;

    val = PyStructSequence_GET_ITEM(obj, i);
    repr = PyObject_Str(val);
    if (repr == NULL)
      return Py_None;
    crepr = PyUnicode_AsUTF8(repr);
    if (crepr == NULL) {
      Py_DECREF(repr);
      return Py_None;
    }

    /* + 3: keep space for ", " */
    len = strlen(crepr) + 2;
    if ((pbuf + len) <= endofbuf) {
      strcpy(pbuf, crepr);
      pbuf += strlen(crepr);
      *pbuf++ = ',';
      *pbuf++ = ' ';
      removelast = 1;
      Py_DECREF(repr);
    } else {
      strcpy(pbuf, "...");
      pbuf += 3;
      removelast = 0;
      Py_DECREF(repr);
      break;
    }
  }
  if (removelast) {
    /* overwrite last ", " */
    pbuf -= 2;
  }
  *pbuf++ = ')';
  *pbuf = '\0';

  return PyUnicode_FromString(buf);
}

static PyObject *structseq_repr(PyObject *iobj) {

/* buffer and type size were chosen well considered. */
#define REPR_BUFFER_SIZE 512
#define TYPE_MAXSIZE 100

  PyStructSequence *obj = (PyStructSequence *)iobj;
  PyTypeObject *typ = Py_TYPE(obj);
  bool removelast = false;
  Py_ssize_t len, i;
  char buf[REPR_BUFFER_SIZE];
  char *endofbuf, *pbuf = buf;
  /* pointer to end of writeable buffer; safes space for "...)\0" */
  endofbuf = &buf[REPR_BUFFER_SIZE - 5];

  /* "typename(", limited to  TYPE_MAXSIZE */
  len =
      strlen(typ->tp_name) > TYPE_MAXSIZE ? TYPE_MAXSIZE : strlen(typ->tp_name);
  strncpy(pbuf, typ->tp_name, len);
  pbuf += len;
  *pbuf++ = '(';

  for (i = 0; i < ((PyStructSequence *)obj)->ob_base.ob_size; i++) {
    PyObject *val, *repr;
    char *crepr;

    val = PyStructSequence_GET_ITEM(obj, i);
    repr = PyObject_Repr(val);
    if (repr == NULL)
      return NULL;
    crepr = PyUnicode_AsUTF8(repr);
    if (crepr == NULL) {
      Py_DECREF(repr);
      return NULL;
    }

    /* + 3: keep space for ", " */
    len = strlen(crepr) + 2;
    if ((pbuf + len) <= endofbuf) {
      strcpy(pbuf, crepr);
      pbuf += strlen(crepr);
      *pbuf++ = ',';
      *pbuf++ = ' ';
      removelast = 1;
      Py_DECREF(repr);
    } else {
      strcpy(pbuf, "...");
      pbuf += 3;
      removelast = 0;
      Py_DECREF(repr);
      break;
    }
  }
  if (removelast) {
    /* overwrite last ", " */
    pbuf -= 2;
  }
  *pbuf++ = ')';
  *pbuf = '\0';

  return PyUnicode_FromString(buf);
}
#endif


PyObject *term_to_nametuple(const char *s, arity_t arity, PyObject *tuple) {
  PyObject *o;
#if PY_MAJOR_VERSION >= 3 
  PyTypeObject *typp;
  PyObject *key = PyUnicode_FromString(s);
  if (py_F2P && PyDict_Contains(py_F2P, key)) {
    typp = (PyTypeObject *)PyDict_GetItem(py_F2P, key);
    Py_INCREF(typp);
  } else {

    typp = calloc(sizeof(PyTypeObject), 1);
    PyStructSequence_Desc *desc = calloc(sizeof(PyStructSequence_Desc), 1);
    desc->name = PyMem_Malloc(strlen(s)+1);
    strcpy(desc->name, s);
    Py_DECREF(key);
    desc->doc = "YAPTerm";
    desc->fields = pnull;
    desc->n_in_sequence = arity;
    if (PyStructSequence_InitType2(typp, desc) < 0)
      return NULL;
    typp->tp_flags &= ~Py_TPFLAGS_HEAPTYPE;
    // typp->tp_str = structseq_str;
    typp->tp_repr = structseq_repr;
    //     typp = PyStructSequence_NewType(desc);
    // don't do this: we cannot add a type as an atribute.
    //PyModule_AddObject(py_Main, s, (PyObject *)typp);
    if (py_F2P)
      PyDict_SetItem(py_F2P, key, (PyObject *)typp);
    Py_INCREF(typp);
  }
  o = PyStructSequence_New(typp);
  arity_t i;
  for (i=0; i < arity; i++) {
  PyObject *pArg = PyTuple_GET_ITEM(tuple, i);
      Py_INCREF(pArg);
  if (pArg)
    PyStructSequence_SET_ITEM(o, i, pArg);
  //PyObject_Print(pArg,stderr,0);fputc('\n',stderr);
 }
  //((PyStructSequence *)o)->ob_base.ob_size = arity;
  //PyObject_Print(o,stderr,0);fputc('\n',stderr);
  return o;
#else
  PyObject *o1;
  o1 = PyTuple_New(2);
  PyTuple_SET_ITEM(o1, 0, PyUnicode_FromString(s));
  PyTuple_SET_ITEM(o1, 1, tuple);
  return o1;
#endif
}

static PyObject *bip_range(term_t t) {
  long ilow = 0, ihigh = 0, istep = 1;
  long bign;
  Py_ssize_t i, n;
  int arity;
  atom_t name;
  term_t arg = PL_new_term_ref();

  PyObject *v;

  if (!PL_get_name_arity(t, &name, &arity))
    return NULL;
  AOK( PL_get_arg(1, t, arg), NULL);
  ilow = get_int(arg, true);
  if (arity == 1) {
    ihigh = ilow;
    ilow = 0;
  } else {
    AOK( PL_get_arg(2, t, arg), NULL );
    ihigh = get_int(arg, true);
    if (arity == 3) {
      AOK( PL_get_arg(3, t, arg), NULL );
      istep = get_int(arg, true);
    }
  }
  if (istep == 0) {
    PyErr_SetString(PyExc_ValueError, "range() step argument must not be zero");
    return NULL;
  }
  if (istep > 0)
    bign = get_len_of_range(ilow, ihigh, istep);
  else
    bign = get_len_of_range(ihigh, ilow, -istep);
  n = (Py_ssize_t)bign;
  AOK ( ( (bign >= 0 && (long)n == bign) || "range() result has too many items" ), NULL );
  v = PyList_New(n);

  if (v == NULL)
    return Py_None;
  for (i = 0; i < n; i++) {
#if PY_MAJOR_VERSION < 3
    PyObject *w = PyInt_FromLong(ilow);
#else
    PyObject *w = PyLong_FromLong(ilow);
#endif
    if (w == NULL) {
      Py_DECREF(v);
      return NULL;
    }
    PyList_SET_ITEM(v, i, w);
    Py_INCREF(w);
    ilow += istep;
  }
  return v;
}

static bool copy_to_dictionary(PyObject *dict, term_t targ, term_t taux,
                              bool eval) {
  PyObject *lhs, *rhs;
  term_t tleft = PL_new_term_ref(), tright = PL_new_term_ref();

    functor_t fun;

    AOK (PL_get_functor(targ, &fun), false);
    while (fun == FUNCTOR_comma2) {
      AOK( PL_get_arg(1, targ, tleft), false);
      if (! copy_to_dictionary(dict, tleft, taux, eval) )
	return false;
      AOK ( PL_get_arg(2, targ, targ), false);
      return copy_to_dictionary(dict, tright, taux, eval);
    }
    // PyObject_Print(dict, stderr, 0); fprintf(stderr,"\n");
    // Py_DECREF(lhs);
    // Py_DECREF(rhs);

    AOK(PL_get_arg(1, targ, tleft), false);
    lhs = atom_to_python_string(tleft);
    if (lhs == NULL) {
      return FALSE;
    }
    AOK(PL_get_arg(2, targ, tright), false);
    rhs = term_to_python(tright, eval, NULL);
    if (rhs == NULL) {
      PyErr_Print();
      return false;
    }
    return PyDict_SetItem(dict, lhs, rhs) >= 0;
    // PyObject_Print(dict, stderr, 0); fprintf(stderr,"\n");
    // Py_DECREF(lhs);
    // Py_DECREF(rhs);
  }


PyObject *compound_to_data(term_t t, PyObject *o, functor_t fun, bool exec) {
  atom_t name;
  int len;

  AOK (PL_get_name_arity(t, &name, &len), o);

  if (fun == FUNCTOR_pointer1) {
    void *ptr;

    AOK (PL_get_arg(1, t, t), NULL);
    AOK (PL_get_pointer(t, &ptr), NULL)
    /* return __main__,s */
    return (PyObject *)ptr;
  }
  if (name == ATOM_t) {
    term_t targ = PL_new_term_ref();
    PyObject *out;
    int i;

    out = PyTuple_New(len);
    DebugPrintf("Tuple %p\n", out);

    if (!out)
      return NULL;
    for (i = 0; i < len; i++) {
      AOK ( PL_get_arg(i + 1, t, targ), NULL);
      PyErr_Clear();
      PyObject *oa = term_to_python(targ, true, o);
      bool rc = PyTuple_SET_ITEM(out, i, oa) == 0;
      if (rc) {
        PyErr_Print();
      }
      Py_INCREF(oa);
    }
    return out;
  } else if (fun == FUNCTOR_div2) {
    term_t targ = PL_new_term_ref();
    PyObject *lhs, *rhs;

    AOK ( PL_get_arg(1, t, targ),  NULL );
    lhs = term_to_python(targ, true, NULL);
    if (!PyNumber_Check(lhs))
      return NULL;
    AOK ( PL_get_arg(2, t, targ), NULL );
    rhs = term_to_python(targ, true, NULL);
    if (!PyNumber_Check(rhs))
      return NULL;
#if PY_MAJOR_VERSION < 3
    return PyNumber_Divide(lhs, rhs);
#else
    return PyNumber_TrueDivide(lhs, rhs);
#endif
  }
  if (fun == FUNCTOR_sqbrackets2) {
    term_t targ = PL_new_term_ref(), trhs = PL_new_term_ref();
    PyObject *v;
    Py_ssize_t min, max;
    AOK (PL_get_arg(2, t, targ), NULL);
    v = term_to_python(targ, true, o);

    AOK (PL_get_arg(1, t, targ), NULL );
    AOK (PL_get_list(targ, trhs, targ) , NULL);
    if (PL_is_functor(trhs, FUNCTOR_colon2)) {
      if (!PySequence_Check(v))
        return NULL;
      min = get_p_int(term_to_python(targ, true, NULL), 0);
      AOK (PL_get_arg(1, trhs, targ), NULL);
      if (PL_is_functor(targ, FUNCTOR_colon2)) {
        return NULL;
      }
      max = get_p_int(term_to_python(targ, true, o), PyObject_Size(v));
      return PySequence_GetSlice(v, min, max);
    } else {
      PyObject *ip = term_to_python(trhs, true, o);
      if (PySequence_Check(v)) {
#if PY_MAJOR_VERSION < 3
  if (PyLong_Check(ip) {
          min = PyLong_AsLong(ip);
  } else if (PyInt_Check(ip) {
          min = PyInt_asInt(ip);
      }
#else
        if (PyLong_Check(ip)) {
          PyObject *o = PySequence_GetItem(v, PyLong_AsLong(ip));
	  if (o == NULL)
	    o = Py_None;
          if (CHECKNULL(t,o) == NULL)
          return NULL;
          Py_INCREF(o);
          return o;
        }
#endif
      } else {
        o = PyObject_GetItem(v, ip);
	if (o == NULL)
	  o = Py_None;
        Py_INCREF(o);
        return o;
      }
    }
  }
  if (fun == FUNCTOR_dollar1) {
    char *s = NULL;
    term_t targ = PL_new_term_ref();
    AOK ( PL_get_arg(1, t, targ), NULL );
    AOK ( PL_get_atom_chars(targ, &s), NULL );
    /* return __main__,s */
    PyObject *o = PyObject_GetAttrString(py_Main, s);
    return o;
  }
  if (fun == FUNCTOR_brackets1) {
    AOK ( PL_get_arg(1, t, t),  NULL );
    return term_to_python(t, true, NULL);
  }
  if (fun == FUNCTOR_complex2) {
    term_t targ = PL_new_term_ref();
    PyObject *lhs, *rhs;
    double d1, d2;

    AOK ( PL_get_arg(1, t, targ), NULL );
    lhs = term_to_python(targ, true, NULL);
    AOK ( PyNumber_Check(lhs), NULL );
    if (PyFloat_Check(lhs)) {
      d1 = PyFloat_AsDouble(lhs);
    } else if (PyLong_Check(lhs)) {
      d1 = PyLong_AsLong(lhs);
#if PY_MAJOR_VERSION < 3
    } else if (PyInt_Check(lhs)) {
      d1 = PyInt_AsLong(lhs);
#endif
    } else {
      return NULL;
    }
    AOK (PL_get_arg(2, t, targ), NULL );
    rhs = term_to_python(targ, true, NULL);
    AOK (PyNumber_Check(rhs), NULL);
    if (PyFloat_Check(rhs)) {
      d2 = PyFloat_AsDouble(rhs);
    } else if (PyLong_Check(rhs)) {
      d2 = PyLong_AsLong(rhs);
#if PY_MAJOR_VERSION < 3
    } else if (PyInt_Check(rhs)) {
      d2 = PyInt_AsLong(rhs);
#endif
    } else {
      return NULL;
    }

    return PyComplex_FromDoubles(d1, d2);
  }
  if (fun == FUNCTOR_curly1) {
    term_t targ = PL_new_term_ref(), taux = PL_new_term_ref();
    PyObject *dict;

    AOK ( PL_get_arg(1, t, t), NULL );
    if (!(dict = PyDict_New()))
      return NULL;
    DebugPrintf("Dict %p\n", dict);

    while (PL_is_functor(t, FUNCTOR_comma2)) {
      AOK ( PL_get_arg(1, t, targ), NULL );
      AOK ( PL_is_functor(targ, FUNCTOR_colon2), NULL);

      AOK( copy_to_dictionary(dict, targ, taux, true), NULL);
      AOK( PL_get_arg(2, t, t) , NULL );
     }

    if (PL_is_functor(t, FUNCTOR_colon2)) {
      AOK ( copy_to_dictionary(dict, t, taux, true), NULL);
    }
    return dict;
  }
  return o;
}

PyObject *compound_to_pytree(term_t t, PyObject *context) {
  PyObject *o = py_Main, *no;
  functor_t fun;
  atom_t name;
  int arity;

  o = find_obj(context, t, false);
  AOK( PL_get_name_arity(t, &name, &arity), NULL );
  if (arity == 0)
    return term_to_python(t, false, o);
  AOK( PL_get_functor(t, &fun), NULL);
  if ((no = compound_to_data(t, o, fun, false)) != o && no) {
    return no;
  }
  if (!arity) {
    char *s = NULL;

    AOK (!PL_get_atom_chars(t, &s), NULL );
    // this should never happen
    return term_to_python(t, false, o);
  } else {
    const char *s;
    if (!(s = PL_atom_chars(name))) {
      return NULL;
    }
    term_t tleft;
    int i;
    PyObject *out = PyTuple_New(arity);
    DebugPrintf("Tuple %p\n", o);
    tleft = PL_new_term_ref();
    for (i = 0; i < arity; i++) {
      PyObject *pArg;
      AOK (PL_get_arg(i + 1, t, tleft), NULL );
      pArg = term_to_python(tleft, false, NULL);
      if (pArg) {
      /* pArg reference stolen here: */
      PyTuple_SET_ITEM(out, i, pArg);
      Py_INCREF(pArg);
      }
    }
    if (CHECKNULL(t, out) == NULL) {
      PyErr_Print();
      return NULL;
    }
    PyObject *c = lookupPySymbol(s, o, NULL);

    if ( c && PyCallable_Check(c)) {
        PyObject *n = PyTuple_New(arity);
        PyTuple_SET_ITEM(n, 0, c);
        PyTuple_SET_ITEM(n, 1, out);
        return n;
    }
    return term_to_nametuple(s, arity, out);
  }
}

PyObject *compound_to_pyeval(term_t t, PyObject *context) {
  PyObject *o = NULL, *no;
  atom_t name;
  int arity;
  functor_t fun;

  o = find_obj(context, t, true);
   AOK (PL_get_name_arity(t, &name, &arity), NULL );
  if (arity == 0)
    return term_to_python(t, true, o);
  if (!PL_get_functor(t, &fun))
    return NULL;
  if ((no = compound_to_data(t, o, fun, true)) != o && no) {
    return no;
  }
  if (fun == FUNCTOR_abs1) {
    return bip_abs(t);
  } else if (fun == FUNCTOR_all1) {
    return bip_all(t);
  } else if (fun == FUNCTOR_any1) {
    return bip_any(t);
  } else if (fun == FUNCTOR_bin1) {
    return bip_bin(t);
  } else if (fun == FUNCTOR_ord1) {
    return bip_ord(t);
  } else if (fun == FUNCTOR_int1) {
    return bip_int(t);
  } else if (fun == FUNCTOR_long1) {
    return bip_long(t);
  } else if (fun == FUNCTOR_float1) {
    return bip_float(t, true);
  } else if (fun == FUNCTOR_iter1) {
    return bip_iter(t);
  } else if (fun == FUNCTOR_range1 || fun == FUNCTOR_range2 ||
             fun == FUNCTOR_range3) {
    return bip_range(t);
  } else if (fun == FUNCTOR_sum1) {
    return bip_sum(t);
  }
  if (fun == FUNCTOR_len1) {
    term_t targ = PL_new_term_ref();
    PyObject *ptr;

    AOK (PL_get_arg(1, t, targ), NULL );
    ptr = term_to_python(targ, true, NULL);
    return PyLong_FromLong(PyObject_Length(ptr));
  }
  if (fun == FUNCTOR_dir1) {
    term_t targ = PL_new_term_ref();
    PyObject *ptr;

    AOK ( PL_get_arg(1, t, targ), NULL );
    ptr = term_to_python(targ, true, NULL);
    return PyObject_Dir(ptr);

  }

  else if (fun == FUNCTOR_plus2) {
    term_t targ = PL_new_term_ref();
    PyObject *lhs, *rhs;

    if (!PL_get_arg(1, t, targ))
      return NULL;
    lhs = term_to_python(targ, true, NULL);
    AOK( PL_get_arg(2, t, targ), NULL );
    rhs = term_to_python(targ, true, NULL);
    if (PySequence_Check(lhs) && PySequence_Check(rhs)) {
      return PySequence_Concat(lhs, rhs);
    }
    if (!PyNumber_Check(lhs))
      return NULL;
    if (!PyNumber_Check(rhs))
      return NULL;
    return PyNumber_Add(lhs, rhs);
  } else if (fun == FUNCTOR_sub2) {
    term_t targ = PL_new_term_ref();
    PyObject *lhs, *rhs;

    if (!PL_get_arg(1, t, targ))
      return NULL;
    lhs = term_to_python(targ, true, NULL);
    if (!PyNumber_Check(lhs))
      return NULL;
    if (!PL_get_arg(2, t, targ))
      return NULL;
    rhs = term_to_python(targ, true, NULL);
    if (!PyNumber_Check(rhs))
      return NULL;
    return PyNumber_Subtract(lhs, rhs);
  } else if (fun == FUNCTOR_mul2) {
    term_t targ = PL_new_term_ref();
    PyObject *lhs, *rhs;

    AOK (PL_get_arg(1, t, targ), NULL );
    (lhs = term_to_python(targ, true, NULL));
    CHECKNULL( targ, lhs);
    AOK (PL_get_arg(2, t, targ), NULL );
    (rhs = term_to_python(targ, true, NULL));
    CHECKNULL( targ, rhs);
    if (PySequence_Check(lhs) && (
#if PY_MAJOR_VERSION < 3
                                     PyInt_Check(rhs) ||
#endif
                                     PyLong_Check(rhs))) {
      return PySequence_Repeat(lhs, get_p_int(rhs, 0));
    }
    if (!PyNumber_Check(lhs) + !PyNumber_Check(rhs))
      return NULL;
    return PyNumber_Multiply(lhs, rhs);
  }
  if (!arity) {
    char *s = NULL;
    PyObject *pValue;

    AOK (PL_get_atom_chars(t, &s), NULL );
    pValue = PyObject_GetAttrString(o, s);
    if (CHECKNULL(t,pValue) == NULL) {
      PyErr_Print();
      return NULL;
    }
    return pValue;
  } else {
    char *s = PL_atom_chars(name);
    o = lookupPySymbol(s, o, NULL);
    if (CHECKNULL(t,o) == NULL) {
      PyErr_Print();
      return NULL;
    }
    PyObject *pArgs = PyTuple_New(arity);
    DebugPrintf("Tuple %p\n", pArgs);
    int i;
    term_t tleft = PL_new_term_ref();
    for (i = 0; i < arity; i++) {
      PyObject *pArg;
      AOK (PL_get_arg(i + 1, t, tleft),  NULL );
      /* ignore (_) */
      if (i == 0 && PL_is_variable(tleft)) {
        pArg = Py_None;
      } else {
	pArg = term_to_python(tleft, true, NULL);
  PyObject_Print(pArg,fdopen(2,"w"),0);
	if (pArg == NULL) {
	  pArg = Py_None;
	}
	/* pArg reference stolen here: */
	Py_INCREF(pArg);
      }

      PyTuple_SetItem(pArgs, i, pArg);
    }
    if ( !PyCallable_Check(o)) {
    return term_to_nametuple(s, arity, pArgs);
    }
PyObject *rc;

    //PyObject_Print(pArgs, stderr, 0);
    //PyObject_Print(o, stderr, 0);
	CHECK_CALL(rc ,t, PyObject_CallObject(o, pArgs));
    Py_DECREF(pArgs);
    Py_DECREF(o);
    DebugPrintf("CallObject %p\n", rc);

    return rc;
  }
}