atsc: Make equililzer taps accessible in python.

4 files changed, 14 insertions, 0 deletions

diff --git a/gr-atsc/src/lib/atsc.i b/gr-atsc/src/lib/atsc.i
index 6e5662ae6..6b83a2315 100644
--- a/gr-atsc/src/lib/atsc.i
+++ b/gr-atsc/src/lib/atsc.i
@@ -225,6 +225,7 @@ class atsc_equalizer : public gr_sync_block
 
 public:
   void reset();
+  std::vector<double> taps();
 };
 
 // ----------------------------------------------------------------
diff --git a/gr-atsc/src/lib/atsc_equalizer.h b/gr-atsc/src/lib/atsc_equalizer.h
index 32b449b62..aca5e63b4 100644
--- a/gr-atsc/src/lib/atsc_equalizer.h
+++ b/gr-atsc/src/lib/atsc_equalizer.h
@@ -25,6 +25,7 @@
 #include <atsc_api.h>
 #include <gr_sync_block.h>
 #include <atsci_equalizer.h>
+#include <vector>
 
 class atsc_equalizer;
 typedef boost::shared_ptr<atsc_equalizer> atsc_equalizer_sptr;
@@ -47,6 +48,10 @@ class ATSC_API atsc_equalizer : public gr_sync_block
 public:
   void forecast (int noutput_items, gr_vector_int &ninput_items_required);
 
+  std::vector<double> taps() {
+    return d_equalizer->taps();
+  }
+
   int work (int noutput_items,
 	    gr_vector_const_void_star &input_items,
 	    gr_vector_void_star &output_items);
diff --git a/gr-atsc/src/lib/atsci_equalizer.h b/gr-atsc/src/lib/atsci_equalizer.h
index a0a1fc674..b0c243b51 100644
--- a/gr-atsc/src/lib/atsci_equalizer.h
+++ b/gr-atsc/src/lib/atsci_equalizer.h
@@ -25,6 +25,7 @@
 
 #include <atsc_api.h>
 #include <atsci_syminfo.h>
+#include <vector>
 
 /*!
  * \brief abstract base class for ATSC equalizer
@@ -54,6 +55,10 @@ public:
   atsci_equalizer ();
   virtual ~atsci_equalizer ();
 
+  virtual std::vector<double> taps () {
+    return std::vector<double>();
+  }
+
   // MANIPULATORS
 
   /*!
diff --git a/gr-atsc/src/lib/atsci_equalizer_lms.h b/gr-atsc/src/lib/atsci_equalizer_lms.h
index 68aa34ca5..eac72f75b 100644
--- a/gr-atsc/src/lib/atsci_equalizer_lms.h
+++ b/gr-atsc/src/lib/atsci_equalizer_lms.h
@@ -37,6 +37,9 @@ public:
   virtual void reset ();
   virtual int ntaps () const;
   virtual int npretaps () const;
+  std::vector<double> taps () {
+    return d_taps;
+  }
 
 protected:
   FILE *trainingfile;