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

refactor(example): get_magnitude improved for linear algebra

Browse Source
This commit is contained in:
Mateusz Pusz
2023-09-13 19:53:32 +02:00
parent dc2d0dfca7
commit 6d619d67ef
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
test/unit_test/runtime/linear_algebra_test.cpp
View File

@@ -56,11 +56,19 @@ namespace {
using namespace mp_units; using namespace mp_units;
using namespace mp_units::si::unit_symbols; using namespace mp_units::si::unit_symbols;
template<typename T> template<QuantitySpec auto QS, QuantityOf<QS> Q>
[[nodiscard]] auto get_magnitude(const vector<T>& v) requires(Q::quantity_spec.character == quantity_character::vector) && (QS.character == quantity_character::scalar)
[[nodiscard]] QuantityOf<QS> auto get_magnitude(const Q& q)
{ {
using namespace std; const auto& v = q.numerical_value_ref_in(q.unit);
return hypot(v(0), v(1), v(2)); return hypot(v(0) * QS[q.unit], v(1) * QS[q.unit], v(2) * QS[q.unit]);
}
template<QuantitySpec auto QS, QuantityOf<QS> T>
requires(T::quantity_spec.character == quantity_character::vector) && (QS.character == quantity_character::scalar)
[[nodiscard]] QuantityOf<QS> auto get_magnitude(const vector<T>& v)
{
return hypot(QS(v(0)), QS(v(1)), QS(v(2)));
} }
template<typename T, typename U> template<typename T, typename U>
@@ -100,8 +108,7 @@ TEST_CASE("vector quantity", "[la]")
SECTION("to scalar magnitude") SECTION("to scalar magnitude")
{ {
const auto v = vector<int>{2, 3, 6} * isq::velocity[km / h]; const auto v = vector<int>{2, 3, 6} * isq::velocity[km / h];
const auto speed = const auto speed = get_magnitude<isq::speed>(v);
get_magnitude(v.numerical_value_ref_in(km / h)) * isq::speed[v.unit]; // TODO can we do better here?
CHECK(speed.numerical_value_ref_in(km / h) == 7); CHECK(speed.numerical_value_ref_in(km / h) == 7);
} }
@@ -308,8 +315,7 @@ TEST_CASE("vector of quantities", "[la]")
SECTION("to scalar magnitude") SECTION("to scalar magnitude")
{ {
const vector<quantity<isq::velocity[km / h], int>> v = {2 * (km / h), 3 * (km / h), 6 * (km / h)}; const vector<quantity<isq::velocity[km / h], int>> v = {2 * (km / h), 3 * (km / h), 6 * (km / h)};
const auto speed = const auto speed = get_magnitude<isq::speed>(v);
get_magnitude(v).numerical_value_ref_in(km / h) * isq::speed[v(0).unit]; // TODO can we do better here?
CHECK(speed.numerical_value_ref_in(km / h) == 7); CHECK(speed.numerical_value_ref_in(km / h) == 7);
} }