update plot.py

microstructure_testsuite/plot.py

@@ -54,19 +54,33 @@ elif case==-1: # Read from outputs
 # 
 
 #    
-N=200
-h=0.0005
+length=.05
+N=300
+h=length/N
+E=np.zeros([N,N])
 X=np.zeros([N,N])
+Y=np.zeros([N,N])
 for i in range(0,N): 
     for j in range(0,N):     
         x=(i-N/2)*h
         y=(j-N/2)*h
+        X[i,j]=(i-N/2)*h
+        Y[i,j]=(j-N/2)*h
         K=xytokappaalpha(x,y)
-        X[N-1-j,i]=energy(K[0],K[1],Q,B) # N-1-j damit die x,y Axen stimmen
+        E[i,j]=energy(K[0],K[1],Q,B) # N-1-j damit die x,y Axen stimmen
 
-fig = plt.figure(figsize=(6,6))
-plt.imshow(np.log(X-np.min(X)+0.0001)) # normalize to min = 0 and log scale to emphasize energy landscape
+fig = plt.figure(figsize=(7,6))
+ax = plt.gca()
+ax.set_aspect(1)
+ax.set_xticks([-length/4,0,length/4])
+ax.set_yticks([])
+#pcm = plt.pcolor(X,Y,E, norm=colors.LogNorm(vmin=E.min(), vmax=E.max()), cmap='winter', shading='auto')
+pcm = plt.pcolor(X,Y,E, norm=colors.PowerNorm(gamma=0.2), cmap='winter', shading='auto')
+plt.colorbar(pcm, extend='max')
+#plt.imshow(np.log(E-np.min(E)+0.0001)) # normalize to min = 0 and log scale to emphasize energy landscape
 # TODO: Beschriftung der Axen sollte von [-h*N/2, h*N/2] sein!
 # Kreis mit Radius 1 einzeichnen
 plt.show()
+#
+