mirror of
https://github.com/triqs/dft_tools
synced 2024-12-27 06:43:40 +01:00
35 lines
1.2 KiB
Cython
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()
|