o Bd4@sxddlZddlZddlZddlZddlZddlmZm Z m Z e dZ e ejd ddZ  dd d Zdd d ZdS)N) attributereadfile writefilenumpymeanc Csgd}||vrd|d}|d|7}t|t|}t|}||dkr)|Stj|jtd}t|dkrt|||ft||f}|d|d d|df}|d vr| |d |||d||f} t &t j d t d |d krtj| dd} n |dkrtj| dd} Wdn1swYn|dkr|t|dd|t|dd|f} nt|dkrNt|d||ftd||f}|d|d d|dd|df}|d vr1| |d |d|||d||f} t (t j d t d |d krtj| dd} n |dkr tj| dd} Wdn 1s+wYn*|dkrM|ddt|dd|t|dd|f} n td|dt|dtj| |jd} | S)a,Apply multilooking (spatial averaging/resampling) to a multi-dimensional array. Link: https://stackoverflow.com/questions/34689519/how-to-coarser-the-2-d-array-data-resolution Parameters: data - 2D / 3D np.array in real or complex lks_y - int, number of multilook in y/azimuth direction lks_x - int, number of multilook in x/range direction method - str, multilook method, mean, median or nearest Returns: out_data - 2D / 3D np.array after multilooking in last two dimension )rmediannearestzun-supported multilook method: z. zAvailable methods: r)dtypeNr)rrignore)categoryr)r)axisrr r)r zUn-supported data dimension: z --> !) ValueErrorintnparrayshapefloatlenfloorastypereshapewarningscatch_warnings simplefilterRuntimeWarningnanmean nanmedianr ) datalks_ylks_xmethodZ method_listmsgr new_shapeZ crop_datatempout_datar*I/home/exouser/operations/rsmas_insar/tools/MintPy/src/mintpy/multilook.pymultilook_datasx       $           r,rc ( Cst|}t|}|drt|} nt|} t| dt| d} } td|td|td|d||r||d} ||d }| d |d |d|d|d }| d |d |d |d |d }| ||}|||}| |||f}nn| rt| }t|dt|d}}| ||d } | ||d }| ||}|||}| |||f}d| kr|kr| krnnd|kr|kr| ksnd |d | d | d}t|ndd| | f}|drt|||||d}|S|s3t tj tj |kr-tj |\}}|d|d|d|}ntj |}tj |d }tj| |||d} |dvrQtj|| |dt|}tdd|D}t}|D];}tdj||tj |d|dvrt|d}t||jd}Wdn 1swYtt|d|d}tt| |dd}t|d}n|d|d }tt|d|d ||}t|D]} |d ||| }!|d ||| d }"t|"|d}"|d|!|d |"f}#t|d|d|t|!|d |t|d |d|t|"|d |f}$td |$t||#d!d"}%|d#krMtj |f||d$|%d}&ntj |fi|%d}&t!|&|||d%}&|&j"dkrzd|&jd|$d |$d|$d|$d g}'n|$d |$d|$d|$d g}'|dvrtj#||&||'d!d&q|&||<qؐqdt$|%d krtj |d dvr| &d'd('d(krtd)| d'td*d(| d'<|dvrt$|%| d+<tj(||| |d,tj )|d-stj )|d.rt*| ||S)/a 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. outfile - str, path of output file max_memory - float, maximum used memory in GB search_win - list(int), ampcor (half) search window in (width, length) xcorr_win - list(int), ampcor cross-correlation window in (width, length) margin - int, ampcor margin off_file - str, path to the ampcor dense offsets file Returns: outfile - str, path of output file .vrtLENGTHWIDTHzmultilooking file: zmultilook method: z$number of looks in y / x direction: z / rrr zCalculated bbox (z%) exceeds input file coverage (width=z , length=)!)boxout_file_Zalks_Zrlks)r1)z.h5z.he5)metadataref_filecss|]}t|VqdS)N)r).0ir*r*r+ sz!multilook_file..z"multilooking {d:<{w}} from {f} ...)dwfrrNi@ rzbox: F) datasetNamer1 print_msgr )xstepystep)r%)r"r>blockr? 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)r2r4r5.xmlz.aux.xml)+rendswithr read_gdal_vrtread_attributeprintrmultilook_gdalosgetcwdpathdirnameabspathsplitextbasenameattrupdate_attribute4multilookr layout_hdf5get_dataset_listmaxdictformath5pyFilerprodrceilrintrangeminreadr,ndimwrite_hdf5_blockrkeysgetupperwriteisfilewrite_isce_xml)(infiler#r$outfiler% max_memoryZ search_winZ xcorr_winmarginZoff_fileatrlengthwidthx0y0out_widout_lenx1y1r1Zatr_offr&fbasefextdsNamesmaxDigitdsDictdsNamer;ds_sizenum_steprow_stepr7r0r1Zbox_iZbox_okwargsr"rBr*r*r+multilook_filews     ((      B        " ((         ' $ rc Cstdddlm}|s.|dr|dd}n|dr&|ddd }ntd |d td |td ||||j}|sMdd|j|jf}|d|d}|d|d}t ||} t ||} |d} |d} | | |} | | |}d| d| d}|d| d| d| d|d 7}|j ||||j d|||j}|j |d||j dddt j|dsddlm}||d|S)aqApply multilooking via gdal_translate. Parameters: in_file - str, path to the input GDAL VRT file lks_y/x - int, number of looks in Y/X direction box - tuple(int), area of interest in x0, y0, x1, y1 out_file - str, path to the output data file Returns: out_file - str, path to the output data file z)apply multilooking via gdal.Translate ...r)gdalz .rdr.full.vrtNiz.rdr.vrtz.mlizun-recognized ISCE VRT file (r0zinput : zoutput: r rrz-of ENVI -a_nodata 0 -outsize  z -srcwin )optionscallbackr-ZVRT)rY)rrF) gdal2isce_xml)rJosgeorrGrOpen GA_ReadOnly RasterXSize RasterYSizerZ TranslateZTermProgress_nocbZTranslateOptionsrLrNrhZisce.applications.gdal2isce_xmlr)in_filer#r$r1r2rdsZin_widZin_lenrsrtZsrc_x0Zsrc_y0Zsrc_x1Zsrc_y1Z options_strZdsorr*r*r+rKs<       "  rK)rrr)NrrNNrN)NN)loggingrLrrZrr mintpy.utilsrrSrr getLoggerZ np_loggersetLevelWARNINGr,rrKr*r*r*r+s   `