layered_medium_general.py 878 B

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