"""MOAB-specific fields
"""
#-----------------------------------------------------------------------------
# Copyright (c) 2013, yt Development Team.
#
# Distributed under the terms of the Modified BSD License.
#
# The full license is in the file COPYING.txt, distributed with this software.
#-----------------------------------------------------------------------------
import numpy as np
from yt.funcs import mylog
from yt.utilities.io_handler import BaseIOHandler
def field_dname(field_name):
return "/tstt/elements/Hex8/tags/{0}".format(field_name)
# TODO all particle bits were removed
[docs]class IOHandlerMoabH5MHex8(BaseIOHandler):
_dataset_type = "moab_hex8"
[docs] def __init__(self, ds):
super(IOHandlerMoabH5MHex8, self).__init__(ds)
self._handle = ds._handle
def _read_fluid_selection(self, chunks, selector, fields, size):
chunks = list(chunks)
assert(len(chunks) == 1)
fhandle = self._handle
rv = {}
for field in fields:
ftype, fname = field
rv[field] = np.empty(size, dtype=fhandle[field_dname(fname)].dtype)
ngrids = sum(len(chunk.objs) for chunk in chunks)
mylog.debug("Reading %s cells of %s fields in %s blocks",
size, [fname for ftype, fname in fields], ngrids)
for field in fields:
ftype, fname = field
ds = np.array(fhandle[field_dname(fname)][:], dtype="float64")
ind = 0
for chunk in chunks:
for g in chunk.objs:
ind += g.select(selector, ds, rv[field], ind) # caches
return rv
[docs]class IOHandlerMoabPyneHex8(BaseIOHandler):
_dataset_type = "moab_hex8_pyne"
def _read_fluid_selection(self, chunks, selector, fields, size):
chunks = list(chunks)
assert(len(chunks) == 1)
tags = {}
rv = {}
pyne_mesh = self.ds.pyne_mesh
mesh = pyne_mesh.mesh
for field in fields:
rv[field] = np.empty(size, dtype="float64")
ngrids = sum(len(chunk.objs) for chunk in chunks)
mylog.debug("Reading %s cells of %s fields in %s blocks",
size, [fname for ftype, fname in fields], ngrids)
for field in fields:
ftype, fname = field
if pyne_mesh.structured:
tag = pyne_mesh.mesh.getTagHandle('idx')
indices = [tag[ent] for ent in pyne_mesh.structured_iterate_hex()]
else:
indices = slice(None)
ds = np.asarray(getattr(pyne_mesh, fname)[indices], 'float64')
ind = 0
for chunk in chunks:
for g in chunk.objs:
ind += g.select(selector, ds, rv[field], ind) # caches
return rv