blocks: added gr::blocks::file_source

8 files changed, 328 insertions, 0 deletions

diff --git a/gr-blocks/grc/blocks_block_tree.xml b/gr-blocks/grc/blocks_block_tree.xml
index 94d09e9eb..cceb6c802 100644
--- a/gr-blocks/grc/blocks_block_tree.xml
+++ b/gr-blocks/grc/blocks_block_tree.xml
@@ -29,6 +29,10 @@
 <cat>
  	<name></name> <!-- Blank for Root Name -->
 	<cat>
+	        <name>Sources (New)</name>
+		<block>blocks_file_source</block>
+	</cat>
+	<cat>
 		<name>Math Operations (New) </name>
 		<block>blocks_add_xx</block>
 		<block>blocks_add_const_vxx</block>
diff --git a/gr-blocks/grc/blocks_file_source.xml b/gr-blocks/grc/blocks_file_source.xml
new file mode 100644
index 000000000..753465bb3
--- /dev/null
+++ b/gr-blocks/grc/blocks_file_source.xml
@@ -0,0 +1,74 @@
+<?xml version="1.0"?>
+<!--
+###################################################
+##File Source
+###################################################
+ -->
+<block>
+	<name>File Source</name>
+	<key>blocks_file_source</key>
+	<import>from gnuradio import blocks</import>
+	<make>blocks.file_source($type.size*$vlen, $file, $repeat)</make>
+	<param>
+		<name>File</name>
+		<key>file</key>
+		<value></value>
+		<type>file_open</type>
+	</param>
+	<param>
+		<name>Output Type</name>
+		<key>type</key>
+		<type>enum</type>
+		<option>
+			<name>Complex</name>
+			<key>complex</key>
+			<opt>size:gr.sizeof_gr_complex</opt>
+		</option>
+		<option>
+			<name>Float</name>
+			<key>float</key>
+			<opt>size:gr.sizeof_float</opt>
+		</option>
+		<option>
+			<name>Int</name>
+			<key>int</key>
+			<opt>size:gr.sizeof_int</opt>
+		</option>
+		<option>
+			<name>Short</name>
+			<key>short</key>
+			<opt>size:gr.sizeof_short</opt>
+		</option>
+		<option>
+			<name>Byte</name>
+			<key>byte</key>
+			<opt>size:gr.sizeof_char</opt>
+		</option>
+	</param>
+	<param>
+		<name>Repeat</name>
+		<key>repeat</key>
+		<value>True</value>
+		<type>enum</type>
+		<option>
+			<name>Yes</name>
+			<key>True</key>
+		</option>
+		<option>
+			<name>No</name>
+			<key>False</key>
+		</option>
+	</param>
+	<param>
+		<name>Vec Length</name>
+		<key>vlen</key>
+		<value>1</value>
+		<type>int</type>
+	</param>
+	<check>$vlen &gt; 0</check>
+	<source>
+		<name>out</name>
+		<type>$type</type>
+		<vlen>$vlen</vlen>
+	</source>
+</block>
diff --git a/gr-blocks/include/blocks/CMakeLists.txt b/gr-blocks/include/blocks/CMakeLists.txt
index 366bf1110..ac674528f 100644
--- a/gr-blocks/include/blocks/CMakeLists.txt
+++ b/gr-blocks/include/blocks/CMakeLists.txt
@@ -101,6 +101,7 @@ install(FILES
     complex_to_arg.h
     conjugate_cc.h
     deinterleave.h
+    file_source.h
     float_to_char.h
     float_to_complex.h
     float_to_int.h
diff --git a/gr-blocks/include/blocks/file_source.h b/gr-blocks/include/blocks/file_source.h
new file mode 100644
index 000000000..6207cdf70
--- /dev/null
+++ b/gr-blocks/include/blocks/file_source.h
@@ -0,0 +1,57 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2012 Free Software Foundation, Inc.
+ *
+ * This file is part of GNU Radio
+ *
+ * GNU Radio is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3, or (at your option)
+ * any later version.
+ *
+ * GNU Radio 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifndef INCLUDED_BLOCKS_FILE_SOURCE_H
+#define INCLUDED_BLOCKS_FILE_SOURCE_H
+
+#include <blocks/api.h>
+#include <gr_sync_block.h>
+
+namespace gr {
+  namespace blocks {
+
+    /*!
+     * \brief Read stream from file
+     * \ingroup source_blk
+     */
+    class BLOCKS_API file_source : virtual public gr_sync_block
+    {
+    public:
+
+      // gr::blocks::file_source::sptr
+      typedef boost::shared_ptr<file_source> sptr;
+
+      static sptr make(size_t itemsize, const char *filename, bool repeat = false);
+
+      /*!
+       * \brief seek file to \p seek_point relative to \p whence
+       *
+       * \param seek_point	sample offset in file
+       * \param whence	one of SEEK_SET, SEEK_CUR, SEEK_END (man fseek)
+       */
+      virtual bool seek(long seek_point, int whence) = 0;
+    };
+
+  } /* namespace blocks */
+} /* namespace gr */
+
+#endif /* INCLUDED_BLOCKS_FILE_SOURCE_H */
diff --git a/gr-blocks/lib/CMakeLists.txt b/gr-blocks/lib/CMakeLists.txt
index 7db4d118c..b06f46675 100644
--- a/gr-blocks/lib/CMakeLists.txt
+++ b/gr-blocks/lib/CMakeLists.txt
@@ -135,6 +135,7 @@ list(APPEND gr_blocks_sources
     complex_to_arg_impl.cc
     conjugate_cc_impl.cc
     deinterleave_impl.cc
+    file_source_impl.cc
     float_to_char_impl.cc
     float_to_complex_impl.cc
     float_array_to_int.cc
diff --git a/gr-blocks/lib/file_source_impl.cc b/gr-blocks/lib/file_source_impl.cc
new file mode 100644
index 000000000..0d20827b7
--- /dev/null
+++ b/gr-blocks/lib/file_source_impl.cc
@@ -0,0 +1,136 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2012 Free Software Foundation, Inc.
+ *
+ * This file is part of GNU Radio
+ *
+ * GNU Radio is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3, or (at your option)
+ * any later version.
+ *
+ * GNU Radio 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include "file_source_impl.h"
+#include <gr_io_signature.h>
+#include <cstdio>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <stdexcept>
+#include <stdio.h>
+
+// win32 (mingw/msvc) specific
+#ifdef HAVE_IO_H
+#include <io.h>
+#endif
+#ifdef O_BINARY
+#define	OUR_O_BINARY O_BINARY
+#else
+#define	OUR_O_BINARY 0
+#endif
+// should be handled via configure
+#ifdef O_LARGEFILE
+#define	OUR_O_LARGEFILE	O_LARGEFILE
+#else
+#define	OUR_O_LARGEFILE 0
+#endif
+
+namespace gr {
+  namespace blocks {
+
+    file_source::sptr file_source::make(size_t itemsize, const char *filename, bool repeat)
+    {
+      return gnuradio::get_initial_sptr(new file_source_impl(itemsize, filename, repeat));
+    }
+
+    file_source_impl::file_source_impl(size_t itemsize, const char *filename, bool repeat)
+      : gr_sync_block("file_source",
+		      gr_make_io_signature(0, 0, 0),
+		      gr_make_io_signature(1, 1, itemsize)),
+	d_itemsize(itemsize), d_fp (0), d_repeat(repeat)
+    {
+      // we use "open" to use to the O_LARGEFILE flag
+
+      int fd;
+
+      if ((fd = open (filename, O_RDONLY | OUR_O_LARGEFILE | OUR_O_BINARY)) < 0){
+	perror (filename);
+	throw std::runtime_error ("can't open file");
+      }
+
+      if ((d_fp = fdopen (fd, "rb")) == NULL) {
+	perror(filename);
+	throw std::runtime_error("can't open file");
+      }
+    }
+
+    file_source_impl::~file_source_impl()
+    {
+      fclose ((FILE *) d_fp);
+    }
+    
+    bool
+    file_source_impl::seek(long seek_point, int whence)
+    {
+      return fseek ((FILE *) d_fp, seek_point *d_itemsize, whence) == 0;
+    }
+
+    int
+    file_source_impl::work(int noutput_items,
+			   gr_vector_const_void_star &input_items,
+			   gr_vector_void_star &output_items)
+    {
+      char *o = (char *) output_items[0];
+      int i;
+      int size = noutput_items;
+
+      while (size) {
+	i = fread(o, d_itemsize, size, (FILE *) d_fp);
+
+	size -= i;
+	o += i * d_itemsize;
+
+	if (size == 0)		// done
+	  break;
+
+	if (i > 0)			// short read, try again
+	  continue;
+
+	// We got a zero from fread.  This is either EOF or error.  In
+	// any event, if we're in repeat mode, seek back to the beginning
+	// of the file and try again, else break
+	
+	if (!d_repeat)
+	  break;
+
+	if (fseek ((FILE *) d_fp, 0, SEEK_SET) == -1) {
+	  fprintf(stderr, "[%s] fseek failed\n", __FILE__);
+	  exit(-1);
+	}
+      }
+
+      if (size > 0) {	     		// EOF or error
+	if (size == noutput_items)      // we didn't read anything; say we're done
+	  return -1;
+	return noutput_items - size;	// else return partial result
+      }
+
+      return noutput_items;
+    }
+
+  } /* namespace blocks */
+} /* namespace gr */
diff --git a/gr-blocks/lib/file_source_impl.h b/gr-blocks/lib/file_source_impl.h
new file mode 100644
index 000000000..c35d563d1
--- /dev/null
+++ b/gr-blocks/lib/file_source_impl.h
@@ -0,0 +1,52 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2012 Free Software Foundation, Inc.
+ *
+ * This file is part of GNU Radio
+ *
+ * GNU Radio is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3, or (at your option)
+ * any later version.
+ *
+ * GNU Radio 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifndef INCLUDED_BLOCKS_FILE_SOURCE_IMPL_H
+#define INCLUDED_BLOCKS_FILE_SOURCE_IMPL_H
+
+#include <blocks/file_source.h>
+
+namespace gr {
+  namespace blocks {
+
+    class BLOCKS_API file_source_impl : public file_source
+    {
+    private:
+      size_t  d_itemsize;
+      void   *d_fp;
+      bool    d_repeat;
+      
+    public:
+      file_source_impl(size_t itemsize, const char *filename, bool repeat);
+      ~file_source_impl();
+
+      bool seek(long seek_point, int whence);
+
+      int work(int noutput_items,
+	       gr_vector_const_void_star &input_items,
+	       gr_vector_void_star &output_items);
+    };
+
+  } /* namespace blocks */
+} /* namespace gr */
+
+#endif /* INCLUDED_BLOCKS_FILE_SOURCE_IMPL_H */
diff --git a/gr-blocks/swig/blocks_swig.i b/gr-blocks/swig/blocks_swig.i
index 82ff0cc84..45b259498 100644
--- a/gr-blocks/swig/blocks_swig.i
+++ b/gr-blocks/swig/blocks_swig.i
@@ -61,6 +61,7 @@
 #include "blocks/divide_ss.h"
 #include "blocks/divide_ii.h"
 #include "blocks/divide_cc.h"
+#include "blocks/file_source.h"
 #include "blocks/float_to_char.h"
 #include "blocks/float_to_complex.h"
 #include "blocks/float_to_int.h"
@@ -144,6 +145,7 @@
 %include "blocks/complex_to_arg.h"
 %include "blocks/conjugate_cc.h"
 %include "blocks/deinterleave.h"
+%include "blocks/file_source.h"
 %include "blocks/divide_ff.h"
 %include "blocks/divide_ss.h"
 %include "blocks/divide_ii.h"
@@ -234,6 +236,7 @@ GR_SWIG_BLOCK_MAGIC2(blocks, divide_ff);
 GR_SWIG_BLOCK_MAGIC2(blocks, divide_ss);
 GR_SWIG_BLOCK_MAGIC2(blocks, divide_ii);
 GR_SWIG_BLOCK_MAGIC2(blocks, divide_cc);
+GR_SWIG_BLOCK_MAGIC2(blocks, file_source);
 GR_SWIG_BLOCK_MAGIC2(blocks, float_to_char);
 GR_SWIG_BLOCK_MAGIC2(blocks, float_to_complex);
 GR_SWIG_BLOCK_MAGIC2(blocks, float_to_int);