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         input_ = format_->create_input(options_);
  53
  54         if (!input_)
  55                 throw QString("Failed to create input");
  56
  57         // open() should add the input device to the session but
  58         // we can't open the device without sending some data first
  59         f = new std::ifstream(file_name_, std::ios::binary);
  60
  61         char buffer[BufferSize];
  62         f->read(buffer, BufferSize);
  63         const std::streamsize size = f->gcount();
  64         if (size == 0)
  65                 return;
  66
  67         input_->send(buffer, size);
  68
  69         try {
  70                 device_ = input_->device();
  71         } catch (sigrok::Error) {
  72                 return;
  73         }
  74
  75         session_->add_device(device_);
  76 }
  77
  78 void InputFile::close()
  79 {
  80         if (session_)
  81                 session_->remove_devices();
  82 }
  83
  84 void InputFile::start()
  85 {
  86 }
  87
  88 void InputFile::run()
  89 {
  90         char buffer[BufferSize];
  91
  92         if (!f) {
  93                 // Previous call to run() processed the entire file already
  94                 f = new std::ifstream(file_name_, std::ios::binary);
  95                 input_->reset();
  96         }
  97
  98         interrupt_ = false;
  99         while (!interrupt_ && !f->eof()) {
 100                 f->read(buffer, BufferSize);
 101                 const std::streamsize size = f->gcount();
 102                 if (size == 0)
 103                         break;
 104
 105                 input_->send(buffer, size);
 106
 107                 if (size != BufferSize)
 108                         break;
 109         }
 110
 111         input_->end();
 112
 113         delete f;
 114         f = nullptr;
 115 }
 116
 117 void InputFile::stop()
 118 {
 119         interrupt_ = true;
 120 }
 121
 122 } // namespace devices
 123 } // namespace pv