Feature/chanedregion

CMakeLists.txt

@@ -19,7 +19,7 @@ set(LIBRARY_DEPS ${OpenCV_LIBS})
 # BUILD
 add_subdirectory(sample_template)
 # Add you directory here
-# add_subdirectory(sample_YOUR_NAME)
+ add_subdirectory(sample_maslova)
 
 # REPORT
 message( STATUS "")

sample_maslova/CMakeLists.txt

@@ -0,0 +1,7 @@
+set(target "sample_maslova")
+
+file(GLOB hdrs "*.hpp")
+file(GLOB srcs "*.cpp")
+
+add_executable(${target} ${srcs} ${hdrs})
+target_link_libraries(${target} ${LIBRARY_DEPS})

sample_maslova/application.cpp

@@ -0,0 +1,197 @@
+#include "application.hpp"
+#include "processing.hpp"
+#include <time.h>
+#include <opencv2/highgui/highgui.hpp>
+
+using namespace cv;
+
+int Application::parseArguments(int argc, const char **argv, 
+        Application::Parameters &params)
+{
+    if (argc < 2)
+    {
+        return 1;
+    }
+    params.imgFileName = std::string(argv[1]);
+    return 0;
+}
+
+int Application::getFrame(const std::string &fileName, Mat& src)
+{
+    src = imread(fileName);
+    if (src.empty())
+    {
+        return 1;
+    }
+    return 0;
+}
+
+int Application::processFrame(const Mat& src, Mat& dst)
+{
+	processor.processFrame(src, dst, guiState.filter);
+
+    if (dst.empty())
+    {
+        return 1;
+    }
+
+    return 0;
+}
+
+int Application::drawButtons(Mat &display)
+{
+    guiState.onButtonPlace = Rect(20, display.rows - 60, 120, 40);
+    guiState.offButtonPlace = Rect(160, display.rows - 60, 120, 40);
+	guiState.saveButtonPlace = Rect(300, display.rows - 60, 120, 40);
+	guiState.medianButtonPlace = Rect(440, display.rows - 60, 120, 40);
+	guiState.grayButtonPlace = Rect(580, display.rows - 60, 120, 40);
+	guiState.pixelButtonPlace = Rect(720, display.rows - 60, 140, 40);
+	guiState.cannyButtonPlace = Rect(880, display.rows - 60, 120, 40);
+    rectangle(display, guiState.onButtonPlace, 
+              Scalar(128, 128, 128), CV_FILLED);
+    rectangle(display, guiState.offButtonPlace, 
+              Scalar(128, 128, 128), CV_FILLED);
+	rectangle(display, guiState.saveButtonPlace, 
+              Scalar(128, 128, 128), CV_FILLED);
+	rectangle(display, guiState.medianButtonPlace, 
+              Scalar(128, 128, 128), CV_FILLED);
+	rectangle(display, guiState.grayButtonPlace, 
+              Scalar(128, 128, 128), CV_FILLED);
+	rectangle(display, guiState.pixelButtonPlace, 
+              Scalar(128, 128, 128), CV_FILLED);
+	rectangle(display, guiState.cannyButtonPlace, 
+              Scalar(128, 128, 128), CV_FILLED);
+
+    putText(display, "on", 
+        Point(guiState.onButtonPlace.x + guiState.onButtonPlace.width / 2 - 15,
+              guiState.onButtonPlace.y + guiState.onButtonPlace.height / 2 + 10),
+        FONT_HERSHEY_SIMPLEX, 1.0, Scalar(0, 0, 0), 2);
+    putText(display, "off", 
+        Point(guiState.offButtonPlace.x + guiState.offButtonPlace.width / 2 - 20,
+              guiState.offButtonPlace.y + guiState.offButtonPlace.height / 2 + 10),
+        FONT_HERSHEY_SIMPLEX, 1.0, Scalar(0, 0, 0), 2);
+	putText(display, "save", 
+        Point(guiState.saveButtonPlace.x + guiState.saveButtonPlace.width / 2 - 20,
+              guiState.saveButtonPlace.y + guiState.saveButtonPlace.height / 2 + 10),
+        FONT_HERSHEY_SIMPLEX, 1.0, Scalar(0, 0, 0), 2);
+	putText(display, "median", 
+		Point(guiState.medianButtonPlace.x + guiState.medianButtonPlace.width / 2 - 60,
+		guiState.medianButtonPlace.y + guiState.medianButtonPlace.height / 2 + 10),
+        FONT_HERSHEY_SIMPLEX, 1.0, Scalar(0, 0, 0), 2);
+	putText(display, "gray", 
+		Point(guiState.grayButtonPlace.x + guiState.grayButtonPlace.width / 2 - 40,
+		guiState.grayButtonPlace.y + guiState.grayButtonPlace.height / 2 + 10),
+        FONT_HERSHEY_SIMPLEX, 1.0, Scalar(0, 0, 0), 2);
+	putText(display, "pixelized", 
+		Point(guiState.pixelButtonPlace.x + guiState.pixelButtonPlace.width / 2 - 60,
+		guiState.pixelButtonPlace.y + guiState.pixelButtonPlace.height / 2 + 10),
+        FONT_HERSHEY_SIMPLEX, 1.0, Scalar(0, 0, 0), 2);
+	putText(display, "canny", 
+		Point(guiState.cannyButtonPlace.x + guiState.cannyButtonPlace.width / 2 - 40,
+		guiState.cannyButtonPlace.y + guiState.cannyButtonPlace.height / 2 + 10),
+        FONT_HERSHEY_SIMPLEX, 1.0, Scalar(0, 0, 0), 2);
+
+    return 0;
+}
+
+int Application::showFrame(const std::string &caption, 
+        const Mat& src, Mat& dst)
+{
+    if (guiState.state == OffFilter)
+    {
+        src.copyTo(dst);
+    }
+    else if (guiState.state == OnFilter)
+    {
+        processFrame(src, dst);
+    }
+    else
+    {
+        return 1;
+    }
+    Mat display(src.rows, src.cols + dst.cols, src.type());
+    Mat srcRoi = display(Rect(0, 0, src.cols, src.rows));
+    src.copyTo(srcRoi);
+    Mat dstRoi = display(Rect(src.cols, 0, dst.cols, dst.rows));
+    dst.copyTo(dstRoi);   
+
+	if(guiState.saveState)
+	{
+		int hours, minutes, seconds;
+		hours = (time(0)/3600)%24;
+		minutes = (time(0)/60)%60;
+		seconds = time(0)%60;
+		std::string s = "";
+		std::stringstream ss;
+		ss<<hours+3<<"-"<<minutes<<"-"<<seconds<<".png";
+		ss>>s;
+		ss.clear();
+		imwrite(s, display);
+		guiState.saveState = false;
+	}
+
+    drawButtons(display);
+
+    namedWindow(caption);  
+    imshow(caption, display);
+    setMouseCallback(caption, onButtonsOnOffClick, &guiState);
+    char key = waitKey(1);
+
+    return key;
+}
+
+void onButtonsOnOffClick(int eventId, int x, int y, int flags, void *userData)
+{
+    if (eventId != EVENT_LBUTTONDOWN)
+    {
+        return;
+    }
+    Application::GUIElementsState *elems = 
+        (Application::GUIElementsState *)userData;
+    if (onButtonClicked(elems->onButtonPlace, x, y))
+    {
+        elems->state = Application::OnFilter;
+        return;
+    }
+    if (onButtonClicked(elems->offButtonPlace, x, y))
+    {
+        elems->state = Application::OffFilter;
+        return;
+    }
+	if (onButtonClicked(elems->saveButtonPlace, x, y))
+	{
+		elems->saveState = true;
+		return;
+	}
+	if (onButtonClicked(elems->medianButtonPlace, x, y))
+	{
+		elems->filter = Processing::MEDIAN;
+		return;
+	}
+	if (onButtonClicked(elems->grayButtonPlace, x, y))
+	{
+		elems->filter = Processing::CVT_CONVERT_GRAY;
+		return;
+	}
+	if (onButtonClicked(elems->pixelButtonPlace, x, y))
+	{
+		elems->filter = Processing::PIXELIZED;
+		return;
+	}
+	if (onButtonClicked(elems->cannyButtonPlace, x, y))
+	{
+		elems->filter = Processing::CANNY;
+		return;
+	}
+}
+
+bool onButtonClicked(cv::Rect buttonPlace, int x, int y)
+{
+    if (x < buttonPlace.x || x > buttonPlace.x + buttonPlace.width ||
+        y < buttonPlace.y || y > buttonPlace.y + buttonPlace.height)
+    {
+        return false;
+    }
+    return true;
+}
+

