optflowcontroller.h

#ifndef CONTROLLER_H
#define CONTROLLER_H
 
#include <QObject>
#include <QWidget>
#include <QLabel>
#include <QSpacerItem>
#include <QLineEdit>
#include <QDoubleValidator>
#include <QCheckBox>
#include <QLayout>
#include <QLabel>
#include <QDoubleSpinBox>
#include <QPushButton>
#include <QComboBox>
#include <opencv2/core/mat.hpp>
#include <iostream>
#include <iomanip>
#include <QtConcurrent/QtConcurrentMap>
#include <QThreadPool>
#include <QTranslator>
#include <QCoreApplication>
#include <vector>
#include <algorithm>
#include <numeric>
#include <future>
 
#include "ialgorithm.h"
#include "reader.h"
#include "factory.h"
#include "imagegatherer.h"
#include "flowcalculator.h"
#include <opencv2/video.hpp>
 
#define PLUGIN_NAME QObject::tr("Stationary camera removal")
// widget
#define SELECTOR_LABEL_TEXT QObject::tr("Stationary threshold")
#define SELECTOR_DESCRIPTION QObject::tr("Removes all frames where the camera is stationary. A frame is declared stationary if its camera movement is lower than given percentage of the median of all camera movements.")
#define DOWNSAMPLE_LABEL_TEXT QObject::tr("Sampling resolution")
#define DOWNSAMPLE_CHECKBOX_TEXT QObject::tr("Activate down sampling")
#define DESCRIPTION_DOWNSAMPLE QObject::tr("If enabled a resolution of 720p will be used for the algorithm to speed up computation. This will hurt the accuracy of the result slightly. It however won't change the export resolution. This parameter will be disabled if the input resolution is lower or equal than 720p.")
#define DESCRIPTION_STYLE "color: rgb(58, 58, 58); border-left: 6px solid  rgb(58, 58, 58); border-top-right-radius: 5px; border-bottom-right-radius: 5px; background-color: lightblue;"
#define INFO_STYLE "color: rgb(58, 58, 58); border-left: 6px solid  rgb(58, 58, 58); border-top-right-radius: 5px; border-bottom-right-radius: 5px; background-color: lightGreen;"
// buffer
#define BUFFER_NAME "StationaryCameraMovementValues"
#define DELIMITER_COORDINATE "|"
#define DELIMITER_ENTITY ","
// settings
#define SETTINGS_SAMPLE_RESOLUTION "Sample resolution"
#define SETTINGS_SELECTOR_THRESHOLD "Selector threshold"
// log file
#define LF_OPT_FLOW_TOTAL "Flow calculation"
#define LF_SELECT_FRAMES "Selection of keyframes"
#define LF_CE_TYPE_ADDITIONAL_INFO "Additional Computation Information"
#define LF_CE_VALUE_USED_BUFFERED "Used buffered values"
#define LF_CE_TYPE_DEBUG "Debug Information"
#define LF_CE_NAME_FLOWVALUE "Flow value"
#define LF_CE_NAME_SAMPLERES "Sampling Resolution"
#define LF_TIMER_BUFFER "Update Buffer"
#define LF_TIMER_CORE "Core Computation"
#define LF_TIMER_SELECTION "Keyframe selection"
 
class SmoothController : public IAlgorithm
{
    Q_OBJECT
    Q_PLUGIN_METADATA(IID "iVS3D.IAlgorithm")   // implement interface as plugin, use the iid as identifier
    Q_INTERFACES(IAlgorithm)                        // declare this as implementation of IAlgorithm interface
 
public:
    SmoothController();
    ~SmoothController() {}
 
    QWidget *getSettingsWidget(QWidget *parent) override;
 
    std::vector<uint> sampleImages(const std::vector<uint> &imageList, Progressable *receiver, volatile bool *stopped, bool useCuda, LogFileParent *logFile) override;
 
    QString getName() const override;
 
    void initialize(Reader *reader, QMap<QString, QVariant> buffer, signalObject *sigObj) override;
 
    void setSettings(QMap<QString, QVariant> settings) override;
 
    QMap<QString, QVariant> generateSettings(Progressable *receiver, bool useCuda, volatile bool *stopped) override;
 
    QMap<QString, QVariant> getSettings() override;
 
private:
    // member variables
    double m_selectorThreshold = 0.3;
    Reader *m_reader = nullptr;
    QPoint m_inputResolution = QPoint(0, 0);
    cv::SparseMat m_bufferMat;
    signalObject *m_sigObj = nullptr;
    //      widget elements
    QWidget *m_settingsWidget = nullptr;
    QDoubleSpinBox *m_selectorThresholdSpinBox = nullptr;
 
    // timing variables
    long m_durationFarnebackMs = 0;
    long m_durationComputationFlowMs = 0;
 
    // functions
    void reportProgress(QString op, int progress, Progressable *receiver);
    void displayMessage(QString txt, Progressable *receiver);
    void createSettingsWidget(QWidget *parent);
    QMap<QString, QVariant> sendBuffer();
    void recreateBufferMatrix(QMap<QString, QVariant> buffer);
    void stringToBufferMat(QString string);
    QVariant bufferMatToVariant(cv::SparseMat bufferMat);
signals:
    void changeUIParameter(QVariant nValue, QString paramName, QString selectorName);
};
 
#endif // CONTROLLER_H