|| || || || || || || ||

---

Avg.h

Go to the documentation of this file.

#ifndef Impala_Core_Vector_Avg_h
#define Impala_Core_Vector_Avg_h

#include "Core/Vector/VectorSet.h"
#include "Core/Vector/AddAssign.h"
#include "Core/Vector/DivAssign.h"

namespace Impala
{
namespace Core
{
namespace Vector
{


template <class ArrayT>
inline void
Avg(typename VectorSet<ArrayT>::VectorT* dst, VectorSet<ArrayT>* src,
    int startIdx, int stepSize, int nrSteps)
{
    typedef typename VectorSet<ArrayT>::VectorT VectorT;

    if (!src->HasConstVectorSize())
    {
        std::cout << "Vector::Avg: works for constant vector size only" << std::endl;
        return;
    }
    VectorT res(src->GetVectorLength(0));
    res = 0;
    int endIdx = startIdx + nrSteps * stepSize;
    for (int i=startIdx ; i<endIdx ; i+=stepSize)
        AddAssign(res, src->GetVector(i, true));
    typename VectorSet<ArrayT>::ElemT factor = nrSteps;
    DivAssign(res, factor);
    if (!dst)
        dst = new VectorT(res);
    else
        *dst = res;
}

template <class ArrayT>
inline void
Avg(typename VectorSet<ArrayT>::VectorT* dst, VectorSet<ArrayT>* src, int size,
    bool* filter)
{
    typedef typename VectorSet<ArrayT>::VectorT VectorT;

    if (!src->HasConstVectorSize())
    {
        std::cout << "Vector::Avg: works for constant vector size only" << std::endl;
        return;
    }
    VectorT res(src->GetVectorLength(0));
    res = 0;
    typename VectorSet<ArrayT>::ElemT factor = 0;
    for (int i=0 ; i<size ; i++)
    {
        if (!filter || filter[i])
        {
            AddAssign(res, src->GetVector(i, true));
            factor++;
        }
    }
    DivAssign(res, factor);
    if (!dst)
        dst = new VectorT(res);
    else
        *dst = res;
}

} // namespace Vector
} // namespace Core
} // namespace Impala

#endif

---