[pulseview.git] / pv / views / decoder_output / view.cpp

/*
 * This file is part of the PulseView project.
 *
 * Copyright (C) 2019 Soeren Apel <soeren@apelpie.net>
 *
 * 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, see <http://www.gnu.org/licenses/>.
 */

#include <climits>

#include <QByteArray>
#include <QDebug>
#include <QFileDialog>
#include <QLabel>
#include <QMenu>
#include <QMessageBox>
#include <QToolBar>
#include <QVBoxLayout>

#include <libsigrokdecode/libsigrokdecode.h>

#include "view.hpp"
#include "QHexView.hpp"

#include "pv/globalsettings.hpp"
#include "pv/session.hpp"
#include "pv/util.hpp"
#include "pv/data/decode/decoder.hpp"

using pv::data::DecodeSignal;
using pv::data::SignalBase;
using pv::data::decode::Decoder;
using pv::util::TimeUnit;
using pv::util::Timestamp;

using std::dynamic_pointer_cast;
using std::numeric_limits;
using std::shared_ptr;

namespace pv {
namespace views {
namespace decoder_output {

const char* SaveTypeNames[SaveTypeCount] = {
	"Binary",
	"Hex Dump, plain",
	"Hex Dump, with offset",
	"Hex Dump, complete"
};


View::View(Session &session, bool is_main_view, QMainWindow *parent) :
	ViewBase(session, is_main_view, parent),

	// Note: Place defaults in View::reset_view_state(), not here
	parent_(parent),
	decoder_selector_(new QComboBox()),
	format_selector_(new QComboBox()),
	class_selector_(new QComboBox()),
	stacked_widget_(new QStackedWidget()),
	hex_view_(new QHexView()),
	save_button_(new QToolButton()),
	save_action_(new QAction(this)),
	signal_(nullptr)
{
	QVBoxLayout *root_layout = new QVBoxLayout(this);
	root_layout->setContentsMargins(0, 0, 0, 0);

	// Create toolbar
	QToolBar* toolbar = new QToolBar();
	toolbar->setContextMenuPolicy(Qt::PreventContextMenu);
	parent->addToolBar(toolbar);

	// Populate toolbar
	toolbar->addWidget(new QLabel(tr("Decoder:")));
	toolbar->addWidget(decoder_selector_);
	toolbar->addWidget(class_selector_);
	toolbar->addSeparator();
	toolbar->addWidget(new QLabel(tr("Show data as")));
	toolbar->addWidget(format_selector_);
	toolbar->addSeparator();
	toolbar->addWidget(save_button_);

	// Add format types
	format_selector_->addItem(tr("Hexdump"), qVariantFromValue(QString("text/hexdump")));

	// Add widget stack
	root_layout->addWidget(stacked_widget_);
	stacked_widget_->addWidget(hex_view_);
	stacked_widget_->setCurrentIndex(0);

	connect(decoder_selector_, SIGNAL(currentIndexChanged(int)),
		this, SLOT(on_selected_decoder_changed(int)));
	connect(class_selector_, SIGNAL(currentIndexChanged(int)),
		this, SLOT(on_selected_class_changed(int)));

	// Configure widgets
	decoder_selector_->setSizeAdjustPolicy(QComboBox::AdjustToContents);
	class_selector_->setSizeAdjustPolicy(QComboBox::AdjustToContents);

	// Configure actions
	save_action_->setText(tr("&Save..."));
	save_action_->setIcon(QIcon::fromTheme("document-save-as",
		QIcon(":/icons/document-save-as.png")));
	save_action_->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_S));
	connect(save_action_, SIGNAL(triggered(bool)),
		this, SLOT(on_actionSave_triggered()));

	QMenu *save_menu = new QMenu();
	connect(save_menu, SIGNAL(triggered(QAction*)),
		this, SLOT(on_actionSave_triggered(QAction*)));

	for (int i = 0; i < SaveTypeCount; i++) {
		QAction *const action =	save_menu->addAction(tr(SaveTypeNames[i]));
		action->setData(qVariantFromValue(i));
	}

	save_button_->setMenu(save_menu);
	save_button_->setDefaultAction(save_action_);
	save_button_->setPopupMode(QToolButton::MenuButtonPopup);

	parent->setSizePolicy(hex_view_->sizePolicy()); // TODO Must be updated when selected widget changes

