1from .graph.defs cimport gint 

2from .graph.defs import gint_dtype 

3  

4  

5cdef gint _bisect(gint[:] a, gint x) 

6  

7cdef int _is_hermitian( 

8 complex[:, :] a, double atol=*, double rtol=* 

9) except -1 

10  

11cdef int _is_hermitian_3d( 

12 complex[:, :, :] a, double atol=*, double rtol=* 

13) except -1 

14  

15cdef _select(gint[:, :] arr, gint[:] indexes) 

16  

17cdef int _check_onsite(complex[:, :] M, gint norbs, 

18 int check_hermiticity) except -1 

19  

20cdef int _check_onsites(complex[:, :, :] M, gint norbs, 

21 int check_hermiticity) except -1 

22  

23cdef int _check_hams(complex[:, :, :] H, gint to_norbs, gint from_norbs, 

24 int check_hermiticity) except -1 

25  

26cdef void _get_orbs(gint[:, :] site_ranges, gint site, 

27 gint *start_orb, gint *norbs) 

28  

29  

30cdef class BlockSparseMatrix: 

31 cdef public gint[:, :] block_offsets, block_shapes 

32 cdef public gint[:] data_offsets 

33 cdef public complex[:] data 

34  

35 cdef complex* get(self, gint block_idx) 

36  

37  

38cdef class _LocalOperator: 

39 cdef public int check_hermiticity, sum 

40 cdef public object syst, onsite, _onsite_param_names, _terms 

41 cdef public gint[:, :] where, _site_ranges 

42 cdef public BlockSparseMatrix _bound_onsite, _bound_hamiltonian 

43  

44 cdef BlockSparseMatrix _eval_onsites(self, args, params) 

45 cdef BlockSparseMatrix _eval_hamiltonian(self, args, params) 

46  

47  

48cdef class Density(_LocalOperator): 

49 pass 

50  

51  

52cdef class Current(_LocalOperator): 

53 pass 

54  

55  

56cdef class Source(_LocalOperator): 

57 pass