#include "ContaminationDetection.h"
#include <vector>
#include <cmath>
#include <cstring>

#define MIN(a, b) ((a) < (b) ? (a) : (b))

ContaminationDetection::ContaminationDetection()
{
}

int ContaminationDetection::detect1(const uchar *data1, const uchar *data2, int length, uchar threshold1, uchar threshold2, int width)
{
	int width_temp = 0;
	for (size_t i = 0; i < length; i++)
	{
		if (MIN(data1[i], data2[i]) > threshold1)
			return 1;

		if (MIN(data1[i], data2[i]) > threshold2)
		{
			width_temp++;
			if (width_temp > width)
				return 2;
		}
		else
			width_temp = 0;
	}
	return 0;
}

int ContaminationDetection::detect2(const uchar *data, int width, int height, int channels, uchar color, int length)
{
	uchar *ptr = const_cast<uchar *>(data);
	std::vector<int> sum_threshold;
	sum_threshold.resize(height);

	uchar table[256];
	memset(table, 1, 256);
	memset(table, 0, color);

	int sum, bit_sum;
	for (size_t y = 0, bytePerLine = width * channels; y < height; y++)
	{
		sum = 0;
		for (size_t x = 0; x < bytePerLine; x += channels)
		{
			bit_sum = 0;
			for (size_t i = 0; i < channels; i++)
				bit_sum |= table[ptr[x + i]];

			sum += bit_sum;
		}
		sum_threshold[y] = sum;
		ptr += bytePerLine;
	}

	if (sum_threshold[0] > length || sum_threshold[height - 1] > length)
	{
		if (std::abs(sum_threshold[0] - sum_threshold[height - 1]) > 10)
			return 1;
		else
			return 2;
	}

	return 0;
}