B dG@sDdZddlmZyddlmZWn ek r@ddlmZYnXddlZddlZddl Z ddl Z ddl Z ddl m Z ddlZddlZyeWnek reZYnXddlmZddlmZdZd ejZejZeGd d d eZy ddlZGd d d ej j!Z!Wn&ek r.Gdd d eZ!YnXGddde"Z#dS)zs This module contains bits and pieces of use when interacting with LAL and LAL-derived code (eg. LALApps programs) )absolute_import)total_orderingN)urllib)segments) git_versionz"Kipp Cannon z git id %sc@seZdZdZd1ddZeddZZeddZZ dd Z d d Z d d Z ddZ ddZddZddZddZddZddZeZddZeZddZd d!Zd"d#ZeZd$d%ZeZd&d'Zd(d)Zd*d+Zd,d-Z d.d/Z!d0S)2 LIGOTimeGPSa7 An object for storing times with nanosecond resolution. LAL defines an equivalent object which is used through-out the search algorithms to represent times. Many LALApps routines input and output times in a manner that meshes well with this object. Internally the time is represented as a signed integer "seconds" part and an unsigned integer "nanoseconds" part. The actual time is always constructed by adding the nanoseconds to the seconds. So -0.5 s is represented by setting seconds = -1, and nanoseconds to 500000000. That's the way LAL does it. rcCszt|tttfs2y t|}Wnt|YnXt|tr`t|\}}t|}||d7}nt|tjsVt|tj tj fr| drdnd}y@d|kr| d\}}t|td|d}nd}t|}Wntd|YnX||7}n\t|dr$t|d r$||j7}|j}n2t|d rNt|d rN||j7}|j}nt||t|d |_t|d |_d S)al Create a LIGOTimeGPS instance. The first parameter is the count of seconds, and the second (optional) parameter is the count of nanoseconds. If the nanoseconds parameter is not supplied, it is assumed to be 0. Either parameter can be a numeric type or an ASCII string. Example: >>> LIGOTimeGPS(100.5) LIGOTimeGPS(100, 500000000) >>> LIGOTimeGPS("100.5") LIGOTimeGPS(100, 500000000) >>> LIGOTimeGPS(100, 500000000) LIGOTimeGPS(100, 500000000) >>> LIGOTimeGPS(0, 100500000000) LIGOTimeGPS(100, 500000000) >>> LIGOTimeGPS(100.2, 300000000) LIGOTimeGPS(100, 500000000) >>> LIGOTimeGPS("0.000000001") LIGOTimeGPS(0, 1) >>> LIGOTimeGPS("0.0000000012") LIGOTimeGPS(0, 1) >>> LIGOTimeGPS("0.0000000018") LIGOTimeGPS(0, 2) >>> LIGOTimeGPS("-0.8") LIGOTimeGPS(-1, 200000000) >>> LIGOTimeGPS("-1.2") LIGOTimeGPS(-2, 800000000) geA-.rz%invalid literal for LIGOTimeGPS(): %s gpsSecondsgpsNanoSecondsseconds nanosecondsiʚ;N) isinstancefloatintlong TypeErrormathmodfsix integer_types binary_type text_typelstrip startswithsplitroundhasattrr r rr_LIGOTimeGPS__seconds_LIGOTimeGPS__nanoseconds)selfrrnssignr%U/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/glue/lal.py__init__\s<      zLIGOTimeGPS.__init__cCs|jS)N)r )r"r%r%r&zLIGOTimeGPS.cCs|jS)N)r!)r"r%r%r&r(r)cCsd|j|jfS)NzLIGOTimeGPS(%d, %u))r r!)r"r%r%r&__repr__szLIGOTimeGPS.__repr__cCsh|jdks|jdkr&d|j|jf}n2|jdkrJd|jdd|jf}ndd|j}|ddS) z= Return an ASCII string representation of a LIGOTimeGPS. rz%d.%09ur r iʚ;z-0.%09u0r )r r!rstrip)r"sr%r%r&__str__s  zLIGOTimeGPS.__str__cCs|j|jdS)zg Convert a LIGOTimeGPS to seconds as a float. Example: >>> float(LIGOTimeGPS(100.5)) 100.5 g& .>)r r!)r"r%r%r& __float__s zLIGOTimeGPS.__float__cCs|jS)zt Return the integer part (seconds) of a LIGOTimeGPS as an int. Example: >>> int(LIGOTimeGPS(100.5)) 100 )r )r"r%r%r&__int__s zLIGOTimeGPS.__int__cCs t|jS)z Return the integer part (seconds) of a LIGOTimeGPS as a long. Example: >>> repr(long(LIGOTimeGPS(100.5))) == '100L' if sys.version_info.major < 3 else True True )rr )r"r%r%r&__long__s zLIGOTimeGPS.__long__cCs|jd|jS)zz Convert a LIGOTimeGPS to a count of nanoseconds as a long. Example: >>> LIGOTimeGPS(100.5).ns() 100500000000 iʚ;)r r!)r"r%r%r&r#s zLIGOTimeGPS.nscCs |j|jAS)N)r r!)r"r%r%r&__hash__szLIGOTimeGPS.__hash__cCs<t|ts,y t|}Wntk r*tSX||kS)N)rrrNotImplementedr#)r"otherr%r%r&__eq__s   zLIGOTimeGPS.__eq__cCs<t|ts,y t|}Wntk r*tSX||kS)a< Compare a value to a LIGOTimeGPS. If the value being compared to the LIGOTimeGPS is not also a LIGOTimeGPS, then an attempt is made to convert it to a LIGOTimeGPS. Example: >>> LIGOTimeGPS(100.5) < LIGOTimeGPS(200) True >>> LIGOTimeGPS(100.5) < 200 True >>> LIGOTimeGPS(100.5) < "200" True )rrrr3r#)r"r4r%r%r&__lt__s   zLIGOTimeGPS.__lt__cCst|jpt|jS)zc Return True if the LIGOTimeGPS is nonzero. Example: >>> bool(LIGOTimeGPS(100.5)) True )boolr r!)r"r%r%r&__bool__s zLIGOTimeGPS.__bool__cCs,t|tst|}t|j|j|j|jS)at Add a value to a LIGOTimeGPS. If the value being added to the LIGOTimeGPS is not also a LIGOTimeGPS, then an attempt is made to convert it to a LIGOTimeGPS. Example: >>> LIGOTimeGPS(100.5) + LIGOTimeGPS(3) LIGOTimeGPS(103, 500000000) >>> LIGOTimeGPS(100.5) + 3 LIGOTimeGPS(103, 500000000) >>> LIGOTimeGPS(100.5) + "3" LIGOTimeGPS(103, 500000000) )rrr rr!r)r"r4r%r%r&__add__s zLIGOTimeGPS.__add__cCs,t|tst|}t|j|j|j|jS)a Subtract a value from a LIGOTimeGPS. If the value being subtracted from the LIGOTimeGPS is not also a LIGOTimeGPS, then an attempt is made to convert it to a LIGOTimeGPS. Example: >>> LIGOTimeGPS(100.5) - LIGOTimeGPS(3) LIGOTimeGPS(97, 500000000) >>> LIGOTimeGPS(100.5) - 3 LIGOTimeGPS(97, 500000000) >>> LIGOTimeGPS(100.5) - "3" LIGOTimeGPS(97, 500000000) )rrr rr!r)r"r4r%r%r&__sub__+s zLIGOTimeGPS.__sub__cCs,t|tst|}t|j|j|j|jS)z* Subtract a LIGOTimeGPS from a value. )rrrr rr!)r"r4r%r%r&__rsub__>s zLIGOTimeGPS.__rsub__c Cs|j}|j}|dkr.|dkr.|d7}|d8}n |dkrN|dkrN|d8}|d7}|d}||}|r~|dtt|ddnd}||}|t|9}d}xF||||||||fD]&}t|\} } || 7}|| d7}qWt|t|S) zi Multiply a LIGOTimeGPS by a number. Example: >>> LIGOTimeGPS(100.5) * 2 LIGOTimeGPS(201, 0) rr iʚ;iggeA) r r!rrlogrrrr) r"r4rrZsloZshiZoloZohiaddendnr-r%r%r&__mul__Fs&  $ "zLIGOTimeGPS.__mul__cCsXtt|t|}x>tdD]2}t|||t|}||7}t|dkrPqW|S)zn Divide a LIGOTimeGPS by a number. Example: >>> LIGOTimeGPS(100.5) / 2 LIGOTimeGPS(50, 250000000) dg& .>)rrrangeabs)r"r4quotientr@Zresidualr%r%r& __truediv__js  zLIGOTimeGPS.__truediv__cCstS)z#Integer division (not implemented).)r3)r"r4r%r%r& __floordiv__~szLIGOTimeGPS.__floordiv__cCst||}|||S)z Compute the remainder when a LIGOTimeGPS is divided by a number. Example: >>> LIGOTimeGPS(100.5) % 3 LIGOTimeGPS(1, 500000000) )r)r"r4rEr%r%r&__mod__s zLIGOTimeGPS.__mod__cCs|S)Nr%)r"r%r%r&__pos__szLIGOTimeGPS.__pos__cCstd| S)Nr)rr#)r"r%r%r&__neg__szLIGOTimeGPS.__neg__cCs|jdkr|S| S)Nr)r )r"r%r%r&__abs__s zLIGOTimeGPS.__abs__N)r)"__name__ __module__ __qualname____doc__r'propertyrr rr r*r.r/r0r1r#r2r5r6r8 __nonzero__r9__radd__r:r;rA__rmul__rFZ__div__rGrHrIrJrKr%r%r%r&rKs8  F     "rcseZdZfddZZS) CacheEntrycstdttt|j||S)NzCglue.lal.CacheEntry is deprecated, use lal.utils.CacheEntry instead)warningswarnDeprecationWarningsuperrTr')r"argskwargs) __class__r%r&r's zCacheEntry.__init__)rLrMrNr' __classcell__r%r%)r[r&rTsrTc@seZdZddZdS)rTcOs tddS)Nz]glue.lal.CacheEntry is deprecated and lal is not available. Use lal.utils.CacheEntry instead.) ValueError)r"rYrZr%r%r&r'szCacheEntry.__init__N)rLrMrNr'r%r%r%r&rTscseZdZdZeZefddZeeZefddZ ee Z efddZ ee Z dd Z d d Z d d Z ddZddZddZddZddZddZddZd'ddZd d!Zd(d#d$Zfd%d&ZZS))Cachez An object representing a LAL cache file. Currently it is possible to add anything to a Cache. This method should check that the thing you are adding is a CacheEntry and throw and error if it is not. csfdd|D}|S)zE Return a Cache object whose entries are read from an open file. csg|]}j|dqS))coltype) entry_class).0line)clsr_r%r& sz"Cache.fromfile..r%)rcfileobjr_cr%)rcr_r&fromfileszCache.fromfilecCs0|}x$|D]}||jt||dq W|S)z` Read Cache objects from the files named and concatenate the results into a single Cache. )r_)extendrgopen)rc filenamesr_cachefilenamer%r%r& fromfilenamess zCache.fromfilenamescs"ddfdd|DS)a0 Return a Cache whose entries are inferred from the URLs in urllist, if possible. PFN lists will also work; for PFNs, the path will be absolutized and "file://" and "localhost" will be assumed for the schemes and hosts. The filenames must be in the format set forth by DASWG in T050017-00. cSsHtj|\}}}}}|dkr*tj|}tj|p6d|p.pfn_to_urlcs g|]}jj|dqS))r_)r`Z from_T050017)raf)rcr_r{r%r&rdsz#Cache.from_urls..r%)rcZurllistr_r%)rcr_r{r& from_urlss zCache.from_urlscCsBt|d}x0t|dddD]\}}||kr |||=q W|S)z2 Remove elements from self that are in other. r Nr )len enumerate)r"r4endielemr%r%r&__isub__s  zCache.__isub__cs|fdd|DS)zJ Return a Cache containing the entries of self that are not in other. csg|]}|kr|qSr%r%)rar)r4r%r&rdsz!Cache.__sub__..)r[)r"r4r%)r4r&r:sz Cache.__sub__cCs||||S)zO Append entries from other onto self without introducing (new) duplicates. )rh)r"r4r%r%r&__ior__sz Cache.__ior__cCs||dd|S)z> Return a Cache containing all entries of self and other. N)r[r)r"r4r%r%r&__or__sz Cache.__or__cCsBt|d}x0t|dddD]\}}||kr |||=q W|S)z4 Remove elements in self that are not in other. r Nr )r~r)r"r4rrrr%r%r&__iand__s  zCache.__iand__cs|fdd|DS)zK Return a Cache containing the entries of self that are also in other. csg|]}|kr|qSr%r%)rar)r4r%r&rdsz!Cache.__and__..)r[)r"r4r%)r4r&__and__ sz Cache.__and__cst|fdd|DS)z Return a Cache which has every element of self, but without duplication. Preserve order. Does not hash, so a bit slow. cs"g|]}|kr|s|qSr%)add)rax)seenr%r&rdsz Cache.unique..)setr[)r"r%)rr&uniquesz Cache.uniquecCs,x|D]}|dt|qW|dS)z= write a cache object to the fileobj as a lal cache file z%s N)writestrclose)r"reentryr%r%r&tofile s z Cache.tofilecCs*x|D]}|d|jqW|dS)z? write a cache object to filename as a plain text pfn file z%s N)rrtr)r"rerr%r%r& topfnfile(s zCache.topfnfilecCs$t}x|D]}||jO}qW|S)z Return a segmentlistdict object describing the instruments and times spanned by the entries in this Cache. The return value is coalesced. )rZsegmentlistdict)r"drr%r%r&to_segmentlistdict0s zCache.to_segmentlistdictNFcs|rfddfddn@|dk r2d|d}|dk rFd|d}fddfdd|}|dk rtt|fdd |D}|dk rtt|fd d |D}dk rЇfd d |D}dk rfd d |D}||S) a Return a Cache object with those CacheEntries that contain the given patterns (or overlap, in the case of segment or segmentlist). If exact_match is True, then non-None ifos, description, and segment patterns must match exactly, and a non-None segmentlist must contain a segment which matches exactly). It makes little sense to specify both segment and segmentlist arguments, but it is not prohibited. Bash-style wildcards (*?) are allowed for ifos and description. cs |jkS)N)segment)e)rr%r&r(Kr)zCache.sieve..cs |jkS)N)r)r) segmentlistr%r&r(Lr)N*cs |jS)N)Z intersectsr)r)rr%r&r(Pr)cs |jS)N)Zintersects_segmentr)r)rr%r&r(Qr)cs g|]}|jdk r|qS)N)matchZ observatory)rar) ifos_regexpr%r&rdWszCache.sieve..cs g|]}|jdk r|qS)N)r description)rar) descr_regexpr%r&rd[scsg|]}|r|qSr%r%)rar) segment_funcr%r&rd^scsg|]}|r|qSr%r%)rar)segmentlist_funcr%r&rdcs)recompilefnmatch translateZcoalescer[)r"ZifosrrrZ exact_matchrfr%)rrrrrrr&sieve;s,     z Cache.sievecCsdd|DS)z+ Return a list of physical file names. cSsg|] }|jqSr%)rt)rarr%r%r&rdksz!Cache.pfnlist..r%)r"r%r%r&pfnlistgsz Cache.pfnlistrVcCs|dkrtdg}g}x0|D](}tj|jr<||q||qWt|dkrdt|t|f}|dkrtjd|n$|dkrt|n|dkrntd | || |fS) an Runs through the entries of the Cache() object and checks each entry if the file which it points to exists or not. If the file does exist then it adds the entry to the Cache() object containing found files, otherwise it adds the entry to the Cache() object containing all entries that are missing. It returns both in the follwing order: Cache_Found, Cache_Missed. Pass on_missing to control how missing files are handled: "warn": print a warning message saying how many files are missing out of the total checked. "error": raise an exception if any are missing "ignore": do nothing )rVerrorignorez0on_missing must be "warn", "error", or "ignore".rz2%d of %d files in the cache were not found on diskrVz warning: %s rrz(Why am I here? Please file a bug report!) r]rsrtisfileappendr~sysstderrrr[)r"Z on_missingZc_foundZc_missedrmsgr%r%r&checkfilesexistms&    zCache.checkfilesexistcs|tt|||S)N)r[rXr^ __getslice__)r"rj)r[r%r&rszCache.__getslice__)NNNNF)rV)rLrMrNrOrTr`rrg classmethodrmr}rr:rrrrrrrrrrrrr\r%r%)r[r&r^s.       + 'r^)$rO __future__r functoolsr ImportErrorZ functools32rrrsrrZ six.movesrrUrr NameErrorrZligorZgluer __author__id __version__date__date__objectrZ lal.utilsZlalutilsrTlistr^r%r%r%r&s@       Z