path: root/DCP/dcp.cpp

#include "dcp_core.h"

using namespace cv::ximgproc;

void dehaze(IplImage *recover, IplImage *input)
{
	int height = input->height;
	int width = input->width;
	
	IplImage *darkchannel = cvCreateImage(cvSize(width, height), IPL_DEPTH_8U, 1);
	IplImage *transmission = cvCreateImage(cvSize(width, height), IPL_DEPTH_8U, 1);
	IplImage *refine_transmission = cvCreateImage(cvSize(width, height), IPL_DEPTH_8U, 1);

	int darkchannelradius = cvRound(MIN(width, height) * 0.02);
	double Airlight[3] = { 0.0, 0.0, 0.0 };

	TicToc t;
	t.tic();
	printf("CalcDarkChannel...");
	CalcDarkChannel(darkchannel, input, darkchannelradius);
	t.toc();

	t.tic();
	printf("CalcAirlight...");
	CalcAirlight(darkchannel, input, Airlight);
	t.toc();

	t.tic();
	printf("CalcTransmission...");
	CalcTransmission(transmission, input, Airlight, darkchannelradius);
	t.toc();

	t.tic();
	printf("GuidedFilterColor...");
	// GuidedFilterColor() is my own implementation of guided filter
	// guidedFilter() is the implementation included in OpenCV 3.0
	// The result is almost the same, however my implementation is much more slower
	// because I haven't spend much time optimizing the code efficiency.

	// See http://research.microsoft.com/en-us/um/people/kahe/eccv10/ for more details
	// ref:
	// He, Kaiming, Jian Sun, and Xiaoou Tang. "Guided image filtering." 
	// Pattern Analysis and Machine Intelligence, IEEE Transactions on 35.6 (2013): 1397-1409.

	GuidedFilterColor(refine_transmission, input, transmission, 1e-6, 60);
	// guidedFilter(cv::cvarrToMat(input), cv::cvarrToMat(transmission), cv::cvarrToMat(refine_transmission), 60, 1e-6);
	t.toc();

	t.tic();
	printf("CalcRecover...");
	CalcRecover(recover, input, refine_transmission, Airlight);	
	t.toc();

	cvReleaseImage(&refine_transmission);
	cvReleaseImage(&transmission);
	cvReleaseImage(&darkchannel);
}