WIP(cgraph):add linear_elasticity_eigen_3d_example

Author: Hukai0

example/cgraph/linear_elasticity_eigen_3d_example.py

@@ -4,28 +4,25 @@
 WORLD_GRAPH = cgraph.WORLD_GRAPH
 
 pde = cgraph.create("LinearElasticityEigen3d")            # PDE 数据（示例）
-mesher = cgraph.create("uniform_tet")                      # 网格生成
-spacer = cgraph.create("FunctionSpace")        # 函数空间节点
+mesher = cgraph.create("Box3d")                      # 网格生成
+spacer = cgraph.create("TensorFunctionSpace")        # 函数空间节点
 isDDof = cgraph.create("BoundaryDof")
 eig_eq = cgraph.create("LinearElasticityEigenEquation")
 eigensolver = cgraph.create("EigenSolver")
 dbc = cgraph.create("DirichletBC")
 
-spacer(mesh=mesher(), p=1)
+spacer(mesh=mesher(), p=1,gd=3)
 
 eig_eq(space=spacer(), q=3, material = pde().material, displacement_bc = pde().displacement_bc, is_displacement_boundary = pde().is_displacement_boundary)
 
 eigensolver(
-    S=eig_eq().S,
-    M=eig_eq().M,
-    space=spacer(),
-
+    S=eig_eq().stiffness,
+    M=eig_eq().mass,
     neigen=6,
     which='SM'
 )
 
-# 输出：把网格和第一个模态输出到 WORLD_GRAPH
-WORLD_GRAPH.output(mesh=mesher(), uh=eigensolver().modes[0])
-WORLD_GRAPH.register_error_hook(print)
+WORLD_GRAPH.output(eig_eq=eigensolver().val, uh=eigensolver().vec)
+WORLD_GRAPH.error_listeners.append(print)
 WORLD_GRAPH.execute()
-print(WORLD_GRAPH.get())
+print(WORLD_GRAPH.get())

fealpy/cgraph/fem/__init__.py

@@ -8,3 +8,4 @@
 from .timoaxle import *
 from .truss import *
 from .beam import *
+from .linear_elasticity_eigen import *

fealpy/cgraph/fem/linear_elasticity_eigen.py

@@ -9,7 +9,7 @@ class LinearElasticityEigenEquation(CNodeType):
     INPUT_SLOTS = [
         PortConf("space", DataType.SPACE, "函数空间"),
         PortConf("q", DataType.INT, title="积分公式", default=3, min_val=1, max_val=17),
-        PortConf("material", DataType.OBJECT, title="材料属性"),
+        PortConf("material", DataType.MENU, title="材料属性"),
         PortConf("displacement_bc", DataType.FUNCTION, title="位移边界条件函数"),
         PortConf("is_displacement_boundary", DataType.FUNCTION, title="位移边界标识函数"),
     ]
@@ -25,7 +25,7 @@ def run(space, q: int, material, displacement_bc, is_displacement_boundary):
         from ...fem import LinearElasticityIntegrator
         from ...fem import ScalarMassIntegrator as MassIntegrator
         from ...fem import DirichletBC
-        from ...backend import backend as bm
+        from ...backend import backend_manager as bm
 
 
         bform_S = BilinearForm(space)

fealpy/cgraph/model/__init__.py

@@ -9,3 +9,4 @@
 from .incompressible_navier_stokes2d import IncompressibleCylinder2d
 from .dipole_antenna3d import DipoleAntenna3D
 from .beam2d import Beam2d
+from .linear_elasticity_eigen3d import LinearElasticityEigen3d

fealpy/cgraph/model/linear_elasticity_eigen3d.py

@@ -7,8 +7,6 @@ class LinearElasticityEigen3d(CNodeType):
     OUTPUT_SLOTS = [
         PortConf("domain", DataType.NONE),
         PortConf("material", DataType.NONE),
-        PortConf("displacement", DataType.FUNCTION),
-        PortConf("body_force", DataType.FUNCTION),
         PortConf("displacement_bc", DataType.FUNCTION),
         PortConf("is_displacement_boundary", DataType.FUNCTION),
     ]
@@ -20,7 +18,7 @@ def run():
         model = CSMModelManager("linear_elasticity").get_example(1)
         return (domain, model.material)+ tuple(
             getattr(model, name)
-            for name in ["displacement", "displacement_bc"]
+            for name in [ "displacement_bc",  "is_displacement_boundary"]
         )
 
 

fealpy/cgraph/solver.py

@@ -90,4 +90,31 @@ def run(*args, **kwargs):
         mgr.set_matrix(A)
         mgr.set_tolerances(rtol=rtol, atol=atol, maxit=maxit)
         x = mgr.solve(b)
-        return x
+        return x
+    
+from .nodetype import CNodeType, PortConf, DataType
+
+__all__ = ["EigenSolver"]
+
+class EigenSolver(CNodeType):
+    TITLE: str = "特征值求解器"
+    PATH: str = "解法器.特征值"
+    INPUT_SLOTS = [
+        PortConf("S", DataType.TENSOR, title="刚度矩阵 S"),
+        PortConf("M", DataType.TENSOR, title="质量矩阵 M"),
+        PortConf("neigen", DataType.INT, title="求取的特征值个数", default=6, min_val=1),
+        PortConf("which", DataType.STRING, title="eigsh which", default='SM'),
+    ]
+    OUTPUT_SLOTS = [
+        PortConf("val", DataType.TENSOR, title="特征值"),
+        PortConf("vec", DataType.TENSOR, title="特征向量"),
+    ]
+    
+    @staticmethod    
+    def run(*args, **kwargs):
+        from scipy.sparse.linalg import eigsh
+        S = kwargs.get('S')
+        M = kwargs.get('M')
+        neigen = kwargs.get('neigen')
+        val, vec = eigsh(S, k=neigen, M=M, which=kwargs.get('which', 'SM'), tol=1e-6, maxiter=1000)
+        return val, vec