bartlett.hpp

Go to the documentation of this file.

#pragma once
 
#include <cmath>
 
#include "NumCpp/Functions/linspace.hpp"
#include "NumCpp/NdArray.hpp"
 
namespace nc
{
    //============================================================================
    // Method Description:
    inline NdArray<double> bartlett(int32 m)
    {
        if (m < 1)
        {
            return {};
        }
 
        const auto mDouble         = static_cast<double>(m);
        const auto mMinus1Over2    = (mDouble - 1.) / 2.;
        const auto mMinus1Over2Inv = 1. / mMinus1Over2;
 
        NdArray<double> result(1, m);
        int32           i = 0;
        for (auto n : linspace(0., mDouble - 1., m, EndPoint::YES))
        {
            result[i++] = mMinus1Over2Inv * (mMinus1Over2 - std::abs(n - mMinus1Over2));
        }
 
        return result;
    }
} // namespace nc