frameDict.cc

Go to the documentation of this file.

/*
 * LSST Data Management System
 *
 * This product includes software developed by the
 * LSST Project (http://www.lsst.org/).
 * See the COPYRIGHT file
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the LSST License Statement and
 * the GNU General Public License along with this program.  If not,
 * see <https://www.lsstcorp.org/LegalNotices/>.
 */
#include <memory>
 
#include <pybind11/pybind11.h>
#include "lsst/cpputils/python.h"
#include <pybind11/stl.h>
 
namespace py = pybind11;
using namespace pybind11::literals;
 
#include "astshim/Channel.h"
#include "astshim/Frame.h"
#include "astshim/FrameSet.h"
#include "astshim/FrameDict.h"
#include "astshim/Mapping.h"
#include "astshim/Stream.h"
 
namespace ast {
namespace {
 
/*
Make a FrameDict from a string output by Object.show()
 
Use this instead of the standard ObjectMaker for FrameDict
because ObjectMaker would return a FrameSet (since the serialization
is the same for both).
*/
class FrameDictMaker {
public:
    FrameDictMaker() = default;
    ~FrameDictMaker() = default;
    std::shared_ptr<Object> operator()(std::string const &state) {
        ast::StringStream stream(state);
        ast::Channel chan(stream);
        auto objPtr = chan.read();
        auto frameSetPtr = std::dynamic_pointer_cast<ast::FrameSet>(objPtr);
        if (!frameSetPtr) {
            throw std::runtime_error("Object being unpickled is a " + objPtr->getClassName() +
                                     " not a FrameSet");
        }
        return std::make_shared<ast::FrameDict>(*frameSetPtr);
    }
};
}
 
void wrapFrameDict(lsst::utils::python::WrapperCollection &wrappers) {
    using PyFrameDictMaker = py::class_<FrameDictMaker, std::shared_ptr<FrameDictMaker>>;
    static auto  clsMaker = wrappers.wrapType(PyFrameDictMaker(wrappers.module, "FrameDictMaker"), [](auto &mod, auto &cls) {
        cls.def(py::init<>());
        cls.def("__call__", &FrameDictMaker::operator());
        cls.def("__reduce__",
                [cls](FrameDictMaker const &self) { return py::make_tuple(cls, py::tuple()); });
    });
 
    using PyFrameDict = py::class_<FrameDict, std::shared_ptr<FrameDict>, FrameSet>;
    wrappers.wrapType(PyFrameDict(wrappers.module, "FrameDict"), [](auto &mod, auto &cls) {
 
        cls.def(py::init<Frame const &, std::string const &>(), "frame"_a, "options"_a = "");
        cls.def(py::init<Frame const &, Mapping const &, Frame const &, std::string const &>(), "baseFrame"_a,
                "mapping"_a, "currentFrame"_a, "options"_a = "");
        cls.def(py::init<FrameSet const &>(), "frameSet"_a);
        cls.def(py::init<FrameDict const &>());
 
        cls.def("copy", &FrameDict::copy);
 
        cls.def("addFrame", py::overload_cast<int, Mapping const &, Frame const &>(&FrameDict::addFrame),
                "iframe"_a, "map"_a, "frame"_a);
        cls.def("addFrame",
                py::overload_cast<std::string const &, Mapping const &, Frame const &>(&FrameDict::addFrame),
                "domain"_a, "map"_a, "frame"_a);
        cls.def("getAllDomains", &FrameDict::getAllDomains);
        cls.def("getFrame", py::overload_cast<int, bool>(&FrameDict::getFrame, py::const_), "index"_a,
                "copy"_a = true);
        cls.def("getFrame", py::overload_cast<std::string const &, bool>(&FrameDict::getFrame, py::const_),
                "domain"_a, "copy"_a = true);
        cls.def("getMapping", py::overload_cast<int, int>(&FrameDict::getMapping, py::const_),
                "from"_a = FrameDict::BASE, "to"_a = FrameDict::CURRENT);
        cls.def("getMapping", py::overload_cast<int, std::string const &>(&FrameDict::getMapping, py::const_),
                "from"_a = FrameDict::BASE, "to"_a = FrameDict::CURRENT);
        cls.def("getMapping", py::overload_cast<std::string const &, int>(&FrameDict::getMapping, py::const_),
                "from"_a = FrameDict::BASE, "to"_a = FrameDict::CURRENT);
        cls.def("getMapping",
                py::overload_cast<std::string const &, std::string const &>(&FrameDict::getMapping, py::const_),
                "from"_a = FrameDict::BASE, "to"_a = FrameDict::CURRENT);
        cls.def("getIndex", &FrameDict::getIndex, "domain"_a);
        cls.def("hasDomain", &FrameDict::hasDomain, "domain"_a);
        cls.def("mirrorVariants", py::overload_cast<int>(&FrameDict::mirrorVariants), "index"_a);
        cls.def("mirrorVariants", py::overload_cast<std::string const &>(&FrameDict::mirrorVariants), "domain"_a);
        cls.def("remapFrame", py::overload_cast<int, Mapping &>(&FrameDict::remapFrame), "index"_a, "map"_a);
        cls.def("remapFrame", py::overload_cast<std::string const &, Mapping &>(&FrameDict::remapFrame),
                "domain"_a, "map"_a);
        cls.def("removeFrame", py::overload_cast<int>(&FrameDict::removeFrame), "index"_a);
        cls.def("removeFrame", py::overload_cast<std::string const &>(&FrameDict::removeFrame), "domain"_a);
        cls.def("setBase", py::overload_cast<int>(&FrameDict::setBase), "index"_a);
        cls.def("setBase", py::overload_cast<std::string const &>(&FrameDict::setBase), "domain"_a);
        cls.def("setCurrent", py::overload_cast<int>(&FrameDict::setCurrent), "index"_a);
        cls.def("setCurrent", py::overload_cast<std::string const &>(&FrameDict::setCurrent), "domain"_a);
        cls.def("setDomain", &FrameDict::setDomain, "domain"_a);
 
        cls.def("__reduce__", [](Object const &self) {
            std::string state = self.show(false);
            auto unpickleArgs = py::make_tuple(state);
            return py::make_tuple(clsMaker(), unpickleArgs);
        });
    });
}
 
}  // namespace ast