支持xml入参

cmd/proc.go

@@ -25,6 +25,8 @@ var procCmd = &cobra.Command{
 		config.GViper = config.InitViper(configFile)
 		logrus.SetLevel(config.GCONFIG.LogLevel)
 
+		initParams()
+
 		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")
@@ -92,3 +94,29 @@ func init() {
 	procCmd.Flags().BoolVarP(&saveStrip, "save-strip", "", false, "save original and registered images as GDAL GTiff")
 	procCmd.Flags().StringVarP(&paramsXML, "params", "x", "", "params xml file path")
 }
+
+func initParams() producer.Params {
+	taskParams := params
+	taskParams.MssTiffFile = filepath.Join(params.OutputDir, strings.TrimSuffix(filepath.Base(params.MssRawFile),
+		filepath.Ext(params.MssRawFile))+".tiff")
+	taskParams.PanTiffFile = filepath.Join(params.OutputDir,
+		strings.TrimSuffix(filepath.Base(params.PanRawFile),
+			filepath.Ext(params.PanRawFile))+".tiff")
+	taskParams.FusTIffFile = strings.Replace(taskParams.MssTiffFile, ".tiff", "_FUS.tiff", 1)
+
+	if paramsXML == "" {
+		return taskParams
+	}
+
+	task, err := producer.ParseXMLImageTask(paramsXML)
+	if err != nil {
+		logrus.Fatal(err)
+	}
+
+	taskParams.PanRawFile = task.InputFileList.PanData
+	taskParams.MssRawFile = task.InputFileList.MssData
+	taskParams.AuxRawFile = task.InputFileList.AuxData
+	taskParams.DoPansharpen = task.Params.DoPansharpen
+	taskParams.OutputDir = task.Params.OutputPath
+	return taskParams
+}