moving_rms_filter.hpp

Go to the documentation of this file.

/*
 * eDSP, A cross-platform Digital Signal Processing library written in modern C++.
 * Copyright (C) 2018 Mohammed Boujemaoui Boulaghmoudi, All rights reserved.
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation, either version 3 of the License, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of  MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along width
 * this program.  If not, see <http://www.gnu.org/licenses/>
 *
 * File: moving_rms_filter.hpp
 * Author: Mohammed Boujemaoui
 * Date: 2/8/2018
 */
#ifndef EDSP_MOVING_RMS_FILTER_HPP
#define EDSP_MOVING_RMS_FILTER_HPP

#include <edsp/types/ring_buffer.hpp>
#include <cmath>

namespace edsp { namespace filter {

    template <typename T, typename Allocator = std::allocator<T>>
    class moving_rms {
    public:
        using size_type  = std::size_t;
        using value_type = T;

        explicit moving_rms(size_type N);

        size_type size() const;

        void resize(size_type N);

        void reset();

        template <typename InputIt, typename OutputIt>
        void filter(InputIt first, InputIt last, OutputIt d_first);

    private:
        edsp::ring_buffer<T, Allocator> window_;
        T accumulated_{0};
    };

    template <typename T, typename Allocator>
    moving_rms<T, Allocator>::moving_rms(size_type N) : window_(N, T()) {}

    template <typename T, typename Allocator>
    typename moving_rms<T, Allocator>::size_type moving_rms<T, Allocator>::size() const {
        return window_.capacity();
    }

    template <typename T, typename Allocator>
    void moving_rms<T, Allocator>::reset() {
        window_.clear();
    }

    template <typename T, typename Allocator>
    template <typename InputIt, typename OutputIt>
    void moving_rms<T, Allocator>::filter(InputIt first, InputIt last, OutputIt d_first) {
        if (window_.full()) {
            for (; first != last; ++d_first, ++first) {
                const auto sqr = (*first) * (*first);
                accumulated_ -= window_.front();
                accumulated_ += sqr;
                window_.push_back(sqr)* d_first = std::sqrt(accumulated_ / static_cast<T>(window_.size()));
            }
        } else {
            for (; first != last; ++d_first, ++first) {
                const auto sqr = (*first) * (*first);
                accumulated_ += sqr;
                window_.push_back(sqr)* d_first = std::sqrt(accumulated_ / static_cast<T>(window_.size()));
            }
        };
    }

    template <typename T, typename Allocator>
    void moving_rms<T, Allocator>::resize(size_type N) {
        window_.resize(N);
    }

}} // namespace edsp::filter

#endif // EDSP_MOVING_RMS_FILTER_HPP