Futile compilation attempt working. see new file.

Author: Iainmon

bridge/include/bridge.h

@@ -107,6 +107,8 @@ proto_bridge_simple(tanhshrink);
 void split_loop(int64_t idx, int64_t n);
 void split_loop_filler(int64_t n,int64_t* ret);
 
+void show_webcam(void);
+
 
 // bridge_tensor_t conv2d(
 //     bridge_tensor_t input,

bridge/include/bridge.h.pch

@@ -16,6 +16,9 @@
 #include <chrono>
 #include <thread>
 
+#include <opencv2/opencv.hpp>
+
+
 #define def_bridge_simple(Name) \
     extern "C" bridge_tensor_t Name(bridge_tensor_t input) { \
         auto t_input = bridge_to_torch(input); \
@@ -397,4 +400,41 @@ extern "C" void split_loop_filler(int64_t n,int64_t* ret) {
         *ret = i;
         std::this_thread::sleep_for(std::chrono::seconds(0));
     }
+}
+
+
+
+cv::VideoCapture open_camera(int cam_index) {
+    cv::VideoCapture cap(cam_index, cv::CAP_AVFOUNDATION);
+    if (!cap.isOpened()) {
+        std::cerr << "Could not open camera index " << cam_index << std::endl;
+        return cv::VideoCapture();
+    }
+    cap.set(cv::CAP_PROP_BUFFERSIZE, 1); // minimal internal buffering
+    cap.set(cv::CAP_PROP_FPS, 60);       // request higher FPS if possible
+    return cap;
+}
+
+
+extern "C" void show_webcam(void) {
+    cv::VideoCapture cap;
+    cap = open_camera(0);
+
+    cv::Mat frame_bgr;
+
+    while (true) {
+        if (!cap.read(frame_bgr) || frame_bgr.empty()) {
+            std::cerr << "[WARN] Empty frame, exiting" << std::endl;
+            break;
+        }
+
+        cv::imshow("webcam", frame_bgr);
+
+        if (cv::waitKey(1) == 27) { // ESC key
+            break;
+        }
+    }
+
+    cap.release();
+    cv::destroyAllWindows();
 }

bridge/lib/bridge.cpp

@@ -0,0 +1 @@
+/usr/bin/clang++ -std=c++20 style_transfer.cpp -o styletransfer -I ../../../libtorch/include -I ../../../libtorch/include/torch/csrc/api/include -I ../include $(pkg-config --cflags --libs opencv4) -L ../../../libtorch/lib -ltorch -ltorch_cpu -lc10 -ltorch_global_deps

demos/video/style-transfer/WORKING.sh

@@ -0,0 +1,26 @@
+#include <opencv2/opencv.hpp>
+#include <iostream>
+
+int main() {
+    cv::VideoCapture cap(0); // Open the default camera (0)
+    if (!cap.isOpened()) {
+        std::cerr << "Error: Could not open camera." << std::endl;
+        return -1;
+    }
+
+    cv::Mat frame;
+    while (true) {
+        cap >> frame; // Capture a new frame
+        if (frame.empty()) {
+            std::cerr << "Error: Could not capture frame." << std::endl;
+            break;
+        }
+
+        cv::imshow("Webcam", frame); // Display the captured frame
+        if (cv::waitKey(30) >= 0) break; // Exit on any key press
+    }
+
+    cap.release(); // Release the camera
+    cv::destroyAllWindows(); // Close all OpenCV windows
+    return 0;
+}

demos/video/style-transfer/mirror.cpp

@@ -1,14 +1,86 @@
+find_package(OpenCV 4 REQUIRED)
+
+find_library(ACCELERATE Accelerate REQUIRED)
+find_library(METAL Metal REQUIRED)
+find_library(FOUNDATION Foundation REQUIRED)
+
+
+
+add_library(bridge_cv OBJECT ${BRIDGE_DIR}/include/bridge.h ${BRIDGE_DIR}/lib/bridge.cpp)
+
+target_link_directories(bridge_cv PRIVATE ${LIBTORCH_DIR}/lib)
+
+target_link_libraries(
+    bridge_cv
+    PRIVATE
+    -ltorch
+    -ltorch_cpu
+    -lc10
+    -ltorch_global_deps
+    ${OpenCV_LIBS}
+    # ${TORCH_LIBRARIES}
+    ${ACCELERATE}
+    ${METAL}
+    ${FOUNDATION}
+)
+
+target_include_directories(
+    bridge_cv
+    PRIVATE
+    ${BRIDGE_DIR}/include
+    ${LIBTORCH_DIR}/include
+    ${LIBTORCH_DIR}/include/torch/csrc/api/include
+    # ${BRIDGE_DIR}/util
+)
+
+# if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
+#   target_compile_options(bridge_cv PRIVATE -Ofast -flto -ffast-math)
+#   target_link_options(bridge_cv PRIVATE -flto)
+# endif()
+
+
+set(BRIDGE_CV_OBJECT_FILES $<TARGET_OBJECTS:bridge_cv>)
+
+
+set(CHAI_CV_LINKER_ARGS
+    -M ${PROJECT_ROOT_DIR}/lib
+    ${BRIDGE_DIR}/include/bridge.h
+    ${BRIDGE_CV_OBJECT_FILES}
+    -L ${LIBTORCH_DIR}/lib
+    ${LIBTORCH_LIBS_LINKER_ARGS}
+    --ldflags "-Wl,-rpath,${LIBTORCH_DIR}/lib"
+)
+
+
+
+
+
+
+
+
 add_executable(SplitLoop 
     ${CMAKE_CURRENT_SOURCE_DIR}/split_loop.chpl
     ${CHAI_LIB_FILES}
 )
-add_dependencies(SplitLoop bridge)
-add_dependencies(SplitLoop ChAI)
+
+add_dependencies(SplitLoop bridge_cv)
+# add_dependencies(SplitLoop ChAI)
 target_link_options(SplitLoop
     PRIVATE
-    ${CHAI_LINKER_ARGS}
+    ${CHAI_CV_LINKER_ARGS}
 )
 
+cmake_print_variables(CHAI_CV_LINKER_ARGS)
+cmake_print_variables(OpenCV_LIBS)
+cmake_print_variables(ACCELERATE)
+cmake_print_variables(METAL)
+cmake_print_variables(FOUNDATION)
+
 set_target_properties(SplitLoop PROPERTIES
     RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}
-)
+)
+
+# if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
+#   target_compile_options(SplitLoop PRIVATE -Ofast -flto -ffast-math)
+#   target_link_options(SplitLoop PRIVATE -flto)
+# endif()

examples/split_loop/CMakeLists.txt

@@ -21,7 +21,7 @@ proc main(args: [] string) {
     }
 
 
-
+    Bridge.showWebcam();
 
     writeln("Done!");
 }

examples/split_loop/split_loop.chpl

@@ -91,6 +91,8 @@ module Bridge {
 
     extern "split_loop_filler" proc splitLoopFiller(n: int(64),ret: c_ptr(int(64))): void;
 
+    extern "show_webcam" proc showWebcam(): void;
+
     // extern "capture_webcam_bridge" proc captureWebcam(
     //     in cam_index: int(32)): bridge_tensor_t;