FlyDetector/mainwindow.h at master · cbodenst/FlyDetector · GitHub

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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#ifndef MAINWINDOW_H
#define MAINWINDOW_H

#include <QGraphicsScene>
#include <QMainWindow>
#include <QString>

#include <opencv2/opencv.hpp>

#include "flycountercontroller.h"

/* view type set */
enum ViewMode
{
    CAMERA    = 0,
    THRESHOLD = 1,
    CLUSTERS  = 2
};

namespace Ui {
    class MainWindow;
}

/*Main GUI class*/
class MainWindow : public QMainWindow
{
    Q_OBJECT

private:
    Ui::MainWindow* ui;

    QPixmap         image;
    QGraphicsScene* scene;
    FlyCounterController      flyCounter;

    /* initialization */
    void setupUI();
    void setupSignals();
    void setupSettings();

    /* image setters */
    void setImage(const cv::Mat& image);
    void setImage(const QPixmap& image);
    void setPixmap();
    void showCameraImage();
    void showClusterImage();
    void showThresholdImage();

    /* interface updates */
    void updateTimeSpinners();
    void userInterfaceEnabled(bool enabled);

    /* settings loading/saving */
    void loadSettings(const QString& path);
    void saveSettings(const QString& path);

private slots:
    void onLoadResize();
    void resizeEvent(QResizeEvent* event);
    void updateImage();

    void on_actionShow_Log_toggled(bool arg1);

public:
    static const QString DEFAULT_PATH;

    static const QString MODE;
    static const QString DISPLAY_VIALS;
    static const QString LEAD_TIME;
    static const QString ROUND_TIME;
    static const QString SHAKE_TIME;
    static const QString EPSILON;
    static const QString MIN_POINTS;
    static const QString PIXELS_PER_FLY;
    static const QString THRESHOLD;
    static const QString VIAL_SIZE;
    static const QString OUTPUT_PATH;
    static const QString SAVE_IMAGES;

    explicit MainWindow(QWidget* parent=nullptr);

    ViewMode    getViewMode();
    QPixmap toPixmap(const cv::Mat& image);

    ~MainWindow();

public slots:
    /* image settings */
    void on_detectDevices_clicked();
    void on_displayVials_stateChanged();
    void on_mode_currentIndexChanged(int index);
    void on_refresh_clicked();

    /* experiment parameter settings */
    void on_leadTime_valueChanged(int time);
    void on_roundTime_valueChanged(int time);
    void on_shakeTime_valueChanged(int time);

    /* analysis parameter setters */
    void on_epsilon_valueChanged(int epsilon);
    void on_minPoints_valueChanged(int minPoints);
    void on_pixelsPerFly_valueChanged(int pixelsPerFly);
    void on_threshold_valueChanged(int threshold);
    void on_vialSize_valueChanged(int arg1);

    /* results */
    void on_outputPath_textChanged(const QString& path);
    void on_outputPathBrowser_clicked();
    void on_saveImages_toggled(bool checked);

    /* experiment execution */
    void on_start_clicked();
    void on_stop_clicked();

    /* interface settings */
    void on_actionLoad_triggered();
    void on_actionSave_triggered();
};

#endif // MAINWINDOW_H