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
|
*********************************************
Decorating Functions to Accept NDData Objects
*********************************************
The `astropy.nddata` module includes a decorator
:func:`~astropy.nddata.support_nddata` that makes it convenient for developers
and users to write functions that can accept :class:`~astropy.nddata.NDData`
objects and also separate arguments.
Consider the following function::
def test(data, wcs=None, unit=None, n_iterations=3):
...
Now say that we want to be able to call the function as ``test(nd)``
where ``nd`` is an :class:`~astropy.nddata.NDData` instance. We can decorate
this function using :func:`~astropy.nddata.support_nddata`::
from astropy.nddata import support_nddata
@support_nddata
def test(data, wcs=None, unit=None, n_iterations=3):
...
Which makes it so that when the user calls ``test(nd)``, the function would
automatically be called with::
test(nd.data, wcs=nd.wcs, unit=nd.unit)
The decorator looks at the signature of the function and checks if any
of the arguments are also properties of the ``NDData`` object, and passes them
as individual arguments. The function can also be called with separate
arguments as if it was not decorated.
A warning is emitted if an ``NDData`` property is set but the function does
not accept it — for example, if ``wcs`` is set, but the function cannot support
WCS objects. On the other hand, if an argument in the function does not exist
in the ``NDData`` object or is not set, it is left to its default value.
If the function call succeeds, then the decorator returns the values from the
function unmodified by default. However, in some cases we may want to return
separate ``data``, ``wcs``, etc. if these were passed in separately, and a new
:class:`~astropy.nddata.NDData` instance otherwise. To do this, you can specify
``repack=True`` in the decorator and provide a list of the names of the output
arguments from the function::
@support_nddata(repack=True, returns=['data', 'wcs'])
def test(data, wcs=None, unit=None, n_iterations=3):
...
With this, the function will return separate values if ``test`` is called with
separate arguments, and an object with the same class type as the input if the
input is an :class:`~astropy.nddata.NDData` or subclass instance.
Finally, the decorator can be made to restrict input to specific ``NDData``
subclasses (and the subclasses of those) using the ``accepts`` option::
@support_nddata(accepts=CCDImage)
def test(data, wcs=None, unit=None, n_iterations=3):
...
|
