pv/storesession.hpp

   1 /*
   2  * This file is part of the PulseView project.
   3  *
   4  * Copyright (C) 2014 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, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 #ifndef PULSEVIEW_PV_STORESESSION_HPP
  21 #define PULSEVIEW_PV_STORESESSION_HPP
  22
  23 #include <stdint.h>
  24
  25 #include <atomic>
  26 #include <fstream>
  27 #include <map>
  28 #include <memory>
  29 #include <mutex>
  30 #include <string>
  31 #include <thread>
  32
  33 #include <glibmm/variant.h>
  34
  35 #include <QObject>
  36
  37 namespace sigrok {
  38 class Output;
  39 class OutputFormat;
  40 }
  41
  42 namespace pv {
  43
  44 class Session;
  45
  46 namespace data {
  47 class SignalBase;
  48 class AnalogSegment;
  49 class LogicSegment;
  50 }
  51
  52 class StoreSession : public QObject
  53 {
  54         Q_OBJECT
  55
  56 private:
  57         static const size_t BlockSize;
  58
  59 public:
  60         StoreSession(const std::string &file_name,
  61                 const std::shared_ptr<sigrok::OutputFormat> &output_format,
  62                 const std::map<std::string, Glib::VariantBase> &options,
  63                 const std::pair<uint64_t, uint64_t> sample_range,
  64                 const Session &session);
  65
  66         ~StoreSession();
  67
  68         std::pair<int, int> progress() const;
  69
  70         const QString& error() const;
  71
  72         bool start();
  73
  74         void wait();
  75
  76         void cancel();
  77
  78 private:
  79         void store_proc(std::vector< std::shared_ptr<data::SignalBase> > achannel_list,
  80                 std::vector< std::shared_ptr<pv::data::AnalogSegment> > asegment_list,
  81                 std::shared_ptr<pv::data::LogicSegment> lsegment);
  82
  83 Q_SIGNALS:
  84         void progress_updated();
  85
  86         void store_successful();
  87
  88 private:
  89         const std::string file_name_;
  90         const std::shared_ptr<sigrok::OutputFormat> output_format_;
  91         const std::map<std::string, Glib::VariantBase> options_;
  92         const std::pair<uint64_t, uint64_t> sample_range_;
  93         const Session &session_;
  94
  95         std::shared_ptr<sigrok::Output> output_;
  96         std::ofstream output_stream_;
  97
  98         std::thread thread_;
  99
 100         std::atomic<bool> interrupt_;
 101
 102         std::atomic<int> units_stored_, unit_count_;
 103
 104         mutable std::mutex mutex_;
 105         QString error_;
 106
 107         uint64_t start_sample_, sample_count_;
 108 };
 109
 110 } // pv
 111
 112 #endif // PULSEVIEW_PV_STORESESSION_HPP