#define NPY_NO_
DEPRECATED_API NPY_1_7_API_VERSION
#define PY_SSIZE_T_CLEAN
#include "eiger_defs.h"
#include "raw_file.h"
#include <Python.h>
#include <numpy/arrayobject.h>
#include <stdio.h>
#include <un
istd.h>
// #define VERBOSE
/* Docstrings */
static char module_docstring[] =
"sls_cmodule provies access to compiled
c++ and ROOT functions supporting "
"detector calibration.
The library uses numpy arrays as input and return "
"values.";
/* Available functions */
static PyObject *read_raw(PyObject *self, PyObject *args);
static PyObject *mt_read(PyObject *self, PyObject *args);
static PyObject *read_m3(PyObject *self, PyObject *args, PyObject *kwargs);
static PyArray_Descr *get_frame_header_dt();
/* Module specification */
static PyMethodDef module_methods[] = {
{"read_frame", (PyCFunction)read_raw, METH_VARARGS, "hej"},
{"mt_read", (PyCFunction)mt_read, METH_VARARGS, "hej"},
{"read_m3", (PyCFunction)read_m3, METH_VARARGS | METH_KEYWORDS, "hej"},
{NULL, NULL, 0, NULL}};
static struct PyModuleDef hdf5maker_def = {
PyModuleDef_HEAD_INIT, "_hdf5maker", module_docstring, -1, module_methods};
/* Initialize
the module */
PyMODINIT_FUNC PyInit__hdf5maker(void) {
PyObject *m = PyModule_Create(&hdf5maker_def);
if (m == NULL) {
return NULL;
}
/* Load `numpy` functionality. */
import_array();
PyArray_Descr *obj = get_frame_header_dt();
PyObject_SetAttrString(m, "frame_header_dt", (PyObject *)obj);
Py_DECREF(obj);
return m;
}
static PyObject *mt_read(PyObject *self, PyObject *args) { return NULL; }
static PyObject *read_raw(PyObject *self, PyObject *args) {
//Passing in
:
// file, dr, n_frames
PyObject *fobj;
int n_frames, dr;
if (!PyArg_ParseTuple(args, "OII", &fobj, &dr, &n_frames))
return NULL;
int fd = PyObject_AsFileDescriptor(fobj);
if (fd < 0)
return NULL;
// Allocate data
for header and frames
PyArray_Descr *dtype = get_frame_header_dt();
const npy_intp header_dims[] = {n_frames};
PyObject *header = PyArray_SimpleNewFromDescr(1, header_dims, dtype);
npy_intp dims[] = {n_frames, PORT_NROWS_WGAP, PORT_NCOLS_WGAP};
PyObject *data = PyArray_SimpleNew(3, dims, dr_to_dtype(dr));
const size_t bytes_to_copy = dr * PORT_NROWS * PORT_NCOLS / 8;
sls_detector_header *h_ptr =
(sls_detector_header *)PyArray_BYTES((PyArrayObject *)header);
uint8_t *frame_ptr = (uint8_t *)PyArray_BYTES((PyArrayObject *)data);
PyArray_FILLWBYTE((PyArrayObject *)data, 0);
const int stride = PyArray_STRIDE((PyArrayObject *)data, 0);
uint8_t *buffer = malloc(bytes_to_copy + sizeof(sls_detector_header));
uint8_t *expanded_buffer = NULL;
if (dr == 4) {
expanded_buffer = malloc(bytes_to_copy * 2);
}
for (int i = 0; i < n_frames; ++i) {
read(fd, buffer, bytes_to_copy + sizeof(sls_detector_header));
memcpy(h_ptr, buffer, sizeof(sls_detector_header));
if (dr == 4) {
uint8_t *dst = expanded_buffer;
uint8_t *src = buffer + sizeof(sls_detector_header);
for (size_t j = 0; j < bytes_to_copy; ++j) {
*dst++ = *src & 0xfu;
*dst++ = *sr
c++ >> 4u;
}
copy_to_place(frame_ptr, expanded_buffer, 8, h_ptr->row,
h_ptr->column);
} else {
copy_to_place(frame_ptr, buffer + sizeof(sls_detector_header), dr,
h_ptr->row, h_ptr->column);
}
frame_ptr += stride;
h_ptr++;
}
free(buffer);
free(expanded_buffer);
PyObject *ret = PyTuple_Pack(2, header, data);
Py_DECREF(header);
Py_DECREF(data);
return ret;
}
static PyObject *read_m3(PyObject *self, PyObject *args, PyObject *kwargs) {
PyObject *fobj;
static char *keywords[] = {"fd","dr","n_frames", "n_chan", NULL};
int n_frames, dr, n_chan;
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OIII", keywords, &fobj, &dr, &n_frames, &n_chan)) {
return NULL;
}
// if (!PyArg_ParseTuple(args, "OIII", &fobj, &dr, &n_frames, &n_chan))
// return NULL;
int fd = PyObject_AsFileDescriptor(fobj);
if (fd < 0)
return NULL;
// Allocate data
for header and frames
PyArray_Descr *dtype = get_frame_header_dt();
const npy_intp header_dims[] = {n_frames};
PyObject *header = PyArray_SimpleNewFromDescr(1, header_dims, dtype);
npy_intp dims[] = {n_frames, n_chan};
PyObject *data = PyArray_SimpleNew(2, dims, dr_to_dtype(dr));
const size_t bytes_to_copy = dr * n_chan / 8;
sls_detector_header *h_ptr =
(sls_detector_header *)PyArray_BYTES((PyArrayObject *)header);
uint8_t *frame_ptr = (uint8_t *)PyArray_BYTES((PyArrayObject *)data);
PyArray_FILLWBYTE((PyArrayObject *)data, 0);
const int stride = PyArray_STRIDE((PyArrayObject *)data, 0);
uint8_t *buffer = malloc(bytes_to_copy + sizeof(sls_detector_header));
uint8_t *expanded_buffer = NULL;
if (dr == 4) {
expanded_buffer = malloc(bytes_to_copy * 2);
}
for (int i = 0; i < n_frames; ++i) {
read(fd, buffer, bytes_to_copy + sizeof(sls_detector_header));
memcpy(h_ptr, buffer, sizeof(sls_detector_header));
memcpy(frame_ptr, buffer + sizeof(sls_detector_header), bytes_to_copy);
frame_ptr += stride;
h_ptr++;
}
free(buffer);
free(expanded_buffer);
PyObject *ret = PyTuple_Pack(2, header, data);
Py_DECREF(header);
Py_DECREF(data);
return ret;
}
static PyArray_Descr *get_frame_header_dt() {
// Move
this to a function that gets run on init
PyObject *dtype_dict;
PyArray_Descr *dtype;
dtype_dict = Py_BuildValue(
"[(s, s), (s, s), (s, s), (s, s), (s, s), (s, s), (s, s), (s, s), (s, "
"s), (s, s), (s, s), (s, s), (s, s), (s, s)]",
"Frame Number", "u8", "SubFrame Number/ExpLength", "u4",
"Packet Number", "u4", "Bunch ID", "u8", "Timestamp", "u8", "Module Id",
"u2", "Row", "u2", "Column", "u2", "Reserved", "u2", "Debug", "u4",
"Round Robin Number", "u2", "Detector Type", "u1", "Header Version",
"u1", "Packets Caught Mask", "V64");
PyArray_DescrConverter(dtype_dict, &dtype);
Py_DECREF(dtype_dict);
return dtype;
}
本文介绍如何通过在common.cpp文件的预处理器定义中添加特定宏，来消除C/C++编译时出现的关于安全性和非标准函数的警告信息。具体步骤包括：在预处理器定义中添加_CRT_SECURE_NO_DEPRECATE和_CRT_NONSTDC_NO_DEPRECATE宏。
扫一扫
2万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
