Source code for pycbc.types.aligned

# Copyright (C) 2014  Josh Willis, Alex Nitz
#
# This program is free software; you can redistribute it and/or modify it
# 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, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

#
# =============================================================================
#
#                                   Preamble
#
# =============================================================================
#
"""
This module provides a class derived from numpy.ndarray that also indicates
whether or not its memory is aligned.  It further provides functions for
creating zeros and empty (unitialized) arrays with this class.
"""
import numpy as _np
from pycbc import PYCBC_ALIGNMENT

[docs]def check_aligned(ndarr):
return ((ndarr.ctypes.data % PYCBC_ALIGNMENT) == 0)

[docs]def zeros(n, dtype):
d = _np.dtype(dtype)
nbytes = (d.itemsize)*int(n)
tmp = _np.zeros(nbytes+PYCBC_ALIGNMENT, dtype=_np.uint8)
ret_ary = tmp[offset:offset+nbytes].view(dtype=d)
del tmp
return ret_ary

[docs]def empty(n, dtype):
d = _np.dtype(dtype)
nbytes = (d.itemsize)*int(n)
tmp = _np.empty(nbytes+PYCBC_ALIGNMENT, dtype=_np.uint8)