[Yt-svn] yt-commit r1524 - trunk/yt/extensions

[dcollins at wrangler.dreamhost.com]

Author: dcollins
Date: Fri Nov 13 10:03:24 2009
New Revision: 1524
URL: http://yt.enzotools.org/changeset/1524

Log:
Added clump_tools for manipulating clumps

Added:
   trunk/yt/extensions/clump_tools.py

Added: trunk/yt/extensions/clump_tools.py
==============================================================================
--- (empty file)
+++ trunk/yt/extensions/clump_tools.py	Fri Nov 13 10:03:24 2009
@@ -0,0 +1,70 @@
+"""
+Clump tools for use with the yt Clump object
+
+Author: David Collins <dcollins at physics.ucsd.edu>
+Affiliation: Center for Astrophysics and Space Sciences, U C San Diego
+Homepage: http://yt.enzotools.org/
+License:
+  Copyright (C) 2009 David Collins.  All Rights Reserved.
+
+  This file is part of yt.
+
+  yt is free software; you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation; either version 3 of the License, or
+  (at your option) any later version.
+
+  This program is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+"""
+
+import numpy as na
+nar = na.array
+
+counter = 0
+def recursive_all_clumps(clump,list,level,parentnumber):
+    """A recursive function to flatten the hierarchy in *clump*.
+    Not to be called directly: please call return_all_clumps, below."""
+    global counter
+    counter += 1
+    clump.number = counter
+    clump.parentnumber = parentnumber
+    counter += 1
+    list.append(clump)
+    clump.level = level
+    if clump.children != None:
+        for child in clump.children:
+            x = recursive_all_clumps(child,list,level+1,clump.number)
+    return list
+
+def return_all_clumps(clump):
+    """Flatten the hierarchy defined by *clump*.
+    Additionally adds three variables to the clump:
+    level        = depth of hierarchy
+    number       = index of clump in the final array
+    parentnumber = index of this clumps parent
+    """
+    global counter
+    counter = 0
+    list = []
+    level = 0
+    clump.level = level
+    parentnumber=-1
+    recursive_all_clumps(clump,list,level,parentnumber)
+    return list
+
+
+def clump_list_sort(clump_list):
+    """Returns a copy of clump_list, sorted by ascending minimum density.
+    This eliminates overlap when passing to yt.raven.ClumpContourCallback"""
+    minDensity = [c['Density'].min() for c in clump_list]
+    
+    args = na.argsort(minDensity)
+    list = nar(clump_list)[args]
+    reverse = range(list.size-1,-1,-1)
+    return list[reverse]