satImageManager/controllerimagefusion.h at master · DmitryMasley/satImageManager · 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
#ifndef IMAGEFUSIONCONTROLLER_H
#define IMAGEFUSIONCONTROLLER_H
#include <QObject>
#include "stdafx.h"
#include <opencv2/opencv.hpp>
#include <opencv2/core/cuda.hpp>
#include "imagetreeview.h"
#include "ui_mainwindow.h"
#include "imagemodel.h"
#include "imageFusion/imagefusionprocessor.h"
#include "lib/image/standardimagemodel.h"

class ControllerImageFusion : public QObject
{
    Q_OBJECT
public:
    explicit ControllerImageFusion(Ui::MainWindow* ui,
                                   QMainWindow *MainWindow);
    ~ControllerImageFusion();
signals:
    void FusionStarted();
    void FusionEnd();
    void FusionProgress(int progress);
    void imageSaved(QString fileName);
public slots:
    void AddSelectedToSoursesList();
    void RemoveFromSoursesList();
    void StartFuision();
    void SetPanImage();
    void ResultPreview();
    void AddColoredImageToResults();
    void SaveImage();
private:
    ImageTreeView* _MainView;
    BasicListView* _TargetSoursesList;
    BasicTreeView* _TargetResultsList;
    QListView* _PanList;
    QPushButton* _AddToSourcesList;
    QPushButton* _RemoveFromSourcesList;
    QPushButton* _SetPan;
    QPushButton* _StartButton;
    QPushButton* _Preview;
    QPushButton* _Save;
    QPushButton* _AddColoredImage;
    StandardImageModel* _TargetSoursesModel;
    StandardImageModel* _TargetResultsModel;
    StandardImageModel* _PanModel;
    QMainWindow* _MainWindow;
    QCheckBox* _UseNormalization;
    QSpinBox* _WindowSize;
    QObject* _parent;
    QComboBox* _OrthogonalizationType;
    void fusion();
    void correctBritness(cv::Mat &image, cv::Mat source);
    void scaledNormalization(cv::Mat &image, cv::Mat source, int windowSize = 2);
    void windowNormalization(cv::Mat &image, cv::Mat source);
    cv::Mat getPanImage();
    QString getName(StandardImageItem* item);
    ImageItem* getPanItem();
    cv::Mat getInputMatrix(int cols, int rows, int type, QList<AbstractItem *> sources);
    cv::Mat getImage(StandardImageItem* item);
};

#endif // IMAGEFUSIONCONTROLLER_H