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

/*
   Appellation: kinds <mod>
   Contrib: FL03 <jo3mccain@icloud.com>
*/
pub use self::{external::*, predict::*};

mod external;
mod predict;

use crate::nn::ModelError;
use strum::{AsRefStr, Display, EnumCount, EnumIs, VariantNames};

#[derive(
    AsRefStr,
    Clone,
    Debug,
    Display,
    EnumCount,
    EnumIs,
    Eq,
    Hash,
    Ord,
    PartialEq,
    PartialOrd,
    VariantNames,
)]
#[cfg_attr(
    feature = "serde",
    derive(serde::Deserialize, serde::Serialize),
    serde(rename_all = "lowercase", tag = "kind")
)]
#[strum(serialize_all = "lowercase")]
pub enum Error {
    IO(String),
    External(ExternalError),
    Predict(PredictError),
    Model(ModelError),
    Shape(String),
}

// impl core::fmt::Display for Error {
//     fn fmt(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result {
//         let msg = match self {
//             Error::IO(ref err) => err.to_string(),
//             Error::Error(ref err) => err.to_string(),
//             Error::Shape(ref err) => err.to_string(),
//         };
//         write!(f, "{}", msg)
//     }
// }

#[cfg(feature = "std")]
error_from!(Error::IO<std::io::Error>);

macro_rules! from_err {
    ($($variant:ident<$err:ty>),* $(,)*) => {
        $(
            from_err!(@impl $variant<$err>);
        )*
    };
    (@impl $variant:ident<$err:ty>) => {
        impl From<$err> for Error {
            fn from(err: $err) -> Self {
                Error::$variant(err)
            }
        }
    };
}

from_err!(
    External<ExternalError>,
    Model<ModelError>,
    Predict<PredictError>,
);