mirror of
https://github.com/triqs/dft_tools
synced 2024-12-26 06:14:14 +01:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
from wrap_generator import *
|
|
|
|
# The operator class
|
|
op = class_(
|
|
py_type = "Operator",
|
|
c_type = "many_body_operator<double>",
|
|
c_type_absolute = "triqs::utility::many_body_operator<double>",
|
|
is_printable= True,
|
|
arithmetic = ("algebra","with_unit","with_unary_minus","double")
|
|
)
|
|
|
|
op.add_constructor(signature="()", doc="create zero operator")
|
|
|
|
# The many_body_operators module
|
|
module = module_(full_name = "pytriqs.operators.operators2", doc = "Doc to be written")
|
|
module.add_include("<triqs/operators/many_body_operator.hpp>")
|
|
module.add_include("<triqs/arrays.hpp>")
|
|
module.add_using("namespace triqs::utility")
|
|
module.add_class(op)
|
|
|
|
# Add various overload of c, c_dag to the module Annihilation & Creation operators
|
|
for name, doc in [("c","annihilation operator"), ("c_dag","creation operator"), ("n","number operator")] :
|
|
for sign in [
|
|
"",
|
|
"std::string ind1",
|
|
"std::string ind1, std::string ind2",
|
|
"int i, std::string ind1",
|
|
"std::string ind1, int i",
|
|
"int i, int j"
|
|
]:
|
|
module.add_function(name = name, signature="many_body_operator<double>(%s)"%sign, doc=doc)
|
|
|
|
module.generate_code()
|
|
|