forked from mpusz/mp-units
pow() and sqrt() for quantities support added
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
// The MIT License (MIT)
|
||||
//
|
||||
// Copyright (c) 2018 Mateusz Pusz
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in all
|
||||
// copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
// SOFTWARE.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <units/quantity.h>
|
||||
#include <cmath>
|
||||
|
||||
namespace std::experimental::units {
|
||||
|
||||
template<std::size_t N, typename U, typename Rep>
|
||||
inline Quantity pow(const quantity<U, Rep>& q) noexcept
|
||||
{
|
||||
using dim = dimension_pow_t<typename U::dimension, N>;
|
||||
using r = ratio_pow<typename U::ratio, N>;
|
||||
return quantity<downcasting_traits_t<unit<dim, r>>, Rep>(std::pow(q.count(), N));
|
||||
}
|
||||
|
||||
template<typename U, typename Rep>
|
||||
inline Quantity sqrt(const quantity<U, Rep>& q) noexcept
|
||||
{
|
||||
using dim = dimension_sqrt_t<typename U::dimension>;
|
||||
using r = ratio_sqrt<typename U::ratio>;
|
||||
return quantity<downcasting_traits_t<unit<dim, r>>, Rep>(std::sqrt(q.count()));
|
||||
}
|
||||
|
||||
} // namespace std::experimental::units
|
||||
@@ -118,6 +118,52 @@ namespace std::experimental::units {
|
||||
template<Ratio R1, Ratio R2>
|
||||
using ratio_divide = detail::ratio_divide_impl<R1, R2>::type;
|
||||
|
||||
// ratio_pow
|
||||
|
||||
namespace detail {
|
||||
|
||||
template<typename R, std::size_t N>
|
||||
struct ratio_pow_impl {
|
||||
using type = ratio_multiply<typename ratio_pow_impl<R, N - 1>::type, R>;
|
||||
};
|
||||
|
||||
template<typename R>
|
||||
struct ratio_pow_impl<R, 0> {
|
||||
using type = R;
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
template<Ratio R, std::size_t N>
|
||||
using ratio_pow = detail::ratio_pow_impl<R, N>::type;
|
||||
|
||||
// ratio_sqrt
|
||||
|
||||
namespace detail {
|
||||
|
||||
constexpr std::intmax_t sqrt_impl(std::intmax_t v, std::intmax_t l, std::intmax_t r)
|
||||
{
|
||||
if(l == r)
|
||||
return r;
|
||||
|
||||
const auto mid = (r + l) / 2;
|
||||
if(mid * mid >= v)
|
||||
return sqrt_impl(v, l, mid);
|
||||
else
|
||||
return sqrt_impl(v, mid + 1, r);
|
||||
}
|
||||
|
||||
static constexpr std::intmax_t sqrt_impl(std::intmax_t v)
|
||||
{
|
||||
return sqrt_impl(v, 1, v);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
template<Ratio R>
|
||||
using ratio_sqrt = ratio<detail::sqrt_impl(R::den), detail::sqrt_impl(R::num)>;
|
||||
|
||||
|
||||
// common_ratio
|
||||
|
||||
namespace detail {
|
||||
|
||||
Reference in New Issue
Block a user