mrc 0.5.1

MRC-2014 file format reader/writer for cryo-EM — SIMD-accelerated, mmap-enabled
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

// ============================================================================
// AGAR — Agard format
// ============================================================================

#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};

/// Size of a single AGAR record, in bytes.
pub const AGAR_RECORD_SIZE: usize = 1024;

/// Agard extended header record.
///
/// Agard extended headers use 1024-byte records.  This format is used by
/// older Agard-style microscopes.  Access the raw bytes for field parsing.
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[derive(Debug, Clone, PartialEq)]
pub struct AgarRecord {
    /// Raw 1024-byte record.
    #[cfg_attr(feature = "serde", serde(with = "crate::serde_byte_array"))]
    pub raw: [u8; AGAR_RECORD_SIZE],
}

impl AgarRecord {
    /// Parse a single Agard record from bytes.
    pub fn from_bytes(bytes: &[u8]) -> Option<Self> {
        if bytes.len() < AGAR_RECORD_SIZE {
            return None;
        }
        let mut raw = [0u8; AGAR_RECORD_SIZE];
        raw.copy_from_slice(&bytes[..AGAR_RECORD_SIZE]);
        Some(Self { raw })
    }
}

/// Parse extended header bytes as Agard records.
pub fn parse_agar_records(bytes: &[u8]) -> Option<Vec<AgarRecord>> {
    if bytes.is_empty() || bytes.len() % AGAR_RECORD_SIZE != 0 {
        return None;
    }
    let count = bytes.len() / AGAR_RECORD_SIZE;
    let mut records = Vec::with_capacity(count);
    for i in 0..count {
        let start = i * AGAR_RECORD_SIZE;
        records.push(AgarRecord::from_bytes(
            &bytes[start..start + AGAR_RECORD_SIZE],
        )?);
    }
    Some(records)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn agar_roundtrip() {
        let buf = vec![77u8; AGAR_RECORD_SIZE];
        let records = super::parse_agar_records(&buf).unwrap();
        assert_eq!(records.len(), 1);
        assert_eq!(records[0].raw[0], 77);
    }

    #[test]
    fn agar_empty() {
        assert!(super::parse_agar_records(&[]).is_none());
    }
}