layered_medium_2.py 793 B

123456789101112131415161718192021222324252627282930
  1. from pysketcher import *
  2. from numpy import exp, linspace
  3. W = 10.
  4. H = 10.
  5. a = [0, 5, 10]
  6. drawing_tool.set_coordinate_system(xmin=-1, xmax=W+1,
  7. ymin=-1, ymax=H+1,
  8. axis=False)
  9. drawing_tool.set_linecolor('black')
  10. drawing_tool.set_fontsize(24)
  11. layers = {'layer%d' % i: Line((0,a[i]), (W,a[i]))
  12. for i in range(len(a))}
  13. symbols_q = {'Omega_k%d' % i: Text(r'$\Omega_%d$: $k_%d$' % (i, i),
  14. (W/2,0.5*(a[i]+a[i+1])))
  15. for i in range(len(a)-1)}
  16. sides ={'left': Line((0,0), (0,H)), 'right': Line((W,0), (W,H))}
  17. d = sides.copy()
  18. d.update(layers)
  19. d.update(symbols_q)
  20. fig = Composition(d)
  21. fig.draw()
  22. drawing_tool.display()
  23. drawing_tool.savefig('tmp2')
  24. input()