U d{j @s,ddlZddddgZd ddZd d d ZdS) NstandardpsdmodellogTcCs^|dkr d}t||\}}|d}|s,|rBt|||}nd}|||}t||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)npZbroadcast_arraysdotsum)yZdyZ center_dataZfit_meanwmuZywr p/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/timeseries/periodograms/lombscargle/utils.pycompute_chi2_refs rcCs<t|}||f}|D]}|tkrt|dq||kr>|Sd|krV|dkrVtd|dkrndt| S|dkrtd| S|dkrt|dS|d krt|dS|d kr|d|S|d kr|d|S|dkrtd |||d |dS|dkr"t|||d d}d||Std|d|ddS)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 '')rZasarrayNORMALIZATIONS ValueErrorexprconvert_normalizationNotImplementedError)ZNrrZchi2_refZfrom_toZnormZ Z_standardr r rr'sR       r)NTT)N)numpyrrrrr r r rs  !