Ui6(ddlZddlmZmZddlZddlmZGddeZGddeZ Gdd eZ Gd d Z Gd d eZ Gdde Z Gdde ZGddZGddZdS)N)ABCabstractmethod) MapWrappercXeZdZdZd dZdZdZedZedZ dZ dS) VertexBasez" Base class for a vertex. Nc||_t|j|_|t||_nt|_||_dS)a Initiation of a vertex object. Parameters ---------- x : tuple or vector The geometric location (domain). nn : list, optional Nearest neighbour list. index : int, optional Index of vertex. N)xhashsetnnindex)selfr r r s x/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/scipy/optimize/_shgo_lib/_vertex.py__init__zVertexBase.__init__ sDLL >"ggDGGeeDG c|jSN)r rs r__hash__zVertexBase.__hash__$s yrc|dvr"tt|d|d|dkr%tj|j|_|jSdS)N)x_az object has no attribute ''r)AttributeErrortypenparrayr r)ritems r __getattr__zVertexBase.__getattr__'sm w   DJJ"-"-%)"-"-"-.. . 5==x''DH8O =rc tdNzKThis method is only implemented with an associated child of the base class.NotImplementedErrorrvs rconnectzVertexBase.connect/!#HII Irc tdr r!r#s r disconnectzVertexBase.disconnect4r&rc\|j|_|j||jS)aReturns the star domain ``st(v)`` of the vertex. Parameters ---------- v : The vertex ``v`` in ``st(v)`` Returns ------- st : set A set containing all the vertices in ``st(v)`` )r staddrs rstarzVertexBase.star9s('  DwrNN) __name__ __module__ __qualname____doc__rrrrr%r(r,rrrr s.II^III^Irrc@eZdZdZ d fd ZdZdZdZdZxZ S) VertexScalarFieldz Add homology properties of a scalar field f: R^n --> R associated with the geometry built from the VertexBase class Nr2clt|||d|_d|_dS)aY Parameters ---------- x : tuple, vector of vertex coordinates field : callable, optional a scalar field f: R^n --> R associated with the geometry nn : list, optional list of nearest neighbours index : int, optional index of the vertex field_args : tuple, optional additional arguments to be passed to field g_cons : callable, optional constraints on the vertex g_cons_args : tuple, optional additional arguments to be passed to g_cons r r TN)superr check_min check_max) rr fieldr r field_argsg_cons g_cons_args __class__s rrzVertexScalarField.__init__Qs8* r///rc||ur[||jvrT|j||j|d|_d|_d|_d|_dSdSdS)zConnects self to another vertex object v. Parameters ---------- v : VertexBase or VertexScalarField object TN)r r+r8r9r#s rr%zVertexScalarField.connecttsj D==Qdg-- GKKNNN DHHTNNN"DN!DNAKAKKK =--rc||jvrR|j||j|d|_d|_d|_d|_dSdS)NT)r remover8r9r#s rr(zVertexScalarField.disconnects^ << GNN1    DKK   "DN!DNAKAKKK z.VertexScalarField.minimiser..+::QDFQSL::::::rF)r8allr _minrs`r minimiserzVertexScalarField.minimisersD > #::::$':::::DI"DNyrcxjr,tfdjD_d_jS)z` Check whether this vertex is strictly greater than all its neighbours. c38K|]}j|jkVdSrrDrFs rrHz.VertexScalarField.maximiser..rIrF)r9rJr _maxrs`r maximiserzVertexScalarField.maximisersD > #::::$':::::DI"DNyr)NNNr2Nr2) r.r/r0r1rr%r(rLrP __classcell__r>s@rr4r4Ks GI*,!!!!!!F"          rr4c*eZdZdZ dfd ZxZS)VertexVectorFieldz Add homology properties of a scalar field f: R^n --> R^m associated with the geometry built from the VertexBase class. Nr2c jt||| td)Nr6z&This class is still a work in progress)r7rr") rr sfieldvfieldr; vfield_argsr<r=r r r>s rrzVertexVectorField.__init__s3 r///!"JKKKr)NNr2r2Nr2NN)r.r/r0r1rrQrRs@rrTrTs^ @B(,04LLLLLLLLLLrrTc*eZdZdZdZdZdZdZdS)VertexCacheBasez7Base class for a vertex cache for a simplicial complex.cRtj|_d|_d|_dS)Nr) collections OrderedDictcachenfevr rs rrzVertexCacheBase.__init__s$ ,..   rc#<K|jD]}|j|VdSr)r_r#s r__iter__zVertexCacheBase.__iter__s1  A*Q-    rc|jdzS)z%Returns the size of the vertex cache.r rs rsizezVertexCacheBase.sizeszA~rc4tdt|jd}td|ztdt|jdtd|z|jD]!}|j|"dS)NzVertex cache of size: :=)lenr_print print_out)rheadlenr$s rrlzVertexCacheBase.print_outsAs4:AAABB cGm 9s4:999::: cGm & &A JqM # # % % % % & &rN)r.r/r0r1rrbrfrlr2rrrZrZsVAA  &&&&&rrZc0eZdZdZdfd ZdZdZxZS) VertexCubezVertex class to be used for a pure simplicial complex with no associated differential geometry (single level domain that exists in R^n)NcPt|||dS)Nr6)r7r)rr r r r>s rrzVertexCube.__init__s( r/////rc||ur?||jvr8|j||j|dSdSdSr)r r+r#s rr%zVertexCube.connectsJ D==Qdg-- GKKNNN DHHTNNNNN =--rc||jvr6|j||j|dSdSr)r rAr#s rr(zVertexCube.disconnectsB << GNN1    DKK      s rrzVertexCacheIndex.__init__s&   rNc |j|S#t$rF|xjdz c_|||j}||j|<|j|cYSwxYw)Nrdre)r_KeyErrorr rvrr r xvals r __getitem__zVertexCacheIndex.__getitem__ss !:a=  ! ! ! JJ!OJJ;;q ;33D!DJqM:a=  !s A AAr)r.r/r0rr{rQrRs@rrtrtsL ! ! ! ! ! ! ! ! ! ! ! ! !rrtcneZdZ dfd ZddZdZdZdZd Zd Z d Z d Z d Z dZ dZdZxZS)VertexCacheFieldNr2rdc:td|_t|_||_||_t|||_||_ ||_ t|||_ t|_t|_d|_||_t%||_|dkr*|j|_||j|_dS|j|_dS|j|_||j|_dS|j|_dS)a Class for a vertex cache for a simplicial complex with an associated field. Parameters ---------- field : callable Scalar or vector field callable. field_args : tuple, optional Any additional fixed parameters needed to completely specify the field function g_cons : dict or sequence of dict, optional Constraints definition. Function(s) ``R**n`` in the form:: g_cons_args : tuple, optional Any additional fixed parameters needed to completely specify the constraint functions workers : int optional Uses `multiprocessing.Pool `) to compute the field functions in parallel. r\FrdN)r7rr r4rvr:r; FieldWrapperwfieldr<r=ConstraintWrapperwgconsr gpoolfpoolsfc_lockworkersr _mapwrapper proc_gpool process_gpoolproc_fpool_nog process_fpool proc_fpool_g pproc_gpoolpproc_fpool_nog pproc_fpool_g)rr:r;r<r=rr>s rrzVertexCacheField.__init__s 0  '  $"5*55  &' << UU UU   %g.. a<|j}|d=|S)Npool)__dict__copy)r self_dicts r __getstate__zVertexCacheField.__getstate__?s#M&&(( f rc|j|||dSr)r<rrproc_minimisersrs r process_poolszVertexCacheField.process_poolsDsF ; "      rcd|_t|j|jD]D\}}t j||jg|Rdkrtj|_d|_dSEdSNTgF) feasiblezipr<r=ranyrinfrE)rr$gargss rfeasibility_checkz"VertexCacheField.feasibility_checkJsx 4;(899  GAtvaaoooo+,, f"    rc |j|jg|jR|_|xjdz c_n!#t $rt j|_YnwxYwt j|jrt j|_dSdS)zCompute the scalar field values of a vertex object `v`. Parameters ---------- v : VertexBase or VertexScalarField object rdN) r:rr;rEr`rrrisnanr#s rcompute_sfieldzVertexCacheField.compute_sfieldSs $*QU5T_555AC IINIII   &ACCC  8AC== &ACCC  s-0A Acx|j|jD]}||t|_dS)zProcess all constraints.N)r<rrr r#s rrzVertexCacheField.proc_gpoolcsA ; "Z * *&&q))))UU rcg}|jD]}||j||jj|}t |j|D] \}}||_ dS)z$Process all constraints in parallel.N)rappendrrrgconsrr)rgpool_lr$Grs rrzVertexCacheField.pproc_gpoolks{ " "A NN15 ! ! ! !   T[. 8 8 A&&  DAqAJJ  rcx|jD]}|jr||t|_dS)z6Process all field functions with constraints supplied.N)rrrr r#s rrzVertexCacheField.proc_fpool_gusA ' 'Az '##A&&&UU rcj|jD]}||t|_dS)z9Process all field functions with no constraints supplied.N)rrr r#s rrzVertexCacheField.proc_fpool_nog}s9 # #A    " " " "UU rc|jjg}|jD]5}|jr||j$t j|_6| |jj|}t||D]1\}}t|}|||_|xj dz c_ 2t|_dS)zT Process all field functions with constraints supplied in parallel. rdN)rfuncrrrrrrrErrtupler`r rfpool_lr$FvarEvts rrzVertexCacheField.pproc_fpool_gs   Az qu%%%%f   T[-w 7 7!__  EBrBDHJ IINIIIUU rcT|jjg}|jD]}||j||jj|}t ||D]1\}}t|}|||_|xj dz c_ 2t|_dS)zW Process all field functions with no constraints supplied in parallel. rdN) rrrrrrrrrEr`r rs rrz VertexCacheField.pproc_fpool_nogs  " "A NN15 ! ! ! !   T[-w 7 7!__  EBrBDHJ IINIIIUU rc`|D]*}||+dS)zCheck for minimisers.N)rLrPr#s rrz VertexCacheField.proc_minimiserss7  A KKMMM KKMMMM  r)Nr2Nr2rdr)r.r/r0rr{rrrrrrrrrrrrQrRs@rr}r}sKM686868686868p ! ! ! !   & rr}ceZdZdZdZdZdS)rz=Object to wrap constraints to pass to `multiprocessing.Pool`.c"||_||_dSr)r<r=)rr<r=s rrzConstraintWrapper.__init__s &rcd}t|j|jD](\}}tj||g|Rdkrd}n)|Sr)rr<r=rr)rv_x_a vfeasiblerrs rrzConstraintWrapper.gconsse 4;(899  GAtvaaoooo+,, !  rN)r.r/r0r1rrr2rrrrs8GG'''rrceZdZdZdZdZdS)rz7Object to wrap field to pass to `multiprocessing.Pool`.c"||_||_dSr)r:r;)rr:r;s rrzFieldWrapper.__init__s $rc |j|g|jR}n#t$rtj}YnwxYwtj|r tj}|Sr)r:r; Exceptionrrr)rrv_fs rrzFieldWrapper.funcsf $*U5T_555CC   &CCC  8C== &C s //N)r.r/r0r1rrr2rrrrs8AA%%%rr)r]abcrrnumpyrscipy._lib._utilrrr4rTrZrortr}rrr2rrrs########''''''????????DWWWWW WWWt L L L L L L L L&&&&&&&&4"!!!!!!!!2ttttttttn         r