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

//! Logs a bunch of big images to test Rerun memory usage.

// Allow unwrap() in tests (allow-unwrap-in-tests doesn't apply)
#![allow(clippy::unwrap_used)]

use mimalloc::MiMalloc;

use re_memory::AccountingAllocator;
use rerun::{
    archetypes::Image,
    external::{image, re_memory},
};

#[global_allocator]
static GLOBAL: AccountingAllocator<MiMalloc> = AccountingAllocator::new(MiMalloc);

fn main() -> Result<(), Box<dyn std::error::Error>> {
    re_memory::accounting_allocator::turn_on_tracking_if_env_var("RERUN_TRACK_ALLOCATIONS");

    let rec = rerun::RecordingStreamBuilder::new("rerun_example_image_memory").spawn()?;
    log_images(&rec).unwrap();

    Ok(())
}

fn log_images(rec: &rerun::RecordingStream) -> Result<(), Box<dyn std::error::Error>> {
    let (w, h) = (2048, 1024);
    let n = 100;

    let image = image::RgbaImage::from_fn(w, h, |x, y| {
        if (x + y) % 2 == 0 {
            image::Rgba([0, 0, 0, 255])
        } else {
            image::Rgba([255, 255, 255, 255])
        }
    });

    for _ in 0..n {
        rec.log("image", &Image::from_image(image.clone())?)?;
    }

    rec.flush_blocking();

    eprintln!(
        "Logged {n} {w}x{h} RGBA images = {}",
        re_format::format_bytes((n * w * h * 4) as _)
    );

    // Give viewer time to load it:
    std::thread::sleep(std::time::Duration::from_secs(2));

    if let Some(allocs) = re_memory::accounting_allocator::global_allocs() {
        eprintln!("{} RAM used", re_format::format_bytes(allocs.size as _));
    }

    Ok(())
}