Add saturated arithmetic operations

Signed-off-by: Mattias Andrée <m@maandree.se>

1 files changed, 40 insertions, 0 deletions

diff --git a/libj2_j2u_add_j2u.c b/libj2_j2u_add_j2u.c
index 8aca375..5f26629 100644
--- a/libj2_j2u_add_j2u.c
+++ b/libj2_j2u_add_j2u.c
@@ -64,6 +64,12 @@ check_(uintmax_t a_high, uintmax_t a_low, uintmax_t b_high, uintmax_t b_low,
 
 	a = a_saved;
 	b = b_saved;
+	libj2_j2u_sat_add_j2u(&a, &b);
+	EXPECT(r_overflow ? libj2_j2u_is_max(&a) : libj2_j2u_eq_j2u(&a, &expected));
+	EXPECT(libj2_j2u_eq_j2u(&b, &b_saved));
+
+	a = a_saved;
+	b = b_saved;
 	EXPECT(libj2_j2u_add_j2u_overflow_p((const struct libj2_j2u *)&a, (const struct libj2_j2u *)&b) == r_overflow);
 	EXPECT(libj2_j2u_eq_j2u(&a, &a_saved));
 	EXPECT(libj2_j2u_eq_j2u(&b, &b_saved));
@@ -87,6 +93,26 @@ check_(uintmax_t a_high, uintmax_t a_low, uintmax_t b_high, uintmax_t b_low,
 	EXPECT(libj2_j2u_add_j2u_to_j2u_overflow(&a, &b, &b) == r_overflow);
 	EXPECT(libj2_j2u_eq_j2u(&b, &expected));
 	EXPECT(libj2_j2u_eq_j2u(&a, &a_saved));
+
+	r = (struct libj2_j2u){111, 222};
+	a = a_saved;
+	b = b_saved;
+	libj2_j2u_sat_add_j2u_to_j2u(&a, &b, &r);
+	EXPECT(r_overflow ? libj2_j2u_is_max(&r) : libj2_j2u_eq_j2u(&r, &expected));
+	EXPECT(libj2_j2u_eq_j2u(&a, &a_saved));
+	EXPECT(libj2_j2u_eq_j2u(&b, &b_saved));
+
+	a = a_saved;
+	b = b_saved;
+	libj2_j2u_sat_add_j2u_to_j2u(&a, &b, &a);
+	EXPECT(r_overflow ? libj2_j2u_is_max(&a) : libj2_j2u_eq_j2u(&a, &expected));
+	EXPECT(libj2_j2u_eq_j2u(&b, &b_saved));
+
+	a = a_saved;
+	b = b_saved;
+	libj2_j2u_sat_add_j2u_to_j2u(&a, &b, &b);
+	EXPECT(r_overflow ? libj2_j2u_is_max(&b) : libj2_j2u_eq_j2u(&b, &expected));
+	EXPECT(libj2_j2u_eq_j2u(&a, &a_saved));
 }
 
 
@@ -160,6 +186,10 @@ check_double(uintmax_t high, uintmax_t low)
 	EXPECT(libj2_j2u_eq_j2u(&a, &expected));
 
 	a = a_saved;
+	libj2_j2u_sat_add_j2u(&a, &a);
+	EXPECT(expected_overflow ? libj2_j2u_is_max(&a) : libj2_j2u_eq_j2u(&a, &expected));
+
+	a = a_saved;
 	EXPECT(libj2_j2u_add_j2u_overflow_p((const struct libj2_j2u *)&a, (const struct libj2_j2u *)&a) == expected_overflow);
 	EXPECT(libj2_j2u_eq_j2u(&a, &a_saved));
 
@@ -172,6 +202,16 @@ check_double(uintmax_t high, uintmax_t low)
 	a = a_saved;
 	EXPECT(libj2_j2u_add_j2u_to_j2u_overflow(&a, &a, &a) == expected_overflow);
 	EXPECT(libj2_j2u_eq_j2u(&a, &expected));
+
+	r = (struct libj2_j2u){111, 222};
+	a = a_saved;
+	libj2_j2u_sat_add_j2u_to_j2u(&a, &a, &r);
+	EXPECT(expected_overflow ? libj2_j2u_is_max(&r) : libj2_j2u_eq_j2u(&r, &expected));
+	EXPECT(libj2_j2u_eq_j2u(&a, &a_saved));
+
+	a = a_saved;
+	libj2_j2u_sat_add_j2u_to_j2u(&a, &a, &a);
+	EXPECT(expected_overflow ? libj2_j2u_is_max(&a) : libj2_j2u_eq_j2u(&a, &expected));
 }