mirror of
https://github.com/NGSolve/netgen.git
synced 2024-09-23 02:02:49 +05:00
120 lines
3.5 KiB
C++
120 lines
3.5 KiB
C++
#include <pybind11/pybind11.h>
|
|
|
|
#include "python_ngcore.hpp"
|
|
|
|
#include "array.hpp"
|
|
#include "flags.hpp"
|
|
#include "logging.hpp"
|
|
|
|
namespace py = pybind11;
|
|
using std::string;
|
|
|
|
namespace ngcore
|
|
{
|
|
|
|
void SetFlag(Flags &flags, string s, py::object value)
|
|
{
|
|
if (py::isinstance<py::dict>(value))
|
|
{
|
|
py::dict vdd(value);
|
|
// call recursively to set dictionary
|
|
for (auto item : vdd) {
|
|
string name = item.first.cast<string>();
|
|
py::object val = py::reinterpret_borrow<py::object>(item.second);
|
|
SetFlag(flags, name, val);
|
|
}
|
|
return;
|
|
}
|
|
|
|
if (py::isinstance<py::bool_>(value))
|
|
flags.SetFlag(s, value.cast<bool>());
|
|
|
|
if (py::isinstance<py::float_>(value))
|
|
flags.SetFlag(s, value.cast<double>());
|
|
|
|
if (py::isinstance<py::int_>(value))
|
|
flags.SetFlag(s, double(value.cast<int>()));
|
|
|
|
if (py::isinstance<py::str>(value))
|
|
flags.SetFlag(s, value.cast<string>());
|
|
|
|
if (py::isinstance<py::list>(value))
|
|
{
|
|
py::list vdl(value);
|
|
if (py::len(vdl) > 0)
|
|
{
|
|
if(py::isinstance<double>(vdl[0]))
|
|
flags.SetFlag(s, makeCArray<double>(vdl));
|
|
if(py::isinstance<py::str>(vdl[0]))
|
|
flags.SetFlag(s, makeCArray<string>(vdl));
|
|
}
|
|
else
|
|
{
|
|
Array<string> dummystr;
|
|
Array<double> dummydbl;
|
|
flags.SetFlag(s,dummystr);
|
|
flags.SetFlag(s,dummydbl);
|
|
}
|
|
}
|
|
|
|
if (py::isinstance<py::tuple>(value))
|
|
{
|
|
py::tuple vdt(value);
|
|
if (py::isinstance<py::float_>(value))
|
|
flags.SetFlag(s, makeCArray<double>(vdt));
|
|
if (py::isinstance<py::int_>(value))
|
|
flags.SetFlag(s, makeCArray<double>(vdt));
|
|
if (py::isinstance<py::str>(value))
|
|
flags.SetFlag(s, makeCArray<string>(vdt));
|
|
}
|
|
}
|
|
|
|
Flags CreateFlagsFromKwArgs(py::object pyclass, const py::kwargs& kwargs, py::list info)
|
|
{
|
|
static std::shared_ptr<Logger> logger = GetLogger("Flags");
|
|
auto flags_doc = pyclass.attr("__flags_doc__")();
|
|
py::dict flags_dict;
|
|
|
|
if (kwargs.contains("flags"))
|
|
{
|
|
logger->warn("WARNING: using flags as kwarg is deprecated in {}, use the flag arguments as kwargs instead!",
|
|
std::string(py::str(pyclass)));
|
|
auto addflags = py::cast<py::dict>(kwargs["flags"]);
|
|
for (auto item : addflags)
|
|
flags_dict[item.first.cast<string>().c_str()] = item.second;
|
|
}
|
|
for (auto item : kwargs)
|
|
if (!flags_doc.contains(item.first.cast<string>().c_str()) &&
|
|
!(item.first.cast<string>() == "flags"))
|
|
logger->warn("WARNING: kwarg '{}' is an undocumented flags option for class {}, maybe there is a typo?",
|
|
item.first.cast<string>(), std::string(py::str(pyclass)));
|
|
|
|
py::dict special;
|
|
if(py::hasattr(pyclass,"__special_treated_flags__"))
|
|
special = pyclass.attr("__special_treated_flags__")();
|
|
for (auto item : kwargs)
|
|
{
|
|
auto name = item.first.cast<string>();
|
|
if (name != "flags")
|
|
{
|
|
if(!special.contains(name.c_str()))
|
|
flags_dict[name.c_str()] = item.second;
|
|
}
|
|
}
|
|
|
|
auto flags = py::cast<Flags>(flags_dict);
|
|
|
|
for (auto item : kwargs)
|
|
{
|
|
auto name = item.first.cast<string>();
|
|
if (name != "flags")
|
|
{
|
|
if(special.contains(name.c_str()))
|
|
special[name.c_str()](item.second, &flags, info);
|
|
}
|
|
}
|
|
return flags;
|
|
}
|
|
|
|
} // namespace ngcore
|