DecodeSignal: Support unitsize > 1 for logic output

[pulseview.git] / pv / data / decodesignal.cpp

diff --git a/pv/data/decodesignal.cpp b/pv/data/decodesignal.cpp
index 9decbe302fe5892864b9c083ad36e28d37d0f3ff..2e23b1ceac8aeef02a67a6640ba50126973ccadf 100644 (file)
--- a/pv/data/decodesignal.cpp
+++ b/pv/data/decodesignal.cpp
@@ -68,6 +68,20 @@ DecodeSignal::~DecodeSignal()
        reset_decode(true);
 }
 
+void DecodeSignal::set_name(QString name)
+{
+       SignalBase::set_name(name);
+
+       update_output_signals();
+}
+
+void DecodeSignal::set_color(QColor color)
+{
+       SignalBase::set_color(color);
+
+       update_output_signals();
+}
+
 const vector< shared_ptr<Decoder> >& DecodeSignal::decoder_stack() const
 {
        return stack_;
@@ -168,6 +182,10 @@ void DecodeSignal::reset_decode(bool shutting_down)
        current_segment_id_ = 0;
        segments_.clear();
 
+       for (const shared_ptr<decode::Decoder>& dec : stack_)
+               if (dec->has_logic_output())
+                       output_logic_[dec->get_srd_decoder()]->clear();
+
        logic_mux_data_.reset();
        logic_mux_data_invalid_ = true;
 
@@ -380,11 +398,12 @@ void DecodeSignal::update_output_signals()
 
                        if (!ch_exists) {
                                shared_ptr<Logic> logic_data = make_shared<Logic>(logic_channels.size());
-                               logic_data->set_samplerate(first_ch.samplerate);
+                               logic_data->set_samplerate(get_samplerate());
                                output_logic_[dec->get_srd_decoder()] = logic_data;
+                               output_logic_muxed_data_[dec->get_srd_decoder()] = vector<uint8_t>();
 
                                shared_ptr<LogicSegment> logic_segment = make_shared<data::LogicSegment>(
-                                       *logic_data, 0, (logic_data->num_channels() + 7) / 8, first_ch.samplerate);
+                                       *logic_data, 0, (logic_data->num_channels() + 7) / 8, get_samplerate());
                                logic_data->push_segment(logic_segment);
 
                                uint index = 0;
@@ -392,20 +411,27 @@ void DecodeSignal::update_output_signals()
                                        shared_ptr<data::SignalBase> signal =
                                                make_shared<data::SignalBase>(nullptr, LogicChannel);
                                        signal->set_internal_name(logic_ch.id);
-                                       signal->set_name(logic_ch.id);
                                        signal->set_index(index);
                                        signal->set_data(logic_data);
                                        output_signals_.push_back(signal);
                                        session_.add_generated_signal(signal);
                                        index++;
                                }
+                       } else {
+                               shared_ptr<Logic> logic_data = output_logic_[dec->get_srd_decoder()];
+                               logic_data->set_samplerate(get_samplerate());
+                               for (shared_ptr<LogicSegment>& segment : logic_data->logic_segments())
+                                       segment->set_samplerate(get_samplerate());
                        }
                }
        }
 
-       // TODO Delete signals that no longer have a corresponding decoder (also from session)
-       // TODO Check whether all sample rates are the same as the session's
-       // TODO Set colors to the same as the decoder's background color
+       for (shared_ptr<SignalBase> s : output_signals_) {
+               s->set_name(s->internal_name() + " (" + name() + ")");
+               s->set_color(color());
+       }
+
+       // TODO Assert that all sample rates are the same as the session's
 }
 
 void DecodeSignal::set_initial_pin_state(const uint16_t channel_id, const int init_state)
@@ -1417,6 +1443,9 @@ void DecodeSignal::start_srd_session()
                return;
        }
 
+       // Update the samplerates for the output logic channels
+       update_output_signals();
+
        // Create the session
        srd_session_new(&srd_session_);
        assert(srd_session_);
@@ -1723,6 +1752,13 @@ void DecodeSignal::logic_output_callback(srd_proto_data *pdata, void *decode_sig
        const srd_proto_data_logic *const pdl = (const srd_proto_data_logic*)pdata->data;
        assert(pdl);
 
+       // FIXME Only one group supported for now
+       if (pdl->logic_group > 0) {
+               qWarning() << "Received logic output state change for group" << pdl->logic_group << "from decoder" \
+                       << QString::fromUtf8(decc->name) << "but only group 0 is currently supported";
+               return;
+       }
+
        shared_ptr<Logic> output_logic = ds->output_logic_.at(decc);
 
        vector< shared_ptr<Segment> > segments = output_logic->segments();
@@ -1733,16 +1769,37 @@ void DecodeSignal::logic_output_callback(srd_proto_data *pdata, void *decode_sig
                last_segment = dynamic_pointer_cast<LogicSegment>(segments.back());
        else {
                // Happens when the data was cleared - all segments are gone then
+               // segment_id is always 0 as it's the first segment
                last_segment = make_shared<data::LogicSegment>(
                        *output_logic, 0, (output_logic->num_channels() + 7) / 8, output_logic->get_samplerate());
                output_logic->push_segment(last_segment);
        }
 
-       last_segment->append_subsignal_payload(pdl->logic_class, (void*)pdl->data, pdl->size);
-
-       qInfo() << "Received" << pdl->size << "bytes /" << pdl->size \
-               << "samples of logic output for class" << pdl->logic_class << "from decoder" \
-               << QString::fromUtf8(decc->name);
+       if (pdata->start_sample < pdata->end_sample) {
+               vector<uint8_t> data;
+               const unsigned int unit_size = last_segment->unit_size();
+               data.resize(unit_size * (1 + pdl->repeat_count));
+
+               if (unit_size == 1)
+                       for (unsigned int i = 0; i <= pdl->repeat_count; i++)
+                               data.data()[i * unit_size] = *((uint8_t*)pdl->data);
+               else if (unit_size == 2)
+                       for (unsigned int i = 0; i <= pdl->repeat_count; i++)
+                               data.data()[i * unit_size] = *((uint16_t*)pdl->data);
+               else if (unit_size <= 4)
+                       for (unsigned int i = 0; i <= pdl->repeat_count; i++)
+                               data.data()[i * unit_size] = *((uint32_t*)pdl->data);
+               else if (unit_size <= 8)
+                       for (unsigned int i = 0; i <= pdl->repeat_count; i++)
+                               data.data()[i * unit_size] = *((uint64_t*)pdl->data);
+               else
+                       for (unsigned int i = 0; i <= pdl->repeat_count; i++)
+                               memcpy((void*)&data.data()[i * unit_size], (void*)pdl->data, unit_size);
+
+               last_segment->append_payload(data.data(), data.size());
+       } else
+               qWarning() << "Ignoring malformed logic output state change for group" << pdl->logic_group << "from decoder" \
+                       << QString::fromUtf8(decc->name) << "from" << pdata->start_sample << "to" << pdata->end_sample;
 }
 
 void DecodeSignal::on_capture_state_changed(int state)