DFT 低通滤波器

2024-06-19 22:42:31 +08:00
parent 6e5ffe1ab2
commit e13038474a
8 changed files with 337 additions and 2 deletions
--- a/pkg/utils/tiff.go
+++ b/pkg/utils/tiff.go
@@ -0,0 +1,95 @@
+package utils
+
+import (
+	"time"
+
+	"github.com/airbusgeo/godal"
+	log "github.com/sirupsen/logrus"
+	"starwiz.cn/sjy01/image-proc/pkg/calculator"
+
+	"gocv.io/x/gocv"
+)
+
+func SavePanToGDALGTiff(pan gocv.Mat, topLeftX, topLeftY float64, tiffFile string, resolution float64) error {
+	// log.Println("Saving PAN image to TIFF file:", tiffFile)
+
+	width := pan.Cols()
+	height := pan.Rows()
+
+	ds, err := godal.Create(godal.GTiff, tiffFile, 1, godal.UInt16, width, height)
+	if err != nil {
+		log.Error("Error creating TIFF file: ", err)
+		return err
+	}
+	defer ds.Close()
+
+	setGeoTransform(ds, topLeftX, topLeftY, resolution)
+	ds.SetMetadata("NBITS", "16")
+
+	// 将通道的数据转换为uint16数组
+	data := make([]uint16, width*height)
+	for y := 0; y < height; y++ {
+		for x := 0; x < width; x++ {
+			data[y*width+x] = uint16(pan.GetShortAt(y, x))
+		}
+	}
+
+	band := ds.Bands()[0]
+	band.SetColorInterp(godal.CIGray)
+	err = band.IO(godal.IOWrite,
+		0, 0,
+		data,
+		width, height,
+		godal.PixelSpacing(2),
+		godal.LineSpacing(width*2))
+	if err != nil {
+		log.Error("Failed to write data to band:", err)
+		return err
+	}
+
+	log.Info("Saved pan image to ", tiffFile)
+
+	return nil
+}
+
+func setGeoTransform(ds *godal.Dataset, topLeftLng, topLeftLat, resolution float64) {
+	// 转换分辨率（米到度）
+	resLat := resolution / calculator.MetersPerDegreeLatitude
+	resLng := calculator.MetersToDegreesLongitude(resolution, topLeftLat)
+	// 设置地理变换（假设左上角坐标为(0,0)，PAN每个像素分辨率为1.2米）
+	geotransform := [6]float64{
+		topLeftLng, // top left x
+		resLng,     // w-e pixel resolution
+		0,          // rotation, 0 if image is "north up"
+		topLeftLat, // top left y
+		0,          // rotation, 0 if image is "north up"
+		-resLat,    // n-s pixel resolution (negative value)
+	}
+	err := ds.SetGeoTransform(geotransform)
+	if err != nil {
+		log.Errorf("Failed to set GeoTransform: %v", err)
+		return
+	}
+
+	// 设置投影为 WGS84
+	srs, err := godal.NewSpatialRefFromEPSG(4326) //  gdal.CreateSpatialReference("")
+	if err != nil {
+		log.Errorf("Failed to set spatial reference: %v", err)
+		return
+	}
+
+	projWKT, err := srs.WKT()
+	if err != nil {
+		log.Errorf("Failed to convert spatial reference to WKT: %v", err)
+		return
+	}
+
+	err = ds.SetProjection(projWKT)
+	if err != nil {
+		log.Errorf("Failed to set projection: %v", err)
+	}
+
+	// 设置一些常见的元数据（可选）
+	ds.SetMetadata("TIFFTAG_DATETIME", time.Now().String())
+	ds.SetMetadata("TIFFTAG_SOFTWARE", "StarWiz-SJY01-IMAGE-PROC")
+}