Skip to content
Snippets Groups Projects
Select Git revision
  • 72-improve-docs-for_optimal_setup
  • master default protected
  • os-path-join
  • develop-GA
  • add-higher-spindown-components
  • v1.3
  • v1.2
  • v1.1.2
  • v1.1.0
  • v1.0.1
10 results

short_transient_search_gridded.py

Blame
  • test_2d_plot.py NaN GiB
    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 *
    import matplotlib.image as mpimg
    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
    
    ad ad1 0 n2
    beam b1 0 n2
    maxtem 0 
    
    xaxis b1 x lin -10.0 10 100
    x2axis b1 y lin -6 6 100
    
    """
    
    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_wireframe(x, y, out["b1"])   
    pl.xlabel(out.xlabel)
    pl.ylabel(out.ylabel)
    pl.show()
    
    pl.figure()
    extent = [np.min(out.x),np.max(out.x),np.min(out.y),np.max(out.y)]
    imgplot = pl.imshow(out["b1"], extent=extent)
    #imgplot.set_interpolation('bicubic')
    imgplot.set_interpolation('nearest')
    pl.colorbar()
    pl.xlabel(out.xlabel)
    pl.ylabel(out.ylabel)
    pl.show()