27052412546

Committed 06 Jun 2026 04:20AM UTC coverage: 50.862% (+1.0%) from 49.905%

Build # 27052412546

Build Type

Pull #73

github

Committed by

web-flow

Commit Message

Merge 734572730 into 88a3c8457

Pull Request Pull Request #73: add portable dependencies wheel support

Coverage Stats

31801 of 79859 branches covered (39.82%)

Branch coverage included in aggregate %.

32491 of 46545 relevant lines covered (69.81%)

9947.11 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

64.56

/src/dftracer/utils/python/arrow_helpers.cpp

#include <dftracer/utils/core/common/config.h>
#ifdef DFTRACER_UTILS_ENABLE_ARROW

#define PY_SSIZE_T_CLEAN
#include <Python.h>
#include <dftracer/utils/python/arrow_helpers.h>
#include <dftracer/utils/python/trace_reader_iterator.h>

namespace dftracer::utils::python {

PyObject *wrap_arrow_result(ArrowExportResult result) {
    if (!result.valid()) {
        PyErr_SetString(PyExc_RuntimeError,
                        "Cannot wrap invalid ArrowExportResult");
        return NULL;
    }

    auto *cap = (ArrowBatchCapsuleObject *)ArrowBatchCapsuleType.tp_alloc(
        &ArrowBatchCapsuleType, 0);
    if (!cap) return NULL;

    cap->result = new ArrowExportResult(std::move(result));
    return (PyObject *)cap;
}

PyObject *wrap_arrow_table(PyObject *batch_list) {
    if (!batch_list) {
        PyErr_SetString(PyExc_RuntimeError, "batch_list is NULL");
        return NULL;
    }

    PyObject *mod = PyImport_ImportModule("dftracer.utils.arrow");
    if (!mod) {
        Py_DECREF(batch_list);
        return NULL;
    }

    PyObject *cls = PyObject_GetAttrString(mod, "ArrowTable");
    Py_DECREF(mod);
    if (!cls) {
        Py_DECREF(batch_list);
        return NULL;
    }

    PyObject *table = PyObject_CallFunctionObjArgs(cls, batch_list, NULL);
    Py_DECREF(cls);
    Py_DECREF(batch_list);
    return table;
}

PyObject *wrap_arrow_stream_table(PyObject *stream_obj) {
    if (!stream_obj) {
        PyErr_SetString(PyExc_RuntimeError, "stream_obj is NULL");
        return NULL;
    }

    PyObject *mod = PyImport_ImportModule("dftracer.utils.arrow");
    if (!mod) {
        Py_DECREF(stream_obj);
        return NULL;
    }

    PyObject *cls = PyObject_GetAttrString(mod, "ArrowTable");
    Py_DECREF(mod);
    if (!cls) {
        Py_DECREF(stream_obj);
        return NULL;
    }

    PyObject *table = PyObject_CallFunctionObjArgs(cls, stream_obj, NULL);
    Py_DECREF(cls);
    Py_DECREF(stream_obj);
    return table;
}

PyObject *arrow_result_to_table(ArrowExportResult result) {
    PyObject *capsule = wrap_arrow_result(std::move(result));
    if (!capsule) return NULL;

    PyObject *list = PyList_New(1);
    if (!list) {
        Py_DECREF(capsule);
        return NULL;
    }
    PyList_SET_ITEM(list, 0, capsule);  // steals ref

    return wrap_arrow_table(list);
}

}  // namespace dftracer::utils::python

#endif  // DFTRACER_UTILS_ENABLE_ARROW

1	#include <dftracer/utils/core/common/config.h>
2	#ifdef DFTRACER_UTILS_ENABLE_ARROW
3
4	#define PY_SSIZE_T_CLEAN
5	#include <Python.h>
6	#include <dftracer/utils/python/arrow_helpers.h>
7	#include <dftracer/utils/python/trace_reader_iterator.h>
8
9	namespace dftracer::utils::python {
10
11	PyObject *wrap_arrow_result(ArrowExportResult result) {	22✔
12	if (!result.valid()) {	22!
13	PyErr_SetString(PyExc_RuntimeError,	×
14	"Cannot wrap invalid ArrowExportResult");
15	return NULL;	×
16	}
17
18	auto cap = (ArrowBatchCapsuleObject )ArrowBatchCapsuleType.tp_alloc(	22✔
19	&ArrowBatchCapsuleType, 0);
20	if (!cap) return NULL;	22!
21
22	cap->result = new ArrowExportResult(std::move(result));	22!
23	return (PyObject *)cap;	22✔
24	}	22✔
25
26	PyObject wrap_arrow_table(PyObject batch_list) {	21✔
27	if (!batch_list) {	21!
28	PyErr_SetString(PyExc_RuntimeError, "batch_list is NULL");	×
29	return NULL;	×
30	}
31
32	PyObject *mod = PyImport_ImportModule("dftracer.utils.arrow");	21✔
33	if (!mod) {	21!
34	Py_DECREF(batch_list);	×
35	return NULL;	×
36	}
37
38	PyObject *cls = PyObject_GetAttrString(mod, "ArrowTable");	21✔
39	Py_DECREF(mod);	21✔
40	if (!cls) {	21!
41	Py_DECREF(batch_list);	×
42	return NULL;	×
43	}
44
45	PyObject *table = PyObject_CallFunctionObjArgs(cls, batch_list, NULL);	21✔
46	Py_DECREF(cls);	21✔
47	Py_DECREF(batch_list);	21✔
48	return table;	21✔
49	}	21✔
50
51	PyObject wrap_arrow_stream_table(PyObject stream_obj) {	13✔
52	if (!stream_obj) {	13!
53	PyErr_SetString(PyExc_RuntimeError, "stream_obj is NULL");	×
54	return NULL;	×
55	}
56
57	PyObject *mod = PyImport_ImportModule("dftracer.utils.arrow");	13✔
58	if (!mod) {	13!
59	Py_DECREF(stream_obj);	×
60	return NULL;	×
61	}
62
63	PyObject *cls = PyObject_GetAttrString(mod, "ArrowTable");	13✔
64	Py_DECREF(mod);	13✔
65	if (!cls) {	13!
66	Py_DECREF(stream_obj);	×
67	return NULL;	×
68	}
69
70	PyObject *table = PyObject_CallFunctionObjArgs(cls, stream_obj, NULL);	13✔
71	Py_DECREF(cls);	13✔
72	Py_DECREF(stream_obj);	13✔
73	return table;	13✔
74	}	13✔
75
76	PyObject *arrow_result_to_table(ArrowExportResult result) {	6✔
77	PyObject *capsule = wrap_arrow_result(std::move(result));	6!
78	if (!capsule) return NULL;	6!
79
80	PyObject *list = PyList_New(1);	6✔
81	if (!list) {	6!
82	Py_DECREF(capsule);	×
83	return NULL;	×
84	}
85	PyList_SET_ITEM(list, 0, capsule); // steals ref	6✔
86
87	return wrap_arrow_table(list);	6✔
88	}	6✔
89
90	} // namespace dftracer::utils::python
91
92	#endif // DFTRACER_UTILS_ENABLE_ARROW

llnl / dftracer-utils / 27052412546

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous