test_btlrec_lamino.py 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. from gi.repository import Ufo
  2. path = [ '/home/timurttv/_UFO-Project/git_framework3/lamino-filters-installed/lib/ufo',
  3. '/home/timurttv/_UFO-Project/git_framework3/ufo-filters-installed/lib/ufo' ]
  4. conf = Ufo.Configuration(paths=path)
  5. g = Ufo.Graph()
  6. pm = Ufo.PluginManager(configuration=conf)
  7. s = Ufo.Scheduler(configuration=conf)
  8. reader = pm.get_filter('reader')
  9. ramp = pm.get_filter('laminoramp')
  10. padding = pm.get_filter('padding2d')
  11. fft_ramp = pm.get_filter('fft')
  12. fft_orig = pm.get_filter('fft')
  13. ifft = pm.get_filter('ifft')
  14. conv = pm.get_filter('laminoftconv')
  15. rec = pm.get_filter('laminobpgeneric')
  16. writer = pm.get_filter('3dedfwriter')
  17. reader.set_properties(path='/home/vogelgesang/data/bug/projections/*.tif', count=300)
  18. padding.set_properties(xl=179, xr=3085, yt=5, yb=107, mode='brep')
  19. ramp.set_properties(width=4096, fwidth=1186, height=512, theta=-1.570796326, tau=1)
  20. fft_ramp.set_properties(dimensions=2)
  21. fft_orig.set_properties(dimensions=2)
  22. ifft.set_properties(dimensions=2)
  23. rec.set_properties( theta=-1.570796326,
  24. psi=0,
  25. angle_step=0.01256637,
  26. vol_sx=832,
  27. vol_sy=832,
  28. vol_sz=400,
  29. vol_ox=416,
  30. vol_oy=416,
  31. vol_oz=200,
  32. proj_ox=592,
  33. proj_oy=205)
  34. writer.set_properties(prefix='./out/vol-')
  35. g.connect_filters(reader, padding)
  36. g.connect_filters(padding, fft_orig)
  37. g.connect_filters(ramp, fft_ramp)
  38. g.connect_filters_full(fft_ramp, 0, conv, 0, Ufo.TransferMode.COPY) # filter
  39. g.connect_filters_full(fft_orig, 0, conv, 1, Ufo.TransferMode.COPY) # projs
  40. g.connect_filters(conv, ifft)
  41. g.connect_filters(ifft, rec)
  42. g.connect_filters(rec, writer)
  43. s.run(g)