Merge pull request #6 from Desilo/chulsoon/add_serializable_galois_keys

갈로아 키 사이즈 최적화

Author: CsYang1248

src/wrapper.cpp

@@ -384,6 +384,20 @@ PYBIND11_MODULE(seal, m)
 		})
 		;
 
+	// serializable.h
+	py::class_<Serializable<GaloisKeys>>(m, "SerializableGaloisKeys")
+		.def("save", [](const Serializable<GaloisKeys> &gk, const std::string &path){
+			std::ofstream out(path, std::ofstream::binary);
+			gk.save(out);
+			out.close();
+		})
+		.def("dumpb", [](const Serializable<GaloisKeys> &gk){
+			std::stringstream out(std::ios::binary | std::ios::out);
+			gk.save(out);
+			return py::bytes(out.str());
+		})
+		;
+
 	// keygenerator.h
 	py::class_<KeyGenerator>(m, "KeyGenerator")
 		.def(py::init<const SEALContext &>())
@@ -407,7 +421,11 @@ PYBIND11_MODULE(seal, m)
 			GaloisKeys gk;
 			keygen.create_galois_keys(gk);
 			return gk;
-		});
+		})
+		.def("create_serializable_galois_keys", [](KeyGenerator &keygen){
+			return keygen.create_galois_keys();
+		})
+		;
 
 	// encryptor.h
 	py::class_<Encryptor>(m, "Encryptor")