Kratos/kratos/python/add_kratos_parameters_to_python.cpp at master · KratosMultiphysics/Kratos · GitHub

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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
//    |  /           |
//    ' /   __| _` | __|  _ \   __|
//    . \  |   (   | |   (   |\__ `
//   _|\_\_|  \__,_|\__|\___/ ____/
//                   Multi-Physics
//
//  License:         BSD License
//                   Kratos default license: kratos/license.txt
//
//  Main authors:    Riccardo Rossi
//

// System includes

// External includes

// Project includes
#include "includes/define_python.h"
#include "includes/kratos_parameters.h"
#include "add_kratos_parameters_to_python.h"

namespace Kratos::Python {

pybind11::list items(Parameters const& self)
{
    pybind11::list t;
    for(Parameters::const_iterator it=self.begin(); it!=self.end(); ++it)
        t.append( std::make_tuple(it.name(), *it) );
    return t;
}

pybind11::list keys(Parameters const& self)
{
    pybind11::list t;
    for(Parameters::const_iterator it=self.begin(); it!=self.end(); ++it)
        t.append(it.name());
    return t;
}

pybind11::list values(Parameters const& self)
{
    pybind11::list t;
    for(Parameters::const_iterator it=self.begin(); it!=self.end(); ++it)
        t.append(*it);
    return t;
}

template< class T >
void Append(Parameters &rParameters, const T& obj) {
    rParameters.Append(obj);
}

Parameters GetValue(Parameters &rParameters, const std::string& rEntry) {
    return rParameters.GetValue(rEntry);
}

Parameters GetArrayItem(Parameters &rParameters, const std::size_t Index) {
    return rParameters.GetArrayItem(Index);
}

void  AddKratosParametersToPython(pybind11::module& m)
{
    namespace py = pybind11;

    py::class_<Parameters, Parameters::Pointer >(m,"Parameters")
    .def(py::init<>())
    .def(py::init<const std::string&>())
    .def(py::init<std::ifstream&>())
    .def(py::init<Parameters const&>())
    .def("__copy__", [](const Parameters& self) {auto a = Kratos::make_shared<Parameters>(self); return a;})
    .def("__copy__", [](const Parameters& self, py::object memo) {auto a = Kratos::make_shared<Parameters>(self); return a;}) // used for the copy module
    .def("__deepcopy__", [](const Parameters& self) {auto a = Kratos::make_shared<Parameters>(); Parameters& r_a = *a;r_a = self; return a;})
    .def("__deepcopy__", [](const Parameters& self, py::object memo) {auto a = Kratos::make_shared<Parameters>(); Parameters& r_a = *a;r_a = self; return a;}) // used for the copy module
    .def("WriteJsonString", &Parameters::WriteJsonString)
    .def("PrettyPrintJsonString", &Parameters::PrettyPrintJsonString)
    .def("Has", &Parameters::Has)
    .def("Clone", &Parameters::Clone)
    .def("AddValue", &Parameters::AddValue)
    .def("AddEmptyValue", &Parameters::AddEmptyValue)
    .def("AddEmptyArray", &Parameters::AddEmptyArray)
    .def("RemoveValue", &Parameters::RemoveValue)
    .def("RemoveValues", &Parameters::RemoveValues)
    .def("ValidateAndAssignDefaults", &Parameters::ValidateAndAssignDefaults)
    .def("RecursivelyValidateAndAssignDefaults", &Parameters::RecursivelyValidateAndAssignDefaults)
    .def("AddMissingParameters", &Parameters::AddMissingParameters)
    .def("RecursivelyAddMissingParameters", &Parameters::RecursivelyAddMissingParameters)
    .def("ValidateDefaults", &Parameters::ValidateDefaults)
    .def("RecursivelyValidateDefaults", &Parameters::RecursivelyValidateDefaults)
    .def("IsEquivalentTo",&Parameters::IsEquivalentTo)
    .def("HasSameKeysAndTypeOfValuesAs",&Parameters::HasSameKeysAndTypeOfValuesAs)
    //.def("GetValue", GetValue) //Do not export this method. users shall adopt the operator [] syntax
    .def("IsNull", &Parameters::IsNull)
    .def("IsNumber", &Parameters::IsNumber)
    .def("IsDouble", &Parameters::IsDouble)
    .def("IsInt", &Parameters::IsInt)
    .def("IsBool", &Parameters::IsBool)
    .def("IsString", &Parameters::IsString)
    .def("IsStringArray", &Parameters::IsStringArray)
    .def("IsArray", &Parameters::IsArray)
    .def("IsVector", &Parameters::IsVector)
    .def("IsMatrix", &Parameters::IsMatrix)
    .def("IsSubParameter", &Parameters::IsSubParameter)
    .def("GetDouble", &Parameters::GetDouble)
    .def("GetInt", &Parameters::GetInt)
    .def("GetBool", &Parameters::GetBool)
    .def("GetString", &Parameters::GetString)
    .def("GetStringArray", &Parameters::GetStringArray)
    .def("GetVector", &Parameters::GetVector)
    .def("GetMatrix", &Parameters::GetMatrix)
    .def("SetDouble", &Parameters::SetDouble)
    .def("SetInt", &Parameters::SetInt)
    .def("SetBool", &Parameters::SetBool)
    .def("SetString", &Parameters::SetString)
    .def("SetStringArray", &Parameters::SetStringArray)
    .def("SetVector", &Parameters::SetVector)
    .def("SetMatrix", &Parameters::SetMatrix)
    .def("AddDouble", &Parameters::AddDouble)
    .def("AddInt", &Parameters::AddInt)
    .def("AddBool", &Parameters::AddBool)
    .def("AddString", &Parameters::AddString)
    .def("AddStringArray", &Parameters::AddStringArray)
    .def("AddVector", &Parameters::AddVector)
    .def("AddMatrix", &Parameters::AddMatrix)
    .def("size", &Parameters::size)
    //.def("GetArrayItem", GetArrayItem) //Do not export this method. users shall adopt the operator [] syntax
    .def("__setitem__", &Parameters::SetValue)
    .def("__getitem__", GetValue)
    .def("__setitem__", &Parameters::SetArrayItem)
    .def("__getitem__", GetArrayItem)
    .def("__iter__", [](Parameters& self){
            KRATOS_ERROR << "iterating through Parameters is ambiguous. Use \"keys\", \"values\" or \"items\" instead.";
            return py::make_iterator(self.begin(), self.end());
        } , py::keep_alive<0,1>())
    .def("items", &items )
    .def("keys", &keys )
    .def("values", &values )
    .def("__str__", PrintObject<Parameters>)
    .def("AddEmptyList", &Parameters::AddEmptyArray)
    .def("Append", Append<int>) // created due to ambiguous overload int/bool...
    .def("Append", Append<bool>) // created due to ambiguous overload int/bool...
    .def("Append", Append<double>) // created due to ambiguous overload int/bool...
    .def("Append", Append<Vector>) // created due to ambiguous overload int/bool...
    .def("Append", Append<Matrix>) // created due to ambiguous overload int/bool...
    .def("Append", Append<std::string>) // created due to ambiguous overload int/bool...
    .def("Append", Append<Parameters>) // created due to ambiguous overload int/bool...
    .def("CopyValuesFromExistingParameters", &Parameters::CopyValuesFromExistingParameters)
    ;
}

} // namespace Kratos::Python.