sigsession.h 8.54 KB
Newer Older
DreamSourceLab's avatar
DreamSourceLab committed
1
/*
DreamSourceLab's avatar
DreamSourceLab committed
2 3
 * This file is part of the DSView project.
 * DSView is based on PulseView.
DreamSourceLab's avatar
DreamSourceLab committed
4 5
 *
 * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
6
 * Copyright (C) 2013 DreamSourceLab <support@dreamsourcelab.com>
DreamSourceLab's avatar
DreamSourceLab committed
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 *
 * This program 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 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
 */


DreamSourceLab's avatar
DreamSourceLab committed
24 25
#ifndef DSVIEW_PV_SIGSESSION_H
#define DSVIEW_PV_SIGSESSION_H
DreamSourceLab's avatar
DreamSourceLab committed
26 27 28 29 30 31 32 33

#include <boost/function.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/weak_ptr.hpp>
#include <boost/thread.hpp>

#include <string>
#include <utility>
DreamSourceLab's avatar
DreamSourceLab committed
34 35 36
#include <map>
#include <set>
#include <string>
DreamSourceLab's avatar
DreamSourceLab committed
37
#include <vector>
38
#include <stdint.h>
DreamSourceLab's avatar
DreamSourceLab committed
39 40 41 42 43 44 45

#include <QObject>
#include <QString>
#include <QLine>
#include <QVector>
#include <QMap>
#include <QVariant>
DreamSourceLab's avatar
DreamSourceLab committed
46
#include <QTimer>
47
#include <QtConcurrent/QtConcurrent>
48
#include <QJsonObject>
DreamSourceLab's avatar
DreamSourceLab committed
49

DreamSourceLab's avatar
DreamSourceLab committed
50
#include <libsigrok4DSL/libsigrok.h>
51
#include <libusb.h>
DreamSourceLab's avatar
DreamSourceLab committed
52

DreamSourceLab's avatar
DreamSourceLab committed
53 54 55
struct srd_decoder;
struct srd_channel;

DreamSourceLab's avatar
DreamSourceLab committed
56 57 58 59 60
namespace pv {

class DeviceManager;

namespace data {
DreamSourceLab's avatar
DreamSourceLab committed
61
class SignalData;
DreamSourceLab's avatar
DreamSourceLab committed
62 63
class Analog;
class AnalogSnapshot;
64 65
class Dso;
class DsoSnapshot;
DreamSourceLab's avatar
DreamSourceLab committed
66 67 68 69
class Logic;
class LogicSnapshot;
class Group;
class GroupSnapshot;
70
class DecoderModel;
DreamSourceLab's avatar
DreamSourceLab committed
71 72
}

DreamSourceLab's avatar
DreamSourceLab committed
73 74 75 76
namespace device {
class DevInst;
}

DreamSourceLab's avatar
DreamSourceLab committed
77 78
namespace view {
class Signal;
DreamSourceLab's avatar
DreamSourceLab committed
79 80
class GroupSignal;
class DecodeTrace;
DreamSourceLab's avatar
DreamSourceLab committed
81
class MathTrace;
DreamSourceLab's avatar
DreamSourceLab committed
82 83 84 85 86 87 88 89 90 91 92 93
}

namespace decoder {
class Decoder;
class DecoderFactory;
}

class SigSession : public QObject
{
        Q_OBJECT

private:
94
    static constexpr float Oversampling = 2.0f;
95
    static const int ViewTime = 50;
96
    static const int RefreshTime = 500;
97
	bool saveFileThreadRunning = false;
DreamSourceLab's avatar
DreamSourceLab committed
98 99 100 101 102 103 104 105 106 107 108 109 110

public:
	enum capture_state {
        Init,
		Stopped,
		Running
	};

public:
	SigSession(DeviceManager &device_manager);

	~SigSession();

DreamSourceLab's avatar
DreamSourceLab committed
111
    boost::shared_ptr<device::DevInst> get_device() const;
DreamSourceLab's avatar
DreamSourceLab committed
112 113 114 115

