def __call__(self,u,v,w,r,bx,by,bz): freq = self.sim.nx/2 plt.subplot(211) for i in self.heights: vals = filter_conv(u[i,:,:],v[i,:,:],w[i,:,:],r[i,:,:]) plt.plot(vals,label="z = " + str(i)) plt.title('Kinetic') plt.semilogy() plt.legend() plt.subplot(212) for i in self.heights: vals = filter_conv(bx[i,:,:],by[i,:,:],bz[i,:,:]) plt.plot(vals,label="z = " + str(i)) plt.title('Magnetic') plt.xlabel('k') plt.semilogy() plt.legend()
def __call__(self,u,v,w,r,b2): eng = r*(u*u + v*v + w*w) plt.subplot(211) for i in self.heights: layer = eng[i,:,:] vals = filter_conv(layer) plt.plot(vals,label="z = " + str(i)) plt.title('Kinetic') plt.semilogy() plt.legend() plt.subplot(212) for i in self.heights: layer = b2[i,:,:] vals = filter_conv(layer) plt.plot(vals,label="z = " + str(i)) plt.title('Magnetic') plt.xlabel('k') plt.semilogy() plt.legend()