diff --git a/python/tamaas_module.cpp b/python/tamaas_module.cpp index b0aa02b..cbcb66b 100644 --- a/python/tamaas_module.cpp +++ b/python/tamaas_module.cpp @@ -1,45 +1,50 @@ /** * @file * * @author Lucas Frérot * * @section LICENSE * * Copyright (©) 2017 EPFL (Ecole Polytechnique Fédérale de * Lausanne) Laboratory (LSMS - Laboratoire de Simulation en Mécanique des * Solides) * * Tamaas is free software: you can redistribute it and/or modify it under the * terms of the GNU Lesser General Public License as published by the Free * Software Foundation, either version 3 of the License, or (at your option) any * later version. * * Tamaas 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 Lesser General Public License for more * details. * * You should have received a copy of the GNU Lesser General Public License * along with Tamaas. If not, see . * */ /* -------------------------------------------------------------------------- */ #include "tamaas.hh" #include "flood_fill.hh" #include "wrap.hh" /* -------------------------------------------------------------------------- */ #include /* -------------------------------------------------------------------------- */ __BEGIN_TAMAAS__ namespace py = pybind11; PYBIND11_MODULE(tamaas, mod) { mod.doc() = "tamaas module for python"; + // Wrapping the base methods + mod.def("initialize", &initialize, py::arg("num_threads") = 0, + "Initialize tamaas with desired number of threads"); + mod.def("finalize", &finalize, "Final cleanup"); + // Wrapping tamaas components wrap::wrapPercolation(mod); } __END_TAMAAS__