phSrSSKrSSKrSSKrSSKJr SrSrSrSr Sr S r S r S r S r"S S\R5rSrSrSrSrSr"SS\R,5r"SS\R05rSrSrg)a PyPhi- and NumPy-aware JSON serialization. To be properly serialized and deserialized, PyPhi objects must implement a ``to_json`` method which returns a dictionary of attribute names and attribute values. These attributes should be the names of arguments passed to the object constructor. If the constructor takes additional, fewer, or different arguments, the object needs to implement a custom ``classmethod`` called ``from_json`` that takes a Python dictionary as an argument and returns a PyPhi object. For example:: class Phi: def __init__(self, phi): self.phi = phi def to_json(self): return {'phi': self.phi, 'twice_phi': 2 * self.phi} @classmethod def from_json(cls, json): return Phi(json['phi']) The object must also be added to ``jsonify._loadable_models``. The JSON encoder adds the name of the object and the current PyPhi version to the JSON stream. The JSON decoder uses this metadata to recursively deserialize the stream to a nested PyPhi object structure. The decoder will raise an exception if current PyPhi version doesn't match the version in the JSON data. N)cache __class__ __version____id__c[R[R[R[R[R R [RR[RR[RR[RR[RR[RR[RR[RR[RR [RR"[RR$[RR&[RR([RR*[RR,[RR.[RR0[RR2[RR4/nUVs0sHoR6U_M sn$s snf)zA dictionary of loadable PyPhi models. These are stored in this function (instead of module scope) to resolve circular import issues. )pyphi DirectionNetwork Subsystem Transitionlabels NodeLabelsmodelsCutKCutNullCutPart Bipartition KPartition Tripartition RepertoireIrreducibilityAnalysis!MaximallyIrreducibleCauseOrEffectMaximallyIrreducibleCauseMaximallyIrreducibleEffectConceptCauseEffectStructureSystemIrreducibilityAnalysis ActualCut"AcRepertoireIrreducibilityAnalysis CausalLinkAccountAcSystemIrreducibilityAnalysis__name__)classesclss G/home/james-whalen/.local/lib/python3.13/site-packages/pyphi/jsonify.py_loadable_modelsr'3sN              !! 55 66 .. //  )) 11  77   331G4*1 1#LL#  11 1s7IclUR5VVs0sHupU[U5_M snn$s snnfN)itemsjsonify)dctkeyvalues r& _jsonify_dictr/Vs+25))+ >+JCC + >> >s0c UR[URR[[ R [[U505 U$r)) update CLASS_KEYrr# VERSION_KEYrrID_KEYhash)r,objs r&_push_metadatar7Zs=JJ3==))U&&S  Jc6U[U[U[4$r))r2r3r4r,s r& _get_metadatar;cs y>3{+S[ 88r8c~UR[5UR[5UR[54$r))popr2r3r4r:s r& _pop_metadatar>gs) 779 sww{3SWWV_ DDr8c[US5(a&UR5n[X5 [U5$[ U[ R 5(aUR5$[ U[ R[ R45(a [U5$[ U[ R5(a [U5$[ U[5(a [U5$[US5(a[UR5$[ U[ ["45(aUVs/sHn[U5PM sn$U$s snf)zReturn a JSON-encodable representation of an object, recursively using any available ``to_json`` methods, converting NumPy arrays and datatypes to native lists and types along the way. to_json__dict__)hasattrr@r7r+ isinstancenpndarraytolistint32int64intfloat64floatdictr/rAlisttuple)r6ditems r&r+r+ks sI KKMqqz#rzz""zz|#"((+,,3x#rzz""Sz#tS!!sJS\\**#e}%%*-.#$ #.. J/s0E c8^\rSrSrSrU4SjrU4SjrSrU=r$)PyPhiJSONEncoderzCJSONEncoder that allows serializing PyPhi objects with ``jsonify``.c4>[TU][U55$)z:Encode the output of ``jsonify`` with the default encoder.)superencoder+)selfr6rs r&rVPyPhiJSONEncoder.encodesw~gcl++r8c 8>[TU]"[U540UD6$)z%Analog to `encode` used by json.dump.)rU iterencoder+)rWr6kwargsrs r&rZPyPhiJSONEncoder.iterencodesw!'#,9&99r8) r# __module__ __qualname____firstlineno____doc__rVrZ__static_attributes__ __classcell__rs@r&rRrRsM,::r8rRc:S[S.nURU5 U$)z>Update kwargs for `dump` and `dumps` to use the PyPhi encoder.),:) separatorsr%)rRr1) user_kwargsr[s r&_encoder_kwargsrjs&/? @F MM+ Mr8c B[R"U40[U5D6$)z+Serialize ``obj`` as JSON-formatted stream.)jsondumpsrj)r6ris r&rmrms ::c :_[9 ::r8c B[R"X40[U5D6$)znSerialize ``obj`` as a JSON-formatted stream and write to ``fp`` (a ``.write()``-supporting file-like object. )rldumprj)r6fpris r&roros 99S = < ==r8cU[R:wa=[RRSR U[R55eg)z9Check whether the JSON version matches the PyPhi version.z^Cannot load JSON from a different version of PyPhi. JSON version = {0}, current version = {1}.N)rr exceptionsJSONVersionErrorformat)versions r&_check_versionrvsH%###// 99?**:,- -$r8c[U;$)z0Check if ``dct`` is a PyPhi model serialization.)r2r:s r& _is_modelrxs  r8c\rSrSrSrSrSrg) _ObjectCachezCCache mapping ids to loaded objects, keyed by the id of the object.c [U5$r))r;)rWr,r[s r&r-_ObjectCache.keys S!!r8r]N)r#r^r_r`rar-rbr]r8r&rzrzs M"r8rzcb^\rSrSrSrU4SjrSr\R"S5S5r Sr U=r $)PyPhiJSONDecoderzoExtension of the default encoder which automatically deserializes PyPhi JSON to the appropriate model classes. c>URUS'[TU]"U0UD6 [5Ul[ 5Ulg)N object_hook) _load_objectrU__init__r'_modelsrz _object_cache)rWargsr[rs r&rPyPhiJSONDecoder.__init__s= $ 1 1} $)&)() *^r8cB^[U[5(aXUR5VVs0sHup#UTRU5_M nnn[ U5(aTR U5$U$[U[ 5(a[U4SjU55$U$s snnf)aiRecursively load a PyPhi object. PyPhi models are recursively loaded, using the model metadata to recreate the original object relations. Lists are cast to tuples because most objects in PyPhi which are serialized to lists (eg. mechanisms and purviews) are ultimately tuples. Other lists (tpms, repertoires) should be cast to the correct type in init methods. c3F># UHnTRU5v M g7fr))r).0rPrWs r& 0PyPhiJSONDecoder._load_object..sAST**400Ss!)rCrLr*rrx _load_modelrMrN)rWr6kvs` r&rPyPhiJSONDecoder._load_objects c4 7:yy{C{tq1d''**{CC~~'',,  T " "ASAA A DsBrc[U5up#n[U5 URUn[US5(aUR U5$U"S0UD6$)z`Load a serialized PyPhi model. The object is memoized for reuse elsewhere in the object graph. from_jsonr])r>rvrrBr)rWr, classnameru_r%s r&rPyPhiJSONDecoder._load_modelsU !.c 2 Awll9% 3 $ $==% %zSzr8)rr) r#r^r_r`rarrrmethodrrbrcrds@r&rrs/,( \\/"#r8rc4[R"U[S9$)z-Deserialize a JSON string to a Python object.r%)rlloadsr)strings r&rrs ::f"2 33r8c4[R"U[S9$)z-Deserialize a JSON stream to a Python object.r)rlloadr)rps r&rrs 99R- ..r8)rarlnumpyrDrrr2r3r4r'r/r7r;r>r+ JSONEncoderrRrjrmrorvrx DictCacherz JSONDecoderrrrr]r8r&rs<      2F?9E"J :t'' :; >- "5??"3t''3l4 /r8