summaryrefslogtreecommitdiffhomepage
path: root/spline-widget/spline-widget.hpp
blob: 1dbdb8e14253d5a96341137059d0b8c1980d4e2c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/* Copyright (c) 2012-2016 Stanislaw Halik <sthalik@misaki.pl>
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 */

// Adapted to FaceTrackNoIR by Wim Vriend.

#pragma once

#include <QWidget>
#include <QtGui>
#include <QPointF>
#include "spline.hpp"
#include "api/plugin-api.hpp"

class SPLINE_WIDGET_EXPORT spline_widget : public QWidget
{
    Q_OBJECT
    Q_PROPERTY(QColor colorBezier READ colorBezier WRITE setColorBezier)
    Q_PROPERTY(bool is_preview_only READ is_preview_only WRITE set_preview_only)
public:
    spline_widget(QWidget *parent = 0);

    spline* config();
    void setConfig(spline* config, const QString &name);

    QColor colorBezier() const
    {
        return spline_color;
    }
    void setColorBezier(QColor color)
    {
        spline_color = color;
        update();
    }
    void force_redraw()
    {
        _background = QPixmap();
        update();
    }
    void set_preview_only(bool val);
    bool is_preview_only() const;
    void set_snap(int x, int y) { snap_x = x; snap_y = y; }
    void get_snap(int& x, int& y) const { x = snap_x; y = snap_y; }
protected slots:
    void paintEvent(QPaintEvent *e) override;
    void mousePressEvent(QMouseEvent *e) override;
    void mouseMoveEvent(QMouseEvent *e) override;
    void mouseReleaseEvent(QMouseEvent *e) override;
private:
    void drawBackground();
    void drawFunction();
    void drawPoint(QPainter *painter, const QPointF &pt, QColor colBG, QColor border = QColor(50, 100, 120, 200));
    void drawLine(QPainter *painter, const QPointF &start, const QPointF &end, QPen& pen);
    bool point_within_pixel(const QPointF& pt, const QPointF& pixel);
protected:
    void resizeEvent(QResizeEvent *) override;
private:
    void update_range();

    QPointF pixel_coord_to_point (const QPointF& point);
    QPointF point_to_pixel(const QPointF& point);

    spline* _config;

    // bounds of the rectangle user can interact with
    QRectF  pixel_bounds;

    int moving_control_point_idx;
    QPointF c;

    QColor spline_color;

    QPixmap _background;
    QPixmap _function;
    int snap_x, snap_y;
    bool _draw_function, _preview_only;

    static constexpr int line_length_pixels = 3;
    static constexpr int point_size = 4;
    static constexpr int point_closeness_limit = 5;
};