_-Ph&O dZddlmZddlZddlmZmZmZm Z ddl m Z ddl m Z ddlmZdd lmZmZmZmZmZmZdd lmZmZmZmZmZmZmZd d lm Z m!Z!m"Z"m#Z#d d l$m%Z%d dl&m'Z'dddddddddifdddifddddfddddf dZ( ddZ)dS)z"Support for aggregation-based AMG.)warnN) csr_matrixisspmatrix_csrisspmatrix_bsrSparseEfficiencyWarning)MultilevelSolver)change_smoothers)relaxation_as_linear_operator)scale_Tget_Cpt_paramseliminate_diag_dom_nodes get_blocksize levelize_strength_or_aggregation%levelize_smooth_or_improve_candidates) classical_strength_of_connection symmetric_strength_of_connection evolution_strength_of_connection#energy_based_strength_of_connectiondistance_strength_of_connectionalgebraic_distanceaffinity_distance)standard_aggregationnaive_aggregationlloyd_aggregationpairwise_aggregation)fit_candidates)energy_prolongation_smoother hermitian symmetricstandardenergyblock_gauss_seidelsweep)r% iterations Fc Ft|sWt|sH t|}tdtn"#t $r}t d|d}~wwxYw|}|dvrtd||_ |j d|j dkrtd|xtj tj t|j dt|z df|j tjt|}ntj||j }t'|j dkr|d d}|j d|j dkrtd |j dt|krtd |j d kr||}ntj||j }t'|j dkr|d d}|j d|j dkrtd|j d|j dkrtdt-|| | \} } }t-|| | \} } }t/| | } t/|| }g}|t3j||d _||d _|j d kr ||d _t'|| krt|d jj dt|d jz | krlt=||||| | | t'|| krDt|d jj dt|d jz | klt3|fi|}t?||||S)a;#Create a multilevel solver using root-node based Smoothed Aggregation (SA). See the notes below, for the major differences with the classical-style smoothed aggregation solver in aggregation.smoothed_aggregation_solver. Parameters ---------- A : csr_matrix, bsr_matrix Sparse NxN matrix in CSR or BSR format B : None, array_like Right near-nullspace candidates stored in the columns of an NxK array. K must be >= the blocksize of A (see reference [2011OlScTu]_). The default value B=None is equivalent to choosing the constant over each block-variable, B=np.kron(np.ones((A.shape[0]/get_blocksize(A), 1)), np.eye(get_blocksize(A))) BH : None, array_like Left near-nullspace candidates stored in the columns of an NxK array. BH is only used if symmetry='nonsymmetric'. K must be >= the blocksize of A (see reference [2011OlScTu]_). The default value B=None is equivalent to choosing the constant over each block-variable, B=np.kron(np.ones((A.shape[0]/get_blocksize(A), 1)), np.eye(get_blocksize(A))) symmetry : string 'symmetric' refers to both real and complex symmetric 'hermitian' refers to both complex Hermitian and real Hermitian 'nonsymmetric' i.e. nonsymmetric in a hermitian sense Note that for the strictly real case, symmetric and hermitian are the same Note that this flag does not denote definiteness of the operator. strength : list Method used to determine the strength of connection between unknowns of the linear system. Method-specific parameters may be passed in using a tuple, e.g. strength=('symmetric',{'theta' : 0.25 }). If strength=None, all nonzero entries of the matrix are considered strong. aggregate : list Method used to aggregate nodes. smooth : list Method used to smooth the tentative prolongator. Method-specific parameters may be passed in using a tuple, e.g. smooth= ('energy',{'krylov' : 'gmres'}). Only 'energy' and None are valid prolongation smoothing options. presmoother : tuple, string, list Defines the presmoother for the multilevel cycling. The default block Gauss-Seidel option defaults to point-wise Gauss-Seidel, if the matrix is CSR or is a BSR matrix with blocksize of 1. See notes below for varying this parameter on a per level basis. postsmoother : tuple, string, list Same as presmoother, except defines the postsmoother. improve_candidates : tuple, string, list The ith entry defines the method used to improve the candidates B on level i. If the list is shorter than max_levels, then the last entry will define the method for all levels lower. If tuple or string, then this single relaxation descriptor defines improve_candidates on all levels. The list elements are relaxation descriptors of the form used for presmoother and postsmoother. A value of None implies no action on B. max_levels : integer Maximum number of levels to be used in the multilevel solver. max_coarse : integer Maximum number of variables permitted on the coarse grid. diagonal_dominance : bool, tuple If True (or the first tuple entry is True), then avoid coarsening diagonally dominant rows. The second tuple entry requires a dictionary, where the key value 'theta' is used to tune the diagonal dominance threshold. keep : bool Flag to indicate keeping extra operators in the hierarchy for diagnostics. For example, if True, then strength of connection (C), tentative prolongation (T), aggregation (AggOp), and arrays storing the C-points (Cpts) and F-points (Fpts) are kept at each level. Other Parameters ---------------- cycle_type : ['V','W','F'] Structrure of multigrid cycle coarse_solver : ['splu', 'lu', 'cholesky, 'pinv', 'gauss_seidel', ... ] Solver used at the coarsest level of the MG hierarchy. Optionally, may be a tuple (fn, args), where fn is a string such as ['splu', 'lu', ...] or a callable function, and args is a dictionary of arguments to be passed to fn. Returns ------- ml : MultilevelSolver Multigrid hierarchy of matrices and prolongation operators See Also -------- MultilevelSolver, aggregation.smoothed_aggregation_solver, classical.ruge_stuben_solver Notes ----- - Root-node style SA differs from classical SA primarily by preserving and identity block in the interpolation operator, P. Each aggregate has a 'root-node' or 'center-node' associated with it, and this root-node is injected from the coarse grid to the fine grid. The injection corresponds to the identity block. - Only smooth={'energy', None} is supported for prolongation smoothing. See reference [2011OlScTu]_ below for more details on why the 'energy' prolongation smoother is the natural counterpart to root-node style SA. - The additional parameters are passed through as arguments to MultilevelSolver. Refer to pyamg.MultilevelSolver for additional documentation. - At each level, four steps are executed in order to define the coarser level operator. 1. Matrix A is given and used to derive a strength matrix, C. 2. Based on the strength matrix, indices are grouped or aggregated. 3. The aggregates define coarse nodes and a tentative prolongation operator T is defined by injection 4. The tentative prolongation operator is smoothed by a relaxation scheme to improve the quality and extent of interpolation from the aggregates to fine nodes. - The parameters smooth, strength, aggregate, presmoother, postsmoother can be varied on a per level basis. For different methods on different levels, use a list as input so that the i-th entry defines the method at the i-th level. If there are more levels in the hierarchy than list entries, the last entry will define the method for all levels lower. Examples are: smooth=[('jacobi', {'omega':1.0}), None, 'jacobi'] presmoother=[('block_gauss_seidel', {'sweep':symmetric}), 'sor'] aggregate=['standard', 'naive', 'lloyd', 'pairwise'] strength=[('symmetric', {'theta':0.25}), ('symmetric', {'theta':0.08})] - Predefined strength of connection and aggregation schemes can be specified. These options are best used together, but aggregation can be predefined while strength of connection is not. For predefined strength of connection, use a list consisting of tuples of the form ('predefined', {'C' : C0}), where C0 is a csr_matrix and each degree-of-freedom in C0 represents a supernode. For instance to predefine a three-level hierarchy, use [('predefined', {'C' : C0}), ('predefined', {'C' : C1}) ]. Similarly for predefined aggregation, use a list of tuples. For instance to predefine a three-level hierarchy, use [('predefined', {'AggOp' : Agg0}), ('predefined', {'AggOp' : Agg1}) ], where the dimensions of A, Agg0 and Agg1 are compatible, i.e. Agg0.shape[1] == A.shape[0] and Agg1.shape[1] == Agg0.shape[0]. Each AggOp is a csr_matrix. Because this is a root-nodes solver, if a member of the predefined aggregation list is predefined, it must be of the form ('predefined', {'AggOp' : Agg, 'Cnodes' : Cnodes}). Examples -------- >>> from pyamg import rootnode_solver >>> from pyamg.gallery import poisson >>> from scipy.sparse.linalg import cg >>> import numpy as np >>> A = poisson((100, 100), format='csr') # matrix >>> b = np.ones((A.shape[0])) # RHS >>> ml = rootnode_solver(A) # AMG solver >>> M = ml.aspreconditioner(cycle='V') # preconditioner >>> x, info = cg(A, b, tol=1e-8, maxiter=30, M=M) # solve with CG References ---------- .. [1996VaMa] Vanek, P. and Mandel, J. and Brezina, M., "Algebraic Multigrid by Smoothed Aggregation for Second and Fourth Order Elliptic Problems", Computing, vol. 56, no. 3, pp. 179--196, 1996. http://citeseer.ist.psu.edu/vanek96algebraic.html .. [2011OlScTu] Olson, L. and Schroder, J. and Tuminaro, R., "A general interpolation strategy for algebraic multigrid using energy minimization", SIAM Journal on Scientific Computing (SISC), vol. 33, pp. 966--991, 2011. zImplicit conversion of A to CSRzQArgument A must have type csr_matrix, bsr_matrix, or be convertible to csr_matrixN)r!r nonsymmetriczOExpected "symmetric", "nonsymmetric" or "hermitian" for the symmetry parameter.rrzexpected square matrixdtypez`The near null-space modes B have incorrect dimensions for matrix Az(B.shape[1] must be >= the blocksize of Ar*zlThe number of left and right near null-space modes B and BH, must be equalzeThe near null-space modes BH have incorrect dimensions for matrix A) rrrrr Exception TypeErrorasfptype ValueErrorsymmetryshapenpkrononesintrr,eyeasarraylenreshapecopyrrappendr LevelABBH_extend_hierarchyr )r?r@rAr2strength aggregatesmooth presmoother postsmootherimprove_candidates max_levels max_coarsediagonal_dominancekeepkwargselevelsmls Z/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/aggregation/rootnode.pyrootnode_solverrRs=\ 1  R!2!2R R1 A 2( * * * * R R RJKKPQ R R AAAAFGG GAJwqzQWQZ1222y GBGSM!,<,|j|j} |dj} ||t|dz \} } | dkrt|fi| }n| dkrt|fi| }n| dkrt|fi| }n| dvr!d | vrt|fi| }nt|| fi| }n| d krt|fi| }n| d kr| d }n^| d krt|fi| }nJ| dkrt!|fi| }n6| |}nt#dt%| ||\}} |rt'||fi| }||t|dz \} } | dkrt)|fi| \}}n| dkrt+|fi| \}}nv| dkrt-|fi| \}}n_| dkrt/|fi| \}}nH| d kr#| d}| d}nt#dt%| ||t|dz \} } | yt1j|jddf|j}t9| | f||| z} | |d_|jdkr#t9| | f| || z} | |d_t;|| dddt=|f\}}~|jdkr-t;|| dddt=|f\}}~dt?||||f}tA||dd|dd}|jdkr)tA||dd|dd}|ddj|djz} |jdkr!|ddj|djz} ||t|dz \} } | dkrtC|||| |djfd|i| }n$| |}nt#dt%| |j}|d kr|j}n|dkr|j}n|dkr||t|dz \} } | dkr8tC| ||| |djfd|i| }|j}n;| |j}nt#dt%| |r||d_"||d_#||d_|dd!|d_$|dd|d_%|dd|d_&|dd"|d_'||d_(||d_)|dd#|d_*|+tYj-||z|z}||_||d_| |d_|jdkr| |d_dSdS)$zExtend the multigrid hierarchy. Service routine to implement the strength of connection, aggregation, tentative prolongation construction, and prolongation smoothing. Called by smoothed_aggregation_solver. cTt|tr|d|dfS|ifS)Nrr) isinstancetuple)vs rQ unpack_argz%_extend_hierarchy..unpack_arg>s/ a   Q41: "u r-r*rr! classicaldistance)ode evolutionr@ energy_based predefinedCraffinityNz,Unrecognized strength of connection method: r"naivelloydpairwiseAggOpCnodesz!Unrecognized aggregation method: rr+TP_II_Fr# Cpt_paramsz+Unrecognized prolongation smoother method: r FptsI_CCpts).r?r@r2T conjugateasformatformatrAr:rrrrrtocsrrrr1strrrrrrr4zerosr3r,r rrr r rr`rerjrgrhrkPRrlr=r r>)rOrCrDrErHrKrLrXr?r@AHrAfnrMr`flagrerfbrmdummyTHdummyHrirtr2rus rQrBrB5sp r Ar Az^## S]]__ % %ah / / BZ]HS[[]344JB [ ,Q 9 9& 9 9 {   ,Q 9 9& 9 9 z   +A 8 8 8 8 # # # &==0==f==AA0A@@@@AA ~   / < >JB ~ Hagaj!_AG 4 4 4 )2v,1 = = Ar : ' '.F|RCCbHBF2JM eQqqq!M!,<,<*<'<%=>>HAu z^###E2aaa=3C3C1C.C+DEE F q&%;;>Az^## RAu-z!}U/C D D 1 evbz|+Az^## ]5 ! #F2JM 1F3v;;q=122JB X~~ (Aq!VBZ\ J J4> JBH J J  Ps2wwPPQQQ zH; CMMOO [ C ^ # #Zs6{{1} 566 F >>,RQF2JMNN8BNFLNNA AA Z AAT3r77TTUU U .r  r r $Q-/r #Au-r #Au-r #Au-r F2JLF2JL mF+F2JO MM"(**+++ A AAJF2JLF2JLz^##r $#rY)FT)*__doc__warningsrnumpyr4 scipy.sparserrrr multilevelr relaxation.smoothingr relaxation.utilsr util.utilsr r rrrrrCrrrrrrrrDrrrr tentativerrErrRrBrYrQrs@((*)))))333333<<<<<<**********************************,,,,,,,,,,,,%%%%%%000000$(;(!5");!7!9"6#*K"8":(<2=78):):(; "b',5XXXXx6:WWWWWWrY