blob: 2a8ffa47f2d265630799eef2b8f9ec139ff0e5f7 [file] [log] [blame]
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
#include <sstream>
#include <pybind11/pybind11.h>
#include "cpc_sketch.hpp"
#include "cpc_union.hpp"
#include "cpc_common.hpp"
#include "common_defs.hpp"
namespace py = pybind11;
namespace datasketches {
namespace python {
cpc_sketch* cpc_sketch_deserialize(py::bytes skBytes) {
std::string skStr = skBytes; // implicit cast
return new cpc_sketch(cpc_sketch::deserialize(skStr.c_str(), skStr.length()));
py::object cpc_sketch_serialize(const cpc_sketch& sk) {
auto serResult = sk.serialize();
return py::bytes((char*), serResult.size());
cpc_sketch* cpc_union_get_result(const cpc_union& u) {
return new cpc_sketch(u.get_result());
namespace dspy = datasketches::python;
void init_cpc(py::module &m) {
using namespace datasketches;
py::class_<cpc_sketch>(m, "cpc_sketch")
.def(py::init<uint8_t, uint64_t>(), py::arg("lg_k"), py::arg("seed")=DEFAULT_SEED)
.def(py::init<const cpc_sketch&>())
.def("__str__", &cpc_sketch::to_string)
.def("to_string", &cpc_sketch::to_string)
.def("serialize", &dspy::cpc_sketch_serialize)
.def_static("deserialize", &dspy::cpc_sketch_deserialize)
.def<void (cpc_sketch::*)(uint64_t)>("update", &cpc_sketch::update, py::arg("datum"))
.def<void (cpc_sketch::*)(double)>("update", &cpc_sketch::update, py::arg("datum"))
.def<void (cpc_sketch::*)(const std::string&)>("update", &cpc_sketch::update, py::arg("datum"))
.def("is_empty", &cpc_sketch::is_empty)
.def("get_estimate", &cpc_sketch::get_estimate)
.def("get_lower_bound", &cpc_sketch::get_lower_bound, py::arg("kappa"))
.def("get_upper_bound", &cpc_sketch::get_upper_bound, py::arg("kappa"))
py::class_<cpc_union>(m, "cpc_union")
.def(py::init<uint8_t, uint64_t>(), py::arg("lg_k"), py::arg("seed")=DEFAULT_SEED)
.def(py::init<const cpc_union&>())
.def("update", (void (cpc_union::*)(const cpc_sketch&)) &cpc_union::update, py::arg("sketch"))
.def("get_result", &dspy::cpc_union_get_result)