Commit 6662c4284837cec9eaa46a11824c9635c89cb409

Authored by Thiago Franco de Moraes
1 parent 4cb5f692
Exists in ff_mask

Updated the setup.py

Showing 2 changed files with 22 additions and 12 deletions   Show diff stats
invesalius/data/floodfill.pyx
... ... @@ -139,13 +139,12 @@ def floodfill_threshold(np.ndarray[image_t, ndim=3] data, list seeds, int t0, in
139 139 out[z, y, x] = fill
140 140  
141 141 for k in xrange(odz):
  142 + zo = z + k - offset_z
142 143 for j in xrange(ody):
  144 + yo = y + j - offset_y
143 145 for i in xrange(odx):
144 146 if strct[k, j, i]:
145 147 xo = x + i - offset_x
146   - yo = y + j - offset_y
147   - zo = z + k - offset_z
148   -
149 148 if 0 <= xo < dx and 0 <= yo < dy and 0 <= zo < dz and out[zo, yo, xo] != fill and t0 <= data[zo, yo, xo] <= t1:
150 149 out[zo, yo, xo] = fill
151 150 c.x = xo
... ...
setup.py
... ... @@ -38,18 +38,23 @@ elif sys.platform == &#39;win32&#39;:
38 38 setup(
39 39 cmdclass = {'build_ext': build_ext},
40 40 ext_modules = cythonize([ Extension("invesalius.data.mips", ["invesalius/data/mips.pyx"],
41   - include_dirs = [numpy.get_include()],
42   - extra_compile_args=['/openmp'],),
  41 + include_dirs = [numpy.get_include()],
  42 + extra_compile_args=['/openmp'],),
43 43  
44   - Extension("invesalius.data.interpolation", ["invesalius/data/interpolation.pyx"],
45   - include_dirs=[numpy.get_include()],
46   - extra_compile_args=['/openmp'],),
  44 + Extension("invesalius.data.interpolation", ["invesalius/data/interpolation.pyx"],
  45 + include_dirs=[numpy.get_include()],
  46 + extra_compile_args=['/openmp'],),
47 47  
48   - Extension("invesalius.data.transforms", ["invesalius/data/transforms.pyx"],
49   - include_dirs=[numpy.get_include()],
50   - extra_compile_args=['/openmp'],),
  48 + Extension("invesalius.data.transforms", ["invesalius/data/transforms.pyx"],
  49 + include_dirs=[numpy.get_include()],
  50 + extra_compile_args=['/openmp'],),
  51 +
  52 + Extension("invesalius.data.floodfill", ["invesalius/data/floodfill.pyx"],
  53 + include_dirs=[numpy.get_include()],
  54 + extra_compile_args=['/openmp',],
  55 + language='c++',),
51 56 ])
52   - )
  57 + )
53 58  
54 59 else:
55 60 setup(
... ... @@ -69,5 +74,11 @@ else:
69 74 include_dirs=[numpy.get_include()],
70 75 extra_compile_args=['-fopenmp',],
71 76 extra_link_args=['-fopenmp',]),
  77 +
  78 + Extension("invesalius.data.floodfill", ["invesalius/data/floodfill.pyx"],
  79 + include_dirs=[numpy.get_include()],
  80 + extra_compile_args=['-fopenmp',],
  81 + extra_link_args=['-fopenmp',],
  82 + language='c++',),
72 83 ])
73 84 )
... ...