Skip to content

Commit 021dda7

Browse files
authored
Add EvalKey serialization (#106)
1 parent b53c5c3 commit 021dda7

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/lib/pke/serialization.cpp

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,10 @@ void bind_serialization(pybind11::module &m) {
106106
py::arg("filename"), py::arg("obj"), py::arg("sertype"));
107107
m.def("DeserializeCiphertext", static_cast<std::tuple<Ciphertext<DCRTPoly>,bool> (*)(const std::string&, const SerType::SERJSON&)>(&DeserializeFromFileWrapper<Ciphertext<DCRTPoly>, SerType::SERJSON>),
108108
py::arg("filename"), py::arg("sertype"));
109+
m.def("SerializeToFile", static_cast<bool (*)(const std::string&, const EvalKey<DCRTPoly>&, const SerType::SERJSON&)>(&Serial::SerializeToFile<EvalKey<DCRTPoly>>),
110+
py::arg("filename"), py::arg("obj"), py::arg("sertype"));
111+
m.def("DeserializeEvalKey", static_cast<std::tuple<EvalKey<DCRTPoly>,bool> (*)(const std::string&, const SerType::SERJSON&)>(&DeserializeFromFileWrapper<EvalKey<DCRTPoly>, SerType::SERJSON>),
112+
py::arg("filename"), py::arg("sertype"));
109113
// Binary Serialization
110114
m.def("SerializeToFile", static_cast<bool (*)(const std::string&,const CryptoContext<DCRTPoly>&, const SerType::SERBINARY&)>(&Serial::SerializeToFile<DCRTPoly>),
111115
py::arg("filename"), py::arg("obj"), py::arg("sertype"));
@@ -122,7 +126,11 @@ void bind_serialization(pybind11::module &m) {
122126
m.def("SerializeToFile", static_cast<bool (*)(const std::string&, const Ciphertext<DCRTPoly>&, const SerType::SERBINARY&)>(&Serial::SerializeToFile<Ciphertext<DCRTPoly>>),
123127
py::arg("filename"), py::arg("obj"), py::arg("sertype"));
124128
m.def("DeserializeCiphertext", static_cast<std::tuple<Ciphertext<DCRTPoly>,bool> (*)(const std::string&, const SerType::SERBINARY&)>(&DeserializeFromFileWrapper<Ciphertext<DCRTPoly>, SerType::SERBINARY>),
125-
py::arg("filename"), py::arg("sertype"));
129+
py::arg("filename"), py::arg("sertype"));
130+
m.def("SerializeToFile", static_cast<bool (*)(const std::string&, const EvalKey<DCRTPoly>&, const SerType::SERBINARY&)>(&Serial::SerializeToFile<EvalKey<DCRTPoly>>),
131+
py::arg("filename"), py::arg("obj"), py::arg("sertype"));
132+
m.def("DeserializeEvalKey", static_cast<std::tuple<EvalKey<DCRTPoly>,bool> (*)(const std::string&, const SerType::SERBINARY&)>(&DeserializeFromFileWrapper<EvalKey<DCRTPoly>, SerType::SERBINARY>),
133+
py::arg("filename"), py::arg("sertype"));
126134

127135
}
128136

0 commit comments

Comments
 (0)