vibration_detector.h

#ifndef VIBRATION_DETECTOR_H
#define VIBRATION_DETECTOR_H
 
// VREMENNO
#define MAIN_WINDOW_NAME "Video"
#define V_MONITOR_WINDOW_NAME "Vibration Monitor"
 
// standart headers
#include <iostream>
#include <vector>
 
// OpenCV headers
#include <opencv2/video/tracking.hpp>
 
// my headers
#include "VVT-V2/lonely_point_handler.h"
#include "VVT-V2/colored_point_handler.h"
#include "VVT-V2/frame_handler.h"
#include "VVT-V2/helper.h"
#include "VVT-V2/contour_handler.h"
 
class VibrationDetector 
{
    enum main_mode
    {
        DEFAULT,
        SELECTINGROI,
        PAUSE
    };
    enum colored_point_mode
    {
        COLORING_DEFAULT,
        COLORING_BASED_ON_FREQUENCY,
        COLORING_BASED_ON_AMPLITUDE
    };
 
public:
    VibrationDetector(std::string input_file_name, std::string output_file_name, std::string window_name);
    ~VibrationDetector();
    void ExecuteVibrationDetection();
 
private:
    void ServeTheQueues();
    void CreateNewPoint(Point2f mouse_coordinates);
    void LeftClickHandler(Point2f mouse_coordinates);
    void CreateNewColoredPoint(Point2f mouse_coordinates);
    void DeletePoints(Point2i mouse_coordinates);
    void DeleteColoredPoints();
    void FindAndDeleteUncofidentPoints();
    static void OnMouse(int event, int x, int y, int flags, void* userdata);
    void DetectEvent(int event, int x, int y, int flags);
    void DefaultModeHandler(int event, int x, int y);
    void RoiSelectionModeHandler(int event, int x, int y);
    Mat MakeWarpedFrame(Mat frame, std::vector<Point2i> warping_figure);
    void TrackAndCalc();
    std::vector<Point2f> FindGoodFeatures(Mat frame, Rect roi);
    void DrawDebugLkWinRectangle(Mat& frame);
    void DrawAndOutput(Mat& frame);
 
private:
    FrameHandler* frame_handler;
    bool running_;
    float res_mp_;
    int point_id_;
    Point last_mouse_position_;
    int current_mode_;
    bool fullscreen_;
 
    // NORMAL MODE
 
    std::vector<Point2i> l_click_queue_;
    std::vector<Point2f> c_point_queue_;
    std::vector<Point2i> delete_queue_;
    std::vector<Point2i> histograms_queue_;
    std::vector<LonelyPointHandler*> vec_lonely_point_handlers_;
    std::vector<ColoredPointHandler*> vec_colored_point_handlers_;
 
    // ADJUST SENSIVITY MODE
 
    int sensivity_in_percents_;
 
    Mat current_tracking_frame_;
    Mat frame_to_be_shown_;
    Mat grad_scale_;
    int colored_point_mode_;
 
    // ROI SELECTING MODE
    // 
    bool roi_selecting_;
    bool roi_selected_;
    Point2i tl_click_coords_;
    
    std::string input_file_name_;
    std::string output_file_name_;
    std::string output_csv_file_name_;
    std::string window_name_;
 
    bool warping_figure_selecting_;
    std::vector<Point> warping_figure_;
 
    Mat prev_img_gray_;
    Mat next_img_gray_;
    int lk_win_size_;
    int level_amount_;
 
 
    // Для быстрого преобразования Фурье
    int fps_;
    double frame_time_;
};
 
#endif