HR-eVhddlmZddlZddlZddlmZmZmZddlZ ddl m Z ddl m Z ddlmZddlmZmZmZmZdd lmZer dd lmZdd lmZgd ZiZeZed dZeddZ Gdde!Z"Gddej#Z$Gddej#Z%dZ&dS)) annotationsN) TYPE_CHECKINGAnyTypeVar)UnifiedReadWriteMethod) classproperty)MetaData)CosmologyFromFormat CosmologyReadCosmologyToFormatCosmologyWrite) Parameter)Mapping) _FormatType) CosmologyCosmologyErrorFlatCosmologyMixin_CosmoTr)bound _FlatCosmoTrceZdZdS)rN)__name__ __module__ __qualname__6lib/python3.11/site-packages/astropy/cosmology/core.pyrr5sDrrcpeZdZUdZeZeeZee Z ee Z ee ZdZded<dZded<fdZedd Zdd Zed Zeejd Zd ddZedZddddZd dZd dZdZ dZ!xZ"S)!raBase-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`. rtuple[str, ...]__parameters____all_parameters__ctgg}|jD]X}t||}t |t r1|jr||n|Y|j D]e\}}|vs*| dst |t s4|jr||n|ffd|j j D}|zt|_|jt|z|_|t |j<dS)N_chg|].}|v|/Sr)popindex).0n parameterss r z/Cosmology.__init_subclass__..vsD   J NN:++A.. / /r)super__init_subclass__r!getattr isinstancerderivedappend__dict__items startswith_init_signaturer*keystupler"_COSMOLOGY_CLASSESr)clsderived_parametersr)pvorderedr* __class__s @rr-zCosmology.__init_subclass__as !!###  # T TAQA!Y'' T01 S"))!,,,z?P?PQR?S?SL&&(( P PDAqJ!,,s"3"3:a;S;S,-I O  % %a ( ( (:;L;LQ;O;O    (388::    z) ":..!$!3e???r)rLc b||s|S|jdz}|d|j|nd||ni}i|j|}i|jd|i|}|jjdi|}t ||ji|j}|j|kr#|s!| |r |j|_ |S)aReturns 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 arguments are given, 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 >>> Planck13.clone(name="Modified Planck 2013", Om0=0.35) FlatLambdaCDM(name="Modified Planck 2013", H0=67.77 km / (Mpc s), Om0=0.35, ... If no name is specified, the new name will note the modification. >>> Planck13.clone(Om0=0.35).name 'Planck13 (modified)' Nz (modified)rOrLr) rO setdefaultrL_init_argumentsr5 bind_partialtypeargskwargs is_equivalentrK)rNrLr[_modnamenew_metanew_initbacloneds rclonezCosmology.clonesL <<K9},&ty/D88$PPP'ttR(di(4(Gd*GFHGGG/T ! . : : : :dRW2 22 K8 # #T #f6J6J46P6P #9FL rcVfdjD}j|d<j|d<|S)Nc2i|]}|t|Srr.)r(r)rNs r z-Cosmology._init_arguments..s% ? ? ?aaq!! ? ? ?rrOrL)r!rOrL)rNkws` rrWzCosmology._init_argumentss?@ ? ? ?4+> ? ? ?Y6 Y6  rF)formatotherrrhrreturnboolcTddlm} |||d|fdS#t$rYdSwxYw)a= Check 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, positional-only The object to which to compare. format : bool or None or str, optional keyword-only Whether to allow, before equivalence is checked, the object to be converted to a |Cosmology|. This allows, e.g. a |Table| to be equivalent to a Cosmology. `False` (default) will not allow conversion. `True` or `None` will, and will use the auto-identification to try to infer the correct format. A `str` is assumed to be the correct format to use when converting. ``format`` is broadcast to match the shape of ``other``. Note that the cosmology arguments are not broadcast against ``format``, so it cannot determine the output shape. Returns ------- bool True if cosmologies are equivalent, False otherwise. Examples -------- Two cosmologies may be equivalent even if not of the same class. In this examples the ``LambdaCDM`` has ``Ode0`` set to the same value calculated in ``FlatLambdaCDM``. >>> import astropy.units as u >>> from astropy.cosmology import LambdaCDM, FlatLambdaCDM >>> cosmo1 = LambdaCDM(70 * (u.km/u.s/u.Mpc), 0.3, 0.7) >>> cosmo2 = FlatLambdaCDM(70 * (u.km/u.s/u.Mpc), 0.3) >>> cosmo1.is_equivalent(cosmo2) True While in this example, the cosmologies are not equivalent. >>> cosmo3 = FlatLambdaCDM(70 * (u.km/u.s/u.Mpc), 0.3, Tcmb0=3 * u.K) >>> cosmo3.is_equivalent(cosmo2) False Also, using the keyword argument, the notion of equivalence is extended to any Python object that can be converted to a |Cosmology|. >>> from astropy.cosmology import Planck18 >>> tbl = Planck18.to_format("astropy.table") >>> Planck18.is_equivalent(tbl, format=True) True The list of valid formats, e.g. the |Table| in this example, may be checked with ``Cosmology.from_format.list_formats()``. As can be seen in the list of formats, not all formats can be auto-identified by ``Cosmology.from_format.registry``. Objects of these kinds can still be checked for equivalence, but the correct format string must be used. >>> tbl = Planck18.to_format("yaml") >>> Planck18.is_equivalent(tbl, format="yaml") True r )cosmology_equalNT)rhallow_equivalentF)funcsrm Exception)rNrirhrms rr\zCosmology.is_equivalentshF +***** "?eT6NT    55  s  ''cjjurtStjtjko t fdjDS)aCosmology equivalence. Use ``.is_equivalent()`` for actual check! Parameters ---------- other : `~astropy.cosmology.Cosmology` subclass instance, positional-only The object in which to compare. Returns ------- bool or `NotImplemented` `NotImplemented` if ``other`` is from a different class. `True` if ``other`` is of the same class and has matching parameters and parameter values. `False` otherwise. c3K|]8}tjt|t|kV9dSrInpallr.r(krirNs r z&Cosmology.__equiv__..QsbU U  F74##wua'8'88 9 9U U U U U U r)r>NotImplementedsetr"ru)rNris``r __equiv__zCosmology.__equiv__<s ?$. 0 0! !4*++s53K/L/LL QTU U U U U ,U U U R R  rcjjurtSjjkoJtjtjko t fdjD}|S)aCheck equality between Cosmologies. Checks the Parameters and immutable fields (i.e. not "meta"). Parameters ---------- other : `~astropy.cosmology.Cosmology` subclass instance, positional-only The object in which to compare. Returns ------- bool `True` if Parameters and names are the same, `False` otherwise. c3K|]8}tjt|t|kV9dSrIrsrvs rrxz#Cosmology.__eq__..oZwtQ''75!+<+<<==r)r>ryrOrzr"ru)rNrieqs`` r__eq__zCosmology.__eq__Vs ?$. 0 0! ! I # D+,,E4L0M0MM   0  rcjjd}j|djdz }fdjD}|d|zdzS)N(zname="z", c3BK|]}|dt|VdS)=Nre)r(rwrNs rrxz%Cosmology.__repr__..~s;HHqA**a((**HHHHHHrz, ))r>rrOr!join)rNnameleadfmtpss` r__repr__zCosmology.__repr__ysnn1444 9 //// /HHHHHD4GHHH$))E***S00rc  |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.tabler9)r) to_format)rNr9copyr[s r__astropy_table__zCosmology.__astropy_table__s!$t~AA3A&AAAr)NN)rirrhrrjrk)rirrjrk)#rrr__doc__r rLrr from_formatr rr readrwriter!__annotations__r"r-rr5rCpropertyrOabcabstractmethodrTrbrWr\r{rrr __classcell__r>s@rrr9s. 8::D)()<==K&&'899I " !- 0 0D " "> 2 2E')N((((*,,,,, 3 3 3 3 3D]%%%%X@@X@!>>>>>@XEJMMMMMM^    4F111BBBBBBBr) metaclassceZdZUdZded<ded<dfd Ze d d!dZeeddZ e dZ e j d"dZd ddd#fdZfdZxZS)$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. r r"r!r9type[_FlatCosmoT]rjNonecVt|jdSrI)r,r-__nonflatclass__)r9r>s rr-z$FlatCosmologyMixin.__init_subclass__s* !!### rNklstype[_CosmoT] | Nonetype[Cosmology] | Nonec|n|}d|jD}|sfd|jD}t|dkrtd|d|d|sdS|S)aFind the corresponding non-flat class. The class' bases are searched recursively. Parameters ---------- kls : :class:`astropy.cosmology.Cosmology` class or None, optional If `None` (default) this class is searched instead of `kls`. Raises ------ TypeError If more than one non-flat class is found at the same level of the inheritance. This is similar to the error normally raised by Python for an inconsistent method resolution order. Returns ------- type A :class:`Cosmology` subclass this class inherits from that is not a :class:`FlatCosmologyMixin` subclass. Ncfh|].}t|tt|t,|/Sr) issubclassrr)r(bs r z6FlatCosmologyMixin._get_nonflat_cls..sM   !Y'' 1;1>P0Q0Q    rcBh|]}|xSrI)_get_nonflat_cls)r(rr9rws rrz6FlatCosmologyMixin._get_nonflat_cls..s833G3G3J3J.Ja-W-W-W-Wrr z?cannot create a consistent non-flat class resolution order for z with bases z at the same inheritance level.) __bases__len TypeErrorr&)r9r_klsnonflatrws` @rrz#FlatCosmologyMixin._get_nonflat_clss4ksss  ^     >G w<>> from astropy.cosmology import Planck13 >>> Planck13.clone(name="Modified Planck 2013", Om0=0.35) FlatLambdaCDM(name="Modified Planck 2013", H0=67.77 km / (Mpc s), Om0=0.35, ... If no name is specified, the new name will note the modification. >>> Planck13.clone(Om0=0.35).name 'Planck13 (modified)' The keyword 'to_nonflat' can be used to clone on the non-flat equivalent cosmology. >>> Planck13.clone(to_nonflat=True) LambdaCDM(name="Planck13", ... >>> Planck13.clone(H0=70, to_nonflat=True) LambdaCDM(name="Planck13 (modified)", H0=70.0 km / (Mpc s), ... rLr)rrbr,)rNrLrr[r>s rrbzFlatCosmologyMixin.clonesO`  ;%4<%::4:6:: :uww}11$1&111rc^ttr!tSt jjstStj tj ko'tfdj Doj }|S)aflat-|Cosmology| equivalence. Use `astropy.cosmology.funcs.cosmology_equal` with ``allow_equivalent=True`` for actual checks! Parameters ---------- other : `~astropy.cosmology.Cosmology` subclass instance The object to which to compare for equivalence. Returns ------- bool or `NotImplemented` `True` if ``other`` is of the same class / non-flat class (e.g. |FlatLambdaCDM| and |LambdaCDM|) has matching parameters and parameter values. `False` if ``other`` is of the same class but has different parameters. `NotImplemented` otherwise. c3K|]8}tjt|t|kV9dSrIrsrvs rrxz/FlatCosmologyMixin.__equiv__..Gr~r) r/rr,r{rr>rryrzr"rur!rT)rNri params_eqr>s`` rr{zFlatCosmologyMixin.__equiv__$s* e/ 0 0 ,77$$U++ + %/4+@AA "! ! ' ( (C0H,I,I I ,   r)r9rrjrrI)rrrjr)rNrrjr)rLrrrk)rrrrrr- classmethodrrrrrTrrrrbr{rrs@rrrs9('''####)-0000[0d%}t)S X OOOO/3u2222222222222222l+++++++++rcddlm}t||r;|dvr7ddl}ddlm}|d|d|d|t||Std td |d ) Nr )flrw)__path__r)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 .) rhasattrwarningsastropy.utils.exceptionsrwarnr.AttributeErrorr)attrrrrs r __getattr__rUstT #t=88FFFFFF  7t 7 7EI 7 7 7 &    tT""" J8JJJJJ K KKr)' __future__rrrAtypingrrrnumpyrtastropy.io.registryrastropy.utils.decoratorsrastropy.utils.metadatar connectr r r r parameterrcollections.abcr"astropy.cosmology.funcs.comparisonr__all____doctest_requires__dictr8rrrprABCMetarrrrrrrs#""""" ..........666666222222++++++ ! ?''''''>>>>>> @ ? ?TVV '); / / /gm+?@@@      Y   [B[B[B[B[B#+[B[B[B[B| xxxxx3;xxxx|LLLLLr