426 lines
15 KiB
C++
426 lines
15 KiB
C++
///\file
|
|
|
|
/******************************************************************************
|
|
The MIT License(MIT)
|
|
|
|
Embedded Template Library.
|
|
https://github.com/ETLCPP/etl
|
|
https://www.etlcpp.com
|
|
|
|
Copyright(c) 2018 John Wellbelove
|
|
|
|
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.
|
|
******************************************************************************/
|
|
|
|
#ifndef ETL_CUMULATIVE_MOVING_AVERAGE_INCLUDED
|
|
#define ETL_CUMULATIVE_MOVING_AVERAGE_INCLUDED
|
|
|
|
#include "type_traits.h"
|
|
#include "iterator.h"
|
|
|
|
namespace etl
|
|
{
|
|
namespace private_pseudo_moving_average
|
|
{
|
|
//***************************************************
|
|
/// add_insert_iterator
|
|
/// An output iterator used to add new values.
|
|
//***************************************************
|
|
template <typename TPseudo_Moving_Average>
|
|
class add_insert_iterator : public etl::iterator<ETL_OR_STD::output_iterator_tag, typename TPseudo_Moving_Average::value_type, void, void, void>
|
|
{
|
|
public:
|
|
|
|
//***********************************
|
|
explicit add_insert_iterator(TPseudo_Moving_Average& pma) ETL_NOEXCEPT
|
|
: p_pma(&pma)
|
|
{
|
|
}
|
|
|
|
//***********************************
|
|
add_insert_iterator& operator*() ETL_NOEXCEPT
|
|
{
|
|
return *this;
|
|
}
|
|
|
|
//***********************************
|
|
add_insert_iterator& operator++() ETL_NOEXCEPT
|
|
{
|
|
return *this;
|
|
}
|
|
|
|
//***********************************
|
|
add_insert_iterator& operator++(int) ETL_NOEXCEPT
|
|
{
|
|
return *this;
|
|
}
|
|
|
|
//***********************************
|
|
add_insert_iterator& operator =(typename TPseudo_Moving_Average::value_type value)
|
|
{
|
|
p_pma->add(value);
|
|
return *this;
|
|
}
|
|
|
|
private:
|
|
|
|
TPseudo_Moving_Average* p_pma;
|
|
};
|
|
}
|
|
|
|
//***************************************************************************
|
|
/// Pseudo Moving Average
|
|
/// \tparam T The sample value type.
|
|
/// \tparam SAMPLE_SIZE The number of samples to average over.
|
|
/// \tparam SCALING The scaling factor applied to samples. Default = 1.
|
|
//***************************************************************************
|
|
template <typename T,
|
|
const size_t SAMPLE_SIZE,
|
|
const size_t SCALING = 1U,
|
|
const bool IsIntegral = etl::is_integral<T>::value,
|
|
const bool IsFloat = etl::is_floating_point<T>::value>
|
|
class pseudo_moving_average;
|
|
|
|
//***************************************************************************
|
|
/// Pseudo Moving Average
|
|
/// For integral types.
|
|
/// \tparam T The sample value type.
|
|
/// \tparam SAMPLE_SIZE The number of samples to average over.
|
|
/// \tparam SCALING The scaling factor applied to samples. Default = 1.
|
|
//***************************************************************************
|
|
template <typename T, const size_t SAMPLE_SIZE_, const size_t SCALING_>
|
|
class pseudo_moving_average<T, SAMPLE_SIZE_, SCALING_, true, false>
|
|
{
|
|
private:
|
|
|
|
typedef pseudo_moving_average<T, SAMPLE_SIZE_, SCALING_, true, false> this_t;
|
|
|
|
typedef typename etl::conditional<etl::is_signed<T>::value, int32_t, uint32_t>::type scale_t;
|
|
typedef typename etl::conditional<etl::is_signed<T>::value, int32_t, uint32_t>::type sample_t;
|
|
|
|
static ETL_CONSTANT sample_t SAMPLES = static_cast<sample_t>(SAMPLE_SIZE_);
|
|
static ETL_CONSTANT scale_t SCALE = static_cast<scale_t>(SCALING_);
|
|
|
|
public:
|
|
|
|
typedef T value_type;
|
|
typedef private_pseudo_moving_average::add_insert_iterator<this_t> add_insert_iterator;
|
|
|
|
static ETL_CONSTANT size_t SAMPLE_SIZE = SAMPLE_SIZE_; ///< The number of samples averaged over.
|
|
static ETL_CONSTANT size_t SCALING = SCALING_; ///< The sample scaling factor.
|
|
|
|
//*************************************************************************
|
|
/// Constructor
|
|
/// \param initial_value The initial value for the average.
|
|
//*************************************************************************
|
|
pseudo_moving_average(const T initial_value)
|
|
: average(initial_value * SCALE)
|
|
{
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Clears the average.
|
|
/// \param initial_value The initial value for the average.
|
|
//*************************************************************************
|
|
void clear(const T initial_value)
|
|
{
|
|
average = (initial_value * SCALE);
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Adds a new sample to the average.
|
|
/// \param new_value The value to add.
|
|
//*************************************************************************
|
|
void add(T new_value)
|
|
{
|
|
average *= SAMPLES;
|
|
average += SCALE * new_value;
|
|
average /= SAMPLES + sample_t(1);
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Gets the current pseudo moving average.
|
|
/// \return The current average.
|
|
//*************************************************************************
|
|
T value() const
|
|
{
|
|
return average;
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Gets an iterator for input.
|
|
/// \return An iterator.
|
|
//*************************************************************************
|
|
add_insert_iterator input()
|
|
{
|
|
return add_insert_iterator(*this);
|
|
}
|
|
|
|
private:
|
|
|
|
T average; ///< The current pseudo moving average.
|
|
};
|
|
|
|
template <typename T, const size_t SAMPLE_SIZE_, const size_t SCALING_>
|
|
ETL_CONSTANT size_t pseudo_moving_average<T, SAMPLE_SIZE_, SCALING_, true, false>::SAMPLE_SIZE;
|
|
|
|
template <typename T, const size_t SAMPLE_SIZE_, const size_t SCALING_>
|
|
ETL_CONSTANT size_t pseudo_moving_average<T, SAMPLE_SIZE_, SCALING_, true, false>::SCALING;
|
|
|
|
//***************************************************************************
|
|
/// Pseudo Moving Average
|
|
/// For integral types.
|
|
/// \tparam T The sample value type.
|
|
/// \tparam SCALING The scaling factor applied to samples. Default = 1.
|
|
//***************************************************************************
|
|
template <typename T, const size_t SCALING_>
|
|
class pseudo_moving_average<T, 0, SCALING_, true, false>
|
|
{
|
|
typedef pseudo_moving_average<T, 0, SCALING_, true, false> this_t;
|
|
|
|
typedef typename etl::conditional<etl::is_signed<T>::value, int32_t, uint32_t>::type scale_t;
|
|
typedef typename etl::conditional<etl::is_signed<T>::value, int32_t, uint32_t>::type sample_t;
|
|
|
|
static ETL_CONSTANT scale_t SCALE = static_cast<scale_t>(SCALING_);
|
|
|
|
public:
|
|
|
|
typedef T value_type;
|
|
typedef private_pseudo_moving_average::add_insert_iterator<this_t> add_insert_iterator;
|
|
|
|
static ETL_CONSTANT size_t SCALING = SCALING_; ///< The sample scaling factor.
|
|
|
|
//*************************************************************************
|
|
/// Constructor
|
|
/// \param initial_value The initial value for the average.
|
|
//*************************************************************************
|
|
pseudo_moving_average(const T initial_value, const size_t sample_size)
|
|
: average(initial_value * SCALE)
|
|
, samples(sample_t(sample_size))
|
|
{
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Clears the average.
|
|
/// \param initial_value The initial value for the average.
|
|
//*************************************************************************
|
|
void clear(const T initial_value)
|
|
{
|
|
average = (initial_value * SCALE);
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Sets the sample size.
|
|
/// \param sample_size The new sample size.
|
|
//*************************************************************************
|
|
void set_sample_size(const size_t sample_size)
|
|
{
|
|
samples = sample_t(sample_size);
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Adds a new sample to the average.
|
|
/// \param new_value The value to add.
|
|
//*************************************************************************
|
|
void add(T new_value)
|
|
{
|
|
average *= samples;
|
|
average += SCALE * new_value;
|
|
average /= samples + sample_t(1);
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Gets the current pseudo moving average.
|
|
/// \return The current average.
|
|
//*************************************************************************
|
|
T value() const
|
|
{
|
|
return average;
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Gets an iterator for input.
|
|
/// \return An iterator.
|
|
//*************************************************************************
|
|
add_insert_iterator input()
|
|
{
|
|
return add_insert_iterator(*this);
|
|
}
|
|
|
|
private:
|
|
|
|
T average; ///< The current pseudo moving average.
|
|
sample_t samples; ///< The number of samples to average over.
|
|
};
|
|
|
|
template <typename T, const size_t SCALING_>
|
|
ETL_CONSTANT size_t pseudo_moving_average<T, 0, SCALING_, true, false>::SCALING;
|
|
|
|
//***************************************************************************
|
|
/// Pseudo Moving Average
|
|
/// For floating point types.
|
|
/// \tparam T The sample value type.
|
|
/// \tparam SAMPLE_SIZE The number of samples to average over.
|
|
//***************************************************************************
|
|
template <typename T, const size_t SAMPLE_SIZE_>
|
|
class pseudo_moving_average<T, SAMPLE_SIZE_, 1U, false, true>
|
|
{
|
|
typedef pseudo_moving_average<T, SAMPLE_SIZE_, 1U, false, true> this_t;
|
|
|
|
public:
|
|
|
|
typedef T value_type;
|
|
typedef private_pseudo_moving_average::add_insert_iterator<this_t> add_insert_iterator;
|
|
|
|
static ETL_CONSTANT size_t SAMPLE_SIZE = SAMPLE_SIZE_;
|
|
|
|
//*************************************************************************
|
|
/// Constructor
|
|
/// \param initial_value The initial value for the average.
|
|
//*************************************************************************
|
|
pseudo_moving_average(const T initial_value)
|
|
: reciprocal_samples_plus_1(T(1.0) / T(SAMPLE_SIZE_ + 1U))
|
|
, average(initial_value)
|
|
{
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Clears the average.
|
|
/// \param initial_value The initial value for the average.
|
|
//*************************************************************************
|
|
void clear(const T initial_value)
|
|
{
|
|
average = initial_value;
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Adds a new sample to the average.
|
|
/// \param new_value The value to add.
|
|
//*************************************************************************
|
|
void add(const T new_value)
|
|
{
|
|
average += (new_value - average) * reciprocal_samples_plus_1;
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Gets the current pseudo moving average.
|
|
/// \return The current average.
|
|
//*************************************************************************
|
|
T value() const
|
|
{
|
|
return average;
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Gets an iterator for input.
|
|
/// \return An iterator.
|
|
//*************************************************************************
|
|
add_insert_iterator input()
|
|
{
|
|
return add_insert_iterator(*this);
|
|
}
|
|
|
|
private:
|
|
|
|
const T reciprocal_samples_plus_1; ///< Reciprocal of one greater than the sample size.
|
|
T average; ///< The current pseudo moving average.
|
|
};
|
|
|
|
template <typename T, const size_t SAMPLE_SIZE_>
|
|
ETL_CONSTANT size_t pseudo_moving_average<T, SAMPLE_SIZE_, 1U, false, true>::SAMPLE_SIZE;
|
|
|
|
//***************************************************************************
|
|
/// Pseudo Moving Average
|
|
/// For floating point types.
|
|
/// \tparam T The sample value type.
|
|
//***************************************************************************
|
|
template <typename T>
|
|
class pseudo_moving_average<T, 0U, 1U, false, true>
|
|
{
|
|
typedef pseudo_moving_average<T, 0U, 1U, false, true> this_t;
|
|
|
|
public:
|
|
|
|
typedef T value_type;
|
|
typedef private_pseudo_moving_average::add_insert_iterator<this_t> add_insert_iterator;
|
|
|
|
//*************************************************************************
|
|
/// Constructor
|
|
/// \param initial_value The initial value for the average.
|
|
//*************************************************************************
|
|
pseudo_moving_average(const T initial_value, const size_t sample_size)
|
|
: reciprocal_samples_plus_1(T(1.0) / T(sample_size + 1U))
|
|
, average(initial_value)
|
|
{
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Clears the average.
|
|
/// \param initial_value The initial value for the average.
|
|
//*************************************************************************
|
|
void clear(const T initial_value)
|
|
{
|
|
average = initial_value;
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Sets the sample size.
|
|
/// \param sample_size The new sample size.
|
|
//*************************************************************************
|
|
void set_sample_size(const size_t sample_size)
|
|
{
|
|
reciprocal_samples_plus_1 = T(1.0) / (T(sample_size) + T(1));
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Adds a new sample to the average.
|
|
/// \param new_value The value to add.
|
|
//*************************************************************************
|
|
void add(const T new_value)
|
|
{
|
|
average += (new_value - average) * reciprocal_samples_plus_1;
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Gets the current pseudo moving average.
|
|
/// \return The current average.
|
|
//*************************************************************************
|
|
T value() const
|
|
{
|
|
return average;
|
|
}
|
|
|
|
//*************************************************************************
|
|
/// Gets an iterator for input.
|
|
/// \return An iterator.
|
|
//*************************************************************************
|
|
add_insert_iterator input()
|
|
{
|
|
return add_insert_iterator(*this);
|
|
}
|
|
|
|
private:
|
|
|
|
T reciprocal_samples_plus_1; ///< Reciprocal of one greater than the sample size.
|
|
T average; ///< The current pseudo moving average.
|
|
};
|
|
}
|
|
|
|
#endif
|