mirror of
https://github.com/triqs/dft_tools
synced 2024-12-25 13:53:40 +01:00
array & gf transpose
- implement transposed_view for arrays. - .transpose method for gf - wrapped to python - add call op. for GfImTime, using C++ - Added ChangeLog - rm matrix_stack - start cleaning old code
This commit is contained in:
Olivier Parcollet
parent
5c8270c437
commit
cfe3532c94
10
ChangeLog
Normal file
10
ChangeLog
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
|
||||||
|
From version 1.0 to 1.2
|
||||||
|
|
||||||
|
Python
|
||||||
|
----------
|
||||||
|
|
||||||
|
gf:
|
||||||
|
|
||||||
|
* transpose method now returns a NEW green function, like conjugate, etc...
|
||||||
|
|
||||||
@ -1,31 +0,0 @@
|
|||||||
############### Density #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/local/functions.hpp":
|
|
||||||
matrix_view density(gf_imfreq &)
|
|
||||||
matrix_view density(gf_legendre &)
|
|
||||||
void enforce_discontinuity(gf_legendre &, array_view[double,TWO])
|
|
||||||
|
|
||||||
############### Fourier #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/local/fourier_matsubara.hpp" :
|
|
||||||
gf_imfreq fourier (gf_imtime & )
|
|
||||||
gf_imtime inverse_fourier (gf_imfreq & )
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/local/fourier_real.hpp" :
|
|
||||||
gf_refreq fourier (gf_retime & )
|
|
||||||
gf_retime inverse_fourier (gf_refreq & )
|
|
||||||
#gf_refreq fourier (gf_retime & ) except +
|
|
||||||
#gf_retime inverse_fourier (gf_refreq & ) except +
|
|
||||||
|
|
||||||
############### Legendre #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/local/legendre_matsubara.hpp" :
|
|
||||||
gf_imfreq legendre_to_imfreq (gf_legendre &)
|
|
||||||
gf_imtime legendre_to_imtime (gf_legendre &)
|
|
||||||
gf_legendre imfreq_to_legendre (gf_imfreq &)
|
|
||||||
gf_legendre imtime_to_legendre (gf_imtime &)
|
|
||||||
|
|
||||||
############### Pade #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/local/pade.hpp":
|
|
||||||
void pade(gf_refreq &, gf_imfreq &, int, double) except +
|
|
||||||
@ -1,62 +0,0 @@
|
|||||||
from dcomplex cimport *
|
|
||||||
from arrays cimport *
|
|
||||||
from libcpp.vector cimport vector
|
|
||||||
from libcpp.string cimport string as std_string
|
|
||||||
from shared_ptr cimport *
|
|
||||||
from extractor cimport *
|
|
||||||
from h5 cimport *
|
|
||||||
|
|
||||||
# -------------------- Some generic tools -------------------------------
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/tools.hpp" namespace "triqs::gfs" :
|
|
||||||
cdef enum statistic_enum "triqs::gfs::statistic_enum" :
|
|
||||||
Boson,Fermion
|
|
||||||
|
|
||||||
cdef cppclass nothing :
|
|
||||||
nothing ()
|
|
||||||
|
|
||||||
cdef cppclass freq_infty:
|
|
||||||
freq_infty ()
|
|
||||||
|
|
||||||
cdef cppclass mesh_pt_generator "triqs::gfs::mesh_pt_generator" [MeshType] :
|
|
||||||
mesh_pt_generator( MeshType * )
|
|
||||||
mesh_pt_generator()
|
|
||||||
complex to_point()
|
|
||||||
mesh_pt_generator operator++()
|
|
||||||
bint at_end()
|
|
||||||
void increment()
|
|
||||||
|
|
||||||
cdef cppclass indices_2_t :
|
|
||||||
indices_2_t( )
|
|
||||||
indices_2_t( vector[vector[std_string]] &)
|
|
||||||
vector[std_string] & operator[](int)
|
|
||||||
vector[vector[std_string]] & operator()()
|
|
||||||
bint same()
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/meshes/linear.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef enum mesh_enum "triqs::gfs::mesh_kind":
|
|
||||||
half_bins, full_bins, without_last
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/block.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass discrete_domain :
|
|
||||||
discrete_domain ()
|
|
||||||
discrete_domain (vector[std_string] &)
|
|
||||||
vector[std_string] & names()
|
|
||||||
|
|
||||||
cdef cppclass discrete_mesh :
|
|
||||||
discrete_mesh ()
|
|
||||||
discrete_mesh (discrete_mesh &)
|
|
||||||
discrete_domain & domain()
|
|
||||||
long size()
|
|
||||||
bint operator ==( discrete_mesh &)
|
|
||||||
|
|
||||||
include "tail.pxd"
|
|
||||||
include "imfreq.pxd"
|
|
||||||
include "imtime.pxd"
|
|
||||||
include "refreq.pxd"
|
|
||||||
include "retime.pxd"
|
|
||||||
include "legendre.pxd"
|
|
||||||
#include "two_real_times.pxd"
|
|
||||||
|
|
||||||
@ -1,26 +0,0 @@
|
|||||||
#!python
|
|
||||||
#cython: embedsignature=True
|
|
||||||
from cython.operator cimport dereference as deref, preincrement as inc #dereference and increment operators
|
|
||||||
cimport cython
|
|
||||||
import numpy
|
|
||||||
import string
|
|
||||||
import warnings
|
|
||||||
from block_gf import BlockGf
|
|
||||||
from math import pi
|
|
||||||
from h5 cimport *
|
|
||||||
|
|
||||||
include "mesh_imfreq.pyx"
|
|
||||||
include "mesh_imtime.pyx"
|
|
||||||
include "mesh_refreq.pyx"
|
|
||||||
include "mesh_retime.pyx"
|
|
||||||
#include "mesh_two_real_times.pyx"
|
|
||||||
include "mesh_legendre.pyx"
|
|
||||||
include "imfreq.pyx"
|
|
||||||
include "imtime.pyx"
|
|
||||||
include "refreq.pyx"
|
|
||||||
include "retime.pyx"
|
|
||||||
#include "two_real_times.pyx"
|
|
||||||
include "legendre.pyx"
|
|
||||||
include "tail.pyx"
|
|
||||||
include "functions.pxd"
|
|
||||||
|
|
||||||
@ -1,57 +0,0 @@
|
|||||||
cdef extern from "triqs/gfs/imfreq.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass imfreq_domain :
|
|
||||||
double beta
|
|
||||||
statistic_enum statistic
|
|
||||||
imfreq_domain ()
|
|
||||||
|
|
||||||
cdef cppclass mesh_imfreq "triqs::gfs::gf_mesh<triqs::gfs::imfreq>" :
|
|
||||||
mesh_imfreq ()
|
|
||||||
mesh_imfreq (mesh_imfreq &)
|
|
||||||
imfreq_domain & domain()
|
|
||||||
double x_min()
|
|
||||||
long size()
|
|
||||||
bint positive_only()
|
|
||||||
bint operator ==( mesh_imfreq &)
|
|
||||||
|
|
||||||
cdef mesh_imfreq make_mesh_imfreq "triqs::gfs::gf_mesh<triqs::gfs::imfreq>" (double beta, statistic_enum S, size_t n_max, bint positive_only)
|
|
||||||
#cdef mesh_imfreq make_mesh_imfreq "triqs::gfs::gf_implementation::gf_factories<triqs::gfs::imfreq,triqs::gfs::matrix>::make_mesh" (double beta, statistic_enum S, size_t n_max)
|
|
||||||
|
|
||||||
cdef cppclass gf_imfreq "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::imfreq> >" :
|
|
||||||
gf_imfreq()
|
|
||||||
gf_imfreq(gf_imfreq &)
|
|
||||||
# The constructor must be no_except, or the cython code won't be correct...
|
|
||||||
gf_imfreq(mesh_imfreq, array_view[dcomplex, THREE], tail, nothing) #except +
|
|
||||||
void operator << (gf_imfreq &)
|
|
||||||
mesh_imfreq mesh()
|
|
||||||
array_view[dcomplex, THREE] data()
|
|
||||||
tail singularity()
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/imfreq.hpp" :
|
|
||||||
cdef void h5_write (h5_group, char *, gf_imfreq &)
|
|
||||||
|
|
||||||
cdef extern from "triqs/utility/serialization.hpp" :
|
|
||||||
cdef std_string boost_serialize "triqs::serialize" (gf_imfreq &)
|
|
||||||
cdef void boost_unserialize_into "triqs::deserialize_into_view" (std_string, gf_imfreq &)
|
|
||||||
|
|
||||||
# Python -> C
|
|
||||||
cdef gf_imfreq as_gf_imfreq (g) except +
|
|
||||||
|
|
||||||
# C -> Python
|
|
||||||
cdef make_GfImFreq (gf_imfreq x, indices_pack=*, name=*)
|
|
||||||
|
|
||||||
############### Blocks of Im Freq #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/block.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass gf_block_imfreq "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::imfreq>>>" :
|
|
||||||
gf_block_imfreq()
|
|
||||||
gf_imfreq & operator [](int)
|
|
||||||
discrete_mesh & mesh()
|
|
||||||
|
|
||||||
cdef gf_block_imfreq make_gf_block_imfreq "triqs::gfs::make_block_gf_view_from_vector_of_cython_proxy<triqs::gfs::gf<triqs::gfs::imfreq>>" (vector[gf_imfreq] &)
|
|
||||||
#cdef gf_block_imfreq make_gf_block_imfreq "triqs::gfs::make_gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::imfreq>>" ( vector[gf_imfreq] &)
|
|
||||||
|
|
||||||
cdef gf_block_imfreq as_gf_block_imfreq (G) except +
|
|
||||||
cdef make_BlockGfImFreq (gf_block_imfreq G, block_indices_pack=*, name=*)
|
|
||||||
|
|
||||||
@ -1,58 +0,0 @@
|
|||||||
cdef extern from "triqs/gfs/imtime.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass imtime_domain :
|
|
||||||
double beta
|
|
||||||
statistic_enum statistic
|
|
||||||
imtime_domain ()
|
|
||||||
|
|
||||||
cdef cppclass mesh_imtime "triqs::gfs::gf_mesh<triqs::gfs::imtime>" :
|
|
||||||
mesh_imtime ()
|
|
||||||
mesh_imtime (mesh_imtime &)
|
|
||||||
imtime_domain & domain()
|
|
||||||
double x_min()
|
|
||||||
long size()
|
|
||||||
long kind()
|
|
||||||
bint operator ==( mesh_imtime &)
|
|
||||||
|
|
||||||
cdef mesh_imtime make_mesh_imtime "triqs::gfs::gf_mesh<triqs::gfs::imtime>" (double beta, statistic_enum S, size_t n_time_slices, mesh_enum mk)
|
|
||||||
#cdef mesh_imtime make_mesh_imtime "triqs::gfs::gf_factories<triqs::gfs::imtime>::make_mesh" (double beta, statistic_enum S, size_t n_time_slices, mesh_enum mk)
|
|
||||||
|
|
||||||
cdef cppclass gf_imtime "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::imtime>>" :
|
|
||||||
gf_imtime()
|
|
||||||
gf_imtime(gf_imtime &)
|
|
||||||
# The constructor must be no_except, or the cython code won't be correct...
|
|
||||||
gf_imtime(mesh_imtime, array_view[double, THREE], tail, nothing) #except +
|
|
||||||
void operator << (gf_imtime &)
|
|
||||||
mesh_imtime mesh()
|
|
||||||
array_view[double, THREE] data()
|
|
||||||
matrix_view[double] operator()(double)
|
|
||||||
tail singularity()
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/imtime.hpp" :
|
|
||||||
cdef void h5_write (h5_group, char *, gf_imtime &)
|
|
||||||
|
|
||||||
cdef extern from "triqs/utility/serialization.hpp" :
|
|
||||||
cdef std_string boost_serialize "triqs::serialize" (gf_imtime &)
|
|
||||||
cdef void boost_unserialize_into "triqs::deserialize_into_view" (std_string, gf_imtime &)
|
|
||||||
|
|
||||||
# Python -> C
|
|
||||||
cdef gf_imtime as_gf_imtime (g) except +
|
|
||||||
|
|
||||||
# C -> Python
|
|
||||||
cdef make_GfImTime (gf_imtime x, indices_pack=*, name=*)
|
|
||||||
|
|
||||||
############### Blocks of Im Time #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/block.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass gf_block_imtime "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::imtime>>>" :
|
|
||||||
gf_block_imtime()
|
|
||||||
gf_imtime & operator [](int)
|
|
||||||
discrete_mesh & mesh()
|
|
||||||
|
|
||||||
cdef gf_block_imtime make_gf_block_imtime "triqs::gfs::make_block_gf_view_from_vector_of_cython_proxy<triqs::gfs::gf<triqs::gfs::imtime>>" (vector[gf_imtime] &)
|
|
||||||
#cdef gf_block_imtime make_gf_block_imtime "triqs::gfs::make_gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::imtime>>" ( vector[gf_imtime] &)
|
|
||||||
|
|
||||||
cdef gf_block_imtime as_gf_block_imtime (G) except +
|
|
||||||
cdef make_BlockGfImTime (gf_block_imtime G, block_indices_pack=*, name=*)
|
|
||||||
|
|
||||||
@ -1,54 +0,0 @@
|
|||||||
cdef extern from "triqs/gfs/legendre.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass legendre_domain :
|
|
||||||
size_t Nmax
|
|
||||||
double beta
|
|
||||||
statistic_enum statistic
|
|
||||||
legendre_domain ()
|
|
||||||
|
|
||||||
cdef cppclass mesh_legendre "triqs::gfs::gf_mesh<triqs::gfs::legendre>" :
|
|
||||||
mesh_legendre ()
|
|
||||||
mesh_legendre (mesh_legendre &)
|
|
||||||
legendre_domain & domain()
|
|
||||||
long size()
|
|
||||||
bint operator == (mesh_legendre &)
|
|
||||||
|
|
||||||
cdef mesh_legendre make_mesh_legendre "triqs::gfs::gf_mesh<triqs::gfs::legendre>" (double beta, statistic_enum S, size_t n_leg)
|
|
||||||
#cdef mesh_legendre make_mesh_legendre "triqs::gfs::gf_factories<triqs::gfs::legendre>::make_mesh" (double beta, statistic_enum S, size_t n_leg)
|
|
||||||
|
|
||||||
cdef cppclass gf_legendre "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::legendre>>" :
|
|
||||||
gf_legendre()
|
|
||||||
gf_legendre(gf_legendre &)
|
|
||||||
gf_legendre(mesh_legendre, array_view[double, THREE], nothing, nothing) #except +
|
|
||||||
void operator << (gf_legendre &)
|
|
||||||
mesh_legendre mesh()
|
|
||||||
array_view[double, THREE] data()
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/legendre.hpp" :
|
|
||||||
cdef void h5_write (h5_group, char *, gf_legendre &)
|
|
||||||
|
|
||||||
cdef extern from "triqs/utility/serialization.hpp" :
|
|
||||||
cdef std_string boost_serialize "triqs::serialize" (gf_legendre &)
|
|
||||||
cdef void boost_unserialize_into "triqs::deserialize_into_view" (std_string, gf_legendre &)
|
|
||||||
|
|
||||||
# Python -> C
|
|
||||||
cdef gf_legendre as_gf_legendre(g) except +
|
|
||||||
|
|
||||||
# C -> Python
|
|
||||||
cdef make_GfLegendre(gf_legendre x, indices_pack=*, name=*)
|
|
||||||
|
|
||||||
############### Blocks of Im Time #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/block.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass gf_block_legendre "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::legendre>>>" :
|
|
||||||
gf_block_legendre()
|
|
||||||
gf_legendre & operator [](int)
|
|
||||||
discrete_mesh & mesh()
|
|
||||||
|
|
||||||
cdef gf_block_legendre make_gf_block_legendre "triqs::gfs::make_block_gf_view_from_vector_of_cython_proxy<triqs::gfs::gf<triqs::gfs::legendre>>" (vector[gf_legendre] &)
|
|
||||||
#cdef gf_block_legendre make_gf_block_legendre "triqs::gfs::make_gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::legendre>>" ( vector[gf_legendre] &)
|
|
||||||
|
|
||||||
cdef gf_block_legendre as_gf_block_legendre (G) except +
|
|
||||||
cdef make_BlockGfLegendre (gf_block_legendre G, block_indices_pack=*, name=*)
|
|
||||||
|
|
||||||
@ -1,17 +0,0 @@
|
|||||||
include "arrays.pxd"
|
|
||||||
include "dcomplex.pxd"
|
|
||||||
|
|
||||||
cdef extern from "<triqs/arrays/matrix_stack_view.hpp>" namespace "triqs::arrays":
|
|
||||||
|
|
||||||
cdef cppclass matrix_stack "triqs::arrays::matrix_stack_view" [T]:
|
|
||||||
|
|
||||||
matrix_stack(object)
|
|
||||||
|
|
||||||
void iadd "operator +=" (matrix_view[T] &) except +
|
|
||||||
void isub "operator -=" (matrix_view[T] &) except +
|
|
||||||
|
|
||||||
void imul "operator *=" (matrix_view[T] &) except +
|
|
||||||
void idiv "operator /=" (matrix_view[T] &) except +
|
|
||||||
|
|
||||||
void invert() except +
|
|
||||||
void onsite_matmul_L_R(matrix_view[T] &, matrix_stack &, matrix_view[T] &) except +
|
|
||||||
@ -1,34 +0,0 @@
|
|||||||
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()
|
|
||||||
@ -1,36 +0,0 @@
|
|||||||
cdef class MeshImFreq:
|
|
||||||
cdef mesh_imfreq _c
|
|
||||||
|
|
||||||
def __init__(self, beta, stat, int n_pts, bint positive_only=True):
|
|
||||||
self._c = make_mesh_imfreq(beta,{'F' :Fermion, 'B' : Boson}[stat], n_pts, positive_only)
|
|
||||||
|
|
||||||
def __len__ (self) : return self._c.size()
|
|
||||||
|
|
||||||
property beta :
|
|
||||||
"""Inverse temperature"""
|
|
||||||
def __get__(self): return self._c.domain().beta
|
|
||||||
|
|
||||||
property statistic :
|
|
||||||
def __get__(self): return 'F' if self._c.domain().statistic==Fermion else 'B'
|
|
||||||
|
|
||||||
property positive_only :
|
|
||||||
def __get__(self): return self._c.positive_only()
|
|
||||||
|
|
||||||
def __iter__(self) : # I use the C++ generator !
|
|
||||||
cdef mesh_pt_generator[mesh_imfreq ] g = mesh_pt_generator[mesh_imfreq ](&self._c)
|
|
||||||
while not g.at_end() :
|
|
||||||
yield g.to_point()
|
|
||||||
g.increment()
|
|
||||||
|
|
||||||
def __richcmp__(MeshImFreq self, MeshImFreq other,int op) :
|
|
||||||
if op ==2 : # ==
|
|
||||||
return self._c == other._c
|
|
||||||
|
|
||||||
def __reduce__(self):
|
|
||||||
return self.__class__, (self.beta, self.statistic, len(self))
|
|
||||||
|
|
||||||
# C -> Python
|
|
||||||
cdef inline make_MeshImFreq (mesh_imfreq x) :
|
|
||||||
return MeshImFreq( x.domain().beta, 'F' if x.domain().statistic==Fermion else 'B', x.size(), x.positive_only() )
|
|
||||||
|
|
||||||
|
|
||||||
@ -1,37 +0,0 @@
|
|||||||
cdef class MeshImTime:
|
|
||||||
cdef mesh_imtime _c
|
|
||||||
|
|
||||||
def __init__(self, beta, stat, int n_max, kind):
|
|
||||||
self._c = make_mesh_imtime(beta, {'F' :Fermion, 'B' : Boson}[stat], n_max, {'H':half_bins, 'F':full_bins, 'W':without_last}[kind])
|
|
||||||
|
|
||||||
def __len__ (self) : return self._c.size()
|
|
||||||
|
|
||||||
property beta :
|
|
||||||
"""Inverse temperature"""
|
|
||||||
def __get__(self): return self._c.domain().beta
|
|
||||||
|
|
||||||
property kind :
|
|
||||||
"""Inverse temperature"""
|
|
||||||
def __get__(self): return {half_bins: 'H', full_bins: 'F', without_last: 'W'}[self._c.kind()]
|
|
||||||
|
|
||||||
property statistic :
|
|
||||||
def __get__(self): return 'F' if self._c.domain().statistic==Fermion else 'B'
|
|
||||||
|
|
||||||
def __iter__(self) : # I use the C++ generator !
|
|
||||||
cdef mesh_pt_generator[mesh_imtime ] g = mesh_pt_generator[mesh_imtime ](&self._c)
|
|
||||||
while not g.at_end() :
|
|
||||||
yield g.to_point()
|
|
||||||
g.increment()
|
|
||||||
|
|
||||||
def __richcmp__(MeshImTime self, MeshImTime other,int op) :
|
|
||||||
if op ==2 : # ==
|
|
||||||
return self._c == other._c
|
|
||||||
|
|
||||||
def __reduce__(self):
|
|
||||||
return self.__class__, (self.beta, self.statistic, len(self), self.kind)
|
|
||||||
|
|
||||||
# C -> Python
|
|
||||||
cdef inline make_MeshImTime ( mesh_imtime x) :
|
|
||||||
return MeshImTime( x.domain().beta, 'F' if x.domain().statistic==Fermion else 'B',
|
|
||||||
x.size(), {half_bins: 'H', full_bins: 'F', without_last: 'W'}[x.kind()] )
|
|
||||||
|
|
||||||
@ -1,34 +0,0 @@
|
|||||||
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 {half_bins: 'H', full_bins: 'F', without_last: 'W'}[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()] )
|
|
||||||
|
|
||||||
@ -1,34 +0,0 @@
|
|||||||
cdef class MeshReTime:
|
|
||||||
cdef mesh_retime _c
|
|
||||||
|
|
||||||
def __init__(self, double t_min, double t_max, int n_max, kind):
|
|
||||||
self._c = make_mesh_retime(t_min, t_max, n_max, {'H':half_bins, 'F':full_bins, 'W':without_last}[kind])
|
|
||||||
|
|
||||||
property t_min:
|
|
||||||
def __get__(self): return self._c.x_min()
|
|
||||||
|
|
||||||
property t_max:
|
|
||||||
def __get__(self): return self._c.x_max()
|
|
||||||
|
|
||||||
property kind:
|
|
||||||
def __get__(self): return {half_bins: 'H', full_bins: 'F', without_last: 'W'}[self._c.kind()]
|
|
||||||
|
|
||||||
def __len__ (self) : return self._c.size()
|
|
||||||
|
|
||||||
def __iter__(self) : # I use the C++ generator !
|
|
||||||
cdef mesh_pt_generator[mesh_retime] g = mesh_pt_generator[mesh_retime](&self._c)
|
|
||||||
while not g.at_end() :
|
|
||||||
yield g.to_point()
|
|
||||||
g.increment()
|
|
||||||
|
|
||||||
def __richcmp__(MeshReTime self, MeshReTime other, int op) :
|
|
||||||
if op ==2 : # ==
|
|
||||||
return self._c == other._c
|
|
||||||
|
|
||||||
def __reduce__(self):
|
|
||||||
return self.__class__, (self.t_min, self.t_max, len(self), self.kind)
|
|
||||||
|
|
||||||
# C -> Python
|
|
||||||
cdef inline make_MeshReTime ( mesh_retime x) :
|
|
||||||
return MeshReTime( x.x_min(), x.x_max(), x.size(), {half_bins: 'H', full_bins: 'F', without_last: 'W'}[x.kind()] )
|
|
||||||
|
|
||||||
@ -1,43 +0,0 @@
|
|||||||
|
|
||||||
################################################################################
|
|
||||||
#
|
|
||||||
# TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
|
||||||
#
|
|
||||||
# Copyright (C) 2011-2012 by M. Ferrero, O. Parcollet
|
|
||||||
#
|
|
||||||
# TRIQS is free software: you can redistribute it and/or modify it under the
|
|
||||||
# terms of the GNU General Public License as published by the Free Software
|
|
||||||
# Foundation, either version 3 of the License, or (at your option) any later
|
|
||||||
# version.
|
|
||||||
#
|
|
||||||
# TRIQS 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
|
|
||||||
# TRIQS. If not, see <http://www.gnu.org/licenses/>.
|
|
||||||
#
|
|
||||||
################################################################################
|
|
||||||
|
|
||||||
class Nothing (object):
|
|
||||||
|
|
||||||
def __init__(self): pass
|
|
||||||
def __getitem__(self, key): return Nothing()
|
|
||||||
def __setitem__(self, key, val): pass
|
|
||||||
def __iadd__(self, arg): return Nothing()
|
|
||||||
def __add__(self, y): return Nothing()
|
|
||||||
def __radd__(self, y): return Nothing()
|
|
||||||
def __isub__(self, arg): return Nothing()
|
|
||||||
def __sub__(self, y): return Nothing()
|
|
||||||
def __rsub__(self, y): return Nothing()
|
|
||||||
def __imul__(self, arg): return Nothing()
|
|
||||||
def __mul__(self, arg): return Nothing()
|
|
||||||
def __rmul__(self, x): return Nothing()
|
|
||||||
def __idiv__(self, arg): return Nothing()
|
|
||||||
def __div__(self, arg): return Nothing()
|
|
||||||
def _make_slice(self, sl1, sl2): return Nothing()
|
|
||||||
def invert(self): pass
|
|
||||||
def transpose(self): return Nothing()
|
|
||||||
def conjugate(self): return Nothing()
|
|
||||||
def copy(self): return Nothing()
|
|
||||||
@ -1,56 +0,0 @@
|
|||||||
cdef extern from "triqs/gfs/refreq.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass refreq_domain :
|
|
||||||
refreq_domain()
|
|
||||||
|
|
||||||
cdef cppclass mesh_refreq "triqs::gfs::gf_mesh<triqs::gfs::refreq>" :
|
|
||||||
mesh_refreq ()
|
|
||||||
mesh_refreq (mesh_refreq &)
|
|
||||||
refreq_domain & domain()
|
|
||||||
double x_min()
|
|
||||||
double x_max()
|
|
||||||
long size()
|
|
||||||
long kind()
|
|
||||||
bint operator ==( mesh_refreq &)
|
|
||||||
|
|
||||||
cdef mesh_refreq make_mesh_refreq "triqs::gfs::gf_mesh<triqs::gfs::refreq>" (double omega_min, double omega_max, size_t n_freq, mesh_enum mk)
|
|
||||||
#cdef mesh_refreq make_mesh_refreq "triqs::gfs::gf_factories<triqs::gfs::refreq>::make_mesh" (double omega_min, double omega_max, size_t n_freq, mesh_enum mk)
|
|
||||||
|
|
||||||
cdef cppclass gf_refreq "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::refreq>>" :
|
|
||||||
gf_refreq()
|
|
||||||
gf_refreq(gf_refreq &)
|
|
||||||
# The constructor must be no_except, or the cython code won't be correct...
|
|
||||||
gf_refreq(mesh_refreq, array_view[dcomplex, THREE], tail, nothing) #except +
|
|
||||||
void operator << (gf_refreq &)
|
|
||||||
mesh_refreq mesh()
|
|
||||||
array_view[dcomplex, THREE] data()
|
|
||||||
tail singularity()
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/refreq.hpp" :
|
|
||||||
cdef void h5_write (h5_group, char *, gf_refreq &)
|
|
||||||
|
|
||||||
cdef extern from "triqs/utility/serialization.hpp" :
|
|
||||||
cdef std_string boost_serialize "triqs::serialize" (gf_refreq &)
|
|
||||||
cdef void boost_unserialize_into "triqs::deserialize_into_view" (std_string, gf_refreq &)
|
|
||||||
|
|
||||||
# Python -> C
|
|
||||||
cdef gf_refreq as_gf_refreq (g) except +
|
|
||||||
|
|
||||||
# C -> Python
|
|
||||||
cdef make_GfReFreq (gf_refreq x, indices_pack=*, name=*)
|
|
||||||
|
|
||||||
############### Blocks of Im Time #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/block.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass gf_block_refreq "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::refreq>>>" :
|
|
||||||
gf_block_refreq()
|
|
||||||
gf_refreq & operator [](int)
|
|
||||||
discrete_mesh & mesh()
|
|
||||||
|
|
||||||
cdef gf_block_refreq make_gf_block_refreq "triqs::gfs::make_block_gf_view_from_vector_of_cython_proxy<triqs::gfs::gf<triqs::gfs::refreq>>" (vector[gf_refreq] &)
|
|
||||||
#cdef gf_block_refreq make_gf_block_refreq "triqs::gfs::make_gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::refreq>>" ( vector[gf_refreq] &)
|
|
||||||
|
|
||||||
cdef gf_block_refreq as_gf_block_refreq (G) except +
|
|
||||||
cdef make_BlockGfReFreq (gf_block_refreq G, block_indices_pack=*, name=*)
|
|
||||||
|
|
||||||
@ -1,79 +0,0 @@
|
|||||||
from gf_refreq import GfReFreq
|
|
||||||
|
|
||||||
cdef class GfReFreq_cython:
|
|
||||||
cdef gf_refreq _c
|
|
||||||
def __init__(self, MeshReFreq mesh, data, TailGf tail):
|
|
||||||
self._c = gf_refreq(mesh._c, array_view[dcomplex,THREE](data), tail._c, nothing())
|
|
||||||
|
|
||||||
def __write_hdf5_cython__ (self, gr , char * key) :
|
|
||||||
h5_write (make_h5_group(gr), key, self._c)
|
|
||||||
|
|
||||||
def set_from_fourier(self, GfReTime_cython gt) :
|
|
||||||
"""Fills self with the Fourier transform of gt"""
|
|
||||||
self._c << fourier( gt._c )
|
|
||||||
|
|
||||||
# put if back with make_gf_from_fourier when approved
|
|
||||||
#def inverse_fourier(self):
|
|
||||||
# return make_GfReTime(inverse_fourier(self._c))
|
|
||||||
|
|
||||||
def set_from_pade(self, GfImFreq_cython gw, n_points = 100, freq_offset = 0.0) :
|
|
||||||
pade(self._c, gw._c, n_points, freq_offset)
|
|
||||||
|
|
||||||
def __dealloc__ (self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
#---------------- Reading from h5 ---------------------------------------
|
|
||||||
|
|
||||||
def h5_read_GfReFreq(gr, key):
|
|
||||||
try:
|
|
||||||
indicesL = gr[key]['indices']['left']
|
|
||||||
indicesR = gr[key]['indices']['right']
|
|
||||||
pack = [indicesL, indicesR]
|
|
||||||
except:
|
|
||||||
pack = []
|
|
||||||
try:
|
|
||||||
name = gr[key]['name']
|
|
||||||
except:
|
|
||||||
name = key
|
|
||||||
return make_GfReFreq(h5_extractor[gf_refreq]()(make_h5_group(gr),key), pack, name)
|
|
||||||
|
|
||||||
from pytriqs.archive.hdf_archive_schemes import register_class
|
|
||||||
register_class (GfReFreq, read_fun = h5_read_GfReFreq)
|
|
||||||
|
|
||||||
#---------------- Convertions functions ---------------------------------------
|
|
||||||
|
|
||||||
# Python -> C
|
|
||||||
cdef gf_refreq as_gf_refreq (g) except +:
|
|
||||||
return (<GfReFreq_cython?>g)._c
|
|
||||||
|
|
||||||
# C -> Python. Do NOT add except +
|
|
||||||
cdef make_GfReFreq (gf_refreq x, indices_pack = [], name = "g"):
|
|
||||||
data = x.data().to_python()
|
|
||||||
if indices_pack == []:
|
|
||||||
indices_pack = [range(data.shape[1]), range(data.shape[2])]
|
|
||||||
return GfReFreq(
|
|
||||||
mesh = make_MeshReFreq (x.mesh()),
|
|
||||||
data = data,
|
|
||||||
tail = make_TailGf (x.singularity()),
|
|
||||||
indices_pack = indices_pack,
|
|
||||||
name = name)
|
|
||||||
|
|
||||||
# Python -> C for blocks
|
|
||||||
cdef gf_block_refreq as_gf_block_refreq (G) except +:
|
|
||||||
cdef vector[gf_refreq] v_c
|
|
||||||
for n,g in G:
|
|
||||||
v_c.push_back(as_gf_refreq(g))
|
|
||||||
return make_gf_block_refreq (v_c)
|
|
||||||
|
|
||||||
# C -> Python for block
|
|
||||||
cdef make_BlockGfReFreq (gf_block_refreq G, block_indices_pack = [], name = "G"):
|
|
||||||
gl = []
|
|
||||||
name_list = G.mesh().domain().names()
|
|
||||||
if block_indices_pack == []:
|
|
||||||
for i,n in enumerate(name_list):
|
|
||||||
sha = G[i].data().to_python().shape[1:3]
|
|
||||||
block_indices_pack.append( [range(sha[0]), range(sha[1])] )
|
|
||||||
for i,n in enumerate(name_list):
|
|
||||||
gl.append( make_GfReFreq(G[i], block_indices_pack[i]) )
|
|
||||||
return BlockGf( name_list = name_list, block_list = gl, name = name )
|
|
||||||
|
|
||||||
@ -1,56 +0,0 @@
|
|||||||
cdef extern from "triqs/gfs/retime.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass retime_domain :
|
|
||||||
retime_domain()
|
|
||||||
|
|
||||||
cdef cppclass mesh_retime "triqs::gfs::gf_mesh<triqs::gfs::retime>" :
|
|
||||||
mesh_retime ()
|
|
||||||
mesh_retime (mesh_retime &)
|
|
||||||
retime_domain & domain()
|
|
||||||
double x_min()
|
|
||||||
double x_max()
|
|
||||||
long size()
|
|
||||||
long kind()
|
|
||||||
bint operator ==( mesh_retime &)
|
|
||||||
|
|
||||||
cdef mesh_retime make_mesh_retime "triqs::gfs::gf_mesh<triqs::gfs::retime>" (double t_min, double t_max, size_t n_freq, mesh_enum mk)
|
|
||||||
#cdef mesh_retime make_mesh_retime "triqs::gfs::gf_factories<triqs::gfs::retime>::make_mesh" (double t_min, double t_max, size_t n_freq, mesh_enum mk)
|
|
||||||
|
|
||||||
cdef cppclass gf_retime "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::retime>>" :
|
|
||||||
gf_retime()
|
|
||||||
gf_retime(gf_retime &)
|
|
||||||
# The constructor must be no_except, or the cython code won't be correct...
|
|
||||||
gf_retime(mesh_retime, array_view[dcomplex, THREE], tail, nothing) #except +
|
|
||||||
void operator << (gf_retime &)
|
|
||||||
mesh_retime mesh()
|
|
||||||
array_view[dcomplex, THREE] data()
|
|
||||||
tail singularity()
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/retime.hpp" :
|
|
||||||
cdef void h5_write (h5_group, char *, gf_retime &)
|
|
||||||
|
|
||||||
cdef extern from "triqs/utility/serialization.hpp" :
|
|
||||||
cdef std_string boost_serialize "triqs::serialize" (gf_retime &)
|
|
||||||
cdef void boost_unserialize_into "triqs::deserialize_into_view" (std_string, gf_retime &)
|
|
||||||
|
|
||||||
# Python -> C
|
|
||||||
cdef gf_retime as_gf_retime (g) except +
|
|
||||||
|
|
||||||
# C -> Python
|
|
||||||
cdef make_GfReTime (gf_retime x, indices_pack=*, name=*)
|
|
||||||
|
|
||||||
############### Blocks of Im Time #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/block.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass gf_block_retime "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::retime>>>" :
|
|
||||||
gf_block_retime()
|
|
||||||
gf_retime & operator [](int)
|
|
||||||
discrete_mesh & mesh()
|
|
||||||
|
|
||||||
cdef gf_block_retime make_gf_block_retime "triqs::gfs::make_block_gf_view_from_vector_of_cython_proxy<triqs::gfs::gf<triqs::gfs::retime>>" (vector[gf_retime] &)
|
|
||||||
#cdef gf_block_retime make_gf_block_retime "triqs::gfs::make_gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::retime>>" ( vector[gf_retime] &)
|
|
||||||
|
|
||||||
cdef gf_block_retime as_gf_block_retime (G) except +
|
|
||||||
cdef make_BlockGfReTime (gf_block_retime G, block_indices_pack=*, name=*)
|
|
||||||
|
|
||||||
@ -1,76 +0,0 @@
|
|||||||
from gf_retime import GfReTime
|
|
||||||
|
|
||||||
cdef class GfReTime_cython:
|
|
||||||
cdef gf_retime _c
|
|
||||||
def __init__(self, MeshReTime mesh, data, TailGf tail):
|
|
||||||
self._c = gf_retime(mesh._c, array_view[dcomplex,THREE](data), tail._c, nothing())
|
|
||||||
|
|
||||||
def __write_hdf5_cython__ (self, gr , char * key) :
|
|
||||||
h5_write (make_h5_group(gr), key, self._c)
|
|
||||||
|
|
||||||
# Put it back ?
|
|
||||||
#def fourier(self):
|
|
||||||
# return make_GfReFreq(fourier(self._c))
|
|
||||||
|
|
||||||
def set_from_inverse_fourier(self, GfReFreq_cython gw) :
|
|
||||||
"""Fills self with the Inverse Fourier transform of gw"""
|
|
||||||
self._c << inverse_fourier( gw._c)
|
|
||||||
|
|
||||||
def __dealloc__ (self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
#---------------- Reading from h5 ---------------------------------------
|
|
||||||
|
|
||||||
def h5_read_GfReTime(gr, key):
|
|
||||||
try:
|
|
||||||
indicesL = gr[key]['indices']['left']
|
|
||||||
indicesR = gr[key]['indices']['right']
|
|
||||||
pack = [indicesL, indicesR]
|
|
||||||
except:
|
|
||||||
pack = []
|
|
||||||
try:
|
|
||||||
name = gr[key]['name']
|
|
||||||
except:
|
|
||||||
name = key
|
|
||||||
return make_GfReTime(h5_extractor[gf_retime]()(make_h5_group(gr),key), pack, name)
|
|
||||||
|
|
||||||
from pytriqs.archive.hdf_archive_schemes import register_class
|
|
||||||
register_class (GfReTime, read_fun = h5_read_GfReTime)
|
|
||||||
|
|
||||||
#---------------- Convertions functions ---------------------------------------
|
|
||||||
|
|
||||||
# Python -> C
|
|
||||||
cdef gf_retime as_gf_retime (g) except +:
|
|
||||||
return (<GfReTime_cython?>g)._c
|
|
||||||
|
|
||||||
# C -> Python. Do NOT add except +
|
|
||||||
cdef make_GfReTime (gf_retime x, indices_pack = [], name = "g"):
|
|
||||||
data = x.data().to_python()
|
|
||||||
if indices_pack == []:
|
|
||||||
indices_pack = [range(data.shape[1]), range(data.shape[2])]
|
|
||||||
return GfReTime(
|
|
||||||
mesh = make_MeshReTime (x.mesh()),
|
|
||||||
data = data,
|
|
||||||
tail = make_TailGf (x.singularity()),
|
|
||||||
indices_pack = indices_pack,
|
|
||||||
name = name)
|
|
||||||
|
|
||||||
# Python -> C for blocks
|
|
||||||
cdef gf_block_retime as_gf_block_retime (G) except +:
|
|
||||||
cdef vector[gf_retime] v_c
|
|
||||||
for n,g in G:
|
|
||||||
v_c.push_back(as_gf_retime(g))
|
|
||||||
return make_gf_block_retime (v_c)
|
|
||||||
|
|
||||||
# C -> Python for block
|
|
||||||
cdef make_BlockGfReTime (gf_block_retime G, block_indices_pack = [], name = "G"):
|
|
||||||
gl = []
|
|
||||||
name_list = G.mesh().domain().names()
|
|
||||||
if block_indices_pack == []:
|
|
||||||
for i,n in enumerate(name_list):
|
|
||||||
sha = G[i].data().to_python().shape[1:3]
|
|
||||||
block_indices_pack.append( [range(sha[0]), range(sha[1])] )
|
|
||||||
for i,n in enumerate(name_list):
|
|
||||||
gl.append( make_GfReTime(G[i], block_indices_pack[i]) )
|
|
||||||
return BlockGf( name_list = name_list, block_list = gl, name = name )
|
|
||||||
|
|
||||||
@ -1,37 +0,0 @@
|
|||||||
from dcomplex cimport *
|
|
||||||
from arrays cimport *
|
|
||||||
cdef extern from "triqs/gfs/local/tail.hpp" :
|
|
||||||
cdef cppclass tail "triqs::python_tools::cython_proxy<triqs::gfs::local::tail_view>" :
|
|
||||||
tail()
|
|
||||||
tail(array_view[dcomplex,THREE], array_view[long,TWO], long) except +
|
|
||||||
matrix_view[dcomplex] operator()(int) except +
|
|
||||||
array_view[dcomplex,THREE] data()
|
|
||||||
array_view[long,TWO] mask_view() except +
|
|
||||||
void operator << (tail &) except +
|
|
||||||
void operator << (dcomplex &) except +
|
|
||||||
long order_min() except +
|
|
||||||
long order_max() except +
|
|
||||||
size_t size() except +
|
|
||||||
size_t shape(int) except +
|
|
||||||
|
|
||||||
cdef tail operator +( tail &, tail &) except +
|
|
||||||
cdef tail operator -( tail &, tail &) except +
|
|
||||||
|
|
||||||
cdef tail operator *( tail&, tail &) except +
|
|
||||||
|
|
||||||
cdef tail operator *( double, tail &) except +
|
|
||||||
cdef tail operator *( tail &, double) except +
|
|
||||||
cdef tail operator /( double, tail &) except +
|
|
||||||
cdef tail operator /( tail &, double) except +
|
|
||||||
|
|
||||||
cdef tail operator *( dcomplex, tail &) except +
|
|
||||||
cdef tail operator *( tail &, dcomplex) except +
|
|
||||||
cdef tail operator /( dcomplex, tail &) except +
|
|
||||||
cdef tail operator /( tail &, dcomplex) except +
|
|
||||||
cdef tail inverse_c "inverse" ( tail &) except +
|
|
||||||
|
|
||||||
cdef void h5_write (h5_group, char *, tail &)
|
|
||||||
|
|
||||||
cdef extern from "triqs/utility/serialization.hpp" :
|
|
||||||
cdef std_string boost_serialize "triqs::serialize" (tail &)
|
|
||||||
cdef void boost_unserialize_into "triqs::deserialize_into_view" (std_string, tail &)
|
|
||||||
@ -8,19 +8,7 @@ cdef class TailGf:
|
|||||||
TailGf ( shape )
|
TailGf ( shape )
|
||||||
TailGf ( data, mask, order_min )
|
TailGf ( data, mask, order_min )
|
||||||
"""
|
"""
|
||||||
c_obj = d.pop('encapsulated_c_object', None)
|
# default values
|
||||||
if c_obj :
|
|
||||||
assert d == {}
|
|
||||||
self._c = extractor [tail] (c_obj) ()
|
|
||||||
return
|
|
||||||
|
|
||||||
bss = d.pop('boost_serialization_string', None)
|
|
||||||
if bss :
|
|
||||||
assert d == {}, "Internal error : boost_serialization_string must be the only argument"
|
|
||||||
boost_unserialize_into(<std_string>bss,self._c)
|
|
||||||
return
|
|
||||||
|
|
||||||
# default values
|
|
||||||
omin = -1
|
omin = -1
|
||||||
omax = 8
|
omax = 8
|
||||||
|
|
||||||
@ -39,50 +27,12 @@ cdef class TailGf:
|
|||||||
|
|
||||||
self._c = tail(array_view[dcomplex,THREE](a), array_view[long,TWO](m), o)
|
self._c = tail(array_view[dcomplex,THREE](a), array_view[long,TWO](m), o)
|
||||||
|
|
||||||
#-------------- Reduction -------------------------------
|
|
||||||
|
|
||||||
def __reduce__(self):
|
|
||||||
return py_deserialize, (self.__class__,boost_serialize(self._c),)
|
|
||||||
|
|
||||||
#-------------- Properties -------------------------------
|
|
||||||
|
|
||||||
property data :
|
|
||||||
"""Access to the data array"""
|
|
||||||
def __get__(self) :
|
|
||||||
return self._c.data().to_python()
|
|
||||||
|
|
||||||
property mask:
|
|
||||||
"""Access to the mask"""
|
|
||||||
def __get__(self) :
|
|
||||||
return self._c.mask_view().to_python()
|
|
||||||
|
|
||||||
property shape :
|
|
||||||
def __get__(self) : return self.data.shape[1:3]
|
|
||||||
|
|
||||||
property order_min :
|
|
||||||
"""Min order of the expansion"""
|
|
||||||
def __get__(self) : return self._c.order_min()
|
|
||||||
|
|
||||||
property order_max :
|
|
||||||
"""Max order of the expansion"""
|
|
||||||
def __get__(self) : return self._c.order_max()
|
|
||||||
|
|
||||||
property N1 :
|
property N1 :
|
||||||
def __get__(self): return self.shape[0]
|
def __get__(self): return self.shape[0]
|
||||||
|
|
||||||
property N2 :
|
property N2 :
|
||||||
def __get__(self): return self.shape[1]
|
def __get__(self): return self.shape[1]
|
||||||
|
|
||||||
property size :
|
|
||||||
"""Length of the expansion"""
|
|
||||||
def __get__(self) : return self._c.size()
|
|
||||||
|
|
||||||
def copy(self) :
|
|
||||||
return self.__class__(data = self.data.copy(), mask = self.mask.copy())
|
|
||||||
|
|
||||||
def copy_from(self, TailGf T) :
|
|
||||||
self._c << T._c
|
|
||||||
|
|
||||||
def _make_slice(self, sl1, sl2):
|
def _make_slice(self, sl1, sl2):
|
||||||
return self.__class__(data = self.data[:,sl1,sl2], mask = self.mask[sl1,sl2])
|
return self.__class__(data = self.data[:,sl1,sl2], mask = self.mask[sl1,sl2])
|
||||||
|
|
||||||
@ -95,85 +45,13 @@ cdef class TailGf:
|
|||||||
else:
|
else:
|
||||||
return string.join([ "%s"%self[r]+ (" /" if r>0 else "") + " Om^%s"%(abs(r)) for r in range(omin, self.order_max+1) ] , " + ")
|
return string.join([ "%s"%self[r]+ (" /" if r>0 else "") + " Om^%s"%(abs(r)) for r in range(omin, self.order_max+1) ] , " + ")
|
||||||
|
|
||||||
def __getitem__(self,i) :
|
|
||||||
"""Returns the i-th coefficient of the expansion, or order Om^i"""
|
|
||||||
if not self.has_coef(i) : raise IndexError, "Index %s is out of range"%i
|
|
||||||
return self.data[i-self.order_min,:,:]
|
|
||||||
|
|
||||||
def __setitem__(self,i, val) :
|
|
||||||
"""Sets the i-th coefficient of the expansion, or order Om^i"""
|
|
||||||
if not self.has_coef(i) : raise IndexError, "Index %s is out of range"%i
|
|
||||||
self.data[i-self.order_min,:,:] = val
|
|
||||||
|
|
||||||
def has_coef(self, i):
|
|
||||||
return (i >= self.order_min) and (i <= self.order_max)
|
|
||||||
|
|
||||||
def __call__(self, x) :
|
def __call__(self, x) :
|
||||||
val = 0.0
|
val = 0.0
|
||||||
for n in range(self.order_min, self.order_max+1):
|
for n in range(self.order_min, self.order_max+1):
|
||||||
val += self[n] * x**(-n)
|
val += self[n] * x**(-n)
|
||||||
return val
|
return val
|
||||||
|
|
||||||
def invert(self) :
|
|
||||||
self._c << inverse_c (self._c)
|
|
||||||
|
|
||||||
######### arithmetic operations #################
|
|
||||||
|
|
||||||
def __iadd__(self, TailGf arg):
|
|
||||||
self._c << self._c + arg._c
|
|
||||||
return self
|
|
||||||
|
|
||||||
def __add__(self, TailGf y):
|
|
||||||
cdef tail t = (<TailGf?>self)._c + y._c
|
|
||||||
return make_TailGf(t)
|
|
||||||
#c = self.copy()
|
|
||||||
#c += y
|
|
||||||
#return c
|
|
||||||
|
|
||||||
def __isub__(self, TailGf arg):
|
|
||||||
self._c << self._c - arg._c
|
|
||||||
return self
|
|
||||||
|
|
||||||
def __sub__(self,TailGf y):
|
|
||||||
cdef tail t = (<TailGf?>self)._c - y._c
|
|
||||||
return make_TailGf(t)
|
|
||||||
#c = self.copy()
|
|
||||||
#c -= y
|
|
||||||
#return c
|
|
||||||
|
|
||||||
def __imul__(self,arg):
|
|
||||||
n = type(arg).__name__
|
|
||||||
if n == 'TailGf' :
|
|
||||||
self._c << self._c * (<TailGf?>arg)._c
|
|
||||||
elif descriptors.is_scalar(arg):
|
|
||||||
self._c << as_dcomplex(arg)* self._c
|
|
||||||
else :
|
|
||||||
raise RuntimeError, " argument type not recognized in imul for %s"%arg
|
|
||||||
return self
|
|
||||||
|
|
||||||
def __mul__(self,arg):
|
|
||||||
c = self.copy()
|
|
||||||
c *= arg
|
|
||||||
return c
|
|
||||||
|
|
||||||
def __idiv__(self,arg):
|
|
||||||
if descriptors.is_scalar(arg):
|
|
||||||
self._c << self._c / as_dcomplex(arg)
|
|
||||||
else:
|
|
||||||
raise RuntimeError, "rhs must be a scalar"
|
|
||||||
return self
|
|
||||||
|
|
||||||
def __div__(self,arg):
|
|
||||||
c = self.copy()
|
|
||||||
c /= arg
|
|
||||||
return c
|
|
||||||
|
|
||||||
#---- other operations ----
|
#---- other operations ----
|
||||||
|
|
||||||
def zero(self) :
|
|
||||||
"""Sets the expansion to 0"""
|
|
||||||
self._c << as_dcomplex(0.0)
|
|
||||||
|
|
||||||
def transpose (self) :
|
def transpose (self) :
|
||||||
"""Transpose the array : new view as in numpy"""
|
"""Transpose the array : new view as in numpy"""
|
||||||
return TailGf(data=self.data.transpose(), mask=self.mask.transpose())
|
return TailGf(data=self.data.transpose(), mask=self.mask.transpose())
|
||||||
@ -182,21 +60,5 @@ cdef class TailGf:
|
|||||||
"""Transpose the array : new view as in numpy"""
|
"""Transpose the array : new view as in numpy"""
|
||||||
return TailGf(data=self.data.conjugate(), mask=self.mask)
|
return TailGf(data=self.data.conjugate(), mask=self.mask)
|
||||||
|
|
||||||
def __write_hdf5__ (self, gr , char * key) :
|
|
||||||
h5_write (make_h5_group(gr), key, self._c)
|
|
||||||
|
|
||||||
#---------------- Reading from h5 ---------------------------------------
|
|
||||||
|
|
||||||
def h5_read_TailGf( gr, std_string key) :
|
|
||||||
return make_TailGf( h5_extractor[tail]()(make_h5_group(gr),key))
|
|
||||||
|
|
||||||
from pytriqs.archive.hdf_archive_schemes import register_class
|
|
||||||
register_class (TailGf, read_fun = h5_read_TailGf)
|
|
||||||
|
|
||||||
#-----------------------------------------------------
|
|
||||||
# C -> Python
|
|
||||||
#-----------------------------------------------------
|
|
||||||
|
|
||||||
cdef inline make_TailGf ( tail x) :
|
|
||||||
return TailGf(encapsulated_c_object = encapsulate (&x))
|
|
||||||
|
|
||||||
|
|||||||
@ -1,59 +0,0 @@
|
|||||||
cdef extern from "triqs/gfs.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass two_real_times_domain :
|
|
||||||
two_real_times_domain()
|
|
||||||
|
|
||||||
cdef cppclass mesh_two_real_times "triqs::gfs::gf_mesh<triqs::gfs::two_real_times>" :
|
|
||||||
mesh_two_real_times ()
|
|
||||||
mesh_two_real_times (mesh_two_real_times &)
|
|
||||||
two_real_times_domain & domain()
|
|
||||||
double x_min()
|
|
||||||
double x_max()
|
|
||||||
long size()
|
|
||||||
double kind()
|
|
||||||
bint operator ==( mesh_two_real_times &)
|
|
||||||
|
|
||||||
cdef mesh_two_real_times make_mesh_two_real_times "triqs::gfs::gf_mesh<triqs::gfs::two_real_times>" (double t_max, double n_time_slices)
|
|
||||||
|
|
||||||
cdef mesh_retime & get_1d_mesh_from_2times_mesh "triqs::gfs::get_1d_mesh_from_2times_mesh" (mesh_two_real_times &)
|
|
||||||
|
|
||||||
cdef cppclass gf_two_real_times "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::two_real_times>>" :
|
|
||||||
gf_two_real_times()
|
|
||||||
gf_two_real_times(gf_two_real_times &)
|
|
||||||
# The constructor must be no_except, or the cython code won't be correct...
|
|
||||||
gf_two_real_times(mesh_two_real_times, array_view[dcomplex, THREE], nothing, nothing) #except +
|
|
||||||
void operator << (gf_two_real_times &)
|
|
||||||
mesh_two_real_times mesh()
|
|
||||||
array_view[dcomplex, THREE] data()
|
|
||||||
tail singularity()
|
|
||||||
|
|
||||||
cdef gf_retime slice1d "triqs::gfs::slice" (gf_two_real_times &, double t) except +
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/two_real_times.hpp" :
|
|
||||||
cdef void h5_write (h5_group, char *, gf_two_real_times &)
|
|
||||||
|
|
||||||
cdef extern from "triqs/utility/serialization.hpp" :
|
|
||||||
cdef std_string boost_serialize "triqs::serialize" (gf_two_real_times &)
|
|
||||||
cdef void boost_unserialize_into "triqs::deserialize_into_view" (std_string, gf_two_real_times &)
|
|
||||||
|
|
||||||
# Python -> C
|
|
||||||
cdef gf_two_real_times as_gf_two_real_times (g) except +
|
|
||||||
|
|
||||||
# C -> Python
|
|
||||||
cdef make_GfTwoRealTime (gf_two_real_times x, indices_pack=*, name=*)
|
|
||||||
|
|
||||||
############### Blocks of Im Time #########################
|
|
||||||
|
|
||||||
cdef extern from "triqs/gfs/block.hpp" namespace "triqs::gfs" :
|
|
||||||
|
|
||||||
cdef cppclass gf_block_two_real_times "triqs::python_tools::cython_proxy<triqs::gfs::gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::two_real_times>>>" :
|
|
||||||
gf_block_two_real_times()
|
|
||||||
gf_two_real_times & operator [](int)
|
|
||||||
discrete_mesh & mesh()
|
|
||||||
|
|
||||||
cdef gf_block_two_real_times make_gf_block_two_real_times "triqs::gfs::make_block_gf_view_from_vector_of_cython_proxy<triqs::gfs::gf<triqs::gfs::two_real_times>>" ( vector[gf_two_real_times] &)
|
|
||||||
#cdef gf_block_two_real_times make_gf_block_two_real_times "triqs::gfs::make_gf_view<triqs::gfs::block_index,triqs::gfs::gf<triqs::gfs::two_real_times>>" ( vector[gf_two_real_times] &)
|
|
||||||
|
|
||||||
cdef gf_block_two_real_times as_gf_block_two_real_times (G) except +
|
|
||||||
cdef make_BlockGfTwoRealTime (gf_block_two_real_times G, block_indices_pack=*, name=*)
|
|
||||||
|
|
||||||
@ -20,7 +20,7 @@ t = class_( py_type = "TailGf",
|
|||||||
is_printable= True,
|
is_printable= True,
|
||||||
arithmetic = ("algebra","double")
|
arithmetic = ("algebra","double")
|
||||||
)
|
)
|
||||||
|
|
||||||
t.add_constructor(signature = "(int N1, int N2, int size=10, int order_min=-1)",
|
t.add_constructor(signature = "(int N1, int N2, int size=10, int order_min=-1)",
|
||||||
doc = "DOC of constructor")
|
doc = "DOC of constructor")
|
||||||
#t.add_constructor(doc = "DOC of constructor", signature = "(array_view<dcomplex,3> d, array_view<long,2> m, long i)", python_precall = "tail_aux.tail_construct")
|
#t.add_constructor(doc = "DOC of constructor", signature = "(array_view<dcomplex,3> d, array_view<long,2> m, long i)", python_precall = "tail_aux.tail_construct")
|
||||||
@ -68,7 +68,7 @@ t.add_call(calling_pattern = "auto result = self_c.evaluate(u)",
|
|||||||
|
|
||||||
t.number_protocol['multiply'].add_overload(calling_pattern = "*", signature = "tail(matrix<dcomplex> x,tail_view y)") #'x'), (self.c_type,'y')], rtype = self.c_type)
|
t.number_protocol['multiply'].add_overload(calling_pattern = "*", signature = "tail(matrix<dcomplex> x,tail_view y)") #'x'), (self.c_type,'y')], rtype = self.c_type)
|
||||||
t.number_protocol['multiply'].add_overload(calling_pattern = "*", signature = "tail(tail_view x,matrix<dcomplex> y)") #'x'), (self.c_type,'y')], rtype = self.c_type)
|
t.number_protocol['multiply'].add_overload(calling_pattern = "*", signature = "tail(tail_view x,matrix<dcomplex> y)") #'x'), (self.c_type,'y')], rtype = self.c_type)
|
||||||
|
|
||||||
# ok, but MISSING CHECK SIZE
|
# ok, but MISSING CHECK SIZE
|
||||||
t.add_getitem(c_name = "operator()",
|
t.add_getitem(c_name = "operator()",
|
||||||
signature = "matrix_view<dcomplex>(int i)",
|
signature = "matrix_view<dcomplex>(int i)",
|
||||||
@ -122,14 +122,10 @@ def make_mesh( py_type, c_tag, has_kind=True, is_im=False) :
|
|||||||
getter = cfunction(calling_pattern="mesh_kind result = self_c.kind()", signature = "mesh_kind()"),
|
getter = cfunction(calling_pattern="mesh_kind result = self_c.kind()", signature = "mesh_kind()"),
|
||||||
doc = "")
|
doc = "")
|
||||||
|
|
||||||
|
|
||||||
#def __richcmp__(MeshImFreq self, MeshImFreq other,int op) :
|
#def __richcmp__(MeshImFreq self, MeshImFreq other,int op) :
|
||||||
# if op ==2 : # ==
|
# if op ==2 : # ==
|
||||||
# return self._c == other._c
|
# return self._c == other._c
|
||||||
|
|
||||||
#def __reduce__(self):
|
|
||||||
# return self.__class__, (self.beta, self.statistic, len(self))
|
|
||||||
|
|
||||||
return m
|
return m
|
||||||
|
|
||||||
########################
|
########################
|
||||||
@ -290,18 +286,23 @@ def make_gf( py_type, c_tag, is_complex_data = True, is_im = False) :
|
|||||||
g.number_protocol['inplace_lshift'] = pyfunction(py_name ="__inplace_lshift__", python_precall = "pytriqs.gf.local._gf_common._ilshift_", arity = 2)
|
g.number_protocol['inplace_lshift'] = pyfunction(py_name ="__inplace_lshift__", python_precall = "pytriqs.gf.local._gf_common._ilshift_", arity = 2)
|
||||||
|
|
||||||
g.add_method(py_name = "invert", calling_pattern = "invert_in_place(self_c)" , signature = "void()", doc = "Invert (in place)")
|
g.add_method(py_name = "invert", calling_pattern = "invert_in_place(self_c)" , signature = "void()", doc = "Invert (in place)")
|
||||||
|
|
||||||
if c_tag != "imtime" :
|
g.add_method(py_name = "transpose",
|
||||||
|
calling_pattern = "auto result = transpose(self_c)",
|
||||||
|
signature = "gf<%s>()"%c_tag,
|
||||||
|
doc = "Returns a NEW gf, with transposed data...")
|
||||||
|
|
||||||
|
if c_tag != "imtime" :
|
||||||
g.add_method(py_name = "conjugate", calling_pattern = "auto result = conj(self_c)" , signature = "gf<%s>()"%c_tag, doc = "Return a new function, conjugate of self.")
|
g.add_method(py_name = "conjugate", calling_pattern = "auto result = conj(self_c)" , signature = "gf<%s>()"%c_tag, doc = "Return a new function, conjugate of self.")
|
||||||
|
|
||||||
g.number_protocol['multiply'].add_overload(calling_pattern = "*", signature = "gf<%s>(matrix<%s> x,gf<%s> y)"%(c_tag,data_type,c_tag)) #'x'), (self.c_type,'y')], rtype = self.c_type)
|
g.number_protocol['multiply'].add_overload(calling_pattern = "*", signature = "gf<%s>(matrix<%s> x,gf<%s> y)"%(c_tag,data_type,c_tag)) #'x'), (self.c_type,'y')], rtype = self.c_type)
|
||||||
g.number_protocol['multiply'].add_overload(calling_pattern = "*", signature = "gf<%s>(gf<%s> x,matrix<%s> y)"%(c_tag,c_tag,data_type)) #'x'), (self.c_type,'y')], rtype = self.c_type)
|
g.number_protocol['multiply'].add_overload(calling_pattern = "*", signature = "gf<%s>(gf<%s> x,matrix<%s> y)"%(c_tag,c_tag,data_type)) #'x'), (self.c_type,'y')], rtype = self.c_type)
|
||||||
|
|
||||||
g.add_method(py_name = "from_L_G_R",
|
g.add_method(py_name = "from_L_G_R",
|
||||||
calling_pattern = "self_c = L_G_R(l,g,r)",
|
calling_pattern = "self_c = L_G_R(l,g,r)",
|
||||||
signature = "void(matrix<%s> l,gf<%s> g,matrix<%s> r)"%(data_type,c_tag,data_type),
|
signature = "void(matrix<%s> l,gf<%s> g,matrix<%s> r)"%(data_type,c_tag,data_type),
|
||||||
doc = "self <<= l * g * r")
|
doc = "self <<= l * g * r")
|
||||||
|
|
||||||
g.add_method(py_name = "zero",
|
g.add_method(py_name = "zero",
|
||||||
calling_pattern = "self_c = 0",
|
calling_pattern = "self_c = 0",
|
||||||
signature = "void()",
|
signature = "void()",
|
||||||
@ -353,6 +354,9 @@ g.add_method(py_name = "set_from_inverse_fourier",
|
|||||||
calling_pattern = "self_c = inverse_fourier(*gw)",
|
calling_pattern = "self_c = inverse_fourier(*gw)",
|
||||||
doc = """Fills self with the Inverse Fourier transform of gw""")
|
doc = """Fills self with the Inverse Fourier transform of gw""")
|
||||||
|
|
||||||
|
# add the call operator using the interpolation
|
||||||
|
g.add_call(signature = "matrix<double>(double tau)", doc = "G(tau) using interpolation")
|
||||||
|
|
||||||
module.add_class(g)
|
module.add_class(g)
|
||||||
|
|
||||||
########################
|
########################
|
||||||
@ -397,4 +401,4 @@ module.add_function(name = "make_gf_from_inverse_fourier", signature="gf_view<re
|
|||||||
if __name__ == '__main__' :
|
if __name__ == '__main__' :
|
||||||
module.generate_code(mako_template = sys.argv[1], wrap_file = sys.argv[2])
|
module.generate_code(mako_template = sys.argv[1], wrap_file = sys.argv[2])
|
||||||
module.generate_py_converter_header(mako_template = sys.argv[3], wrap_file = sys.argv[4])
|
module.generate_py_converter_header(mako_template = sys.argv[3], wrap_file = sys.argv[4])
|
||||||
|
|
||||||
|
|||||||
@ -1,89 +0,0 @@
|
|||||||
#include "./common.hpp"
|
|
||||||
#include <triqs/arrays/matrix_stack_view.hpp>
|
|
||||||
|
|
||||||
using namespace triqs::arrays;
|
|
||||||
using namespace triqs;
|
|
||||||
|
|
||||||
int main() {
|
|
||||||
|
|
||||||
clef::placeholder<0> i_;
|
|
||||||
clef::placeholder<1> j_;
|
|
||||||
clef::placeholder<2> k_;
|
|
||||||
|
|
||||||
{
|
|
||||||
std::cerr<< "------------- Testing inversion ---------------"<< std::endl;
|
|
||||||
array <double,3> A(4,2,2);
|
|
||||||
|
|
||||||
A(k_,i_,j_) << clef::if_else(i_ == j_, i_ + k_+1, 0); // + 2.3* j_;
|
|
||||||
|
|
||||||
matrix_stack_view<double> S (std::move(A));
|
|
||||||
|
|
||||||
std::cout << S(0) << S(1)<< S(2) << std::endl ;
|
|
||||||
|
|
||||||
S.invert();
|
|
||||||
std::cout << S(0) << S(1)<< S(2)<< std::endl ;
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
std::cerr<< "------------- Testing += matrix ---------------"<< std::endl;
|
|
||||||
array <double,3> A(4,2,3);
|
|
||||||
matrix<double> M(2,3); M()= 1;
|
|
||||||
|
|
||||||
A(k_,i_,j_) << i_ + 2*j_ + k_/2.0;
|
|
||||||
|
|
||||||
matrix_stack_view<double> S (std::move(A));
|
|
||||||
|
|
||||||
std::cout << S(0) << S(1)<< S(2) << std::endl ;
|
|
||||||
|
|
||||||
S += M;
|
|
||||||
std::cout << S(0) << S(1)<< S(2)<< std::endl ;
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
std::cerr<< "------------- Testing += me---------------"<< std::endl;
|
|
||||||
array <double,3> A(4,2,3);
|
|
||||||
matrix<double> M(2,3); M()= 1;
|
|
||||||
|
|
||||||
A(k_,i_,j_) << i_ + 2*j_ + k_/2.0;
|
|
||||||
|
|
||||||
matrix_stack_view<double> S (std::move(A));
|
|
||||||
|
|
||||||
std::cout << S(0) << S(1)<< S(2) << std::endl ;
|
|
||||||
|
|
||||||
S += S;
|
|
||||||
std::cout << S(0) << S(1)<< S(2)<< std::endl ;
|
|
||||||
}
|
|
||||||
{
|
|
||||||
std::cerr<< "------------- Testing *= 2---------------"<< std::endl;
|
|
||||||
array <double,3> A(4,2,3);
|
|
||||||
matrix<double> M(2,3); M()= 1;
|
|
||||||
|
|
||||||
A(k_,i_,j_) << i_ + 2*j_ + k_/2.0;
|
|
||||||
|
|
||||||
matrix_stack_view<double> S (std::move(A));
|
|
||||||
|
|
||||||
std::cout << S(0) << S(1)<< S(2) << std::endl ;
|
|
||||||
|
|
||||||
S *= 2;
|
|
||||||
std::cout << S(0) << S(1)<< S(2)<< std::endl ;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
{
|
|
||||||
std::cerr<< "------------- Testing R, L mul ---------------"<< std::endl;
|
|
||||||
array <double,3> A(4,3,3);
|
|
||||||
matrix<double> L(2,3); L()= 2;
|
|
||||||
matrix<double> R(3,2); R()= 3;
|
|
||||||
|
|
||||||
A(k_,i_,j_) << i_ + 2*j_ + k_/2.0;
|
|
||||||
|
|
||||||
matrix_stack_view<double> S (std::move(A));
|
|
||||||
|
|
||||||
std::cout << S(0) << S(1)<< S(2) << std::endl ;
|
|
||||||
|
|
||||||
auto Sb = matmul_L_R(L,S,R);
|
|
||||||
std::cout << Sb(0) << Sb(1)<< Sb(2)<< std::endl ;
|
|
||||||
std::cout << S(0) << std::endl ;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -95,8 +95,11 @@ namespace triqs { namespace arrays {
|
|||||||
TRIQS_DEFINE_COMPOUND_OPERATORS(array_view);
|
TRIQS_DEFINE_COMPOUND_OPERATORS(array_view);
|
||||||
// to forbid serialization of views...
|
// to forbid serialization of views...
|
||||||
//template<class Archive> void serialize(Archive & ar, const unsigned int version) = delete;
|
//template<class Archive> void serialize(Archive & ar, const unsigned int version) = delete;
|
||||||
|
|
||||||
};
|
template <typename... INT> friend array_view transposed_view(array_view const& a, INT... is) {
|
||||||
|
return array_view{transpose(a.indexmap_, is...), a.storage_};
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
#undef IMPL_TYPE
|
#undef IMPL_TYPE
|
||||||
|
|
||||||
@ -213,6 +216,13 @@ namespace triqs { namespace arrays {
|
|||||||
|
|
||||||
TRIQS_DEFINE_COMPOUND_OPERATORS(array);
|
TRIQS_DEFINE_COMPOUND_OPERATORS(array);
|
||||||
|
|
||||||
|
template <typename... INT> friend const_view_type transposed_view(array const& a, INT... is) {
|
||||||
|
return view_type{transpose(a.indexmap_, is...), a.storage_};
|
||||||
|
};
|
||||||
|
template <typename... INT> friend view_type transposed_view(array& a, INT... is) {
|
||||||
|
return view_type{transpose(a.indexmap_, is...), a.storage_};
|
||||||
|
};
|
||||||
|
|
||||||
};//array class
|
};//array class
|
||||||
|
|
||||||
#undef IMPL_TYPE
|
#undef IMPL_TYPE
|
||||||
|
|||||||
@ -48,10 +48,9 @@ namespace triqs { namespace arrays { namespace indexmaps { namespace cuboid {
|
|||||||
|
|
||||||
domain_t () =default;
|
domain_t () =default;
|
||||||
domain_t (const domain_t & C) = default;
|
domain_t (const domain_t & C) = default;
|
||||||
domain_t (domain_t && C) { *this = std::move(C);}
|
domain_t (domain_t && C) = default;
|
||||||
friend void swap( domain_t & a, domain_t & b) { swap(a.lengths_,b.lengths_);}
|
|
||||||
domain_t & operator =( domain_t const &) = default;
|
domain_t & operator =( domain_t const &) = default;
|
||||||
domain_t & operator =( domain_t && x) { swap(*this,x); return *this;}
|
domain_t & operator =( domain_t && x)= default;
|
||||||
|
|
||||||
domain_t (n_uple lengths):lengths_(std::move(lengths)) {}
|
domain_t (n_uple lengths):lengths_(std::move(lengths)) {}
|
||||||
domain_t (mini_vector<int,Rank> const & lengths):lengths_(lengths) {}
|
domain_t (mini_vector<int,Rank> const & lengths):lengths_(lengths) {}
|
||||||
|
|||||||
@ -60,31 +60,36 @@ namespace triqs { namespace arrays { namespace indexmaps { namespace cuboid {
|
|||||||
typedef domain_t<Rank> domain_type;
|
typedef domain_t<Rank> domain_type;
|
||||||
domain_type const & domain() const { return mydomain;}
|
domain_type const & domain() const { return mydomain;}
|
||||||
|
|
||||||
// basic construction
|
// semi-regular type
|
||||||
map (memory_layout<Rank> const & ml = memory_layout<Rank>(traversal_order)):mydomain(), start_shift_(0), memory_order_(ml) {}
|
map (): start_shift_(0), memory_order_(memory_layout<Rank>(traversal_order)) {}
|
||||||
map(domain_type const & C): mydomain(C), start_shift_(0), memory_order_(traversal_order) {compute_stride_compact();}
|
|
||||||
map(domain_type const & C, memory_layout<Rank> ml): mydomain(C), start_shift_(0), memory_order_(ml) {compute_stride_compact();}
|
|
||||||
|
|
||||||
/// Construction from the length, the stride, start_shift
|
|
||||||
map(lengths_type const & Lengths, strides_type const & strides, std::ptrdiff_t start_shift ):
|
|
||||||
mydomain(Lengths), strides_(strides), start_shift_(start_shift),
|
|
||||||
memory_order_ (memory_layout_from_strides(strides_)) {}
|
|
||||||
|
|
||||||
/// Construction from the length, the stride, start_shift
|
|
||||||
map(lengths_type && Lengths, strides_type && strides, std::ptrdiff_t start_shift ):
|
|
||||||
mydomain(std::move(Lengths)), strides_(std::move(strides)), start_shift_(start_shift),
|
|
||||||
memory_order_ (memory_layout_from_strides(strides_)) {}
|
|
||||||
|
|
||||||
/// Construction from another map with the same order (used in grouping indices)
|
|
||||||
template<ull_t Opt2, ull_t To2> map (map<Rank,Opt2,To2> const & C):
|
|
||||||
mydomain(C.domain()), strides_(C.strides()), start_shift_(C.start_shift()), memory_order_ (C.memory_indices_layout()) {}
|
|
||||||
|
|
||||||
// regular type
|
|
||||||
map (map const & C) = default;
|
map (map const & C) = default;
|
||||||
map (map && C) = default;
|
map (map && C) = default;
|
||||||
map & operator = (map const & m) = default;
|
map & operator = (map const & m) = default;
|
||||||
map & operator = (map && m) = default;
|
map & operator = (map && m) = default;
|
||||||
|
|
||||||
|
// basic construction
|
||||||
|
map(memory_layout<Rank> const & ml):mydomain(), start_shift_(0), memory_order_(ml) {}
|
||||||
|
map(domain_type const & C): mydomain(C), start_shift_(0), memory_order_(traversal_order) {compute_stride_compact();}
|
||||||
|
map(domain_type const & C, memory_layout<Rank> ml): mydomain(C), start_shift_(0), memory_order_(ml) {compute_stride_compact();}
|
||||||
|
|
||||||
|
/// Construction from the length, the stride, start_shift
|
||||||
|
map(lengths_type Lengths, strides_type strides, std::ptrdiff_t start_shift ):
|
||||||
|
mydomain(std::move(Lengths)), strides_(std::move(strides)), start_shift_(start_shift),
|
||||||
|
memory_order_ (memory_layout_from_strides(strides_)) {}
|
||||||
|
|
||||||
|
/// Construction from the length, the stride, start_shift, ml
|
||||||
|
map(lengths_type Lengths, strides_type strides, std::ptrdiff_t start_shift, memory_layout<Rank> const & ml ):
|
||||||
|
mydomain(std::move(Lengths)), strides_(std::move(strides)), start_shift_(start_shift), memory_order_ (ml) {}
|
||||||
|
|
||||||
|
/// Construction from another map with the same order (used in grouping indices)
|
||||||
|
template<ull_t Opt2, ull_t To2> map (map<Rank,Opt2,To2> const & C):
|
||||||
|
mydomain(C.domain()), strides_(C.strides()), start_shift_(C.start_shift()), memory_order_ (C.memory_indices_layout()) {}
|
||||||
|
|
||||||
|
// transposition
|
||||||
|
template <typename... INT> friend map transpose(map const& m, INT... is) {
|
||||||
|
return map{{m.domain().lengths()[is]...}, {m.strides_[is]...}, m.start_shift_, transpose(m.memory_order_, is...)};
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns the shift in position of the element key.
|
/// Returns the shift in position of the element key.
|
||||||
template <typename KeyType>
|
template <typename KeyType>
|
||||||
size_t operator[] (KeyType const & key ) const {
|
size_t operator[] (KeyType const & key ) const {
|
||||||
@ -192,6 +197,9 @@ namespace triqs { namespace arrays { namespace indexmaps { namespace cuboid {
|
|||||||
};
|
};
|
||||||
|
|
||||||
}; //------------- end class ---------------------
|
}; //------------- end class ---------------------
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
}//namespace cuboid
|
}//namespace cuboid
|
||||||
|
|
||||||
template<int R1, int R2, ull_t OptFlags1, ull_t OptFlags2, ull_t To1, ull_t To2>
|
template<int R1, int R2, ull_t OptFlags1, ull_t OptFlags2, ull_t To1, ull_t To2>
|
||||||
|
|||||||
@ -65,8 +65,6 @@ namespace triqs { namespace arrays {
|
|||||||
template< int rank, ull_t fl, ull_t to> struct get_traversal_order { static constexpr ull_t value = _get_traversal_order (rank,fl,to); };
|
template< int rank, ull_t fl, ull_t to> struct get_traversal_order { static constexpr ull_t value = _get_traversal_order (rank,fl,to); };
|
||||||
}}
|
}}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
struct memory_layout_fortran {};
|
struct memory_layout_fortran {};
|
||||||
struct memory_layout_c {};
|
struct memory_layout_c {};
|
||||||
|
|
||||||
@ -90,18 +88,23 @@ namespace triqs { namespace arrays {
|
|||||||
static_assert( sizeof...(in)==Rank-2, "Error");
|
static_assert( sizeof...(in)==Rank-2, "Error");
|
||||||
}
|
}
|
||||||
memory_layout (const memory_layout & C) = default;
|
memory_layout (const memory_layout & C) = default;
|
||||||
memory_layout (memory_layout && C) { *this = std::move(C);}
|
memory_layout (memory_layout && C) = default;
|
||||||
friend void swap( memory_layout & a, memory_layout & b){ std::swap(a.value,b.value);}
|
|
||||||
memory_layout & operator =( memory_layout const &) = default;
|
memory_layout & operator =( memory_layout const &) = default;
|
||||||
memory_layout & operator =( memory_layout && x) { swap(*this,x); return *this;}
|
memory_layout & operator =( memory_layout && x) = default;
|
||||||
|
|
||||||
bool operator ==( memory_layout const & ml) const { return value == ml.value;}
|
bool operator ==( memory_layout const & ml) const { return value == ml.value;}
|
||||||
bool operator !=( memory_layout const & ml) const { return value != ml.value;}
|
bool operator !=( memory_layout const & ml) const { return value != ml.value;}
|
||||||
|
|
||||||
friend std::ostream & operator <<( std::ostream & out, memory_layout const & s) { permutations::print(out,s.value); return out;}
|
friend std::ostream &operator<<(std::ostream &out, memory_layout const &s) {
|
||||||
|
permutations::print(out, s.value);
|
||||||
|
return out;
|
||||||
|
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
//template<int Rank> void swap( memory_layout<Rank> & a, memory_layout<Rank> & b) { std::swap(a.value,b.value);}
|
template <int R, typename... INT> memory_layout<R> transpose(memory_layout<R> ml, INT... is) {
|
||||||
|
static_assert(sizeof...(INT)==R, "!");
|
||||||
|
return memory_layout<R>{permutations::compose(ml.value, permutations::inverse(permutations::permutation(is...)))};
|
||||||
|
}
|
||||||
}}//namespace triqs::arrays
|
}}//namespace triqs::arrays
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@ -1,86 +0,0 @@
|
|||||||
/*******************************************************************************
|
|
||||||
*
|
|
||||||
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
|
||||||
*
|
|
||||||
* Copyright (C) 2013 by O. Parcollet
|
|
||||||
*
|
|
||||||
* TRIQS is free software: you can redistribute it and/or modify it under the
|
|
||||||
* terms of the GNU General Public License as published by the Free Software
|
|
||||||
* Foundation, either version 3 of the License, or (at your option) any later
|
|
||||||
* version.
|
|
||||||
*
|
|
||||||
* TRIQS 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
|
|
||||||
* TRIQS. If not, see <http://www.gnu.org/licenses/>.
|
|
||||||
*
|
|
||||||
******************************************************************************/
|
|
||||||
#ifndef TRIQS_ARRAYS_MATRIX_STACK_VIEW_H
|
|
||||||
#define TRIQS_ARRAYS_MATRIX_STACK_VIEW_H
|
|
||||||
#include "./array.hpp"
|
|
||||||
#include "./matrix.hpp"
|
|
||||||
#include "./matrix_tensor_proxy.hpp"
|
|
||||||
#include <triqs/arrays/linalg/det_and_inverse.hpp>
|
|
||||||
|
|
||||||
namespace triqs { namespace arrays {
|
|
||||||
|
|
||||||
template<typename T> class matrix_stack_view {
|
|
||||||
array_view<T,3> a;
|
|
||||||
public:
|
|
||||||
typedef array_view<T,3> array_view_t;
|
|
||||||
|
|
||||||
matrix_stack_view( typename array_view_t::view_type const & a_):a(a_) {}
|
|
||||||
matrix_stack_view( array_view_t && a_):a(std::move(a_)) {}
|
|
||||||
|
|
||||||
#ifdef TRIQS_WITH_PYTHON_SUPPORT
|
|
||||||
/// Build from a numpy.array (X is a borrowed reference) : throws if X is not a numpy.array
|
|
||||||
explicit matrix_stack_view (PyObject * X):a(typename array_view_t::view_type (X)){}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
//matrix_view_proxy <array_view_t,0> operator()(size_t i) { return matrix_view_proxy <array_view_t,0>(a,i);}
|
|
||||||
//const_matrix_view_proxy<array_view_t,0> operator()(size_t i) const { return const_matrix_view_proxy <array_view_t,0>(a,i);}
|
|
||||||
auto operator()(long i) DECL_AND_RETURN(make_matrix_proxy(this->a, i));
|
|
||||||
auto operator()(long i) const DECL_AND_RETURN(make_const_matrix_proxy(this->a, i));
|
|
||||||
|
|
||||||
matrix_view<T> view(size_t i) const { return a(i,range(),range());}
|
|
||||||
|
|
||||||
size_t size() const { return a.shape(0);}
|
|
||||||
// BE CAREFUL to the shift : it is 1, 2, not 0,1, because of the stack !
|
|
||||||
friend size_t first_dim (matrix_stack_view const & m) { return second_dim(m.a);}
|
|
||||||
friend size_t second_dim (matrix_stack_view const & m) { return third_dim(m.a);}
|
|
||||||
|
|
||||||
matrix_stack_view & operator +=(matrix_stack_view const & arg) { a += arg.a; return *this; }
|
|
||||||
matrix_stack_view & operator -=(matrix_stack_view const & arg) { a -= arg.a; return *this; }
|
|
||||||
|
|
||||||
template<typename RHS>
|
|
||||||
typename std::enable_if<RHS::rank ==2, matrix_stack_view &>::type
|
|
||||||
operator +=(RHS const & arg) { for (size_t i=0; i<size(); ++i) view(i) +=arg; return *this; }
|
|
||||||
|
|
||||||
template<typename RHS>
|
|
||||||
typename std::enable_if<RHS::rank ==2, matrix_stack_view &>::type
|
|
||||||
operator -=(RHS const & arg) { for (size_t i=0; i<size(); ++i) view(i) -=arg; return *this; }
|
|
||||||
|
|
||||||
template<typename RHS> matrix_stack_view & operator *=(RHS const & arg) { a*= arg; return *this; }
|
|
||||||
template<typename RHS> matrix_stack_view & operator /=(RHS const & arg) { a/= arg; return *this; }
|
|
||||||
|
|
||||||
void invert() {for (size_t i=0; i<size(); ++i) { auto v = view(i); v = inverse(v);} }
|
|
||||||
|
|
||||||
friend matrix_stack_view matmul_L_R ( matrix_view<T> const & L, matrix_stack_view const & M, matrix_view<T> const & R) {
|
|
||||||
matrix_stack_view res (typename array_view_t::regular_type (M.size(), first_dim(L), second_dim(R)));
|
|
||||||
for (size_t i=0; i<M.size(); ++i) { res.view(i) = L * M.view(i) * R; }
|
|
||||||
return res;
|
|
||||||
}
|
|
||||||
|
|
||||||
void onsite_matmul_L_R ( matrix_view<T> const & L, matrix_stack_view const & M, matrix_view<T> const & R) {
|
|
||||||
if ((first_dim(*this) != first_dim(L)) || (second_dim(*this) != second_dim(R)) || (second_dim(L) != first_dim(R)))
|
|
||||||
TRIQS_RUNTIME_ERROR << "dimensions do not match!";
|
|
||||||
for (size_t i=0; i<M.size(); ++i) { view(i) = L * M.view(i) * R; }
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
}}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
@ -676,14 +676,13 @@ namespace gfs {
|
|||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
// ---- transpose : a new view
|
// ---- transpose : a new gf
|
||||||
/*
|
|
||||||
template <typename Variable, typename Opt>
|
|
||||||
gf_view<Variable, matrix_valued, Opt> invert_in_place(gf_view<Variable, matrix_valued, Opt> g) {
|
|
||||||
return { g.mesh(), transpose( g.data(), xxxx), transpose(g.singularity()), g.symmetry(), g.indices() XXX, g.name};
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
|
|
||||||
|
template <typename Variable, typename Opt>
|
||||||
|
gf<Variable, matrix_valued, Opt> transpose(gf_view<Variable, matrix_valued, Opt> g) {
|
||||||
|
return {g.mesh(), transposed_view(g.data(), 0, 2, 1), transpose(g.singularity()), g.symmetry(), transpose(g.indices()), g.name};
|
||||||
|
}
|
||||||
|
|
||||||
// ---- conjugate : always a new function -> changelog
|
// ---- conjugate : always a new function -> changelog
|
||||||
|
|
||||||
template <typename Variable, typename Opt> gf<Variable, matrix_valued, Opt> conj(gf_view<Variable, matrix_valued, Opt> g) {
|
template <typename Variable, typename Opt> gf<Variable, matrix_valued, Opt> conj(gf_view<Variable, matrix_valued, Opt> g) {
|
||||||
|
|||||||
@ -281,6 +281,8 @@ namespace triqs { namespace gfs { namespace local {
|
|||||||
|
|
||||||
inline tail conj(tail_view t) { return {conj(t.data()), t.mask_view(),t.order_min()};}
|
inline tail conj(tail_view t) { return {conj(t.data()), t.mask_view(),t.order_min()};}
|
||||||
|
|
||||||
|
inline tail transpose(tail_view t) { return {transposed_view(t.data(),0,2,1), transposed_view(t.mask_view(),1,0),t.order_min()};}
|
||||||
|
|
||||||
/// Slice in orbital space
|
/// Slice in orbital space
|
||||||
//template<bool V> tail_view slice_target(tail_impl<V> const & t, tqa::range R1, tqa::range R2) {
|
//template<bool V> tail_view slice_target(tail_impl<V> const & t, tqa::range R1, tqa::range R2) {
|
||||||
inline tail_view slice_target(tail_view t, tqa::range R1, tqa::range R2) {
|
inline tail_view slice_target(tail_view t, tqa::range R1, tqa::range R2) {
|
||||||
|
|||||||
@ -142,6 +142,16 @@ namespace gfs {
|
|||||||
TRIQS_RUNTIME_ERROR << "Not implemented : slice of string indices";
|
TRIQS_RUNTIME_ERROR << "Not implemented : slice of string indices";
|
||||||
}
|
}
|
||||||
|
|
||||||
|
inline indices_2 transpose(indices_2 const &x) {
|
||||||
|
auto ind2 = x.ind;
|
||||||
|
if (x.ind.size() > 0) {
|
||||||
|
int im = x.ind.size(), jm = x.ind[0].size();
|
||||||
|
for (int i = 0; i < im; ++i)
|
||||||
|
for (int j = 0; j < jm; ++j) ind2[i][j] = x.ind[j][i];
|
||||||
|
}
|
||||||
|
return indices_2(std::move(ind2));
|
||||||
|
}
|
||||||
|
|
||||||
//------------------------------------------------------
|
//------------------------------------------------------
|
||||||
|
|
||||||
// A simple replacement of tail when there is none to maintain generic code simple...
|
// A simple replacement of tail when there is none to maintain generic code simple...
|
||||||
@ -162,6 +172,7 @@ namespace gfs {
|
|||||||
template<typename A> bool check_size(A) {return true;}
|
template<typename A> bool check_size(A) {return true;}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
inline nothing transpose(nothing) { return {};}
|
||||||
template <typename... T> nothing slice_target(nothing, T...) { return nothing(); }
|
template <typename... T> nothing slice_target(nothing, T...) { return nothing(); }
|
||||||
template <typename T> nothing operator+(nothing, T const &) { return nothing(); }
|
template <typename T> nothing operator+(nothing, T const &) { return nothing(); }
|
||||||
template <typename T> nothing operator-(nothing, T const &) { return nothing(); }
|
template <typename T> nothing operator-(nothing, T const &) { return nothing(); }
|
||||||
|
|||||||
@ -1,4 +1,5 @@
|
|||||||
#pragma once
|
|
||||||
|
#pragma once
|
||||||
#include <Python.h>
|
#include <Python.h>
|
||||||
#include "structmember.h"
|
#include "structmember.h"
|
||||||
#include <string>
|
#include <string>
|
||||||
@ -198,11 +199,14 @@ template <> struct py_converter<double> {
|
|||||||
template <> struct py_converter<std::complex<double>> {
|
template <> struct py_converter<std::complex<double>> {
|
||||||
static PyObject *c2py(std::complex<double> x) { return PyComplex_FromDoubles(x.real(), x.imag()); }
|
static PyObject *c2py(std::complex<double> x) { return PyComplex_FromDoubles(x.real(), x.imag()); }
|
||||||
static std::complex<double> py2c(PyObject *ob) {
|
static std::complex<double> py2c(PyObject *ob) {
|
||||||
auto r = PyComplex_AsCComplex(ob);
|
if (PyComplex_Check(ob)) {
|
||||||
return {r.real, r.imag};
|
auto r = PyComplex_AsCComplex(ob);
|
||||||
|
return {r.real, r.imag};
|
||||||
|
}
|
||||||
|
return PyFloat_AsDouble(ob);
|
||||||
}
|
}
|
||||||
static bool is_convertible(PyObject *ob, bool raise_exception) {
|
static bool is_convertible(PyObject *ob, bool raise_exception) {
|
||||||
if (PyComplex_Check(ob)) return true;
|
if (PyComplex_Check(ob) || PyFloat_Check(ob) || PyInt_Check(ob)) return true;
|
||||||
if (raise_exception) { PyErr_SetString(PyExc_TypeError, "Can not convert to complex");}
|
if (raise_exception) { PyErr_SetString(PyExc_TypeError, "Can not convert to complex");}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user