* along with this program; if not, see <http://www.gnu.org/licenses/>.
*/
+#include <QApplication>
+#include <QDebug>
#include <QString>
#include "pv/views/tabular_decoder/view.hpp"
+#include "view.hpp"
+
+#include "pv/util.hpp"
+#include "pv/globalsettings.hpp"
+
using std::make_shared;
+using pv::util::Timestamp;
+using pv::util::format_time_si;
+using pv::util::format_time_minutes;
+using pv::util::SIPrefix;
+
namespace pv {
namespace views {
namespace tabular_decoder {
AnnotationCollectionModel::AnnotationCollectionModel(QObject* parent) :
- QAbstractItemModel(parent)
+ QAbstractTableModel(parent),
+ all_annotations_(nullptr),
+ dataset_(nullptr),
+ signal_(nullptr),
+ first_hidden_column_(0),
+ prev_segment_(0),
+ prev_last_row_(0),
+ had_highlight_before_(false),
+ hide_hidden_(false)
+{
+ // Note: when adding entries, consider ViewVisibleFilterProxyModel::filterAcceptsRow()
+
+ uint8_t i = 0;
+ header_data_.emplace_back(tr("Sample")); i++; // Column #0
+ header_data_.emplace_back(tr("Time")); i++; // Column #1
+ header_data_.emplace_back(tr("Decoder")); i++; // Column #2
+ header_data_.emplace_back(tr("Ann Row")); i++; // Column #3
+ header_data_.emplace_back(tr("Ann Class")); i++; // Column #4
+ header_data_.emplace_back(tr("Value")); i++; // Column #5
+
+ first_hidden_column_ = i;
+ header_data_.emplace_back("End Sample"); // Column #6, hidden
+}
+
+int AnnotationCollectionModel::get_hierarchy_level(const Annotation* ann) const
{
- vector<QVariant> header_data;
- header_data.emplace_back(tr("ID")); // Column #0
- header_data.emplace_back(tr("Start Time")); // Column #1
- header_data.emplace_back(tr("End Time")); // Column #2
- header_data.emplace_back(tr("Ann Row Name")); // Column #3
- header_data.emplace_back(tr("Class Row Name")); // Column #4
- header_data.emplace_back(tr("Value")); // Column #5
- root_ = make_shared<AnnotationCollectionItem>(header_data);
+ int level = 0;
+
+ const unsigned int ann_stack_level = ann->row_data()->row()->decoder()->get_stack_level();
+ level = (signal_->decoder_stack().size() - 1 - ann_stack_level);
+
+ return level;
+}
+
+QVariant AnnotationCollectionModel::data_from_ann(const Annotation* ann, int index) const
+{
+ switch (index) {
+ case 0: return QVariant((qulonglong)ann->start_sample()); // Column #0, Start Sample
+ case 1: { // Column #1, Start Time
+ Timestamp t = ann->start_sample() / signal_->get_samplerate();
+ QString unit = signal_->get_samplerate() ? tr("s") : tr("sa");
+ QString s;
+ if ((t < 60) || (signal_->get_samplerate() == 0)) // i.e. if unit is sa
+ s = format_time_si(t, SIPrefix::unspecified, 3, unit, false);
+ else
+ s = format_time_minutes(t, 3, false);
+ return QVariant(s);
+ }
+ case 2: return QVariant(ann->row()->decoder()->name()); // Column #2, Decoder
+ case 3: return QVariant(ann->row()->description()); // Column #3, Ann Row
+ case 4: return QVariant(ann->ann_class_description()); // Column #4, Ann Class
+ case 5: return QVariant(ann->longest_annotation()); // Column #5, Value
+ case 6: return QVariant((qulonglong)ann->end_sample()); // Column #6, End Sample
+ default: return QVariant();
+ }
}
QVariant AnnotationCollectionModel::data(const QModelIndex& index, int role) const
{
- if (!index.isValid())
+ if (!signal_ || !index.isValid() || !index.internalPointer())
return QVariant();
- if (role == Qt::DisplayRole)
- {
- AnnotationCollectionItem* item =
- static_cast<AnnotationCollectionItem*>(index.internalPointer());
+ const Annotation* ann =
+ static_cast<const Annotation*>(index.internalPointer());
+
+ if ((role == Qt::DisplayRole) || (role == Qt::ToolTipRole))
+ return data_from_ann(ann, index.column());
+
+ if (role == Qt::ForegroundRole) {
+ if (index.column() >= get_hierarchy_level(ann)) {
+ // Invert the text color if this cell is highlighted
+ const bool must_highlight = (highlight_sample_num_ > 0) &&
+ ((int64_t)ann->start_sample() <= highlight_sample_num_) &&
+ ((int64_t)ann->end_sample() >= highlight_sample_num_);
+
+ if (must_highlight) {
+ if (GlobalSettings::current_theme_is_dark())
+ return QApplication::palette().brush(QPalette::Window);
+ else
+ return QApplication::palette().brush(QPalette::WindowText);
+ }
+ }
- return item->data(index.column());
+ return QApplication::palette().brush(QPalette::WindowText);
}
- if ((role == Qt::FontRole) && (index.parent().isValid()) && (index.column() == 0))
- {
- QFont font;
- font.setItalic(true);
- return QVariant(font);
+ if (role == Qt::BackgroundRole) {
+ // Only use custom cell background color if column index reached the hierarchy level
+ if (index.column() >= get_hierarchy_level(ann)) {
+
+ QColor color;
+ const bool must_highlight = (highlight_sample_num_ > 0) &&
+ ((int64_t)ann->start_sample() <= highlight_sample_num_) &&
+ ((int64_t)ann->end_sample() >= highlight_sample_num_);
+
+ if (must_highlight)
+ color = ann->color();
+ else
+ color = GlobalSettings::current_theme_is_dark() ?
+ ann->dark_color() : ann->bright_color();
+
+ return QBrush(color);
+ }
}
return QVariant();
Qt::ItemFlags AnnotationCollectionModel::flags(const QModelIndex& index) const
{
if (!index.isValid())
- return nullptr;
+ return Qt::NoItemFlags;
- return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+ return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemNeverHasChildren;
+}
+
+uint8_t AnnotationCollectionModel::first_hidden_column() const
+{
+ return first_hidden_column_;
}
QVariant AnnotationCollectionModel::headerData(int section, Qt::Orientation orientation,
int role) const
{
- if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
- return root_->data(section);
+ if ((section < 0) || (section >= (int)header_data_.size()))
+ return QVariant();
+
+ if ((orientation == Qt::Horizontal) && (role == Qt::DisplayRole))
+ return header_data_[section];
return QVariant();
}
QModelIndex AnnotationCollectionModel::index(int row, int column,
const QModelIndex& parent_idx) const
{
- if (!hasIndex(row, column, parent_idx))
- return QModelIndex();
+ (void)parent_idx;
+ assert(column >= 0);
- AnnotationCollectionItem* parent = root_.get();
+ if (!dataset_ || (row < 0))
+ return QModelIndex();
- if (parent_idx.isValid())
- parent = static_cast<AnnotationCollectionItem*>(parent_idx.internalPointer());
+ QModelIndex idx;
- AnnotationCollectionItem* subItem = parent->subItem(row).get();
+ if ((size_t)row < dataset_->size())
+ idx = createIndex(row, column, (void*)dataset_->at(row));
- return subItem ? createIndex(row, column, subItem) : QModelIndex();
+ return idx;
}
QModelIndex AnnotationCollectionModel::parent(const QModelIndex& index) const
{
- if (!index.isValid())
- return QModelIndex();
-
- AnnotationCollectionItem* subItem =
- static_cast<AnnotationCollectionItem*>(index.internalPointer());
-
- shared_ptr<AnnotationCollectionItem> parent = subItem->parent();
+ (void)index;
- return (parent == root_) ? QModelIndex() :
- createIndex(parent->row(), 0, parent.get());
+ return QModelIndex();
}
int AnnotationCollectionModel::rowCount(const QModelIndex& parent_idx) const
{
- AnnotationCollectionItem* parent = root_.get();
+ (void)parent_idx;
- if (parent_idx.column() > 0)
+ if (!dataset_)
return 0;
- if (parent_idx.isValid())
- parent = static_cast<AnnotationCollectionItem*>(parent_idx.internalPointer());
-
- return parent->subItemCount();
+ return dataset_->size();
}
int AnnotationCollectionModel::columnCount(const QModelIndex& parent_idx) const
{
- if (parent_idx.isValid())
- return static_cast<AnnotationCollectionItem*>(
- parent_idx.internalPointer())->columnCount();
+ (void)parent_idx;
+
+ return header_data_.size();
+}
+
+void AnnotationCollectionModel::set_signal_and_segment(data::DecodeSignal* signal, uint32_t current_segment)
+{
+ layoutAboutToBeChanged();
+
+ if (!signal) {
+ all_annotations_ = nullptr;
+ dataset_ = nullptr;
+ signal_ = nullptr;
+
+ dataChanged(QModelIndex(), QModelIndex());
+ layoutChanged();
+ return;
+ }
+
+ disconnect(this, SLOT(on_annotation_visibility_changed()));
+
+ all_annotations_ = signal->get_all_annotations_by_segment(current_segment);
+ signal_ = signal;
+
+ for (const shared_ptr<Decoder>& dec : signal_->decoder_stack())
+ connect(dec.get(), SIGNAL(annotation_visibility_changed()),
+ this, SLOT(on_annotation_visibility_changed()));
+
+ if (hide_hidden_)
+ update_annotations_without_hidden();
else
- return root_->columnCount();
+ dataset_ = all_annotations_;
+
+ if (!dataset_ || dataset_->empty()) {
+ prev_segment_ = current_segment;
+ return;
+ }
+
+ const size_t new_row_count = dataset_->size() - 1;
+
+ // Force the view associated with this model to update when the segment changes
+ if (prev_segment_ != current_segment) {
+ dataChanged(index(0, 0), index(new_row_count, 0));
+ layoutChanged();
+ } else {
+ // Force the view associated with this model to update when we have more annotations
+ if (prev_last_row_ < new_row_count) {
+ dataChanged(index(prev_last_row_, 0), index(new_row_count, 0));
+ layoutChanged();
+ }
+ }
+
+ prev_segment_ = current_segment;
+ prev_last_row_ = new_row_count;
+}
+
+void AnnotationCollectionModel::set_hide_hidden(bool hide_hidden)
+{
+ layoutAboutToBeChanged();
+
+ hide_hidden_ = hide_hidden;
+
+ if (hide_hidden_) {
+ dataset_ = &all_annotations_without_hidden_;
+ update_annotations_without_hidden();
+ } else {
+ dataset_ = all_annotations_;
+ all_annotations_without_hidden_.clear(); // To conserve memory
+ }
+
+ if (dataset_)
+ dataChanged(index(0, 0), index(dataset_->size() - 1, 0));
+ else
+ dataChanged(QModelIndex(), QModelIndex());
+
+ layoutChanged();
+}
+
+void AnnotationCollectionModel::update_annotations_without_hidden()
+{
+ uint64_t count = 0;
+
+ if (!all_annotations_ || all_annotations_->empty()) {
+ all_annotations_without_hidden_.clear();
+ return;
+ }
+
+ for (const Annotation* ann : *all_annotations_) {
+ if (!ann->visible())
+ continue;
+
+ if (all_annotations_without_hidden_.size() < (count + 100))
+ all_annotations_without_hidden_.resize(count + 100);
+
+ all_annotations_without_hidden_[count++] = ann;
+ }
+
+ all_annotations_without_hidden_.resize(count);
+}
+
+QModelIndex AnnotationCollectionModel::update_highlighted_rows(QModelIndex first,
+ QModelIndex last, int64_t sample_num)
+{
+ bool has_highlight = false;
+ QModelIndex result;
+
+ highlight_sample_num_ = sample_num;
+
+ if (!dataset_ || dataset_->empty())
+ return result;
+
+ if (sample_num >= 0) {
+ last = last.sibling(last.row() + 1, 0);
+
+ // Check if there are any annotations visible in the table view that
+ // we would need to highlight - only then do we do so
+ QModelIndex index = first;
+ do {
+ const Annotation* ann = static_cast<const Annotation*>(index.internalPointer());
+ if (!ann) // Can happen if the table is being modified at this exact time
+ return result;
+
+ if (((int64_t)ann->start_sample() <= sample_num) &&
+ ((int64_t)ann->end_sample() >= sample_num)) {
+ result = index;
+ has_highlight = true;
+ break;
+ }
+
+ index = index.sibling(index.row() + 1, 0);
+ } while (index != last);
+ }
+
+ if (has_highlight || had_highlight_before_)
+ dataChanged(first, last);
+
+ had_highlight_before_ = has_highlight;
+
+ return result;
}
+void AnnotationCollectionModel::on_annotation_visibility_changed()
+{
+ if (!hide_hidden_)
+ return;
+
+ layoutAboutToBeChanged();
+
+ update_annotations_without_hidden();
+
+ if (dataset_)
+ dataChanged(index(0, 0), index(dataset_->size() - 1, 0));
+ else
+ dataChanged(QModelIndex(), QModelIndex());
+
+ layoutChanged();
+}
} // namespace tabular_decoder
} // namespace views