U cY@sddlZddlZddlZddlZddlmZmZmZm Z m Z ddl mZdZdZdddZddd Zd d Zd d ZddZddZddZejfddZd ddZddZd!ddZedkreejdddS)"N)ptimereadfile writefileutils attribute)create_argument_parsera template ## if both yx and lalo are specified, use lalo option unless a) no lookup file AND b) dataset is in radar coord mintpy.subset.yx = auto #[1800:2000,700:800 / no], auto for no mintpy.subset.lalo = auto #[31.5:32.5,130.5:131.0 / no], auto for no aexample: subset.py inputs/ifgramStack.h5 -y 400 1500 -x 200 600 subset.py geo_velocity.h5 -l 30.5 30.8 -L 130.3 130.9 subset.py 030405_090801.unw -t SinabungT495F50AlosA.template # subset to the same coverage as the reference file subset.py geo_incidence.h5 -r subset_geo_velocity.h5 # multiple files input subset.py *velocity*.h5 timeseries*.h5 -y 400 1500 -x 200 600 # crop to larger area with custom fill value subset.py geo_velocity.h5 -l 32.2 33.5 --outfill-nan subset.py Mask.h5 -x 500 3500 --outfill 0 # "tight" subset for geocoded lookup table larger than data file subset.py geomap_4rlks.trans --tight c Csd}tdt}tdd}t|||||d}|jdddd |jd d d d tddd|jddddtddd|jdddddtddd|jddddd tdd!d|jd"d#d$d%d&|jd'd(d)d*|jd+d,d-d.|jd/d0td1d2|jd3d4d5d6d&|d7d8}|jd9d:d;d&|S)|sz#get_coverage_box..)Y_STEPX_STEPY_FIRSTX_FIRSTrBrCrDrENc3s |]}d|kVqdS)ZSUBSET_Nr:r<r?r+r,rAs)YMINXMINYMAXXMAX SUBSET_XMIN SUBSET_YMIN SUBSET_XMAX SUBSET_YMAX)r&allr') r@lengthwidthlat_steplon_stepZul_latZul_lonZlr_latZlr_longeo_boxpix_boxr+r?r,get_coverage_boxps&            rUc Cst|}zhdd|ddD}tdd|ddD\}}tdd|d dD\}}||||f}Wnd }YnXzhd d|d dD}td d|ddD\}} tdd|d dD\} } | || | f} Wnd } YnX| |fS)ziRead mintpy.subset.lalo/yx option from template file into box type Return None if not specified. cSs$g|]}|ddddqS[]stripreplacer=ir+r+r, sz,read_subset_template2box..zmintpy.subset.lalo,cSsg|]}t|qSr+r'r[r]r+r+r,r_sr:cSsg|]}t|qSr+rar]r+r+r,r_sr/NcSs$g|]}|ddddqSrVrZr]r+r+r,r_szmintpy.subset.yxcSsg|]}t|qSr+r&r[r]r+r+r,r_scSsg|]}t|qSr+rcr]r+r+r,r_s)r read_templater$sorted) rtmploptslat0lat1lon0lon1rSy0y1x0x1rTr+r+r,read_subset_template2boxs        rpcCsp|r*|d|dg|_|d|dg|_n d|_d|_|r`|d|dg|_|d|dg|_n d|_d|_|S)z9Update inps.subset_y/x/lat/lon from pixel_box and geo_boxrrr/N)rrrr)r5rTrSr+r+r,subset_box2inpssrrc Cs t|d|d}t|d|d}t|d|d}t|d|d}||ksX||krd}|d|7}|d|7}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 Inputs: box1/2 : 4-tuple of int, indicating coverage of box1/2 defining in (x0, y0, x1, y1) Outputs: 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/rrqz(No overlap between two input box range! z box 1: {} z box 2: {} )maxminformat ValueError) Zbox1Zbox2rnrlrormmsgZ overlap_boxZoverlap_idx_box1Zoverlap_idx_box2r+r+r,get_box_overlap_indexs* rxc Cstt|d}tt|d}t|}|ddrJ|j|ddd}n|dr\|d}nd|g}|d dr|j|d d d}n|d r|d }nd|g}t|}t|}|d|d|d |d f}||}||fS) aOConvert subset inputs dict into box in radar and/or geo coord. Inputs: 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 Outputs: # box defined by 4-tuple of number, defining (left, upper, right, lower) coordinate, # (UL_X, UL_Y, LR_X, LR_Y ) pixel_box : 4-tuple of int, in pixel unit - 1 geo_box : 4-tuple of float, in lat/lon unit - degree None if file is in radar coordinate. example: 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} pixel_box = subset_input_dict2box(subset_dict, meta_dict)[0] pixel_box, geo_box = subset_input_dict2box(subset_dict, meta_dict) r8r7rNlatitude) coord_typerrr longituderr/)r&r'r1 coordinategetlalo2yxre box_pixel2geo) subset_dict meta_dictrPrOcoordsub_ysub_x pixel_boxrSr+r+r,subset_input_dict2boxs$      rc Cstd|d|dtj|dtj|||ddd}|j}t|}|dkrt |d |d |d|df|j |} || |d |d |d|df<nh|d kr t |d|d |d |d|df|j |} || dd|d |d |d|df<| S) Nzreading z in z from  ...F) datasetNamebox print_msgrrrqr/) printospathbasenamerreadshaper3nponesdtype) fnamedsNamerT pix_box4datapix_box4subsetrdatads_shapeds_ndimdata_outr+r+r,subset_dataset%sF$  rc Cst|}t|d}t|d}|d}td|d|d|}t||\}} t|} d} d|kr|drd } nd} | s| |}t j |d<| |} d d ||f} td | td |td | | td | || krtd|St| |\} }|sttjtj|krd|krv|drvd tj|d tj|d}ndtj|}n tj|}td|t||}t|}tj|d}|dkr tj|||d|D]6}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}d t|dd t|dg}tj"||||d d|dkrt#j$|d d}t%|d D]}||| d| d| d | df}t d|d|d|d|d f|j |d}||dd|d|d|d |df<||dd t|dd t|dg}tj"||||dd|j&|dd |d|d d!q|'td" |W5QRXqnt(}|D]"}t)|||| ||dd#||<q*t*||d$<tj+||||d%tj,|d&stj,|d'rt-|||S)(a?Subset 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 fill_value : float, optional. filled value for area outside of data coverage. default=None None/not-existed to subset within data coverage only. tight : bool, tight subset or not, for lookup table file, i.e. geomap*.trans out_file : str, path/name of output file Outputs: out_file : str, path/name of output file out_file = 'sub_'+fname, if fname is in current directory; out_file = fname, if fname is not in the current directory. r8r7 FILE_TYPEzsubset z file: rFrTrz!data range in (x0,y0,x1,y1): {}z!subset range in (x0,y0,x1,y1): {}z data range in (W, N, E, S): {}z subset range in (W, N, E, S): {}z7Subset range == data coverage, no need to subset. Skip.tightz {}_tight{}r/Zsub_z writing >>> )z.h5z.he5)metadataref_filerz cropping {d} in {b} from {f} ...)dbfrrq)r)rrblockr)maxValueNz{}/{})suffixzfinished writing to file: {})rBANDS)out_filerrz.xmlz.aux.xml).rread_attributer&rcopyrr1r|r;check_box_within_data_coveragernanrrurxrgetcwdrdirnameabspathsplitextrattrupdate_attribute4subsetget_dataset_listr layout_hdf5h5pyFilerndimrrarraywrite_hdf5_blockr progressBarrangeupdateclosedictrr3writeisfilewrite_isce_xml)rZsubset_dict_inputrr@rPrOkrrTrSrZoutfilldata_boxrrdsNamesextrfidsrrrrrprog_barr^dsDictr+r+r, subset_fileBs                  $&$ rc Cstdd|j|j|j|jfDr|jrTt|j}t|\}}t d|jnT|j rzt |j \}}t d|j n.|j rt |j|_|jstdt|j}t |}d|krJtj|jddd}tj|d d \}}t||t|k\} } t| d t| d t| d t| d f}||}~nTtj|jd dd} tj|jd dd} t| t| t| t| f}d}~ ~ ntdt|||}|S)Ncss|] }| VqdSr9r+r]r+r+r,rAsz'read_aux_subset2inps..zusing subset info from z s*  1 %(<  2