_-PhYdZddlmZddlmZmZmZddlZddl m Z ddl m Z ddl mZmZmZmZmZmZmZddlmZmZd d lmZd d lmZd d difdddifd dddifdddifdddfdZdZdS)z Classical AMG (Ruge-Stuben AMG).)warn) csr_matrixisspmatrix_csrSparseEfficiencyWarningN)MultilevelSolver)change_smoothers) classical_strength_of_connection symmetric_strength_of_connection evolution_strength_of_connectiondistance_strength_of_connection#energy_based_strength_of_connectionalgebraic_distanceaffinity_distance)direct_interpolationclassical_interpolation)split)CR classicalthetag?RS second_passF gauss_seidelsweep symmetric c  tjg} t|sH t|}t dt n"#t $r} td| d} ~ wwxYw|}|j d|j dkrtd|| d_ t| |kra| dj j d|krEt| ||||} | rn/t| |kr| dj j d|kEt| fi| } t| ||| S)am Create a multilevel solver using Classical AMG (Ruge-Stuben AMG). Parameters ---------- A : csr_matrix Square matrix in CSR format strength : str Valid strings are ['symmetric', 'classical', 'evolution', 'distance', 'algebraic_distance','affinity', 'energy_based', None]. 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. CF : str or tuple, default 'RS' Method used for coarse grid selection (C/F splitting) Supported methods are RS, PMIS, PMISc, CLJP, CLJPc, and CR. interpolation : str, default 'classical' Method for interpolation. Options include 'direct', 'classical'. presmoother : str or dict Method used for presmoothing at each level. Method-specific parameters may be passed in using a tuple, e.g. presmoother=('gauss_seidel',{'sweep':'symmetric}), the default. postsmoother : str or dict Postsmoothing method with the same usage as presmoother max_levels : int, default 30 Maximum number of levels to be used in the multilevel solver. max_coarse : int, default 20 Maximum number of variables permitted on the coarse grid. keep : bool, default False Flag to indicate keeping strength of connection (C) in the hierarchy for diagnostics. Returns ------- ml : MultilevelSolver Multigrid hierarchy of matrices and prolongation operators Examples -------- >>> from pyamg.gallery import poisson >>> from pyamg import ruge_stuben_solver >>> A = poisson((10,),format='csr') >>> ml = ruge_stuben_solver(A,max_coarse=3) Notes ----- "coarse_solver" is an optional argument and is the solver used at the coarsest grid. The default is a pseudo-inverse. Most simply, coarse_solver can be one of ['splu', 'lu', 'cholesky, 'pinv', 'gauss_seidel', ... ]. Additionally, coarse_solver 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. See [2001TrOoSc]_ for additional details. References ---------- .. [2001TrOoSc] Trottenberg, U., Oosterlee, C. W., and Schuller, A., "Multigrid" San Diego: Academic Press, 2001. Appendix A See Also -------- aggregation.smoothed_aggregation_solver, MultilevelSolver, aggregation.rootnode_solver zImplicit conversion of A to CSRzEArgument A must have type csr_matrix, or be convertible to csr_matrixNrrzexpected square matrix)rLevelrrrr Exception TypeErrorasfptypeshape ValueErrorAlen_extend_hierarchyr)r&strengthCF interpolation presmoother postsmoother max_levels max_coarsekeepkwargslevelsebottommls Y/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/classical/classical.pyruge_stuben_solverr7swP$&& 'F !  F F1 A 2( * * * * F F F>??DE F F AwqzQWQZ1222F2JL f++ " "vbz|'9!'.unpack_arg{s/ a   Q41: "u rrrdistance)ode evolution energy_basedraffinityNz,Unrecognized strength of connection method: rPMISPMIScCLJPCLJPcrzUnknown C/F splitting method rTdirectzUnknown interpolation method F) r&r r r r r rrr%rrrDrErFrGrnpsumr'rrTtocsrCastypebool splittingPRappendrr )r2r)r*r+r0r=r&fnr1rMrPnum_fptsrQrRs r6r(r(ysl r AH%%JB [ ,Q 9 9& 9 9 {   ,Q 9 9& 9 9 z   +A 8 8 8 8 # # # ,Q 9 9& 9 9 ~   / < >>vi  HC NN""A tM**JB [ #Aq) > >v > > x Ay ; ;F ; ;HHHIII  A r $++D11F2JF2JLF2JL MM"(**+++ A AF2JL 5r>)__doc__warningsr scipy.sparserrrnumpyrIpyamg.multilevelrpyamg.relaxation.smoothingrpyamg.strengthr r r r r rrpyamg.classical.interpolaterrrcrrr7r(r>r6ras|&&LLLLLLLLLL------777777******************VUUUUUUU"-wo >-!78%0$2Wk4J#K%3g{5K$L"$% bbbbLPPPPPr>