basis.probes(quad0_nodes) fails with a stack trace...

[prusso630]

from scipy.ndimage import laplace
from skfem import MeshTri
from scipy.sparse.linalg import spsolve
from skfem import MeshTri, ElementTriP1, InteriorBasis
from skfem import MeshQuad
from skfem import ElementQuad1
from skfem.assembly import BilinearForm, LinearForm, asm
from scipy.sparse import lil_matrix
import matplotlib.pyplot as plt
from skfem import *



# Define the size of the plate
length = 24
height = 24

# Define the number of elements along each dimension
n_elems = 10

# Create the mesh
m = MeshQuad().init_tensor(np.linspace(0, length, n_elems), np.linspace(0, height, n_elems))

# Define the element
element = ElementQuad1()

# Create the InteriorBasis object
basis = InteriorBasis(m, element)

# Get the degrees of freedom
dofs = basis.dofs

print(dofs.nodal_dofs)
print(dofs.element_dofs)

# We have the 2D sense. u(x,y) that is being appended to grad_u
grad_u = []
for i in range(m.p.shape[1]):
    u = m.p[:, i]
    print(f"u: {u}")
    grad_u.append(laplace(u))

# Starting temperatures...
start_temp = []
for i in range(m.p.shape[1]):
    start_temp.append(10)

# Starting kappa's
kappa = []
for i in range(m.p.shape[1]):
    kappa.append(0.83)

# Starting forces...
force = []
for i in range(m.p.shape[1]):
    u = m.p[:, i]
    if (u[0] == 0.0):
        force.append(10)
    else:
        force.append(0)

def grad(f):
    return np.array([laplace(f[:, i]) for i in range(f.shape[1])]).T

@BilinearForm
def bilinear_form(u, v, w):
    return kappa * np.sum(grad(v) * grad(u), axis=0)



# Get the nodes indices of a quadrilateral....
quad0_nodes = m.t[:, 0]
free_nodes=np.array([quad0_nodes])

v = basis.probes(quad0_nodes)
'''
??? I'm not quite sure yet why this would be happening here at the creation of v, or yet how to find a resolution...

Traceback (most recent call last):
  File "c:\Users\prusso\demo\scratchpad\scikit-FEM20.py", line 76, in <module>
    v = basis.probes(quad0_nodes)
        ^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\prusso\demo\.venv\Lib\site-packages\skfem\assembly\basis\cell_basis.py", line 179, in probes
    cells = self.mesh.element_finder(mapping=self.mapping)(*x)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\prusso\demo\.venv\Lib\site-packages\skfem\mesh\mesh_quad_1.py", line 220, in finder
    return tri_finder(*args) % self.t.shape[1]
           ^^^^^^^^^^^^^^^^^
TypeError: MeshTri1.element_finder.<locals>.finder() takes from 2 to 3 positional arguments but 4 were given
'''



##u = basis.probes(quad0_nodes)
##K = lil_matrix((quad0_nodes.shape[0], quad0_nodes.shape[0]))
##F = np.zeros(quad0_nodes.shape[0])

## A global matrix for the heat problem is to be built...

print('done...')

Anyone can help me to understand why that v = basis.probes(quad0_nodes) is throwing the stack trace posted in the multi-line comment area of the attached code?

[kinnala]

basis.probes takes in point coordinates as an argument. You have passed in node indices.

[prusso630]

It does seem to accept v = basis.probes(m.p). I'm not really for sure how correct that is yet but it works. Thanks...