def __init__(self, sim, filename, **kwargs):
self.sim = sim
self.filename = filename
f = filename.replace('.hdf5','')
self.number = int(f.split('_')[-1])
self.file_id = h5py.File(filename, 'r')
self.header = Header(self.file_id)
kwargs['refine'] = kwargs.pop('refine_nbody', False)
self.dm = PartTypeNbody(self.file_id, 1, sim, **kwargs)
kwargs['refine'] = kwargs.pop('refine_gas', False)
self.gas = PartTypeSPH(self.file_id, 0, sim, **kwargs)
self.sinks = []
if kwargs.get('track_sinks', False):
self.sinks = self.gas.get_sink_properties()
class File(object):
"""
Class for Gadget2 HDF5 snapshot files.
"""
def __init__(self, sim, filename, **kwargs):
self.sim = sim
self.filename = filename
f = filename.replace('.hdf5','')
self.number = int(f.split('_')[-1])
self.file_id = h5py.File(filename, 'r')
self.header = Header(self.file_id)
kwargs['refine'] = kwargs.pop('refine_nbody', False)
self.dm = PartTypeNbody(self.file_id, 1, sim, **kwargs)
kwargs['refine'] = kwargs.pop('refine_gas', False)
self.gas = PartTypeSPH(self.file_id, 0, sim, **kwargs)
self.sinks = []
if kwargs.get('track_sinks', False):
self.sinks = self.gas.get_sink_properties()
def __getstate__(self):
result = self.__dict__.copy()
del result['file_id']
return result
# def __setstate__(self, in_dict):
# self.__dict__ = in_dict
# self.file_id = h5py.File(self.filename, 'r')
# self.header = hdf5.Header(self.file_id)
def update_sink_coordinates(self, x,y,z):
for s in self.sinks:
s.update_coordinates(x,y,z)
def update_sink_velocities(self, u,v,w):
for s in self.sinks:
s.update_velocities(u,v,w)
def update_sink_frame_ofR(self, xyz, uvw):
self.update_sink_coordinates(xyz[:,0], xyz[:,1], xyz[:,2])
self.update_sink_velocities(uvw[:,0], uvw[:,1], uvw[:,2])
def keys(self):
for key in self.file_id.keys():
print key
def close(self):
self.file_id.close()
