use malachite_base::test_util::bench::{BenchmarkType, run_benchmark};
use malachite_base::test_util::generators::common::{GenConfig, GenMode};
use malachite_base::test_util::runner::Runner;
use malachite_q::gaussian_rational::GaussianRational;
use malachite_q::test_util::bench::bucketers::{
pair_gaussian_rational_max_bit_bucketer, vec_gaussian_rational_sum_bits_bucketer,
};
use malachite_q::test_util::gaussian_rational::arithmetic::add::gaussian_rational_sum_naive;
use malachite_q::test_util::generators::{gaussian_rational_pair_gen, gaussian_rational_vec_gen};
use std::iter::Sum;
pub(crate) fn register(runner: &mut Runner) {
register_demo!(runner, demo_gaussian_rational_add);
register_demo!(runner, demo_gaussian_rational_add_val_ref);
register_demo!(runner, demo_gaussian_rational_add_ref_val);
register_demo!(runner, demo_gaussian_rational_add_ref_ref);
register_demo!(runner, demo_gaussian_rational_add_assign);
register_demo!(runner, demo_gaussian_rational_add_assign_ref);
register_demo!(runner, demo_gaussian_rational_sum);
register_demo!(runner, demo_gaussian_rational_ref_sum);
register_bench!(runner, benchmark_gaussian_rational_add_evaluation_strategy);
register_bench!(
runner,
benchmark_gaussian_rational_add_assign_evaluation_strategy
);
register_bench!(runner, benchmark_gaussian_rational_sum_algorithms);
register_bench!(runner, benchmark_gaussian_rational_sum_evaluation_strategy);
}
fn demo_gaussian_rational_add(gm: GenMode, config: &GenConfig, limit: usize) {
for (x, y) in gaussian_rational_pair_gen().get(gm, config).take(limit) {
let x_old = x.clone();
let y_old = y.clone();
println!("({x_old}) + ({y_old}) = {}", x + y);
}
}
fn demo_gaussian_rational_add_val_ref(gm: GenMode, config: &GenConfig, limit: usize) {
for (x, y) in gaussian_rational_pair_gen().get(gm, config).take(limit) {
let x_old = x.clone();
println!("({x_old}) + (&{y}) = {}", x + &y);
}
}
fn demo_gaussian_rational_add_ref_val(gm: GenMode, config: &GenConfig, limit: usize) {
for (x, y) in gaussian_rational_pair_gen().get(gm, config).take(limit) {
let y_old = y.clone();
println!("(&{x}) + ({y_old}) = {}", &x + y);
}
}
fn demo_gaussian_rational_add_ref_ref(gm: GenMode, config: &GenConfig, limit: usize) {
for (x, y) in gaussian_rational_pair_gen().get(gm, config).take(limit) {
println!("(&{x}) + (&{y}) = {}", &x + &y);
}
}
fn demo_gaussian_rational_add_assign(gm: GenMode, config: &GenConfig, limit: usize) {
for (mut x, y) in gaussian_rational_pair_gen().get(gm, config).take(limit) {
let x_old = x.clone();
let y_old = y.clone();
x += y;
println!("x := {x_old}; x += {y_old}; x = {x}");
}
}
fn demo_gaussian_rational_add_assign_ref(gm: GenMode, config: &GenConfig, limit: usize) {
for (mut x, y) in gaussian_rational_pair_gen().get(gm, config).take(limit) {
let x_old = x.clone();
x += &y;
println!("x := {x_old}; x += &{y}; x = {x}");
}
}
#[allow(unused_must_use)]
fn benchmark_gaussian_rational_add_evaluation_strategy(
gm: GenMode,
config: &GenConfig,
limit: usize,
file_name: &str,
) {
run_benchmark(
"GaussianRational + GaussianRational",
BenchmarkType::EvaluationStrategy,
gaussian_rational_pair_gen().get(gm, config),
gm.name(),
limit,
file_name,
&pair_gaussian_rational_max_bit_bucketer("x", "y"),
&mut [
("GaussianRational + GaussianRational", &mut |(x, y)| {
no_out!(x + y);
}),
("GaussianRational + &GaussianRational", &mut |(x, y)| {
no_out!(x + &y);
}),
("&GaussianRational + GaussianRational", &mut |(x, y)| {
no_out!(&x + y);
}),
("&GaussianRational + &GaussianRational", &mut |(x, y)| {
no_out!(&x + &y);
}),
],
);
}
#[allow(unused_must_use)]
fn benchmark_gaussian_rational_add_assign_evaluation_strategy(
gm: GenMode,
config: &GenConfig,
limit: usize,
file_name: &str,
) {
run_benchmark(
"GaussianRational += GaussianRational",
BenchmarkType::EvaluationStrategy,
gaussian_rational_pair_gen().get(gm, config),
gm.name(),
limit,
file_name,
&pair_gaussian_rational_max_bit_bucketer("x", "y"),
&mut [
("GaussianRational += GaussianRational", &mut |(mut x, y)| {
no_out!(x += y);
}),
(
"GaussianRational += &GaussianRational",
&mut |(mut x, y)| no_out!(x += &y),
),
],
);
}
fn demo_gaussian_rational_sum(gm: GenMode, config: &GenConfig, limit: usize) {
for xs in gaussian_rational_vec_gen().get(gm, config).take(limit) {
println!(
"sum({:?}) = {}",
xs.clone(),
GaussianRational::sum(xs.into_iter())
);
}
}
fn demo_gaussian_rational_ref_sum(gm: GenMode, config: &GenConfig, limit: usize) {
for xs in gaussian_rational_vec_gen().get(gm, config).take(limit) {
println!("sum({:?}) = {}", xs, GaussianRational::sum(xs.iter()));
}
}
fn benchmark_gaussian_rational_sum_algorithms(
gm: GenMode,
config: &GenConfig,
limit: usize,
file_name: &str,
) {
run_benchmark(
"GaussianRational::sum(Iterator<Item=GaussianRational>)",
BenchmarkType::Algorithms,
gaussian_rational_vec_gen().get(gm, config),
gm.name(),
limit,
file_name,
&vec_gaussian_rational_sum_bits_bucketer(),
&mut [
("default", &mut |xs| {
no_out!(GaussianRational::sum(xs.into_iter()));
}),
("naive", &mut |xs| {
no_out!(gaussian_rational_sum_naive(xs.into_iter()));
}),
],
);
}
fn benchmark_gaussian_rational_sum_evaluation_strategy(
gm: GenMode,
config: &GenConfig,
limit: usize,
file_name: &str,
) {
run_benchmark(
"GaussianRational::sum(Iterator<Item=GaussianRational>)",
BenchmarkType::EvaluationStrategy,
gaussian_rational_vec_gen().get(gm, config),
gm.name(),
limit,
file_name,
&vec_gaussian_rational_sum_bits_bucketer(),
&mut [
(
"GaussianRational::sum(Iterator<Item=GaussianRational>)",
&mut |xs| {
no_out!(GaussianRational::sum(xs.into_iter()));
},
),
(
"GaussianRational::sum(Iterator<Item=&GaussianRational>)",
&mut |xs| {
no_out!(GaussianRational::sum(xs.iter()));
},
),
],
);
}