This commit is contained in:
nuknal
2024-10-29 16:45:28 +08:00
parent 9c2d93dff4
commit 0e920ce9f0
10 changed files with 57 additions and 57 deletions

View File

@@ -45,20 +45,20 @@ var procCmd = &cobra.Command{
calculator.EOP = calculator.NewEOPTable()
calculator.EOP.Load(eopData, eopp5Line)
reg := producer.NewRegistrator(producer.DownSampled)
reg.Params = initParams()
processor := producer.NewImgProc(producer.DownSampled)
processor.Params = initParams()
if err := reg.LoadAuxData(); err != nil {
if err := processor.LoadAuxData(); err != nil {
logrus.Fatal(err)
}
// reg.AuxPrint()
if err := reg.LoadMssRaw(); err != nil {
if err := processor.LoadMssRaw(); err != nil {
logrus.Fatal(err)
}
if err := reg.LoadPanRaw(); err != nil {
if err := processor.LoadPanRaw(); err != nil {
logrus.Fatal(err)
}
@@ -71,14 +71,14 @@ var procCmd = &cobra.Command{
os.MkdirAll(params.OutputDir, 0755)
if doLUTRRC {
reg.DoRRCbyLUT(lutDir)
processor.DoRRCbyLUT(lutDir)
}
if doMomentMatching {
reg.DoMomentMatching()
processor.DoMomentMatching()
}
if err := reg.DoPhaseCorrelation(); err != nil {
if err := processor.DoPhaseCorrelation(); err != nil {
logrus.Fatal(err)
}
@@ -87,33 +87,33 @@ var procCmd = &cobra.Command{
params.OutputDir,
strings.TrimSuffix(filepath.Base(params.MssRawFile), filepath.Ext(params.MssRawFile))+"_registered.RAW",
)
reg.SaveRegisteredMssToRaw(registerdMSSRAW)
processor.SaveRegisteredMssToRaw(registerdMSSRAW)
}
runtime.GC()
panScenes, mssScenes, err := reg.SubScenes()
panScenes, mssScenes, err := processor.SubScenes()
if err != nil {
logrus.Error(err)
}
reg.OutputL1A(panScenes, mssScenes)
processor.OutputL1A(panScenes, mssScenes)
producer.CleanScenes(panScenes)
producer.CleanScenes(mssScenes)
runtime.GC()
if saveStrip {
reg.SaveOriginalPanToGDALGTiff(reg.Params.PanTiffFile)
reg.SaveRegisteredMssToGDALGTiff(reg.Params.MssTiffFile)
processor.SaveOriginalPanToGDALGTiff(processor.Params.PanTiffFile)
processor.SaveRegisteredMssToGDALGTiff(processor.Params.MssTiffFile)
}
if reg.Params.DoPansharpen {
reg.DoScenePansharpen(panScenes, mssScenes)
if processor.Params.DoPansharpen {
processor.DoScenePansharpen(panScenes, mssScenes)
}
reg.Report()
processor.Report()
reg.Clean()
processor.Clean()
},
}