X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=main.cpp;h=17431aa46f127bb06fb33cd0dc9ec705bdcfc430;hb=0a58b6906096d32f8d588ee18e898d487c06072b;hp=810bcf3d9d27974db482764d462d486a1154f060;hpb=b3e4f4b8458a052a7a8e252404f116a7b3c92864;p=nageru

diff --git a/main.cpp b/main.cpp
index 810bcf3..17431aa 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,19 +1,56 @@
+#include <assert.h>
 #include <stdio.h>
+#include <stdint.h>
+
+#include <chrono>
 #include <memory>
+#include <mutex>
+#include <string>
+#include <thread>
+#include <vector>
 
 extern "C" {
 #include <libavformat/avformat.h>
 }
 
+#include <QApplication>
+
+#include "mainwindow.h"
 #include "ffmpeg_raii.h"
 
+#define MAX_STREAMS 16
+
 using namespace std;
+using namespace std::chrono;
+
+string filename_for_frame(unsigned stream_idx, int64_t pts)
+{
+	char filename[256];
+	snprintf(filename, sizeof(filename), "frames/cam%d-pts%09ld.jpeg", stream_idx, pts);
+	return filename;
+}
+
+mutex frame_mu;
+vector<int64_t> frames[MAX_STREAMS];
+
+int thread_func();
 
-int main(void)
+int main(int argc, char **argv)
 {
 	av_register_all();
 	avformat_network_init();
 
+	QApplication app(argc, argv);
+	MainWindow mainWindow;
+	mainWindow.show();
+
+	thread(thread_func).detach();
+
+	return app.exec();
+}
+
+int thread_func()
+{
 	auto format_ctx = avformat_open_input_unique("example.mp4", nullptr, nullptr);
 	if (format_ctx == nullptr) {
 		fprintf(stderr, "%s: Error opening file\n", "example.mp4");
@@ -32,15 +69,19 @@ int main(void)
 		}
 		fprintf(stderr, "Got a frame from camera %d, pts = %ld, size = %d\n",
 			pkt.stream_index, pkt.pts, pkt.size);
-		char filename[256];
-		snprintf(filename, sizeof(filename), "frames/cam%d-pts%09ld.jpeg", pkt.stream_index, pkt.pts);
-		FILE *fp = fopen(filename, "wb");
+		string filename = filename_for_frame(pkt.stream_index, pkt.pts);
+		FILE *fp = fopen(filename.c_str(), "wb");
 		if (fp == nullptr) {
-			perror(filename);
+			perror(filename.c_str());
 			exit(1);
 		}
 		fwrite(pkt.data, pkt.size, 1, fp);
 		fclose(fp);
+
+		assert(pkt.stream_index < MAX_STREAMS);
+		frames[pkt.stream_index].push_back(pkt.pts);
+
+		this_thread::sleep_for(milliseconds(1000) / 120);
 	}
 
 	return 0;