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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
//# PycArray.h: Class to convert an Array to/from Python
//# Copyright (C) 2006
//# Associated Universities, Inc. Washington DC, USA.
//#
//# This library is free software; you can redistribute it and/or modify it
//# under the terms of the GNU Library General Public License as published by
//# the Free Software Foundation; either version 2 of the License, or (at your
//# option) any later version.
//#
//# This library is distributed in the hope that it will be useful, but WITHOUT
//# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
//# FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public
//# License for more details.
//#
//# You should have received a copy of the GNU Library General Public License
//# along with this library; if not, write to the Free Software Foundation,
//# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
//#
//# Correspondence concerning AIPS++ should be addressed as follows:
//# Internet email: casa-feedback@nrao.edu.
//# Postal address: AIPS++ Project Office
//# National Radio Astronomy Observatory
//# 520 Edgemont Road
//# Charlottesville, VA 22903-2475 USA
#ifndef PYRAP_PYCARRAY_H
#define PYRAP_PYCARRAY_H
//# Includes
// include first to avoid _POSIX_C_SOURCE redefined warnings
#include <boost/python.hpp>
#include <boost/python/object.hpp>
#include <casacore/casa/Arrays/Array.h>
#include <casacore/casa/Containers/ValueHolder.h>
#include <casacore/casa/Utilities/DataType.h>
#include <casacore/casa/Exceptions/Error.h>
#include <iostream>
namespace casacore { namespace python {
// <summary>
// A class to convert an Array to/from Python objects.
// </summary>
// <use visibility=export>
// <reviewed reviewer="" date="" tests="">
// </reviewed>
// <synopsis>
// </synopsis>
// Check if the PyObject is an array object.
Bool PycArrayCheck (PyObject* obj_ptr);
// Check if the PyObject is an array scalar object.
Bool PycArrayScalarCheck (PyObject* obj_ptr);
// Get the data type of the array scalar object.
// It returns TpBool, TpInt, TpFloat, or TpComplex.
// TpOther is returned if unrecognized.
DataType PycArrayScalarType (PyObject* obj_ptr);
struct casa_array_from_python
{
// Constructs an Array from a Python object.
// If copyData=False, the array data is only copied if needed meaning
// that the Array object in the ValueHolder can reference the data in
// Python array.
// That should only be used if the ValueHolder and its Array will be
// destructed before the Python array.
static ValueHolder makeArray(PyObject* obj_ptr, Bool copyData=False);
// Construct an Array<String> from a special Python dict object.
static ValueHolder makeArrayFromDict (PyObject* obj_ptr);
// Construct a scalar from an array scalar (i.e. element in array).
static ValueHolder makeScalar (PyObject* obj_ptr);
};
// Do the actual making of the PyArrayObject.
// Specialize for strings.
// <group>
template <typename T>
boost::python::object makePyArrayObject (casacore::Array<T> const& arr);
template <>
boost::python::object makePyArrayObject (casacore::Array<String> const& arr);
// </group>
// Convert Array to Python.
template <typename T>
struct casa_array_to_python
{
static boost::python::object makeobject (Array<T> const& arr)
{ return makePyArrayObject (arr); }
static PyObject* convert (Array<T> const& c)
{ return boost::python::incref(makeobject(c).ptr()); }
};
}}
#endif
|
