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