	/**
	 * Sets device instance that will be used in the next capture session.
	 */
DreamSourceLab's avatar
DreamSourceLab committed
116 117
    void set_device(boost::shared_ptr<device::DevInst> dev_inst)
        throw(QString);
DreamSourceLab's avatar
DreamSourceLab committed
118

119
    void set_file(QString name)
DreamSourceLab's avatar
DreamSourceLab committed
120
        throw(QString);
DreamSourceLab's avatar
DreamSourceLab committed
121

122
    void save_file(const QString name, QWidget* parent, int type);
DreamSourceLab's avatar
DreamSourceLab committed
123

DreamSourceLab's avatar
DreamSourceLab committed
124
    void set_default_device(boost::function<void (const QString)> error_handler);
125
    void export_file(const QString name, QWidget* parent, const QString ext);
126

DreamSourceLab's avatar
DreamSourceLab committed
127 128
    void release_device(device::DevInst *dev_inst);

DreamSourceLab's avatar
DreamSourceLab committed
129 130
	capture_state get_capture_state() const;

131 132 133
    uint64_t cur_samplerate() const;
    uint64_t cur_samplelimits() const;
    double cur_sampletime() const;
134 135
    QDateTime get_trigger_time() const;
    uint64_t get_trigger_pos() const;
136

DreamSourceLab's avatar
DreamSourceLab committed
137
    void start_capture(bool instant,
DreamSourceLab's avatar
DreamSourceLab committed
138
		boost::function<void (const QString)> error_handler);
139
    void capture_init();
DreamSourceLab's avatar
DreamSourceLab committed
140 141
	void stop_capture();

DreamSourceLab's avatar
DreamSourceLab committed
142 143
    std::set< boost::shared_ptr<data::SignalData> > get_data() const;

DreamSourceLab's avatar
DreamSourceLab committed
144 145 146
	std::vector< boost::shared_ptr<view::Signal> >
		get_signals();

DreamSourceLab's avatar
DreamSourceLab committed
147 148
    std::vector< boost::shared_ptr<view::GroupSignal> >
        get_group_signals();
149
    QList<QString> getSuportedExportFormats();
DreamSourceLab's avatar
DreamSourceLab committed
150

DreamSourceLab's avatar
DreamSourceLab committed
151 152
#ifdef ENABLE_DECODE
    bool add_decoder(srd_decoder *const dec);
DreamSourceLab's avatar
DreamSourceLab committed
153

DreamSourceLab's avatar
DreamSourceLab committed
154 155
    std::vector< boost::shared_ptr<view::DecodeTrace> >
        get_decode_signals() const;
DreamSourceLab's avatar
DreamSourceLab committed
156

DreamSourceLab's avatar
DreamSourceLab committed
157
    void remove_decode_signal(view::DecodeTrace *signal);
DreamSourceLab's avatar
DreamSourceLab committed
158

DreamSourceLab's avatar
DreamSourceLab committed
159
    void remove_decode_signal(int index);
DreamSourceLab's avatar
DreamSourceLab committed
160

DreamSourceLab's avatar
DreamSourceLab committed
161
    void rst_decoder(int index);
DreamSourceLab's avatar
DreamSourceLab committed
162

DreamSourceLab's avatar
DreamSourceLab committed
163
    void rst_decoder(view::DecodeTrace *signal);
DreamSourceLab's avatar
DreamSourceLab committed
164

165 166
    pv::data::DecoderModel* get_decoder_model() const;

DreamSourceLab's avatar
DreamSourceLab committed
167 168 169
    std::vector< boost::shared_ptr<view::MathTrace> >
        get_math_signals();

DreamSourceLab's avatar
DreamSourceLab committed
170
#endif
DreamSourceLab's avatar
DreamSourceLab committed
171

DreamSourceLab's avatar
DreamSourceLab committed
172
    void init_signals();
DreamSourceLab's avatar
DreamSourceLab committed
173

DreamSourceLab's avatar
DreamSourceLab committed
174
    void add_group();
DreamSourceLab's avatar
DreamSourceLab committed
175

DreamSourceLab's avatar
DreamSourceLab committed
176
    void del_group();
DreamSourceLab's avatar
DreamSourceLab committed
177

178
    const void* get_buf(int& unit_size, uint64_t& length);
DreamSourceLab's avatar
DreamSourceLab committed
179

180 181
    void start_hotplug_proc(boost::function<void (const QString)> error_handler);
    void stop_hotplug_proc();
DreamSourceLab's avatar
DreamSourceLab committed
182
	void register_hotplug_callback();
183
    void deregister_hotplug_callback();
DreamSourceLab's avatar
DreamSourceLab committed
184

DreamSourceLab's avatar
DreamSourceLab committed
185
    uint16_t get_ch_num(int type);
DreamSourceLab's avatar
DreamSourceLab committed
186
    
DreamSourceLab's avatar
DreamSourceLab committed
187
    bool get_instant();
DreamSourceLab's avatar
v0.3  
DreamSourceLab committed
188

189 190
    bool get_data_lock();

DreamSourceLab's avatar
DreamSourceLab committed
191 192
    void mathTraces_rebuild();

DreamSourceLab's avatar
DreamSourceLab committed
193 194 195 196
private:
	void set_capture_state(capture_state state);

private:
DreamSourceLab's avatar
DreamSourceLab committed
197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213
    /**
     * Attempts to autodetect the format. Failing that
     * @param filename The filename of the input file.
     * @return A pointer to the 'struct sr_input_format' that should be
     * 	used, or NULL if no input format was selected or
     * 	auto-detected.
     */
    static sr_input_format* determine_input_file_format(
        const std::string &filename);

