1 files changed, 74 insertions, 0 deletions
diff --git a/gnuradio-core/src/lib/runtime/gr_block.h b/gnuradio-core/src/lib/runtime/gr_block.h
index c89138bb3..98339080d 100644
--- a/gnuradio-core/src/lib/runtime/gr_block.h
+++ b/gnuradio-core/src/lib/runtime/gr_block.h
@@ -288,6 +288,76 @@ class GR_CORE_API gr_block : public gr_basic_block {
    */
   bool is_set_max_noutput_items();
 
+  /*
+   * Used to expand the vectors that hold the min/max buffer sizes.
+   *
+   * Specifically, when -1 is used, the vectors are just initialized
+   * with 1 value; this is used by the flat_flowgraph to expand when
+   * required to add a new value for new ports on these blocks.
+   */
+  void expand_minmax_buffer(int port) {
+    if((size_t)port >= d_max_output_buffer.size())
+      set_max_output_buffer(port, -1);
+    if((size_t)port >= d_min_output_buffer.size())
+      set_min_output_buffer(port, -1);
+  }
+
+  /*!
+   * \brief Returns max buffer size on output port \p i.
+   */
+  long max_output_buffer(size_t i) {
+    if(i >= d_max_output_buffer.size())
+      throw std::invalid_argument("gr_basic_block::max_output_buffer: port out of range.");
+    return d_max_output_buffer[i];
+  }
+
+  /*!
+   * \brief Sets max buffer size on all output ports.
+   */
+  void set_max_output_buffer(long max_output_buffer) { 
+    for(int i = 0; i < output_signature()->max_streams(); i++) {
+      set_max_output_buffer(i, max_output_buffer);
+    }
+  }
+
+  /*!
+   * \brief Sets max buffer size on output port \p port.
+   */
+  void set_max_output_buffer(int port, long max_output_buffer) {
+    if((size_t)port >= d_max_output_buffer.size())
+      d_max_output_buffer.push_back(max_output_buffer);
+    else
+      d_max_output_buffer[port] = max_output_buffer; 
+  }
+
+  /*!
+   * \brief Returns min buffer size on output port \p i.
+   */
+  long min_output_buffer(size_t i) {
+    if(i >= d_min_output_buffer.size())
+      throw std::invalid_argument("gr_basic_block::min_output_buffer: port out of range.");
+    return d_min_output_buffer[i];
+  }
+
+  /*!
+   * \brief Sets min buffer size on all output ports.
+   */
+  void set_min_output_buffer(long min_output_buffer) {
+    for(int i=0; i<output_signature()->max_streams(); i++) {
+      set_min_output_buffer(i, min_output_buffer);
+    }
+  }
+
+  /*!
+   * \brief Sets min buffer size on output port \p port.
+   */
+  void set_min_output_buffer(int port, long min_output_buffer) {
+    if((size_t)port >= d_min_output_buffer.size())
+      d_min_output_buffer.push_back(min_output_buffer);
+    else
+      d_min_output_buffer[port] = min_output_buffer; 
+  }
+
   // ----------------------------------------------------------------------------
 
  private:
@@ -384,6 +454,10 @@ class GR_CORE_API gr_block : public gr_basic_block {
 			 uint64_t abs_end,
 			 const pmt::pmt_t &key);
 
+  std::vector<long>    d_max_output_buffer;
+  std::vector<long>    d_min_output_buffer;
+
+
   // These are really only for internal use, but leaving them public avoids
   // having to work up an ever-varying list of friend GR_CORE_APIs