package main

import (
	"image"
	"image/color"
	"os"
)

// 얼룩 제거 알고리즘 적용 함수
func removeSpots(img image.Image) image.Image {
	bounds := img.Bounds()
	newImg := image.NewRGBA(bounds)

	for y := bounds.Min.Y; y < bounds.Max.Y; y++ {
		for x := bounds.Min.X; x < bounds.Max.X; x++ {
			r, g, b, _ := img.At(x, y).RGBA()
			// 얼룩에 해당하는 픽셀 처리
			if isSpot(r, g, b) {
				// 주변 픽셀과 함께 평균값으로 대체
				avgColor := calculateAverageColor(img, x, y, 3)
				newImg.Set(x, y, avgColor)
			} else {
				// 얼룩이 아닌 경우 기존 값 그대로 유지
				newImg.Set(x, y, img.At(x, y))
			}
		}
	}
	return newImg
}

// 해당 픽셀이 얼룩인지 판별하는 함수
func isSpot(r, g, b uint32) bool {
	// 간단한 임계값을 기준으로 얼룩 여부를 판별
	threshold := uint32(10000)
	return (r < threshold && g < threshold && b < threshold)
}

// 인접한 픽셀의 색상 평균을 계산하는 함수
func calculateAverageColor(img image.Image, x, y, size int) color.RGBA {
	var r, g, b, count uint32
	bounds := img.Bounds()
	for j := y - size/2; j < y+size/2; j++ {
		for i := x - size/2; i < x+size/2; i++ {
			if i >= bounds.Min.X && i < bounds.Max.X && j >= bounds.Min.Y && j < bounds.Max.Y {
				c := img.At(i, j)
				cr, cg, cb, _ := c.RGBA()
				r += cr
				g += cg
				b += cb
				count++
			}
		}
	}

	return color.RGBA{uint8(r / count), uint8(g / count), uint8(b / count), 255}
}

func main() {
	// 이미지 파일 열기
	file, err := os.Open("input.jpg")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	// 이미지 디코딩
	img, _, err := image.Decode(file)
	if err != nil {
		panic(err)
	}

	// 얼룩 제거 알고리즘 적용
	newImg := removeSpots(img)

	// 결과 이미지 파일로 저장
	outFile, err := os.Create("output.jpg")
	if err != nil {
		panic(err)
	}
	defer outFile.Close()
	jpeg.Encode(outFile, newImg, nil)
}