U cX5@sddlZddlZddlZddlZddlZddlZedZe ej ddl m Z m Z mZmZddlmZdZdddZddd Zd d ZdddZdd ddddgdfddZdddZedkreejd ddS)Nnumpy)readfile writefileutils1 attribute)create_argument_parserzexample: multilook.py velocity.h5 -r 15 -a 15 multilook.py srtm30m.dem -r 10 -a 10 -o srtm30m_300m.dem # Ignore / skip marginal pixels multilook.py ../../geom_reference/hgt.rdr.full -r 300 -a 100 --margin 58 58 58 58 -o hgt.rdr c Csd}t}tdd}t|||||d}|jdddd|jd d d d td dd|jddddtd dd|jdddd|jdddtddddgdd |jd!d"td#d$d%d%d%d%gd&d'|S)(NzMultilook the input file.)synopsis descriptionepilog subparsersfile+zFile(s) to multilook)nargshelpz-rz--rangez-xlks_xzEnumber of multilooking in range /x direction (default: %(default)s).)desttypedefaultrz-az --azimuthz-ylks_yzEnumber of multilooking in azimuth/y direction (default: %(default)s).z-oz --outfilez7Output file name. Disabled when more than 1 input files)rz-mz--methodmethodmeanmediannearestz_downsampling method (default: %(default)s) e.g. nearest for geometry, average for observations)rrrchoicesrz--marginmargin)ZTOPZBOTTOMLEFTRIGHTrz?number of pixels on the margin to skip, (default: %(default)s).)rrrmetavarrr)EXAMPLE__name__splitr add_argumentintstr)r r r nameparserr*G/home/exouser/operations/rsmas_insar/sources/MintPy/mintpy/multilook.py create_parser's8 r,cCsbt}|j|d}t|j|_|jdkr<|jdkrtj|||||ddd }n&tj|||dd d }t||||d!}|jd krd |jd |d |d |d |d g}n|d |d |d |d g}| dkrtj ||||dd"n|||<qbqt!|"d krDtj|d dkrD|#d#d$$d$krDtd%|d#td&d$|d#<| dkrt!|"|d'<tj%||||d(tj&|d)stj&|d*rt'|||S)+aA Multilook input file Parameters: infile - str, path of input file to be multilooked. lks_y - int, number of looks in y / row direction. lks_x - int, number of looks in x / column direction. margin - list of 4 int, number of pixels to be skipped during multilooking. useful for offset product, where the marginal pixels are ignored during cross correlation matching. outfile - str, path of output file Returns: outfile - str, path of output file LENGTHWIDTH FILE_TYPEzmultilooking {} {} file: {} PROCESSORz,number of looks in y / azimuth direction: %dz,number of looks in x / range direction: %dzmultilook method: {}rrLrMrz@number of pixels to skip in top/bottom/left/right boundaries: {}_Zalks_Zrlks)box)z.h5z.he5)metadataref_filecSsg|] }t|qSr*)r2).0ir*r*r+ sz"multilook_file..z"multilooking {d:<{w}} from {f} ...)dwfrIri@ zbox: {}rF) datasetNamercxstepystep print_msg)rmrcrp)r)rWrmblockrp INTERLEAVEBILz=the input binary file has 2 bands with band interleave as: {}zIfor the output binary file, change the band interleave to BIL as default.BANDS)out_filerdrez.xmlz.aux.xml)(r&rread_attributer4rPospathsplitextgetcwddirnameabspathr'basenameattrupdate_attribute4multilookr layout_hdf5get_dataset_listmaxdicth5pyFiler<prodr;ceilrintrCminreadr]ndimwrite_hdf5_blockr2keysgetupperwriteisfilewrite_isce_xml)infilerrr3rr max_memoryatrlengthwidthkrcextdsNamesmaxDigitZdsDictdsNamerkdsds_sizenum_steprow_steprgr0r1Zbox_iZbox_orWrqr*r*r+multilook_files    $ 2       "      $ rc Cs>t|}|jD]"}t||j|j|j|j|jdqtddS)N)rrr3rrzDone.) r7rrrrr3rrr4)r5r6rr*r*r+mainEs r__main__)N)N)rrr)N)rwsysr?rrr<logging getLoggerZ np_loggersetLevelWARNING mintpy.utilsrrrr/rr~Zmintpy.utils.arg_utilsrr"r,r7rJr]rrr#argvr*r*r*r+ s&      Y