refactor
This commit is contained in:
34
cmd/proc.go
34
cmd/proc.go
@@ -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()
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user