Changeset 3db96b0 in sasmodels

Timestamp:

Feb 23, 2018 11:46:44 AM (7 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

master, core_shell_microgels, magnetic_model, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests

Children:

58a1be9

Parents:

aa90015

Message:

improve plotting for 2d monte carlo calculator

File:

• explore/realspace.py (modified) (6 diffs)

explore/realspace.py

@@ -423 +423 @@
 
 # -------------- plotters ----------------
-def plot_calc(r, Pr, q, Iq, theory=None):
+def plot_calc(r, Pr, q, Iq, theory=None, title=None):
     import matplotlib.pyplot as plt
     plt.subplot(211)
@@ -429 +429 @@
     plt.xlabel('r (A)')
     plt.ylabel('Pr (1/A^2)')
+    if title is not None:
+        plt.title(title)
     plt.subplot(212)
     plt.loglog(q, Iq, '-', label='from Pr')
@@ -437 +439 @@
         plt.legend()
 
-def plot_calc_2d(qx, qy, Iqxy, theory=None):
+def plot_calc_2d(qx, qy, Iqxy, theory=None, title=None):
     import matplotlib.pyplot as plt
     qx, qy = bin_edges(qx), bin_edges(qy)
@@ -443 +445 @@
     if theory is not None:
         plt.subplot(121)
-    plt.pcolormesh(qx, qy, np.log10(Iqxy))
+    #plt.pcolor(qx, qy, np.log10(Iqxy))
+    extent = [qx[0], qx[-1], qy[0], qy[-1]]
+    plt.imshow(np.log10(Iqxy), extent=extent, interpolation="nearest",
+               origin='lower')
     plt.xlabel('qx (1/A)')
     plt.ylabel('qy (1/A)')
+    plt.axis('equal')
+    plt.axis(extent)
+    #plt.grid(True)
+    if title is not None:
+        plt.title(title)
     if theory is not None:
         plt.subplot(122)
-        plt.pcolormesh(qx, qy, np.log10(theory))
+        plt.imshow(np.log10(theory), extent=extent, interpolation="nearest",
+                   origin='lower')
+        plt.axis('equal')
+        plt.axis(extent)
         plt.xlabel('qx (1/A)')
 
@@ -663 +676 @@
     if show_points:
          plot_points(rho, points); pylab.figure()
-    plot_calc(r, Pr, q, Iq, theory=theory)
+    plot_calc(r, Pr, q, Iq, theory=theory, title=title)
     pylab.gcf().canvas.set_window_title(title)
     pylab.show()
@@ -688 +701 @@
     if show_points:
         plot_points(rho, points); pylab.figure()
-    plot_calc_2d(qx, qy, Iqxy, theory=theory)
+    plot_calc_2d(qx, qy, Iqxy, theory=theory, title=title)
     pylab.gcf().canvas.set_window_title(title)
     pylab.show()