working simple streamer

examples/numpy_write_example.cpp

@@ -1,6 +1,6 @@
 // Your First C++ Program
-#include "aare/file_io/File.hpp"
 #include "aare/core/Frame.hpp"
+#include "aare/file_io/File.hpp"
 #include <iostream>
 
 #define AARE_ROOT_DIR_VAR "PROJECT_ROOT_DIR"

examples/restream_example.cpp

@@ -1,26 +1,88 @@
-#include <iostream>
-#include <string>
-#include <filesystem>
-#include <vector>
-#include <memory>
 #include <boost/program_options.hpp>
+#include <chrono>
+#include <thread>
 
+#include "aare/file_io/File.hpp"
 #include "aare/network_io/ZmqSocketSender.hpp"
 
 #include <boost/program_options/options_description.hpp>
 
-
 using namespace aare;
 using namespace std;
 namespace po = boost::program_options;
 
 int main(int argc, char **argv) {
+    aare::logger::set_verbosity(aare::logger::DEBUG);
+
     po::options_description desc("Allowed options");
-    desc.add_options()
-    ("help", "produce help message")
-    // ("input,i", po::value<string>(), "input file");
-    ("port,p", po::value<int>(), "port number");
-    // ("loop,l", "loop over the file");
-    // po::positional_options_description pd;
-    // pd.add("input,i", 1);
+    desc.add_options()("help", "produce help message")("file,f", po::value<string>(), "input file")(
+        "port,p", po::value<uint16_t>(), "port number")("fps", po::value<uint16_t>()->default_value(1),
+                                                        "frames per second (default 1)")("loop,l",
+                                                                                         "loop over the file");
+    po::positional_options_description pd;
+    pd.add("file", -1);
+
+    po::variables_map vm;
+    try {
+        auto parsed = po::command_line_parser(argc, argv).options(desc).positional(pd).run();
+        po::store(parsed, vm);
+        po::notify(vm);
+
+    } catch (const boost::program_options::error &e) {
+        cout << e.what() << "\n";
+        cout << desc << "\n";
+        return 1;
+    }
+
+    if (vm.count("help")) {
+        cout << desc << "\n";
+        return 1;
+    }
+    if (vm.count("file") != 1) {
+        aare::logger::error("file is required");
+        cout << desc << "\n";
+        return 1;
+    }
+    if (vm.count("port") != 1) {
+        aare::logger::error("file is required");
+        cout << desc << "\n";
+        return 1;
+    }
+
+    std::string path = vm["file"].as<string>();
+    uint16_t port = vm["port"].as<uint16_t>();
+    bool loop = vm.count("loop") == 1 ? true : false;
+    uint16_t fps = vm["fps"].as<uint16_t>();
+
+    aare::logger::debug("ARGS: file:", path, "port:", port, "fps:", fps, "loop:", loop);
+    auto d = round<std::chrono::milliseconds>(std::chrono::duration<double>{1. / fps});
+    aare::logger::debug("sleeping for", d.count(), "ms");
+
+    if (!std::filesystem::exists(path)) {
+        aare::logger::error("file does not exist");
+        return 1;
+    }
+
+    std::filesystem::path tmp(path);
+
+    File file(tmp, "r");
+    string endpoint = "tcp://*:" + std::to_string(port);
+    ZmqSocketSender sender(endpoint);
+    sender.bind();
+    std::this_thread::sleep_for(d); // slow joiner problem should fix this
+
+    for (size_t frameidx = 0; frameidx < file.total_frames(); frameidx++) {
+
+        Frame frame = file.read();
+        ZmqHeader header;
+        header.frameNumber = frameidx;
+        header.data = true;
+        header.npixelsx = frame.rows();
+        header.npixelsy = frame.cols();
+        header.dynamicRange = frame.bitdepth();
+        header.imageSize = frame.size();
+
+        sender.send({header, frame});
+        std::this_thread::sleep_for(d);
+    }
 }

examples/zmq_receiver_example.cpp

@@ -7,7 +7,7 @@
 using namespace aare;
 
 int main() {
-    // aare::logger::set_verbosity(aare::logger::DEBUG);
+    aare::logger::set_verbosity(aare::logger::DEBUG);
     std::string endpoint = "tcp://localhost:5555";
     aare::ZmqSocketReceiver socket(endpoint);
     socket.connect();
@@ -19,15 +19,15 @@ int main() {
         aare::logger::info("Frame size:", v[0].frame.size());
         aare::logger::info("Header:", v[0].header.to_string());
 
-        for (ZmqFrame zmq_frame : v) {
-            auto &[header, frame] = zmq_frame;
-            for (int i = 0; i < 1024; i++) {
-                for (int j = 0; j < 1024; j++) {
-                    assert(*(uint32_t *)frame.get(i, j) == (uint32_t)i + j);
-                }
-            }
-            aare::logger::info("Frame verified");
-        }
+        // for (ZmqFrame zmq_frame : v) {
+        //     auto &[header, frame] = zmq_frame;
+        //     for (int i = 0; i < 1024; i++) {
+        //         for (int j = 0; j < 1024; j++) {
+        //             assert(*(uint32_t *)frame.get(i, j) == (uint32_t)i + j);
+        //         }
+        //     }
+        //     aare::logger::info("Frame verified");
+        // }
     }
     return 0;
 }