U }-_ @sdddlZddlZddlmZddlZddlmZddlm Z dZ dde dfddZ Gd d d Z dS) N)datetime) coordinate)readfilez0/data/lxr/insarlab/GPS/CGPS/IndexForStations.txtTcCsBtj|tdddddt}|dddf}|ddddftjj\}}|t|d d 8}td d |ddd ftD} td d |dddftD} ||dk||dk||dk||dk} |rt |gdd} | | | k9} |r,t |gdd} | | | k9} || || || fS)aHUsed for search available GPS sites within the geo bounding box from CGPS database Parameters: SNWE : tuple of 4 float, indicating (South, North, West, East) in degrees start_date : string in YYYYMMDD format end_date : string in YYYYMMDD format site_list_file : string. min_num_solution : int, minimum number of solutions available Returns: site_names : 1D np.array of string for GPS station names site_lats : 1D np.array for lat site_lons : 1D np.array for lon ,)rrGBKdtype delimiterskiprowsusecolsencodingNrrgv@cSsg|]}t|dqSz%Y%m%ddtstrptime.0irI/home/centos/operations/rsmas_insar/sources/MimtPy/mimtpy/objects/cgps.py &szsearch_gps..r cSsg|]}t|dqSrrrrrrr'sr r) nploadtxtstrastypefloat32Troundarrayptimedate_list2vector)SNWE start_dateend_datesite_list_fileZ pring_msgtxt_data site_names site_lons site_latst_startt_endidxt0t1rrr search_gpss$ "&&    r4c@sZeZdZdZdddZdddZddd Zdd d ZdeedddZ de dddZ dS)CGPSaxGPS class for CGPS velocity Example: import matplotlib.pyplot as plt from mimtpy.objects.cgps import CGPS from mintpy.utils import utils as ut gps_obj = GPS(site='GV05', data_dir='~/insarlab/GPS') gps_obj.open() vel_los = ut.enu2los(gps_obj.vel_e, gps_obj.vel_n, gps_obj.vel_u) /data/lxr/insarlab/GPS/CGPS/cCs||_||_d|_d|_dS)Nz6GNSS data products of China earthquake adiministrationz'/data/lxr/insarlab/GPS/CGPS/GPS_vel.txt)sitedata_dirsourcefile)selfr7r8rrr__init__Esz CGPS.__init__TcCs&tj|jstd|j|ddS)N3GPS velocity dataset cannot be found. Please check! print_msg)ospathisfiler: Exception read_velocity)r;r?rrropenLsz CGPS.opencCsv|r tdtj|js"tdtj|jtddddd t}|ddd f}||j k}t |ddd f||_ t |ddd f||_ t |ddd f|d |_t |dddf|d |_t |dddf|d |_t |dddf|d |_|dddf| tjd d |_|dddf| tjd d |_|j |j |j|j|j|j|j|jfS)zread velocity datasetzRead velocity datasetr=rr) rrrrr r r  r r Nrrrr drFr r rGrH)printr@rArBr:rCrrrr r7floatsite_latsite_lonvel_estd_evel_nstd_nfloat64vel_ustd_u)r;r?datar,Z id_positionrrrrDQs, && zCGPS.read_velocityFc Cs|j}|j}t|trt|}t||d}|j|||ddd\}}|||d|df} tj|d| |ddd} tj|d | |ddd} t | } nLt|t rt j |d|d } t |d } | d d krt | } ntd|| | fS)N) lookup_filer>rrrincidenceAngle) datasetNameboxr?)rr azimuthAngle) dimensionr?HEADINGf@gF@z)input geom_obj is neight str nor dict: {})rLrM isinstancerrread_attributer geo2radarreadutazimuth2heading_angledictincidence_anglerK ValueErrorformat) r;geom_objr?latlonatrcoordyxrY inc_angleaz_angle head_anglerrrget_los_geometryjs"        zCGPS.get_los_geometryenu2los)rorqcCs|tjd9}|tjd9}t|t|dt|t|t|g}|}|dkrbn<|dkrtd|d<n*|dkrd|d<d|d <ntd t||j|d|j|d |j |d|_ |j |dd|j |d d|j |ddd |_|j |jfS) a)Convert displacement in ENU to LOS direction Parameters: inc_angle : float, local incidence angle in degree head_angle : float, satellite orbit heading direction in degree from the north, defined as positive in clock-wise direction gps_comp : string, GPS components used to convert to LOS direction Returns: dis_los : 1D np.array for displacement in LOS direction std_los : 1D np.array for displacement standard deviation in LOS direction r])rs)en2loshz2losgr)u2losup2losrrzUn-known input gps components:g?)rpisincoslowerrfrrNrPrSvel_losrOrQrTstd_los)r;rorqgps_compunit_vecrrrdisplacement_enu2loss8      zCGPS.displacement_enu2losrcCs"||\}}|j|||ddS)aRead GPS velocity in LOS direction Parameters: geom_obj : dict / str, metadata of InSAR file, or geometry file path gps_comp : string, GPS components used to convert to LOS direction Retruns: vel : 1D np.array of displacement in meters std : 1D np.array of displacement uncertainty in meters rN)rrr)r;rhrr?rorqrrrread_gps_los_velocityszCGPS.read_gps_los_velocityN)r6)T)T)F)rs)rsF) __name__ __module__ __qualname____doc__r<rErDrrrKrrrrrrrr58s    &r5) r@codecsrrnumpyrmintpy.objects.coordr mintpy.utilsrZ CGPS_datasetr4r5rrrr s   #