3
0
mirror of https://github.com/triqs/dft_tools synced 2024-12-25 13:53:40 +01:00
dft_tools/pytriqs/gf/local/matrix_stack.pyx
Michel Ferrero fe1aeb8dbf Fix for compatibility with 32-bit systems
This is a fix for issue #18. The type float128 doesn't exist on 32-bit
machines. I removed it from the matrix_stack.pyx detection of a scalar.

  modified:   pytriqs/gf/local/matrix_stack.pyx
2013-09-26 21:59:21 +02:00

35 lines
1.2 KiB
Cython

import numpy
class MatrixStack:
def __init__(self, A):
"""
"""
self.data = A
self.is_double = A.dtype in [numpy.float, numpy.float32, numpy.float64, numpy.float_]
def add(self, M):
if self.is_double:
matrix_stack[double](self.data).iadd(matrix_view[double](M))
else:
matrix_stack[dcomplex](self.data).iadd(matrix_view[dcomplex](M.astype(numpy.complex)))
def sub(self, M):
if self.is_double:
matrix_stack[double](self.data).isub(matrix_view[double](M))
else:
matrix_stack[dcomplex](self.data).isub(matrix_view[dcomplex](M.astype(numpy.complex)))
def matmul_L_R(self, L, D, R):
if self.is_double:
matrix_stack[double](self.data).onsite_matmul_L_R(matrix_view[double](L), matrix_stack[double](D), matrix_view[double](R))
else:
matrix_stack[dcomplex](self.data).onsite_matmul_L_R(
matrix_view[dcomplex](L.astype(numpy.complex)), matrix_stack[dcomplex](D), matrix_view[dcomplex](R.astype(numpy.complex)))
def invert(self):
if self.is_double:
matrix_stack[double](self.data).invert()
else:
matrix_stack[dcomplex](self.data).invert()