b[eps_m].coeffs += beta[1] * g0
g0 = pde.assemble_rhs(am_f, basis=b[zero].basis, f=Constant(0.0))
g0[dofs] = 0
b[zero].coeffs += beta[0] * g0
# b[eps_m].coeffs[dofs] += beta[1] * pde.assemble_rhs(am_f, basis=b[eps_m].basis, f=Constant(0.0))[dofs]
# b[zero].coeffs[dofs] += beta[0] * pde.assemble_rhs(am_f, basis=b[zero].basis, f=Constant(0.0))[dofs]
b0 = 1 * b
if True:
b = 0 * w
w0 = 1 * w
b = 1 * b0
zero = Multiindex()
b[zero].coeffs = pde.assemble_rhs(coeff_field.mean_func, basis=b[zero].basis)
pde.set_dirichlet_bc_entries(w0[mu], homogeneous=False)
for mu in w0.active_indices():
pde.set_dirichlet_bc_entries(w0[mu], homogeneous=bool(mu.order != 0))
d = A * w0
pde.copy_dirichlet_bc(d, b)
# b[zero].coeffs = pde.assemble_rhs(coeff_field.mean_func, basis=b[zero].basis)
b1 = b
b2 = prepare_rhs(A, w, coeff_field, pde)
bl = 0 * b
for mu in w.active_indices():
bl[mu].coeffs[dofs] = 1
