wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-11 18:00:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

added better logic to the thread dispatcher.

Browse Source
This commit is contained in:
sccolbert
2009-11-08 05:39:02 +01:00
parent 196ab7530a
commit 4beb9eb16b
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scikits/image/io/_plugins/util.py
+11 -7
View File
@@ -193,12 +193,11 @@ class ThreadDispatch(object):
self.cores = CPU_COUNT
self.threads = []
self.chunks = []
if self.cores == 2:
self.chunks.append((img[:, :(width/2), :],
stateimg[:, :(width/2), :]))
self.chunks.append((img[:, (width/2):, :],
stateimg[:, (width/2):, :]))
elif self.cores == 4:
if self.cores == 1:
self.chunks.append((img, stateimg))
elif self.cores >= 4:
self.chunks.append((img[:(height/2), :(width/2), :],
stateimg[:(height/2), :(width/2), :]))
self.chunks.append((img[:(height/2), (width/2):, :],
@@ -207,8 +206,13 @@ class ThreadDispatch(object):
stateimg[(height/2):, :(width/2), :]))
self.chunks.append((img[(height/2):, (width/2):, :],
stateimg[(height/2):, (width/2):, :]))
# if they dont have 1, or 4 or more, 2 is good.
else:
raise ValueError('Cant handle your weird machine')
self.chunks.append((img[:, :(width/2), :],
stateimg[:, :(width/2), :]))
self.chunks.append((img[:, (width/2):, :],
stateimg[:, (width/2):, :]))
for i in range(self.cores):
self.threads.append(ImgThread(func, self.chunks[i][0],