X-Git-Url: http://git.code-monkey.de/?a=blobdiff_plain;ds=inline;f=pv%2Fdevices%2Finputfile.cpp;h=58c60d700bbd25375222b20f09c7f35a41097e3a;hb=d93ac5e8770e9e5c7bf64d0ed7961357e04370ba;hp=13af3db9cd85c791935da6c80bf1a14220d492a0;hpb=4d6c6ea3e6b069787c270d4911083dae05eae4c6;p=pulseview.git
diff --git a/pv/devices/inputfile.cpp b/pv/devices/inputfile.cpp
index 13af3db..58c60d7 100644
--- a/pv/devices/inputfile.cpp
+++ b/pv/devices/inputfile.cpp
@@ -14,8 +14,7 @@
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+ * along with this program; if not, see .
*/
#include
@@ -25,74 +24,104 @@
#include "inputfile.hpp"
+using std::map;
+using std::shared_ptr;
+using std::streamsize;
+using std::string;
+using std::ifstream;
+using std::ios;
+
namespace pv {
namespace devices {
-const std::streamsize InputFile::BufferSize = 16384;
+const streamsize InputFile::BufferSize = 16384;
-InputFile::InputFile(const std::shared_ptr &context,
- const std::string &file_name,
- std::shared_ptr format,
- const std::map &options) :
+InputFile::InputFile(const shared_ptr &context,
+ const string &file_name,
+ shared_ptr format,
+ const map &options) :
File(file_name),
context_(context),
- input_(format->create_input(options)),
- interrupt_(false) {
- if (!input_)
- throw QString("Failed to create input");
+ format_(format),
+ options_(options),
+ interrupt_(false)
+{
}
-void InputFile::open() {
+void InputFile::open()
+{
if (session_)
close();
+ else
+ session_ = context_->create_session();
- session_ = context_->create_session();
+ input_ = format_->create_input(options_);
+
+ if (!input_)
+ throw QString("Failed to create input");
+
+ // open() should add the input device to the session but
+ // we can't open the device without sending some data first
+ f = new ifstream(file_name_, ios::binary);
+
+ char buffer[BufferSize];
+ f->read(buffer, BufferSize);
+ const streamsize size = f->gcount();
+ if (size == 0)
+ return;
+
+ input_->send(buffer, size);
+
+ try {
+ device_ = input_->device();
+ } catch (sigrok::Error) {
+ return;
+ }
+
+ session_->add_device(device_);
}
-void InputFile::close() {
+void InputFile::close()
+{
if (session_)
session_->remove_devices();
}
-void InputFile::start() {
+void InputFile::start()
+{
}
-void InputFile::run() {
+void InputFile::run()
+{
char buffer[BufferSize];
- bool need_device = true;
- assert(session_);
- assert(input_);
+ if (!f) {
+ // Previous call to run() processed the entire file already
+ f = new ifstream(file_name_, ios::binary);
+ input_->reset();
+ }
interrupt_ = false;
- std::ifstream f(file_name_);
- while (!interrupt_ && f) {
- f.read(buffer, BufferSize);
- const std::streamsize size = f.gcount();
+ while (!interrupt_ && !f->eof()) {
+ f->read(buffer, BufferSize);
+ const streamsize size = f->gcount();
if (size == 0)
break;
input_->send(buffer, size);
- if (need_device) {
- try {
- device_ = input_->device();
- } catch (sigrok::Error) {
- break;
- }
-
- session_->add_device(device_);
- need_device = false;
- }
-
if (size != BufferSize)
break;
}
input_->end();
+
+ delete f;
+ f = nullptr;
}
-void InputFile::stop() {
+void InputFile::stop()
+{
interrupt_ = true;
}