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
79
80
81
82
83
84
85
86
87
88
89

use std::fmt;
use std::path::PathBuf;

use nom::HexDisplay;

bitflags! {
    pub(crate) struct Flags: u64 {
        const BYTE_COUNT_MASK = 0x4000_0000;
        const BYTE_COUNT_VMASK = 0x4000;      // 16384
        const CLASS_MASK = 0x8000_0000;
        const NEW_CLASSTAG = 0xFFFF_FFFF;
    }
}
bitflags! {
    pub(crate) struct TObjectFlags: u32 {
        const IS_ON_HEAP = 0x0100_0000;
        const IS_REFERENCED = 1 << 4;
    }
}

/// Used in `TStreamerInfo`
/// Describes if the following entry is a new class or if it was
/// already described.
#[derive(Debug)]
pub enum ClassInfo {
    /// Class name of the new class
    New(String),
    /// Byte offset of new class tag in record, + 2; whatever... followed by object
    Exists(u64),
    /// Byte offset of new class tag in record, + 2; whatever... ref to object
    References(u64),
}

/// The most basic ROOT object from which almost everything inherits
#[derive(Debug, Clone)]
pub struct TObject {
    pub(crate) ver: u16,
    pub(crate) id: u32,
    pub(crate) bits: TObjectFlags,
}

/// A ROOT object with a name and a title
#[derive(Debug, Clone)]
pub struct TNamed {
    // pub(crate) ver: u16,
    // pub(crate) tobj: TObject,
    pub name: String,
    pub title: String,
}

/// A "list" (implemented as `Vec`) of arbitrary objects
#[derive(Debug)]
pub struct TList<'a> {
    pub(crate) ver: u16,
    pub(crate) tobj: TObject,
    pub(crate) name: String,
    pub(crate) len: usize,
    pub(crate) objs: Vec<Raw<'a>>
}

/// A type holding nothing but the original data and a class info object
pub struct Raw<'s> {
    pub(crate) classinfo: String,
    pub(crate) obj: &'s[u8]
}

/// The context from which we are currently parsing
#[derive(Debug)]
pub struct Context<'s> {
    /// Path to file of this context
    pub path: PathBuf,
    /// Offset between the beginning of `s` and to where absolute
    /// positions in the buffer point (e.g. for class defs)
    /// Usually something like TKey-length + 4
    pub offset: u64,
    /// The full buffer we are working on 
    pub s: &'s[u8],
}

impl<'s> fmt::Debug for Raw<'s> {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{} \n {}", self.classinfo, self.obj.to_hex(16))
    }
}

// Types which are so far unused:
// pub type TArrayD = Vec<f64>;
// pub type TArrayI = Vec<i32>;
// pub type TArrayL64 = Vec<i64>;