sample_maslova/application.hpp

@@ -0,0 +1,58 @@
+#pragma once
+
+#include <string>
+#include <iostream>
+#include <opencv2/core/core.hpp>
+
+#include "processing.hpp"
+
+bool onButtonClicked(cv::Rect buttonPlace, int x, int y);
+void onButtonsOnOffClick(int eventId, int x, int y, int flags, void *userData);
+
+class Application
+{
+ public:
+    enum WindowState
+    {
+        OnFilter,
+        OffFilter
+    };
+    struct Parameters
+    {
+        std::string imgFileName;
+    };
+    struct GUIElementsState
+    {
+        WindowState state;
+        cv::Rect onButtonPlace;
+        cv::Rect offButtonPlace;
+		cv::Rect saveButtonPlace;
+		cv::Rect medianButtonPlace;
+		cv::Rect grayButtonPlace;
+		cv::Rect pixelButtonPlace;
+		cv::Rect cannyButtonPlace;
+		bool saveState;
+		Processing::FilterType filter;
+    };
+    int parseArguments(int argc, const char **argv, Parameters &params);
+    int getFrame(const std::string &fileName, cv::Mat& src);
+    int processFrame(const cv::Mat& src, cv::Mat& dst);
+    int showFrame(const std::string &caption, 
+                  const cv::Mat& src, cv::Mat& dst);
+    friend void onButtonsOnOffClick(int eventId, int x, int y, 
+                                    int flags, void *userData);
+    Application() 
+    { 
+        guiState.state = OnFilter;
+		guiState.saveState = false;
+		guiState.filter = Processing::MEDIAN;
+    };
+
+ private:
+    Processing processor;
+    GUIElementsState guiState;
+
+    int drawButtons(cv::Mat &display);
+
+    friend bool onButtonClicked(cv::Rect buttonPlace, int x, int y);
+};

