ngsolve 6.2.2506.post216.dev0__cp314-cp314-manylinux_2_28_x86_64.whl

ngsolve-6.2.2506.post216.dev0.data/data/lib/python3.14/site-packages/ngsolve/utils.py

@@ -0,0 +1,167 @@
+from ngsolve.ngstd import IntRange
+from ngsolve.fem import *
+from ngsolve.comp import *
+from ngsolve.bla import Norm
+from netgen import TimeFunction, Timer
+
+x = CoordCF(0)
+y = CoordCF(1)
+z = CoordCF(2)
+
+dx = DifferentialSymbol(VOL)
+ds = DifferentialSymbol(BND)
+
+
+def Laplace (coef):
+    return BFI("laplace", coef=coef)
+
+def Mass (coef):
+    return BFI("mass", coef=coef)
+
+def Source (coef):
+    return LFI("source", coef=coef)
+
+def Neumann (coef):
+    return LFI("neumann", coef=coef)
+
+
+# VectorFacet = TangentialFacetFESpace
+def VectorFacet (mesh, **args):
+    print ("deprecated warning: VectorFacet is renamed to TangentialFacetFESpace")
+    return TangentialFacetFESpace(mesh, **args)
+
+def grad(func):
+    try:
+        # now in C++ code
+        # if "normal vector" in str(func):
+        # return specialcf.Weingarten(func.dim)
+        if func.derivname == "grad":
+            return func.Deriv()
+    except:
+        pass
+    add = func.Operator("grad")
+    if add:
+        return add        
+    #     if func.derivname != "grad":
+    raise Exception("cannot form grad")
+    # return func.Deriv()
+
+def Grad(func):
+    """ Jacobi-matrix"""
+    # now in C++ code    
+    # if "normal vector" in str(func):        
+    # return specialcf.Weingarten(func.dim)
+    try:
+        return func.Operator("Grad")
+    except:
+        return grad(func).trans
+    
+
+
+def curl(func):
+    if func.derivname == "curl":
+        return func.Deriv()
+    add = func.Operator("curl")
+    if add:
+        return add
+    raise Exception("cannot form curl")    
+
+def div(func):
+    if func.derivname == "div":
+        return func.Deriv()
+    try:
+        return func.Operator("div")
+    except:
+        pass
+    try:
+        return Trace(grad(func))
+    except:
+        pass
+    #if func.derivname == "grad" and len(func.dims)==2:  # should check for square
+    #   return Trace(grad(func))
+    raise Exception("cannot form div")    
+
+
+def ConstantCF(val):
+    print ("Warning: ConstantCF deprecated, just use CoefficientFunction(val)")
+    return CoefficientFunction(val)
+
+def DomainConstantCF(values):
+    print ("Warning: DomainConstantCF deprecated, just use CoefficientFunction([values])")
+    return CoefficientFunction(values)
+
+
+def Deviator(mat):
+    return mat - 1/mat.dims[0]*Trace(mat)*Id(mat.dims[0])
+
+
+def PyId(dim):
+    return CoefficientFunction( tuple( [1 if i==j else 0 for i in range(dim) for j in range(dim)]), dims=(dim,dim) )
+
+def PyTrace(mat):
+    return sum( [mat[i,i] for i in range(mat.dims[0]) ])
+
+def PyDet(mat):
+    if mat.dims[0] == 1:
+        return mat[0,0]
+    elif mat.dims[0] == 2:
+        return mat[0,0]*mat[1,1]-mat[0,1]*mat[1,0]
+    elif mat.dims[0] == 3:
+        return mat[0,0]*(mat[1,1]*mat[2,2]-mat[1,2]*mat[2,1]) \
+              +mat[1,0]*(mat[2,1]*mat[0,2]-mat[2,2]*mat[0,1]) \
+              +mat[2,0]*(mat[0,1]*mat[1,2]-mat[0,2]*mat[1,1])
+
+def PyCross(a,b):
+    return CoefficientFunction( (a[1]*b[2]-a[2]*b[1],a[2]*b[0]-a[0]*b[2],a[0]*b[1]-a[1]*b[0]) )
+
+def PyCof(m):
+    if m.dims[0] == 1:
+        return CoefficientFunction(1, dims=(1,1))
+    elif m.dims[0] == 2:
+        return CoefficientFunction( (m[1,1], -m[1,0], -m[0,1], m[0,0]), dims=(2,2) )
+    elif m.dims[0] == 3:
+        return CoefficientFunction( 
+                    ( m[1,1]*m[2,2]-m[2,1]*m[1,2],
+                     -m[1,0]*m[2,2]+m[2,0]*m[1,2],
+                      m[1,0]*m[2,1]-m[2,0]*m[1,1],
+    
+                     -m[0,1]*m[2,2]+m[2,1]*m[0,2],
+                      m[0,0]*m[2,2]-m[2,0]*m[0,2], 
+                     -m[0,0]*m[2,1]+m[2,0]*m[0,1],
+    
+                      m[0,1]*m[1,2]-m[1,1]*m[0,2], 
+                     -m[0,0]*m[1,2]+m[1,0]*m[0,2], 
+                      m[0,0]*m[1,1]-m[1,0]*m[0,1] ), dims=(3,3) )
+
+def PyInv(m):
+    return 1/Det(m)*Cof(m).trans
+
+def PySym(m):
+    return 0.5*(m+m.trans)
+
+def PySkew(m):
+    return 0.5*(m-m.trans)
+
+def OuterProduct(a, b):
+    # return CoefficientFunction( (a,), dims=(a.dim,1)) * CoefficientFunction( (b,), dims=(b.dim,1)).trans
+    # return CoefficientFunction(a, dims=(a.dim,1)) * CoefficientFunction(b, dims=(1, b.dim))
+    return a.Reshape((a.dim,1)) * b.Reshape((1, b.dim))
+
+
+PrivateSpace = Hidden
+# def PrivateSpace(fes):
+# TODO: make space wrapper which also works after refinement
+# fes.SetCouplingType(IntRange(0,fes.ndof), COUPLING_TYPE.HIDDEN_DOF)
+# return Compress(fes)
+# return Hidden(fes)
+
+def Normalize (v):
+    return 1/Norm(v) * v
+
+def printonce (*args):
+    from mpi4py.MPI import COMM_WORLD
+    if COMM_WORLD.rank == 0:
+        print (*args)
+        
+
+def dt(u): return u.dt