1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
/*
* SPDX-FileCopyrightText: All Contributors to the PyTango project
*
* SPDX-License-Identifier: LGPL-3.0-or-later
*/
// This header file is just some template functions moved apart from
// wattribute.cpp, and should only be included there.
#pragma once
#include "tango_numpy.h"
namespace PyWAttribute
{
template <long tangoTypeConst>
void __get_write_value_array_numpy(Tango::WAttribute &att, bopy::object *obj)
{
typedef typename TANGO_const2type(tangoTypeConst) TangoScalarType;
const TangoScalarType *buffer;
att.get_write_value(buffer);
size_t length = att.get_write_value_length();
// Copy buffer in a python raw buffer
const char *original_ch_buffer = reinterpret_cast<const char *>(buffer);
PyObject *str_guard = PyBytes_FromStringAndSize(original_ch_buffer, length * sizeof(TangoScalarType));
if(!str_guard)
{
bopy::throw_error_already_set();
}
// Create a numpy object based on it...
static const int typenum = TANGO_const2numpy(tangoTypeConst);
npy_intp dims[2];
int nd = 1;
char *ch_buffer = PyBytes_AsString(str_guard);
if(att.get_data_format() == Tango::IMAGE)
{
nd = 2;
dims[1] = att.get_w_dim_x();
dims[0] = att.get_w_dim_y();
}
else
{
nd = 1;
dims[0] = att.get_w_dim_x();
}
PyObject *array = PyArray_SimpleNewFromData(nd, dims, typenum, ch_buffer);
if(!array)
{
Py_XDECREF(str_guard);
bopy::throw_error_already_set();
}
PyArray_SetBaseObject(to_PyArrayObject(array), str_guard);
*obj = bopy::object(bopy::handle<>(array));
}
template <>
void __get_write_value_array_numpy<Tango::DEV_STRING>(Tango::WAttribute &att, bopy::object *obj)
{
__get_write_value_array_lists<Tango::DEV_STRING>(att, obj);
}
template <>
void __get_write_value_array_numpy<Tango::DEV_ENCODED>(Tango::WAttribute &att, bopy::object *obj)
{
__get_write_value_array_lists<Tango::DEV_STRING>(att, obj);
}
} // namespace PyWAttribute
|
