File: PyObjectPtr.cpp

package info (click to toggle)
bornagain 23.0-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 103,936 kB
  • sloc: cpp: 423,131; python: 40,997; javascript: 11,167; awk: 630; sh: 318; ruby: 173; xml: 130; makefile: 51; ansic: 24
file content (76 lines) | stat: -rw-r--r-- 2,073 bytes parent folder | download | duplicates (2)
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
//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      PyCore/Embed/PyObjectPtr.cpp
//! @brief     Implements class PyObjectPtr.
//!
//! @homepage  http://www.bornagainproject.org
//! @license   GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors   Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
//  ************************************************************************************************

#include "PyObjectPtr.h"
#include "PyInterpreter.h"

PyObjectPtr::PyObjectPtr(PyObjectPtr&& other)
    : PyObjectPtr(other.release())
{
}
PyObject* PyObjectPtr::release()
{
    PyObject* pyobject_ptr{m_ptr};
    reset();
    return pyobject_ptr;
}

void PyObjectPtr::reset()
{
    m_ptr = nullptr;
}

void PyObjectPtr::discard()
{
    if (m_borrowed) {
        // borrowed reference need no reference decrement
        reset();
        return;
    }

    PyInterpreter::DecRef(m_ptr);
    reset();
}


NumpyArrayWrapper::NumpyArrayWrapper(const std::size_t datasize, const std::size_t n_dims,
                                     const std::size_t* dims, const std::string& dtype,
                                     const void* array_ptr, PyObject* py_array_ptr,
                                     const ArrayMetadata& metadata)
    : m_datasize{datasize}
    , m_dtype{dtype}
    , m_metadata{metadata}
    , m_array_ptr{array_ptr}
    , m_py_array_ptr{py_array_ptr}
{
    // check validity
    if (m_datasize < 1 || n_dims < 1 || !dims || !m_array_ptr || !m_py_array_ptr)
        return;

    for (std::size_t i_d = 0; i_d < n_dims; ++i_d) {
        if (dims[i_d] < 1)
            return;
    }

    // initialize the array descriptor
    m_dims.resize(n_dims);
    for (std::size_t i_d = 0; i_d < n_dims; ++i_d)
        m_dims[i_d] = dims[i_d];
}

NumpyArrayWrapper::~NumpyArrayWrapper()
{
    if (m_py_array_ptr)
        PyInterpreter::DecRef(m_py_array_ptr);
}