X-Git-Url: http://git.code-monkey.de/?a=blobdiff_plain;f=pv%2Fdata%2Fdecoder.cpp;h=a9168ae2bdd33f8ec73fc46cfbf08c04e272c512;hb=4e5a4405482a296ebb6014e627298ad156c78d55;hp=cdfe57f9540b3bd8de3c8c787ed8121490567523;hpb=708c552391bc942e91bc09bc808577e2564347bd;p=pulseview.git

diff --git a/pv/data/decoder.cpp b/pv/data/decoder.cpp
index cdfe57f..a9168ae 100644
--- a/pv/data/decoder.cpp
+++ b/pv/data/decoder.cpp
@@ -18,29 +18,221 @@
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  */
 
+#include <libsigrokdecode/libsigrokdecode.h>
+
+#include <boost/thread/thread.hpp>
+
+#include <stdexcept>
+
+#include <QDebug>
+
 #include "decoder.h"
 
-#include <pv/view/signal.h>
+#include <pv/data/logic.h>
+#include <pv/data/logicsnapshot.h>
+#include <pv/view/logicsignal.h>
+#include <pv/view/decode/annotation.h>
+
+using namespace boost;
+using namespace std;
 
 namespace pv {
 namespace data {
 
-Decoder::Decoder(const srd_decoder *const dec,
-	std::map<const srd_probe*,
-		boost::shared_ptr<pv::view::Signal> > probes) :
+const double Decoder::DecodeMargin = 1.0;
+const double Decoder::DecodeThreshold = 0.2;
+const int64_t Decoder::DecodeChunkLength = 4096;
+
+mutex Decoder::_global_decode_mutex;
+
+Decoder::Decoder(const srd_decoder *const dec) :
 	_decoder(dec),
-	_probes(probes)
+	_options(g_hash_table_new_full(g_str_hash,
+		g_str_equal, g_free, (GDestroyNotify)g_variant_unref))
+{
+}
+
+Decoder::~Decoder()
 {
+	_decode_thread.interrupt();
+	_decode_thread.join();
+
+	g_hash_table_destroy(_options);
 }
 
-const srd_decoder* Decoder::get_decoder() const
+const srd_decoder* Decoder::decoder() const
 {
 	return _decoder;
 }
 
+const map<const srd_probe*, shared_ptr<view::LogicSignal> >&
+Decoder::probes() const
+{
+	return _probes;
+}
+
+void Decoder::set_probes(std::map<const srd_probe*,
+	boost::shared_ptr<view::LogicSignal> > probes)
+{
+	_probes = probes;
+	begin_decode();
+}
+
+const GHashTable* Decoder::options() const
+{
+	return _options;
+}
+
+void Decoder::set_option(const char *id, GVariant *value)
+{
+	g_variant_ref(value);
+	g_hash_table_replace(_options, (void*)g_strdup(id), value);
+	begin_decode();
+}
+
+const vector< shared_ptr<view::decode::Annotation> >
+	Decoder::annotations() const
+{
+	lock_guard<mutex> lock(_mutex);
+	return _annotations;
+}
+
+QString Decoder::error_message()
+{
+	lock_guard<mutex> lock(_mutex);
+	return _error_message;
+}
+
+void Decoder::begin_decode()
+{
+	_decode_thread.interrupt();
+	_decode_thread.join();
+
+	_annotations.clear();
+
+	if (_probes.empty())
+		return;
+
+	// Get the samplerate and start time
+	shared_ptr<pv::view::LogicSignal> logic_signal =
+		dynamic_pointer_cast<pv::view::LogicSignal>(
+			(*_probes.begin()).second);
+	if (logic_signal) {
+		shared_ptr<pv::data::Logic> data(
+			logic_signal->data());
+		if (data) {
+			_start_time = data->get_start_time();
+			_samplerate = data->get_samplerate();
+			if (_samplerate == 0.0)
+				_samplerate = 1.0;
+		}
+	}
+
+	// We get the logic data of the first probe in the list.
+	// This works because we are currently assuming all
+	// LogicSignals have the same data/snapshot
+	shared_ptr<pv::view::LogicSignal> sig = (*_probes.begin()).second;
+	assert(sig);
+	shared_ptr<data::Logic> data = sig->data();
+
+	_decode_thread = boost::thread(&Decoder::decode_proc, this,
+		data);
+}
+
 void Decoder::clear_snapshots()
 {
 }
 
+void Decoder::decode_proc(shared_ptr<data::Logic> data)
+{
+	srd_session *session;
+	uint8_t chunk[DecodeChunkLength];
+
+	assert(data);
+
+	const deque< shared_ptr<pv::data::LogicSnapshot> > &snapshots =
+		data->get_snapshots();
+	if (snapshots.empty())
+		return;
+
+	const shared_ptr<pv::data::LogicSnapshot> &snapshot =
+		snapshots.front();
+	const int64_t sample_count = snapshot->get_sample_count() - 1;
+
+	// Create the session
+	srd_session_new(&session);
+	assert(session);
+
+	srd_session_metadata_set(session, SRD_CONF_SAMPLERATE,
+		g_variant_new_uint64((uint64_t)_samplerate));
+
+	srd_pd_output_callback_add(session, SRD_OUTPUT_ANN,
+		Decoder::annotation_callback, this);
+
+	// Create the decoder instance
+	srd_decoder_inst *const decoder_inst = srd_inst_new(
+		session, _decoder->id, _options);
+	if(!decoder_inst) {
+		_error_message = tr("Failed to initialise decoder");
+		return;
+	}
+
+	// Setup the probes
+	GHashTable *const probes = g_hash_table_new_full(g_str_hash,
+		g_str_equal, g_free, (GDestroyNotify)g_variant_unref);
+
+	for(map<const srd_probe*, shared_ptr<view::LogicSignal> >::
+		const_iterator i = _probes.begin();
+		i != _probes.end(); i++)
+	{
+		shared_ptr<view::Signal> signal((*i).second);
+		GVariant *const gvar = g_variant_new_int32(
+			signal->probe()->index);
+		g_variant_ref_sink(gvar);
+		g_hash_table_insert(probes, (*i).first->id, gvar);
+	}
+
+	srd_inst_probe_set_all(decoder_inst, probes);
+
+	// Start the session
+	srd_session_start(session);
+
+	for (int64_t i = 0;
+		!this_thread::interruption_requested() && i < sample_count;
+		i += DecodeChunkLength)
+	{
+		lock_guard<mutex> decode_lock(_global_decode_mutex);
+
+		const int64_t chunk_end = min(
+			i + DecodeChunkLength, sample_count);
+		snapshot->get_samples(chunk, i, chunk_end);
+
+		if (srd_session_send(session, i, i + sample_count,
+				chunk, chunk_end - i) != SRD_OK) {
+			_error_message = tr("Failed to initialise decoder");
+			break;
+		}
+	}
+
+	// Destroy the session
+	srd_session_destroy(session);
+}
+
+void Decoder::annotation_callback(srd_proto_data *pdata, void *decoder)
+{
+	using namespace pv::view::decode;
+
+	assert(pdata);
+	assert(decoder);
+
+	Decoder *const d = (Decoder*)decoder;
+
+	shared_ptr<Annotation> a(new Annotation(pdata));
+	lock_guard<mutex> lock(d->_mutex);
+	d->_annotations.push_back(a);
+
+	d->new_decode_data();
+}
+
 } // namespace data
 } // namespace pv