mmd-anim-cli 0.3.0

CLI diagnostics and roundtrip tools for mmd-anim
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use std::{fs, process::ExitCode};

use crate::schema::MmdDumperOracleDump;
use serde::Serialize;

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub(crate) struct OracleSummaryMetrics {
    pub(crate) frames: usize,
    pub(crate) models: usize,
    pub(crate) first_model_bones: usize,
    pub(crate) first_model_morphs: usize,
}

pub(crate) fn oracle_summary_metrics(dump: &MmdDumperOracleDump) -> OracleSummaryMetrics {
    OracleSummaryMetrics {
        frames: dump.frames.len(),
        models: dump
            .frames
            .iter()
            .map(|frame| frame.models.len())
            .max()
            .unwrap_or(0),
        first_model_bones: dump
            .frames
            .first()
            .and_then(|frame| frame.models.first())
            .map(|model| model.bones.len())
            .unwrap_or(0),
        first_model_morphs: dump
            .frames
            .first()
            .and_then(|frame| frame.models.first())
            .map(|model| model.morphs.len())
            .unwrap_or(0),
    }
}

#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
struct OracleSummarySourceJson {
    mmd_version: String,
    dumper_version: String,
}

#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub(crate) struct OracleSummaryJsonReport {
    status: &'static str,
    command: &'static str,
    mode: &'static str,
    input: String,
    frames: usize,
    models: usize,
    first_model_bones: usize,
    first_model_morphs: usize,
    source: OracleSummarySourceJson,
}

pub(crate) fn oracle_summary_json_report(
    path: &str,
    dump: &MmdDumperOracleDump,
) -> OracleSummaryJsonReport {
    let metrics = oracle_summary_metrics(dump);
    OracleSummaryJsonReport {
        status: "ok",
        command: "verify",
        mode: "oracle",
        input: path.to_owned(),
        frames: metrics.frames,
        models: metrics.models,
        first_model_bones: metrics.first_model_bones,
        first_model_morphs: metrics.first_model_morphs,
        source: OracleSummarySourceJson {
            mmd_version: dump.source.mmd_version.clone(),
            dumper_version: dump.source.dumper_version.clone(),
        },
    }
}

pub(crate) fn oracle_summary(
    path: &str,
    use_json: bool,
) -> Result<ExitCode, Box<dyn std::error::Error>> {
    let content = fs::read_to_string(path)?;
    let dump = MmdDumperOracleDump::from_jsonl_str(&content, None)?;
    let metrics = oracle_summary_metrics(&dump);

    if use_json {
        println!(
            "{}",
            serde_json::to_string(&oracle_summary_json_report(path, &dump))?
        );
    } else {
        println!(
            "MMDDumper oracle: frames={} models={} firstModelBones={} firstModelMorphs={} mmd={} dumper={}",
            metrics.frames,
            metrics.models,
            metrics.first_model_bones,
            metrics.first_model_morphs,
            dump.source.mmd_version,
            dump.source.dumper_version
        );
    }
    Ok(ExitCode::SUCCESS)
}