U c?;@sddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m ZmZedZdZdZddd Zdd d Zd d ZddZddZddZddZdddZedkreejdddS)N)resample)get_template_content) arg_utilsreadfile writefileutils attributegeocodeaexample: geocode.py velocity.h5 geocode.py velocity.h5 -b -0.5 -0.25 -91.3 -91.1 geocode.py velocity.h5 timeseries.h5 -t smallbaselineApp.cfg --outdir ./geo --update # geocode file using ISCE-2 lat/lon.rdr file geocode.py filt_fine.int --lat-file ../../geom_reference/lat.rdr --lon-file ../../geom_reference/lon.rdr # radar-code file in geo coordinates geocode.py swbdLat_S02_N01_Lon_W092_W090.wbd -l geometryRadar.h5 -o waterMask.rdr --geo2radar geocode.py geo_velocity.h5 --geo2radar z degrees --> meters on equator 0.000925926 --> 100 0.000833334 --> 90 0.000555556 --> 60 0.000462963 --> 50 0.000277778 --> 30 0.000185185 --> 20 0.000092593 --> 10 c Csd}tdt}tdd}tj|||||d}|jdddd |jd d d d |jddddd|jdddd|jdddd|jdddddd|jdd d!d"d|d#}|jd$d%d&td'd(d)d*|jd+d,d-td.d/d0 t d*|d1}|jd2d3d4d5d5d6hd7d8|jd9d:tt j d;d<|jd=d>d?t d@dAd<|jdBdCdDdDdEhdFd8|jdGdHdIdJd|jdKdLdMdNd|jdOdPdQdRdt|}|S)SNz=Resample radar-coded files into geo-coordinates or vice versa .)synopsis descriptionepilog subparsersfile+zFile(s) to be geocoded)nargshelpz-dz--dsetzdataset to be geocoded, for example: height for geometryRadar.h5 unwrapPhase-20100114_20101017 for ifgramStack.h5)rz-lz--lookup lookupFilez0Lookup table file generated by InSAR processors.)destrz --lat-filelatFilezlookup table file for latitude.z --lon-filelonFilez lookup table file for longitude.z --geo2radarz --geo2rdr radar2geo store_falsezcresample geocoded files into radar coordinates. ONLY for lookup table in radar-coord (ISCE, Doris).)ractionrz-tz --template templateFilez%Template file with geocoding options.zgrid in geo-coordinatesz-bz--bboxSNWE)SNWEzBounding box for the area of interest. using coordinates of the uppler left corner of the first pixel and the lower right corner of the last pixel for radar2geo, it's the output spatial extent for geo2radar, it's the input spatial extent)rtypermetavarrz--laloz --lalo-steplaloStep)ZLAT_STEPZLON_STEPz6output pixel size in degree in latitude / longitude.{} interpolationz-iz--interp interpMethodnearestlinearz7interpolation/resampling method (default: %(default)s).)rdefaultchoicesrz--fill fillValuez4Fill value for extrapolation (default: %(default)s).)rr#r+rz-nz--nprocsnprocsznumber of processors to be used for calculation (default: %(default)s). Note: Do not use more processes than available processor cores.z --softwaresoftware pyresamplescipyzhsoftware/module used for interpolation (default: %(default)s) Note: --bbox is not supported for -p scipyz--update updateMode store_truez?skip resampling if output file exists and newer than input filez-oz--outputoutfilez,output file name. Default: add prefix 'geo_'z--outdirz --output-dirout_dirzoutput directory.)TEMPLATEEXAMPLE__name__splitrcreate_argument_parser add_argumentadd_argument_groupfloatformat DEG2METERnpnanintadd_memory_argument)rr rnameparseroutinterprIE/home/exouser/operations/rsmas_insar/sources/MintPy/mintpy/geocode.py create_parser5sh         rKcCs0t}|j|d}|jr$t|j|}t|}|S)N)args)rK parse_argsrread_template2inps _check_inps)iargsrFinpsrIrIrJcmd_line_parseqs   rRcCsnt|j|_|jstdnt|jdkr2d|_|jsht|j|_|jsZ|jrZ|j|_|jsht dt |jd}d| kr|j rtdtdtdtdn$d| kr|j std td|jr|j std tdt |j}d| krtd tdt|j|_|j sj|jrLtd td|jd krjtdtd|S)NzERROR: no input file found!r/z2No lookup table found! Can not geocode without it.rY_FIRSTzinput file is already geocodedzIto resample geocoded files into radar coordinates, use --geo2radar optionzexit without doing anything.zGinput file is already in radar coordinates, exit without doing anythingzAERROR: "--lalo-step" can NOT be used together with "--geo2radar"!zOERROR: "--lalo-step" can NOT be used with lookup table file in geo-coordinates!zsz&read_template2inps..)rr%cSsg|] }t|qSrI)r>rdrIrIrJrjs,)r(r-rBzmintpy.compute.maxMemory)r\rRvarsr read_templaterTcheck_template_auto_valuelistr[r:lowerrArBr> maxMemory) template_filerQZ inps_dictkey_listkeyvaluerIrgrJrNs.      rNcCsN|s.dtjkrttd}nttd}tt|}td||S)a2Check number of processors Note by Yunjun, 2019-05-02: 1. conda install pyresample will install pykdtree and openmp, but it seems not working: geocode.py is getting slower with more processors Test on a TS HDF5 file in size of (241, 2267, 2390) Memory: up to 10GB Run time: 2.5 mins for nproc=1, 3 mins for nproc=4 2. macports seems to have minor speedup when more processors Thus, default number of processors is set to 1; although the capability of using multiple processors is written here. OMP_NUM_THREADSr&z"number of processor to be used: {})osenvironrCgetenv cpu_countminr\r?)r.rIrIrJr_s  r_cCst|jdkr|jr|jS|jr&d}nd}|jr@d||j}nd|tj|}|j rtj |j st |j t d|j tj |j |}|S)Nr/geo_Zrdr_z{}{}.h5z{}{}zcreate directory: {})rWrr5rdsetr?rwpathbasenamer6isdirmakedirsr\join)infilerQrhr5rIrIrJauto_output_filenames rc Cs<t}tdd|jD}t|j|j|j|j|j dd}|j r\|j r\|j |d<|j |d<t f|j |j||j|jd|}|||jD]x}tdd |tj||jd }t||}|jrtd tj|||j gd d krq|jrtj||d}ntj||d}tj !|ddk}|rNt"|} t#j$|||| dnt} tj%||jd } t&dd| D} | D]R} |st'|j(|j)f| | <t*|j+D]}|j,|}|j-|}tddj| | |tj .|dtj/|| |ddd}|j0||d}|j1dkr:d|j2d|d|d|d|dg}n|d|d|d|dg}|rtd||t#j3||| |ddn(|| | |d|d|d|df<q|sztj4| | |j5d| | <qz|st6| 7|d<t#j8| |||d|j r|j rt#j9||d qt:t|d!\}}td"|||S)#zgeocode all input filescSsg|]}tj|qSrI)rwr~getsizerdrIrIrJrj szrun_geocode..T) interp_method fill_valuer. max_memoryr0 print_msglat_filelon_file)lut_filesrc_filer lalo_stepz2--------------------------------------------------z resampling file: {}) datasetNamezupdate mode: ON)in_fileskip)res_objr/)z.h5z.he5)metadataref_file compressioncSsg|] }t|qSrI)rWrdrIrIrJrj>sz+ reading {d:<{w}} in block {b} from {f} ...)dwbfF)rboxrr)src_databox_indr&z"write data in block {} to file: {})datarblockr)dtypeBANDS)out_filerr)fname<z(time used: {:02.0f} mins {:02.1f} secs. );timerAargmaxrdictr(r-r.rqr0rrrrrr%openpreparer\r?rrZr}rr3rT run_or_skiprattrupdate_attribute4radar2geoupdate_attribute4geo2radarrwr~splitextget_hdf5_compressionr layout_hdf5get_dataset_listmaxzeroslengthwidthrangenum_box src_box_list dest_box_listrread run_resamplendimshapewrite_hdf5_blockarrayrrWr[writewrite_isce_xmldivmod)rQ start_timeZind_maxkwargsrrr`r5Z file_is_hdf5rdsDictdsNamesmaxDigitdsNamerfsrc_boxdest_boxrrmsrIrIrJ run_geocodes               rcCst|}t|dS)N)rRr)rPrQrIrIrJmain|sr__main__r/)N)N)N)rwr]rnumpyrAmintpy.objects.resamplermintpy.defaults.templater mintpy.utilsrrrrrTrrr7r8r@rKrRrOrNr_rrrr9argvrIrIrIrJ s&     < =!t