root/04/bugfix/sgfparserC.cc

#include <Python.h>

typedef PyObject* PYOptr;

static PyObject* parseC(PyObject* self, PyObject* args) {

  char *sgf;
  int sgfLength;

  PyObject *parseExc;

  if (!PyArg_ParseTuple(args, "s#O", &sgf, &sgfLength, &parseExc))
    return NULL; 
  PYOptr current = PyList_New(0);

  int p = -1;
  PYOptr c[500];
  int cIndex = 0;
  char last = ')';
  bool inbrackets = false;

  int i = 0;

  // skip everything before first (; :

  int gameStart = -1;
  while (i < sgfLength && gameStart == -1) {
    while (i < sgfLength && sgf[i] != '(') i++;
    gameStart = i;
    i++;
    while (i < sgfLength && (sgf[i] == ' ' || sgf[i] == '\t' || sgf[i] == '\r' || sgf[i] == '\n')) i++;
    if (i < sgfLength && sgf[i] != ';') gameStart = -1;
  }
  
  if (i >= sgfLength) {
    PyErr_SetString(parseExc, "SGF parse error: game start not found");
    return NULL;
  }

  i = gameStart;
        
  while (i < sgfLength) {
    if (inbrackets) {
      if (sgf[i]==']') {
        int numberBackslashes = 0;
        int j = i-1;
        while (sgf[j] == '\\') {
          numberBackslashes++;
          j--;
        }
        if (!(numberBackslashes % 2)) inbrackets = false;
      }
      i++;
      continue;
    }

    if (sgf[i] == '[') inbrackets = true;
        
    if (sgf[i] == '(') {
      if (last != ')') {
        if (p != -1) {
          PyObject *temp = PyString_FromStringAndSize(sgf+p, i-p);
          if (PyList_Append(current, temp)) 
            return NULL;   // PyList_Append returns 0 on success
          Py_DECREF(temp);
        }
        PYOptr newlist = PyList_New(0);
        PyList_Append(current, newlist);
        Py_XDECREF(current);
        current = newlist;
      }
                
      PYOptr newlist = PyList_New(0);
      PyList_Append(current, newlist);
      c[cIndex++] = current;
      current = newlist;
                
      p = -1;
      last = '(';
    }

    if (sgf[i] == ')') {
      if (last != ')' && p != -1) {
        PyObject *temp = PyString_FromStringAndSize(sgf+p, i-p);
        if (PyList_Append(current, temp)) 
          return NULL;   // PyList_Append returns 0 on success
        Py_DECREF(temp);
      }
      
      if (!cIndex) {
        PyErr_SetString(parseExc, "SGF parse error: mismatched parentheses");
        return NULL;
      }
      Py_XDECREF(current);
      current = c[--cIndex];
      last = ')';
    }

    if (sgf[i] == ';') {
      if (p != -1) {
        PyObject *temp = PyString_FromStringAndSize(sgf+p, i-p);
        if (PyList_Append(current, temp)) 
            return NULL;   // PyList_Append returns 0 on success
        Py_DECREF(temp);
      }
      p = i;
    }
    i++;
  }
   
  if (inbrackets) {
    PyErr_SetString(parseExc, "SGF parse error: unexpected end of data");
    return NULL;
  }

  if (cIndex) {
    while (cIndex) {
      cIndex--;
      Py_DECREF(c[cIndex]);
    }
    PyErr_SetString(parseExc, "SGF parse error: mismatched parentheses");
    return NULL;
  }

  return current;
}


static PyMethodDef sgfparserMethods[] = {
  {"parseC",  parseC, METH_VARARGS},
  {NULL,      NULL}        /* Sentinel */
};


extern "C" {
  void initsgfparserC() {
    (void) Py_InitModule("sgfparserC", sgfparserMethods);
  }
}