git.gag.com Git - debian/gnuradio/blob - gr-atsc/src/lib/atsci_equalizer_lms2.h

   1 /* -*- c++ -*- */
   2 /*
   3  * Copyright 2002 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 _ATSC_EQUALIZER_LMS2_H_
  24 #define _ATSC_EQUALIZER_LMS2_H_
  25
  26 #include <atsci_equalizer.h>
  27 #include <vector>
  28 #include <stdio.h>
  29
  30 class atsci_equalizer_lms2 : public atsci_equalizer
  31 {
  32 public:
  33   atsci_equalizer_lms2 ();
  34   virtual ~atsci_equalizer_lms2 ();
  35
  36   virtual void reset ();
  37   virtual int ntaps () const;
  38   virtual int npretaps () const;
  39
  40 protected:
  41   FILE *trainingfile;
  42   virtual void filter_normal (const float *input_samples,
  43                               float *output_samples,
  44                               int   nsamples);
  45
  46   virtual void filter_data_seg_sync (const float *input_samples,
  47                                      float *output_samples,
  48                                      int   nsamples,
  49                                      int   offset);
  50
  51   virtual void filter_field_sync (const float *input_samples,
  52                                   float *output_samples,
  53                                   int   nsamples,
  54                                   int   offset,
  55                                   int   which_field);
  56
  57 private:
  58   std::vector<double>   d_taps_ff;
  59   std::vector<double>   d_taps_fb;
  60   std::vector<float>    d_old_output;
  61
  62   int d_output_ptr;
  63
  64   void filterN (const float *input_samples,
  65                 float *output_samples,
  66                 int nsamples);
  67
  68   void adaptN (const float *input_samples,
  69                const float *training_pattern,
  70                float *output_samples,
  71                int    nsamples);
  72
  73   float filter1 (const float input[]);
  74   float adapt1 (const float input[], float ideal_output);
  75
  76 };
  77
  78
  79 #endif /* _ATSC_EQUALIZER_LMS2_H_ */