point-in-cell.py

from firedrake import *
import numpy as np
 
Dx = 0.1; Nx = 10
mesh = IntervalMesh(Nx, Nx*Dx)
V = FunctionSpace(mesh, "DG", 0)

def coord2idx(x, coords):
    for i, cell in enumerate(coords.cell_node_map().values):
        a, b = coords.dat.data_ro_with_halos[cell]
        a, b = (a, b) if a < b else (b, a)
        if a <= x < b:
            return i
    return None
 
print "Rank " + str(op2.MPI.comm.rank)
point = 0.5
idx = coord2idx(point, mesh.coordinates)
print idx