From 0a41fbc3766914c1e681bbca8896fe61d9bbcb5e Mon Sep 17 00:00:00 2001 From: xiong-jie-y Date: Sat, 20 Jun 2020 23:41:49 +0900 Subject: [PATCH] Added python so generation rule. --- WORKSPACE | 12 ++++++++++++ mediapipe/examples/desktop/BUILD | 14 ++++++++++++++ mediapipe/examples/desktop/cameravtuber.cc | 10 ++++++++++ 3 files changed, 36 insertions(+) create mode 100644 mediapipe/examples/desktop/cameravtuber.cc diff --git a/WORKSPACE b/WORKSPACE index 1901344ba..522e24092 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -299,3 +299,15 @@ maven_install( fetch_sources = True, version_conflict_policy = "pinned", ) + +new_local_repository( + name = "python_dev", + path = "/home/yusuke/miniconda3/envs/py37_ubuntu_vtuber/include/python3.7m", + build_file_content = """ +package(default_visibility = ["//visibility:public"]) +cc_library( + name = "headers", + hdrs = glob(["**/*.h"]) +) +""" +) \ No newline at end of file diff --git a/mediapipe/examples/desktop/BUILD b/mediapipe/examples/desktop/BUILD index 768cbdbab..8612979d5 100644 --- a/mediapipe/examples/desktop/BUILD +++ b/mediapipe/examples/desktop/BUILD @@ -76,3 +76,17 @@ cc_library( "//mediapipe/calculators/core:packet_cloner_calculator" ], ) + +cc_binary( + name = "cameravtuber.so", + linkshared = 1, + srcs = [ + "cameravtuber.cc" + ], + deps = [ + ":demo_run_graph_main_gpu", + "@python_dev//:headers" + ], + copts = ["-Iexternal/python_dev", "-DPIC", "-shared", "-fPIC"], + linkopts = ["-lboost_python3"] +) \ No newline at end of file diff --git a/mediapipe/examples/desktop/cameravtuber.cc b/mediapipe/examples/desktop/cameravtuber.cc new file mode 100644 index 000000000..42f94478c --- /dev/null +++ b/mediapipe/examples/desktop/cameravtuber.cc @@ -0,0 +1,10 @@ +#include + +std::string hello() { + return "hello world"; +} + +BOOST_PYTHON_MODULE(cameravtuber) { + using namespace boost::python; + def("hello", &hello); +} \ No newline at end of file