X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Fdecklink_capture.cpp;h=e239ec964f6f70d17fc13c24494ead77e7d726b4;hb=HEAD;hp=6ccbadc339791ca12816dcc57920855f049ad404;hpb=9a0f617093e7cde02a824b90f82f129c47193939;p=nageru

diff --git a/nageru/decklink_capture.cpp b/nageru/decklink_capture.cpp
index 6ccbadc..90d86b1 100644
--- a/nageru/decklink_capture.cpp
+++ b/nageru/decklink_capture.cpp
@@ -1,10 +1,16 @@
 #include "decklink_capture.h"
+#include "defs.h"
 
 #include <DeckLinkAPI.h>
 #include <DeckLinkAPIConfiguration.h>
 #include <DeckLinkAPIDiscovery.h>
 #include <DeckLinkAPIModes.h>
+#include <DeckLinkAPITypes.h>
+#include <LinuxCOM.h>
 #include <assert.h>
+#include <errno.h>
+#include <sched.h>
+#include <string>
 #ifdef __SSE2__
 #include <immintrin.h>
 #endif
@@ -24,8 +30,6 @@
 #include "shared/memcpy_interleaved.h"
 #include "v210_converter.h"
 
-#define FRAME_SIZE (8 << 20)  // 8 MB.
-
 using namespace std;
 using namespace std::chrono;
 using namespace std::placeholders;
@@ -246,13 +250,20 @@ HRESULT STDMETHODCALLTYPE DeckLinkCapture::VideoInputFrameArrived(
 		const int stride = video_frame->GetRowBytes();
 		const BMDPixelFormat format = video_frame->GetPixelFormat();
 		assert(format == pixel_format_to_bmd(current_pixel_format));
-		if (global_flags.ten_bit_input) {
+		if (global_flags.bit_depth > 8) {
 			assert(stride == int(v210Converter::get_v210_stride(width)));
 		} else {
 			assert(stride == width * 2);
 		}
 
-		current_video_frame = video_frame_allocator->create_frame(width, height, stride);
+		if (width * stride > FRAME_SIZE) {
+			// TODO: If we had an OpenGL context here, calling create_frame()
+			// would be completely fine.
+			fprintf(stderr, "Card %u: Captured frame %d x %d (stride %d) would be larger than supported frame size (%d > %d), skipping.\n",
+				card_index, width, height, stride, width * stride, FRAME_SIZE);
+		} else {
+			current_video_frame = video_frame_allocator->create_frame(width, height, stride);
+		}
 		if (current_video_frame.data != nullptr) {
 			const uint8_t *src;
 			video_frame->GetBytes((void **)&src);