Use time at end of time step for Neumann BC

BenjaminRodenberg · Nov 16, 2023 · 29fa60b · 29fa60b
1 parent fef8f25
commit 29fa60b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/partitioned-heat-conduction/fenics/heat.py b/partitioned-heat-conduction/fenics/heat.py
@@ -171,14 +171,14 @@ def determine_gradient(V_g, u, flux):
         vs = split(v)
         for i in range(tsm.num_stages):
             bc.append(DirichletBC(Vbig.sub(i), du_dt[i], remaining_boundary))
-            F += vs[i] * coupling_expressions[i] * dolfin.ds
+            F += vs[i] * coupling_expressions[-1] * dolfin.ds
 else:
     if problem is ProblemType.DIRICHLET:
         bc.append(DirichletBC(Vbig, du_dt[0], remaining_boundary))
         bc.append(DirichletBC(Vbig, coupling_expressions[0], coupling_boundary))
     else:
         bc.append(DirichletBC(Vbig, du_dt[0], remaining_boundary))
-        F += v * coupling_expressions[0] * dolfin.ds
+        F += v * coupling_expressions[-1] * dolfin.ds
 
 
 # get lhs and rhs of variational form