U Sæb·ã@s&ddlZddddddgZd d d „ZdS) éNÚlinearÚ linear_rangeÚlinear_azimuthÚ quadraticÚquadratic_rangeÚquadratic_azimuth瀄.ATc sâ|j}|dd…\}} || } t|ƒdkrXt |dd¡}| | d¡}tj|dd ¡} n| dd¡}t |¡ ¡} |dkrŽtj|| ftj d}|dk ¡} ~| t  | ¡9} |r¼| | d k9} ~ |rFt  | ¡|krFt t  t t  | ¡|¡¡ƒ} | dkrFtj|| ftjd}d|t | d ƒd| …t | d ƒd| …f<| | ¡9} ~t t d| ¡t d|¡¡\}}tj|tj d dd¡}tj|tj d dd¡}tj|jtj d}|d krÄt |||f¡}n°|d kròt |d |d |||||f¡}n‚|d kr t ||f¡}nh|dkr&t ||f¡}nN|dkrFt |d ||f¡}n.|dkrft |d ||f¡}ntd |¡ƒ‚t tjj|| dd…fdd|| dd…f¡}t ||¡}tj||jd}|dk r*t|dƒP}t|jjdƒD]8}d dd„|j|dd…fDƒ¡}| d |¡¡qæW5QRXˆr‚t‡fdd„dDƒƒr‚t ˆdƒt ˆdƒ}}|| |}|||dd…f8}|r”d||dk<||}t|ƒdkrÆt |dd¡}t |dd¡}| |¡}| |¡}||fS)aRemove ramp from input data matrix based on pixel marked by mask Ignore data with nan or zero value. Parameters: data : 2D / 3D np.ndarray, data to be derampped If 3D, it's in size of (num_date, length, width) mask_in : 2D np.ndarray, mask of pixels used for ramp estimation ramp_type : str, name of ramp to be estimated. metadata : dict, containing reference pixel info, REF_Y/X max_num_sample : float, max number of pixel sample, above which the uniform sampling is applied to reduce sample size coeff_file : str, path to the text file to save the estimated ramp coefficients ignore_zero_value : bool, ignore pixels with zero values, default is True Recommend: True for phase data and False for offset data Returns: data_out : 2D / 3D np.ndarray, data after deramping ramp : 2D / 3D np.ndarray, estimated ramp éþÿÿÿNéréÿÿÿÿ)Úaxisé)Údtypegérrrrrrzun-recognized ramp type: {}gV瞯Ò<)ÚrcondÚaz cSsg|]}d t|ƒ¡‘qS)z{:16.6e})ÚformatÚfloat)Ú.0Úc©rúJ/home/exouser/operations/rsmas_insar/sources/MintPy/mintpy/objects/ramp.pyÚ jszderamp..z{} c3s|]}|ˆ ¡kVqdS)N)Úkeys)rÚkey©ÚmetadatarrÚ nszderamp..)ÚREF_XÚREF_Yrr) ÚshapeÚlenÚnpÚmoveaxisÚreshapeÚmeanÚflattenÚarrayÚonesÚfloat32ÚisnanÚsumÚintÚceilÚsqrtÚzerosÚbool_ÚmeshgridÚarangeÚhstackÚ ValueErrorrÚdotÚlinalgÚpinvrÚopenÚrangeÚTÚjoinÚwriteÚall)ÚdataZmask_inZ ramp_typerZmax_num_sampleZ coeff_fileZignore_zero_valueZdshapeÚlengthÚwidthZ num_pixelÚdmeanÚmaskÚstepZ sample_flagZxxÚyyr(ÚGÚXÚrampÚfÚiZ coeff_strZref_yZref_xZref_idxZdata_outrrrÚderampsŠ      ÿ ÿÿ  ÿ  $    0   "    rJ)NrNrNT)Únumpyr"Z RAMP_LISTrJrrrrÚ sú ÿ