pv/view/decode/annotation.cpp

   1 /*
   2  * This file is part of the PulseView project.
   3  *
   4  * Copyright (C) 2012 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, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  19  */
  20
  21 extern "C" {
  22 #include <libsigrokdecode/libsigrokdecode.h>
  23 }
  24
  25 #include <extdef.h>
  26
  27 #include <algorithm>
  28
  29 #include <boost/foreach.hpp>
  30
  31 #include <QPainter>
  32
  33 #include "annotation.h"
  34
  35 using namespace std;
  36
  37 namespace pv {
  38 namespace view {
  39 namespace decode {
  40
  41 const double Annotation::EndCapWidth = 5;
  42 const int Annotation::DrawPadding = 100;
  43
  44 const QColor Annotation::Colours[7] = {
  45         QColor(0xFC, 0xE9, 0x4F),       // Light Butter
  46         QColor(0xFC, 0xAF, 0x3E),       // Light Orange
  47         QColor(0xE9, 0xB9, 0x6E),       // Light Chocolate
  48         QColor(0x8A, 0xE2, 0x34),       // Light Green
  49         QColor(0x72, 0x9F, 0xCF),       // Light Blue
  50         QColor(0xAD, 0x7F, 0xA8),       // Light Plum
  51         QColor(0xEF, 0x29, 0x29)        // Light Red
  52 };
  53
  54 Annotation::Annotation(const srd_proto_data *const pdata) :
  55         _start_sample(pdata->start_sample),
  56         _end_sample(pdata->end_sample)
  57 {
  58         assert(pdata);
  59         const srd_proto_data_annotation *const pda =
  60                 (const srd_proto_data_annotation*)pdata->data;
  61         assert(pda);
  62
  63         const char *const *annotations = (char**)pda->ann_text;
  64         while(*annotations) {
  65                 _annotations.push_back(QString(*annotations));
  66                 annotations++;
  67         }
  68 }
  69
  70 void Annotation::paint(QPainter &p, QColor text_color, int text_height,
  71         int left, int right, double samples_per_pixel, double pixels_offset,
  72         int y)
  73 {
  74         const int h = (text_height * 3) / 2;
  75         const double start = _start_sample / samples_per_pixel -
  76                 pixels_offset;
  77         const double end = _end_sample / samples_per_pixel -
  78                 pixels_offset;
  79         const QColor fill = Colours[(_format * (countof(Colours) / 2 + 1)) %
  80                 countof(Colours)];
  81         const QColor outline(fill.darker());
  82
  83         if (start > right + DrawPadding || end < left - DrawPadding)
  84                 return;
  85
  86         if (_start_sample == _end_sample)
  87                 draw_instant(p, fill, outline, text_color, h,
  88                         start, y);
  89         else
  90                 draw_range(p, fill, outline, text_color, h,
  91                         start, end, y);
  92 }
  93
  94 void Annotation::draw_instant(QPainter &p, QColor fill, QColor outline,
  95         QColor text_color, int h, double x, int y)
  96 {
  97         const QString text = _annotations.empty() ?
  98                 QString() : _annotations.back();
  99         const double w = min(p.boundingRect(QRectF(), 0, text).width(),
 100                 0.0) + h;
 101         const QRectF rect(x - w / 2, y - h / 2, w, h);
 102
 103         p.setPen(outline);
 104         p.setBrush(fill);
 105         p.drawRoundedRect(rect, h / 2, h / 2);
 106
 107         p.setPen(text_color);
 108         p.drawText(rect, Qt::AlignCenter | Qt::AlignVCenter, text);
 109 }
 110
 111 void Annotation::draw_range(QPainter &p, QColor fill, QColor outline,
 112         QColor text_color, int h, double start, double end, int y)
 113 {
 114         const double cap_width = min((end - start) / 2, EndCapWidth);
 115
 116         QPointF pts[] = {
 117                 QPointF(start, y + .5f),
 118                 QPointF(start + cap_width, y + .5f - h / 2),
 119                 QPointF(end - cap_width, y + .5f - h / 2),
 120                 QPointF(end, y + .5f),
 121                 QPointF(end - cap_width, y + .5f + h / 2),
 122                 QPointF(start + cap_width, y + .5f + h / 2)
 123         };
 124
 125         p.setPen(outline);
 126         p.setBrush(fill);
 127         p.drawConvexPolygon(pts, countof(pts));
 128
 129         if (_annotations.empty())
 130                 return;
 131
 132         QRectF rect(start + cap_width, y - h / 2,
 133                 end - start - cap_width * 2, h);
 134         p.setPen(text_color);
 135
 136         // Try to find an annotation that will fit
 137         QString best_annotation;
 138         int best_width = 0;
 139
 140         BOOST_FOREACH(QString &a, _annotations) {
 141                 const int w = p.boundingRect(QRectF(), 0, a).width();
 142                 if (w <= rect.width() && w > best_width)
 143                         best_annotation = a, best_width = w;
 144         }
 145
 146         if (best_annotation.isEmpty())
 147                 best_annotation = _annotations.back();
 148
 149         // If not ellide the last in the list
 150         p.drawText(rect, Qt::AlignCenter, p.fontMetrics().elidedText(
 151                 best_annotation, Qt::ElideRight, rect.width()));
 152 }
 153
 154 } // namespace decode
 155 } // namespace view
 156 } // namespace pv