use crate::Float;
use crate::InnerFloat::{Infinity, NaN, Zero};
use crate::test_util::common::rug_float_significant_bits;
use malachite_base::num::arithmetic::traits::NegAssign;
use malachite_base::num::conversion::traits::ExactFrom;
use malachite_base::rounding_modes::RoundingMode::{self, *};
use malachite_q::Rational;
use rug::float::Round;
use rug::ops::AssignRound;
use std::cmp::Ordering::{self, *};
use std::cmp::max;
pub fn rug_add_prec_round(
x: &rug::Float,
y: &rug::Float,
prec: u64,
rm: Round,
) -> (rug::Float, Ordering) {
let mut sum = rug::Float::with_val(u32::exact_from(prec), 0);
let o = sum.assign_round(x + y, rm);
(sum, o)
}
#[inline]
pub fn rug_add_round(x: &rug::Float, y: &rug::Float, rm: Round) -> (rug::Float, Ordering) {
rug_add_prec_round(
x,
y,
max(rug_float_significant_bits(x), rug_float_significant_bits(y)),
rm,
)
}
#[inline]
pub fn rug_add_prec(x: &rug::Float, y: &rug::Float, prec: u64) -> (rug::Float, Ordering) {
rug_add_prec_round(x, y, prec, Round::Nearest)
}
pub fn rug_add(x: &rug::Float, y: &rug::Float) -> rug::Float {
rug_add_prec_round(
x,
y,
max(rug_float_significant_bits(x), rug_float_significant_bits(y)),
Round::Nearest,
)
.0
}
pub fn rug_add_rational_prec_round(
x: &rug::Float,
y: &rug::Rational,
prec: u64,
rm: Round,
) -> (rug::Float, Ordering) {
let mut sum = rug::Float::with_val(u32::exact_from(prec), 0);
let o = sum.assign_round(x + y, rm);
(sum, o)
}
pub fn rug_add_rational_round(
x: &rug::Float,
y: &rug::Rational,
rm: Round,
) -> (rug::Float, Ordering) {
rug_add_rational_prec_round(x, y, rug_float_significant_bits(x), rm)
}
pub fn rug_add_rational_prec(
x: &rug::Float,
y: &rug::Rational,
prec: u64,
) -> (rug::Float, Ordering) {
rug_add_rational_prec_round(x, y, prec, Round::Nearest)
}
pub fn rug_add_rational(x: &rug::Float, y: &rug::Rational) -> rug::Float {
rug_add_rational_prec_round(x, y, rug_float_significant_bits(x), Round::Nearest).0
}
pub fn add_prec_round_naive(x: Float, y: Float, prec: u64, rm: RoundingMode) -> (Float, Ordering) {
assert_ne!(prec, 0);
match (x, y) {
(float_nan!(), _)
| (_, float_nan!())
| (float_infinity!(), float_negative_infinity!())
| (float_negative_infinity!(), float_infinity!()) => (float_nan!(), Equal),
(float_infinity!(), _) | (_, float_infinity!()) => (float_infinity!(), Equal),
(float_negative_infinity!(), _) | (_, float_negative_infinity!()) => {
(float_negative_infinity!(), Equal)
}
(float_zero!(), float_negative_zero!()) | (float_negative_zero!(), float_zero!()) => (
if rm == Floor {
float_negative_zero!()
} else {
float_zero!()
},
Equal,
),
(float_either_zero!(), mut z) | (mut z, float_either_zero!()) => {
let o = z.set_prec_round(prec, rm);
(z, o)
}
(x, y) => {
let (mut sum, o) = Float::from_rational_prec_round(
Rational::exact_from(x) + Rational::exact_from(y),
prec,
rm,
);
if rm == Floor && o == Equal && sum == 0u32 {
sum.neg_assign();
}
(sum, o)
}
}
}
pub fn add_rational_prec_round_naive(
x: Float,
y: Rational,
prec: u64,
rm: RoundingMode,
) -> (Float, Ordering) {
assert_ne!(prec, 0);
match (x, y) {
(x @ Float(NaN | Infinity { .. }), _) => (x, Equal),
(float_negative_zero!(), y) => {
if y == 0u32 {
(float_negative_zero!(), Equal)
} else {
Float::from_rational_prec_round(y, prec, rm)
}
}
(float_zero!(), y) => Float::from_rational_prec_round(y, prec, rm),
(x, y) => {
let (mut sum, o) =
Float::from_rational_prec_round(Rational::exact_from(x) + y, prec, rm);
if rm == Floor && sum == 0u32 {
sum.neg_assign();
}
(sum, o)
}
}
}