qwt_plot_series_data_picker.h

/******************************************************************************
 * Qwt Widget Library
 * Copyright (C) 2024   ChenZongYan <czy.t@163.com>
 *****************************************************************************/
#ifndef QWT_PLOT_SERIES_DATA_PICKER_H
#define QWT_PLOT_SERIES_DATA_PICKER_H
#include <QList>
#include <QPointF>
#include "qwt_canvas_picker.h"
#include "qwt_text.h"
class QwtPlot;
class QwtPlotItem;
class QWT_EXPORT QwtPlotSeriesDataPicker : public QwtCanvasPicker
{
    Q_OBJECT
    QWT_DECLARE_PRIVATE(QwtPlotSeriesDataPicker)
public:
    enum PickSeriesMode
    {
        PickYValue,  
        PickNearestPoint  
    };
 
    enum TextPlacement
    {
        TextPlaceAuto,         
        TextFollowOnTop,       
        TextFollowOnBottom,    
        TextFollowMouse,       
        TextOnCanvasTopRight,  
        TextOnCanvasTopLeft,   
        TextOnCanvasBottomRight,  
        TextOnCanvasBottomLeft,   
        TextOnCanvasTopAuto,  
        TextOnCanvasBottomAuto  
    };
 
    enum InterpolationMode
    {
        NoInterpolation,     
        LinearInterpolation  
    };
 
    struct FeaturePoint
    {
        QwtPlotItem* item;  
        QPointF feature;    
        size_t index;       
    };
 
public:
    explicit QwtPlotSeriesDataPicker(QWidget* canvas);
    ~QwtPlotSeriesDataPicker();
 
    // 拾取模式
    void setPickMode(PickSeriesMode mode);
    PickSeriesMode pickMode() const;
 
    // 设置文字显示的位置
    void setTextArea(TextPlacement t);
    TextPlacement textArea() const;
 
    // 插值模式
    void setInterpolationMode(InterpolationMode mode);
    InterpolationMode interpolationMode() const;
    // 判断是否插值
    bool isInterpolation() const;
 
    // 临近点搜索窗口大小，窗口大小决定了临近点搜索的范围，避免全曲线遍历
    void setNearestSearchWindowSize(int windowSize);
    int nearestSearchWindowSize() const;
 
    // 是否绘制特征点,如果是，picker会把捕获的特征点绘制在曲线上
    void setEnableDrawFeaturePoint(bool on = true);
    bool isEnableDrawFeaturePoint() const;
 
    // 设置绘制的特征点的大小
    void setDrawFeaturePointSize(int px);
    int drawFeaturePointSize() const;
 
    // 设置文字的背景颜色
    void setTextBackgroundBrush(const QBrush& br);
    QBrush textBackgroundBrush() const;
 
    // 文字的对其方式
    void setTextAlignment(Qt::Alignment al);
    Qt::Alignment textAlignment() const;
 
    // 是否显示x值
    void setEnableShowXValue(bool on);
    bool isEnableShowXValue() const;
 
    // 顶部矩形文字
    QwtText trackerText(const QPoint& pos) const QWT_OVERRIDE;
 
    // 让矩形在最顶部
    QRect trackerRect(const QFont& f) const QWT_OVERRIDE;
 
    // 绘制rubberband
    virtual void drawRubberBand(QPainter* painter) const QWT_OVERRIDE;
 
    // 手动设置位置
    virtual void setTrackerPosition(const QPoint& pos) QWT_OVERRIDE;
 
private:
    // 获取绘图区域屏幕坐标pos上，的所有可拾取的y值,返回获取的个数
    int pickYValue(const QwtPlot* plot, const QPoint& pos, bool interpolate = false);
    // 获取绘图区域屏幕坐标pos上，可拾取的最近的一个点，(基于窗口实现快速索引)
    int pickNearestPoint(const QwtPlot* plot, const QPoint& pos, int windowSize = -5);
private Q_SLOTS:
    // item删除的槽，用于更新记录
    void onPlotItemDetached(QwtPlotItem* item, bool on);
    void onParasitePlotAttached(QwtPlot* parasiteplot, bool on);
 
protected:
    // 生成一个item的文字内容
    virtual QString valueString(const QList< FeaturePoint >& fps) const;
    // 绘制特征点，所谓特征点就是捕获到的点
    virtual void drawFeaturePoints(QPainter* painter) const;
    // 鼠标移动
    virtual void move(const QPoint& pos) QWT_OVERRIDE;
    // 格式化为坐标轴对应的内容，针对时间轴，value是一个大浮点数，用户需要看到的是2024-10-01这样的数字
    QString formatAxisValue(double value, int axisId, QwtPlot* plot) const;
    // 更新特征点
    void updateFeaturePoint(const QPoint& pos);
};
 
#endif  // QWT_PLOT_SERIES_DATA_PICKER_H