1 files changed, 52 insertions, 0 deletions
diff --git a/lib/block_allocator.cpp b/lib/block_allocator.cpp
new file mode 100644
index 0000000..b3f8c4d
--- /dev/null
+++ b/lib/block_allocator.cpp
@@ -0,0 +1,52 @@
+//
+// Copyright 2012 Josh Blum
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Lesser General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public License
+// along with io_sig program.  If not, see <http://www.gnu.org/licenses/>.
+
+#include "element_impl.hpp"
+#include <boost/bind.hpp>
+
+using namespace gnuradio;
+
+//TODO will need more complicated later
+static void simple_allocator(
+    const tsbe::BufferToken &tok,
+    const size_t num_bytes,
+    const size_t num_buffs
+){
+    for (size_t i = 0; i < num_buffs; i++)
+    {
+        tsbe::BufferConfig config;
+        config.memory = NULL;
+        config.length = num_bytes;
+        config.token = tok;
+        tsbe::Buffer buff(config);
+        //buffer derefs here and the token messages it back to the block
+    }
+}
+
+void ElementImpl::handle_allocation(const tsbe::TaskInterface &task_iface)
+{
+    //allocate output buffers which will also wake up the task
+    const size_t num_outputs = task_iface.get_num_outputs();
+    for (size_t i = 0; i < num_outputs; i++)
+    {
+        size_t items = this->hint;
+        if (items == 0) items = 1024;
+        items = std::max(items, this->output_multiple_items[i]);
+
+        const size_t bytes = items * this->output_items_sizes[i];
+        this->block.set_output_port_allocator(i, boost::bind(&simple_allocator, _1, bytes, 2));
+    }
+}