[firedrake] Is there a way to calculate a flux on the edges of each cell in Firedrake