mirror of
https://github.com/triqs/dft_tools
synced 2024-11-01 03:33:50 +01:00
35 lines
1.2 KiB
Cython
35 lines
1.2 KiB
Cython
|
cdef class MeshReFreq:
|
||
|
|
cdef mesh_refreq _c
|
||
|
|
|
||
|
|
def __init__(self, double omega_min, double omega_max, int n_max, kind):
|
||
|
|
self._c = make_mesh_refreq(omega_min, omega_max, n_max, {'H':half_bins, 'F':full_bins, 'W':without_last}[kind])
|
||
|
|
|
||
|
|
property omega_min:
|
||
|
|
def __get__(self): return self._c.x_min()
|
||
|
|
|
||
|
|
property omega_max:
|
||
|
|
def __get__(self): return self._c.x_max()
|
||
|
|
|
||
|
|
property kind:
|
||
|
|
def __get__(self): return self._c.kind()
|
||
|
|
|
||
|
|
def __len__ (self) : return self._c.size()
|
||
|
|
|
||
|
|
def __iter__(self) : # I use the C++ generator !
|
||
|
|
cdef mesh_pt_generator[mesh_refreq] g = mesh_pt_generator[mesh_refreq](&self._c)
|
||
|
|
while not g.at_end() :
|
||
|
|
yield g.to_point()
|
||
|
|
g.increment()
|
||
|
|
|
||
|
|
def __richcmp__(MeshReFreq self, MeshReFreq other, int op) :
|
||
|
|
if op ==2 : # ==
|
||
|
|
return self._c == other._c
|
||
|
|
|
||
|
|
def __reduce__(self):
|
||
|
|
return self.__class__, (self.omega_min, self.omega_max, len(self), self.kind)
|
||
|
|
|
||
|
|
# C -> Python
|
||
|
|
cdef inline make_MeshReFreq ( mesh_refreq x) :
|
||
|
|
return MeshReFreq( x.x_min(), x.x_max(), x.size(), {half_bins: 'H', full_bins: 'F', without_last: 'W'}[x.kind()] )
|
||
|
|
|