LSST Applications  21.0.0+04719a4bac,21.0.0-1-ga51b5d4+f5e6047307,21.0.0-11-g2b59f77+a9c1acf22d,21.0.0-11-ga42c5b2+86977b0b17,21.0.0-12-gf4ce030+76814010d2,21.0.0-13-g1721dae+760e7a6536,21.0.0-13-g3a573fe+768d78a30a,21.0.0-15-g5a7caf0+f21cbc5713,21.0.0-16-g0fb55c1+b60e2d390c,21.0.0-19-g4cded4ca+71a93a33c0,21.0.0-2-g103fe59+bb20972958,21.0.0-2-g45278ab+04719a4bac,21.0.0-2-g5242d73+3ad5d60fb1,21.0.0-2-g7f82c8f+8babb168e8,21.0.0-2-g8f08a60+06509c8b61,21.0.0-2-g8faa9b5+616205b9df,21.0.0-2-ga326454+8babb168e8,21.0.0-2-gde069b7+5e4aea9c2f,21.0.0-2-gecfae73+1d3a86e577,21.0.0-2-gfc62afb+3ad5d60fb1,21.0.0-25-g1d57be3cd+e73869a214,21.0.0-3-g357aad2+ed88757d29,21.0.0-3-g4a4ce7f+3ad5d60fb1,21.0.0-3-g4be5c26+3ad5d60fb1,21.0.0-3-g65f322c+e0b24896a3,21.0.0-3-g7d9da8d+616205b9df,21.0.0-3-ge02ed75+a9c1acf22d,21.0.0-4-g591bb35+a9c1acf22d,21.0.0-4-g65b4814+b60e2d390c,21.0.0-4-gccdca77+0de219a2bc,21.0.0-4-ge8a399c+6c55c39e83,21.0.0-5-gd00fb1e+05fce91b99,21.0.0-6-gc675373+3ad5d60fb1,21.0.0-64-g1122c245+4fb2b8f86e,21.0.0-7-g04766d7+cd19d05db2,21.0.0-7-gdf92d54+04719a4bac,21.0.0-8-g5674e7b+d1bd76f71f,master-gac4afde19b+a9c1acf22d,w.2021.13
LSST Data Management Base Package
frameDict.cc
Go to the documentation of this file.
1 /*
2  * LSST Data Management System
3  *
4  * This product includes software developed by the
5  * LSST Project (http://www.lsst.org/).
6  * See the COPYRIGHT file
7  *
8  * This program is free software: you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation, either version 3 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the LSST License Statement and
19  * the GNU General Public License along with this program. If not,
20  * see <https://www.lsstcorp.org/LegalNotices/>.
21  */
22 #include <memory>
23 
24 #include <pybind11/pybind11.h>
25 #include <pybind11/stl.h>
26 
27 namespace py = pybind11;
28 using namespace pybind11::literals;
29 
30 #include "astshim/Channel.h"
31 #include "astshim/Frame.h"
32 #include "astshim/FrameSet.h"
33 #include "astshim/FrameDict.h"
34 #include "astshim/Mapping.h"
35 #include "astshim/Stream.h"
36 
37 namespace ast {
38 namespace {
39 
40 /*
41 Make a FrameDict from a string output by Object.show()
42 
43 Use this instead of the standard ObjectMaker for FrameDict
44 because ObjectMaker would return a FrameSet (since the serialization
45 is the same for both).
46 */
47 class FrameDictMaker {
48 public:
49  FrameDictMaker() = default;
50  ~FrameDictMaker() = default;
51  std::shared_ptr<Object> operator()(std::string const &state) {
52  std::cout << "FrameDictMaker() called\n";
53  ast::StringStream stream(state);
54  ast::Channel chan(stream);
55  auto objPtr = chan.read();
56  auto frameSetPtr = std::dynamic_pointer_cast<ast::FrameSet>(objPtr);
57  if (!frameSetPtr) {
58  throw std::runtime_error("Object being unpickled is a " + objPtr->getClassName() +
59  " not a FrameSet");
60  }
61  return std::make_shared<ast::FrameDict>(*frameSetPtr);
62  }
63 };
64 
65 PYBIND11_MODULE(frameDict, mod) {
66  py::module::import("astshim.frameSet");
67 
68  py::class_<FrameDictMaker, std::shared_ptr<FrameDictMaker>> makerCls(mod, "FrameDictMaker");
69  makerCls.def(py::init<>());
70  makerCls.def("__call__", &FrameDictMaker::operator());
71  makerCls.def("__reduce__",
72  [makerCls](FrameDictMaker const &self) { return py::make_tuple(makerCls, py::tuple()); });
73 
74  py::class_<FrameDict, std::shared_ptr<FrameDict>, FrameSet> cls(mod, "FrameDict");
75 
76  cls.def(py::init<Frame const &, std::string const &>(), "frame"_a, "options"_a = "");
77  cls.def(py::init<Frame const &, Mapping const &, Frame const &, std::string const &>(), "baseFrame"_a,
78  "mapping"_a, "currentFrame"_a, "options"_a = "");
79  cls.def(py::init<FrameSet const &>(), "frameSet"_a);
80  cls.def(py::init<FrameDict const &>());
81 
82  cls.def("copy", &FrameDict::copy);
83 
84  cls.def("addFrame", py::overload_cast<int, Mapping const &, Frame const &>(&FrameDict::addFrame),
85  "iframe"_a, "map"_a, "frame"_a);
86  cls.def("addFrame",
87  py::overload_cast<std::string const &, Mapping const &, Frame const &>(&FrameDict::addFrame),
88  "domain"_a, "map"_a, "frame"_a);
89  cls.def("getAllDomains", &FrameDict::getAllDomains);
90  cls.def("getFrame", py::overload_cast<int, bool>(&FrameDict::getFrame, py::const_), "index"_a,
91  "copy"_a = true);
92  cls.def("getFrame", py::overload_cast<std::string const &, bool>(&FrameDict::getFrame, py::const_),
93  "domain"_a, "copy"_a = true);
94  cls.def("getMapping", py::overload_cast<int, int>(&FrameDict::getMapping, py::const_),
95  "from"_a = FrameDict::BASE, "to"_a = FrameDict::CURRENT);
96  cls.def("getMapping", py::overload_cast<int, std::string const &>(&FrameDict::getMapping, py::const_),
97  "from"_a = FrameDict::BASE, "to"_a = FrameDict::CURRENT);
98  cls.def("getMapping", py::overload_cast<std::string const &, int>(&FrameDict::getMapping, py::const_),
99  "from"_a = FrameDict::BASE, "to"_a = FrameDict::CURRENT);
100  cls.def("getMapping",
101  py::overload_cast<std::string const &, std::string const &>(&FrameDict::getMapping, py::const_),
102  "from"_a = FrameDict::BASE, "to"_a = FrameDict::CURRENT);
103  cls.def("getIndex", &FrameDict::getIndex, "domain"_a);
104  cls.def("hasDomain", &FrameDict::hasDomain, "domain"_a);
105  cls.def("mirrorVariants", py::overload_cast<int>(&FrameDict::mirrorVariants), "index"_a);
106  cls.def("mirrorVariants", py::overload_cast<std::string const &>(&FrameDict::mirrorVariants), "domain"_a);
107  cls.def("remapFrame", py::overload_cast<int, Mapping &>(&FrameDict::remapFrame), "index"_a, "map"_a);
108  cls.def("remapFrame", py::overload_cast<std::string const &, Mapping &>(&FrameDict::remapFrame),
109  "domain"_a, "map"_a);
110  cls.def("removeFrame", py::overload_cast<int>(&FrameDict::removeFrame), "index"_a);
111  cls.def("removeFrame", py::overload_cast<std::string const &>(&FrameDict::removeFrame), "domain"_a);
112  cls.def("setBase", py::overload_cast<int>(&FrameDict::setBase), "index"_a);
113  cls.def("setBase", py::overload_cast<std::string const &>(&FrameDict::setBase), "domain"_a);
114  cls.def("setCurrent", py::overload_cast<int>(&FrameDict::setCurrent), "index"_a);
115  cls.def("setCurrent", py::overload_cast<std::string const &>(&FrameDict::setCurrent), "domain"_a);
116  cls.def("setDomain", &FrameDict::setDomain, "domain"_a);
117 
119  cls.def("__reduce__", [makerCls](Object const &self) {
120  std::string state = self.show(false);
121  auto unpickleArgs = py::make_tuple(state);
122  return py::make_tuple(makerCls(), unpickleArgs);
123  });
124 }
125 
126 } // namespace
127 } // namespace ast
Channel provides input/output of AST objects.
Definition: Channel.h:60
String-based source and sink for channels.
Definition: Stream.h:180
AST wrapper classes and functions.
PYBIND11_MODULE(_cameraGeom, mod)
Definition: _cameraGeom.cc:38