init

Telegram/ThirdParty/cld3/gcld3/pybind_ext.cc

@@ -0,0 +1,43 @@
+#include <pybind11/pybind11.h>
+#include <pybind11/pytypes.h>
+#include <pybind11/stl.h>
+
+#include "../src/nnet_language_identifier.h"
+
+namespace pybind11 {
+
+using chrome_lang_id::NNetLanguageIdentifier;
+
+// This is conventional.
+namespace py = pybind11;
+
+PYBIND11_MODULE(pybind_ext, py_module) {
+  py::class_<NNetLanguageIdentifier>(py_module, "NNetLanguageIdentifier")
+      .def(py::init<const int, const int>(), py::arg("min_num_bytes"),
+           py::arg("max_num_bytes"))
+      .def("FindLanguage", &NNetLanguageIdentifier::FindLanguage,
+           py::arg("text"))
+      .def("FindTopNMostFreqLangs",
+           &NNetLanguageIdentifier::FindTopNMostFreqLangs, py::arg("text"),
+           py::arg("num_langs"))
+      .def_readonly_static("kUnknown", &NNetLanguageIdentifier::kUnknown)
+      .def_readonly_static("kMinNumBytesToConsider",
+                           &NNetLanguageIdentifier::kMinNumBytesToConsider)
+      .def_readonly_static("kMaxNumBytesToConsider",
+                           &NNetLanguageIdentifier::kMaxNumBytesToConsider)
+      .def_readonly_static("kMaxNumInputBytesToConsider",
+                           &NNetLanguageIdentifier::kMaxNumInputBytesToConsider)
+      .def_readonly_static("kReliabilityThreshold",
+                           &NNetLanguageIdentifier::kReliabilityThreshold)
+      .def_readonly_static("kReliabilityHrBsThreshold",
+                           &NNetLanguageIdentifier::kReliabilityHrBsThreshold);
+
+  py::class_<NNetLanguageIdentifier::Result>(py_module, "Result")
+      .def_readwrite("language", &NNetLanguageIdentifier::Result::language)
+      .def_readwrite("probability",
+                     &NNetLanguageIdentifier::Result::probability)
+      .def_readwrite("is_reliable",
+                     &NNetLanguageIdentifier::Result::is_reliable)
+      .def_readwrite("proportion", &NNetLanguageIdentifier::Result::proportion);
+}
+}  // namespace pybind11