]> sigrok.org Git - pulseview.git/blob - pv/devices/inputfile.cpp
InputFile: Don't try to create device twice
[pulseview.git] / pv / devices / inputfile.cpp
1 /*
2  * This file is part of the PulseView project.
3  *
4  * Copyright (C) 2015 Joel Holdsworth <joel@airwebreathe.org.uk>
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
19  */
20
21 #include <cassert>
22 #include <fstream>
23
24 #include <QString>
25
26 #include "inputfile.hpp"
27
28 namespace pv {
29 namespace devices {
30
31 const std::streamsize InputFile::BufferSize = 16384;
32
33 InputFile::InputFile(const std::shared_ptr<sigrok::Context> &context,
34         const std::string &file_name,
35         std::shared_ptr<sigrok::InputFormat> format,
36         const std::map<std::string, Glib::VariantBase> &options) :
37         File(file_name),
38         context_(context),
39         format_(format),
40         options_(options),
41         interrupt_(false)
42 {
43 }
44
45 void InputFile::open()
46 {
47         if (session_)
48                 close();
49         else
50                 session_ = context_->create_session();
51 }
52
53 void InputFile::close()
54 {
55         if (session_)
56                 session_->remove_devices();
57 }
58
59 void InputFile::start()
60 {
61 }
62
63 void InputFile::run()
64 {
65         char buffer[BufferSize];
66         bool need_device = true;
67
68         assert(session_);
69
70         input_ = format_->create_input(options_);
71
72         if (!input_)
73                 throw QString("Failed to create input");
74
75         interrupt_ = false;
76         std::ifstream f(file_name_, std::ios::binary);
77         while (!interrupt_ && f) {
78                 f.read(buffer, BufferSize);
79                 const std::streamsize size = f.gcount();
80                 if (size == 0)
81                         break;
82
83                 input_->send(buffer, size);
84
85                 if (need_device) {
86                         try {
87                                 device_ = input_->device();
88                         } catch (sigrok::Error) {
89                                 break;
90                         }
91
92                         session_->remove_devices(); // Remove instance from previous run
93                         session_->add_device(device_);
94                         need_device = false;
95                 }
96
97                 if (size != BufferSize)
98                         break;
99         }
100
101         input_->end();
102 }
103
104 void InputFile::stop()
105 {
106         interrupt_ = true;
107 }
108
109 } // namespace devices
110 } // namespace pv