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

aux

Browse Source
This commit is contained in:
nuknal
2024-05-20 11:08:24 +08:00
parent f609b2b023
commit efd824cc8a
14 changed files with 332 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
cmd/extract.go
View File

@@ -9,6 +9,7 @@ import (
var (
dataId string
batch bool
)
var extractCmd = &cobra.Command{
@@ -16,18 +17,31 @@ 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) {
params := extract.Params{
InputData: fmt.Sprintf("demo/data/%s.dat", dataId),
OutputPath: "demo/output",
TempPath: "demo/temp",
DataId: dataId,
Satellite: "SJY01",
}
p := extract.NewExtractor(&params)
aos, _ := p.ExtractAosData()
dats, _ := p.ExtractOriginalImageData(aos)
for i, d := range dats {
p.SeprateAuxAndImgData(d, i)
if batch {
ps := params()
for _, p := range ps {
e := extract.NewExtractor(p)
aos, _ := e.ExtractAosData()
dats, _ := e.ExtractOriginalImageData(aos)
for i, d := range dats {
e.SeprateAuxAndImgData(d, i)
}
}
} else {
p := &extract.Params{
InputData: fmt.Sprintf("demo/data/%s.dat", dataId),
OutputPath: fmt.Sprintf("demo/output/%s", dataId),
TempPath: fmt.Sprintf("demo/temp/%s", dataId),
DataId: dataId,
Satellite: "SJY01",
}
e := extract.NewExtractor(p)
aos, _ := e.ExtractAosData()
dats, _ := e.ExtractOriginalImageData(aos)
for i, d := range dats {
e.SeprateAuxAndImgData(d, i)
}
}
},
}
@@ -35,5 +49,23 @@ var extractCmd = &cobra.Command{
func init() {
rootCmd.AddCommand(extractCmd)
extractCmd.Flags().StringVarP(&dataId, "data-id", "d", "051513", "051513")
extractCmd.Flags().StringVarP(&dataId, "data-id", "d", "051622", "051622")
extractCmd.Flags().BoolVarP(&batch, "batch", "b", false, "true | false")
}
func params() []*extract.Params {
var params []*extract.Params
datas := []string{"051513", "051522", "051613", "051622",
"051712", "051721", "051821", "051823", "051921", "051922", "Q051723"}
for _, d := range datas {
params = append(params, &extract.Params{
InputData: fmt.Sprintf("demo/data/%s.dat", d),
OutputPath: fmt.Sprintf("demo/output/%s", d),
TempPath: fmt.Sprintf("demo/temp/%s", d),
DataId: d,
Satellite: "SJY01",
})
}
return params
}