IR-e &ddlZgdZddZddZdS)N)standardpsdmodellogTc|d}tj||\}}|dz}|s|r+tj|||z }nd}||z |z }tj||S)aCompute the reference chi-square for a particular dataset. Note: this is not valid center_data=False and fit_mean=False. Parameters ---------- y : array-like data values dy : float, array, or None, optional data uncertainties center_data : bool specify whether data should be pre-centered fit_mean : bool specify whether model should fit the mean of the data Returns ------- chi2_ref : float The reference chi-square for the periodogram of this data Ngr)npbroadcast_arraysdotsum)ydy center_datafit_meanwmuyws Qlib/python3.11/site-packages/astropy/timeseries/periodograms/lombscargle/utils.pycompute_chi2_refrs|* z   2 & &EAr DAh VAq\\AEEGG #  b&BB 6"b>>c|tj|}||f}|D]}|tvrt|d||kr|Sd|vr|td|dkrdtj| z S|dkrtjd|z  S|dkrtj|dz S|d krtj|dzS|d kr|d|zz S|d kr|d|z z S|dkrt d |z |z|d |S|dkrt |||d }d|z|zStd|d|d)ahConvert power from one normalization to another. This currently only works for standard & floating-mean models. Parameters ---------- Z : array-like the periodogram output N : int the number of data points from_normalization, to_normalization : str the normalization to convert from and to. Options are ['standard', 'model', 'log', 'psd'] chi2_ref : float The reference chi-square, required for converting to or from the psd normalization. Returns ------- Z_out : ndarray The periodogram in the new normalization z is not a valid normalizationrNzHmust supply reference chi^2 when converting to or from psd normalization)rrr)rr)rr)rr)rr)rrr)from_normalizationto_normalizationg?zconversion from 'z' to '')r asarrayNORMALIZATIONS ValueErrorexprconvert_normalizationNotImplementedError)ZNrrchi2_reffrom_tonorm Z_standards rr r 's. 1 A!#34GSS ~ % % 2QQQRR R &--- H, V   %%%261"::~ ' ' 'q1u ~ $ $ $vayy1} $ $ $va!e}} ) ) )AE{ ) ) )AE{ u $ $$ L1  )-     U " "* q%7*   X~ **! M 2 M M:J M M M   r)NTT)N)numpyr rrr rrr*sU444BA A A A A A r