a ߙfb2@sddlZddlZddlZddlmZmZddlZddlm Z ddl m Z ddl mZddlmZddlmZmZmZmZddlmZgd ZiZeZGd d d eZGd d d ejdZGdddejdZ ddZ!dS)N) FunctionTypeMappingProxyType)UnifiedReadWriteMethod) classproperty)MetaData)CosmologyFromFormat CosmologyReadCosmologyToFormatCosmologyWrite) Parameter) CosmologyCosmologyErrorFlatCosmologyMixinc@s eZdZdS)rN)__name__ __module__ __qualname__rr5lib/python3.9/site-packages/astropy/cosmology/core.pyr!srcseZdZdZeZeeZee Z ee Z ee ZdZdZfddZeddddZfd d Zdd d ZeddZd dddZddZddZddZddZddZZS)r aBase-class for all Cosmologies. Parameters ---------- *args Arguments into the cosmology; used by subclasses, not this base class. name : str or None (optional, keyword-only) The name of the cosmology. meta : dict or None (optional, keyword-only) Metadata for the cosmology, e.g., a reference. **kwargs Arguments into the cosmology; used by subclasses, not this base class. Notes ----- Class instances are static -- you cannot (and should not) change the values of the parameters. That is, all of the above attributes (except meta) are read only. For details on how to create performant custom subclasses, see the documentation on :ref:`astropy-cosmology-fast-integrals`. rcsXtt|jj|jj|jj|jj|jjd}t ||j}|j j |_ |j }|jtddgt|jd}||_||_gg}|jD]2}t||}t|tr|jr||n|q|jD]B\}}|vs|dst|tsq|jr||n|qʇfdd|j jD}|t|_|jt||_|t |j!<dS)N)nameZargdefsZclosureclsr parameters_cs$g|]}|vr|qSr)popindex).0nrrr osz/Cosmology.__init_subclass__..)"super__init_subclass__r__new____code__ __globals__r __defaults__ __closure__ functoolsupdate_wrapper__init____kwdefaults___init_signaturereplaceinspectr listrvaluesZ __signature____parameters__getattr isinstanceZderivedappend__dict__items startswithkeystuple__all_parameters___COSMOLOGY_CLASSESr)rnewsigZderived_parametersrpvZordered __class__rrr Ms8   $      zCosmology.__init_subclass__T)ZlazycCs.t|j}|jt|jddd}|S)z*Initialization signature (without 'self').rNr)r,Z signaturer(r+r-rr.)rr;rrrr*zs zCosmology._init_signaturecs2t|}|jj|i|}||j|_|SN)rr!r* bind_partialZapply_defaultsZ arguments_init_arguments)rargskwargsselfbar>rrr!s  zCosmology.__new__NcCs||_|j|pidSr@)_namemetaupdate)rErrHrrrr(szCosmology.__init__cCs|jS)z#The name of the Cosmology instance.)rGrErrrrszCosmology.name)rHcKs~|dur|s|S|d|jdur*|jdndi|j|p>i}i|jd|i|}|jjfi|}|j|ji|jS)a:Returns a copy of this object with updated parameters, as specified. This cannot be used to change the type of the cosmology, so ``clone()`` cannot be used to change between flat and non-flat cosmologies. Parameters ---------- meta : mapping or None (optional, keyword-only) Metadata that will update the current metadata. **kwargs Cosmology parameter (and name) modifications. If any parameter is changed and a new name is not given, the name will be set to "[old name] (modified)". Returns ------- newcosmo : `~astropy.cosmology.Cosmology` subclass instance A new instance of this class with updated parameters as specified. If no modifications are requested, then a reference to this object is returned instead of copy. Examples -------- To make a copy of the ``Planck13`` cosmology with a different matter density (``Om0``), and a new name: >>> from astropy.cosmology import Planck13 >>> newcosmo = Planck13.clone(name="Modified Planck 2013", Om0=0.35) If no name is specified, the new name will note the modification. >>> Planck13.clone(Om0=0.35).name 'Planck13 (modified)' Nrz (modified)rH) setdefaultrrHrBr*rAr?rCrD)rErHrDZnew_metaZnew_initrFrrrclones$  zCosmology.clonecCs6||}|tur&t|dr&||}|tur2|SdS)aCheck equivalence between Cosmologies. Two cosmologies may be equivalent even if not the same class. For example, an instance of ``LambdaCDM`` might have :math:`\Omega_0=1` and :math:`\Omega_k=0` and therefore be flat, like ``FlatLambdaCDM``. Parameters ---------- other : `~astropy.cosmology.Cosmology` subclass instance The object in which to compare. Returns ------- bool True if cosmologies are equivalent, False otherwise. __equiv__F)rMNotImplementedhasattr)rEotherZequivrrr is_equivalents  zCosmology.is_equivalentcsBjjurtStjtjkosz&Cosmology.__equiv__..)r?rNsetr8rS)rErPZ params_eqrrVrrMs zCosmology.__equiv__cCs.|j|jurtS||}|j|jk}|o,|S)a}Check equality between Cosmologies. Checks the Parameters and immutable fields (i.e. not "meta"). Parameters ---------- other : `~astropy.cosmology.Cosmology` subclass instance The object in which to compare. Returns ------- bool True if Parameters and names are the same, False otherwise. )r?rNrMr)rErPZ equivalentZname_eqrrr__eq__s    zCosmology.__eq__cszfddjD}fddjDjjd}jdurR|djd7}fdd|D}|d |d S) Ncsi|]}|t|qSr)r0rTrJrr z&Cosmology.__repr__..csi|]}|tj|qSr)r0r?rTrJrrrZ r[(zname="z", c3s6|].\}}|dt||dur&|jndVqdS)=N)format format_spec)rrUr=)cpsrrrW&sz%Cosmology.__repr__..z, ))r/r?rrr4join)rEZpsZnameleadZfmtpsr)rarEr__repr__s  zCosmology.__repr__cKs|jdd|i|S)aReturn a `~astropy.table.Table` of type ``cls``. Parameters ---------- cls : type Astropy ``Table`` class or subclass. copy : bool Ignored. **kwargs : dict, optional Additional keyword arguments. Passed to ``self.to_format()``. See ``Cosmology.to_format.help("astropy.table")`` for allowed kwargs. Returns ------- `astropy.table.Table` or subclass instance Instance of type ``cls``. astropy.tabler)re) to_format)rErcopyrDrrr__astropy_table__+szCosmology.__astropy_table__)NN)rrr__doc__rrHrrZ from_formatr rfr readr writer/r8r rr*r!r(propertyrrLrQrMrYrdrh __classcell__rrr>rr %s* -  9 r ) metaclassc@seZdZdZdS)raZ Mixin class for flat cosmologies. Do NOT instantiate directly. Note that all instances of ``FlatCosmologyMixin`` are flat, but not all flat cosmologies are instances of ``FlatCosmologyMixin``. As example, ``LambdaCDM`` **may** be flat (for the a specific set of parameter values), but ``FlatLambdaCDM`` **will** be flat. N)rrrrirrrrr@srcCshddlm}t||rNddl}ddlm}|d|d|d|t||Stdt d |d dS) Nr)flrwr)AstropyDeprecationWarningz`astropy.cosmology.core.zX` has been moved (since v5.0) and should be imported as ``from astropy.cosmology import z+``. In future this will raise an exception.zmodule z has no attribute .) r^rorOwarningsZastropy.utils.exceptionsrpwarnr0AttributeErrorr)attrrorrrprrr __getattr__Ns    rv)"abcr&r,typesrrZnumpyrRZ astropy.unitsZunitsuZastropy.io.registryrZastropy.utils.decoratorsrZastropy.utils.metadatarZconnectrr r r Z parameterr __all__Z__doctest_requires__dictr9 ExceptionrABCMetar rrvrrrrs&