void bind_density_sketch()

in src/density_wrapper.cpp [38:98]

• 60 lines of code
• 1 McCabe index (conditional complexity)

void bind_density_sketch(nb::module_ &m, const char* name) {
  using namespace datasketches;

  nb::class_<density_sketch<T, K>>(m, name)
    .def("__init__", [](density_sketch<T, K>* sk, uint16_t k, uint32_t dim, kernel_function* kernel)
        { K holder(kernel);
          new (sk) density_sketch<T, K>(k, dim, holder);
        },
        nb::arg("k"), nb::arg("dim"), nb::arg("kernel"),
        "Creates a new density sketch\n\n"
        ":param k: controls the size and error of the sketch\n:type k: int\n"
        ":param dim: dimension of the input data\n:type dim: int\n"
        ":param kernel: instance of a kernel\n:type kernel: KernelFunction\n"
        )
    .def("__copy__", [](const density_sketch<T,K>& sk){ return density_sketch<T,K>(sk); })
    .def("update", static_cast<void (density_sketch<T, K>::*)(const std::vector<T>&)>(&density_sketch<T, K>::update), nb::arg("vector"),
        "Updates the sketch with the given vector")
    .def("merge", static_cast<void (density_sketch<T, K>::*)(const density_sketch<T, K>&)>(&density_sketch<T, K>::merge), nb::arg("sketch"),
        "Merges the provided sketch into this one")
    .def("is_empty", &density_sketch<T, K>::is_empty,
        "Returns True if the sketch is empty, otherwise False")
    .def_prop_ro("k", &density_sketch<T, K>::get_k,
        "The configured parameter k")
    .def_prop_ro("dim", &density_sketch<T, K>::get_dim,
        "The configured parameter dim")
    .def_prop_ro("n", &density_sketch<T, K>::get_n,
        "The length of the input stream")
    .def_prop_ro("num_retained", &density_sketch<T, K>::get_num_retained,
        "The number of retained items (samples) in the sketch")
    .def("is_estimation_mode", &density_sketch<T, K>::is_estimation_mode,
        "Returns True if the sketch is in estimation mode, otherwise False")
    .def("get_estimate", &density_sketch<T, K>::get_estimate, nb::arg("point"),
        "Returns an approximate density at the given point")
    .def("__str__", [](const density_sketch<T, K>& sk) { return sk.to_string(); },
        "Produces a string summary of the sketch")
    .def("to_string", &density_sketch<T, K>::to_string, nb::arg("print_levels")=false, nb::arg("print_items")=false,
        "Produces a string summary of the sketch")
    .def("__iter__", [](const density_sketch<T, K> &sk) {
                        return nb::make_iterator(nb::type<density_sketch<T,K> >(),
                                                 "density_iterator",
                                                 sk.begin(),
                                                 sk.end());
                      },
        nb::keep_alive<0,1>())
    .def("serialize",
        [](const density_sketch<T, K>& sk) {
          auto bytes = sk.serialize();
          return nb::bytes(reinterpret_cast<const char*>(bytes.data()), bytes.size());
        },
        "Serializes the sketch into a bytes object"
    )
    .def_static(
        "deserialize",
          [](const nb::bytes& bytes, kernel_function* kernel) {
          K holder(kernel);
          return density_sketch<T, K>::deserialize(bytes.c_str(), bytes.size(), holder);
        },
        nb::arg("bytes"), nb::arg("kernel"),
        "Reads a bytes object and returns the corresponding density_sketch"
    );
}