保留可配置入口,未启用

cmd/proc.go

@@ -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")

cmd/root.go

@@ -6,6 +6,10 @@ import (
 	"github.com/spf13/cobra"
 )
 
+var (
+	configFile string
+)
+
 var rootCmd = &cobra.Command{
 	Use:   "SJY01 Image processing tools",
 	Short: "Preprocessing tools for SJY01 original image data",
@@ -14,3 +18,7 @@ var rootCmd = &cobra.Command{
 		fmt.Println("preprocessing tools for SJY01 original image data")
 	},
 }
+
+func init() {
+	rootCmd.PersistentFlags().StringVarP(&configFile, "config", "c", "config/config.yaml", "config file (default is config/config.yaml)")
+}