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

//! Types and functions for all supported versions.

use std::io::{Read, Seek};

use crate::{
    low::{FbxHeader, FbxVersion},
    pull_parser::{
        self,
        reader::{PlainSource, SeekableSource},
        ParserSource, ParserVersion,
    },
};

pub use self::error::{Error, Result};

mod error;

/// FBX tree type with any supported version.
#[non_exhaustive]
pub enum AnyParser<R> {
    /// FBX 7.4 or later.
    V7400(pull_parser::v7400::Parser<R>),
}

impl<R: ParserSource> AnyParser<R> {
    /// Returns the parser version.
    pub fn parser_version(&self) -> ParserVersion {
        match self {
            AnyParser::V7400(_) => pull_parser::v7400::Parser::<R>::PARSER_VERSION,
        }
    }

    /// Returns the FBX version.
    pub fn fbx_version(&self) -> FbxVersion {
        match self {
            AnyParser::V7400(parser) => parser.fbx_version(),
        }
    }
}

/// Returns the parser version for the FBX data.
fn parser_version(header: FbxHeader) -> Result<ParserVersion> {
    header
        .parser_version()
        .ok_or_else(|| Error::UnsupportedVersion(header.version()))
}

/// Loads a tree from the given reader.
///
/// This works for seekable readers (which implement [`std::io::Seek`]), but
/// [`from_seekable_reader`] should be used for them, because it is more
/// efficent.
///
/// [`std::io::Seek`]: https://doc.rust-lang.org/stable/std/io/trait.Seek.html
/// [`from_seekable_reader`]: fn.from_seekable_reader.html
pub fn from_reader<R: Read>(mut reader: R) -> Result<AnyParser<PlainSource<R>>> {
    let header = FbxHeader::load(&mut reader)?;
    match parser_version(header)? {
        ParserVersion::V7400 => {
            let parser = pull_parser::v7400::from_reader(header, reader).unwrap_or_else(|e| {
                panic!(
                    "Should never fail: FBX version {:?} should be supported by v7400 parser: {}",
                    header.version(),
                    e
                )
            });
            Ok(AnyParser::V7400(parser))
        }
    }
}

/// Loads a tree from the given seekable reader.
pub fn from_seekable_reader<R: Read + Seek>(mut reader: R) -> Result<AnyParser<SeekableSource<R>>> {
    let header = FbxHeader::load(&mut reader)?;
    match parser_version(header)? {
        ParserVersion::V7400 => {
            let parser =
                pull_parser::v7400::from_seekable_reader(header, reader).unwrap_or_else(|e| {
                    panic!(
                    "Should never fail: FBX version {:?} should be supported by v7400 parser: {}",
                    header.version(),
                    e
                )
                });
            Ok(AnyParser::V7400(parser))
        }
    }
}