ZfLddlZddlZddlZddlmZmZmZm Z m Z dZ dZ dZdZdZej"fdZd d Zd Zy) N) attributeptimereadfileutils writefilectd}td}tfddDrOtd}td}td}td}|||zz}|||zz}||||f} nd } tfd d Dr:td td tdtdf} | | fSd } | | fS)zGet Coverage Box of data in geo and pixel coordinates. Parameters: atr - dict, meta data dictionary Returns: pix_box - 4-tuple of int, defining in (UL_X, UL_Y, LR_X, LR_Y) geo_box - 4-tuple of float in lat/lon LENGTHWIDTHc3BK|]}|jvywNkeys.0xatrs F/home/exouser/operations/rsmas_insar/tools/MintPy/src/mintpy/subset.py z#get_coverage_box..#s Oq1 ? Os)Y_STEPX_STEPY_FIRSTX_FIRSTrrrrNc3HK|]}d|jvyw)SUBSET_Nr rs rrz#get_coverage_box../s" Q1WQC=CHHJ & Qs")YMINXMINYMAXXMAX SUBSET_XMIN SUBSET_YMIN SUBSET_XMAX SUBSET_YMAX)intallfloat) rlengthwidthlat_steplon_stepul_latul_lonlr_latlr_longeo_boxpix_boxs ` rget_coverage_boxr0sX F G E O$N OOX'X's9~&s9~&(6/)(5.(6662 Q0P QQ M" # M" # M" # M" #   G  G cd\}}tj|}d}||jvr||jddjdd}|jdd}|j dDcgc]}|j }}t |dk(rd |D\}} } } | || | f}d }||jvr||jddjdd}|jdd}|j dDcgc]}|j }}t |dk(rd |D\} } }}|| || f}||fScc}wcc}w) zRead mintpy.subset.lalo/yx option from template file into box type. Parameters: template_file - str, path to the template file Returns pix/geo_box - tuple of 4 int or None )NNzmintpy.subset.yx[]:,c32K|]}t|ywr )r#ris rrz+read_subset_template2box..Rs;c!f;zmintpy.subset.laloc32K|]}t|ywr )r%r:s rrz+read_subset_template2box.._s%E1eAh%Er<)r read_templaterreplacesplitstriplen) template_filer/r.tmplkeyopt_strr; opt_str_listy0y1x0x1lat0lat1lon0lon1s rread_subset_template2boxrP<sd"GW  ! !- 0D C diiks)##C+33C;//#c*+2==+=>a > > |  !;l;NBB2r2&G C diiks)##C+33C;//#c*+2==+=>a > > |  !%E %E "D$dT4.G G %??s 9EEc|r|d|dg|_|d|dg|_nd|_d|_|r |d|dg|_|d|dg|_|Sd|_d|_|S)z9Update inps.subset_y/x/lat/lon from pixel_box and geo_boxrN) subset_lon subset_latsubset_xsubset_y)inpsr/r.s rsubset_box2inpsrZes"1:wqz2"1:wqz2 WQZ0  WQZ0  K  Kr1ct|d|d}t|d|d}t|d|d}t|d|d}||k\s||k\rd}|d|dz }|d|dz }t|||||f}|d|dz |d|dz |d|dz |d|dz f}|d|dz |d|dz |d|dz |d|dz f} || fS) a{Get index box overlap area of two input boxes. Parameters: box1/2 - 4-tuple of int, indicating coverage of box1/2 defining in (x0, y0, x1, y1) Returns: overlap_idx_box1/2 - 4-tuple of int, indicating index of overlap area in box1/2 defining in (idx_x0, idx_y0, idx_x1, idx_y1) rrSrRrTz(No overlap between two input box range! zbox 1:  zbox 2: )maxmin ValueError) box1box2rJrHrKrImsg overlap_boxoverlap_idx_box1overlap_idx_box2s rget_box_overlap_indexrfvsF T!Wd1g B T!Wd1g B T!Wd1g B T!Wd1g B Rx289 b!! b!!or2r"K$Aa0#Aa0#Aa0#Aa02 $Aa0#Aa0#Aa0#Aa02 - --r1cVtt|d}tt|d}tj|}d|j vrd}nd}|r|j ddr|j |ddd}nd|g}|j d dr|j d|d d }nd|g}t|}t|}|d|d|d |d f}|j|} || fS|j ddrF|d d|dd |d d |ddf} |j| d }|j| }|| fS|d r|d }nd|g}|d r|d }nd|g}t|}t|}|d|d|d |d f}|j|} || fS)aConvert subset inputs dict into box in radar and/or geo coord. Parameters: subset_dict - dict, including the following 4 objects: subset_x : list of 2 int, subset in x direction, default=None subset_y : list of 2 int, subset in y direction, default=None subset_lat : list of 2 float, subset in lat direction, default=None subset_lon : list of 2 float, subset in lon direction, default=None meta_dict - dict, including the following items: 'WIDTH' : int 'LENGTH': int 'X_FIRST' : float, optional 'Y_FIRST' : float, optional 'X_STEP' : float, optional 'Y_STEP' : float, optional Returns: pix_box - 4-tuple of int, in pixel unit of 1, in (x0, y0, x1, y1) geo_box - 4-tuple of float, in lat/lon unit (degree) None if file is in radar coordinate. Examples: subset_dict = {'subset_x': None, 'subset_y': None, 'subset_lat': [30.5, 31.0], 'subset_lon': [130.0, 131.0]} subset_dict = {'subset_x': [100, 1100], 'subset_y': [2050, 2550], 'subset_lat': None, 'subset_lon': None} pix_box = subset_input_dict2box(subset_dict, meta_dict)[0] pix_box, geo_box = subset_input_dict2box(subset_dict, meta_dict) r r rTFrVNrrUrS)bufrXrW) r#r%ut coordinatergetlalo2yxsorted box_pixel2geobbox_geo2radar) subset_dict meta_dictr'r&coord geocoded_flagsub_ysub_xr/r.s rsubset_input_dict2boxrvs!2 i() *E y*+ ,F MM) $EINN$$   ??< .MM+l";TB1EEKE ??< .MM$ L(AB1EEJEu u 8U1XuQxq:%%g.6 G / ??< ."<03[5Nq5QS^_kSlmnSoq|~JrKLMrNOG**7*:G**73G( G #:&#J/F :&#J/E 5ME5MEQxq58U1X>G))'2G G r1c .td|d|dtjj|dt j |||dd}|j }t|}|dk(rPtj|d |d z |d|dz f|j|z} || |d |d |d|df<| S|d k(rUtj|d|d |d z |d|dz f|j|z} || dd|d |d |d|df< S) Nzreading z in z from  ...F) datasetNamebox print_msgrrRrTrS) printospathbasenamerreadshaperBnponesdtype) fnamedsNamer/ pix_box4datapix_box4subset fill_valuedatads_shapeds_ndimdata_outs rsubset_datasetrs| HVHDfRWW5E5Ee5L4MT RS ==F PU VWX YDzzH(mG!|77GAJ3#AJ35::')349= ">!#44">!#445 6 O A77HQK#AJ3#AJ35 ::'*44 9= ">!#44">!#445 6 Or1c J tj|}t|d}t|d}td|dd|d|j }t ||\}}t j|} d} d|jvr|drd } nd} | s$| j|}tj|d<| j|}d d ||f} td | td |td | j| td||| k(r td|St| |\} } |stjtj j#tj j%|k(rdd|jvr/|dr*tj j'|\}}|d|}nBdtj j)|z}ntj j)|}td|zt+j,||}tj.|}tj j'|d}tj j'|d}|dvr|dvrt1j2|||n t5}|D]z}t7j8|d5}||}|j:}|j<}tdj?|| tj j)||dk(r|| d| d| d | df}tj@|d|dz |d|d z f|jB|dz}||| d| d| d | df<tjD||jB}|dvr:d t|dd t|dg}t1jF||||d n5||<n.|dk(r(|dvrtId|dtKjL|d  }tO|d D]}||| d| d| d | df}tj@d|d|dz |d|d z f|jB|dz}||d!d!| d| d| d | df<||dzd t|dd t|dg}t1jF||||d|jQ|dz|dzd"|d #|jStd$|d!d!d!}|dvrt1jT||%|St5}|D]}tW|||| | |d&||<tY|j|d'<t1jT||||(|dvrZtj j[|d)zs"tj j[|d*zrt1j\|||S#1swYcxYw)+aiSubset file with Parameters: fname - str, path/name of file subset_dict - dict, subsut parameter, including the following items: subset_x : list of 2 int, subset in x direction, default=None subset_y : list of 2 int, subset in y direction, default=None subset_lat : list of 2 float, subset in lat direction, default=None subset_lon : list of 2 float, subset in lon direction, default=None tight : bool, tight subset or not, for lookup table file, i.e. geomap*.trans fill_value : float, optional. filled value for area outside of data coverage. default=None None/not-existed to subset within data coverage only. out_file - str, path/name of output file Returns: out_file - str, path/name of output file default: add prefix 'sub_', if fname in the current directory; keep the same fname, if fname not in the current directory. r r zsubset FILE_TYPEz file: rxFrTrzdata range in (x0,y0,x1,y1): zsubset range in (x0,y0,x1,y1): zdata range in (W, N, E, S): zsubset range in (W, N, E, S): z7Subset range == data coverage, no need to subset. Skip.tight_tightsub_z writing >>> rS)z.h5z.he5)metadataref_filerz cropping {d} in {b} from {f} ...)dbfrRrT)r)rryblockr{zWriting 3D dataset z# into binary file is NOT supported!)maxValueN/)suffixzfinished writing to file: )out_filer)rBANDS)rrrz.xmlz.aux.xml)/rread_attributer#r|copyrvrirjrcheck_box_within_data_coveragernanrnrfr}getcwdr~dirnameabspathsplitextrattrupdate_attribute4subsetget_dataset_listr layout_hdf5dicth5pyFilerndimformatrrarraywrite_hdf5_blockr_r progressBarrangeupdateclosewriterrBisfilewrite_isce_xml)rsubset_dict_inputrrr'r&rpr/r.rroutfilldata_boxrrfbasefextdsNamesin_extout_extdsDictrfidsrrrrrprog_barr;s r subset_filer s$  ! !% (C G E X F GC $%WUG4 89#((*K,[#>GW MM# EG{''))k,.G 66w?$&FF L!!!'*G1eV$H +H: 67 +G9 56 *5+>+>x+H*I JK *7) 45( GH $97#K L.  99;"''//"''//%*@A A+**,,W1E gg..u5 t#WF4&1!"''"2"25"99ww''.H . !"  & &sG 4C''.G WW  e $Q 'Fggx(+G   o %  ! !(S5 IVFD CF5#&C C"Z88''8??"gg&&u-@/0 a<l1ol1o=*1ol1o=>?D "ww WQZ(?(/ WQZ(?(A'+zz 35@5N OHEI^A.~a/@@+A.~a/@@AB!xx CH/1!"CH $63s7|;L M!22$!)(."'&* ,*2v\o5(+>vhFi)jkk$00(1+FH"8A;/L!!".q/,q/"A".q/,q/"A#B C $&77A,3AJ,C,3AJ,C,E,0::$7:E\9R$S IM!!/!2>!3D!D!/!2>!3D!D"EF "#AaCCH ,>3s7|CT U!22$!)(."'&+ -!!qse1Xa[M4JK1L2NN$6xjABGC CC CD CN / ) OOFX D, O% F+&|4 F6N 6;;=)G C%P / )rww~~eFl/Krww~~^cdn^nOo  $ $S( 3 O{C CC Cs IXX" c td|j|j|j|jfDr|j rGt j|j }t|\}}td|j znd|jr2t|j\}}td|jzn&|jrtj|j|_|js t!dt j|j}tj"|}d|j%vrt j&|jdd}t)j*|d \}}t)j,||t)j.|j1k7\} } t)j2| d z t)j2| d z t)j4| d zt)j4| d zf}|j7|}~nt j&|jd d} t j&|jd d} t)j8| t)j:| t)j:| t)j8| f}d}~ ~ n t!d t=|||}|S)Nc3"K|]}|  ywr r:s rrz'read_aux_subset2inps..s -Qq5 -s zusing subset info from z>--dnn=G/8 GW +DNN: ;  78J8JK GW +D,>,>> ?ZZ!11$2B2BCD ## ^__--d.>.>?GMM'*EGLLN*!t'7'7WMaP$&IIfT$J! 6#%88Fi F@S@Z@Z@\6],]#^ 66'?R/21E66'?R/21EG--g6mmD$4$4*MaPmmD$4$4+NqQ99S>299S>99S>299S>;56 6tWg6 Kr1r )r}rnumpyr mintpy.utilsrrrrrrirr0rPrZrfrvrrrrrr1rrsV "J&R"".LPhUWTZTZ:qh0r1