main.cpp

#include <opencv2/core/core.hpp>
#include <opencv2/videoio/videoio.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <iostream>
#include "rgb2name.h"

using namespace cv;
using namespace std;

int main(int argc, char **argv) {
    std::cerr << "start" << std::endl;
    int64 startTick = cv::getTickCount();

    cv::VideoCapture capture = cv::VideoCapture("Clip00001.avi");

    namedWindow("orjinal", WINDOW_AUTOSIZE);
    namedWindow("gray&gaussian", WINDOW_AUTOSIZE);
    namedWindow("canny", WINDOW_AUTOSIZE);
    namedWindow("found", WINDOW_AUTOSIZE);
    moveWindow("orjinal", 100, 300);
    moveWindow("gray&gaussian", 500, 300);
    moveWindow("canny", 900, 300);
    moveWindow("found", 1300, 300);

    cv::Mat currFrame;
    vector<vector<vector<double>>> totalFoundColors;
    int i = 0;
    for (;; i++) {
        capture >> currFrame;
        if (currFrame.empty()) {
            break;
        }
        //resize(currFrame, currFrame, Size(1280,720));
        currFrame = currFrame(CvRect(1340, 160, 350, 350));
        imshow("orjinal", currFrame);
        cv::Mat tempFrame;
        cvtColor(currFrame, tempFrame, cv::COLOR_BGR2GRAY);

        cv::GaussianBlur(tempFrame, tempFrame, cv::Size(5, 5), 0, 0);
        threshold(tempFrame, tempFrame, 150, 255, THRESH_BINARY_INV);
        imshow("gray&gaussian", tempFrame);

        //cv::Canny(tempFrame, tempFrame, 500, 500);
        imshow("canny", tempFrame);

        vector<vector<Point>> contours;
        vector<Vec4i> hierarchy;
        findContours(tempFrame, contours, hierarchy, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_TC89_L1, Point(0, 0));

        Mat foundFrame(350, 350, CV_8UC3, Scalar(0, 0, 0));
        int contourSize = contours.size();
        vector<vector<double>> foundColorsInFrame;
        for (int j = 0; j < contourSize; j++) {
            vector<Point2f> approx;
            if (contours[j].size() == 0) continue;
            approxPolyDP(contours[j], approx, arcLength(contours[j], true) * 0.02, true);
            if (approx.size() == 4) {

                Rect boundRect = boundingRect(approx);
                Mat roi = currFrame(boundRect);
                Mat1b mask(roi.rows, roi.cols);
                Scalar meanVals = mean(roi, mask);
                rectangle(foundFrame, boundRect, meanVals, -1);
                foundColorsInFrame.push_back({meanVals[2], meanVals[1], meanVals[0]});
                //cout << meanVals[0] << " " << meanVals[1] << " " << meanVals[2];
            }
        }
        if (foundColorsInFrame.size() == 16) {//TODO: sort foundColorsInFrame s before pushing
            totalFoundColors.push_back(foundColorsInFrame);
        }
        imshow("found", foundFrame);
        waitKey(1);
    }
    int64 endTick = cv::getTickCount();
    std::cout << "read " << i << " frames in " << ((endTick - startTick) / cv::getTickFrequency())
              << std::endl;
    string lastNameMatrix("");
    for (std::vector<int>::size_type a = totalFoundColors.size() - 1;
         a != (std::vector<int>::size_type) -1; a--) {
        string nameMatrix("");
        for (std::vector<int>::size_type b = totalFoundColors[a].size() - 1;
             b != (std::vector<int>::size_type) -1; b--) {
            string colorName = rgb2name(totalFoundColors[a][b]);
            auto format = "%6s\t";
            auto size = snprintf(nullptr, 0, format, colorName.c_str());
            std::string output((unsigned int) size, ' ');
            sprintf(&output[0], format, colorName.c_str());
            nameMatrix += output;
            if (b % 4 == 0) nameMatrix += "\n";
        }
        if (nameMatrix.compare(lastNameMatrix) != 0) {
            printf("%s\n", nameMatrix.c_str());
            lastNameMatrix = nameMatrix;
        }
    }
    std::cout << "wrote color names in " << ((getTickCount() - endTick) / cv::getTickFrequency())
              << std::endl;
    return 0;
}