Split annotation data set into rows, and improved painting.

[pulseview.git] / pv / data / decode / rowdata.cpp

diff --git a/pv/data/decode/rowdata.cpp b/pv/data/decode/rowdata.cpp
new file mode 100644 (file)
index 0000000..a06af61
--- /dev/null
+++ b/pv/data/decode/rowdata.cpp
@@ -0,0 +1,118 @@
+/*
+ * This file is part of the PulseView project.
+ *
+ * Copyright (C) 2014 Joel Holdsworth <joel@airwebreathe.org.uk>
+ *
+ * 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, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
+ */
+
+#include <boost/bind.hpp>
+
+#include <assert.h>
+
+#include "rowdata.h"
+
+using std::vector;
+
+namespace pv {
+namespace data {
+namespace decode {
+
+RowData::RowData()
+{
+}
+
+uint64_t RowData::get_max_sample() const
+{
+       if (_annotations.empty())
+               return 0;
+       return _annotations.back().end_sample();
+}
+
+void RowData::get_annotation_subset(
+       std::vector<pv::data::decode::Annotation> &dest,
+       uint64_t start_sample, uint64_t end_sample) const
+{
+       const vector<size_t>::const_iterator start_iter =
+               lower_bound(_ann_end_index.begin(),
+                       _ann_end_index.end(), start_sample,
+                       bind(&RowData::index_entry_end_sample_lt,
+                               this, _1, _2));
+
+       const vector<size_t>::const_iterator end_iter =
+               upper_bound(_ann_start_index.begin(),
+                       _ann_start_index.end(), end_sample,
+                       bind(&RowData::index_entry_start_sample_gt,
+                               this, _1, _2));
+
+       for (vector<size_t>::const_iterator i = start_iter;
+               i != _ann_end_index.end() && *i != *end_iter; i++)
+               dest.push_back(_annotations[*i]);
+}
+
+void RowData::push_annotation(const Annotation &a)
+{
+       const size_t offset = _annotations.size();
+       _annotations.push_back(a);
+
+       // Insert the annotation into the start index
+       vector<size_t>::iterator i = _ann_start_index.end();
+       if (!_ann_start_index.empty() &&
+               _annotations[_ann_start_index.back()].start_sample() >
+                       a.start_sample())
+               i = upper_bound(_ann_start_index.begin(),
+                       _ann_start_index.end(), a.start_sample(),
+                       bind(&RowData::index_entry_start_sample_gt,
+                               this, _1, _2));
+
+       _ann_start_index.insert(i, offset);
+
+       // Insert the annotation into the end index
+       vector<size_t>::iterator j = _ann_end_index.end();
+       if (!_ann_end_index.empty() &&
+               _annotations[_ann_end_index.back()].end_sample() <
+                       a.end_sample())
+               j = upper_bound(_ann_end_index.begin(),
+                       _ann_end_index.end(), a.end_sample(),
+                       bind(&RowData::index_entry_end_sample_gt,
+                               this, _1, _2));
+
+       _ann_end_index.insert(j, offset);
+}
+
+bool RowData::index_entry_start_sample_gt(
+       const uint64_t sample, const size_t index) const
+{
+       assert(index < _annotations.size());
+       return _annotations[index].start_sample() > sample;
+}
+
+bool RowData::index_entry_end_sample_lt(
+       const size_t index, const uint64_t sample) const
+{
+       assert(index < _annotations.size());
+       return _annotations[index].end_sample() < sample;
+}
+
+bool RowData::index_entry_end_sample_gt(
+       const uint64_t sample, const size_t index) const
+{
+       assert(index < _annotations.size());
+       return _annotations[index].end_sample() > sample;
+}
+
+} // decode
+} // data
+} // pv