package rrc

import (
	"bufio"
	"fmt"
	"os"

	log "github.com/sirupsen/logrus"
	"gocv.io/x/gocv"
)

// Relative Radiation Correction
// 采用在轨统计定标方法。利用卫星在轨后获取的常规影像数据，统计每个探元出现的灰度频次。

const (
	PANCameraProbeNum = 9344  // 全色探元数
	MSSCameraProbeNum = 2336  // 多光谱探元数
	MaxGrayLevel      = 65536 // 16bit像素值
)

type RRC struct {
	Histograms [5]ProbeHistogram
}

func NewRRC() *RRC {
	r := RRC{}
	r.Histograms[0].init(PANCameraProbeNum)
	for i := 1; i < 5; i++ {
		r.Histograms[i].init(MSSCameraProbeNum)
	}

	os.MkdirAll("data/rrc", 0755)

	return &r
}

func (rrc *RRC) Statistical(dsPAN, dsMSS string) error {
	rrc.StatisticalPAN(dsPAN)
	rrc.StatisticalMSS(dsMSS)

	return nil
}

func (rrc *RRC) Close() {}

// 统计探元灰度的累积概率密度
func (rrc *RRC) StatisticalPAN(dsfile string) error {
	f, err := os.Open(dsfile)
	if err != nil {
		return err
	}
	defer f.Close()

	scanner := bufio.NewScanner(f)
	for scanner.Scan() {
		l0 := scanner.Text()
		log.Println("statistical PAN RAW: ", l0)
		data, err := os.ReadFile(l0)
		if err != nil {
			log.Error("Error reading file: ", err)
			continue
		}

		height := len(data) / (PANCameraProbeNum * 2)
		img, err := gocv.NewMatFromBytes(height, PANCameraProbeNum, gocv.MatTypeCV16U, data)
		if err != nil {
			log.Error("Error creating Mat from bytes: ", err)
			return err
		}

		rrc.Histograms[0].statistical(img)
		img.Close()
	}

	log.Println("compute PAN histogram...")
	rrc.Histograms[0].compute()
	log.Println("save PAN gray table matrix.")
	rrc.Histograms[0].save("data/rrc/pan_gray_table.dat")

	return nil
}

func (rrc *RRC) StatisticalMSS(dsfile string) error {
	f, err := os.Open(dsfile)
	if err != nil {
		return err
	}
	defer f.Close()

	scanner := bufio.NewScanner(f)
	for scanner.Scan() {
		l0 := scanner.Text()
		log.Println("statistical MSS RAW: ", l0)
		data, err := os.ReadFile(l0)
		if err != nil {
			log.Error("Error reading file: ", err)
			continue
		}

		width := MSSCameraProbeNum
		height := len(data) / (width * 2)
		mssData := make([][]byte, 4)
		for h := 0; h < height; h++ {
			row := data[h*width*4*2 : (h+1)*width*4*2]
			for i := 0; i < 4; i++ {
				mssData[i] = append(mssData[i], row[i*width*2:(i+1)*width*2]...)
			}
		}

		var mssImages [4]gocv.Mat
		for i := 0; i < 4; i++ {
			mssImages[i], err = gocv.NewMatFromBytes(height, width, gocv.MatTypeCV16U, mssData[i])
			if err != nil {
				log.Error("Error creating Mat from bytes: ", err)
				return err
			}

			rrc.Histograms[i+1].statistical(mssImages[i])
			mssImages[i].Close()
		}
	}

	for i := 1; i < 5; i++ {
		log.Println("compute MSS histogram...")
		rrc.Histograms[i].compute()
		log.Println("save MSS gray table matrix.")
		rrc.Histograms[i].save(fmt.Sprintf("data/rrc/mss%d_gray_table.dat", i))
	}

	return nil
}