24 #include <pybind11/pybind11.h>
25 #include <pybind11/stl.h>
28 using namespace pybind11::literals;
47 class FrameDictMaker {
49 FrameDictMaker() =
default;
50 ~FrameDictMaker() =
default;
54 auto objPtr = chan.read();
55 auto frameSetPtr = std::dynamic_pointer_cast<ast::FrameSet>(objPtr);
60 return std::make_shared<ast::FrameDict>(*frameSetPtr);
65 py::module::import(
"astshim.frameSet");
67 py::class_<FrameDictMaker, std::shared_ptr<FrameDictMaker>> makerCls(mod,
"FrameDictMaker");
68 makerCls.def(py::init<>());
69 makerCls.def(
"__call__", &FrameDictMaker::operator());
70 makerCls.def(
"__reduce__",
71 [makerCls](FrameDictMaker
const &
self) {
return py::make_tuple(makerCls, py::tuple()); });
73 py::class_<FrameDict, std::shared_ptr<FrameDict>, FrameSet> cls(mod,
"FrameDict");
75 cls.def(py::init<Frame const &, std::string const &>(),
"frame"_a,
"options"_a =
"");
76 cls.def(py::init<Frame const &, Mapping const &, Frame const &, std::string const &>(),
"baseFrame"_a,
77 "mapping"_a,
"currentFrame"_a,
"options"_a =
"");
78 cls.def(py::init<FrameSet const &>(),
"frameSet"_a);
79 cls.def(py::init<FrameDict const &>());
81 cls.def(
"copy", &FrameDict::copy);
83 cls.def(
"addFrame", py::overload_cast<int, Mapping const &, Frame const &>(&FrameDict::addFrame),
84 "iframe"_a,
"map"_a,
"frame"_a);
86 py::overload_cast<std::string const &, Mapping const &, Frame const &>(&FrameDict::addFrame),
87 "domain"_a,
"map"_a,
"frame"_a);
88 cls.def(
"getAllDomains", &FrameDict::getAllDomains);
89 cls.def(
"getFrame", py::overload_cast<int, bool>(&
FrameDict::getFrame, py::const_),
"index"_a,
91 cls.def(
"getFrame", py::overload_cast<std::string const &, bool>(&
FrameDict::getFrame, py::const_),
92 "domain"_a,
"copy"_a =
true);
93 cls.def(
"getMapping", py::overload_cast<int, int>(&FrameDict::getMapping, py::const_),
94 "from"_a = FrameDict::BASE,
"to"_a = FrameDict::CURRENT);
95 cls.def(
"getMapping", py::overload_cast<int, std::string const &>(&FrameDict::getMapping, py::const_),
96 "from"_a = FrameDict::BASE,
"to"_a = FrameDict::CURRENT);
97 cls.def(
"getMapping", py::overload_cast<std::string const &, int>(&FrameDict::getMapping, py::const_),
98 "from"_a = FrameDict::BASE,
"to"_a = FrameDict::CURRENT);
100 py::overload_cast<std::string const &, std::string const &>(&FrameDict::getMapping, py::const_),
101 "from"_a = FrameDict::BASE,
"to"_a = FrameDict::CURRENT);
102 cls.def(
"getIndex", &FrameDict::getIndex,
"domain"_a);
103 cls.def(
"hasDomain", &FrameDict::hasDomain,
"domain"_a);
104 cls.def(
"mirrorVariants", py::overload_cast<int>(&FrameDict::mirrorVariants),
"index"_a);
105 cls.def(
"mirrorVariants", py::overload_cast<std::string const &>(&FrameDict::mirrorVariants),
"domain"_a);
106 cls.def(
"remapFrame", py::overload_cast<int, Mapping &>(&FrameDict::remapFrame),
"index"_a,
"map"_a);
107 cls.def(
"remapFrame", py::overload_cast<std::string const &, Mapping &>(&FrameDict::remapFrame),
108 "domain"_a,
"map"_a);
109 cls.def(
"removeFrame", py::overload_cast<int>(&FrameDict::removeFrame),
"index"_a);
110 cls.def(
"removeFrame", py::overload_cast<std::string const &>(&FrameDict::removeFrame),
"domain"_a);
111 cls.def(
"setBase", py::overload_cast<int>(&FrameDict::setBase),
"index"_a);
112 cls.def(
"setBase", py::overload_cast<std::string const &>(&FrameDict::setBase),
"domain"_a);
113 cls.def(
"setCurrent", py::overload_cast<int>(&FrameDict::setCurrent),
"index"_a);
114 cls.def(
"setCurrent", py::overload_cast<std::string const &>(&FrameDict::setCurrent),
"domain"_a);
115 cls.def(
"setDomain", &FrameDict::setDomain,
"domain"_a);
118 cls.def(
"__reduce__", [makerCls](Object
const &
self) {
120 auto unpickleArgs = py::make_tuple(state);
121 return py::make_tuple(makerCls(), unpickleArgs);
Channel provides input/output of AST objects.
String-based source and sink for channels.
AST wrapper classes and functions.
PYBIND11_MODULE(_cameraGeom, mod)