use std::error::Error;
use std::fmt;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[derive(Debug)]
pub struct Failed {
err: FailedError,
msg: String,
}
#[non_exhaustive]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[derive(Copy, Clone, Debug)]
pub enum FailedError {
FitFailed = 1,
PredictFailed,
TransformFailed,
FindFailed,
DecompositionFailed,
SolutionFailed,
ParametersError,
}
impl Failed {
#[inline]
pub fn error(&self) -> FailedError {
self.err
}
pub fn fit(msg: &str) -> Self {
Failed {
err: FailedError::FitFailed,
msg: msg.to_string(),
}
}
pub fn predict(msg: &str) -> Self {
Failed {
err: FailedError::PredictFailed,
msg: msg.to_string(),
}
}
pub fn transform(msg: &str) -> Self {
Failed {
err: FailedError::TransformFailed,
msg: msg.to_string(),
}
}
pub fn because(err: FailedError, msg: &str) -> Self {
Failed {
err,
msg: msg.to_string(),
}
}
}
impl PartialEq for FailedError {
#[inline(always)]
fn eq(&self, rhs: &Self) -> bool {
*self as u8 == *rhs as u8
}
}
impl PartialEq for Failed {
#[inline(always)]
fn eq(&self, rhs: &Self) -> bool {
self.err == rhs.err && self.msg == rhs.msg
}
}
impl fmt::Display for FailedError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let failed_err_str = match self {
FailedError::FitFailed => "Fit failed",
FailedError::PredictFailed => "Predict failed",
FailedError::TransformFailed => "Transform failed",
FailedError::FindFailed => "Find failed",
FailedError::DecompositionFailed => "Decomposition failed",
FailedError::SolutionFailed => "Can't find solution",
FailedError::ParametersError => "Error in input, check parameters",
};
write!(f, "{failed_err_str}")
}
}
impl fmt::Display for Failed {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}: {}", self.err, self.msg)
}
}
impl Error for Failed {}