Класс для создания начального меню выбора режима работы программы Подробнее...

#include <main_menu.h>

Открытые члены
	MainMenu (std::string txt_file_name)
	Конструктор этого класса Подробнее...

Закрытые типы
enum	MODE { CALIBRATION , UNDISTORTION , VIBRATION_MONITORING , MOVEMENT_ENLARGEMENT }
	Перечисление со списком основных режимов работы Подробнее...

Закрытые члены
void	LoadPaths ()
	Открывает нужные пути Подробнее...

void	ExecuteCameraCalibration (std::string chessboards_path)
	Запускает калибровку камеры по картинкам шахматных досок Подробнее...

void	ExecuteVideoUndistortion (std::string input_file_name, std::string output_file_name)
	Запускает удаление дисторсии Подробнее...

void	ExecuteVibrationMonitoring (std::string input_file_name, std::string output_file_name)
	Запускает детекцию вибрации (основной режим) Подробнее...

void	ExecuteMotionMagnification (std::string input_file_name, std::string output_file_name)
	Запускает увеличение движения Подробнее...

void	CreateMenu ()
	Строит сетку меню с кнопками и надписями на Mat-объекте menu_. Подробнее...

void	UpdateBtnColorAndText ()
	Раскрашивает кнопки и наносит на них текст Подробнее...

void	DetectEvent (int event, int x, int y, int flags)
	callback функция для определения события (используется для хэндлинга клика мышью) Подробнее...

void	Idle ()

Закрытые статические члены
static void	OnMouse (int event, int x, int y, int flags, void *userdata)
	callback функция для определения события (используется для хэндлинга клика мышью) Подробнее...

Закрытые данные
std::string	params_file_name_
	Путь файла с параметрами Подробнее...

std::string	winname_
	Название окна Подробнее...

Mat	menu_
	Изображение меню, на котором будет происходить отрисовка Подробнее...

std::vector< Rect >	buttons_
	Контейнер, хранящий в себе прямоугольники-кнопки (фактически тут хранятся координаты кнопок) Подробнее...

std::string	button_text_ [4]
	Тексты кнопок Подробнее...

int	menu_width_
	Ширина меню Подробнее...

int	menu_height_
	Высота меню Подробнее...

int	button_amount_
	Количество кнопок Подробнее...

int	button_width_
	Ширина кнопки Подробнее...

int	button_height_
	Высота кнопки Подробнее...

int	button_spacing_
	Отступ между кнопками Подробнее...

int	top_offset_
	Отступ от застаканных кнопок сверху Подробнее...

int	bottom_offset_
	Отступ от застаканных кнопок снизу Подробнее...

int	width_offset_
	Отступ с двух сторон от кнопки по ширине Подробнее...

int	font
	Шрифт Подробнее...

double	font_scale
	Величина шрифта Подробнее...

int	thickness
	Толщина шрифта Подробнее...

int	baseline
	Параметр для cv::getText() Подробнее...

Point2i	last_mouse_position_
	Последние найденные в callback координаты курсора Подробнее...

std::vector< Scalar >	buttons_color_
	Контейнер для хранения цвета каждой кнопки Подробнее...

bool	mode_chosen_
	Флаг выбранного режима Подробнее...

bool	wanna_quit_
	Флаг выхода из программы Подробнее...

std::string	input_calibration_video_
	Имя (путь) входного файла (видео) для калибровки Подробнее...

std::string	input_distorted_video_
	Имя (путь) входного файла (видео) для удаления дисторсии Подробнее...

std::string	output_undistorted_video_
	Имя (путь) выходного файла (видео) после удаления дисторсии Подробнее...

std::string	input_vibration_detection_
	Имя (путь) входного файла (видео) для детекции вибрации Подробнее...

std::string	output_vibration_detection_
	Имя (путь) выходного файла (видео) после детекции вибрации Подробнее...

std::string	chessboards_path_
	Путь к папке с фотографиями шахматной доски Подробнее...

std::string	input_movement_enlargment_
	Имя (путь) входного файла (видео) для усиления движения Подробнее...

std::string	output_movement_enlargement_
	Имя (путь) выходного файла (видео) для усиления движения Подробнее...

