mpusz/mp-units
1
0
Fork 1
mirror of https://github.com/mpusz/mp-units.git synced 2025-08-06 13:44:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add fma to math.h

Browse Source
This commit is contained in:
Nick Thompson
2023-12-09 13:16:57 -08:00
parent 173c500886
commit 51f8b47071
3 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/utility/include/mp-units/math.h
View File

@@ -136,6 +136,30 @@ template<auto R, typename Rep>
return {static_cast<Rep>(abs(q.numerical_value_ref_in(q.unit))), R}; return {static_cast<Rep>(abs(q.numerical_value_ref_in(q.unit))), R};
} }
/**
* @brief Computes the fma of 3 quantities
*
* @param a: Multiplicand
* @param x: Multiplicand
* @param b: Addend
* @return Quantity: The nearest floating point representable to ax+b
*/
template<auto R, auto S, typename Rep>
[[nodiscard]] constexpr quantity<R, Rep> fma(const quantity<R, Rep>& a, const quantity<S, Rep>& x,
const quantity<R, Rep>& b) noexcept
requires requires {
fma(a.numerical_value_ref_in(a.unit), x.numerical_value_ref_in(x.unit), b.numerical_value_ref_in(b.unit));
} || requires {
std::fma(a.numerical_value_ref_in(a.unit), x.numerical_value_ref_in(x.unit), b.numerical_value_ref_in(b.unit));
}
{
using std::fma;
return {static_cast<Rep>(
fma(a.numerical_value_ref_in(a.unit), x.numerical_value_ref_in(x.unit), b.numerical_value_ref_in(b.unit))),
R};
}
/** /**
* @brief Returns the epsilon of the quantity * @brief Returns the epsilon of the quantity
* *

6
test/unit_test/runtime/math_test.cpp
View File

@@ -62,6 +62,12 @@ TEST_CASE("'cbrt()' on quantity changes the value and the dimension accordingly"
REQUIRE(cbrt(8 * isq::volume[m3]) == 2 * isq::length[m]); REQUIRE(cbrt(8 * isq::volume[m3]) == 2 * isq::length[m]);
} }
TEST_CASE("'fma()' on quantity changes the value and the dimension accordingly", "[math][cbrt]")
{
REQUIRE(fma(1.0 * isq::length[m], 2.0, 2.0 * isq::length[m]) == 4.0 * isq::length[m]);
}
TEST_CASE("'pow<Num, Den>()' on quantity changes the value and the dimension accordingly", "[math][pow]") TEST_CASE("'pow<Num, Den>()' on quantity changes the value and the dimension accordingly", "[math][pow]")
{ {
REQUIRE(pow<1, 4>(16 * isq::area[m2]) == sqrt(4 * isq::length[m])); REQUIRE(pow<1, 4>(16 * isq::area[m2]) == sqrt(4 * isq::length[m]));

5
test/unit_test/static/math_test.cpp
View File

@@ -40,6 +40,11 @@ template<typename T1, typename T2, typename... Ts>
#if __cpp_lib_constexpr_cmath || MP_UNITS_COMP_GCC #if __cpp_lib_constexpr_cmath || MP_UNITS_COMP_GCC
static_assert(compare(fma(2 * m, 3 * m, 1 * m2), 7 * m2));
static_assert(compare(fma(2.0 * s, 3.0 * Hz, 1.0), 7.0));
static_assert(compare(fma(2 * s, 3 * Hz, 1), 7));
static_assert(compare(fma(2.0, 3.0*m, 1.0*m), 7*m);
static_assert(compare(fma(2.0*m, 3.0, 1.0*m), 7*m));
static_assert(compare(pow<0>(2 * m), 1 * one)); static_assert(compare(pow<0>(2 * m), 1 * one));
static_assert(compare(pow<1>(2 * m), 2 * m)); static_assert(compare(pow<1>(2 * m), 2 * m));
static_assert(compare(pow<2>(2 * m), 4 * pow<2>(m), 4 * m2)); static_assert(compare(pow<2>(2 * m), 4 * pow<2>(m), 4 * m2));