use crate::common::data::Data;
use crate::core::extract::binop;
pub fn add(data: Data) -> Result<Data, String> {
let result = match binop(data) {
(Data::Real(l), Data::Real(r)) => Data::Real(l + r),
(Data::Integer(l), Data::Integer(r)) => Data::Integer(l + r),
(Data::String(l), Data::String(r)) => Data::String(format!("{}{}", l, r)),
_ => return Err("Addition between unsupported datatypes".to_string()),
};
Ok(result)
}
pub fn sub(data: Data) -> Result<Data, String> {
let result = match binop(data) {
(Data::Real(l), Data::Real(r)) => Data::Real(l - r),
(Data::Integer(l), Data::Integer(r)) => Data::Integer(l - r),
_ => return Err("Subtraction between unsupported datatypes".to_string()),
};
Ok(result)
}
pub fn neg(data: Data) -> Result<Data, String> {
let result = match data {
Data::Real(n) => Data::Real(-n),
Data::Integer(n) => Data::Integer(-n),
_ => return Err("Subtraction between unsupported datatypes".to_string()),
};
Ok(result)
}
pub fn mul(data: Data) -> Result<Data, String> {
let result = match binop(data) {
(Data::Real(l), Data::Real(r)) => Data::Real(l * r),
(Data::Integer(l), Data::Integer(r)) => Data::Integer(l * r),
_ => return Err("Multiplication between unsupported datatypes".to_string()),
};
Ok(result)
}
pub fn div(data: Data) -> Result<Data, String> {
let result = match binop(data) {
(Data::Real(_), Data::Real(n)) if n == 0.0 => return Err("Division by zero".to_string()),
(Data::Real(l), Data::Real(r)) => Data::Real(l / r),
(Data::Integer(_), Data::Integer(n)) if n == 0 => return Err("Division by zero".to_string()),
(Data::Integer(l), Data::Integer(r)) => Data::Integer(l / r),
_ => return Err("Division between unsupported datatypes".to_string()),
};
Ok(result)
}
pub fn rem(data: Data) -> Result<Data, String> {
let result = match binop(data) {
(Data::Real(_), Data::Real(r)) if r == 0.0 => return Err("Division by zero".to_string()),
(Data::Real(l), Data::Real(r)) => Data::Real(l.rem_euclid(r)),
(Data::Integer(_), Data::Integer(n)) if n == 0 => return Err("Division by zero".to_string()),
(Data::Integer(l), Data::Integer(r)) => Data::Integer(l.rem_euclid(r)),
_ => return Err("Division between unsupported datatypes".to_string()),
};
Ok(result)
}
pub fn pow(data: Data) -> Result<Data, String> {
let result = match binop(data) {
(Data::Real(l), Data::Real(r)) => Data::Real(l.powf(r)),
(Data::Integer(l), Data::Integer(r)) => Data::Integer(l.pow(r as u32)),
_ => return Err("Exponentiation between unsupported datatypes".to_string()),
};
Ok(result)
}