package imageproc

import (
	"fmt"
	"image"

	"gocv.io/x/gocv"
)

const (
	GDALPansharpenCommand = "gdal_pansharpen.py {{.PANGTiff}} {{.MSSGTiff}} {{.FUSGTiff}} -r cubic -of GTiff"
)

// IHS 变换是一种将 RGB 空间转换为亮度、色调和饱和度的变换方法
func PansharpenIHS(panImage, mssImage gocv.Mat) gocv.Mat {
	// pan32 := gocv.NewMat()
	// defer pan32.Close()
	// panImage.ConvertTo(&pan32, gocv.MatTypeCV32FC1)

	mss32F := gocv.NewMat()
	mssImage.ConvertTo(&mss32F, gocv.MatTypeCV32FC4)

	// 调整 MSS 图像的大小到与 PAN 图像相同的大小
	mssResized := gocv.NewMat()
	defer mssResized.Close()
	gocv.Resize(mss32F, &mssResized,
		image.Point{X: panImage.Cols(), Y: panImage.Rows()}, 0, 0, gocv.InterpolationCubic)

	// 将 MSS 图像从 BGR 转换为 HLS
	mssHLS := gocv.NewMat()
	defer mssHLS.Close()
	gocv.CvtColor(mssResized, &mssHLS, gocv.ColorBGRToHLS)

	// 分离 HLS 通道 只有3个通道
	// Hue
	// Lightness
	// Saturation
	hlsChannels := gocv.Split(mssHLS)

	fmt.Println("size of channels:", len(hlsChannels), mssImage.Channels(), mssResized.Channels(), mssHLS.Channels())

	// 使用 PAN 图像替换亮度分量（Intensity）
	// hlsChannels[1] = pan32

	// 将 PAN 图像转换为与 HLS 亮度分量相同的范围
	panFloat := gocv.NewMat()
	defer panFloat.Close()
	panImage.ConvertTo(&panFloat, gocv.MatTypeCV32F)
	gocv.Normalize(panFloat, &panFloat, 0, 255, gocv.NormMinMax)

	hlsFloat := gocv.NewMat()
	defer hlsFloat.Close()
	hlsChannels[1].ConvertTo(&hlsFloat, gocv.MatTypeCV32F)
	gocv.Normalize(hlsFloat, &hlsFloat, 0, 255, gocv.NormMinMax)

	// 替换亮度分量
	hlsChannels[1] = panFloat

	// 合并通道
	gocv.Merge(hlsChannels, &mssHLS)

	// 将图像从 HLS 转换回 BGR
	fusedImage := gocv.NewMat()
	defer fusedImage.Close()
	gocv.CvtColor(mssHLS, &fusedImage, gocv.ColorHLSToBGR)

	// 交换 B 和 R 通道
	channels := gocv.Split(fusedImage)
	defer channels[0].Close() // B
	defer channels[1].Close() // G
	defer channels[2].Close() // R

	// 交换 B 和 R 通道
	swappedChannels := []gocv.Mat{channels[2], channels[1], channels[0]}
	gocv.Merge(swappedChannels, &fusedImage)

	fused16U := gocv.NewMat()
	fusedImage.ConvertTo(&fused16U, gocv.MatTypeCV16UC3)

	return fused16U
}