[pulseview.git] / pv / views / decoder_binary / 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::Timestamp;

using std::shared_ptr;

namespace pv {
namespace views {
namespace decoder_binary {

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


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"), QVariant(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")));
#if QT_VERSION >= QT_VERSION_CHECK(6, 0, 0)
	save_action_->setShortcut(QKeySequence(Qt::CTRL | Qt::Key_S));
#else
	save_action_->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_S));
#endif
	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(QVariant::fromValue(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

	// Set up metadata event handler
	session_.metadata_obj_manager()->add_observer(this);

	reset_view_state();
}

View::~View()
{
	session_.metadata_obj_manager()->remove_observer(this);
}

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

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
{
	ViewBase::save_settings(settings);
}

void View::restore_settings(QSettings &settings)
{
	// Note: It is assumed that this function is only called once,
	// immediately after restoring a previous session.
	ViewBase::restore_settings(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;

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

	hex_view_->set_data(bin_class);

	if (!binary_data_exists_)
		return;

	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;
		data.resize(selection.second - selection.first + 1);

		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;
		data.resize(selection.second - selection.first + 1);

		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;
		uint64_t n = hex_view_->get_bytes_per_line();
		QString s;

		while (offset < selection.second) {
			size_t end = std::min((uint64_t)(selection.second), offset + n);
			offset = hex_view_->create_hex_line(offset, end, &s, with_offset, with_ascii);
#if QT_VERSION >= QT_VERSION_CHECK(5, 15, 0)
			out_stream << s << Qt::endl;
#else
			out_stream << s << endl;
#endif
		}

#if QT_VERSION >= QT_VERSION_CHECK(5, 15, 0)
		out_stream << Qt::endl;
#else
		out_stream << endl;
#endif

		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_) ?
		signal_->get_binary_data_chunk_count(current_segment_, decoder_, bin_class_id_) :
		false;

	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::on_metadata_object_changed(MetadataObject* obj,
	MetadataValueType value_type)
{
	// Check if we need to update the model's data range. We only work on the
	// end sample value because the start sample value is updated first and
	// we need both
	if ((obj->type() == MetadataObjMainViewRange) &&
		(value_type == MetadataValueEndSample)) {

		int64_t start_sample = obj->value(MetadataValueStartSample).toLongLong();
		int64_t end_sample = obj->value(MetadataValueEndSample).toLongLong();

		hex_view_->set_visible_sample_range(start_sample, end_sample);
	}

	if (obj->type() == MetadataObjMousePos)
		hex_view_->set_highlighted_data_sample(obj->value(MetadataValueStartSample).toLongLong());
}

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

	update_data();
}


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