[firedrake] Flux calculation