wehub-resource-sync
82 lines
2.7 KiB
C++
82 lines
2.7 KiB
C++
#include <rerun.hpp>
|
|
#include <rerun/demo_utils.hpp>
|
|
|
|
#include <algorithm> // std::generate
|
|
#include <random>
|
|
#include <vector>
|
|
|
|
using namespace rerun::demo;
|
|
using namespace std::chrono_literals;
|
|
|
|
static constexpr size_t NUM_POINTS = 100;
|
|
|
|
int main() {
|
|
const auto rec = rerun::RecordingStream("rerun_example_dna_abacus");
|
|
rec.spawn().exit_on_failure();
|
|
|
|
std::vector<rerun::Position3D> points1, points2;
|
|
std::vector<rerun::Color> colors1, colors2;
|
|
color_spiral(NUM_POINTS, 2.0f, 0.02f, 0.0f, 0.1f, points1, colors1);
|
|
color_spiral(NUM_POINTS, 2.0f, 0.02f, TAU * 0.5f, 0.1f, points2, colors2);
|
|
|
|
rec.set_time_duration("stable_time", 0s);
|
|
|
|
rec.log_static(
|
|
"dna/structure/left",
|
|
rerun::Points3D(points1).with_colors(colors1).with_radii({0.08f})
|
|
);
|
|
rec.log_static(
|
|
"dna/structure/right",
|
|
rerun::Points3D(points2).with_colors(colors2).with_radii({0.08f})
|
|
);
|
|
|
|
std::vector<rerun::LineStrip3D> lines;
|
|
for (size_t i = 0; i < points1.size(); ++i) {
|
|
lines.emplace_back(rerun::LineStrip3D({points1[i].xyz, points2[i].xyz}));
|
|
}
|
|
|
|
rec.log_static(
|
|
"dna/structure/scaffolding",
|
|
rerun::LineStrips3D(lines).with_colors(rerun::Color(128, 128, 128))
|
|
);
|
|
|
|
std::default_random_engine gen;
|
|
std::uniform_real_distribution<float> dist(0.0f, 1.0f);
|
|
std::vector<float> offsets(NUM_POINTS);
|
|
std::generate(offsets.begin(), offsets.end(), [&] { return dist(gen); });
|
|
|
|
std::vector<rerun::Position3D> beads_positions(lines.size());
|
|
std::vector<rerun::Color> beads_colors(lines.size());
|
|
|
|
for (int t = 0; t < 400; t++) {
|
|
auto time = std::chrono::duration<float>(t) * 0.01f;
|
|
|
|
rec.set_time_duration("stable_time", time);
|
|
|
|
for (size_t i = 0; i < lines.size(); ++i) {
|
|
float time_offset = time.count() + offsets[i];
|
|
auto c = static_cast<uint8_t>(bounce_lerp(80.0f, 230.0f, time_offset * 2.0f));
|
|
|
|
beads_positions[i] = rerun::Position3D(
|
|
bounce_lerp(lines[i].points[0].x(), lines[i].points[1].x(), time_offset),
|
|
bounce_lerp(lines[i].points[0].y(), lines[i].points[1].y(), time_offset),
|
|
bounce_lerp(lines[i].points[0].z(), lines[i].points[1].z(), time_offset)
|
|
);
|
|
beads_colors[i] = rerun::Color(c, c, c);
|
|
}
|
|
|
|
rec.log(
|
|
"dna/structure/scaffolding/beads",
|
|
rerun::Points3D(beads_positions).with_colors(beads_colors).with_radii({0.06f})
|
|
);
|
|
|
|
rec.log(
|
|
"dna/structure",
|
|
rerun::archetypes::Transform3D(rerun::RotationAxisAngle(
|
|
{0.0f, 0.0f, 1.0f},
|
|
rerun::Angle::radians(time.count() / 4.0f * TAU)
|
|
))
|
|
);
|
|
}
|
|
}
|