77 lines
2.0 KiB
Python
77 lines
2.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import isce
|
||
|
|
from isceobj.Sensor import createSensor
|
||
|
|
import shelve
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
from isceobj.Util import Poly1D
|
||
|
|
from isceobj.Planet.AstronomicalHandbook import Const
|
||
|
|
from mroipac.dopiq.DopIQ import DopIQ
|
||
|
|
import copy
|
||
|
|
|
||
|
|
|
||
|
|
def cmdLineParse():
|
||
|
|
'''
|
||
|
|
Command line parser.
|
||
|
|
'''
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description='Unpack RISAT raw data and store metadata in pickle file.')
|
||
|
|
parser.add_argument('-i','--input', dest='indir', type=str,
|
||
|
|
required=True, help='Input CSK frame')
|
||
|
|
parser.add_argument('-o', '--output', dest='slc', type=str,
|
||
|
|
required=True, help='Output SLC file')
|
||
|
|
parser.add_argument('-p', '--polar', dest='polar', type=str,
|
||
|
|
default='RH', help='Polarization to extract')
|
||
|
|
|
||
|
|
return parser.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
def unpack(hdf5, slcname, polar='RH'):
|
||
|
|
'''
|
||
|
|
Unpack HDF5 to binary SLC file.
|
||
|
|
'''
|
||
|
|
|
||
|
|
obj = createSensor('RISAT1')
|
||
|
|
obj._imageFile = os.path.join(hdf5, 'scene_'+polar, 'dat_01.001')
|
||
|
|
obj._leaderFile = os.path.join(hdf5, 'scene_'+polar,'lea_01.001')
|
||
|
|
if not os.path.isdir(slcname):
|
||
|
|
os.mkdir(slcname)
|
||
|
|
|
||
|
|
date = os.path.basename(slcname)
|
||
|
|
obj.output = os.path.join(slcname, date + '.raw')
|
||
|
|
|
||
|
|
obj.extractImage()
|
||
|
|
|
||
|
|
obj.frame.getImage().renderHdr()
|
||
|
|
|
||
|
|
#####Estimate doppler
|
||
|
|
dop = DopIQ()
|
||
|
|
dop.configure()
|
||
|
|
|
||
|
|
img = copy.deepcopy(obj.frame.getImage())
|
||
|
|
img.setAccessMode('READ')
|
||
|
|
|
||
|
|
dop.wireInputPort('frame', object=obj.frame)
|
||
|
|
dop.wireInputPort('instrument', object=obj.frame.instrument)
|
||
|
|
dop.wireInputPort('image', object=img)
|
||
|
|
dop.calculateDoppler()
|
||
|
|
dop.fitDoppler()
|
||
|
|
fit = dop.quadratic
|
||
|
|
coef = [fit['a'], fit['b'], fit['c']]
|
||
|
|
|
||
|
|
print(coef)
|
||
|
|
obj.frame._dopplerVsPixel = [x*obj.frame.PRF for x in coef]
|
||
|
|
|
||
|
|
pickName = os.path.join(slcname, 'raw')
|
||
|
|
with shelve.open(pickName) as db:
|
||
|
|
db['frame'] = obj.frame
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
'''
|
||
|
|
Main driver.
|
||
|
|
'''
|
||
|
|
|
||
|
|
inps = cmdLineParse()
|
||
|
|
unpack(inps.indir, inps.slc, polar=inps.polar)
|