remotia-profilation-utils 0.1.0

Profilation utilities for remotia, an open source framework for the development of remote rendering software in pure Rust
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

use std::{
    fs::{create_dir_all, File},
    io::Write,
    path::PathBuf,
};

use async_trait::async_trait;

use log::debug;
use remotia_core::{traits::FrameProcessor, types::FrameData};

pub struct RawFrameDumper {
    buffer_id: String,

    key: String,
    extension: String,

    folder: PathBuf,
}

impl RawFrameDumper {
    pub fn new(buffer_id: &str, folder: PathBuf) -> Self {
        create_dir_all(folder.clone()).unwrap();
        Self {
            buffer_id: buffer_id.to_string(),
            key: "capture_timestamp".to_string(),
            extension: "raw".to_string(),
            folder,
        }
    }

    pub fn key(mut self, key: &str) -> Self {
        self.key = key.to_string();
        self
    }

    pub fn extension(mut self, value: &str) -> Self {
        self.extension = value.to_string();
        self
    }
}

#[async_trait]
impl FrameProcessor for RawFrameDumper {
    async fn process(&mut self, mut frame_data: FrameData) -> Option<FrameData> {
        let frame_id = frame_data.get(&self.key);
        let buffer = frame_data.get_writable_buffer_ref(&self.buffer_id).unwrap();

        debug!("Dumping frame {}", frame_id);

        let mut file_path = self.folder.clone();
        file_path.push(format!("{}.{}", frame_id, self.extension));
        let mut output_file = File::create(file_path.as_path()).unwrap();
        output_file.write_all(buffer).unwrap();

        Some(frame_data)
    }
}