1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
def RunTest():
from petsc4py import PETSc
import example1
OptDB = PETSc.Options()
N = OptDB.getInt('N', 100)
draw = OptDB.getBool('draw', False)
A = PETSc.Mat()
A.create(comm=PETSc.COMM_WORLD)
A.setSizes([N,N])
A.setType(PETSc.Mat.Type.PYTHON)
A.setPythonContext(example1.Laplace1D())
A.setUp()
x, b = A.getVecs()
b.set(1)
ksp = PETSc.KSP()
ksp.create(comm=PETSc.COMM_WORLD)
ksp.setType(PETSc.KSP.Type.PYTHON)
ksp.setPythonContext(example1.ConjGrad())
pc = ksp.getPC()
pc.setType(PETSc.PC.Type.PYTHON)
pc.setPythonContext(example1.Jacobi())
ksp.setOperators(A, A)
ksp.setFromOptions()
ksp.solve(b, x)
r = b.duplicate()
A.mult(x, r)
r.aypx(-1, b)
rnorm = r.norm()
PETSc.Sys.Print('error norm = %g' % rnorm,
comm=PETSc.COMM_WORLD)
if draw:
viewer = PETSc.Viewer.DRAW(x.getComm())
x.view(viewer)
PETSc.Sys.sleep(2)
if __name__ == '__main__':
import sys, petsc4py
petsc4py.init(sys.argv)
RunTest()
|
