U xbM@sddlZddlZddlZddlmZddlmZmZm Z m Z ddl m Z m Z mZddlTddlTddlmZddlmZdd d ZdddZd ddZd!ddZd"ddZddZd#ddZdS)$N)map_coordinates)geometryDatasetNamesgeometry ifgramStack timeseries)ptimereadfile attribute)*) coordinate)resample./TFc s|s t}tj|}tj|d}dddg}t|ddrt|jddtfd d |Drd |d }|d |7}t |dj krdj krt d|nt t jtt j|t|}d|krdnd}|d}tj|dd|d} td} t| ddrPt| jdd| j krft d| d| nt t jtt j| tj|d} t| d|d} |dkr4| dk rt| jdd|dkrtdd}n,|d krtdd!}nd"|d#}t||D]&} | j krtd| d| qnt t jtt j| nt d$tj|d%} d} t| ddrt| jdd| j krt d| d| dj krdj krt d| nd} |r|rtj|n|}| rtj| n| } | rtj| n| } | rtj| n| } |rd&|}|d'|d(7}|d)|7}|| rZd*| nd+7}|d,| 7}|d-| 7}|d.7}t ||| | | fS)/aCheck the loaded input files, following two rules: 1. file existance 2. file attribute readability Parameters: work_dir - str, MintPy working directory print_msg - bool, print out message Returns: stack_file - str, path to the interferogram stack file geom_file - str, path to the geometry file lookup_file - str, path to the look up table file, for radar-coord dataset only. ion_file - str, path to the ionosphere stack file Example: work_dir = os.path.expandvars('./FernandinaSenDT128/mintpy') stack_file, geom_file, lookup_file = ut.check_loaded_dataset(work_dir)[:3] zinputs/ifgramStack.h5 unwrapPhase rangeOffset azimuthOffsetT)abspathF print_msgc3s|]}|jkVqdS)N) datasetNames).0xobjI/home/exouser/operations/rsmas_insar/sources/MintPy/mintpy/utils/utils.py 8sz'check_loaded_dataset..z$required dataset is missing in file z: z OR coherencez(WARNING: "coherence" is missing in file Y_FIRSTZGEOZRADAR PROCESSORinputsrz.h5rzrequired dataset "z" is missing in file zinputs/geometry*.h5)rrN)isceZdoris)gammaroipaczUnknown InSAR processor: z to locate look up table!z8Input data seems to be geocoded. Lookup file not needed.zinputs/ionStack.h5z0Loaded dataset are processed by InSAR software: z Loaded dataset are in z coordinatesz Interferogram Stack: z Ionosphere Stack: z Geometry File : z Lookup Table File : z3 --------------------------------------------------)osgetcwdpathrjoinZ is_file_existropenall ValueErrorrprintFileNotFoundErrorerrnoENOENTstrerrorrread_attributekeys capitalizerrZget_lookup_fileAttributeError Exceptionrelpath) Zwork_dirrr8Z stack_fileZdnamesmsgatr coord_type processor geom_filedname lookup_fileZion_filerrrcheck_loaded_datasets                  r@r!mmeanc  Cst|} t| |d} | ||dd\} }| rRtd||td| |d\}}|dk rz| ||dd\}}t| |||||||| dd \}}}|||fS) a Read time-series of one pixel with input lat/lon Parameters: lat/lon - float, latitude/longitude ts_file - string, filename of time-series HDF5 file lookup_file - string, filename of lookup table file ref_lat/lon - float, latitude/longitude of reference pixel zero_first - bool, shift the time-series so that it starts from zero win_size - int, windows size centered at point of interest unit - str, output displacement unit method - str, method to calculate the output displacement and its dispersity Returns: dates - 1D np.ndarray of datetime.datetime objects, i.e. datetime.datetime(2010, 10, 20, 0, 0) dis - 1D np.ndarray of float32, displacement dis_std - 1D np.ndarray of float32, displacement dispersity )r?rzinput lat / lon: {} / {}zcorresponding y / x: {} / {})NNNF)ref_yref_x zero_firstwin_sizeunitmethodr)rr3r Z geo2radarr.formatread_timeseries_yx)latlonts_filer?ref_latref_lonrFrGrHrIrr:coordyrrDrEdatesdisdis_stdrrrread_timeseries_lalos&   rVc Cst|} | jddt| jd} t| } | rBtd|||||d|df} t j || dd} d}|dkrt |d}||||||d||df}t j ||dd | j d }|d krtj|dd } tj|dd }n0|d kr tj|dd } t|}ntd ||dk rL|||d|df}| t j ||dd8} |r^| | d8} |dkrjnb|dkr| d9} |dkrdn|d}n8|dkr| d9} |dkrdn|d}ntd|| | |fS)aE Read time-series of one pixel with input y/x Parameters: y/x - int, row/column number of interest ts_file - string, filename of time-series HDF5 file ref_y/x - int, row/column number of reference pixel zero_first - bool, shift the time-series so that it starts from zero win_size - int, windows size centered at point of interest unit - str, output displacement unit method - str, method to calculate the output displacement and its dispersity Returns: dates - 1D np.ndarray of datetime.datetime objects, i.e. datetime.datetime(2010, 10, 20, 0, 0) dis - 1D np.ndarray of float32, displacement dis_std - 1D np.ndarray of float32, displacement dispersity Frrzinput y / x: {} / {}r!)boxNrCrB)axismedianzun-recognized method: {}rAcmgY@mmg@@zun-supported output unit: {})rr+rdate_list2vectordateListnparrayr.rJrreadintreshapenumDatenanmeannanstd nanmedianmedian_abs_deviationr-)rRrrNrDrErFrGrHrIrrrSrWrTrUbufZbox_winZdis_winZref_boxrrrrKsF    $       rKnearestcs|}|\}}|\}}t|dt|dtfddt||g||gDsd} | d|7} | d|7} | d7} t| tt||||} tj||| tj d } tj||| tj d } |d kr|t |  tjt |  tjf} nbd d d d}|| krBd|} | d| 7} t| ||}t |t| | f|d} d}d}d| krt|j||gdd\}}||d}t|dtjd|t|tjd}t|dtjd|}n8zt|}t|}Wn"tk r"d }d }d}YnXt| ||| ||}t| }|| dk9}i}| ||d<| ||d<| ||d <|||d!<||d"<|S)#a{Extract 2D matrix (z) value along the line [x0,y0;x1,y1] Link: http://stackoverflow.com/questions/7878398/how-to-extract-an-arbitrary-line-of-values-from-a-numpy-array Parameters: z : (np.array) 2D data matrix atr : (dict) attribute start_yx : (list) y,x coordinate of start point end_yx : (list) y,x coordinate of end point interpolation : str, sampling/interpolation method, including: 'nearest' - nearest neighbour 'linear' - linear spline interpolation (order of 1) 'cubic' - cubic spline interpolation (order of 3) 'quintic' - quintic spline interpolation (order of 5) Returns: transect: (dict) containing 1D matrix: 'X' - 1D np.array for X/column coordinates in float32 'Y' - 1D np.array for Y/row. coordinates in float32 'value' - 1D np.array for z value in float32 'distance' - 1D np.array for distance in float32 Example: transect = transect_yx(dem, demRsc, [10,15], [100,115]) LENGTHWIDTHc3sB|]:\}}d|kokno8d|ko4knVqdS)rNr)rijlengthwidthrrrsztransect_yx..z-input start/end point is out of data coveragez start_yx: {}z end_yx:{}z data size: ({}, {})dtyperjr!r"r%)linearZcubicZquinticz%un-supported interpolation method: {}z available methods: {})ordergTXArArrRr;g@X_STEPf@Y_STEPpixelgYXvaluedistanceZ distance_unit)lowerrbr,ziprJr-r_hypotlinspacefloat32rintastyper4rvstackr Zyx2lalofloatpicosrange_ground_resolutionazimuth_ground_resolutionKeyErrorisnan)zr:Zstart_yxZend_yx interpolationy0x0y1x1r9Znum_ptsysxsZz_lineZinterpolate_name2orderZ interp_order earth_radiusZ dist_unitlat0lat1lat_cx_stepy_stepZ dist_linemasktransectrror transect_yxsf& *   .         rc Cs`t|}|j|d|dgdd\}}|j|d|dgdd\}} t||||g|| g|} | S)zAExtract 2D matrix (z) value along the line [start_lalo, end_lalo]rrLrvr!rM)r Zlalo2yxr) rr: start_laloend_lalorrQrrrrrrrr transect_lalo]s rc Csi}d}g|d<tt|D]}||d||d}}d|krVt||||}nt||||}|d|7<|dkr|D]\} } tj| tjd|| <qn(|D]\} } t || | f|| <q|d ||dd}qtj|dtjd|d<|S)aExtract 2D matrix (z) value along multiple lines Parameters: z : 2D np.ndarray in size of (l,w) atr : dict, metadata of matrix z lines : list of lines with each line is defined as: [[lat0, lon0], [lat1, lon1]] for geo coordinates [[y0, x0], [y1, x1]] for radar coordinates Returns: transect : (dict) containing 1D matrix: 'X' - 1D np.array for X/column coordinates in float32 'Y' - 1D np.array for Y/row. coordinates in float32 'value' - 1D np.array for z value in float32 'distance' - 1D np.array for distance in float32 rstart_distancer!rrrrrX) rangelenr4rritemsr_r`r concatenateappend) rr:linesrrrmrrsegkeyr~rrrtransect_linesfs$  rradcCs|td|t|}td|tj|ddd}dt|krxtd|tdtj|ddd}t|}n&td tj|j tj d t |d }d | krHtd tdt ||d}|||jd}|j||d|d|d|dfd}|j||d|d|d|dfd}tj||d}|dds.t|dt|d}} t |d } t |d} t |d} t |d} | | |}| | | }t|d| | \}}t|d||\}}|||}||| }||d <||d<||d<||d<d|d<d|d<tj||dk<tj||dk<|d rr|tjd!9}|tjd!9}|||fS)"aPrepare LOS geometry data/info in geo-coordinates. Parameters: geom_file - str, path of geometry file unit - str, rad or deg, output angle unit Returns: inc_angle - 2D np.ndarray, incidence angle in radians / degrees head_angle - 2D np.ndarray, heading angle in radians / degrees atr - dict, metadata in geo-coordinate z5prepare LOS geometry in geo-coordinates from file: {}z!read incidenceAngle from file: {}incidenceAngle) datasetNamer azimuthAnglez!read azimuthAngle from file: {}z&convert azimuth angle to heading anglez3use the HEADING attribute as the mean heading anglerrHEADINGrz2--------------------------------------------------z,geocoding the incidence / heading angles ...)Zlut_fileZsrc_filer!r"rC)Zsrc_data)res_objY_UNITdegrkrlX_FIRSTrzrw OG_FILE_PATHdegreesX_UNITZrrx)r.rJrr3raget_dataset_listazimuth2heading_angler_onesshaperrr4r r+prepareZ src_box_listZ run_resampleattrZupdate_attribute4radar2geor startswithrb to_latlonnanr)r=rHr: inc_angleaz_angle head_anglerrWrprqNWrrSErlon0rlon1lat_steplon_steprrrprepare_geo_los_geometrysZ     ,,         r)r TF)NNNTr!rArBT)NNTr!rArBT)rj)rj)r)r'r0numpyr_Z scipy.ndimagermintpy.objectsrrrr mintpy.utilsrrr rZmintpy.utils.utils0Zmintpy.utils.utils1Zmintpy.objects.coordr Zmintpy.objects.resampler r@rVrKrrrrrrrr s6    u ' D _ -