Подробное описание

Класс для создания начального меню выбора режима работы программы

См. определение в файле main_menu.h строка 25

Перечисления

◆ MODE

enum MainMenu::MODE

private

Перечисление со списком основных режимов работы

См. определение в файле main_menu.h строка 30

    {
        CALIBRATION,
        UNDISTORTION,
        VIBRATION_MONITORING,
        MOVEMENT_ENLARGEMENT
    };

Конструктор(ы)

◆ MainMenu()

MainMenu::MainMenu ( std::string txt_file_name )

Конструктор этого класса

Аргументы

txt_file_name имя (путь) текстового документа, содержащий нужные для работы пути к входным и выходным расположениям

См. определение в файле main_menu.cpp строка 3

                                             :
    params_file_name_{ params_file_name },
    winname_{ MAIN_MENU_WINDOW_NAME },
    menu_width_{ 500 },
    menu_height_{ 500 },
    button_amount_{ 4 },
    mode_chosen_{ false }
{
    namedWindow(winname_, WINDOW_AUTOSIZE);
    // Инициализируем отступы
    int x = 20;
    button_spacing_ = x;
    top_offset_ = 3 * x;
    bottom_offset_ = 5 * x;
    width_offset_ = 3 * x;
    button_width_ = menu_width_ - 2 * width_offset_;
    button_height_ = (menu_height_ - (button_amount_ - 1) * button_spacing_ - top_offset_ - bottom_offset_) / button_amount_;
    // Надписи на кнопках
    button_text_[0] = "CALIBRATION";
    button_text_[1] = "UNDISTORTION";
    button_text_[2] = "VIBRATION MONITORING";
    button_text_[3] = "MOVEMENT ENLARGEMENT";
    // Инициализируем контейнер цветов кнопок
    for (int i = 0; i < button_amount_; i++)
    {
        buttons_color_.push_back(Scalar(255, 255, 255));
    }
    // Инициализируем атрибуты текста
    font = FONT_HERSHEY_PLAIN;
    font_scale = 1;
    thickness = 2;
    baseline = 0;
    // Создание меню на Mat-объекте
    CreateMenu();
}

Методы

◆ CreateMenu()

void MainMenu::CreateMenu ( )

private

Строит сетку меню с кнопками и надписями на Mat-объекте menu_.

См. определение в файле main_menu.cpp строка 89

{
    // Создадим кнопки в соответствии с их позициями
    for (int i = 0; i < button_amount_; i++)
    {
        buttons_.push_back
        (
            Rect(
                Point2i(
                    width_offset_,
                    top_offset_ + i * (button_height_ + button_spacing_)
                ),
                Point2i(
                    width_offset_ + button_width_,
                    top_offset_ + button_height_ + i * (button_height_ + button_spacing_)
                )
            )
        );
    }
    
    menu_ = Mat(Size(menu_width_, menu_height_), CV_8UC3, Scalar(0, 0, 0)); 
 
    // Подсказка для пользователя
    std::string tip = "Press 'Q' to quit";
    putText(
        menu_,
        tip,
        Point(
            (menu_width_ - getTextSize(tip, font, font_scale, thickness, &baseline).width) / 2,
            menu_height_ - (bottom_offset_ - getTextSize(tip, font, font_scale, thickness, &baseline).height) / 2),
        font,
        font_scale,
        Scalar(75, 75, 75),
        thickness / 2,
        4
    );
 
    Idle();
}

◆ DetectEvent()

void MainMenu::DetectEvent	(	int	event,
		int	x,
		int	y,
		int	flags
	)

private

callback функция для определения события (используется для хэндлинга клика мышью)

См. определение в файле main_menu.cpp строка 162

