Looping over cells in the mesh

[cpraveen]

I am trying to convert code from old fenics to firedrake.

Here I am computing some local error indicator and loop over cells

    # Construct error indicator
    n = FacetNormal(mesh)
    dudn = dot(grad(u), n)
    Z = FunctionSpace(mesh, 'DG', 0)
    z = TestFunction(Z)
    chi = assemble(2*avg(z)*jump(dudn)**2*dS)
    eta = numpy.array([0.5*numpy.sqrt(c.h()*chi[c.index()]) \
                        for c in cells(mesh)])

cells(mesh) does not work in firedrake. Could you suggest an alternative for this ?

[cpraveen]

Full fenics code is here https://github.com/cpraveen/fem/blob/master/fenics/step-10/demo.py

[cpraveen]

I figured out a solution, so closing this.