From e211fd7e5d9d6ca2e0c521988de0c11da10dd5b7 Mon Sep 17 00:00:00 2001
From: Stefan Neukamm <stefan.neukamm@tu-dresden.de>
Date: Thu, 7 Jul 2022 10:37:43 +0200
Subject: [PATCH] update plot.py

---
 microstructure_testsuite/plot.py | 24 +++++++++++++++++++-----
 1 file changed, 19 insertions(+), 5 deletions(-)

diff --git a/microstructure_testsuite/plot.py b/microstructure_testsuite/plot.py
index eec5776d..ca353510 100644
--- a/microstructure_testsuite/plot.py
+++ b/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()
+#
+
 
-- 
GitLab