28693295402

Committed 04 Jul 2026 03:17AM UTC coverage: 52.408% (+0.1%) from 52.278%

Build # 28693295402

Build Type

push

github

Committed by

hariharan-devarajan

Commit Message

feat: silence noisy warnings on aarch64

Coverage Stats

37318 of 92666 branches covered (40.27%)

Branch coverage included in aggregate %.

33462 of 42389 relevant lines covered (78.94%)

20557.64 hits per line

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

81.48

/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) {
    // dftracer.utils.arrow.ArrowTable accepts either a batch list or a stream
    // object, so wrapping is identical for both.
    return wrap_arrow_table(stream_obj);
}

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) {	44✔
12	if (!result.valid()) {	44!
13	PyErr_SetString(PyExc_RuntimeError,	×
14	"Cannot wrap invalid ArrowExportResult");
15	return NULL;	×
16	}
17
18	auto cap = (ArrowBatchCapsuleObject )ArrowBatchCapsuleType.tp_alloc(	44✔
19	&ArrowBatchCapsuleType, 0);
20	if (!cap) return NULL;	44✔
21
22	cap->result = new ArrowExportResult(std::move(result));	44!
23	return (PyObject *)cap;	44✔
24	}	22✔
25
26	PyObject wrap_arrow_table(PyObject batch_list) {	68✔
27	if (!batch_list) {	68✔
28	PyErr_SetString(PyExc_RuntimeError, "batch_list is NULL");	×
29	return NULL;	×
30	}
31
32	PyObject *mod = PyImport_ImportModule("dftracer.utils.arrow");	68✔
33	if (!mod) {	68✔
34	Py_DECREF(batch_list);
35	return NULL;	×
36	}
37
38	PyObject *cls = PyObject_GetAttrString(mod, "ArrowTable");	68✔
39	Py_DECREF(mod);	34✔
40	if (!cls) {	68✔
41	Py_DECREF(batch_list);
42	return NULL;	×
43	}
44
45	PyObject *table = PyObject_CallFunctionObjArgs(cls, batch_list, NULL);	68✔
46	Py_DECREF(cls);	34✔
47	Py_DECREF(batch_list);	34✔
48	return table;	68✔
49	}	34✔
50
51	PyObject wrap_arrow_stream_table(PyObject stream_obj) {	26✔
52	// dftracer.utils.arrow.ArrowTable accepts either a batch list or a stream
53	// object, so wrapping is identical for both.
54	return wrap_arrow_table(stream_obj);	26✔
55	}
56
57	PyObject *arrow_result_to_table(ArrowExportResult result) {	12✔
58	PyObject *capsule = wrap_arrow_result(std::move(result));	12!
59	if (!capsule) return NULL;	12✔
60
61	PyObject *list = PyList_New(1);	12✔
62	if (!list) {	12✔
63	Py_DECREF(capsule);
64	return NULL;	×
65	}
66	PyList_SET_ITEM(list, 0, capsule); // steals ref	12✔
67
68	return wrap_arrow_table(list);	12✔
69	}	6✔
70
71	} // namespace dftracer::utils::python
72
73	#endif // DFTRACER_UTILS_ENABLE_ARROW

llnl / dftracer-utils / 28693295402

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