jan/gridlock
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

more member docs

Browse Source
This commit is contained in:
Jan Petykiewicz 2019-11-27 19:32:53 -08:00
parent 00dd0718fa
commit d4b0177386
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
gridlock/grid.py
View File

@ -216,16 +216,24 @@ class Grid(object):
documentation. List of 3 bool, or a single bool that gets broadcast. Default False. documentation. List of 3 bool, or a single bool that gets broadcast. Default False.
:raises: GridError :raises: GridError
""" """
self.exyz = [numpy.unique(pixel_edge_coordinates[i]) for i in range(3)] self.exyz = [numpy.unique(pixel_edge_coordinates[i]) for i in range(3)] # type: List[numpy.ndarray]
"""Cell edges. Monotonically increasing without duplicates."""
self.grids = None # type: numpy.ndarray
"""epsilon (or mu, or whatever) grids. shape is (num_grids, X, Y, Z)"""
for i in range(3): for i in range(3):
if len(self.exyz[i]) != len(pixel_edge_coordinates[i]): if len(self.exyz[i]) != len(pixel_edge_coordinates[i]):
warnings.warn('Dimension {} had duplicate edge coordinates'.format(i), stacklevel=2) warnings.warn('Dimension {} had duplicate edge coordinates'.format(i), stacklevel=2)
if is_scalar(periodic): if is_scalar(periodic):
periodic = [periodic] * 3 periodic = [periodic] * 3
self.periodic = periodic self.periodic = periodic # type: List[bool]
"""For each axis, determines how far the rightmost boundary gets shifted. """
self.shifts = numpy.array(shifts, dtype=float) # type: numpy.ndarray
"""Offsets `[[x0, y0, z0], [x1, y1, z1], ...]` for grid `0,1,...`"""
self.shifts = numpy.array(shifts, dtype=float)
if len(self.shifts.shape) != 2: if len(self.shifts.shape) != 2:
raise GridError('Misshapen shifts: shifts must have two axes! ' raise GridError('Misshapen shifts: shifts must have two axes! '
' The given shifts has shape {}'.format(self.shifts.shape)) ' The given shifts has shape {}'.format(self.shifts.shape))