import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import Normalize
from matplotlib.cm import ScalarMappable
from scipy.ndimage import convolve, correlate
from mplEasyAnimate import animation
from matplotlib.cm import viridis
from tqdm.notebook import tqdm
plt.style.use(pubStyle)