diff --git a/Examples/python/simulation/ex06_Reflectometry/BasicPolarizedReflectometry.py b/Examples/python/simulation/ex06_Reflectometry/BasicPolarizedReflectometry.py index 164d8a072b6c17b2d600f7a5e79a5ac44c6b3e2a..32b55eff4e354744972b67e5bdc414356c85cf93 100644 --- a/Examples/python/simulation/ex06_Reflectometry/BasicPolarizedReflectometry.py +++ b/Examples/python/simulation/ex06_Reflectometry/BasicPolarizedReflectometry.py @@ -6,6 +6,7 @@ magnetized sample. import bornagain as ba from bornagain import deg, angstrom +import matplotlib.pyplot as plt def get_sample(): """ @@ -43,8 +44,8 @@ def get_simulation(scan_size=500): return simulation -def run_simulation(polarization=ba.kvector_t(0.0, 1.0, 0.0), - analyzer=ba.kvector_t(0.0, 1.0, 0.0)): +def run_simulation(polarization=ba.kvector_t(0, 1, 0), + analyzer=ba.kvector_t(0, 1, 0)): """ Runs simulation and returns its result. """ @@ -60,20 +61,28 @@ def run_simulation(polarization=ba.kvector_t(0.0, 1.0, 0.0), return simulation.result() -def plot(pp, mm): - from matplotlib import pyplot as plt - - ba.plot_simulation_result(pp, postpone_show=True) - plt.semilogy(mm.axis(), mm.array()) - plt.legend(['Up-Up', 'Down-Down'], - loc='upper right') - +def plot(data, labels): + + plt.figure() + for d, l in zip(data, labels): + plt.semilogy( d.axis(), + d.array(), + label=l, linewidth=1) + + plt.legend( loc='upper right' ) + plt.gca().yaxis.set_ticks_position('both') + plt.gca().xaxis.set_ticks_position('both') + + plt.xlabel(r"$\alpha_i$ [deg]") + plt.ylabel("Reflectivity") + + plt.tight_layout() plt.show() if __name__ == '__main__': - results_pp = run_simulation(ba.kvector_t(0.0, 1.0, 0.0), - ba.kvector_t(0.0, 1.0, 0.0)) - results_mm = run_simulation(ba.kvector_t(0.0, -1.0, 0.0), - ba.kvector_t(0.0, -1.0, 0.0)) - plot(results_pp, results_mm) + results_pp = run_simulation(ba.kvector_t(0, 1, 0), + ba.kvector_t(0, 1, 0)) + results_mm = run_simulation(ba.kvector_t(0, -1, 0), + ba.kvector_t(0, -1, 0)) + plot([results_pp, results_mm], ["$++$", "$--$"])