StrIter.h

Go to the documentation of this file.

#pragma once
#define PY_SSIZE_T_CLEAN
#include <Python.h>
 
#include <string>
#include <typeinfo>
#include <vector>
 
#include "utils.h"
 
namespace k1a {
 
typedef std::string (*transformF)(std::string);
 
class StrIter {
   public:
    PyObject *pyObj;  
    bool done;        
 
    StrIter();
    StrIter(PyObject *pyObj);
    virtual std::string next();
    StrIter *transform(std::vector<transformF> fs, bool decref = false);
    long length();
    ~StrIter();
 
    template <class T>
    static PyObject *Py_next(PyObject *self);
    template <class T>
    static PyObject *Py_repr(PyObject *self);
    template <class T>
    static Py_ssize_t Py_length(PyObject *self);
    template <class T>
    static void Py_dealloc(PyObject *self);
    template <class T>
    static PySequenceMethods *Py_as_sequence();
};
 
template <class T>
PySequenceMethods *StrIter::Py_as_sequence() {
    PySequenceMethods *answer = new PySequenceMethods();
    *answer = {
        (lenfunc)StrIter::Py_length<T>, /* sq_length */
        0,                              /* sq_concat */
        0,                              /* sq_repeat */
        0,                              /* sq_item */
        0,                              /* sq_slice */
        0,                              /* sq_ass_item */
        0,                              /* sq_ass_slice */
        0,                              /* sq_contains */
        0,                              /* sq_inplace_concat */
        0,                              /* sq_inplace_repeat */
    };
    return answer;
}
 
template <class T>
PyObject *StrIter::Py_next(PyObject *self) {
    T *it = (T *)self;
    if (debug) log_println(std::string("StrIter::Py_next - ") + demangle(typeid(T).name()));
    std::string res = it->val->next();
    if (it->val->done)
        return NULL;
    else
        return PyUnicode_FromString(res.c_str());
}
 
template <class T>
PyObject *StrIter::Py_repr(PyObject *self) {
    return PyUnicode_FromString(std::string("<").append(demangle(typeid(T).name())).append(" object>").c_str());
}
 
template <class T>
Py_ssize_t StrIter::Py_length(PyObject *self) {
    return ((T *)self)->val->length();
}
 
template <class T>
void StrIter::Py_dealloc(PyObject *self) {
    if (debug) log_println(std::string("StrIter::Py_dealloc - ") + demangle(typeid(T).name()));
    T *_self = (T *)self;
    delete _self->val;
    Py_TYPE(_self)->tp_free(self);
}
}  // namespace k1a