保留可配置入口,未启用

This commit is contained in:
nuknal
2024-05-30 11:46:50 +08:00
parent 07ee4d88d4
commit e4d6b35702
11 changed files with 159 additions and 28 deletions

View File

@@ -8,6 +8,7 @@ import (
"github.com/airbusgeo/godal"
"github.com/sirupsen/logrus"
"github.com/spf13/cobra"
"starwiz.cn/sjy01/image-proc/pkg/config"
producer "starwiz.cn/sjy01/image-proc/pkg/producer"
)
@@ -21,6 +22,8 @@ var procCmd = &cobra.Command{
Short: "process images",
Long: `process images`,
Run: func(cmd *cobra.Command, args []string) {
_ = config.InitViper(configFile)
reg := producer.NewRegistrator(producer.DownSampled)
reg.Params = params
reg.Params.MssTiffFile = filepath.Join(params.OutputDir, strings.TrimSuffix(filepath.Base(params.MssRawFile), filepath.Ext(params.MssRawFile))+".tiff")
@@ -41,7 +44,7 @@ var procCmd = &cobra.Command{
if err := reg.DoPhaseCorrelation(); err != nil {
panic(err)
}
reg.DoCoRegestration()
reg.DoCoRegistration()
if params.SaveRegisteredMssRaw {
registerdMSSRAW := filepath.Join(
@@ -63,7 +66,7 @@ var procCmd = &cobra.Command{
reg.SaveRegisteredMssToGDALGTiff(reg.Params.MssTiffFile)
}
if params.PansharpenIHS {
if params.DoPansharpen {
reg.DoScenePansharpen(panScenes, mssScenes)
}
@@ -78,7 +81,7 @@ func init() {
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().BoolVarP(&params.DoPansharpen, "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")
procCmd.Flags().BoolVarP(&saveStrip, "save-strip", "", false, "save original and registered images as GDAL GTiff")