X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=main.cpp;h=dcb6f10674e280a70d8e19b394b03f1503538e2b;hb=e6dd4055a226d31ecb29bc0746bb896e6ff7ff66;hp=0a976c3d438994e6c95dce9faabc4556cf766715;hpb=2f390da4e0c47247a9080df7f3761b2e7ad011c5;p=nageru diff --git a/main.cpp b/main.cpp index 0a976c3..dcb6f10 100644 --- a/main.cpp +++ b/main.cpp @@ -3,6 +3,7 @@ #include #include +#include #include #include #include @@ -15,16 +16,29 @@ extern "C" { #include +#include +#include + +#include "clip_list.h" +#include "context.h" +#include "defs.h" #include "mainwindow.h" #include "ffmpeg_raii.h" +#include "httpd.h" +#include "player.h" #include "post_to_main_thread.h" +#include "ref_counted_gl_sync.h" +#include "timebase.h" #include "ui_mainwindow.h" -#define MAX_STREAMS 16 - using namespace std; using namespace std::chrono; +std::mutex RefCountedGLsync::fence_lock; + +// TODO: Replace by some sort of GUI control, I guess. +int64_t current_pts = 0; + string filename_for_frame(unsigned stream_idx, int64_t pts) { char filename[256]; @@ -34,31 +48,68 @@ string filename_for_frame(unsigned stream_idx, int64_t pts) mutex frame_mu; vector frames[MAX_STREAMS]; +HTTPD *global_httpd; -int thread_func(); +int record_thread_func(); int main(int argc, char **argv) { - av_register_all(); avformat_network_init(); + global_httpd = new HTTPD; + global_httpd->start(DEFAULT_HTTPD_PORT); + + QCoreApplication::setAttribute(Qt::AA_ShareOpenGLContexts, true); + + QSurfaceFormat fmt; + fmt.setDepthBufferSize(0); + fmt.setStencilBufferSize(0); + fmt.setProfile(QSurfaceFormat::CoreProfile); + fmt.setMajorVersion(4); + fmt.setMinorVersion(5); + + // Turn off vsync, since Qt generally gives us at most frame rate + // (display frequency) / (number of QGLWidgets active). + fmt.setSwapInterval(0); + + QSurfaceFormat::setDefaultFormat(fmt); + + QGLFormat::setDefaultFormat(QGLFormat::fromSurfaceFormat(fmt)); QApplication app(argc, argv); + global_share_widget = new QGLWidget(); + if (!global_share_widget->isValid()) { + fprintf(stderr, "Failed to initialize OpenGL. Futatabi needs at least OpenGL 4.5 to function properly.\n"); + exit(1); + } + + // Initialize Movit. + { + QSurface *surface = create_surface(); + QOpenGLContext *context = create_context(surface); + make_current(context, surface); + CHECK(movit::init_movit(MOVIT_SHADER_DIR, movit::MOVIT_DEBUG_OFF)); + delete_context(context); + // TODO: Delete the surface, too. + } + MainWindow mainWindow; mainWindow.show(); - thread(thread_func).detach(); + thread(record_thread_func).detach(); return app.exec(); } -int thread_func() +int record_thread_func() { - auto format_ctx = avformat_open_input_unique("example.mp4", nullptr, nullptr); + auto format_ctx = avformat_open_input_unique("multiangle.mp4", nullptr, nullptr); if (format_ctx == nullptr) { fprintf(stderr, "%s: Error opening file\n", "example.mp4"); return 1; } + int64_t last_pts = -1; + for ( ;; ) { AVPacket pkt; unique_ptr pkt_cleanup( @@ -69,8 +120,14 @@ int thread_func() if (av_read_frame(format_ctx.get(), &pkt) != 0) { break; } - fprintf(stderr, "Got a frame from camera %d, pts = %ld, size = %d\n", - pkt.stream_index, pkt.pts, pkt.size); + + // Convert pts to our own timebase. + // TODO: Figure out offsets, too. + AVRational stream_timebase = format_ctx->streams[pkt.stream_index]->time_base; + pkt.pts = av_rescale_q(pkt.pts, stream_timebase, AVRational{ 1, TIMEBASE }); + + //fprintf(stderr, "Got a frame from camera %d, pts = %ld, size = %d\n", + // pkt.stream_index, pkt.pts, pkt.size); string filename = filename_for_frame(pkt.stream_index, pkt.pts); FILE *fp = fopen(filename.c_str(), "wb"); if (fp == nullptr) { @@ -82,16 +139,25 @@ int thread_func() post_to_main_thread([pkt] { if (pkt.stream_index == 0) { - global_mainwindow->ui->input1_display->setFrame(pkt.stream_index, pkt.pts); + global_mainwindow->ui->input1_display->setFrame(pkt.stream_index, pkt.pts, /*interpolated=*/false); } else if (pkt.stream_index == 1) { - global_mainwindow->ui->input2_display->setFrame(pkt.stream_index, pkt.pts); + global_mainwindow->ui->input2_display->setFrame(pkt.stream_index, pkt.pts, /*interpolated=*/false); + } else if (pkt.stream_index == 2) { + global_mainwindow->ui->input3_display->setFrame(pkt.stream_index, pkt.pts, /*interpolated=*/false); + } else if (pkt.stream_index == 3) { + global_mainwindow->ui->input4_display->setFrame(pkt.stream_index, pkt.pts, /*interpolated=*/false); } }); assert(pkt.stream_index < MAX_STREAMS); frames[pkt.stream_index].push_back(pkt.pts); - this_thread::sleep_for(milliseconds(1000) / 120); + // Hack. Remove when we're dealing with live streams. + if (last_pts != -1) { + this_thread::sleep_for(microseconds((pkt.pts - last_pts) * 1000000 / TIMEBASE)); + } + last_pts = pkt.pts; + current_pts = pkt.pts; } return 0;