{
    // Загружаем названия директорий
    LoadPaths();
 
    switch (event)
    {
    case EVENT_LBUTTONDOWN:
    {
        for (int i = 0; i < buttons_.size(); i++)
        {
            // Если клик пользователя был по кнопке
            if (buttons_[i].contains(Point(x, y)))
            {
                switch (i)
                {
                case CALIBRATION:
                {
                    std::cout << "u've chosen camera calibration" << std::endl;
                    ExecuteCameraCalibration(chessboards_path_);
                    break;
                }
                case UNDISTORTION:
                {
                    std::cout << "u've chosen undistortion" << std::endl;
                    ExecuteVideoUndistortion(input_distorted_video_, output_undistorted_video_);
                    break;
                }
                case VIBRATION_MONITORING:
                {
                    std::cout << "u've chosen vibration monitoring" << std::endl;
                    ExecuteVibrationMonitoring(input_vibration_detection_, output_vibration_detection_);
                    break;
                }
                case MOVEMENT_ENLARGEMENT:
                {
                    std::cout << "u've chosen movement enlarging" << std::endl;
                    ExecuteMotionMagnification(input_movement_enlargment_, output_movement_enlargement_);
                    break;
                }
                }
            }
        }
        break;
    }
    case EVENT_MOUSEMOVE:
    {
        last_mouse_position_.x = x;
        last_mouse_position_.y = y;
        break;
    }
    }
}

◆ ExecuteCameraCalibration()

void MainMenu::ExecuteCameraCalibration ( std::string chessboards_path )

private

Запускает калибровку камеры по картинкам шахматных досок

Аргументы

chessboards_path имя (путь) к папке с изображениями шахматных досок

См. определение в файле main_menu.cpp строка 59

{
    CameraCalibrator camera_calibrator(chessboards_path);
    camera_calibrator.ExecuteCameraCalibration();
}

◆ ExecuteMotionMagnification()

void MainMenu::ExecuteMotionMagnification	(	std::string	input_file_name,
		std::string	output_file_name
	)

private

Запускает увеличение движения

Аргументы

input_file_name	имя (путь) входного видео
output_file_name	имя (путь) выходного файла

См. определение в файле main_menu.cpp строка 83

{
    MovementEnlarger movement_enlarger(input_file_name, output_file_name);
    movement_enlarger.ExecuteLaplaceEnlargement();
}

◆ ExecuteVibrationMonitoring()

void MainMenu::ExecuteVibrationMonitoring	(	std::string	input_file_name,
		std::string	output_file_name
	)

private

Запускает детекцию вибрации (основной режим)

Аргументы

input_file_name	имя (путь) входного видео
output_file_name	имя (путь) выходного файла

См. определение в файле main_menu.cpp строка 77

{
    VibrationDetector vibration_detector(input_file_name, output_file_name, MAIN_WINDOW_NAME);
    vibration_detector.ExecuteVibrationDetection();
}

◆ ExecuteVideoUndistortion()

void MainMenu::ExecuteVideoUndistortion	(	std::string	input_file_name,
		std::string	output_file_name
	)

private

Запускает удаление дисторсии

Аргументы

input_file_name	имя (путь) входного видео
output_file_name	имя (путь) выходного файла

См. определение в файле main_menu.cpp строка 65

{
    std::string params;
 
    std::cout << "Please attach the parameters.txt file: ";
    std::cin >> params;
    std::cout << std::endl;
 
    VideoUndistorter video_undistorter(input_file_name, output_file_name, params);
    video_undistorter.ExecuteVideoUndistortion();
}

◆ Idle()

void MainMenu::Idle ( )

private

Запускает меню

См. определение в файле main_menu.cpp строка 216

{
    // Устанавливаем хэндлер окна
    setMouseCallback(winname_, OnMouse, (void*)this);
 
    while (!mode_chosen_)
    {
        for (int i = 0; i < buttons_.size(); i++)
        {
            if (buttons_[i].contains(last_mouse_position_))
            {
                buttons_color_[i] = Scalar(0, 0, 255);
            }
            else buttons_color_[i] = Scalar(25, 25, 25);
        }
        // Обновляем текущие цвета кнопок
        UpdateBtnColorAndText();
 
        imshow(winname_, menu_);
        switch (waitKey(10))
        {
        case 'q':
        {
            mode_chosen_ = true;
            break;
        }
        }
    }
}

◆ LoadPaths()

void MainMenu::LoadPaths ( )

private

Открывает нужные пути

См. определение в файле main_menu.cpp строка 39

