lankun/sjy01-image-proc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

do not use IHS

Browse Source
This commit is contained in:
nuknal
2024-05-28 15:15:53 +08:00
parent 538fb71110
commit 5eb5ae6a72
12 changed files with 346 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
cmd/main.go
View File

@@ -1,18 +1,5 @@
package main
import (
"github.com/sirupsen/logrus"
)
func init() {
logrus.SetFormatter(&logrus.TextFormatter{
ForceColors: true,
DisableColors: false,
TimestampFormat: "2006-01-02 15:04:05",
FullTimestamp: true,
})
}
func main() {
rootCmd.Execute()
}

56
cmd/proc.go
View File

@@ -6,17 +6,13 @@ import (
"strings"
"github.com/airbusgeo/godal"
log "github.com/sirupsen/logrus"
"github.com/spf13/cobra"
imageproc "starwiz.cn/sjy01/image-proc"
)
var (
PanRawFile string
MssRawFile string
AuxRawFile string
SaveRegisteredMssRaw bool
PansharpenIHS bool
OutputDir string
params imageproc.Params
)
var procCmd = &cobra.Command{
@@ -25,20 +21,9 @@ var procCmd = &cobra.Command{
Long: `process images`,
Run: func(cmd *cobra.Command, args []string) {
reg := imageproc.NewRegistrator(imageproc.DownSampled)
reg.Params = imageproc.Params{
PanRawFile: PanRawFile,
MssRawFile: MssRawFile,
AuxRawFile: AuxRawFile,
SaveRegisteredMssRaw: SaveRegisteredMssRaw,
PansharpenIHS: PansharpenIHS,
OutputDir: OutputDir,
MssTiffFile: filepath.Join(OutputDir,
strings.TrimSuffix(filepath.Base(MssRawFile),
filepath.Ext(MssRawFile))+".tiff"),
PanTiffFile: filepath.Join(OutputDir,
strings.TrimSuffix(filepath.Base(PanRawFile),
filepath.Ext(PanRawFile))+".tiff"),
}
reg.Params = params
reg.Params.MssTiffFile = filepath.Join(params.OutputDir, strings.TrimSuffix(filepath.Base(params.MssRawFile), filepath.Ext(params.MssRawFile))+".tiff")
reg.Params.PanTiffFile = filepath.Join(params.OutputDir, strings.TrimSuffix(filepath.Base(params.PanRawFile), filepath.Ext(params.PanRawFile))+".tiff")
reg.Params.FusTIffFile = strings.Replace(reg.Params.MssTiffFile, ".tiff", "_FUS.tiff", 1)
if err := reg.LoadMssRaw(); err != nil {
@@ -50,26 +35,32 @@ var procCmd = &cobra.Command{
}
godal.RegisterAll()
os.MkdirAll(OutputDir, 0755)
os.MkdirAll(params.OutputDir, 0755)
if err := reg.DoPhaseCorrelation(); err != nil {
panic(err)
}
reg.DoCoRegestration()
if SaveRegisteredMssRaw {
if params.SaveRegisteredMssRaw {
registerdMSSRAW := filepath.Join(
OutputDir,
strings.TrimSuffix(filepath.Base(MssRawFile), filepath.Ext(MssRawFile))+"_registered.RAW",
params.OutputDir,
strings.TrimSuffix(filepath.Base(params.MssRawFile), filepath.Ext(params.MssRawFile))+"_registered.RAW",
)
reg.SaveRegisteredMssToRaw(registerdMSSRAW)
}
panScenes, mssScenes, err := reg.SubScenes()
if err != nil {
log.Error(err)
}
reg.SaveScenesToTiff(panScenes, mssScenes)
reg.SaveOriginalPanToGDALGTiff(reg.Params.PanTiffFile)
reg.SaveRegisteredMssToGDALGTiff(reg.Params.MssTiffFile)
if PansharpenIHS {
reg.SavePansharpenedToGDALGTiff(reg.Params.FusTIffFile)
if params.PansharpenIHS {
reg.DoScenePansharpen(panScenes, mssScenes)
}
reg.Clean()
@@ -79,10 +70,11 @@ var procCmd = &cobra.Command{
func init() {
rootCmd.AddCommand(procCmd)
procCmd.Flags().StringVarP(&PanRawFile, "pan", "p", "pan.raw", "PAN image raw file path")
procCmd.Flags().StringVarP(&MssRawFile, "mss", "m", "mss.raw", "MSS image raw file path")
procCmd.Flags().StringVarP(&AuxRawFile, "aux", "a", "pms.aux", "AUX image raw file path")
procCmd.Flags().BoolVarP(&SaveRegisteredMssRaw, "srmss", "s", false, "save registered MSS image raw file")
procCmd.Flags().BoolVarP(&PansharpenIHS, "fus", "", false, "pansharpen using IHS")
procCmd.Flags().StringVarP(&OutputDir, "output-dir", "o", "data", "output directory")
procCmd.Flags().StringVarP(&params.PanRawFile, "pan", "p", "pan.raw", "PAN image raw file path")
procCmd.Flags().StringVarP(&params.MssRawFile, "mss", "m", "mss.raw", "MSS image raw file path")
procCmd.Flags().StringVarP(&params.AuxRawFile, "aux", "a", "pms.aux", "AUX image raw file path")
procCmd.Flags().BoolVarP(&params.SaveRegisteredMssRaw, "srmss", "s", false, "save registered MSS image raw file")
procCmd.Flags().BoolVarP(&params.PansharpenIHS, "fus", "", false, "pansharpen using IHS")
procCmd.Flags().StringVarP(&params.OutputDir, "output-dir", "o", "data", "output directory")
procCmd.Flags().BoolVarP(&params.SubScenes, "sub-scenes", "", false, "process sub-scenes")
}