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

diff --git a/decklink_util.cpp b/decklink_util.cpp
index 0e775de..d808196 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;
@@ -18,6 +20,7 @@ map<uint32_t, VideoMode> summarize_video_modes(IDeckLinkDisplayModeIterator *mod
 			mode.name = "Unknown mode";
 		} else {
 			mode.name = mode_name;
+			free((char *)mode_name);
 		}
 
 		mode.autodetect = false;
@@ -43,3 +46,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);
+	}
+}