B Qaj9@sddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZe dZdZdZdd Zdd d Zd d ZddZddZddZddZdddZedkreejdddS)N)resample)get_template_content) arg_groupreadfile 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 Cs~tjdtjtdtd}|jdddd|jdd d d |jd d ddd|jdddd|jdddd|jdddddd|jdddd d|d!}|jd"d#d$td%d&d'd(|jd)d*d+td,d-d.t d(|d/}|jd0d1d2d3d3d4hd5d6|jd7d8tt j d9d:|jd;dd?d:|jd@dAdBdBdChdDd6|jdEdFdGdHd|jdIdJdKdLd|jdMdNdOdPdt |}|S)QNz;Resample radar coded files into geo coordinates, or reverse ) descriptionformatter_classepilogfile+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 --softwaresoftwareZ 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.)argparseArgumentParserRawTextHelpFormatterTEMPLATEEXAMPLE add_argumentadd_argument_groupfloatformat DEG2METERnpnanintradd_memory_argument)parseroutinterprDD/home/centos/operations/rsmas_insar/sources/MintPy/mintpy/geocode.py create_parser6sF        rFcCs0t}|j|d}|jr$t|j|}t|}|S)N)args)rF parse_argsrread_template2inps _check_inps)iargsrAinpsrDrDrEcmd_line_parseps   rMcCs.t|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|_|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 anythingz=ERROR: --lalo-step can NOT be used together with --geo2radar!zMERROR: --lalo-step can NOT be used with lookup table file in geo-coordinates!)ut get_file_listr Exceptionlenr1rget_lookup_filerFileNotFoundErrorrread_attributekeysrprintsysexitr"check_num_processorr+)rLatrrDrDrErJ|s<        rJcstd||st}t|}t|tdfddt|D}x|D]}|}|rX|dkr| dd dd d d }d d| D||<qX|d kr|||<qX|d krXd| krt j ||<qXt|||<qXWd}|kr|rt||_|S)z/Read input template options into Namespace inpsz&read input option from template file: zmintpy.geocode.cs g|]}|kr|qSrD)rV).0i)prefixtemplaterDrE sz&read_template2inps..)rr"[], cSsg|] }t|qSrD)r:)r\r]rDrDrEr`s)r%r*r>zmintpy.compute.maxMemory)rWrMvarsr read_templaterOcheck_template_auto_valuelistrVreplacesplitlowerr=r>r: maxMemory) template_filerLZ inps_dictkey_listkeyvaluerD)r^r_rErIs0        rIcCsN|s.dtjkrttd}nttd}tt|}td||S)aCheck 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 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: {})osenvironr?getenv cpu_countminrWr;)r+rDrDrErZs  rZcCst|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: {})rRrr1rdsetr;rspathbasenamer2isdirmakedirsrWjoin)infilerLr^r1rDrDrEauto_output_filenames rc CsNt}tdd|jD}t|j|j|j|j|j dd}|j r\|j r\|j |d<|j |d<t f|j |j||j|jd|}||x|jD]}tdd |tj||jd }t||}|jrtd tj|||j gd d krq|jr tj||d}ntj||d}tj !|ddk}|rRt"|} t#j$|||| dnt} tj%||jd } t&dd| D} xb| D]X} |st'|j(|j)f| | <xt*|j+D]}|j,|}|j-|}tddj| | |tj .|dtj/|| |ddd}|j0||d}|j1dkrFd|j2d|d|d|d|dg}n|d|d|d|dg}|rtd||t#j3||| |ddn(|| | |d|d|d|df<qW|stj4| | |j5d| | <qW|st6| 7|d<t#j8| |||d|j r|j rt#j9||d qWt:t|d!\}}td"|||S)#zgeocode all input filescSsg|]}tj|qSrD)rsrzgetsize)r\r]rDrDrEr`szrun_geocode..T)Z interp_method fill_valuer+ max_memoryr- print_msgZlat_fileZlon_file)Zlut_fileZsrc_filer lalo_stepz2--------------------------------------------------z resampling file: {}) datasetNamezupdate mode: ON)in_fileskip)res_objr,)z.h5z.he5)metadataref_file compressioncSsg|] }t|qSrD)rR)r\r]rDrDrEr`0sz+ reading {d:<{w}} in block {b} from {f} ...)dwbfF)rboxrr)Zsrc_dataZbox_indr#z"write data in block {} to file: {})datarblockr)dtypeBANDS)out_filerr)fname<z(time used: {:02.0f} mins {:02.1f} secs. );timer=argmaxrdictr%r*r+rmr-rrrrrr"openZpreparerWr;rrUryrr/rO run_or_skiprattrZupdate_attribute4radar2geoZupdate_attribute4geo2radarrsrzsplitextget_hdf5_compressionr layout_hdf5get_dataset_listmaxzeroslengthwidthrangenum_boxZ src_box_listZ dest_box_listr{readZ run_resamplendimshapewrite_hdf5_blockarrayrrRrVwritewrite_isce_xmldivmod)rL start_timeZind_maxkwargsrrr[r1Z file_is_hdf5rdsDictdsNamesmaxDigitdsNamer]Zsrc_boxZdest_boxrrmsrDrDrE run_geocodes             . rcCst|}t|dS)N)rMr)rKrLrDrDrEmainnsr__main__r,)N)N)rsrXrr3numpyr=Zmintpy.objects.resamplermintpy.defaults.templater mintpy.utilsrrrrrOrrr6r7r<rFrMrJrIrZrrr__name__argvrDrDrDrE s(     : 2"t