X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=decklink_util.cpp;h=a6bdfd9cd65fe4831491a515346bd53322d600e7;hb=96cb6414f85e0ef4d660b7bd56267303e80fcd05;hp=0e775dea4981b709077701389e30c40b4061b8ed;hpb=00bcd63104da3965714baaffd3bf6526e85a6f26;p=nageru

diff --git a/decklink_util.cpp b/decklink_util.cpp
index 0e775de..a6bdfd9 100644
--- a/decklink_util.cpp
+++ b/decklink_util.cpp
@@ -1,6 +1,8 @@
 #include <DeckLinkAPI.h>
 #include <DeckLinkAPIModes.h>
 
+#include <assert.h>
+
 #include "decklink_util.h"
 
 using namespace bmusb;
@@ -43,3 +45,43 @@ map<uint32_t, VideoMode> summarize_video_modes(IDeckLinkDisplayModeIterator *mod
 
 	return video_modes;
 }
+
+BMDVideoConnection pick_default_video_connection(IDeckLink *card, BMDDeckLinkAttributeID attribute_id, unsigned card_index)
+{
+	assert(attribute_id == BMDDeckLinkVideoInputConnections ||
+	       attribute_id == BMDDeckLinkVideoOutputConnections);
+
+	IDeckLinkAttributes *attr;
+	if (card->QueryInterface(IID_IDeckLinkAttributes, (void**)&attr) != S_OK) {
+		fprintf(stderr, "Card %u has no attributes\n", card_index);
+		exit(1);
+	}
+
+	int64_t connection_mask;
+	if (attr->GetInt(attribute_id, &connection_mask) != S_OK) {
+		if (attribute_id == BMDDeckLinkVideoInputConnections) {
+			fprintf(stderr, "Failed to enumerate video inputs for card %u\n", card_index);
+		} else {
+			fprintf(stderr, "Failed to enumerate video outputs for card %u\n", card_index);
+		}
+		exit(1);
+	}
+	attr->Release();
+	if (connection_mask == 0) {
+		if (attribute_id == BMDDeckLinkVideoInputConnections) {
+			fprintf(stderr, "Card %u has no input connections\n", card_index);
+		} else {
+			fprintf(stderr, "Card %u has no outpu connectionss\n", card_index);
+		}
+		exit(1);
+	}
+
+	if (connection_mask & bmdVideoConnectionHDMI) {
+		return bmdVideoConnectionHDMI;
+	} else if (connection_mask & bmdVideoConnectionSDI) {
+		return bmdVideoConnectionSDI;
+	} else {
+		// Fallback: Return lowest-set bit, whatever that might be.
+		return connection_mask & (-connection_mask);
+	}
+}