{
    std::ifstream params_file;
    params_file.open(params_file_name_);
 
    if (!params_file.is_open())
    {
        std::cout << "Wrong filename" << std::endl;
    }
 
    std::getline(params_file, input_calibration_video_);
    std::getline(params_file, input_distorted_video_);
    std::getline(params_file, output_undistorted_video_);
    std::getline(params_file, input_vibration_detection_);
    std::getline(params_file, output_vibration_detection_);
    std::getline(params_file, chessboards_path_);
    std::getline(params_file, input_movement_enlargment_);
    std::getline(params_file, output_movement_enlargement_);
}

◆ OnMouse()

void MainMenu::OnMouse	(	int	event,
		int	x,
		int	y,
		int	flags,
		void *	userdata
	)

staticprivate

callback функция для определения события (используется для хэндлинга клика мышью)

См. определение в файле main_menu.cpp строка 156

{
    MainMenu* d = static_cast<MainMenu*>(userdata);
    d->DetectEvent(event, x, y, flags);
}

◆ UpdateBtnColorAndText()

void MainMenu::UpdateBtnColorAndText ( )

private

Раскрашивает кнопки и наносит на них текст

См. определение в файле main_menu.cpp строка 129

{
    for (int i = 0; i < button_amount_; i++)
    {
        // Рисуем очертания кнопки
        rectangle(menu_, buttons_[i], buttons_color_[i], FILLED);
        // Размеры текста в пикселях
        Size current_text_size = getTextSize(button_text_[i], font, font_scale, thickness, &baseline);
        // Координаты текущего текста
        Point text_coordinates = Point(
            buttons_[i].tl().x + button_width_ / 2 - current_text_size.width / 2,
            buttons_[i].tl().y + button_height_ / 2 + (current_text_size.height) / 2);
        // Рисуем текст кнопки
        putText(
            menu_,
            button_text_[i],
            text_coordinates,
            font,
            font_scale,
            Scalar(255, 255, 255),
            thickness,
            4 // тип линии. вроде отличаются сглаживанием. выбрал 4-й просто так :)
        );
 
    }
}

Данные класса

◆ baseline

int MainMenu::baseline

private

Параметр для cv::getText()

См. определение в файле main_menu.h строка 166

◆ bottom_offset_

int MainMenu::bottom_offset_

private

Отступ от застаканных кнопок снизу

См. определение в файле main_menu.h строка 146

◆ button_amount_

int MainMenu::button_amount_

private

Количество кнопок

См. определение в файле main_menu.h строка 126

◆ button_height_

int MainMenu::button_height_

private

Высота кнопки

См. определение в файле main_menu.h строка 134

◆ button_spacing_

int MainMenu::button_spacing_

private

Отступ между кнопками

См. определение в файле main_menu.h строка 138

◆ button_text_

std::string MainMenu::button_text_[4]

private

Тексты кнопок

См. определение в файле main_menu.h строка 114

◆ button_width_

int MainMenu::button_width_

private

Ширина кнопки

См. определение в файле main_menu.h строка 130

◆ buttons_

std::vector<Rect> MainMenu::buttons_

private

Контейнер, хранящий в себе прямоугольники-кнопки (фактически тут хранятся координаты кнопок)

См. определение в файле main_menu.h строка 110

◆ buttons_color_

std::vector<Scalar> MainMenu::buttons_color_

private

Контейнер для хранения цвета каждой кнопки

См. определение в файле main_menu.h строка 175

◆ chessboards_path_

std::string MainMenu::chessboards_path_

private

Путь к папке с фотографиями шахматной доски

См. определение в файле main_menu.h строка 209

◆ font

int MainMenu::font

private

Шрифт

См. определение в файле main_menu.h строка 154

◆ font_scale

double MainMenu::font_scale

private

Величина шрифта

См. определение в файле main_menu.h строка 158

◆ input_calibration_video_

std::string MainMenu::input_calibration_video_

private

Имя (путь) входного файла (видео) для калибровки

См. определение в файле main_menu.h строка 189

◆ input_distorted_video_

std::string MainMenu::input_distorted_video_

private

Имя (путь) входного файла (видео) для удаления дисторсии

