xml params

cmd/extract.go

@@ -3,14 +3,16 @@ package cmd
 import (
 	"fmt"
 
+	"github.com/k0kubun/pp/v3"
 	"github.com/spf13/cobra"
 	"starwiz.cn/sjy01/preprocessing/extract"
 )
 
 var (
-	dataId string
-	batch  bool
-	output string
+	dataId    string
+	batch     bool
+	output    string
+	paramsXML string
 )
 
 var extractCmd = &cobra.Command{
@@ -18,6 +20,24 @@ var extractCmd = &cobra.Command{
 	Short: "Extract data from raw data files",
 	Long:  `Extract data from raw data files`,
 	Run: func(cmd *cobra.Command, args []string) {
+		if paramsXML != "" {
+			params, err := extract.LoadL0Params(paramsXML)
+			if err != nil {
+				panic(err)
+			}
+
+			pp.Println(params)
+
+			e := extract.NewExtractor(params)
+			aos, _ := e.ExtractAosData()
+			dats, _ := e.ExtractOriginalImageData(aos)
+			for i, d := range dats {
+				e.SeprateAuxAndImgData(d, i)
+			}
+			e.Cleanup()
+
+			return
+		}
 
 		if batch {
 			ps := params()
@@ -55,6 +75,7 @@ func init() {
 	extractCmd.Flags().StringVarP(&dataId, "data-id", "d", "051622", "051622")
 	extractCmd.Flags().BoolVarP(&batch, "batch", "b", false, "true | false")
 	extractCmd.Flags().StringVarP(&output, "out", "o", "demo/output", "demo/output")
+	extractCmd.Flags().StringVarP(&paramsXML, "params", "x", "", "parameters file path")
 }
 
 func params() []*extract.Params {