[yt-svn] commit/yt: 2 new changesets

[commits-noreply at bitbucket.org]

2 new commits in yt:

https://bitbucket.org/yt_analysis/yt/commits/1b01901aeeba/
changeset:   1b01901aeeba
branch:      yt
user:        mqk
date:        2013-03-14 23:17:14
summary:     Added field_parameters keywords to GDF writer.save_field().
affected #:  1 file

diff -r bda5c9c8fe1bfdda64bdd5f476690ad43fff9b4b -r 1b01901aeebaab2eea6485bae9f27ca32380d11a yt/utilities/grid_data_format/writer.py
--- a/yt/utilities/grid_data_format/writer.py
+++ b/yt/utilities/grid_data_format/writer.py
@@ -57,7 +57,7 @@
     # don't forget to close the file.
     f.close()
 
-def save_field(pf, field_name):
+def save_field(pf, field_name, field_parameters=None):
     """
     Write a single field associated with the parameter file pf to the
     backup file.
@@ -68,6 +68,8 @@
         The yt parameter file that the field is associated with.
     field_name : string
         The name of the field to save.
+    field_parameters : dictionary
+        A dictionary of field parameters to set.
     """
 
     field_obj = pf.field_info[field_name]
@@ -85,12 +87,12 @@
                        particle_type_name="dark_matter")
 
     # now save the field
-    _write_field_to_gdf(pf, f, field_name, particle_type_name="dark_matter")
+    _write_field_to_gdf(pf, f, field_name, particle_type_name="dark_matter", field_parameters=field_parameters)
 
     # don't forget to close the file.
     f.close()
         
-def _write_field_to_gdf(pf, fhandle, field_name, particle_type_name):
+def _write_field_to_gdf(pf, fhandle, field_name, particle_type_name, field_parameters=None):
 
     # add field info to field_types group
     g = fhandle["field_types"]
@@ -122,6 +124,12 @@
     # now add actual data, grid by grid
     g = fhandle["data"]     
     for grid in pf.h.grids:
+
+        # set field parameters, if specified
+        if field_parameters is not None:
+            for k,v in field_parameters.iteritems():
+                grid.set_field_parameter(k,v)
+
         grid_group = g["grid_%010i" % (grid.id - grid._id_offset)]
         particles_group = grid_group["particles"]
         pt_group = particles_group[particle_type_name]


https://bitbucket.org/yt_analysis/yt/commits/22228acb73d5/
changeset:   22228acb73d5
branch:      yt
user:        ngoldbaum
date:        2013-03-16 18:37:16
summary:     Merged in mqk/yt_clean (pull request #461)

Added field_parameters keywords to GDF writer.save_field(). (Fixes #526)
affected #:  1 file

diff -r 5b79077a918d2c91fd74b734946138d81dbd83d5 -r 22228acb73d5af081289f362ddfe82753c394ae8 yt/utilities/grid_data_format/writer.py
--- a/yt/utilities/grid_data_format/writer.py
+++ b/yt/utilities/grid_data_format/writer.py
@@ -57,7 +57,7 @@
     # don't forget to close the file.
     f.close()
 
-def save_field(pf, field_name):
+def save_field(pf, field_name, field_parameters=None):
     """
     Write a single field associated with the parameter file pf to the
     backup file.
@@ -68,6 +68,8 @@
         The yt parameter file that the field is associated with.
     field_name : string
         The name of the field to save.
+    field_parameters : dictionary
+        A dictionary of field parameters to set.
     """
 
     field_obj = pf.field_info[field_name]
@@ -85,12 +87,12 @@
                        particle_type_name="dark_matter")
 
     # now save the field
-    _write_field_to_gdf(pf, f, field_name, particle_type_name="dark_matter")
+    _write_field_to_gdf(pf, f, field_name, particle_type_name="dark_matter", field_parameters=field_parameters)
 
     # don't forget to close the file.
     f.close()
         
-def _write_field_to_gdf(pf, fhandle, field_name, particle_type_name):
+def _write_field_to_gdf(pf, fhandle, field_name, particle_type_name, field_parameters=None):
 
     # add field info to field_types group
     g = fhandle["field_types"]
@@ -122,6 +124,12 @@
     # now add actual data, grid by grid
     g = fhandle["data"]     
     for grid in pf.h.grids:
+
+        # set field parameters, if specified
+        if field_parameters is not None:
+            for k,v in field_parameters.iteritems():
+                grid.set_field_parameter(k,v)
+
         grid_group = g["grid_%010i" % (grid.id - grid._id_offset)]
         particles_group = grid_group["particles"]
         pt_group = particles_group[particle_type_name]

Repository URL: https://bitbucket.org/yt_analysis/yt/

--

This is a commit notification from bitbucket.org. You are receiving
this because you have the service enabled, addressing the recipient of
this email.