From 044af4a96b58f001eb3cfbe1fc1f5804c6a816fa Mon Sep 17 00:00:00 2001
From: Daniel Brown <ddb@star.sr.bham.ac.uk>
Date: Sat, 14 Dec 2013 01:58:27 +0000
Subject: [PATCH] adding 2d plot test - plotting in 3d with matplotlib is a
 pain...

---
 bin/test_2d_plot.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 bin/test_2d_plot.py

diff --git a/bin/test_2d_plot.py b/bin/test_2d_plot.py
new file mode 100644
index 0000000..debb606
--- /dev/null
+++ b/bin/test_2d_plot.py
@@ -0,0 +1,35 @@
+from pykat import finesse
+from pykat.utilities.optics.gaussian_beams import gauss_param
+from pykat.detectors import *
+from pykat.components import *
+from pykat.commands import *
+from pykat.structs import *
+from mpl_toolkits.mplot3d.axes3d import Axes3D
+
+import numpy as np
+import pylab as pl
+
+code = """
+l l1 1 0 0 n1
+s s1 10 1 n1 n2
+
+beam b1 0 n2
+maxtem 0 
+
+xaxis b1 x lin -7.0 7.0 50
+x2axis b1 y lin -7.0 7.0 50
+
+"""
+
+kat = finesse.kat()
+
+kat.parseCommands(code)
+kat.s1.n1.q = gauss_param(w0=1e-3, z=0)
+
+out = kat.run(printout=0,printerr=0)
+
+fig = pl.figure()
+ax = fig.add_subplot(1, 1, 1, projection='3d')
+x, y = np.meshgrid(out.x, out.y)
+p = ax.plot_surface(x, y, out.z[0,:])   
+pl.show()
-- 
GitLab