[firedrake] Converting FENICS code to Firedrake (iterations over mesh cells)