dupeguru/core/pe/modules/cache.c

/* Created By: Virgil Dupras
 * Created On: 2010-01-30
 * Copyright 2014 Hardcoded Software (http://www.hardcoded.net)
 *
 * This software is licensed under the "BSD" License as described in the "LICENSE" file, 
 * which should be included with this package. The terms are also available at 
 * http://www.hardcoded.net/licenses/bsd_license
 */

#include "common.h"

static PyObject*
cache_bytes_to_colors(PyObject *self, PyObject *args)
{
    char *y;
    Py_ssize_t char_count, i, color_count;
    PyObject *result;
    unsigned long r, g, b;
    Py_ssize_t ci;
    PyObject *color_tuple;

    if (!PyArg_ParseTuple(args, "y#", &y, &char_count)) {
        return NULL;
    }
    
    color_count = char_count / 3;
    result = PyList_New(color_count);
    if (result == NULL) {
        return NULL;
    }
    
    for (i=0; i<color_count; i++) {
        ci = i * 3;
        r = (unsigned char) y[ci];
        g = (unsigned char) y[ci+1];
        b = (unsigned char) y[ci+2];
        
        color_tuple = inttuple(3, r, g, b);
        if (color_tuple == NULL) {
            Py_DECREF(result);
            return NULL;
        }
        PyList_SET_ITEM(result, i, color_tuple);
    }
    
    return result;
}

static PyMethodDef CacheMethods[] = {
    {"bytes_to_colors",  cache_bytes_to_colors, METH_VARARGS, "Transform the bytes 's' into a list of 3 sized tuples."},
    {NULL, NULL, 0, NULL}        /* Sentinel */
};

static struct PyModuleDef CacheDef = {
    PyModuleDef_HEAD_INIT,
    "_cache",
    NULL,
    -1,
    CacheMethods,
    NULL,
    NULL,
    NULL,
    NULL
};

PyObject *
PyInit__cache(void)
{
    PyObject *m = PyModule_Create(&CacheDef);
    if (m == NULL) {
        return NULL;
    }
    return m;
}