Cleaner computation of einstein summations in Electrostatic.py

[sannant]

Adhesion/Adhesion/Interactions/Electrostatic.py

Lines 200 to 210 in 49a1f4f

	E_y = np.fft.ifft2(
	np.einsum(
	"pq, pq, pqz, pqz-> pqz",
	-complex("j") * q2_axis[:,:,0] / q_norm[:,:,0],
	fft_magnitude / (self.epsilon_material + self.epsilon_gap),
	1 + np.exp(-q_norm*gap_axis),
	1 / (1 - B*decay),
	),
	s=self.num_grid_points,
	axes=(0, 1),
	)

There are probably smarter and more transparent ways to handle these arrays.