o ldH@sddlZddlZddlZddlmZmZmZm Z m Z ddZ ddZ ddZd d Zd d Zejfd dZdddZddZdS)N) attributeptimereadfileutils writefilec std}td}tfdddDrBtd}td}td}td }|||}|||}||||f} nd } tfd dd Dritd tdtdtdf} | | 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 LENGTHWIDTHc3s|] }|vVqdSNkeys.0xatrF/home/exouser/operations/rsmas_insar/tools/MintPy/src/mintpy/subset.py #sz#get_coverage_box..)Y_STEPX_STEPY_FIRSTX_FIRSTrrrrNc3s"|] }d|vVqdS)ZSUBSET_Nr r rrrr/s )YMINXMINYMAXXMAXZ SUBSET_XMINZ SUBSET_YMINZ SUBSET_XMAXZ SUBSET_YMAX)intallfloat) rlengthwidthZlat_stepZlon_stepZul_latZul_lonZlr_latZlr_longeo_boxpix_boxrrrget_coverage_boxs(             r#cCsd\}}t|}d}||vrD||dddd}|dd}dd |dD}t|d krDd d |D\}}} } | || |f}d }||vr||dddd}|dd}dd |dD}t|d krdd |D\} } } }| | || f}||fS)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[]:,cSg|]}|qSrstripr irrr Pz,read_subset_template2box..cs|]}t|VqdSr )rr,rrrrRz+read_subset_template2box..zmintpy.subset.lalocSr)rr*r,rrrr.]r/csr1r )rr,rrrr_r2)rZ read_templater replacesplitlen) template_filer"r!ZtmplkeyZopt_strZ opt_str_listy0y1x0x1Zlat0Zlat1Zlon0Zlon1rrrread_subset_template2box<s&         r<cCsr|r|d|dg|_|d|dg|_nd|_d|_|r1|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!rrrsubset_box2inpsesrEc Cst|d|d}t|d|d}t|d|d}t|d|d}||ks,||krBd}|d|d7}|d|d7}t|||||f}|d|d|d|d|d|d|d|df}|d|d|d|d|d|d|d|df} || 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) rr>r=r?z(No overlap between two input box range! zbox 1:  zbox 2: )maxmin ValueError) Zbox1Zbox2r:r8r;r9msgZ overlap_boxZoverlap_idx_box1Zoverlap_idx_box2rrrget_box_overlap_indexvs*  rKc Cstt|d}tt|d}t|}|ddr%|j|ddd}n |dr.|d}nd|g}|d drB|j|d d d}n |d rK|d }nd|g}t|}t|}|d|d|d |d f}||}||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) rrrANlatitude)Z coord_typerCrr@ longituderBr>)rrut coordinategetZlalo2yxsorted box_pixel2geo) subset_dictZ meta_dictr rcoordZsub_yZsub_xr"r!rrrsubset_input_dict2boxs$      rUc Cstd|d|dtj|dtj|||ddd}|j}t|}|dkrSt |d |d |d|df|j |} || |d |d |d|df<| S|d krt |d|d |d |d|df|j |} || dd|d |d |d|df<| S) Nzreading z in z from  ...F) datasetNamebox print_msgrr=r?r>) printospathbasenamerreadshaper5nponesdtype) fnamedsNamer" pix_box4datapix_box4subset fill_valuedatads_shapeds_ndimdata_outrrrsubset_datasets@$  rlc Cst|}t|d}t|d}td|dd|d|}t||\}}t|} d} d|vr=|dr=d } nd} | sK| |}t j |d<| |}d d ||f} td | td |td | | td||| krtd|St | |\} } |sttjtj|krd|vr|drtj|\}}|d|}ndtj|}ntj|}td|t||}t|}tj|d}tj|d}|dvrF|dvrtj|||dnt}|D]>}t|d,}||}|j}|j}tdj|| tj|d|dkr|| d| d| d | df}t |d|d|d|d f|j!|d}||| d| d| d | df<t j"||j!d}|dvrd t|dd t|dg}tj#||||d dn|||<n|dkr&|dvrt$d|dt%j&|d d }t'|d D]s}||| d| d| d | df}t d|d|d|d|d f|j!|d}||d!d!| d| d| d | df<||dd t|dd t|dg}tj#||||dd|j(|d|dd"|d d#q|)td$|Wd!n 1s1wYq|dvrDtj*|||d%|St}|D]}t+|||| | |dd&||<qKt,||d'<tj*||||d(|dvrtj-|d)stj-|d*rt.|||S)+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. rrzsubset FILE_TYPEz file: rVFrgTrzdata 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_tightZsub_z writing >>> r>)z.h5z.he5)metadataref_filerz cropping {d} in {b} from {f} ...)dbfr=r?)rb)rhrWblockrYzWriting 3D dataset z# into binary file is NOT supported!)ZmaxValueN/)suffixzfinished writing to file: )out_filerp)rgZBANDS)ryrprqz.xmlz.aux.xml)/rread_attributerrZcopyrUrNrOr check_box_within_data_coverager`nanrRrKr[getcwdr\dirnameabspathsplitextr]attrZupdate_attribute4subsetZget_dataset_listrZ layout_hdf5dicth5pyZFiler_ndimformatrarbarrayZwrite_hdf5_blockrIrZ progressBarrangeupdateclosewriterlr5isfileZwrite_isce_xml)rcZsubset_dict_inputryrr rrSr"r!rTZoutfilldata_boxrerffbasefextZdsNamesZin_extZout_extZdsDictrdfidsrirjrhrkrvZprog_barr-rrr subset_files                    $( F . rc Cstdd|j|j|j|jfDr|jr(t|j}t|\}}t d|jn|j r:t |j \}}t d|j n|j rt |j|_|jsKtdt|j}t |}d|vrtj|jddd}tj|d d \}}t||t|k\} } t| d t| d t| d t| d f}||}~n/tj|jd dd} tj|jd dd} t| t| t| t| f}d}~ ~ ntdt|||}|S)Ncss|]}| VqdSr rr,rrrrsz'read_aux_subset2inps..zusing subset info from zs %)&8  5