1 files changed, 31 insertions, 3 deletions

diff --git a/utils/wbcap.cpp b/utils/wbcap.cpp
index 38a8cc0..5a94a70 100644
--- a/utils/wbcap.cpp
+++ b/utils/wbcap.cpp
@@ -2,6 +2,7 @@
 #include <poll.h>
 #include <unistd.h>
 #include <algorithm>
+#include <fstream>
 
 #include <kms++/kms++.h>
 #include <kms++util/kms++util.h>
@@ -56,7 +57,7 @@ public:
                 m_capdev.stream_off();
         }
 
-        void Dequeue()
+        DumbFramebuffer* Dequeue()
         {
                 auto fb = m_capdev.dequeue();
 
@@ -64,6 +65,8 @@ public:
                 s_wb_fbs.erase(iter);
 
                 s_ready_fbs.insert(s_ready_fbs.begin(), fb);
+
+                return fb;
         }
 
         void Queue()
@@ -248,7 +251,10 @@ static const char* usage_str =
                 "Options:\n"
                 "  -s, --src=CONN            Source connector\n"
                 "  -d, --dst=CONN            Destination connector\n"
-                "  -f, --format=4CC          Format"
+                "  -m, --smode=MODE          Source connector videomode\n"
+                "  -M, --dmode=MODE          Destination connector videomode\n"
+                "  -f, --format=4CC          Format\n"
+                "  -w, --write               Write captured frames to wbcap.raw file\n"
                 "  -h, --help                Print this help\n"
                 ;
 
@@ -259,6 +265,7 @@ int main(int argc, char** argv)
         string dst_conn_name;
         string dst_mode_name;
         PixelFormat pixfmt = PixelFormat::XRGB8888;
+        bool write_file = false;
 
         OptionSet optionset = {
                 Option("s|src=", [&](string s)
@@ -281,6 +288,10 @@ int main(int argc, char** argv)
                 {
                         pixfmt = FourCCToPixelFormat(s);
                 }),
+                Option("w|write", [&]()
+                {
+                        write_file = true;
+                }),
                 Option("h|help", [&]()
                 {
                         puts(usage_str);
@@ -366,6 +377,13 @@ int main(int argc, char** argv)
         fds[2].fd = card.fd();
         fds[2].events =  POLLIN;
 
+        uint32_t dst_frame_num = 0;
+
+        const string filename = "wbcap.raw";
+        unique_ptr<ofstream> os;
+        if (write_file)
+                os = unique_ptr<ofstream>(new ofstream(filename, ofstream::binary));
+
         while (true) {
                 int r = poll(fds.data(), fds.size(), -1);
                 ASSERT(r > 0);
@@ -376,7 +394,17 @@ int main(int argc, char** argv)
                 if (fds[1].revents) {
                         fds[1].revents = 0;
 
-                        wb.Dequeue();
+                        DumbFramebuffer* fb = wb.Dequeue();
+
+                        if (write_file) {
+                                printf("Writing frame %u to %s\n", dst_frame_num, filename.c_str());
+
+                                for (unsigned i = 0; i < fb->num_planes(); ++i)
+                                        os->write((char*)fb->map(i), fb->size(i));
+
+                                dst_frame_num++;
+                        }
+
                         wbflipper.queue_next();
                 }