o  h6@sddlZddlmZmZddlZddlmZGdddeZGdddeZ Gdd d eZ Gd d d Z Gd d d eZ Gddde Z Gddde ZGdddZGdddZdS)N)ABCabstractmethod) MapWrapperc@sJeZdZdZdddZddZddZed d Zed d Z d dZ dS) VertexBasez" Base class for a vertex. NcCs8||_t|j|_|durt||_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)xhashsetnnindexselfrr r r T/var/www/vscode/kcb/lib/python3.10/site-packages/scipy/optimize/_shgo_lib/_vertex.py__init__ s    zVertexBase.__init__cCs|jSN)rr r r r__hash__$szVertexBase.__hash__cCs@|dvrtt|d|d|dkrt|j|_|jSdS)N)x_az object has no attribute ''r)AttributeErrortypenparrayrr)r itemr r r __getattr__'s  zVertexBase.__getattr__cCtdNzKThis method is only implemented with an associated child of the base class.NotImplementedErrorr vr r rconnect/zVertexBase.connectcCrrrrr r r disconnect4r"zVertexBase.disconnectcCs|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 staddrr r rstar9s zVertexBase.starNN) __name__ __module__ __qualname____doc__rrrrr!r#r&r r r rr s    rcsFeZdZdZ  dfdd ZddZdd Zd d Zd d ZZ S)VertexScalarFieldz Add homology properties of a scalar field f: R^n --> R associated with the geometry built from the VertexBase class Nr cs"tj|||dd|_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)r rfieldr r field_argsg_cons g_cons_args __class__r rrQs zVertexScalarField.__init__cCsN||ur#||jvr%|j||j|d|_d|_d|_d|_dSdSdS)zConnects self to another vertex object v. Parameters ---------- v : VertexBase or VertexScalarField object TN)r r%r/r0rr r rr!ts   zVertexScalarField.connectcCsB||jvr|j||j|d|_d|_d|_d|_dSdS)NT)r remover/r0rr r rr#s    zVertexScalarField.disconnectc,jrtfddjD_d_jS)zMCheck whether this vertex is strictly less than all its neighboursc3s|] }j|jkVqdSrf.0r rr r z.VertexScalarField.minimiser..F)r/allr _minrr rr minimiserszVertexScalarField.minimisercr8)z` Check whether this vertex is strictly greater than all its neighbours. c3s|] }j|jkVqdSrr9r;rr rr=r>z.VertexScalarField.maximiser..F)r0r?r _maxrr rr maximiserszVertexScalarField.maximiser)NNNr Nr ) r(r)r*r+rr!r#rArC __classcell__r r r5rr,Ks#  r,cs(eZdZdZ   dfdd ZZS)VertexVectorFieldz Add homology properties of a scalar field f: R^n --> R^m associated with the geometry built from the VertexBase class. Nr c stj||| dtd)Nr-z&This class is still a work in progress)r.rr) r rsfieldvfieldr2 vfield_argsr3r4r r r5r rrszVertexVectorField.__init__)NNr r Nr NN)r(r)r*r+rrDr r r5rrEs rEc@s0eZdZdZddZddZddZdd Zd S) VertexCacheBasez7Base class for a vertex cache for a simplicial complex.cCst|_d|_d|_dS)Nr) collections OrderedDictcachenfevr rr r rrs  zVertexCacheBase.__init__ccs|jD]}|j|VqdSr)rMrr r r__iter__s zVertexCacheBase.__iter__cCs |jdS)z%Returns the size of the vertex cache.r rr r rsizes zVertexCacheBase.sizecCsbtdt|jd}td|tdt|jdtd||jD] }|j|q%dS)NzVertex cache of size: :=)lenrMprint print_out)r headlenr r r rrWs   zVertexCacheBase.print_outN)r(r)r*r+rrOrRrWr r r rrIs  rIcs2eZdZdZd fdd ZddZddZZS) VertexCubezVertex class to be used for a pure simplicial complex with no associated differential geometry (single level domain that exists in R^n)Ncstj|||ddS)Nr-)r.rr r5r rrszVertexCube.__init__cCs6||ur||jvr|j||j|dSdSdSr)r r%rr r rr!s zVertexCube.connectcCs*||jvr|j||j|dSdSr)r r7rr r rr#s  zVertexCube.disconnectr')r(r)r*r+rr!r#rDr r r5rrYs rYcs&eZdZfddZdddZZS)VertexCacheIndexcstt|_dS)z Class for a vertex cache for a simplicial complex without an associated field. Useful only for building and visualising a domain complex. Parameters ---------- N)r.rrYVertexrr5r rrs  zVertexCacheIndex.__init__NcCsRz|j|WSty(|jd7_|j||jd}||j|<|j|YSw)NrPrQ)rMKeyErrorr r[r rr xvalr r r __getitem__s   zVertexCacheIndex.__getitem__r)r(r)r*rr_rDr r r5rrZs  rZcseZdZ  dfdd ZdddZdd Zd d Zd d ZddZddZ ddZ ddZ ddZ ddZ ddZddZZS) VertexCacheFieldNr rPcstd|_t|_||_||_t|||_||_ ||_ t |||_ t |_t |_d|_||_t||_|dkrN|j|_|durH|j|_dS|j|_dS|j|_|dur\|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. rJFrPN)r.rr r,r[r1r2 FieldWrapperwfieldr3r4ConstraintWrapperwgconsrgpoolfpoolsfc_lockworkersr _mapwrapper proc_gpool process_gpoolproc_fpool_nog process_fpool proc_fpool_g pproc_gpoolpproc_fpool_nog pproc_fpool_g)r r1r2r3r4rhr5r rrs.        zVertexCacheField.__init__c Cs|z|j|WSty=|jd7_|j||j||j|j|j|jd}||j|<|j ||j ||j|YSw)NrP)r1r r r2r3r4) rMr\r r[r1r2r3r4rer%rfr]r r rr_0s     zVertexCacheField.__getitem__cCs|j}|d=|S)Npool)__dict__copy)r self_dictr r r __getstate__?s zVertexCacheField.__getstate__cCs&|jdur |||dSr)r3rkrmproc_minimisersrr r r process_poolsDs  zVertexCacheField.process_poolscCsRd|_t|j|jD]\}}t||jg|Rdkr&tj|_d|_dSq dSNTgF) feasiblezipr3r4ranyrinfr:)r r gargsr r rfeasibility_checkJsz"VertexCacheField.feasibility_checkcCsbz|j|jg|jR|_|jd7_Wn ty"tj|_Ynwt|jr/tj|_dSdS)zCompute the scalar field values of a vertex object `v`. Parameters ---------- v : VertexBase or VertexScalarField object rPN) r1rr2r:rNrrr}isnanrr r rcompute_sfieldSs    zVertexCacheField.compute_sfieldcCs,|jdur|jD]}||qt|_dS)zProcess all constraints.N)r3rerrrr r rrjcs    zVertexCacheField.proc_gpoolcCsLg}|jD]}||jq||jj|}t|j|D]\}}||_qdS)z$Process all constraints in parallel.N)reappendrrirdgconsr{rz)r gpool_lr Gr~r r rroks zVertexCacheField.pproc_gpoolcCs(|jD] }|jr ||qt|_dS)z6Process all field functions with constraints supplied.N)rfrzrrrr r rrnus   zVertexCacheField.proc_fpool_gcCs"|jD]}||qt|_dS)z9Process all field functions with no constraints supplied.N)rfrrrr r rrl}s   zVertexCacheField.proc_fpool_nogcCs|jjg}|jD]}|jr||jq tj|_q | |jj|}t ||D]\}}t |}|||_|j d7_ q't |_dS)zT Process all field functions with constraints supplied in parallel. rPN)rbfuncrfrzrrrr}r:rir{tuplerNrr fpool_lr Fvar:vtr r rrqs    zVertexCacheField.pproc_fpool_gcCst|jjg}|jD]}||jq ||jj|}t||D]\}}t|}|||_|j d7_ qt |_dS)zW Process all field functions with no constraints supplied in parallel. rPN) rbrrfrrrir{rr:rNrrr r rrps   z VertexCacheField.pproc_fpool_nogcCs|D] }||qdS)zCheck for minimisers.N)rArCrr r rrws z VertexCacheField.proc_minimisers)Nr Nr rPr)r(r)r*rr_rvrxrrrjrornrlrqrprwrDr r r5rr`s  8  r`c@ eZdZdZddZddZdS)rcz=Object to wrap constraints to pass to `multiprocessing.Pool`.cC||_||_dSr)r3r4)r r3r4r r rr zConstraintWrapper.__init__cCsDd}t|j|jD]\}}t||g|Rdkrd}|Sq |Sry)r{r3r4rr|)r v_x_a vfeasibler~rr r rrszConstraintWrapper.gconsN)r(r)r*r+rrr r r rrc rcc@r)raz7Object to wrap field to pass to `multiprocessing.Pool`.cCrr)r1r2)r r1r2r r rrrzFieldWrapper.__init__cCsFz |j|g|jR}Wn tytj}Ynwt|r!tj}|Sr)r1r2 Exceptionrr}r)r rv_fr r rrs   zFieldWrapper.funcN)r(r)r*r+rrr r r rrarra)rKabcrrnumpyrscipy._lib._utilrrr,rErIrYrZr`rcrar r r rs BZ8