#!/usr/bin/env python

#

# Copyright 2007 Free Software Foundation, Inc.

# 

# This file is part of GNU Radio

# 

# GNU Radio 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, or (at your option)

# any later version.

# 

# GNU Radio 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 with GNU Radio; see the file COPYING.  If not, write to

# the Free Software Foundation, Inc., 51 Franklin Street,

# Boston, MA 02110-1301, USA.

# 

from gnuradio import gr, gr_unittest

class test_moving_average_vector (gr_unittest.TestCase):

    def setUp (self):

        self.tb = gr.top_block ()

    def tearDown (self):

        self.tb = None

    def test_001(self):

        src_data = (1,1,0, 2,2,0, 3,3,0, 4,4,0, 5,5,0)

        vlen = 3

        MAlen = 3

        expected_result = (1.0/3,1.0/3,0, 1,1,0, 2,2,0, 3,3,0, 4,4,0)

        src = gr.vector_source_f(src_data, False, vlen)

        MA = gr.moving_average_vff(MAlen, vlen, 1.0/3)

        dst = gr.vector_sink_f(vlen)

        self.tb.connect(src, MA, dst)

        self.tb.run()

        result_data = dst.data()

        self.assertFloatTuplesAlmostEqual(expected_result, result_data)

if __name__ == '__main__':

    gr_unittest.main ()

	qa_vector_sink_source.py	\

	qa_moving_average_vXX.py

/* -*- c++ -*- */

/*

 * Copyright 2008 Free Software Foundation, Inc.

 * 

 * This file is part of GNU Radio

 * 

 * GNU Radio 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, or (at your option)

 * any later version.

 * 

 * GNU Radio 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 with GNU Radio; see the file COPYING.  If not, write to

 * the Free Software Foundation, Inc., 51 Franklin Street,

 * Boston, MA 02110-1301, USA.

 */

// @WARNING@

GR_SWIG_BLOCK_MAGIC(gr,@BASE_NAME@);

@SPTR_NAME@ gr_make_@BASE_NAME@ (int length, int vlen, @O_TYPE@ scale, int max_iter=4096);

class @NAME@ : public gr_sync_block

{

private:

  @NAME@ ();

};

/* -*- c++ -*- */

/*

 * Copyright 2008 Free Software Foundation, Inc.

 * 

 * This file is part of GNU Radio

 * 

 * GNU Radio 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, or (at your option)

 * any later version.

 * 

 * GNU Radio 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 with GNU Radio; see the file COPYING.  If not, write to

 * the Free Software Foundation, Inc., 51 Franklin Street,

 * Boston, MA 02110-1301, USA.

 */

// @WARNING@

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <@NAME@.h>

#include <gr_io_signature.h>

#include <vector>

@SPTR_NAME@ 

gr_make_@BASE_NAME@ (int length, int vlen, @O_TYPE@ scale, int max_iter)

{

  return @SPTR_NAME@ (new @NAME@ (length, vlen, scale, max_iter));

}

@NAME@::@NAME@ (int length, int vlen, @O_TYPE@ scale, int max_iter)

  : gr_sync_block ("@BASE_NAME@",

		   gr_make_io_signature (1, 1, sizeof (@I_TYPE@) * vlen),

		   gr_make_io_signature (1, 1, sizeof (@O_TYPE@) * vlen)),

    d_length(length),

    d_vlen(vlen),

    d_scale(scale),

    d_max_iter(max_iter)

{

  set_history(length);

}

@NAME@::~@NAME@ ()

{

}

int

@NAME@::work (int noutput_items,

	      gr_vector_const_void_star &input_items,

	      gr_vector_void_star &output_items)

{

  const @I_TYPE@ *in = (const @I_TYPE@ *) input_items[0];

  @O_TYPE@ *out = (@O_TYPE@ *) output_items[0];

  std::vector<@I_TYPE@> sum(d_vlen, 0);

  int num_iter = (noutput_items>d_max_iter) ? d_max_iter : noutput_items;

  for (int i = 0; i < d_length-1 ; i++) {

    const @I_TYPE@ *in_vec = &in[i * d_vlen];

    for (int k = 0; k < d_vlen; k++) {

      sum[k] += in_vec[k];

    }

  }

  for (int i = 0; i < num_iter; i++) {

    const @I_TYPE@ *in_vec_old = &in[i * d_vlen];

    const @I_TYPE@ *in_vec_new = &in[(i+d_length-1) * d_vlen];

    @O_TYPE@ *out_vec = &out[i * d_vlen];

    for (int k = 0; k < d_vlen; k++) {

      sum[k] += in_vec_new[k];

      out_vec[k] = sum[k] * d_scale;

      sum[k] -= in_vec_old[k];

    }

  }

  return num_iter;

}

	gr_moving_average_vcc.h \

	gr_moving_average_vff.h \

	gr_moving_average_vii.h \

	gr_moving_average_vss.h \

	gr_moving_average_vcc.i \

	gr_moving_average_vff.i \

	gr_moving_average_vii.i \

	gr_moving_average_vss.i \

	gr_moving_average_vcc.cc \

	gr_moving_average_vff.cc \

	gr_moving_average_vii.cc \

	gr_moving_average_vss.cc \

/* -*- c++ -*- */

/*

 * Copyright 2008 Free Software Foundation, Inc.

 * 

 * This file is part of GNU Radio

 * 

 * GNU Radio 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, or (at your option)

 * any later version.

 * 

 * GNU Radio 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 with GNU Radio; see the file COPYING.  If not, write to

 * the Free Software Foundation, Inc., 51 Franklin Street,

 * Boston, MA 02110-1301, USA.

 */

// @WARNING@

#ifndef @GUARD_NAME@

#define @GUARD_NAME@

#include <gr_sync_block.h>

class @NAME@;

typedef boost::shared_ptr<@NAME@> @SPTR_NAME@;

@SPTR_NAME@ gr_make_@BASE_NAME@ (int length, int vlen, @O_TYPE@ scale, int max_iter = 4096);

/*!

 * \brief output is the moving sum of the last N samples, scaled by the \p scale factor.

 *

 * The moving average of the vectors is calculated per element.

 *

 * \p max_iter limits how long we go without flushing the accumulator. This is necessary

 * to avoid numerical instability for float and complex.

 *

 * \ingroup filter

 */

class @NAME@ : public gr_sync_block

{

private:

  friend @SPTR_NAME@ gr_make_@BASE_NAME@(int length, int vlen, @O_TYPE@ scale, int max_iter);

  @NAME@ (int length, int vlen, @O_TYPE@ scale, int max_iter = 4096);

  int d_length;

  int d_vlen;

  @O_TYPE@ d_scale;

  int d_max_iter;

public:

  ~@NAME@ ();

  int work (int noutput_items,

	    gr_vector_const_void_star &input_items,

	    gr_vector_void_star &output_items);

};

#endif /* @GUARD_NAME@ */

    'gr_moving_average_vXX'

	gr_moving_average_XX.i.t	\

	gr_moving_average_vXX.cc.t	\

	gr_moving_average_vXX.h.t	\

	gr_moving_average_vXX.i.t