См. определение в файле main_menu.h строка 193

◆ input_movement_enlargment_

std::string MainMenu::input_movement_enlargment_

private

Имя (путь) входного файла (видео) для усиления движения

См. определение в файле main_menu.h строка 213

◆ input_vibration_detection_

std::string MainMenu::input_vibration_detection_

private

Имя (путь) входного файла (видео) для детекции вибрации

См. определение в файле main_menu.h строка 201

◆ last_mouse_position_

Point2i MainMenu::last_mouse_position_

private

Последние найденные в callback координаты курсора

См. определение в файле main_menu.h строка 171

◆ menu_

Mat MainMenu::menu_

private

Изображение меню, на котором будет происходить отрисовка

См. определение в файле main_menu.h строка 106

◆ menu_height_

int MainMenu::menu_height_

private

Высота меню

См. определение в файле main_menu.h строка 122

◆ menu_width_

int MainMenu::menu_width_

private

Ширина меню

См. определение в файле main_menu.h строка 118

◆ mode_chosen_

bool MainMenu::mode_chosen_

private

Флаг выбранного режима

См. определение в файле main_menu.h строка 179

◆ output_movement_enlargement_

std::string MainMenu::output_movement_enlargement_

private

Имя (путь) выходного файла (видео) для усиления движения

См. определение в файле main_menu.h строка 217

◆ output_undistorted_video_

std::string MainMenu::output_undistorted_video_

private

Имя (путь) выходного файла (видео) после удаления дисторсии

См. определение в файле main_menu.h строка 197

◆ output_vibration_detection_

std::string MainMenu::output_vibration_detection_

private

Имя (путь) выходного файла (видео) после детекции вибрации

См. определение в файле main_menu.h строка 205

◆ params_file_name_

std::string MainMenu::params_file_name_

private

Путь файла с параметрами

См. определение в файле main_menu.h строка 98

◆ thickness

int MainMenu::thickness

private

Толщина шрифта

См. определение в файле main_menu.h строка 162

◆ top_offset_

int MainMenu::top_offset_

private

Отступ от застаканных кнопок сверху

См. определение в файле main_menu.h строка 142

◆ wanna_quit_

bool MainMenu::wanna_quit_

private

Флаг выхода из программы

См. определение в файле main_menu.h строка 183

◆ width_offset_

int MainMenu::width_offset_

private

Отступ с двух сторон от кнопки по ширине

См. определение в файле main_menu.h строка 150

◆ winname_

std::string MainMenu::winname_

private

Название окна

См. определение в файле main_menu.h строка 102

Объявления и описания членов классов находятся в файлах:

C:/Users/seeyo/source/repos/Visual-Vibration-Tracking-V2/include/GUI/main_menu.h
C:/Users/seeyo/source/repos/Visual-Vibration-Tracking-V2/source/main_menu.cpp

Открытые члены

Закрытые типы

Закрытые члены

Закрытые статические члены

Закрытые данные

Подробное описание

Перечисления

◆ MODE

Конструктор(ы)

◆ MainMenu()

Методы

◆ CreateMenu()

◆ DetectEvent()

◆ ExecuteCameraCalibration()

◆ ExecuteMotionMagnification()

◆ ExecuteVibrationMonitoring()

◆ ExecuteVideoUndistortion()

◆ Idle()

◆ LoadPaths()

◆ OnMouse()

◆ UpdateBtnColorAndText()

Данные класса

◆ baseline

◆ bottom_offset_

◆ button_amount_

◆ button_height_

◆ button_spacing_

◆ button_text_

◆ button_width_

◆ buttons_

◆ buttons_color_

◆ chessboards_path_

◆ font

◆ font_scale

◆ input_calibration_video_

◆ input_distorted_video_

◆ input_movement_enlargment_

◆ input_vibration_detection_

◆ last_mouse_position_

◆ menu_

◆ menu_height_

◆ menu_width_

◆ mode_chosen_

◆ output_movement_enlargement_

◆ output_undistorted_video_

◆ output_vibration_detection_

◆ params_file_name_

◆ thickness

◆ top_offset_

◆ wanna_quit_

◆ width_offset_

◆ winname_