wehub-resource-sync
67 lines
2.0 KiB
C++
67 lines
2.0 KiB
C++
// Use `send_column` to send an entire column of custom data to Rerun.
|
|
|
|
#include <rerun.hpp>
|
|
|
|
#include <arrow/array/builder_primitive.h>
|
|
#include <cmath>
|
|
#include <cstdio>
|
|
#include <numeric>
|
|
|
|
arrow::Status run_main() {
|
|
const auto rec =
|
|
rerun::RecordingStream("rerun_example_any_batch_value_column_updates");
|
|
rec.spawn().exit_on_failure();
|
|
|
|
constexpr int64_t STEPS = 64;
|
|
|
|
std::vector<int64_t> times(STEPS);
|
|
std::iota(times.begin(), times.end(), 0);
|
|
|
|
std::shared_ptr<arrow::Array> arrow_array;
|
|
|
|
arrow::DoubleBuilder one_per_timestamp_builder;
|
|
for (int64_t i = 0; i < STEPS; i++) {
|
|
ARROW_RETURN_NOT_OK(
|
|
one_per_timestamp_builder.Append(sin(static_cast<double>(i) / 10.0))
|
|
);
|
|
}
|
|
ARROW_RETURN_NOT_OK(one_per_timestamp_builder.Finish(&arrow_array));
|
|
auto one_per_timestamp_result = rerun::ComponentBatch::from_arrow_array(
|
|
std::move(arrow_array),
|
|
"custom_component_single"
|
|
);
|
|
auto one_per_timestamp = one_per_timestamp_result.value_or_throw();
|
|
|
|
arrow::DoubleBuilder ten_per_timestamp_builder;
|
|
for (int64_t i = 0; i < STEPS * 10; i++) {
|
|
ARROW_RETURN_NOT_OK(ten_per_timestamp_builder.Append(
|
|
cos(static_cast<double>(i) / 100.0)
|
|
));
|
|
}
|
|
ARROW_RETURN_NOT_OK(ten_per_timestamp_builder.Finish(&arrow_array));
|
|
auto ten_per_timestamp_result = rerun::ComponentBatch::from_arrow_array(
|
|
std::move(arrow_array),
|
|
"custom_component_multi"
|
|
);
|
|
auto ten_per_timestamp = ten_per_timestamp_result.value_or_throw();
|
|
|
|
rec.send_columns(
|
|
"/",
|
|
rerun::TimeColumn::from_sequence("step", std::move(times)),
|
|
one_per_timestamp.partitioned().value_or_throw(),
|
|
ten_per_timestamp.partitioned(std::vector<uint32_t>(STEPS, 10))
|
|
.value_or_throw()
|
|
);
|
|
|
|
return arrow::Status::OK();
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
arrow::Status status = run_main();
|
|
if (!status.ok()) {
|
|
printf("%s\n", status.ToString().c_str());
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|