ZfJdZddlZddlZddlZddlZddlmZdZ dZ dZ dZ dZ d Zd Zdd Zd Zd ZdZdZdZdZdZdZddZddZddZddZddZdZy)z#Utilities for date/time operations.N) progressBarct|tr|dn|}t|}|jdd}|jdd}|jdd}|jdd }|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_str iso_formats K/home/exouser/operations/rsmas_insar/tools/MintPy/src/mintpy/utils/ptime.pyget_compact_isoformatrsp )48x{hH$X.J##C-J##Hj9J##Hj9J##Fg6J c~t|tr|dn|} |jd}d}tt j d|dkDrd}|Stt j d|dkDrd}|Stt j d|dkDrd }|Stt j d |dkDrd }|Stt j d |dkDrd }|Stt j d|dkDrd}|Stt j d|dkDrd}|Stt j d|dkDrd}|Stt j d|dkDrd}|Stt j d|dkDrd}|St d|d#YxYw)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}:\d{6}z %Y%m%d:%H%M%Sz \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 "z"!)r r decodelenrefindall ValueError)r date_str_formats rr r "s)48x{hH ??6*O 2:: B'4 1 RZZ,h 7 81 <$. + RZZ 1 2Q 6)( % RZZ 1 2Q 6)"  RZZ 1 2Q 6'  RZZ 1 2Q 6'  RZZ( + ,q 0"  RZZ( + ,q 0" A(2NOOG  s F77F<cgd}d}|jtjddd}|D]J}|D]?}tt j ||dkDs&t j ||d}n|sJn|st d|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_path date12_fmts date12_strpartspart date12_fmts rget_date12_from_pathr&XsKJ OOBFF #DbD )E% J2::j$/014ZZ D9!<     8 1EFF rcz|}|jdk\r|tjdz }|jdS)zRound datetime object to the nearest second. Link: https://stackoverflow.com/questions/47792242/rounding-time-off-to-the-nearest-second-python gAsecondsr) microsecond)r+dt timedeltar ) datetime_objdatetime_obj_outs r round_secondsr0ys@$##s*BLL33  # # # 22rct|}tjj|dj j }d}|dks|dkDrd}|S|dkrd}|S|dkrd }|Sd }|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<iJWINTERSPRINGSUMMERFALL)yyyymmddr,datetimestrptime timetupletm_yday)r ydayseasons ryyyymmdd2seasonr@s!H ;;  ( 3 = = ? G GDF byD3J M  M  M Mrctjjt|tj|dzztjdz S)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-)datenums rdatenum2datetimerHsE ;; " "3w< 0 \\w{ + , \\s # $$rcd}t|ttjtjt fr ||}|St|t r"g}|D]}|j|||Stdt|d)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 cDt|}tj|jt}tj||z dzjtdz}|dd|d} t j j|d}|S#td|xYw)Nv@r(d-z%Y-%jzwrong format: ) floatnpfloorastyperFr,r:r;r)xyearr>x2xts rdecimal_year2datetime1z5decimal_year2datetime..decimal_year2datetime1s !Hxx{!!#&xxTV+,33C81<Qxqa ! 1%%b'2B  1-a0 0s / BBzunrecognized input format: z$. Only float/str/list are supported.) r rNrOfloat32float64strr appendrtype)yearsrVyears_dtrSs rdecimal_year2datetimer^s %%RZZ=>)%0 O E4  :D OO248 9 : O6tE{mCghiirct|tr|g}n t|}t|d}g}|D]}tj j ||}|j|jjdz dz z|jdz z|jdz z|jdz z}|rd|vr|t|dz z }|j|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(rK@ A~~AT)r rYr r r,r:r;rSr<r=hourminutesecondrNrZ)datesr* date_list date_formatr\r rLrs ryyyymmdd2yearsrjs%G K %il3K E  KK ; 7 VVq{{},,q0F: : VV{ #$ XX) *+XX. /0 s+- w#89 9A Q %a Lrc.|ddk(rd|z}|Sd|z}|S)z/Convert date str from YYMMDD to YYYYMMDD formatr91920)dates ryymmdd2yyyymmddrq- Aw#~Dy KDy Krct|tr]|dd|dd}}tjt |ddtj t |z}|j d}|St|trug}|D]l}|dd|dd}}tjt |ddtj t |z}|j|j dn|Sy)zConvert GMTSAR folder name in YYYYDDD into YYYYMMDD. Parameters: date_in - str/list, GMTSAR date format in YYYYDDD, where DDD is the day of year - 1 Returns: date_out - str/list, date in YYYYMMDD format Nr(rBr) r rYr,r:rFr-strftimer rZ)date_inrSdoydt_objdate_outr s ryyyyddd2yyyymmddr{s '3BQKcSY1- #c(0KK??8, O GT " 7H ! hrsm#D[[TAq1BLLc#h4OOF OOFOOH5 6 7 Orc.|ddk(rd|z}|Sd|z}|S)z'Convert year str from YY to YYYY formatrrlrmrnro)rSs ryy2yyyyr}rrrc4t|tr1t|jdddk(r t |}|S|}|St|t rGg}|D]>}t|jdddk(r t |}|j |@|Sy)zAConvert date str from (YY)YYMMDD(THHMM) to YYYYMMDD(THHMM) formatrcrN)r rYrrrqr rZrgdatesOutrps rr9r9s% u{{3" #q (&u-H OH O E4  "D4::c?1%&!+&t, OOD ! " Orct|trt|dk(r|dd}|S|}|St|tr/g}|D]&}t|dk(r|dd}|j |(|Sy)z/Convert date str in (YY)YYMMDD to YYMMDD formatN)r rYrr rZrs ryymmddr#s% u:?QqzH OH O E4  "D4yA~Aay OOD ! " Orc t|tr|g}n t|}t|Dcgc]&}|j ddj dd(c}}t|Dcgc]&}|j ddj dd(c}}t ||Dcgc] \}}|d|}}}t|tr|d}|Scc}wcc}wcc}}w)zConvert date12 into YYYYMMDD_YYYYMMDD format Parameters: date12_list_in - (list of) str Returns: date12_list_out - (list of) str in YYYYMMDD_YYYYMMDD format rM_rr()r rYr r9r rzipdate12_list_in date12_listim_datess_datesmsdate12_list_outs ryyyymmdd_date12r7s .#&%& >* ;Oa #s+11#6q9OPG;Oa #s+11#6q9OPG.1'7.CDda!AaSzDOD.#&)!, POD+C$+C%Cc t|tr|g}n t|}t|Dcgc]&}|j ddj dd(c}}t|Dcgc]&}|j ddj dd(c}}t ||Dcgc] \}}|d|}}}t|tr|d}|Scc}wcc}wcc}}w)zConvert date12 into YYMMDD-YYMMDD format Parameters: date12_list_in - (list of) str Returns: date12_list_out - (list of) str in YYMMDD-YYMMDD format rMrrr()r rYr rr rrrs r yymmdd_date12rNs .#&%& >* MAaiiS)//4Q7MNGMAaiiS)//4Q7MNG.1'7.CDda!AaSzDOD.#&)!, NMDrcg}tjj|rXt|5}|j j }ddd|Dcgc]}|s| }}t t|}|S#1swY1xYwcc}w)zRead Date List from txt fileN)rpathisfileopenread splitlinessortedr9) date_filerhfrRs r read_date_txtrfs}I ww~~i )_ .++-I .!*/1QQ/ /8I./  . .0sA<B B<Bc |sgSt|tr|g}g}|D]L}|jdr1tjj |s4t |}||z }E|g}||z }Ntttt|}|r,ttt|j|}|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) r rYendswithrrrrrr9r set intersection) date_list_in date_list_all date_list_outrLdss rread_date_listrws  L# &$~ M   ::, -ww~~a "1%# B R M 8D]);$<=>MtC $6$C$CM$RST rc4t|}g}|r'|tt||z }td||r`td|tt |}t ||D]/\}}||ks ||vstd||j |1|r`td|tt |} t ||D]/\}}|| kDs ||vstd||j |1ttt|}|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 )rzexclude date: zstart date: z remove date: zend date: ) rjrr printr9rrZrr) rh start_dateend_date exclude_date year_list ex_date_listyear_minrSr year_maxs rget_exclude_date_listrs)y)ILtL'9SS  |n-. zl+,!(:"67!)Y7 .ND(h8<#?z23##H- .  xj)*!(8"45!)Y7 .ND(h8<#?z23##H- . $s<012L rct|}|r|n|d}t|}tt|}|Dcgc]"}tjj ||$}}tjj |t|}g}|D]2}||z }|j |j|jdz z4i} t|D] \}} ||| | <|| fScc}w)aGet temporal Baseline in days with respect to the 1st date Parameters: date_list - list(str), date in YYYYMMDD or YYMMDD format ref_date - str, reference date in YYYYMMDD format Returns: tbase - list(int), temporal baseline in days dateDict - dict with key - string, date in YYYYMMDD format value - int, temporal baseline in days riQ) r9r rYr,r:r;rZrCr* enumerate) rhref_datedt_fmtrdt_listref_dttbasedt_idelta_dtdateDictr s rdate_list2tbasers#I#x1H!H!Y 0F8AB1r{{##Av.BGB [[ ! !(,?,I JF EH&= X]]X%5%5%FFGH H +& 8"1X& (?Cs'C$c t|}tt|}|Dcgc]"}tjj ||$}}|r3d|vr/|Dcgc]$}|tj t|z&}}g}|D]p}|j|jjdz dz z|jdz z|jdz z|jdz z}|j|r||fScc}wcc}w)aGet time in datetime format: datetime.datetime(2006, 5, 26, 0, 0) Parameters: date_list - list of string, date in YYYYMMDD or YYMMDD format seconds - float, float or str, acquisition time of the day info in seconds. 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 rcr)r(rKr`rarb)r9r rYr,r:r;r-rNrSr<r=rdrerfrZ) rhr*rirrgrR datevectorrLdate_vecs rdate_list2vectorrs #I%c)n5K;D EaR[[ ! !![ 1 EE E 3k)CHIaR\\%.99IIJ $FFakkm33a76AAFFk*+HH 012HH 567 (# $ * # F Js 'C=)Dct|}tjtjj ||j }tjtjj ||j }tj||}tj|||z|djd} | D cgc]} | j|} } | Scc} w)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)dtypeO) r rOrr,r:r; isoformat timedelta64arangerQrv) dmindmaxdstepdunitout_fmtrt1t2tstepdt_objsobjrs rget_date_rangers*$/O r{{++D/BLLN OB r{{++D/BLLN OB NN5% (EiiBuHe<@GGLG078s||G$8G8 N9sC/cddlm}m}m}|jdzdk(r'|jdzdk7r|jdzdk7rd}nd}d|z|z |j j d z t|jd z d z zz}d d d||zzd||zz d|d|zzz d|d|zzz z}|d|zz}|jdz|jz|jdz z|z} tjj|jtjdtj | z} | 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)cospisinrtdirDimrr( g(\¥l@ga2U0*?gY+^?gmWel?g?x?g- &(?r2)minutes)mathrrrrSr<r=rNrdrerfr,r:combinerptimer-) utc_time longituderrryear_lengammaeqtime time_offsettst solar_times rutc2solar_timersP"!}}qA(--#"5":x}}s?RVW?W FX !3!3!5!=!=!AE(--Z\J\D]`bDb!b cE x(SZ"77(SZ:OO#a%i.()+3c!e)n+DEFF 1y=(K --" x .21E E SC$$X]]_bggajABLLY\D]]J r)r)N)NNN)r(Dr)__doc__r:r,rrnumpyrOmintpy.objects.progressrrr r&r0r@rHr^rjrqr{r}r9rrrrrrrrrrrorrrs) / 3lB3.$:"J.((.0">&V<>,r