mirror of
https://github.com/triqs/dft_tools
synced 2024-12-26 22:33:48 +01:00
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
|
from wrap_generator import *
|
||
|
|
|
||
|
|
module = module_(full_name = "pytriqs.random_generator", doc = "")
|
||
|
|
module.add_include("<triqs/mc_tools/random_generator.hpp>")
|
||
|
|
module.add_using("namespace triqs::mc_tools")
|
||
|
|
# Not needed. Reorganize the hpp wrapper tool
|
||
|
|
module.add_include("<triqs/h5.hpp>")
|
||
|
|
module.add_include("<triqs/arrays.hpp>")
|
||
|
|
|
||
|
|
# --------- RandomGenerator ----------------------------------
|
||
|
|
|
||
|
|
r = class_(py_type = "RandomGenerator",
|
||
|
|
c_type = "random_generator",
|
||
|
|
c_type_absolute = "triqs::mc_tools::random_generator",
|
||
|
|
)
|
||
|
|
|
||
|
|
r.add_constructor(signature = "(std::string name, int seed)",
|
||
|
|
doc =
|
||
|
|
"""
|
||
|
|
This is a random number generator class based on boost.
|
||
|
|
|
||
|
|
name Name of the random number generator
|
||
|
|
seed Random number seed
|
||
|
|
""")
|
||
|
|
|
||
|
|
r.add_call(signature = "int(int N)", doc = """Generate an integer random number in [0,N-1]""")
|
||
|
|
r.add_call(signature = "double()", doc = """Generate a float random number in [0,1[""")
|
||
|
|
|
||
|
|
module.add_class(r)
|
||
|
|
|
||
|
|
# --------- Module functions ----------------------------------
|
||
|
|
|
||
|
|
module.add_function(name = "random_generator_names_list",
|
||
|
|
signature = "std::vector<std::string>()",
|
||
|
|
doc = """Get a list of available random generator names"""
|
||
|
|
)
|
||
|
|
|
||
|
|
########################
|
||
|
|
## Code generation
|
||
|
|
########################
|
||
|
|
|
||
|
|
if __name__ == '__main__' :
|
||
|
|
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])
|
||
|
|
|
||
|
|
|