kll/kll_sketch_float.cpp

/*
 * 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
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

#include <emscripten/bind.h>

#include <kll_sketch.hpp>
#include <kolmogorov_smirnov.hpp>

using kll_sketch_float = datasketches::kll_sketch<float>;

const emscripten::val Uint8Array = emscripten::val::global("Uint8Array");

EMSCRIPTEN_BINDINGS(kll_sketch_float) {
  emscripten::function("getExceptionMessage", emscripten::optional_override([](intptr_t ptr) {
    return std::string(reinterpret_cast<std::exception*>(ptr)->what());
  }));

  emscripten::register_vector<float>("VectorFloat");

  emscripten::constant("DEFAULT_K", datasketches::kll_constants::DEFAULT_K);

  emscripten::class_<kll_sketch_float>("kll_sketch_float")
    .constructor(emscripten::optional_override([](uint16_t k) {
      return new kll_sketch_float(k);
    }))
    .function("isEmpty", &kll_sketch_float::is_empty)
    .function("update", emscripten::optional_override([](kll_sketch_float& self, float value) {
      self.update(value);
    }))
    .function("merge", emscripten::optional_override([](kll_sketch_float& self, const std::string& bytes) {
      self.merge(kll_sketch_float::deserialize(bytes.data(), bytes.size()));
    }))
    .function("serializeAsUint8Array", emscripten::optional_override([](const kll_sketch_float& self) {
      auto bytes = self.serialize();
      return Uint8Array.new_(emscripten::typed_memory_view(bytes.size(), bytes.data()));
    }))
    .class_function("deserialize", emscripten::optional_override([](const std::string& bytes) {
      return new kll_sketch_float(kll_sketch_float::deserialize(bytes.data(), bytes.size()));
    }), emscripten::allow_raw_pointers())
    .function("getN", &kll_sketch_float::get_n)
    .function("getNumRetained", &kll_sketch_float::get_num_retained)
    .function("getMinValue", &kll_sketch_float::get_min_item)
    .function("getMaxValue", &kll_sketch_float::get_max_item)
    .function("getRank", &kll_sketch_float::get_rank)
    .function("getQuantile", &kll_sketch_float::get_quantile)
    .function("getPMF", emscripten::optional_override([](const kll_sketch_float& self, const std::vector<float>& split_points, bool inclusive) {
      const auto pmf = self.get_PMF(split_points.data(), split_points.size(), inclusive);
      return emscripten::val::array(pmf.begin(), pmf.end());
    }))
    .function("getCDF", emscripten::optional_override([](const kll_sketch_float& self, const std::vector<float>& split_points, bool inclusive) {
      const auto cdf = self.get_CDF(split_points.data(), split_points.size(), inclusive);
      return emscripten::val::array(cdf.begin(), cdf.end());
    }))
    .function("toString", emscripten::optional_override([](const kll_sketch_float& self) {
      return self.to_string();
    }))
    .function("getNormalizedRankError", emscripten::optional_override([](const kll_sketch_float& self, bool pmf) {
      return self.get_normalized_rank_error(pmf);
    }))
    ;

  emscripten::function("kolmogorovSmirnovTest", emscripten::optional_override([](const std::string& sketch_bytes1, const std::string& sketch_bytes2, double pvalue) {
    return datasketches::kolmogorov_smirnov::test(
      kll_sketch_float::deserialize(sketch_bytes1.data(), sketch_bytes1.size()),
      kll_sketch_float::deserialize(sketch_bytes2.data(), sketch_bytes2.size()),
      pvalue
    );
  }));
}