NeuralError.h

Go to the documentation of this file.

#pragma once
 
#include <ostream>
#include <string>
 
namespace NN {
 
enum class ErrorCode {
    InvalidArgument,
    OutOfMemory,
    RuntimeError
};
 
 
class NeuralError {
public:
    NeuralError(ErrorCode code, const std::string& message)
        : m_code(code), m_message(message) {}
 
    ErrorCode code() const { return m_code; }
    std::string message() const { return m_message; }
 
private:
    ErrorCode m_code;
    std::string m_message;
};
 
// Helper function to convert ErrorCode to string
inline const char* to_string(ErrorCode code) {
    switch (code) {
        case ErrorCode::InvalidArgument: return "InvalidArgument";
        case ErrorCode::OutOfMemory:     return "OutOfMemory";
        case ErrorCode::RuntimeError:    return "RuntimeError";
        default:                         return "UnknownError";
    }
}
 
// Overload operator<< for NeuralError
inline std::ostream& operator<<(std::ostream& os, const NeuralError& err) {
    os << "[" << to_string(err.code()) << "] " << err.message();
    return os;
}
 
} // namespace NN