set(PythonInterp_FIND_VERSION_MAJOR 3)
include(FindPythonInterp)
if(NOT PYTHONINTERP_FOUND)
- message(FATAL_ERROR "Please install Python (version 3 or higher).")
+ message(FATAL_ERROR "Please install Python (version 3 or higher) to compile SimGrid.")
endif()
SET(LIBRARY_OUTPUT_PATH ${CMAKE_BINARY_DIR}/lib)
include(${CMAKE_HOME_DIRECTORY}/tools/cmake/Java.cmake)
endif()
+# Python binding, generated with pybind11
+set(PYBIND11_CPP_STANDARD -std=c++11)
+find_package(pybind11)
+option(enable_python "Whether the Python bindings are activated." ${pybind11_FOUND}) # ON by default if dependencies are met
+
+if(enable_python)
+ if(pybind11_FOUND)
+ pybind11_add_module(simgrid_python src/bindings/python/simgrid_python.cpp)
+ set_source_files_properties(src/bindings/python/simgrid_python.cpp PROPERTIES COMPILE_FLAGS -Wno-attributes)
+ set_source_files_properties(src/bindings/python/simgrid_python.cpp PROPERTIES COMPILE_FLAGS -std=gnu++14)
+ target_link_libraries(simgrid_python PUBLIC simgrid)
+ set_target_properties(simgrid_python PROPERTIES LIBRARY_OUTPUT_NAME simgrid)
+ else()
+ message(FATAL_ERROR "Please install pybind11-dev to build the Python bindings (or disable that option).")
+ endif()
+endif()
+
### Make tests
if(enable_memcheck_xml)
set(enable_memcheck true)
else()
message(" Compile Java ................: NO")
endif()
+if(pybind11_FOUND)
+ message(" Compile Python bindings .....: ${enable_python}")
+ message(" module ....................: ${PYTHON_MODULE_PREFIX}simgrid${PYTHON_MODULE_EXTENSION}")
+else()
+ message(" Compile Python bindings .....: NO (disabled, or pybind11 not found)")
+endif()
message(" Compile Lua .................: ${SIMGRID_HAVE_LUA}")
message(" Compile Smpi ................: ${HAVE_SMPI}")
message(" Smpi fortran ..............: ${SMPI_FORTRAN}")
--- /dev/null
+#include <functional>
+#include <memory>
+#include <string>
+#include <vector>
+
+#include <pybind11/functional.h>
+#include <pybind11/pybind11.h>
+#include <pybind11/stl.h>
+
+#include <simgrid/config.h>
+#include <xbt/log.h>
+#include <xbt/string.hpp>
+
+#include <simgrid/s4u/Actor.hpp>
+#include <simgrid/s4u/Engine.hpp>
+#include <simgrid/s4u/Host.hpp>
+
+#include <boost/intrusive_ptr.hpp>
+
+namespace py = pybind11;
+using simgrid::s4u::Actor;
+using simgrid::s4u::ActorPtr;
+using simgrid::s4u::Engine;
+using simgrid::s4u::Host;
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_python, "S4U python");
+
+PYBIND11_DECLARE_HOLDER_TYPE(T, boost::intrusive_ptr<T>);
+
+namespace {
+
+static std::string get_simgrid_version()
+{
+ int major, minor, patch;
+ sg_version_get(&major, &minor, &patch);
+ return simgrid::xbt::string_printf("%i.%i.%i", major, minor, patch);
+}
+
+static std::string simgrid_version = get_simgrid_version();
+
+} // namespace
+
+PYBIND11_MODULE(simgrid, m)
+{
+
+ m.doc() = "SimGrid userspace API";
+
+ m.attr("simgrid_version") = simgrid_version;
+
+ m.def("info", [](char* s) { XBT_INFO("%s", s); }, "Display a logging message of default priority.");
+
+ m.def("execute", py::overload_cast<double>(&simgrid::s4u::this_actor::execute),
+ "Block the actor, computing the given amount of flops");
+ m.def("yield_", &simgrid::s4u::this_actor::yield);
+
+ py::class_<Engine>(m, "Engine")
+ .def(py::init([](std::vector<std::string> args) -> simgrid::s4u::Engine* {
+ static char noarg[] = {'\0'};
+ int argc = args.size();
+ std::unique_ptr<char* []> argv(new char*[argc + 1]);
+ for (int i = 0; i != argc; ++i)
+ argv[i] = args[i].empty() ? noarg : &args[i].front();
+ argv[argc] = nullptr;
+ // Currently this can be dangling, we should wrap this somehow.
+ return new simgrid::s4u::Engine(&argc, argv.get());
+ }))
+ .def("load_platform", &Engine::load_platform)
+ .def("load_deployment", &Engine::load_deployment)
+ .def("run", &Engine::run)
+ .def("register_function", [](Engine*, const char* name, std::function<void(std::vector<std::string>)> f) {
+ simgrid::simix::register_function(
+ name, [f](std::vector<std::string> args) -> simgrid::simix::ActorCode { return [args, f]() { f(args); }; });
+ });
+
+ // Currently, Host lead to segfault:
+ py::class_<simgrid::s4u::Host, std::unique_ptr<Host, py::nodelete>>(m, "Host").def("by_name", &Host::by_name);
+
+ simgrid::s4u::ActorPtr (*create_actor)(std::string, Host*, std::function<void()>) = &Actor::create;
+
+ py::class_<simgrid::s4u::Actor, ActorPtr>(m, "Actor");
+
+ m.def("create_actor", create_actor);
+
+ m.def("create_actor", [](std::string name, Host* host) -> std::function<ActorPtr(std::function<void()>)> {
+ return [name, host](std::function<void()> f) -> ActorPtr {
+ return simgrid::s4u::Actor::create(name.c_str(), host, std::move(f));
+ };
+ });
+}
