디지털 영상처리 - Add Noise

double uniform()
{
	return((double)(rand() & RAND_MAX) / RAND_MAX - 0.5);
}

double gaussian() // 가우시안 난수 생성
{
	static int ready = 0;
	static double gstore;
	double v1, v2, r, fac, gaus;
	double uniform();

	if (ready == 0)
	{
		do {
			v1 = 2. * uniform();
			v2 = 2. * uniform();
			r = v1 * v1 + v2 * v2;
		} while (r > 1.0);

		fac = sqrt(-2. * log(r) / r);
		gstore = v1 * fac;
		gaus = v2 * fac;
		ready = 1;

	}
	else
	{
		ready = 0;
		gaus = gstore;
	}

	return(gaus);
}

void AddNoise(uchar** img, uchar** outimg, int Row, int Col) // 가우시안 난수를 더해 잡음 생성
{
	int i, j;
	int temp;
	for (i = 0; i < Row; i++)
		for (j = 0; j < Col; j++) {

			outimg[i][j] = img[i][j] + gaussian() * 50;
		}

}

void RmNoise(uchar** img, int** tempimg, uchar** outimg, int Row, int Col, int count) // 잡음영상을 더해 잡음 제거
{
	int i, j;
	int k;
	int temp;
	for (k = 0; k < count; k++) { // count : 잡음영상 더할 횟수
		for (i = 0; i < Row; i++)
			for (j = 0; j < Col; j++) {
				tempimg[i][j] += (img[i][j] + gaussian() * 50);
				// unsigned char 는 0~255 의 값만을 지니기 때문에
				// int형으로 선언하여 누적시킴
//				printf("%lf\n", gaussian() * 50);
			}
	}

	for (i = 0; i < Row; i++)
		for (j = 0; j < Col; j++) {
			temp = tempimg[i][j] / count; // 누적 값을 평균내어 temp에 저장
//			printf("temp : %d\n", temp);

			if (temp < 0) // clipping
			{
				temp = 0;
				outimg[i][j] = temp;
			}
			else if (temp > 255)
			{
				temp = 255;
				outimg[i][j] = temp;
			}
			else
			{
				outimg[i][j] = temp;
			}
//			printf("outimg[i][j] : %d\n", outimg[i][j]);
		}

}