pv/data/decode/rowdata.cpp

   1 /*
   2  * This file is part of the PulseView project.
   3  *
   4  * Copyright (C) 2014 Joel Holdsworth <joel@airwebreathe.org.uk>
   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
  20 #include <pv/data/decode/decoder.hpp>
  21 #include <pv/data/decode/row.hpp>
  22 #include <pv/data/decode/rowdata.hpp>
  23
  24 using std::vector;
  25
  26 namespace pv {
  27 namespace data {
  28 namespace decode {
  29
  30 RowData::RowData(Row* row) :
  31         row_(row),
  32         prev_ann_start_sample_(0)
  33 {
  34         assert(row);
  35 }
  36
  37 uint64_t RowData::get_max_sample() const
  38 {
  39         if (annotations_.empty())
  40                 return 0;
  41         return annotations_.back().end_sample();
  42 }
  43
  44 uint64_t RowData::get_annotation_count() const
  45 {
  46         return annotations_.size();
  47 }
  48
  49 void RowData::get_annotation_subset(
  50         vector<const pv::data::decode::Annotation*> &dest,
  51         uint64_t start_sample, uint64_t end_sample) const
  52 {
  53         // Determine whether we must apply per-class filtering or not
  54         bool all_ann_classes_enabled = true;
  55         bool all_ann_classes_disabled = true;
  56
  57         uint32_t max_ann_class_id = 0;
  58         for (AnnotationClass* c : row_->ann_classes()) {
  59                 if (!c->visible)
  60                         all_ann_classes_enabled = false;
  61                 else
  62                         all_ann_classes_disabled = false;
  63                 if (c->id > max_ann_class_id)
  64                         max_ann_class_id = c->id;
  65         }
  66
  67         if (all_ann_classes_enabled) {
  68                 // No filtering, send everyting out as-is
  69                 dest.reserve(dest.size() + annotations_.size());
  70                 for (const auto& annotation : annotations_)
  71                         if ((annotation.end_sample() > start_sample) &&
  72                                 (annotation.start_sample() <= end_sample))
  73                                 dest.push_back(&annotation);
  74         } else {
  75                 if (!all_ann_classes_disabled) {
  76                         // Filter out invisible annotation classes
  77                         vector<size_t> class_visible;
  78                         class_visible.resize(max_ann_class_id + 1, 0);
  79                         for (AnnotationClass* c : row_->ann_classes())
  80                                 if (c->visible)
  81                                         class_visible[c->id] = 1;
  82
  83                         dest.reserve(dest.size() + annotations_.size());
  84                         for (const auto& annotation : annotations_)
  85                                 if ((class_visible[annotation.ann_class()]) &&
  86                                         (annotation.end_sample() > start_sample) &&
  87                                         (annotation.start_sample() <= end_sample))
  88                                         dest.push_back(&annotation);
  89                 }
  90         }
  91 }
  92
  93 void RowData::emplace_annotation(srd_proto_data *pdata)
  94 {
  95         // We insert the annotation in a way so that the annotation list
  96         // is sorted by start sample. Otherwise, we'd have to sort when
  97         // painting, which is expensive
  98
  99         if (pdata->start_sample < prev_ann_start_sample_) {
 100                 // Find location to insert the annotation at
 101
 102                 auto it = annotations_.end();
 103                 do {
 104                         it--;
 105                 } while ((it->start_sample() > pdata->start_sample) && (it != annotations_.begin()));
 106
 107                 // Allow inserting at the front
 108                 if (it != annotations_.begin())
 109                         it++;
 110
 111                 annotations_.insert(it, Annotation(pdata, row_));
 112         } else {
 113                 annotations_.emplace_back(pdata, row_);
 114                 prev_ann_start_sample_ = pdata->start_sample;
 115         }
 116 }
 117
 118 }  // namespace decode
 119 }  // namespace data
 120 }  // namespace pv