66 lines
1.7 KiB
C++
66 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#ifdef SIMDJSON_COMPETITION_SAJSON
|
|
|
|
#include "kostya.h"
|
|
|
|
namespace kostya {
|
|
|
|
struct sajson {
|
|
static constexpr diff_flags DiffFlags = diff_flags::IMPRECISE_FLOATS;
|
|
|
|
size_t ast_buffer_size{0};
|
|
size_t *ast_buffer{nullptr};
|
|
|
|
simdjson_really_inline double get_double(const ::sajson::value &obj, std::string_view key) {
|
|
using namespace sajson;
|
|
|
|
auto val = obj.get_value_of_key({key.data(), key.length()});
|
|
switch (val.get_type()) {
|
|
case TYPE_INTEGER:
|
|
case TYPE_DOUBLE:
|
|
return val.get_number_value();
|
|
default:
|
|
throw "field not double";
|
|
}
|
|
}
|
|
|
|
bool run(simdjson::padded_string &json, std::vector<point> &result) {
|
|
using namespace sajson;
|
|
|
|
if (!ast_buffer) {
|
|
ast_buffer_size = json.size();
|
|
ast_buffer = (size_t *)std::malloc(ast_buffer_size * sizeof(size_t));
|
|
}
|
|
auto doc = parse(
|
|
bounded_allocation(ast_buffer, ast_buffer_size),
|
|
mutable_string_view(json.size(), json.data())
|
|
);
|
|
if (!doc.is_valid()) { return false; }
|
|
|
|
auto root = doc.get_root();
|
|
if (root.get_type() != TYPE_OBJECT) { return false; }
|
|
auto points = root.get_value_of_key({"coordinates", strlen("coordinates")});
|
|
if (points.get_type() != TYPE_ARRAY) { return false; }
|
|
|
|
for (size_t i=0; i<points.get_length(); i++) {
|
|
auto point = points.get_array_element(i);
|
|
if (point.get_type() != TYPE_OBJECT) { return false; }
|
|
result.emplace_back(json_benchmark::point{
|
|
get_double(point, "x"),
|
|
get_double(point, "y"),
|
|
get_double(point, "z")
|
|
});
|
|
}
|
|
|
|
return true;
|
|
}
|
|
};
|
|
|
|
BENCHMARK_TEMPLATE(kostya, sajson)->UseManualTime();
|
|
|
|
} // namespace kostya
|
|
|
|
#endif // SIMDJSON_COMPETITION_SAJSON
|
|
|