re_arrow_store 0.2.0-alpha.0

An in-memory time series database for Rerun log data, based on Apache Arrow
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

//! Demonstrates usage of [`DataStore::to_dataframe`].
//!
//! ```text
//! POLARS_FMT_MAX_ROWS=100 cargo r -p re_arrow_store --example dump_dataframe
//! ```

use re_arrow_store::{test_bundle, DataStore};
use re_log_types::{
    component_types::InstanceKey,
    datagen::{
        build_frame_nr, build_log_time, build_some_instances, build_some_instances_from,
        build_some_point2d, build_some_rects,
    },
    msg_bundle::Component as _,
    EntityPath, Time,
};

// ---

fn main() {
    let mut store = DataStore::new(InstanceKey::name(), Default::default());

    let ent_paths = [
        EntityPath::from("this/that"),
        EntityPath::from("and/this/other/thing"),
    ];

    for ent_path in &ent_paths {
        let bundle1 = test_bundle!(ent_path @ [
            build_frame_nr(1.into()), build_log_time(Time::now()),
        ] => [build_some_instances(2), build_some_rects(2)]);
        store.insert(&bundle1).unwrap();
    }

    for ent_path in &ent_paths {
        let bundle2 = test_bundle!(ent_path @ [
            build_frame_nr(2.into())
        ] => [build_some_instances(2), build_some_point2d(2)]);
        store.insert(&bundle2).unwrap();
        // Insert timelessly too!
        let bundle2 =
            test_bundle!(ent_path @ [] => [build_some_instances(2), build_some_point2d(2)]);
        store.insert(&bundle2).unwrap();

        let bundle3 = test_bundle!(ent_path @ [
            build_frame_nr(3.into()), build_log_time(Time::now()),
        ] => [build_some_instances_from(25..29), build_some_point2d(4)]);
        store.insert(&bundle3).unwrap();
        // Insert timelessly too!
        let bundle3 = test_bundle!(ent_path @ [] => [build_some_instances_from(25..29), build_some_point2d(4)]);
        store.insert(&bundle3).unwrap();
    }

    for ent_path in &ent_paths {
        let bundle4_1 = test_bundle!(ent_path @ [
            build_frame_nr(4.into()), build_log_time(Time::now()),
        ] => [build_some_instances_from(20..23), build_some_rects(3)]);
        store.insert(&bundle4_1).unwrap();

        let bundle4_15 = test_bundle!(ent_path @ [
            build_frame_nr(4.into()),
        ] => [build_some_instances_from(20..23), build_some_point2d(3)]);
        store.insert(&bundle4_15).unwrap();

        let bundle4_2 = test_bundle!(ent_path @ [
            build_frame_nr(4.into()), build_log_time(Time::now()),
        ] => [build_some_instances_from(25..28), build_some_rects(3)]);
        store.insert(&bundle4_2).unwrap();

        let bundle4_25 = test_bundle!(ent_path @ [
            build_frame_nr(4.into()), build_log_time(Time::now()),
        ] => [build_some_instances_from(25..28), build_some_point2d(3)]);
        store.insert(&bundle4_25).unwrap();
    }

    let df = store.to_dataframe();
    println!("{df}");
}