git.sesse.net Git - nageru/blob - nageru/decklink_util.cpp

   1 #include "DeckLinkAPI.h"
   2 #include "DeckLinkAPIDiscovery.h"
   3 #include "DeckLinkAPIModes.h"
   4 #include "DeckLinkAPITypes.h"
   5 #include "LinuxCOM.h"
   6
   7 #include <assert.h>
   8 #include <bmusb/bmusb.h>
   9 #include <stdint.h>
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <map>
  13 #include <stdlib.h>
  14 #include <utility>
  15
  16 #include "decklink_util.h"
  17 #include "flags.h"
  18
  19 using namespace bmusb;
  20 using namespace std;
  21
  22 map<uint32_t, VideoMode> summarize_video_modes(IDeckLinkDisplayModeIterator *mode_it, unsigned card_index)
  23 {
  24         map<uint32_t, VideoMode> video_modes;
  25
  26         for (IDeckLinkDisplayMode *mode_ptr; mode_it->Next(&mode_ptr) == S_OK; mode_ptr->Release()) {
  27                 VideoMode mode;
  28
  29                 const char *mode_name;
  30                 if (mode_ptr->GetName(&mode_name) != S_OK) {
  31                         mode.name = "Unknown mode";
  32                 } else {
  33                         mode.name = mode_name;
  34                         free((char *)mode_name);
  35                 }
  36
  37                 mode.autodetect = false;
  38
  39                 mode.width = mode_ptr->GetWidth();
  40                 mode.height = mode_ptr->GetHeight();
  41
  42                 BMDTimeScale frame_rate_num;
  43                 BMDTimeValue frame_rate_den;
  44                 if (mode_ptr->GetFrameRate(&frame_rate_den, &frame_rate_num) != S_OK) {
  45                         fprintf(stderr, "Could not get frame rate for mode '%s' on card %d\n", mode.name.c_str(), card_index);
  46                         abort();
  47                 }
  48                 mode.frame_rate_num = frame_rate_num;
  49                 mode.frame_rate_den = frame_rate_den;
  50
  51                 // TODO: Respect the TFF/BFF flag.
  52                 mode.interlaced = (mode_ptr->GetFieldDominance() == bmdLowerFieldFirst || mode_ptr->GetFieldDominance() == bmdUpperFieldFirst);
  53
  54                 uint32_t id = mode_ptr->GetDisplayMode();
  55                 video_modes.insert(make_pair(id, mode));
  56         }
  57
  58         return video_modes;
  59 }
  60
  61 BMDVideoConnection pick_default_video_connection(IDeckLink *card, BMDDeckLinkAttributeID attribute_id, unsigned card_index)
  62 {
  63         assert(attribute_id == BMDDeckLinkVideoInputConnections ||
  64                attribute_id == BMDDeckLinkVideoOutputConnections);
  65
  66         IDeckLinkAttributes *attr;
  67         if (card->QueryInterface(IID_IDeckLinkAttributes, (void**)&attr) != S_OK) {
  68                 fprintf(stderr, "Card %u has no attributes\n", card_index);
  69                 abort();
  70         }
  71
  72         int64_t connection_mask;
  73         if (attr->GetInt(attribute_id, &connection_mask) != S_OK) {
  74                 if (attribute_id == BMDDeckLinkVideoInputConnections) {
  75                         fprintf(stderr, "Failed to enumerate video inputs for card %u\n", card_index);
  76                 } else {
  77                         fprintf(stderr, "Failed to enumerate video outputs for card %u\n", card_index);
  78                 }
  79                 abort();
  80         }
  81         attr->Release();
  82         if (connection_mask == 0) {
  83                 if (attribute_id == BMDDeckLinkVideoInputConnections) {
  84                         fprintf(stderr, "Card %u has no input connections\n", card_index);
  85                 } else {
  86                         fprintf(stderr, "Card %u has no output connections\n", card_index);
  87                 }
  88                 abort();
  89         }
  90
  91         if ((connection_mask & bmdVideoConnectionHDMI) &&
  92             global_flags.default_hdmi_input) {
  93                 return bmdVideoConnectionHDMI;
  94         } else if (connection_mask & bmdVideoConnectionSDI) {
  95                 return bmdVideoConnectionSDI;
  96         } else if (connection_mask & bmdVideoConnectionHDMI) {
  97                 return bmdVideoConnectionHDMI;
  98         } else {
  99                 // Fallback: Return lowest-set bit, whatever that might be.
 100                 return connection_mask & (-connection_mask);
 101         }
 102 }
 103
 104 bool decklink_card_is_active(IDeckLink *card, unsigned card_index)
 105 {
 106         IDeckLinkStatus *status;
 107         if (card->QueryInterface(IID_IDeckLinkStatus, (void**)&status) != S_OK) {
 108                 fprintf(stderr, "Card %u has no status interface\n", card_index);
 109                 abort();
 110         }
 111
 112         int64_t duplex_mode;
 113         if (status->GetInt(bmdDeckLinkStatusDuplexMode, &duplex_mode) != S_OK) {
 114                 fprintf(stderr, "Could not query duplex mode for card %u\n", card_index);
 115                 abort();
 116         }
 117
 118         status->Release();
 119
 120         return (duplex_mode != bmdDuplexStatusInactive);
 121 }