    static sr_input* load_input_file_format(
        const std::string &filename,
        boost::function<void (const QString)> error_handler,
        sr_input_format *format = NULL);

    void sample_thread_proc(boost::shared_ptr<device::DevInst> dev_inst,
                            boost::function<void (const QString)> error_handler);
DreamSourceLab's avatar
DreamSourceLab committed
214

DreamSourceLab's avatar
v0.3  
DreamSourceLab committed
215
    // data feed
DreamSourceLab's avatar
DreamSourceLab committed
216 217 218
	void feed_in_header(const sr_dev_inst *sdi);
	void feed_in_meta(const sr_dev_inst *sdi,
		const sr_datafeed_meta &meta);
DreamSourceLab's avatar
v0.3  
DreamSourceLab committed
219
    void feed_in_trigger(const ds_trigger_pos &trigger_pos);
DreamSourceLab's avatar
DreamSourceLab committed
220
	void feed_in_logic(const sr_datafeed_logic &logic);
DreamSourceLab's avatar
v0.3  
DreamSourceLab committed
221
    void feed_in_dso(const sr_datafeed_dso &dso);
DreamSourceLab's avatar
DreamSourceLab committed
222 223
	void feed_in_analog(const sr_datafeed_analog &analog);
	void data_feed_in(const struct sr_dev_inst *sdi,
DreamSourceLab's avatar
v0.3  
DreamSourceLab committed
224 225 226
		const struct sr_datafeed_packet *packet);
	static void data_feed_in_proc(const struct sr_dev_inst *sdi,
		const struct sr_datafeed_packet *packet, void *cb_data);
DreamSourceLab's avatar
DreamSourceLab committed
227

DreamSourceLab's avatar
DreamSourceLab committed
228
    // thread for hotplug
DreamSourceLab's avatar
DreamSourceLab committed
229 230 231
    void hotplug_proc(boost::function<void (const QString)> error_handler);
    static int hotplug_callback(struct libusb_context *ctx, struct libusb_device *dev,
                                libusb_hotplug_event event, void *user_data);
DreamSourceLab's avatar
v0.3  
DreamSourceLab committed
232

DreamSourceLab's avatar
DreamSourceLab committed
233 234 235 236 237 238
private:
	DeviceManager &_device_manager;

