B îq\û ã@s,ddlZddddgZd dd„Zd d d „ZdS) éNÚstandardÚpsdÚmodelÚlogTcCs^|dkr d}t ||¡\}}|d}|s,|rBt ||¡| ¡}nd}|||}t ||¡S)a Compute 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 : boolean specify whether data should be pre-centered fit_mean : boolean specify whether model should fit the mean of the data Returns ------- chi2_ref : float The reference chi-square for the periodogram of this data NégÀr)ÚnpZbroadcast_arraysÚdotÚsum)ÚyZdyZ center_dataZfit_meanÚwZmuZyw©r ú>lib/python3.7/site-packages/astropy/stats/lombscargle/utils.pyÚcompute_chi2_refs rcCs:t |¡}||f}x"|D]}|tkrtd |¡ƒ‚qW||krB|Sd|krZ|dkrZtdƒ‚|dkrrdt | ¡S|dkrŠt d|¡ S|dkr t |¡dS|d kr¶t |d¡S|d krÊ|d|S|d krÞ|d|S|dkrtd |||d |dS|dkr&t|||d d}d||Std ||¡ƒ‚dS)apConvert power from one normalization to another. This currently only works for standard & floating-mean models. Parameters ---------- Z : array_like the periodogram output N : integer the number of data points from_normalization, to_normalization : strings 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 {0} is not a valid normalizationrNzHmust supply reference chi^2 when converting to or from psd normalization)rrr)rr)rr)rr)rr)rrér)Úfrom_normalizationÚto_normalizationgà?zconversion from '{0}' to '{1}') rZasarrayÚNORMALIZATIONSÚ ValueErrorÚformatZexprÚconvert_normalizationÚNotImplementedError)ÚZÚNrrZchi2_refZfrom_toZnormZ Z_standardr r r r(sD        r)NTT)N)Znumpyrrrrr r r r Ús  "