cervo 0.2.0

High level APIs for RL in games.
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

// Author: Tom Solberg <tom.solberg@embark-studios.com>
// Copyright © 2022, Embark Studios AB, all rights reserved.
// Created: 13 May 2022

/*!

*/

use anyhow::{bail, Result};
use cervo_asset::AssetData;
use clap::Parser;
use std::{fs::File, path::PathBuf};

/// Shortly describe the model file.
#[derive(Parser, Debug)]
#[clap()]
pub(crate) struct DescribeArgs {
    file: PathBuf,
}

pub(super) fn describe(config: DescribeArgs) -> Result<()> {
    let mut reader = File::open(&config.file)?;

    if cervo_nnef::is_nnef_tar(&config.file) {
        println!("a NNEF file")
    } else {
        match config.file.extension().and_then(|ext| ext.to_str()) {
            Some("onnx") => println!("an ONNX file"),
            Some("crvo") => {
                let asset = AssetData::deserialize(&mut reader)?;
                println!("a native cervo file containing {} data", asset.kind())
            }
            Some(other) => bail!("unknown file type {:?}", other),
            None => bail!("missing file extension {:?}", config.file),
        }
    }

    Ok(())
}