]> sigrok.org Git - pulseview.git/blobdiff - pv/view/decodetrace.cpp
DecodeTrace: Try to keep annotation labels within the view
[pulseview.git] / pv / view / decodetrace.cpp
index f2aadd1b18ff3c7e4b8904a53aeef1b8f7c8d0a9..eb1f7f83d01324d1527e81ffdb890e17a2194dcd 100644 (file)
@@ -337,33 +337,63 @@ void DecodeTrace::draw_annotations(vector<pv::data::decode::Annotation> annotati
        using namespace pv::data::decode;
 
        vector<Annotation> a_block;
-       int prev_ann_pos = INT_MIN;
+       int p_end = INT_MIN;
 
        double samples_per_pixel, pixels_offset;
        tie(pixels_offset, samples_per_pixel) =
                get_pixels_offset_samples_per_pixel();
 
+       // Sort the annotations by start sample so that decoders
+       // can't confuse us by creating annotations out of order
+       stable_sort(annotations.begin(), annotations.end(),
+               [](const Annotation &a, const Annotation &b) {
+                       return a.start_sample() < b.start_sample(); });
+
        // Gather all annotations that form a visual "block" and draw them as such
        for (const Annotation &a : annotations) {
 
-               const int end = a.end_sample() / samples_per_pixel - pixels_offset;
-               const int delta = end - prev_ann_pos;
+               const int a_start = a.start_sample() / samples_per_pixel - pixels_offset;
+               const int a_end = a.end_sample() / samples_per_pixel - pixels_offset;
+               const int a_width = a_end - a_start;
+
+               const int delta = a_end - p_end;
+
+               bool a_is_separate = false;
+
+               // Annotation wider than the threshold for a useful label width?
+               if (a_width > 20) {
+                       for (const QString &ann_text : a.annotations()) {
+                               const int w = p.boundingRect(QRectF(), 0, ann_text).width();
+                               // Annotation wide enough to fit a label? Don't put it in a block then
+                               if (w <= a_width) {
+                                       a_is_separate = true;
+                                       break;
+                               }
+                       }
+               }
 
-               // Some annotations are in reverse order, so we cannot
-               // simply check for delta > 1
-               if (abs(delta) > 1) {
-                       // Block was broken, draw it
-                       if (a_block.size() == 1)
+               // Were the previous and this annotation more than a pixel apart?
+               if ((abs(delta) > 1) || a_is_separate) {
+                       // Block was broken, draw annotations that form the current block
+                       if (a_block.size() == 1) {
                                draw_annotation(a_block.front(), p, h, pp, y, base_colour);
+                       }
                        else
-                               if (a_block.size() > 0)
-                                       draw_annotation_block(a_block, p, h, y, base_colour);
+                               draw_annotation_block(a_block, p, h, y, base_colour);
 
                        a_block.clear();
                }
 
-               a_block.push_back(a);
-               prev_ann_pos = end;
+               if (a_is_separate) {
+                       draw_annotation(a, p, h, pp, y, base_colour);
+                       // Next annotation must start a new block. delta will be > 1
+                       // because we set p_end to INT_MIN but that's okay since
+                       // a_block will be empty, so nothing will be drawn
+                       p_end = INT_MIN;
+               } else {
+                       a_block.push_back(a);
+                       p_end = a_end;
+               }
        }
 
        if (a_block.size() == 1)
@@ -395,7 +425,7 @@ void DecodeTrace::draw_annotation(const pv::data::decode::Annotation &a,
        if (a.start_sample() == a.end_sample())
                draw_instant(a, p, fill, outline, h, start, y);
        else
-               draw_range(a, p, fill, outline, h, start, end, y);
+               draw_range(a, p, fill, outline, h, start, end, y, pp);
 }
 
 void DecodeTrace::draw_annotation_block(
@@ -404,6 +434,9 @@ void DecodeTrace::draw_annotation_block(
 {
        using namespace pv::data::decode;
 
+       if (annotations.empty())
+               return;
+
        double samples_per_pixel, pixels_offset;
        tie(pixels_offset, samples_per_pixel) =
                get_pixels_offset_samples_per_pixel();
@@ -415,36 +448,22 @@ void DecodeTrace::draw_annotation_block(
 
        const double top = y + .5 - h / 2;
        const double bottom = y + .5 + h / 2;
-       const double cap_width = min((end - start) / 4, EndCapWidth);
-
-       QPointF pts[] = {
-               QPointF(start, y + .5f),
-               QPointF(start + cap_width, top),
-               QPointF(end - cap_width, top),
-               QPointF(end, y + .5f),
-               QPointF(end - cap_width, bottom),
-               QPointF(start + cap_width, bottom)
-       };
 
        const size_t colour = (base_colour + annotations.front().format()) %
                countof(Colours);
 
        // Check if all annotations are of the same type (i.e. we can use one color)
        // or if we should use a neutral color (i.e. gray)
-       bool single_format = true;
-       int format = annotations.front().format();
-
-       for (const Annotation &a : annotations)
-               if (a.format() != format) {
-                       single_format = false;
-                       break;
-               }
+       const int format = annotations.front().format();
+       const bool single_format = std::all_of(
+               annotations.begin(), annotations.end(),
+               [&](const Annotation &a) { return a.format() == format; });
 
        p.setPen((single_format ? OutlineColours[colour] : Qt::gray));
        p.setBrush(QBrush((single_format ? Colours[colour] : Qt::gray),
                Qt::Dense4Pattern));
-
-       p.drawConvexPolygon(pts, countof(pts));
+       p.drawRoundedRect(
+               QRectF(start, top, end - start, bottom - top), h/4, h/4);
 }
 
 void DecodeTrace::draw_instant(const pv::data::decode::Annotation &a, QPainter &p,
@@ -466,7 +485,7 @@ void DecodeTrace::draw_instant(const pv::data::decode::Annotation &a, QPainter &
 
 void DecodeTrace::draw_range(const pv::data::decode::Annotation &a, QPainter &p,
        QColor fill, QColor outline, int h, double start,
-       double end, int y) const
+       double end, int y, const ViewItemPaintParams &pp) const
 {
        const double top = y + .5 - h / 2;
        const double bottom = y + .5 + h / 2;
@@ -497,8 +516,15 @@ void DecodeTrace::draw_range(const pv::data::decode::Annotation &a, QPainter &p,
        if (annotations.empty())
                return;
 
-       QRectF rect(start + cap_width, y - h / 2,
-               end - start - cap_width * 2, h);
+       const int ann_start = start + cap_width;
+       const int ann_end = end - cap_width;
+
+       const int real_start = std::max(ann_start, pp.left());
+       const int real_end = std::min(ann_end, pp.right());
+
+       const int real_width = real_end - real_start;
+
+       QRectF rect(real_start, y - h / 2, real_width, h);
        if (rect.width() <= 4)
                return;