[Yt-svn] yt: Bug fixes in the SFG.

[hg at spacepope.org]

hg Repository: yt
details:   yt/rev/1c6d36026f5b
changeset: 1548:1c6d36026f5b
user:      Stephen Skory <stephenskory at yahoo.com>
date:
Sun Apr 11 14:41:26 2010 -0700
description:
Bug fixes in the SFG.

diffstat:

 yt/lagos/StructureFunctionGenerator.py |  6 +++---
 1 files changed, 3 insertions(+), 3 deletions(-)

diffs (27 lines):

diff -r 9cd931a41384 -r 1c6d36026f5b yt/lagos/StructureFunctionGenerator.py
--- a/yt/lagos/StructureFunctionGenerator.py	Sat Apr 10 17:52:49 2010 -0700
+++ b/yt/lagos/StructureFunctionGenerator.py	Sun Apr 11 14:41:26 2010 -0700
@@ -171,7 +171,7 @@
         self.stored_fields = {}
         for field in self.fields:
             self.stored_fields[field] = self.ds[field].copy()
-            self.ds.clear_data()
+        self.ds.clear_data()
         # If the arrays haven't been sorted yet and need to be, do that.
         if not self.sort_done:
             for field in self.fields:
@@ -484,12 +484,12 @@
                 # Get the fields values.
                 results = self._get_fields_vals(mypoints)
                 # Put this into self.fields_vals.
-                self.fields_vals.shape = (self.comm_size*2, 3)
+                self.fields_vals.shape = (self.comm_size*2, len(self.fields))
                 self.fields_vals[select] = results
             
             # Put our arrays back into their original shapes cheaply!
             if mypoints.size > 0:
-                self.fields_vals.shape = (self.comm_size, 6)
+                self.fields_vals.shape = (self.comm_size, len(self.fields)*2)
             self.points.shape = (self.comm_size, 6)
             
             # To run the structure functions, what is key is that the