def corrector_y(u, b1, b2, a1, a2):
Bu = matrix(u.line + 1, u.col + 1)
Bu.clear()
for i in xrange(Bu.line):
Bu.data[i][0] = u.data[i][0] * b1
Bu.data[i][Bu.col - 1] = u.data[i][u.col - 1]*b2
for j in xrange(1, Bu.col - 1):
Bu.data[i][j] = u.data[i][j]*b1 + u.data[i][j-1]*b2
#progonka A
y = matrix(Bu.line, Bu.col)
y.clear()
gradu = matrix(Bu.line, Bu.col)
gradu.clear()
alpha = vector(Bu.col)
beta = vector(Bu.col)
for i in xrange(y.line):
alpha.data[0] = a1
beta.data[0] = a2/alpha.data[0]
y.data[i][0] = Bu.data[i][0]/alpha.data[0]
for j in xrange(1, y.col - 1):
alpha.data[j] = 2*a1 - a2*beta.data[j - 1]
beta.data[j] = a2/alpha.data[j]
y.data[i][j] = (Bu.data[i][j] - a2*y.data[i][j -1])/alpha.data[j]
alpha.data[y.col - 1] = a1 - a2*beta.data[y.col - 2]
y.data[i][y.col - 1] = (Bu.data[i][y.col - 1] - a2*y.data[i][y.col - 2])/alpha.data[y.col - 1]
gradu.data[i][y.col - 1] = y.data[i][y.col - 1]
for j in range(y.col - 2, -1, -1):
gradu.data[i][j] = y.data[i][j] - beta.data[j]*gradu.data[i][j + 1]
gradu.printmatrix()
return gradu
def operator_Ax(u):
new_matrix = matrix(amnty, amntx + 1)
new_matrix.clear()
new_solution = matrix(amnty, amntx + 1)
new_solution.clear()
alpha = vector(amntx + 1)
beta = vector(amntx + 1)
for i in xrange(new_matrix.line):
alpha.data[0] = hy*hx/3
beta.data[0] = hy*hx/6/alpha.data[0]
new_matrix.data[i][0] = u.data[i][0]/alpha.data[0]
for j in xrange(1, new_matrix.col - 1):
alpha.data[j] = 2*hy*hx/3 - hy*hx/6*beta.data[j - 1]
beta.data[j] = hy*hx/6/alpha.data[j]
new_matrix.data[i][j] = (u.data[i][j] - hy*hx/6*new_matrix.data[i][j -1])/alpha.data[j]
alpha.data[new_matrix.col - 1] = hy*hx/3 - hy*hx/6*beta.data[new_matrix.col - 2]
new_matrix.data[i][new_matrix.col - 1] = (u.data[i][new_matrix.col - 1] - (hy*hx/6)*new_matrix.data[i][new_matrix.col - 2])/alpha.data[new_matrix.col - 1]
new_solution.data[i][new_matrix.col - 1] = new_matrix.data[i][new_matrix.col - 1]
for j in range(new_matrix.col - 2, -1, -1):
new_solution.data[i][j] = new_matrix.data[i][j] - beta.data[j]*new_solution.data[i][j + 1]
new_solution.printmatrix()
return new_solution