	/**
	 * The device instance that will be used in the next capture session.
	 */
DreamSourceLab's avatar
DreamSourceLab committed
239
    boost::shared_ptr<device::DevInst> _dev_inst;
DreamSourceLab's avatar
DreamSourceLab committed
240

241
    mutable boost::mutex _sampling_mutex;
DreamSourceLab's avatar
DreamSourceLab committed
242
	capture_state _capture_state;
DreamSourceLab's avatar
DreamSourceLab committed
243
    bool _instant;
244 245
    uint64_t _cur_samplerate;
    uint64_t _cur_samplelimits;
DreamSourceLab's avatar
DreamSourceLab committed
246

247
    //mutable boost::mutex _signals_mutex;
DreamSourceLab's avatar
DreamSourceLab committed
248
	std::vector< boost::shared_ptr<view::Signal> > _signals;
DreamSourceLab's avatar
DreamSourceLab committed
249
    std::vector< boost::shared_ptr<view::GroupSignal> > _group_traces;
DreamSourceLab's avatar
DreamSourceLab committed
250
#ifdef ENABLE_DECODE
DreamSourceLab's avatar
DreamSourceLab committed
251
    std::vector< boost::shared_ptr<view::DecodeTrace> > _decode_traces;
252
    pv::data::DecoderModel *_decoder_model;
DreamSourceLab's avatar
DreamSourceLab committed
253
#endif
DreamSourceLab's avatar
DreamSourceLab committed
254
    std::vector< boost::shared_ptr<view::MathTrace> > _math_traces;
DreamSourceLab's avatar
DreamSourceLab committed
255

DreamSourceLab's avatar
DreamSourceLab committed
256
    mutable boost::mutex _data_mutex;
DreamSourceLab's avatar
DreamSourceLab committed
257 258
	boost::shared_ptr<data::Logic> _logic_data;
	boost::shared_ptr<data::LogicSnapshot> _cur_logic_snapshot;
259 260
    boost::shared_ptr<data::Dso> _dso_data;
    boost::shared_ptr<data::DsoSnapshot> _cur_dso_snapshot;
DreamSourceLab's avatar
DreamSourceLab committed
261 262 263 264 265 266
	boost::shared_ptr<data::Analog> _analog_data;
	boost::shared_ptr<data::AnalogSnapshot> _cur_analog_snapshot;
    boost::shared_ptr<data::Group> _group_data;
    boost::shared_ptr<data::GroupSnapshot> _cur_group_snapshot;
    int _group_cnt;

DreamSourceLab's avatar
DreamSourceLab committed
267
	std::unique_ptr<boost::thread> _sampling_thread;
DreamSourceLab's avatar
DreamSourceLab committed
268

DreamSourceLab's avatar
DreamSourceLab committed
269
	libusb_hotplug_callback_handle _hotplug_handle;
DreamSourceLab's avatar
DreamSourceLab committed
270
    std::unique_ptr<boost::thread> _hotplug;
DreamSourceLab's avatar
DreamSourceLab committed
271 272 273
    bool _hot_attach;
    bool _hot_detach;

DreamSourceLab's avatar
DreamSourceLab committed
274
    QTimer _view_timer;
275 276
    QTimer _refresh_timer;
    bool _data_lock;
277
    bool _data_updated;
DreamSourceLab's avatar
DreamSourceLab committed
278

279 280 281
    QDateTime _trigger_time;
    uint64_t _trigger_pos;

DreamSourceLab's avatar
DreamSourceLab committed
282 283 284 285 286 287 288
signals:
	void capture_state_changed(int state);

	void signals_changed();

	void data_updated();

DreamSourceLab's avatar
DreamSourceLab committed
289
    void start_timer(int);
DreamSourceLab's avatar
DreamSourceLab committed
290 291 292 293 294 295 296 297 298 299

    void receive_data(quint64 length);

    void device_attach();
    void device_detach();

    void test_data_error();

    void receive_trigger(quint64 trigger_pos);

300 301
    void receive_header();

DreamSourceLab's avatar
v0.3  
DreamSourceLab committed
302 303
    void dso_ch_changed(uint16_t num);

DreamSourceLab's avatar
DreamSourceLab committed
304 305 306
    void frame_began();

    void data_received();
DreamSourceLab's avatar
DreamSourceLab committed
307

DreamSourceLab's avatar
DreamSourceLab committed
308 309 310 311
    void frame_ended();

    void device_setted();

DreamSourceLab's avatar
DreamSourceLab committed
312 313 314
    void malloc_error();

    void zero_adj();
315
    void progressSaveFileValueChanged(int percent);
DreamSourceLab's avatar
DreamSourceLab committed
316

317 318 319 320
    void decode_done();

    void show_region(uint64_t start, uint64_t end);

321 322
    void hardware_connect_failed();

DreamSourceLab's avatar
DreamSourceLab committed
323 324
public slots:
    void reload();
325
    void refresh(int holdtime);
DreamSourceLab's avatar
DreamSourceLab committed
326

327 328
private slots:
    void cancelSaveFile();
329
    void data_unlock();
330
    void check_update();
331

DreamSourceLab's avatar
DreamSourceLab committed
332 333 334 335 336 337 338 339 340
private:
	// TODO: This should not be necessary. Multiple concurrent
	// sessions should should be supported and it should be
	// possible to associate a pointer with a ds_session.
	static SigSession *_session;
};

} // namespace pv

DreamSourceLab's avatar
DreamSourceLab committed
341
#endif // DSVIEW_PV_SIGSESSION_H