re_data_loader 0.31.4

Handles loading of Rerun data from file using data loader plugins
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

//! Test utilities for MCAP data loader snapshot testing.

use std::path::{Path, PathBuf};

use re_chunk::{Chunk, EntityPath};

use crate::loader_mcap::McapLoader;
use crate::{DataLoader as _, DataLoaderSettings, LoadedData};

// Helper function to get the path to a test asset file.
pub fn test_asset(name: &str) -> PathBuf {
    PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("src/loader_mcap/tests/assets")
        .join(name)
}

/// Loads an MCAP file using the default loader configuration for testing purposes.
pub fn load_mcap(path: impl AsRef<Path>) -> LoadedMcap {
    let path = path.as_ref();

    let loader = McapLoader::default();

    let (tx, rx) = crossbeam::channel::bounded(1024);
    let settings = DataLoaderSettings::recommended("test");

    loader
        .load_from_path(&settings, path.to_path_buf(), tx)
        .unwrap_or_else(|err| {
            panic!("Failed to load MCAP file at {}: {err}", path.display());
        });

    let chunks: Vec<Chunk> = rx.iter().filter_map(LoadedData::into_chunk).collect();

    if 10_000 < chunks.len() {
        re_log::warn!(
            "MCAP file contained {} chunks. Consider running `rerun rrd compact` on the output.",
            re_format::format_uint(chunks.len()),
        );
    }

    LoadedMcap { chunks }
}

/// Result of loading a test MCAP file, providing convenient access to chunks.
pub struct LoadedMcap {
    chunks: Vec<Chunk>,
}

impl LoadedMcap {
    /// Returns chunks for a specific entity path.
    pub fn chunks_for_entity(&self, path: &str) -> Vec<&Chunk> {
        let entity_path: EntityPath = path.into();
        self.chunks
            .iter()
            .filter(|c| c.entity_path() == &entity_path)
            .collect()
    }
}