Imported Upstream version 3.2.2
[debian/gnuradio] / gnuradio-core / src / lib / filter / gr_rotator.h
1 /* -*- c++ -*- */
2 /*
3  * Copyright 2003,2008 Free Software Foundation, Inc.
4  * 
5  * This file is part of GNU Radio
6  * 
7  * GNU Radio is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3, or (at your option)
10  * any later version.
11  * 
12  * GNU Radio is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  * 
17  * You should have received a copy of the GNU General Public License
18  * along with GNU Radio; see the file COPYING.  If not, write to
19  * the Free Software Foundation, Inc., 51 Franklin Street,
20  * Boston, MA 02110-1301, USA.
21  */
22
23 #ifndef _GR_ROTATOR_H_
24 #define _GR_ROTATOR_H_
25
26 #include <gr_complex.h>
27
28 class gr_rotator {
29   gr_complex    d_phase;
30   gr_complex    d_phase_incr;
31   unsigned int  d_counter;
32
33  public:
34   gr_rotator () : d_phase (1), d_phase_incr (1), d_counter(0) { }
35
36   void set_phase (gr_complex phase)     { d_phase = phase / abs(phase); }
37   void set_phase_incr (gr_complex incr) { d_phase_incr = incr / abs(incr); }
38
39   gr_complex rotate (gr_complex in){
40     d_counter++;
41
42     gr_complex z = in * d_phase;    // rotate in by phase
43     d_phase *= d_phase_incr;        // incr our phase (complex mult == add phases)
44
45     if ((d_counter % 512) == 0)
46       d_phase /= abs(d_phase);      // Normalize to ensure multiplication is rotation
47
48     return z;
49   }
50
51 };
52
53 #endif /* _GR_ROTATOR_H_ */