U xbD@sddlZddlZddlZddlZddlZddlmZddZ ddZ ddZ d d Z d d Z d dZddZd0ddZddZddZddZddZddZddZdd Zd1d!d"Zd2d#d$Zd%d&Zd'd(Zd3d,d-Zd.d/ZdS)4N) progressBarcCsRt|tr|dn|}t|}|dd}|dd}|dd}|dd }|S) zGet the "compact-looking" isoformat of the input datetime string. Parameters: date_str - str, an example date string Returns: iso_format - str, date string in "compact" iso format ryY%Y%m%d%Y-%m-%dz%H%M%Sz%H:%M:%Sz%H%Mz%H:%M) isinstancelistget_date_str_formatreplace)date_strZ iso_formatr I/home/exouser/operations/rsmas_insar/sources/MintPy/mintpy/utils/ptime.pyget_compact_isoformats    rcCs6t|tr|dn|}z|d}Wn YnXd}ttd|dkrPd}nttd|dkrjd}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}rz \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}rz\d{6}z%y%m%dz*un-recognized date string format for "{}"!)rrdecodelenrefindall ValueErrorformat)r date_str_formatr r r r !s4 r cCsdddg}d}|tjddd}|D]>}|D],}tt||dkr0t||d}q^q0|r(qhq(|s|td|d|S) zGet date12 str from a given file path. Parameters: file_path - str, path to a file that contains date1/2 info Returns: date12_str - str, date12 in (YY)YYMMDD(THHMM)[-_](YY)YYMMDD(THHMM) format z\d{8}T\d{4}[-_]\d{8}T\d{4}z\d{8}[-_]\d{8}z\d{6}[-_]\d{6}NrzNO date12 str found in path: !)splitosseprrrr) file_pathZ date12_fmtsZ date12_strpartspartZ date12_fmtr r r get_date12_from_pathTs 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)r"dt timedeltar )Z datetime_objZdatetime_obj_outr r r round_secondsus r%cCsZt|}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'] rN<iJZWINTERZSPRINGZSUMMERZFALL)yyyymmddr#datetimestrptime timetupletm_yday)r ydayZseasonr r r yyyymmdd2seasonsr/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 )daysn)r#r* fromordinalintr$)Zdatenumr r r datenum2datetimes  r4cCsfdd}t|ttjtjtfr(||}n:t|trPg}|D]}|||q:ntd t ||S)zread date in 2002.40657084 to datetime format Parameters: years : (list of) float or str for years Returns: years_dt : (list of) datetime.datetime objects cSsnt|}t|t}t||dtd}d||}ztj|d}Wnt d|YnX|S)Nv@r z {:d}-{:d}z%Y-%jzwrong format: ) floatnpfloorastyper3rr#r*r+r)xyearr.x2Zxtr r r decimal_year2datetime1s z5decimal_year2datetime..decimal_year2datetime1zAunrecognized input format: {}. Only float/str/list are supported.) rr6r7float32float64strrappendrrtype)yearsr=Zyears_dtr;r r r decimal_year2datetimes   rDcCst|tr|g}nt|}t|d}g}|D]n}tj||}|j|j dd|j d|j d|j d}|rd|kr|t |d7}||q.t|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 rr r5@ A~~AT)rr@rr r#r*r+r;r,r-hourminutesecondr6rA)datesr! date_list date_formatrCr drr r r yyyymmdd2yearss(     rPcCs"|ddkrd|}nd|}|S)z/Convert date str from YYMMDD to YYYYMMDD formatr91920r )dater r r yymmdd2yyyymmdds  rUcCs"|ddkrd|}nd|}|S)z'Convert year str from YY to YYYY formatrrQrRrSr )r;r r r yy2yyyys  rVcCszt|tr0t|dddkr*t|}qv|}nFt|trrg}|D],}t|dddkrdt|}||qBndS|S)zAConvert date str from (YY)YYMMDD(THHMM) to YYYYMMDD(THHMM) formatrHrN)rr@rrrUrrArLZdatesOutrTr r r r)s   r)cCsnt|tr*t|dkr$|dd}qj|}n@t|trfg}|D]&}t|dkrX|dd}||q(sz#yyyymmdd_date12..cSs"g|]}|ddddqSr]r^r r_r`r r r rc)scSsg|]\}}d||qS)z{}_{}rramsr r r rc*sr)rr@rr)zipZdate12_list_in date12_listZm_datesZs_datesZdate12_list_outr r r yyyymmdd_date12s  rlcCsht|tr|g}nt|}tdd|D}tdd|D}ddt||D}t|trd|d}|S)zConvert date12 into YYMMDD-YYMMDD format Parameters: date12_list_in - (list of) str Returns: date12_list_out - (list of) str in YYMMDD-YYMMDD format cSs"g|]}|ddddqSr\r_r`r r r rc?sz!yymmdd_date12..cSs"g|]}|ddddqSrdr_r`r r r rc@scSsg|]\}}d||qS)z{}-{}rerfr r r rcAsr)rr@rr[rirjr r r yymmdd_date123s  rmc CsBg}tj|r>t|d}|}W5QRXtt|}|S)zRead Date List from txt filer)rpathisfileopenread splitlinessortedr))Z date_filerMfr r r read_date_txtKs    rvcCs|sgSt|tr|g}g}|D]:}|drLtj|rZt|}||7}q |g}||7}q ttt 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) rr@endswithrrorprvrtr)rset intersection)Z date_list_in date_list_allZ date_list_outrOdsr r r read_date_listZs      r|c Cst|}g}|r2|tt||d7}td||rtd|tt|}t||D]0\}}||krZ||krZtd|||qZ|rtd|tt|} t||D]0\}}|| kr||krtd|||qttt|}|S)aGet exclude date list from input options (start/end/ex_date). Parameters: date_list - list of str, all dates in YYYYMMDD(THHMM) format start_date - str, starting date end_date - str, ending date exclude_date - list of str, exclude date in YYYYMMDD or text file Returns: ex_date_list - list of str, exclude date )rzzexclude date: zstart date: z remove date: zend date: ) rPr|rprintr)rirArtrx) rMZ start_dateZend_dateZ exclude_dateZ year_listZ ex_date_listZyear_minr;r Zyear_maxr r r get_exclude_date_listys*     r~c st|}tt|fdd|D}g}|D]*}||d}|j|jd}||q.i}t|D]\}}||||<qf||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 r#r*r+r`rNr r rcsz#date_list2tbase..riQ)r)r r@r0r!rA enumerate) rMrLtbaserTZ date_deltaZtbase_iZdateDictrbr r rr date_list2tbases   rcs|t|}tt|fdd|D}g}|D]D}|j|jdd|jd|jd|jd}| |q.||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 rr`rr r rcsz$date_list2vector..r r5rErFrG) r)r r@r;r,r-rIrJrKrA)rMrLZ datevectorrOZdate_vecr rr date_list2vectors  rr Drc 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)raobjout_fmtr r rcsz"get_date_range..) r r7rr#r*r+ isoformat timedelta64aranger9) ZdminZdmaxZdstepZdunitrrt1t2ZtstepZdt_objsZdt_listr rr get_date_ranges  rc 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)picossindir1imrZr g(\¥l@ga2U0*?gY+^?gmWel?g?x?g- &(?r&)minutes)mathrrrr;r,r-r6rIrJrKr#r*combinerTtimer$) Zutc_time longituderrrZyear_lengammaZeqtimeZ time_offsetZtstZ solar_timer r r utc2solar_times *, $r)r)N)NNN)r rr)rrrr*r#numpyr7Zmintpy.objects.progressrrr rr%r/r4rDrPrUrVr)r[rlrmrvr|r~rrrrr r r r  s4 3!   %    +