	reset_view_state();
}

View::~View()
{
}

ViewType View::get_type() const
{
	return ViewTypeDecoderOutput;
}

void View::reset_view_state()
{
	ViewBase::reset_view_state();

	decoder_selector_->clear();
	class_selector_->clear();
	format_selector_->setCurrentIndex(0);
	save_button_->setEnabled(false);

	hex_view_->clear();
}

void View::clear_decode_signals()
{
	ViewBase::clear_decode_signals();

	reset_data();
	reset_view_state();
}

void View::add_decode_signal(shared_ptr<data::DecodeSignal> signal)
{
	ViewBase::add_decode_signal(signal);

	connect(signal.get(), SIGNAL(name_changed(const QString&)),
		this, SLOT(on_signal_name_changed(const QString&)));
	connect(signal.get(), SIGNAL(decoder_stacked(void*)),
		this, SLOT(on_decoder_stacked(void*)));
	connect(signal.get(), SIGNAL(decoder_removed(void*)),
		this, SLOT(on_decoder_removed(void*)));

	// Add all decoders provided by this signal
	auto stack = signal->decoder_stack();
	if (stack.size() > 1) {
		for (const shared_ptr<Decoder>& dec : stack)
			// Only add the decoder if it has binary output
			if (dec->get_binary_class_count() > 0) {
				QString title = QString("%1 (%2)").arg(signal->name(), dec->name());
				decoder_selector_->addItem(title, QVariant::fromValue((void*)dec.get()));
			}
	} else
		if (!stack.empty()) {
			shared_ptr<Decoder>& dec = stack.at(0);
			if (dec->get_binary_class_count() > 0)
				decoder_selector_->addItem(signal->name(), QVariant::fromValue((void*)dec.get()));
		}
}

void View::remove_decode_signal(shared_ptr<data::DecodeSignal> signal)
{
	// Remove all decoders provided by this signal
	for (const shared_ptr<Decoder>& dec : signal->decoder_stack()) {
		int index = decoder_selector_->findData(QVariant::fromValue((void*)dec.get()));

		if (index != -1)
			decoder_selector_->removeItem(index);
	}

	ViewBase::remove_decode_signal(signal);

	if (signal.get() == signal_) {
		reset_data();
		update_data();
		reset_view_state();
	}
}

void View::save_settings(QSettings &settings) const
{
	(void)settings;
}

void View::restore_settings(QSettings &settings)
{
	// Note: It is assumed that this function is only called once,
	// immediately after restoring a previous session.
	(void)settings;
}

void View::reset_data()
{
	signal_ = nullptr;
	decoder_ = nullptr;
	bin_class_id_ = 0;
	binary_data_exists_ = false;

	hex_view_->clear();
}

void View::update_data()
{
	if (!signal_)
		return;

	if (!binary_data_exists_)
		return;

	const DecodeBinaryClass* bin_class =
		signal_->get_binary_data_class(current_segment_, decoder_, bin_class_id_);

	hex_view_->set_data(bin_class);

	if (!save_button_->isEnabled())
		save_button_->setEnabled(true);
}

void View::save_data() const
{
	assert(decoder_);
	assert(signal_);

	if (!signal_)
		return;

	GlobalSettings settings;
	const QString dir = settings.value("MainWindow/SaveDirectory").toString();

	const QString file_name = QFileDialog::getSaveFileName(
		parent_, tr("Save Binary Data"), dir, tr("Binary Data Files (*.bin);;All Files (*)"));

	if (file_name.isEmpty())
		return;

	QFile file(file_name);
	if (file.open(QIODevice::WriteOnly | QIODevice::Truncate)) {
		pair<size_t, size_t> selection = hex_view_->get_selection();

		vector<uint8_t> data;
		signal_->get_merged_binary_data_chunks_by_offset(current_segment_, decoder_,
			bin_class_id_, selection.first, selection.second, &data);

		int64_t bytes_written = file.write((const char*)data.data(), data.size());

		if ((bytes_written == -1) || ((uint64_t)bytes_written != data.size())) {
			QMessageBox msg(parent_);
			msg.setText(tr("Error") + "\n\n" + tr("File %1 could not be written to.").arg(file_name));
			msg.setStandardButtons(QMessageBox::Ok);
			msg.setIcon(QMessageBox::Warning);
			msg.exec();
			return;
		}
	}
}

void View::save_data_as_hex_dump(bool with_offset, bool with_ascii) const
{
	assert(decoder_);
	assert(signal_);

	if (!signal_)
		return;

	GlobalSettings settings;
	const QString dir = settings.value("MainWindow/SaveDirectory").toString();

	const QString file_name = QFileDialog::getSaveFileName(
		parent_, tr("Save Binary Data"), dir, tr("Hex Dumps (*.txt);;All Files (*)"));

	if (file_name.isEmpty())
		return;

	QFile file(file_name);
	if (file.open(QIODevice::WriteOnly | QIODevice::Truncate | QIODevice::Text)) {
		pair<size_t, size_t> selection = hex_view_->get_selection();

		vector<uint8_t> data;
		signal_->get_merged_binary_data_chunks_by_offset(current_segment_, decoder_,
			bin_class_id_, selection.first, selection.second, &data);

		QTextStream out_stream(&file);

		uint64_t offset = selection.first;
		unsigned int n = hex_view_->get_bytes_per_line();
		QString s;

		while (offset < selection.second) {
			size_t end = std::min(selection.second, offset + n);
			offset = hex_view_->create_hex_line(offset, end, &s, with_offset, with_ascii);
			out_stream << s << endl;
		}

		out_stream << endl;

		if (out_stream.status() != QTextStream::Ok) {
			QMessageBox msg(parent_);
			msg.setText(tr("Error") + "\n\n" + tr("File %1 could not be written to.").arg(file_name));
			msg.setStandardButtons(QMessageBox::Ok);
			msg.setIcon(QMessageBox::Warning);
			msg.exec();
			return;
		}
	}
}

void View::on_selected_decoder_changed(int index)
{
	if (signal_)
		disconnect(signal_, SIGNAL(new_binary_data(unsigned int, void*, unsigned int)));

	reset_data();

	decoder_ = (Decoder*)decoder_selector_->itemData(index).value<void*>();

	// Find the signal that contains the selected decoder
	for (const shared_ptr<DecodeSignal>& ds : decode_signals_)
		for (const shared_ptr<Decoder>& dec : ds->decoder_stack())
			if (decoder_ == dec.get())
				signal_ = ds.get();

	class_selector_->clear();

	if (signal_) {
		// Populate binary class selector
		uint32_t bin_classes = decoder_->get_binary_class_count();
		for (uint32_t i = 0; i < bin_classes; i++) {
			const data::decode::DecodeBinaryClassInfo* class_info = decoder_->get_binary_class(i);
			class_selector_->addItem(class_info->description, QVariant::fromValue(i));
		}

		connect(signal_, SIGNAL(new_binary_data(unsigned int, void*, unsigned int)),
			this, SLOT(on_new_binary_data(unsigned int, void*, unsigned int)));
	}

	update_data();
}

void View::on_selected_class_changed(int index)
{
	bin_class_id_ = class_selector_->itemData(index).value<uint32_t>();

	binary_data_exists_ =
		signal_->get_binary_data_chunk_count(current_segment_, decoder_, bin_class_id_);

	update_data();
}

void View::on_signal_name_changed(const QString &name)
{
	(void)name;

	SignalBase* sb = qobject_cast<SignalBase*>(QObject::sender());
	assert(sb);

	DecodeSignal* signal = dynamic_cast<DecodeSignal*>(sb);
	assert(signal);

	// Update all decoder entries provided by this signal
	auto stack = signal->decoder_stack();
	if (stack.size() > 1) {
		for (const shared_ptr<Decoder>& dec : stack) {
			QString title = QString("%1 (%2)").arg(signal->name(), dec->name());
			int index = decoder_selector_->findData(QVariant::fromValue((void*)dec.get()));

			if (index != -1)
				decoder_selector_->setItemText(index, title);
		}
	} else
		if (!stack.empty()) {
			shared_ptr<Decoder>& dec = stack.at(0);
			int index = decoder_selector_->findData(QVariant::fromValue((void*)dec.get()));

			if (index != -1)
				decoder_selector_->setItemText(index, signal->name());
		}
}

void View::on_new_binary_data(unsigned int segment_id, void* decoder, unsigned int bin_class_id)
{
	if ((segment_id == current_segment_) && (decoder == decoder_) && (bin_class_id == bin_class_id_))
		if (!delayed_view_updater_.isActive())
			delayed_view_updater_.start();
}

void View::on_decoder_stacked(void* decoder)
{
	// TODO This doesn't change existing entries for the same signal - but it should as the naming scheme may change

	Decoder* d = static_cast<Decoder*>(decoder);

	// Only add the decoder if it has binary output
	if (d->get_binary_class_count() == 0)
		return;

	// Find the signal that contains the selected decoder
	DecodeSignal* signal = nullptr;

	for (const shared_ptr<DecodeSignal>& ds : decode_signals_)
		for (const shared_ptr<Decoder>& dec : ds->decoder_stack())
			if (d == dec.get())
				signal = ds.get();

	assert(signal);

	// Add the decoder to the list
	QString title = QString("%1 (%2)").arg(signal->name(), d->name());
	decoder_selector_->addItem(title, QVariant::fromValue((void*)d));
}

void View::on_decoder_removed(void* decoder)
{
	Decoder* d = static_cast<Decoder*>(decoder);

	// Remove the decoder from the list
	int index = decoder_selector_->findData(QVariant::fromValue((void*)d));

	if (index != -1)
		decoder_selector_->removeItem(index);
}

void View::on_actionSave_triggered(QAction* action)
{
	int save_type = SaveTypeBinary;
	if (action)
		save_type = action->data().toInt();

	switch (save_type)
	{
	case SaveTypeBinary: save_data(); break;
	case SaveTypeHexDumpPlain: save_data_as_hex_dump(false, false); break;
	case SaveTypeHexDumpWithOffset: save_data_as_hex_dump(true, false); break;
	case SaveTypeHexDumpComplete: save_data_as_hex_dump(true, true); break;
	}
}

void View::perform_delayed_view_update()
{
	if (!binary_data_exists_)
		if (signal_->get_binary_data_chunk_count(current_segment_, decoder_, bin_class_id_))
			binary_data_exists_ = true;

	update_data();
}


} // namespace decoder_output
} // namespace views
} // namespace pv
Commit	Line	Data
2bdc5796 SA	1	/*
	2	* This file is part of the PulseView project.
	3	*
	4	* Copyright (C) 2019 Soeren Apel <soeren@apelpie.net>
	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
4b97fe09	20	#include <climits>
2bdc5796	21
4b97fe09 SA	22	#include <QByteArray>
4b97fe09 SA	23	#include <QDebug>
13b726cd	24	#include <QFileDialog>
bdbc561f	25	#include <QLabel>
2bdc5796	26	#include <QMenu>
13b726cd	27	#include <QMessageBox>
a24412db	28	#include <QToolBar>
2bdc5796 SA	29	#include <QVBoxLayout>
2bdc5796 SA	30
4b97fe09 SA	31	#include <libsigrokdecode/libsigrokdecode.h>
4b97fe09 SA	32
2bdc5796	33	#include "view.hpp"
560f8377	34	#include "QHexView.hpp"
2bdc5796	35
13b726cd	36	#include "pv/globalsettings.hpp"
2bdc5796 SA	37	#include "pv/session.hpp"
2bdc5796 SA	38	#include "pv/util.hpp"
e77de61f	39	#include "pv/data/decode/decoder.hpp"
2bdc5796	40
4b97fe09	41	using pv::data::DecodeSignal;
bdbc561f	42	using pv::data::SignalBase;
e77de61f	43	using pv::data::decode::Decoder;
2bdc5796 SA	44	using pv::util::TimeUnit;
	45	using pv::util::Timestamp;
	46
e77de61f	47	using std::dynamic_pointer_cast;
4b97fe09	48	using std::numeric_limits;
2bdc5796 SA	49	using std::shared_ptr;
	50
	51	namespace pv {
	52	namespace views {
	53	namespace decoder_output {
	54
03408f5f SA	55	const char* SaveTypeNames[SaveTypeCount] = {
03408f5f SA	56	"Binary",
d5645564 SA	57	"Hex Dump, plain",
	58	"Hex Dump, with offset",
	59	"Hex Dump, complete"
03408f5f SA	60	};
	61
	62
a24412db	63	View::View(Session &session, bool is_main_view, QMainWindow *parent) :
bdbc561f	64	ViewBase(session, is_main_view, parent),
2bdc5796 SA	65
2bdc5796 SA	66	// Note: Place defaults in View::reset_view_state(), not here
6961eab0	67	parent_(parent),
e77de61f	68	decoder_selector_(new QComboBox()),
560f8377	69	format_selector_(new QComboBox()),
e77de61f	70	class_selector_(new QComboBox()),
560f8377	71	stacked_widget_(new QStackedWidget()),
4b97fe09	72	hex_view_(new QHexView()),
03408f5f SA	73	save_button_(new QToolButton()),
03408f5f SA	74	save_action_(new QAction(this)),
628b45cc	75	signal_(nullptr)
2bdc5796 SA	76	{
	77	QVBoxLayout *root_layout = new QVBoxLayout(this);
	78	root_layout->setContentsMargins(0, 0, 0, 0);
	79
bdbc561f SA	80	// Create toolbar
	81	QToolBar* toolbar = new QToolBar();
	82	toolbar->setContextMenuPolicy(Qt::PreventContextMenu);
	83	parent->addToolBar(toolbar);
a24412db	84
bdbc561f SA	85	// Populate toolbar
bdbc561f SA	86	toolbar->addWidget(new QLabel(tr("Decoder:")));
e77de61f SA	87	toolbar->addWidget(decoder_selector_);
e77de61f SA	88	toolbar->addWidget(class_selector_);
861ab3a4 SA	89	toolbar->addSeparator();
	90	toolbar->addWidget(new QLabel(tr("Show data as")));
	91	toolbar->addWidget(format_selector_);
03408f5f SA	92	toolbar->addSeparator();
03408f5f SA	93	toolbar->addWidget(save_button_);
861ab3a4 SA	94
	95	// Add format types
	96	format_selector_->addItem(tr("Hexdump"), qVariantFromValue(QString("text/hexdump")));
a24412db	97
560f8377 SA	98	// Add widget stack
	99	root_layout->addWidget(stacked_widget_);
	100	stacked_widget_->addWidget(hex_view_);
4b97fe09 SA	101	stacked_widget_->setCurrentIndex(0);
4b97fe09 SA	102
e77de61f SA	103	connect(decoder_selector_, SIGNAL(currentIndexChanged(int)),
e77de61f SA	104	this, SLOT(on_selected_decoder_changed(int)));
b2b18d3a SA	105	connect(class_selector_, SIGNAL(currentIndexChanged(int)),
b2b18d3a SA	106	this, SLOT(on_selected_class_changed(int)));
4b97fe09	107
a13d836f SA	108	// Configure widgets
	109	decoder_selector_->setSizeAdjustPolicy(QComboBox::AdjustToContents);
	110	class_selector_->setSizeAdjustPolicy(QComboBox::AdjustToContents);
	111
03408f5f SA	112	// Configure actions
	113	save_action_->setText(tr("&Save..."));
	114	save_action_->setIcon(QIcon::fromTheme("document-save-as",
	115	QIcon(":/icons/document-save-as.png")));
	116	save_action_->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_S));
	117	connect(save_action_, SIGNAL(triggered(bool)),
	118	this, SLOT(on_actionSave_triggered()));
	119
	120	QMenu *save_menu = new QMenu();
	121	connect(save_menu, SIGNAL(triggered(QAction*)),
	122	this, SLOT(on_actionSave_triggered(QAction*)));
	123
	124	for (int i = 0; i < SaveTypeCount; i++) {
	125	QAction *const action = save_menu->addAction(tr(SaveTypeNames[i]));
	126	action->setData(qVariantFromValue(i));
	127	}
	128
	129	save_button_->setMenu(save_menu);
	130	save_button_->setDefaultAction(save_action_);
	131	save_button_->setPopupMode(QToolButton::MenuButtonPopup);
	132
6961eab0 SA	133	parent->setSizePolicy(hex_view_->sizePolicy()); // TODO Must be updated when selected widget changes
6961eab0 SA	134
2bdc5796 SA	135	reset_view_state();
	136	}
	137
	138	View::~View()
	139	{
	140	}
	141
db298158 SA	142	ViewType View::get_type() const
	143	{
	144	return ViewTypeDecoderOutput;
	145	}
	146
2bdc5796 SA	147	void View::reset_view_state()
	148	{
	149	ViewBase::reset_view_state();
2bdc5796	150
03408f5f SA	151	decoder_selector_->clear();
	152	class_selector_->clear();
	153	format_selector_->setCurrentIndex(0);
	154	save_button_->setEnabled(false);
	155
	156	hex_view_->clear();
2bdc5796 SA	157	}
	158
	159	void View::clear_decode_signals()
	160	{
e77de61f SA	161	ViewBase::clear_decode_signals();
e77de61f SA	162
03408f5f SA	163	reset_data();
03408f5f SA	164	reset_view_state();
2bdc5796 SA	165	}
	166
	167	void View::add_decode_signal(shared_ptr<data::DecodeSignal> signal)
	168	{
e77de61f SA	169	ViewBase::add_decode_signal(signal);
e77de61f SA	170
2bdc5796	171	connect(signal.get(), SIGNAL(name_changed(const QString&)),
bdbc561f	172	this, SLOT(on_signal_name_changed(const QString&)));
e77de61f SA	173	connect(signal.get(), SIGNAL(decoder_stacked(void*)),
	174	this, SLOT(on_decoder_stacked(void*)));
	175	connect(signal.get(), SIGNAL(decoder_removed(void*)),
	176	this, SLOT(on_decoder_removed(void*)));
	177
	178	// Add all decoders provided by this signal
	179	auto stack = signal->decoder_stack();
	180	if (stack.size() > 1) {
b2b18d3a SA	181	for (const shared_ptr<Decoder>& dec : stack)
	182	// Only add the decoder if it has binary output
	183	if (dec->get_binary_class_count() > 0) {
	184	QString title = QString("%1 (%2)").arg(signal->name(), dec->name());
	185	decoder_selector_->addItem(title, QVariant::fromValue((void*)dec.get()));
	186	}
e77de61f SA	187	} else
	188	if (!stack.empty()) {
	189	shared_ptr<Decoder>& dec = stack.at(0);
b2b18d3a SA	190	if (dec->get_binary_class_count() > 0)
b2b18d3a SA	191	decoder_selector_->addItem(signal->name(), QVariant::fromValue((void*)dec.get()));
e77de61f	192	}
2bdc5796 SA	193	}
	194
	195	void View::remove_decode_signal(shared_ptr<data::DecodeSignal> signal)
	196	{
e77de61f SA	197	// Remove all decoders provided by this signal
	198	for (const shared_ptr<Decoder>& dec : signal->decoder_stack()) {
	199	int index = decoder_selector_->findData(QVariant::fromValue((void*)dec.get()));
bdbc561f	200
e77de61f SA	201	if (index != -1)
	202	decoder_selector_->removeItem(index);
	203	}
	204
	205	ViewBase::remove_decode_signal(signal);
4b97fe09 SA	206
4b97fe09 SA	207	if (signal.get() == signal_) {
03408f5f	208	reset_data();
4b97fe09	209	update_data();
03408f5f	210	reset_view_state();
4b97fe09	211	}
2bdc5796 SA	212	}
	213
	214	void View::save_settings(QSettings &settings) const
	215	{
	216	(void)settings;
	217	}
	218
	219	void View::restore_settings(QSettings &settings)
	220	{
	221	// Note: It is assumed that this function is only called once,
	222	// immediately after restoring a previous session.
	223	(void)settings;
	224	}
	225
03408f5f SA	226	void View::reset_data()
	227	{
	228	signal_ = nullptr;
	229	decoder_ = nullptr;
	230	bin_class_id_ = 0;
	231	binary_data_exists_ = false;
	232
	233	hex_view_->clear();
	234	}
	235
4b97fe09 SA	236	void View::update_data()
4b97fe09 SA	237	{
03408f5f	238	if (!signal_)
4b97fe09	239	return;
4b97fe09	240
03408f5f	241	if (!binary_data_exists_)
4b97fe09	242	return;
4b97fe09	243
628b45cc SA	244	const DecodeBinaryClass* bin_class =
628b45cc SA	245	signal_->get_binary_data_class(current_segment_, decoder_, bin_class_id_);
b2b18d3a	246
d5645564	247	hex_view_->set_data(bin_class);
13b726cd SA	248
	249	if (!save_button_->isEnabled())
	250	save_button_->setEnabled(true);
	251	}
	252
	253	void View::save_data() const
	254	{
	255	assert(decoder_);
	256	assert(signal_);
	257
	258	if (!signal_)
	259	return;
	260
	261	GlobalSettings settings;
	262	const QString dir = settings.value("MainWindow/SaveDirectory").toString();
	263
	264	const QString file_name = QFileDialog::getSaveFileName(
	265	parent_, tr("Save Binary Data"), dir, tr("Binary Data Files (.bin);;All Files ()"));
	266
	267	if (file_name.isEmpty())
	268	return;
	269
	270	QFile file(file_name);
	271	if (file.open(QIODevice::WriteOnly \| QIODevice::Truncate)) {
	272	pair<size_t, size_t> selection = hex_view_->get_selection();
	273
	274	vector<uint8_t> data;
	275	signal_->get_merged_binary_data_chunks_by_offset(current_segment_, decoder_,
	276	bin_class_id_, selection.first, selection.second, &data);
	277
	278	int64_t bytes_written = file.write((const char*)data.data(), data.size());
	279
	280	if ((bytes_written == -1) \|\| ((uint64_t)bytes_written != data.size())) {
	281	QMessageBox msg(parent_);
	282	msg.setText(tr("Error") + "\n\n" + tr("File %1 could not be written to.").arg(file_name));
	283	msg.setStandardButtons(QMessageBox::Ok);
	284	msg.setIcon(QMessageBox::Warning);
	285	msg.exec();
	286	return;
	287	}
	288	}
	289	}
	290
d5645564	291	void View::save_data_as_hex_dump(bool with_offset, bool with_ascii) const
13b726cd	292	{
d5645564 SA	293	assert(decoder_);
	294	assert(signal_);
	295
	296	if (!signal_)
	297	return;
	298
	299	GlobalSettings settings;
	300	const QString dir = settings.value("MainWindow/SaveDirectory").toString();
	301
	302	const QString file_name = QFileDialog::getSaveFileName(
	303	parent_, tr("Save Binary Data"), dir, tr("Hex Dumps (.txt);;All Files ()"));
	304
	305	if (file_name.isEmpty())
	306	return;
	307
	308	QFile file(file_name);
	309	if (file.open(QIODevice::WriteOnly \| QIODevice::Truncate \| QIODevice::Text)) {
	310	pair<size_t, size_t> selection = hex_view_->get_selection();
	311
	312	vector<uint8_t> data;
	313	signal_->get_merged_binary_data_chunks_by_offset(current_segment_, decoder_,
	314	bin_class_id_, selection.first, selection.second, &data);
	315
	316	QTextStream out_stream(&file);
	317
	318	uint64_t offset = selection.first;
	319	unsigned int n = hex_view_->get_bytes_per_line();
	320	QString s;
13b726cd	321
d5645564 SA	322	while (offset < selection.second) {
	323	size_t end = std::min(selection.second, offset + n);
	324	offset = hex_view_->create_hex_line(offset, end, &s, with_offset, with_ascii);
	325	out_stream << s << endl;
	326	}
	327
	328	out_stream << endl;
	329
	330	if (out_stream.status() != QTextStream::Ok) {
	331	QMessageBox msg(parent_);
	332	msg.setText(tr("Error") + "\n\n" + tr("File %1 could not be written to.").arg(file_name));
	333	msg.setStandardButtons(QMessageBox::Ok);
	334	msg.setIcon(QMessageBox::Warning);
	335	msg.exec();
	336	return;
	337	}
	338	}
4b97fe09 SA	339	}
4b97fe09 SA	340
e77de61f	341	void View::on_selected_decoder_changed(int index)
4b97fe09 SA	342	{
4b97fe09 SA	343	if (signal_)
b2b18d3a	344	disconnect(signal_, SIGNAL(new_binary_data(unsigned int, void*, unsigned int)));
4b97fe09	345
03408f5f SA	346	reset_data();
03408f5f SA	347
e77de61f SA	348	decoder_ = (Decoder)decoder_selector_->itemData(index).value<void>();
	349
	350	// Find the signal that contains the selected decoder
b2b18d3a SA	351	for (const shared_ptr<DecodeSignal>& ds : decode_signals_)
	352	for (const shared_ptr<Decoder>& dec : ds->decoder_stack())
	353	if (decoder_ == dec.get())
	354	signal_ = ds.get();
e77de61f	355
b2b18d3a SA	356	class_selector_->clear();
	357
	358	if (signal_) {
	359	// Populate binary class selector
ac9494ef SA	360	uint32_t bin_classes = decoder_->get_binary_class_count();
ac9494ef SA	361	for (uint32_t i = 0; i < bin_classes; i++) {
b2b18d3a	362	const data::decode::DecodeBinaryClassInfo* class_info = decoder_->get_binary_class(i);
cbf428c2	363	class_selector_->addItem(class_info->description, QVariant::fromValue(i));
b2b18d3a SA	364	}
	365
	366	connect(signal_, SIGNAL(new_binary_data(unsigned int, void*, unsigned int)),
	367	this, SLOT(on_new_binary_data(unsigned int, void*, unsigned int)));
e77de61f	368	}
4b97fe09	369
b2b18d3a SA	370	update_data();
	371	}
	372
	373	void View::on_selected_class_changed(int index)
	374	{
ac9494ef	375	bin_class_id_ = class_selector_->itemData(index).value<uint32_t>();
e77de61f	376
03408f5f SA	377	binary_data_exists_ =
	378	signal_->get_binary_data_chunk_count(current_segment_, decoder_, bin_class_id_);
	379
e77de61f	380	update_data();
4b97fe09 SA	381	}
4b97fe09 SA	382
bdbc561f	383	void View::on_signal_name_changed(const QString &name)
2bdc5796	384	{
e77de61f SA	385	(void)name;
	386
	387	SignalBase* sb = qobject_cast<SignalBase*>(QObject::sender());
bdbc561f SA	388	assert(sb);
bdbc561f SA	389
e77de61f SA	390	DecodeSignal* signal = dynamic_cast<DecodeSignal*>(sb);
	391	assert(signal);
	392
	393	// Update all decoder entries provided by this signal
	394	auto stack = signal->decoder_stack();
	395	if (stack.size() > 1) {
	396	for (const shared_ptr<Decoder>& dec : stack) {
	397	QString title = QString("%1 (%2)").arg(signal->name(), dec->name());
	398	int index = decoder_selector_->findData(QVariant::fromValue((void*)dec.get()));
	399
	400	if (index != -1)
	401	decoder_selector_->setItemText(index, title);
	402	}
	403	} else
	404	if (!stack.empty()) {
	405	shared_ptr<Decoder>& dec = stack.at(0);
	406	int index = decoder_selector_->findData(QVariant::fromValue((void*)dec.get()));
	407
	408	if (index != -1)
	409	decoder_selector_->setItemText(index, signal->name());
	410	}
2bdc5796 SA	411	}
2bdc5796 SA	412
b2b18d3a	413	void View::on_new_binary_data(unsigned int segment_id, void* decoder, unsigned int bin_class_id)
4b97fe09	414	{
b2b18d3a	415	if ((segment_id == current_segment_) && (decoder == decoder_) && (bin_class_id == bin_class_id_))
516d2128 SA	416	if (!delayed_view_updater_.isActive())
516d2128 SA	417	delayed_view_updater_.start();
4b97fe09 SA	418	}
4b97fe09 SA	419
e77de61f SA	420	void View::on_decoder_stacked(void* decoder)
	421	{
	422	// TODO This doesn't change existing entries for the same signal - but it should as the naming scheme may change
	423
	424	Decoder* d = static_cast<Decoder*>(decoder);
	425
b2b18d3a SA	426	// Only add the decoder if it has binary output
	427	if (d->get_binary_class_count() == 0)
	428	return;
	429
e77de61f SA	430	// Find the signal that contains the selected decoder
	431	DecodeSignal* signal = nullptr;
	432
	433	for (const shared_ptr<DecodeSignal>& ds : decode_signals_)
	434	for (const shared_ptr<Decoder>& dec : ds->decoder_stack())
	435	if (d == dec.get())
	436	signal = ds.get();
	437
	438	assert(signal);
	439
	440	// Add the decoder to the list
	441	QString title = QString("%1 (%2)").arg(signal->name(), d->name());
	442	decoder_selector_->addItem(title, QVariant::fromValue((void*)d));
	443	}
	444
	445	void View::on_decoder_removed(void* decoder)
	446	{
	447	Decoder* d = static_cast<Decoder*>(decoder);
	448
	449	// Remove the decoder from the list
	450	int index = decoder_selector_->findData(QVariant::fromValue((void*)d));
	451
	452	if (index != -1)
	453	decoder_selector_->removeItem(index);
	454	}
	455
03408f5f SA	456	void View::on_actionSave_triggered(QAction* action)
03408f5f SA	457	{
13b726cd SA	458	int save_type = SaveTypeBinary;
	459	if (action)
	460	save_type = action->data().toInt();
	461
d5645564 SA	462	switch (save_type)
	463	{
	464	case SaveTypeBinary: save_data(); break;
	465	case SaveTypeHexDumpPlain: save_data_as_hex_dump(false, false); break;
	466	case SaveTypeHexDumpWithOffset: save_data_as_hex_dump(true, false); break;
	467	case SaveTypeHexDumpComplete: save_data_as_hex_dump(true, true); break;
	468	}
03408f5f SA	469	}
03408f5f SA	470
516d2128 SA	471	void View::perform_delayed_view_update()
516d2128 SA	472	{
03408f5f	473	if (!binary_data_exists_)
13b726cd	474	if (signal_->get_binary_data_chunk_count(current_segment_, decoder_, bin_class_id_))
03408f5f SA	475	binary_data_exists_ = true;
03408f5f SA	476
516d2128 SA	477	update_data();
	478	}
	479
e77de61f	480
2bdc5796 SA	481	} // namespace decoder_output
	482	} // namespace views
	483	} // namespace pv