#include "opencv2/opencv.hpp"
#include <iostream>

using namespace std;
using namespace cv;

Mat imgProc(Mat imgIn) {
    Mat imgOut;
    imgIn.copyTo(imgOut);
    int thre = 40, maxv = 255;

    cvtColor(imgOut, imgOut, COLOR_BGR2GRAY);//BGR

    //ֵ
    threshold(imgOut, imgOut, thre, maxv, THRESH_BINARY);//pixel = (pixel < thre) ? 0 : maxv;
    
    dilate(imgOut, imgOut, Mat());

    floodFill(imgOut, Point(5, 50), Scalar(255), 0, FLOODFILL_FIXED_RANGE);

    erode(imgOut, imgOut, Mat());
    erode(imgOut, imgOut, Mat());

    return imgOut;
}

void imageIO() {
    //input image path
    string imgPath;
    cout << "Select a file to Open: \n";
    cin >> imgPath;

    //Ctrl + D
    if(imgPath.empty()) imgPath = "../TestFiles/Energy02.jpg";

    //read image
    Mat img = imread(imgPath);

    //show image
    imshow("Display Image", imgProc(img));

    //quit
    while(waitKey(0) != 'q') cout << "Press Q to Quit\n";//ASCII
}

void videoIO() {
    Mat frame;
    VideoCapture video;
    video.open("../TestFiles/Energy.mp4");

    do {
        video >> frame;//cin >> a;

        if(frame.empty()) break;

        imshow("Video", frame);

        imshow("Proc", imgProc(frame));
    }
    while(waitKey(0) != 'q');//100ms
}

int main() {
    imageIO();
    return 0;
}