B d3o@sdZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZmZddlmZddlmZmZdd lmZmZmZd d Zd d ZiZdddZddZdejkreejddGddde Z!ddZ"Gddde Z#dddZ$dS)zdThis module provides utilities for calculating detector responses and timing between observatories. N) TimeSeries)InterpolatingConfigParser)Time) constants coordinatesunits)rotation_matrix)sdaymeter)cossinpicCst|dddddj}|S)Ngpsutc)rr)formatscalelocationZmean)rZ sidereal_timerad)gps_timegmstr[/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/detector.py gmst_accurate.srcsJtjddD}fdd|D}fdd|D}tt||S)aeReturn list of detectors known in the currently sourced lalsuite. This function will query lalsuite about which detectors are known to lalsuite. Detectors are identified by a two character string e.g. 'K1', but also by a longer, and clearer name, e.g. KAGRA. This function returns both. As LAL doesn't really expose this functionality we have to make some assumptions about how this information is stored in LAL. Therefore while we hope this function will work correctly, it's possible it will need updating in the future. Better if lal would expose this information properly. cSsg|]}d|kr|qS)ZDETECTOR_PREFIXr).0jrrr @sz+get_available_detectors..csg|] }|qSrr)rk)ldrrrAscsg|]}|ddqS)PREFIXNAME)replace)rr)rrrrBs)lal__dict__keyslistzip)Zknown_lal_namesZknown_prefixesZ known_namesr)rrget_available_detectors4s r&cCsJ|dkr|tjd}td|td|}}t| |dg||dgdddgg}td|td|}}t| |dg||dgdddgg} | |d} t|tjd} ttjd|tjd} t| | } t| | } t| j | } t j j |tj|tj|tj d}t|jj|jj|jjg}|| |||d d d t|<dS) a Add a new detector on the earth Parameters ---------- name: str two-letter name to identify the detector longitude: float Longitude in radians using geodetic coordinates of the detector latitude: float Latitude in radians using geodetic coordinates of the detector yangle: float Azimuthal angle of the y-arm (angle drawn from pointing north) xangle: float Azimuthal angle of the x-arm (angle drawn from point north). If not set we assume a right angle detector following the right-hand rule. height: float The height in meters of the detector above the standard reference ellipsoidal earth Ng@rg@zy)heightg)rresponseyanglexangler* xaltitude yaltitude)npr r r arrayrrrmatmulTr EarthLocationZ from_geodeticr xvaluer)r(_custom_ground_detectors)name longitudelatituder,r-r*abZxrespZyresprespZrm1Zrm2Zrmlocrrradd_detector_on_earthGs2$$     r?c sdtddgfi}t|}|d}x|D]}t|d|y|d\}}Wn&tk r|td| YnXxD]}t ||<qWyfdd |D}Wn0tk r} ztd |Wd d } ~ XYnX|| f|q(Wd S) z Add custom detectors from a configuration file Parameters ---------- config_files: str or list of strs The config file(s) which specify new detectors Z earth_normalr9r:detectorz detector-{}methodz(Missing or unkown method, options are {}csg|]}|qSr)pop)rarg)kwdsrrrsz(load_detector_config..z2missing required detector argument {} are requiredN) r?rZget_subsectionsdictitemsrrBKeyError ValueErrorr#floatupper) config_filesmethodsconfZdetsdetrA arg_namesrargser)rDrload_detector_configs&    rRZPYCBC_DETECTOR_CONFIG:c@s~eZdZdZd ddZddZddZd d Zd d Zd!ddZ ddZ ddZ ddZ d"ddZ ddZddZddZdS)#Detectorz"A gravitational wave detector VAcCst||_|ddtDkrNtjd}||j|_|jj|_|jj |_ n:|t krzt ||_ |j d|_|j d|_ nt d |tj|j d|j d|j d td }|jj|_|jj|_||_d |_d |_d S) a Create class representing a gravitational-wave detector Parameters ---------- detector_name: str The two-character detector string, i.e. H1, L1, V1, K1, I1 reference_time: float Default is time of GW150914. In this case, the earth's rotation will be estimated from a reference time. If 'None', we will calculate the time for each gps time requested explicitly using a slower but higher precision method. cSsg|] \}}|qSrr)rpfxr8rrrrsz%Detector.__init__.. lalsimulationr+rzUnkown detector {}rr')unitN)strr8r&pycbcZlibutilsZimport_optionalZDetectorPrefixToLALDetector_lalr+rr7inforHrrr4r Zlatrr:Zlonr9reference_timer gmst_reference)selfZ detector_namer^Zlalsimr>rrr__init__s(         zDetector.__init__cCs2|jdk r&ttjj|_t|j|_ntddS)Nzconvrrr get_icrs_posszDetector.get_icrs_posc CsR|||||\}}t|} dd| | } || d|| dd} || S)a Distance scaled to account for amplitude factors The effective distance of the source. This scales the distance so that the amplitude is equal to a source which is optimally oriented with respect to the detector. For fixed detector-frame intrinsic parameters this is a measure of the expected signal strength. Parameters ---------- distance: float Source luminosity distance in megaparsecs ra: float The right ascension in radians dec: float The declination in radians pol: float Polarization angle of the gravitational wave in radians time: float GPS time in seconds inclination: The inclination of the binary's orbital plane Returns ------- eff_dist: float The effective distance of the source g?g?g@)r~r0r ) r`ZdistancerrZpolrZ inclinationrrZiciprrrreffective_distance s  zDetector.effective_distance)rU)rl)r!N)__name__ __module__ __qualname____doc__rardr!rhrkr~rrrrrrrrrrrrTs %  P# LrTcCstjd|}ddt|t|td|td|t|td|td|}ddt|t|td|td|t|td|td|}||fS)uEReturn the antenna pattern factors F+ and Fx as a function of sky location and polarization angle for a hypothetical interferometer located at the north pole. Angles are in radians. Declinations of ±π/2 correspond to the normal to the detector plane (i.e. overhead and underneath) while the point with zero right ascension and declination is the direction of one of the interferometer arms. Parameters ---------- right_ascension: float declination: float polarization: float Returns ------- f_plus: float f_cros: float g@gg?g?)r0r r r )rtrurvthetaZf_plusZf_crossrrroverhead_antenna_pattern/s ."."rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)LISAzFor LISA detector cCsdS)Nr)r`rrrraTsz LISA.__init__c Cs4t|dddj}ttdd}d\}}dtj|d|}|ddtd|}d \}}|d|td } |t||| t|t|t|dt|d t|} |t||| t|t|t|dt|d t|} td  || t||} t| | | g|_ |j S) aWReturn the position of LISA detector for a given reference time Parameters ---------- ref_time : numpy.ScalarType Returns ------- location : numpy.ndarray of shape (3,3) Returns the position of all 3 sattelites with each row correspoding to a single axis. rr)valrrrX)rrg@g@)g?g5Þ?r') rZjyearr0r1ranger sqrtr r r) r`Zref_timenkappaZ_lambda_alphaZbeta_nr;LrQr5r)r(rrrget_posWs DD z LISA.get_poscCs|}tj|d|d|dtjdddd}d|_t|jj tj  }t t|jt|jt|jg|}|S)ac Transforms ICRS frame to GCRS frame Parameters ---------- loc : numpy.ndarray shape (3,1) units: AU Cartesian Coordinates of the location in ICRS frame Returns ---------- loc : numpy.ndarray shape (3,1) units: meters GCRS coordinates in cartesian system rrXr'rr)r5r)r(rYrrr)rrrrrrr0rr5rYrrrr1r)r()r`rr>rrrr get_gcrs_posrszLISA.get_gcrs_posc CsV|j|}t|}|t|}|t| }t|} t||| g} || tjjS)aReturn the time delay from the LISA detector to detector for a signal with the given sky location. In other words return `t1 - t2` where `t1` is the arrival time in this detector and `t2` is the arrival time in the other location. Units(AU) Parameters ---------- other_location : numpy.ndarray of coordinates in ICRS frame A detector instance. right_ascension : float The right ascension (in rad) of the signal. declination : float The declination (in rad) of the signal. t_gps : float The GPS time (in s) of the signal. Returns ------- numpy.ndarray The arrival time difference between the detectors. ) rr r r0r1rirrjr6) r`rrtrurwryrrrrrrrrrs  zLISA.time_delay_from_locationcCst||}|||||S)aReturn the time delay from the LISA detector for a signal with the given sky location in ICRS frame; i.e. return `t1 - t2` where `t1` is the arrival time in this detector and `t2` is the arrival time in the other detector. Parameters ---------- other_detector : detector.Detector A detector instance. right_ascension : float The right ascension (in rad) of the signal. declination : float The declination (in rad) of the signal. t_gps : float The GPS time (in s) of the signal. Returns ------- numpy.ndarray The arrival time difference between the detectors. )rTrr)r`rNrtrurwr>rrrrszLISA.time_delay_from_detectorc Cs^t|ddd}tjd|ddd}d|_|tt|j t|j t|j g|||S) zBReturn the time delay from the earth center in ICRS frame rr)rrrearthN)rrr) rrget_bodyrrrr0r1rr5r)r()r`rtrurwrrrrrsz!LISA.time_delay_from_earth_centerN) rrrrrarrrrrrrrrrQsr, cCs|r|t|SdS)zPretty-print a list (or set) of detectors: return a string listing the given detectors alphabetically and separated by the given string (comma by default). z no detectors)joinsorted)Zifos separatorrrrppdetssr)rNr)r)%rosrr0r!Zpycbc.libutilsr[Z pycbc.typesrZpycbc.types.configrZ astropy.timerZastropyrrrZ$astropy.coordinates.matrix_utilitiesrZastropy.units.sir r r r r rr&r7r?rRenvironsplitrprTrrrrrrrs2     8 "