sample_maslova/main.cpp

@@ -0,0 +1,44 @@
+#include <opencv2/core/core.hpp>
+#include <iostream>
+
+#include "application.hpp"
+
+using namespace std;
+using namespace cv;
+
+enum ErrorCode {
+    OK,
+    WRONG_ARGUMENTS,
+    WRONG_INPUT,
+    CANT_PROCESS
+};
+
+int main(int argc, const char **argv)
+{
+    Application app;
+    Application::Parameters params;
+
+    if (app.parseArguments(argc, argv, params) != 0)
+    {
+        cout << "sample_template <image_name>" << endl;
+        cout << "<image_name> - image name for filtering" << endl;
+        return WRONG_ARGUMENTS;
+    }
+
+    Mat src;
+    if (app.getFrame(params.imgFileName, src) != 0)
+    {
+        cout << "Error: \'src\' image is null or empty!" << endl;
+        return WRONG_INPUT;
+    }
+
+    const std::string caption = "OpenCV Sample";
+    char key = 0;
+    Mat dst(src.rows, src.cols, src.type());
+    while (key != 27) // Esc
+    {
+        key = app.showFrame(caption, src, dst);
+    }
+
+	return OK;
+}

sample_maslova/processing.cpp

@@ -0,0 +1,56 @@
+#include "processing.hpp"
+
+#include <opencv2/imgproc/imgproc.hpp>
+#include <time.h>
+#include <math.h>
+using namespace cv;
+#define PI 3.1415926
+
+void Processing::processFrame(const cv::Mat& src, cv::Mat& dst, FilterType filter = MEDIAN)
+{
+	srand((unsigned)time(NULL));
+	int h, w;
+	w = src.rows/8+rand()%(src.rows/8-1);
+	h = src.rows/8+rand()%(src.cols/8-1);
+    src.copyTo(dst);
+	static double diff = 0, dir=0.01;
+	if(diff<0||diff>2*PI) dir*=-1;
+	diff+=dir;
+	int x = src.rows/2+src.rows/4*cos(diff);
+	int y = src.cols/2+src.cols/4*sin(diff);
+    cv::Rect region(x-w, y-h, w, h);
+    Mat roi = dst(region);
+	int kSize, i=0, j=0;
+	cv::Rect rect;
+	Mat gray;
+	switch (filter){
+	case MEDIAN: 
+		kSize=11;
+		medianBlur(roi, roi, kSize); 
+		break;
+	case CVT_CONVERT_GRAY: 
+		cvtColor(roi, gray, COLOR_BGR2GRAY); cvtColor(gray, roi, COLOR_GRAY2BGR);
+		break;
+	case PIXELIZED: 
+		kSize=3;
+		for (i = 0; i < roi.rows; i += kSize)
+		{
+			for (int j = 0; j < roi.cols; j += kSize)
+			{
+				rect = cv::Rect(j, i, kSize, kSize) & 
+                            cv::Rect(0, 0, roi.cols, roi.rows);
+
+				cv::Mat sub_dst(roi, rect);
+				sub_dst.setTo(cv::mean(roi(rect)));
+			}
+		}
+		break;
+	case CANNY:
+		cvtColor(roi, gray, COLOR_BGR2GRAY); 
+		Canny(gray, gray, 0, 50);
+		cvtColor(gray,roi,COLOR_GRAY2BGR);
+		break;
+
+	}
+    rectangle(dst, region, Scalar(255, 0, 0));
+}

sample_maslova/processing.hpp

@@ -0,0 +1,18 @@
+#pragma once
+
+#include <opencv2/core/core.hpp>
+
+class Processing
+{
+ public:
+
+	enum FilterType
+	{
+		MEDIAN,
+		CVT_CONVERT_GRAY,
+		PIXELIZED,
+		CANNY
+	};
+
+    void processFrame(const cv::Mat& src, cv::Mat& dst, FilterType filter);
+};