[Yt-svn] yt: This should fix parallelism for the 6-channel images

[hg at spacepope.org]

hg Repository: yt
details:   yt/rev/37368e122ba8
changeset: 1513:37368e122ba8
user:      Matthew Turk <matthewturk at gmail.com>
date:
Thu Apr 01 14:44:36 2010 -0700
description:
This should fix parallelism for the 6-channel images

diffstat:

 yt/lagos/ParallelTools.py |  4 ++--
 1 files changed, 2 insertions(+), 2 deletions(-)

diffs (21 lines):

diff -r b409318f6420 -r 37368e122ba8 yt/lagos/ParallelTools.py
--- a/yt/lagos/ParallelTools.py	Thu Apr 01 11:30:20 2010 -0700
+++ b/yt/lagos/ParallelTools.py	Thu Apr 01 14:44:36 2010 -0700
@@ -848,7 +848,7 @@
         if MPI.COMM_WORLD.rank == 0:
             cc = MPI.Compute_dims(MPI.COMM_WORLD.size, 2)
             nsize = final[0]/cc[0], final[1]/cc[1]
-            new_image = na.zeros((final[0], final[1], 4), dtype='float64')
+            new_image = na.zeros((final[0], final[1], 6), dtype='float64')
             new_image[0:nsize[0],0:nsize[1],:] = data[:]
             for i in range(1,MPI.COMM_WORLD.size):
                 cy, cx = na.unravel_index(i, cc)
@@ -856,7 +856,7 @@
                     i, nsize[0]*cx,nsize[0]*(cx+1),
                        nsize[1]*cy,nsize[1]*(cy+1))
                 buf = _recv_array(source=i, tag=0).reshape(
-                    (nsize[0],nsize[1],4))
+                    (nsize[0],nsize[1],6))
                 new_image[nsize[0]*cy:nsize[0]*(cy+1),
                           nsize[1]*cx:nsize[1]*(cx+1),:] = buf[:]
             data = new_image