o gRec@sddlZddlZddlZddlmZedddlmZddl Z ddl Z ddl Z ddl ZddlmZddlZddlZddlmZddlmZmZmZmZddlmZdGdd Zd d Zd d ZddZddZ ddZ!ddZ"ddZ#ddZ$ddZ%ddZ&ddZ'd d!Z(d"d#Z)d$d%Z*d&d'Z+d(d)Z,d*d+Z-d,d-Z.dHd0d1Z/dId2d3Z0dId4d5Z1d6d7Z2d8d9Z3d:d;Z4dJd=d>Z5d?d@Z6dAdBZ7dCdDZ8dEdFZ9dS)KNagg) LightSource)sensor)ptimereadfile writefileutils) timeseriescCs>|D]}t|tr|dt|}q|dt|}q|S)z1 parse list array(list item or values) to string  ) isinstancelistseparate_string_by_spacestrstrip)arrdstkrV/home/exouser/operations/rsmas_insar/tools/MimtPy/mimtpy/utils/multitrack_utilities.pyrs  rcCs2tjtj|\}}tj|\}}|||fS)z; return directory(filepath), filename(shotname), extension )ospathsplitabspathsplitext)rfilepath tempfilenamefilename extensionrrrseparate_filename_extension"s rcCs&|jd|d|jd|j}|S)zset output directory/_)outdir startDateendDate)inpsZ modelsoftwaredirnamerrr set_outdir(s"r'cCsvtd}d|dgg}g}|D]"}t|D]}tjtj||r3t||dkr3||qqt |}|S)8find the project folder and sort the folder in [*AT *DT] SCRATCHDIRr r ) rgetenvjoinlistdirrisdirrfindappendsortedrdirfoldersZproject_folderfolderxZproject_folder_sortrrr find_folder-s & r7cCsg}d}d}t|D]!}tj|ddkr,t||dkr,t||dkr,||q g}|D] }t|t|kr=|}q1|S)Hfind timeseries***.h5 file. The best results is timeseries_***_demErr.h5r Residual.h5r*rr-rrrr/r0len)datadir datafileskey1key2filedatafilerrrfind_timeseries9s  rDcCs:g}t|D]}tj|ddkr||q|dS)zfind full name for S1 datatyper:.he5r)rr-rrr0r>r?rBrrrfind_HDFEOS_fullnameHs  rGcCZg}t|D]!}tj|ddkr(tj|ddddkr(||q|dS)z!find full name for S1**_*PSDS.he5r:rEr!r*ZPSDSrrr-rrbasenamerr/r0rFrrrfind_PSDS_HDFEOS_fullnameP 4 rKcCrH)zfind full name for S1**_*PS.he5r:rEr!r*ZPSrrIrFrrrfind_PS_HDFEOS_fullnameYrLrMc Cst|}ttd|dddg}|dkrtdtdtjdr/tdd}| }| d |dd }g}d }|D]2} t | |d krptt| d d t|d d |krptt| d d t|d d }| }q>|S)z$get the date close to the given datezinfo.pyz--date>z date_list.txtr:zerror when runing info.pyrrr r*)rGrsystemrprintexitrisfileopen readlinescloser,rr/absintr) r>daterCcompletion_statusflinesZ date_partdate2subdatesrrrfind_nearest_datebs&  ($rccCs>|dkr|}nt||}|dkr|}||fSt||}||fS)z,find the startdate and enddate of each trackNone)rc)r>r#r$Z startdate2Zenddate2rrrfind_start_end_datews  recCsg}g}|D]8}tdtdd|dgtdtdd|dg}t|}||d||dqtt |dkrKt dtt |dkrWt d d S) zWfor S1*h5 file, check whether the lat_step and lon_step are same for different projectsr r)r z/mintpy/X_STEPY_STEPr:z5error! lon_Step between different tracks is not same!z5error! lat_Step between different tracks is not same!T) rchdirr,r+rGrread_attributer0r=set Exception)r4x_stepy_stepprojectrCatrrrrcheck_X_Y_steps$" rpc std|ji}t|j}t|}|jdfdd|D}g}|D]}|dd}| |q(t||D]}||}|r|dkr\t t dd|dD||<q=|d kre|||<q=|d krzt t d d|dD||<q=|d vrt |||<q=|d vr|||<q=|dvr|||<q=|dvrt t dd|dD||<q=|dvr|||<q=|dvr|||<q=|S)z/Read input template options into Namespace inpsz&read input option from template file: .cs g|] }t|dkr|qS)r*)rr/.0iprefixrr s z&read_template2inps..r*ZDataSetcSg|]}|qSrrrrrrrrw,DataTypeSNWEcSsg|]}t|qSr)floatrrrrrrw)ZlatStepZlonStep)r#r$) mask_file)ref_lalocSrxrrrrrrrrwry)maskfile)outname) rT templateFiler read_templateutcheck_template_auto_valueZ ModelSoftwarekeysrr0r tupler}) r% inps_dicttemplateZ key_list_tmpkey_listrtZkey_wordkeyvaluerrurread_template2inpssD     " "  " rcCstd\}}t|j|jgd\}}|||jd9}dtjt |d}||9}d|d<d|d <d |d <d d |j|j}t j |||ddS)zbcalculated displacement during startDate_endDate period based on linear assumption and velocity.h5zgeo_velocity.h5rgv@g WAVELENGTHroipac PROCESSORz.unw FILE_TYPEradianUNITgeo_z {}_{}.unw)out_filemetadataN) rreadrdate_list2vectorr#r$daysnppir}formatrwrite)r%datarodt1dt2 range2phaserrrrvelocity_displacementsrcCsJt|D]}tj|d|kr"t||dkr"t|d|qdS)zSdelete all geo_*.h5 files in $MODLEDIR/project/SenAT(DT)/geodmod_startdate_enddate/r:r*r N)rr-rrrr/remove)r>r@rArBrrrdelete_temporalGeofiles rcsfddtt|dD}d}d}tt|D]=}ttD]4}|||krU|||d||d|d<||d|d|krU||d|d}|d}q!q|||||fS)Ncs&g|]}ddttdDqS)cSsg|]}dqSrrrrrrrrwryz5find_intersection_part...r:ranger=)rsjs2rrrws&z*find_intersection_part..r:rr)s1rmmmaxprtrrrrfind_intersection_parts rcCsld|dgg}g}|D]"}t|D]}tjtj||r.t||dkr.||qq t|}|S)r(r r r*) r,rr-rr.rr/r0r1r2rrrfind_folder_horzverts& rcCsg}d}d}d}t|D])}tj|ddkr6t||dkr6t||dkr6t||dkr6||q g}|D] }t|t|krG|}q;|S)r8r r9mskr:r;r*r<)r>r?r@rAZkey3rBrCrrrfind_timeseries_horzverts0 rc Csd}d}|tjd}|tjd}|dddfdk}|d|f|d}|d|ddd |dd d |d d |d|fd |ddd |ddd|d dtd|d|fd|dd d|d dtd|d|fd|d dtd |d|f}|d|ddd |dd d |d d |dd |ddd |ddd|d dtd|dd|dd d|d dtd|dd|d dtd |d}|td|dt|d|fd}|t|d|f} tjdt|ftd} |dt|d|ft| | d|f<|||dt|d|fdt| | d|f<|d|f|d}||| d|f<| | d|f<| d} | S)zConverts from longitude and latitude to local coorindates given an origin. llh (lon; lat; height) and origin should be in decimal degrees. Note that heights are ignored and that xy is in kmg@TXAg4n-$?r:Nr@rPrR -i#i )dtype) rrsinsqrtzerosr=r}tancos) ZllhoriginaezZdlambdaMM0NExyrrrllh2xys$,,8rc Cs|}|}d}|tjd9}|dkr|d7}|tjd9}t|}t|t|} t|t|} |||| || } | S)zyinc is 1*1 value, head is 1*1 value,north_disp/east_disp/up_disp is n*1 matrix positive value means north/east/upZgf@ggv@)rrrr) incheadZ north_dispZ east_dispZup_disp inc_angle head_angleaz_angleZA_upZA_northZA_eastZlos_simrrrcalculate_LOS_value+s r=Tc Csi}i}d}g}tj|rt|d}|}n t|tr"|d}|D]{} | } dd| |dD} t | dks@| dr`| d rJi}d }n|r]| d s]| d s]d}| |t q$| d } t | dddd d } tj| } tj| } |rt| r| || <q$d}| || || <q$| dkr| || <q$tj|r||r| |t |d krt||d<|S)a{Reads the template file into a python dictionary structure. Parameters: fname : str full path to the template file delimiter : str string to separate the key and value print_msg : bool print message or not Returns: template_dict : dict file content Examples: tmpl = read_template(KyushuT424F610_640AlosA.template) tmpl = read_template(R1_54014_ST5_L0_F898.000.pi, ':') from mintpy.defaults.auto_path import isceAutoPath tmpl = read_template(isceAutoPath, print_msg=False) FrO cSsg|]}|qSr)rrrrrrrwir~z!read_template..r:r)%#rNTrrrr plotAttributes)rrrVrWrXr rrrr= startswithr0nextreplace expanduser expandvarsZis_plot_attributerYjsondumps) fname delimiter print_msgZ template_dictZplotAttributeDictZinsidePlotObjectrr^r_linecZatrNameZatrValuerrrrGsP                rcCs |tjt|||}|S)av Function to calculate displacements/velocities due to slip on a deep screw dislocation (infinitely long strike slip fault. After Savage and Burford (1973). v = (s/pi)*arctan((x+xc)/d) INPUTS x = vector of distances from fault s = slip or slip rate on deep dislocation d = locking depth [same units as x] c = scalar offset in y [same unit as s] xc = offset to fault location [same units as x] OUTPUTS v = vector of displacements or velocities at locations defined by x [same units as s] USEAGE: v = deepdisloc(x, s, d) rrarctan)r6sdrxcvrrr screw_discsrcCs\|tjt|||||dtjt||||dkd|dkd}|S)a  Model for interseismic strain accumulation and fault creep. Taken from Hussain et al. (2016). INPUTS x = vector of distances from fault s1 = slip or slip rate on deep dislocation s2 = creep or creep rate d1 = locking depth d2 = depth to bottom of creeping section c = scalar offset in y [same unit as s] xc = offset of fault location OUTPUTS v = vector of displacements or velocities at locations defined by x [same units as s] r:rg?r)r6rrd1d2rrrrrr fault_creepsXrc CsF|t||d|d|d}tdtt||t|}|S)a@ INPUTS x = vector of distances from fault v = velocities at locations defined by x m = model parameters, [0] = slip (mm/yr), [1] = locking depth (km), [2] = scalar offset (mm/yr) W = weight matrix (inverse of the VCM) OUTPUTS ll = value of the loglikelihood function rr:rg{Gz)rrnansumdot transpose)r6rrWdiffllrrrloglikes $rcCs&tt||kt||k@}|S)z INPUTS m = model values m_min = lower model limits m_max = upper model limits OUTPUTS lp = true if within limits, false if any aren't )rall)rm_minm_maxlprrrlogpriors" rcCstt||d}|S)zs INPUTS a,b = two arrays of same length OUTPUTS rms = rms misfit between a and b (a-b) r)rrnanmean)rbrmsrrr rms_misfitsr'c Cst|D]}|}|dtjjddddd|d<|dtjjddddd|d<|dtjjddddd|d<t|ttsQ|d7}|t |ddf<qt t t |t }t t t |t }t||tjjddddkrz|}|}|d7}n|d7}|t |ddf<|t|<qt dddfdt dddf<t dddfdt dddf<t dddfdt dddf<t ttddf}dS) z Bayesian monte carlo inversion taken from the main notebook. Re-packaged as a function here to minimise code repeats. rg@r:)locscalesizerrN)lowhighr)rcopyrrandomnormallibrrrZ models_savedrr6rrexpuniformZll_saved nanargmax) n_iterationsiiZ m_currentZm_trialZn_rejectZ ll_currentZll_trialZn_acceptZ best_modelrrr run_inversions, """      rc) Cst||\}}t|}t|}t|d|dd|d|dd}td||dd} | dd| ddd} tdd|dd} ||} |d| | d} |d| | d}|dd| tj| }| |d}| |d}||d}||d}t|dd}t|}t |d|df|d|df|d|df|d|dfg}| t | | g}| |j}||}||}||}td|dD]b}t||||d||d||g}t||||d||d||g}t |d|df|d|df|d|df|d|dfg}| t ||g}||}t|||<qt|d|d|d|dg}t|d|d|d|dg}t||fj} t| t| d | d gf} t |d|df|d|df|d|df|d|dfg}| t ||g}||}!||}"||}#|d|d|d|d}$|d|#|d|"}%t|%|$d|$|%}&t|d|#d|d|"d}'|'t|&}(|| |!|(| fS) a Generates a profile through gridded data. INPUTS: data = numpy array of values to profile x = vector of coords for the x axis y = vector of coords for the y axis prof_start = (x, y) pair for the start of the profile line prof_end = (x, y) pair for the end of the profile line params = dictionary of parameters for the profiler (currently nbins and width) r:rrnbinsNr*widthrrrrr:)rmeshgridarrayrlinspacelinalgnorm zeros_likerPathcontains_pointsrflattenreshapeshaperrvstackTrr))r6yr prof_startprof_endparamsxxyyZ prof_distZprof_bin_edgesZ prof_bin_midsZbin_midsZbin_gradZx_midsZy_midsZ bin_grad_normZbin_x1Zbin_x2Zbin_y1Zbin_y2Zbin_valZbin_stdZ full_polyZ poly_pointsZ trim_dataZtrim_xxZtrim_yyrZpoly_xZpoly_ypolyZ in_poly_valsZ points_polyZ points_valZpoints_xZpoints_yprof_mZpoints_mZpoints_prof_angleZpoints2prof_startZ points_distrrr profile_data"sZ  .     B **B"" B &r+c Cst|jddD]}t||||ddf||dddfr$|}nq |||dddf}|d|d|d|d}|d|d|d|d}t||d||}t|||dddf|dddfg} t| td f} t| d| d} t| d| d } t| | \} }}t | |||g}t |d|dd|d|dd}|t |fS) aX Calculates the distance along a profile at which it intersects a fault, and the angle between the two at this point. INPUTS: prof_start = (x,y) coord prof_end = (x,y) coords fault_trace = x and y coords of fault trace OUTPUTS: intersection_distance intersection_angle rr:Nrr)r:r:r)r:r)rr:r) rr  intersectrrr!hstackonescrossrrrad2deg)r$r%Z fault_traceindZ inter_indZfault_inter_coordsr*Zfault_mZintersection_anglerhl1l2r6r#rZintersection_pointZintersection_distancerrrprofile_fault_intersectionzs"*  *.r5cCs@|d|d|d|d|d|d|d|dkS)Nr:rr)ABCrrrccws@r9cCs0t|||t|||kot|||t|||kS)N)r9)r6r7r8Drrrr,s0r,)r )rTr)r):rargparsestringmatplotlib.pyplotpyplotpltswitch_backendmatplotlib.colorsrshutilnumpyrre subprocessZsubpZmatplotlib.pathrmintpymintpy.workflowmintpy.objectsr mintpy.utilsrrrrrr rrr'r7rDrGrKrMrcrerprrrrrrrrrrrrrrrr+r5r9r,rrrrs\           (  ,  L  /X+