B Qa_5@sddlZddlZddlZddlZddlZddlmZddZ ddZ ddZ d d Z d d Z d*d dZddZddZddZddZddZddZddZd+ddZdd Zd!d"Zd,d&d'Zd(d)ZdS)-N) progressBarcCs2t|tr|d}y|d}Wn YnXd}ttd|dkrLd}nttd|dkrfd}nttd|dkrd }nttd |dkrd }nttd |dkrd }nzttd|dkrd}n`ttd|dkrd}nFttd|dkrd}n*ttd|dkr d}ntd||S)a~Get the datetime string format as defined in: https://docs.python.org/3.7/library/datetime.html#strftime-and-strptime-behavior Parameters: date_str - str, date in one of the following formats: YYYYMMDDTHHMM YYYYMMDD YYMMDD Returns: date_str_format - str, datetime string format rutf8Nz#\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}z%Y-%m-%dT%H:%M:%Sz\d{4}-\d{2}-\d{2}T\d{2}:\d{2}z%Y-%m-%dT%H:%Mz\d{4}-\d{2}-\d{2}T\d{2}z %Y-%m-%dT%Hz\d{4}-\d{2}-\d{2}Tz%Y-%m-%dz \d{8}T\d{6}z %Y%m%dT%H%M%Sz \d{8}T\d{4}z %Y%m%dT%H%Mz \d{6}T\d{4}z %y%m%dT%H%Mz\d{8}z%Y%m%dz\d{6}z%y%m%dz*un-recognized date string format for "{}"!) isinstancelistdecodelenrefindall ValueErrorformat)date_strdate_str_formatrH/home/centos/operations/rsmas_insar/sources/MintPy/mintpy/utils/ptime.pyget_date_str_formats6 rcCs*|}|jdkr|tjdd7}|jddS)zRound datetime object to the nearest second. Link: https://stackoverflow.com/questions/47792242/rounding-time-off-to-the-nearest-second-python gA)secondsr) microsecond)rdt timedeltareplace)Z datetime_objZdatetime_obj_outrrr round_secondsFs rcCsZt|}tj|dj}d}|dks0|dkr6d}n |dkrDd}n|dkrRd }nd }|S) zDetermine the season of input date in YYYYMMDD format Parameters: date_str - str, date in YYYYMMDD format Returns: season - str, season in ['WINTER', 'SPRING', 'SUMMER', 'FALL'] z%Y%m%dN<iJZWINTERZSPRINGZSUMMERZFALL)yyyymmddrdatetimestrptime timetupletm_yday)r ydayZseasonrrryyyymmdd2seasonPsr!cCs,tjt|tj|ddtjddS)zConvert Matlab datenum into Python datetime. Parameters: datenum : Date in datenum format, i.e. 731763.5 Returns: datetime: Date in datetime.datetime format, datetime.datetime(2003, 7, 1, 12, 0) r)daysin)rr fromordinalintr)Zdatenumrrrdatenum2datetimegsr%cCsjdd}t|ttjtjtfr(||}n>t|trTg}x.|D]}|||q.decimal_year2datetime1zAunrecognized input format: {}. Only float/str/list are supported.) rr&r'float32float64strrappendr r type)yearsr-Zyears_dtr+rrrdecimal_year2datetimeqs    r4cCst|tr|g}nt|}t|d}g}xv|D]n}tj||}|j|j dd|j d|j d|j d}|rd|kr|t |d7}||q0Wt|tr|d}|S)aConvert date(s) string into float number in the unit of year Parameters: dates - (list of) str, date in YYYYMMDD format seconds - float or str, time of the day info in seconds Returns: years - (list of) float, years including the date and time info rrgv@g@g Ag~~AT)rr0rrrrrr+rrhourminutesecondr&r1)datesr date_list date_formatr3r dyrrryyyymmdd2yearss   *   r>cCs"|ddkrd|}nd|}|S)z/Convert date str from YYMMDD to YYYYMMDD formatr91920r)daterrryymmdd2yyyymmdds  rCcCs"|ddkrd|}nd|}|S)z'Convert year str from YY to YYYY formatrr?r@rAr)r+rrryy2yyyys  rDcCs~t|tr0t|dddkr*t|}qz|}nJt|trvg}x:|D],}t|dddkrft|}||qDWndS|S)zAConvert date str from (YY)YYMMDD(THHMM) to YYYYMMDD(THHMM) formatr5rN)rr0rsplitrCrr1)r9datesOutrBrrrrs    rcCsrt|tr*t|dkr$|dd}qn|}nDt|trjg}x4|D]&}t|dkrZ|dd}||q>WndS|S)z/Convert date str in (YY)YYMMDD to YYMMDD formatN)rr0rrr1)r9rGrBrrryymmdds      rJcCs<tdd|D}tdd|D}ddt||D}|S)z,Convert date12 into YYYYMMDD_YYYYMMDD formatcSs"g|]}|ddddqS)-_r)rrF).0irrr sz#yyyymmdd_date12..cSs"g|]}|ddddqS)rKrLr)rrF)rMrNrrrrOscSsg|]\}}d||qS)z{}_{})r )rMmsrrrrOs)rzip) date12_listm_datess_datesrrryyyymmdd_date12srVcCs<tdd|D}tdd|D}ddt||D}|S)z(Convert date12 into YYMMDD-YYMMDD formatcSs"g|]}|ddddqS)rKrLr)rrF)rMrNrrrrOsz!yymmdd_date12..cSs"g|]}|ddddqS)rKrLr)rrF)rMrNrrrrOscSsg|]\}}d||qS)z{}-{})r )rMrPrQrrrrOs)rJrR)rSrTrUrrr yymmdd_date12srWc CsBg}tj|r>t|d}|}WdQRXtt|}|S)zRead Date List from txt filerN)ospathisfileopenread splitlinessortedr)Z date_filer:frrr read_date_txts    racCs|sgSt|tr|g}g}xB|D]:}|drNtj|r\t|}||7}q"|g}||7}q"Wttt t |}|rtt t | |}|S)zRead Date List Parameters: date_list_in : list of str / text file date_list_all : list of str in YYYYMMDD format Returns: date_list_out : list of str in YYYYMMDD format )z.txtz.cfgz.dat) rr0endswithrYrZr[rar_rrset intersection)Z date_list_inZ date_list_allZ date_list_outr<dsrrrread_date_list s       rfc st|}tt|fdd|D}g}x2|D]*}||d}|j|jd}||q0Wi}x t|D]\}}||||<qlW||fS)akGet temporal Baseline in days with respect to the 1st date Parameters: date_list - list of string, date in YYYYMMDD or YYMMDD format Returns: tbase - list of int, temporal baseline in days dateDict - dict with key - string, date in YYYYMMDD format value - int, temporal baseline in days csg|]}tj|qSr)rrr)rMrN)r;rrrO7sz#date_list2tbase..riQ)rrr0r"rr1 enumerate) r:r9tbaserBZ date_deltaZtbase_iZdateDictrNr r)r;rdate_list2tbase-s   ricst|}tt|fdd|D}g}xL|D]D}|j|jdd|jd|jd|jd}| |q0W||fS)aKGet time in datetime format: datetime.datetime(2006, 5, 26, 0, 0) Parameters: date_list - list of string, date in YYYYMMDD or YYMMDD format Returns: dates - list of datetime.datetime objects, i.e. datetime.datetime(2010, 10, 20, 0, 0) datevector - list of float, years, i.e. 2010.8020547945205 csg|]}tj|qSr)rrr)rMrN)r;rrrOPsz$date_list2vector..rgv@g@g Ag~~A) rrr0r+rrr6r7r8r1)r:r9Z datevectorr<Zdate_vecr)r;rdate_list2vectorHs  * rjrD%Y%m%dc svt|}ttj||}ttj||}t||}tj||||dd d} fdd| D} | S)aMake a list of dates with one-day (or given days) interval for [dmin, dmax] Parameters: dmin - str in format supported by get_date_str_format() dmax - str in format supported by get_date_str_format() dstep - int, interval in number of dunit dunit - str, unit of interval, e.g. Y, M, W, D, h, m, s out_fmt - str, output datetime string format Returns: dt_list - list of str in YYYYMMDD format datetime64)dtypeOcsg|]}|qSr)strftime)rMobj)out_fmtrrrOpsz"get_date_range..) rr'rmrrr isoformat timedelta64aranger)) ZdminZdmaxZdstepZdunitrrr t1t2ZtstepZdt_objsZdt_listr)rrrget_date_range_s  rxc Csddlm}m}m}|jddkrD|jddkrD|jddkrDd}nd}d|||jd t|jd d }d d d||d||d|d|d|d|}|d|}|jd|j |j d|} t j |t dt j| d} | S)aConvert UTC time to solar local time. Solar time: https://en.wikipedia.org/wiki/Solar_time Link: https://stackoverflow.com/questions/13314626 Parameters: utc_time - datetime.datetime object for the UTC time longitude - float, longitude of the observer in degrees Returns: solar_time - datetime.datetime object for the local solar time Example: utc_time = dt.datetime(2015, 2, 9, 3, 18, 48) solar_time = ptime.utc2solar_time(utc_time, 130.7) r)picossindiinimrIr g(\¥l@ga2U0*?gY+^?gmWel?g?x?g- &(?r)Zminutes)mathryrzr{r+rrr&r6r7r8rrcombinerBtimer) Zutc_time longituderyrzr{Zyear_lengammaZeqtimeZ time_offsetZtstZ solar_timerrrutc2solar_timeus *,> $r)r)N)rrkrl)rYrrrrnumpyr'Zmintpy.objects.progressrrrr!r%r4r>rCrDrrJrVrWrarfrirjrxrrrrr s. 4   %