merging branch trondeau/ofdm r7971:8143. Fixing subcarriers and other issues to impro...
[debian/gnuradio] / gnuradio-core / src / lib / general / gr_ofdm_frame_sink.h
1 /* -*- c++ -*- */
2 /*
3  * Copyright 2007 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 INCLUDED_GR_OFDM_FRAME_SINK_H
24 #define INCLUDED_GR_OFDM_FRAME_SINK_H
25
26 #include <gr_sync_block.h>
27 #include <gr_msg_queue.h>
28
29 class gr_ofdm_frame_sink;
30 typedef boost::shared_ptr<gr_ofdm_frame_sink> gr_ofdm_frame_sink_sptr;
31
32 gr_ofdm_frame_sink_sptr 
33 gr_make_ofdm_frame_sink (const std::vector<gr_complex> &sym_position, 
34                          const std::vector<unsigned char> &sym_value_out,
35                          gr_msg_queue_sptr target_queue, unsigned int occupied_tones,
36                          float phase_gain=0.25, float freq_gain=0.25*0.25/4.0);
37
38 /*!
39  * \brief Takes an OFDM symbol in, demaps it into bits of 0's and 1's, packs
40  * them into packets, and sends to to a message queue sink.
41  * \ingroup sink
42  * NOTE: The mod input parameter simply chooses a pre-defined demapper/slicer. Eventually,
43  * we want to be able to pass in a reference to an object to do the demapping and slicing
44  * for a given modulation type.
45  */
46 class gr_ofdm_frame_sink : public gr_sync_block
47 {
48   friend gr_ofdm_frame_sink_sptr 
49   gr_make_ofdm_frame_sink (const std::vector<gr_complex> &sym_position, 
50                            const std::vector<unsigned char> &sym_value_out,
51                            gr_msg_queue_sptr target_queue, unsigned int occupied_tones,
52                            float phase_gain, float freq_gain);
53
54  private:
55   enum state_t {STATE_SYNC_SEARCH, STATE_HAVE_SYNC, STATE_HAVE_HEADER};
56
57   static const int MAX_PKT_LEN    = 4096;
58   static const int HEADERBYTELEN   = 4;
59
60   gr_msg_queue_sptr  d_target_queue;            // where to send the packet when received
61   state_t            d_state;
62   unsigned int       d_header;                  // header bits
63   int                d_headerbytelen_cnt;       // how many so far
64
65   unsigned char      *d_bytes_out;              // hold the current bytes produced by the demapper    
66
67   unsigned int       d_occupied_carriers;
68   unsigned int       d_byte_offset;
69   unsigned int       d_partial_byte;
70
71   unsigned char      d_packet[MAX_PKT_LEN];     // assembled payload
72   int                d_packetlen;               // length of packet
73   int                d_packet_whitener_offset;  // offset into whitener string to use
74   int                d_packetlen_cnt;           // how many so far
75
76   gr_complex * d_derotated_output;  // Pointer to output stream to send deroated symbols out
77
78   std::vector<gr_complex>    d_sym_position;
79   std::vector<unsigned char> d_sym_value_out;
80   std::vector<gr_complex>    d_dfe;
81   unsigned int d_nbits;
82
83   unsigned char d_resid;
84   unsigned int d_nresid;
85   float d_phase;
86   float d_freq;
87   float d_phase_gain;
88   float d_freq_gain;
89   float d_eq_gain;
90
91   std::vector<int> d_subcarrier_map;
92
93  protected:
94   gr_ofdm_frame_sink(const std::vector<gr_complex> &sym_position, 
95                      const std::vector<unsigned char> &sym_value_out,
96                      gr_msg_queue_sptr target_queue, unsigned int occupied_tones,
97                      float phase_gain, float freq_gain);
98
99   void enter_search();
100   void enter_have_sync();
101   void enter_have_header();
102   
103   bool header_ok()
104   {
105     // confirm that two copies of header info are identical
106     return ((d_header >> 16) ^ (d_header & 0xffff)) == 0;
107   }
108   
109   unsigned char slicer(const gr_complex x);
110   unsigned int demapper(const gr_complex *in,
111                         unsigned char *out);
112
113   bool set_sym_value_out(const std::vector<gr_complex> &sym_position, 
114                          const std::vector<unsigned char> &sym_value_out);
115
116  public:
117   ~gr_ofdm_frame_sink();
118
119   int work(int noutput_items,
120            gr_vector_const_void_star &input_items,
121            gr_vector_void_star &output_items);
122 };
123
124 #endif /* INCLUDED_GR_OFDM_FRAME_SINK_H */