SQUID noise from a conducting plane

Compute thermal noise in SQUID pickup loops near a conducting plane.

This example requires left_auditory-ave.fif from the MNE example results. Set SAVE_DIR to its directory before running the script.

import pyvista as pv
from bfieldtools._example_plots import triangle_mesh, vector_glyphs, set_view

pv.set_plot_theme("document")

import numpy as np
import matplotlib.pyplot as plt
import trimesh
from scipy.linalg import eigh

from bfieldtools.thermal_noise import (
    compute_current_modes,
    visualize_current_modes,
    noise_covar,
    noise_var,
    sensornoise_covar,
)
from bfieldtools.mesh_magnetics import magnetic_field_coupling


from bfieldtools import resources

import mne

font = {"family": "normal", "weight": "normal", "size": 16}

plt.rc("font", **font)

# Fix the simulation parameters
d = 5e-3
sigma = 3.8e7
T = 293
kB = 1.38064852e-23
mu0 = 4 * np.pi * 1e-7


Nchunks = 4
quad_degree = 2

SAVE_DIR = ""

evoked = mne.Evoked(SAVE_DIR + "left_auditory-ave.fif")
evoked.pick_types(meg="mag")

coils = mne.forward._create_meg_coils(evoked.info["chs"], acc=2)

p = np.zeros((len(coils), coils[0]["rmag"].shape[0], 3))
n = np.zeros((len(coils), coils[0]["rmag"].shape[0], 3))
w = np.zeros((len(coils), coils[0]["rmag"].shape[0]))

for i in range(len(coils)):
    p[i] = coils[i]["rmag"]
    n[i] = coils[i]["cosmag"]
    w[i] = coils[i]["w"]

mesh = trimesh.load(resources.files("bfieldtools", "example_meshes/10x10_plane.obj"))
mesh.vertices, mesh.faces = trimesh.remesh.subdivide(mesh.vertices, mesh.faces)

mesh.apply_scale(1e-1)
theta = np.pi / 2


R = np.zeros((4, 4))
R[0] = np.array((np.cos(theta), -1 * np.sin(theta), 0, 0))
R[1] = np.array((np.sin(theta), np.cos(theta), 0, 0))
R[2] = np.array((0, 0, 1, 0))
R[3] = np.array((0, 0, 0, 0))

mesh.apply_transform(R)
mesh.apply_translation((40e-2, 0, 0))

plotter = scene = pv.Plotter(window_size=(800, 800))
s = plotter.add_mesh(
    triangle_mesh(*mesh.vertices.T, mesh.faces),
    color=(0, 0, 0),
    opacity=0.5,
    show_scalar_bar=False,
)

plotter.add_mesh(
    vector_glyphs(
        *p.reshape(p.shape[0] * p.shape[1], p.shape[2]).T,
        *n.reshape(n.shape[0] * n.shape[1], n.shape[2]).T,
        mode="arrow"
    ),
    show_scalar_bar=False,
)
set_view(plotter, azimuth=90, elevation=90, distance=2.5)

plotter = scene = pv.Plotter(window_size=(800, 800))
s = plotter.add_mesh(
    triangle_mesh(*mesh.vertices.T, mesh.faces),
    color=(0, 0, 0),
    opacity=0.5,
    show_scalar_bar=False,
)
plotter.add_mesh(vector_glyphs(*p.T, *n.T, mode="arrow"), show_scalar_bar=False)
set_view(plotter, azimuth=180, elevation=90, distance=2.5)

vl, u = compute_current_modes(
    obj=mesh, T=T, resistivity=1 / sigma, thickness=d, mode="AC", return_eigenvals=True
)

noisecov = sensornoise_covar(mesh, p, n, w, vl)
noisecov = noisecov[:, :, 0]

noisecov += noisecov.T - np.diag(np.diag(noisecov))

plt.figure(figsize=(7.5, 7.5))
plt.imshow(noisecov * 1e30, cmap="RdBu_r")
cbar = plt.colorbar()
plt.clim((-300, 300))
plt.xlabel("Sensor index")
plt.ylabel("Sensor index")
cbar.set_label(r"Noise covariance (fT$^2$/Hz)")
plt.tight_layout()


evoked1 = evoked.copy()
evoked1.data = np.tile(np.sqrt(np.diag(noisecov)), (evoked.times.shape[0], 1)).T
evoked1.plot_topomap(times=0.080, ch_type="mag", vmin=5, vmax=18)

# Display the assembled scene (Sphinx-Gallery captures it during docs builds).
if "plotter" in globals() and not plotter._closed:
    plotter.show()