1 #ifndef _DECKLINK_CAPTURE_H
2 #define _DECKLINK_CAPTURE_H 1
4 #include <DeckLinkAPI.h>
10 #include "bmusb/bmusb.h"
13 class IDeckLinkDisplayMode;
15 // TODO: Adjust CaptureInterface to be a little less bmusb-centric.
16 // There are too many member functions here that don't really do anything.
17 class DeckLinkCapture : public CaptureInterface, IDeckLinkInputCallback
20 DeckLinkCapture(IDeckLink *card, int card_index);
23 // IDeckLinkInputCallback.
24 HRESULT STDMETHODCALLTYPE QueryInterface(REFIID, LPVOID *) override;
25 ULONG STDMETHODCALLTYPE AddRef() override;
26 ULONG STDMETHODCALLTYPE Release() override;
27 HRESULT STDMETHODCALLTYPE VideoInputFormatChanged(
28 BMDVideoInputFormatChangedEvents,
29 IDeckLinkDisplayMode*,
30 BMDDetectedVideoInputFormatFlags) override;
31 HRESULT STDMETHODCALLTYPE VideoInputFrameArrived(
32 IDeckLinkVideoInputFrame *video_frame,
33 IDeckLinkAudioInputPacket *audio_frame) override;
36 void set_video_frame_allocator(FrameAllocator *allocator) override
38 video_frame_allocator = allocator;
41 FrameAllocator *get_video_frame_allocator() override
43 return video_frame_allocator;
46 // Does not take ownership.
47 void set_audio_frame_allocator(FrameAllocator *allocator) override
49 audio_frame_allocator = allocator;
52 FrameAllocator *get_audio_frame_allocator() override
54 return audio_frame_allocator;
57 void set_frame_callback(frame_callback_t callback) override
59 frame_callback = callback;
62 void set_dequeue_thread_callbacks(std::function<void()> init, std::function<void()> cleanup) override
64 dequeue_init_callback = init;
65 dequeue_cleanup_callback = cleanup;
66 has_dequeue_callbacks = true;
69 std::string get_description() const override
74 void configure_card() override;
75 void start_bm_capture() override;
76 void stop_dequeue_thread() override;
78 std::map<uint32_t, VideoMode> get_available_video_modes() const override { return video_modes; }
79 void set_video_mode(uint32_t video_mode_id) override;
80 uint32_t get_current_video_mode() const override { return current_video_mode; }
82 std::map<uint32_t, std::string> get_available_video_inputs() const override { return video_inputs; }
83 void set_video_input(uint32_t video_input_id) override;
84 uint32_t get_current_video_input() const override { return current_video_input; }
86 std::map<uint32_t, std::string> get_available_audio_inputs() const override { return audio_inputs; }
87 void set_audio_input(uint32_t audio_input_id) override;
88 uint32_t get_current_audio_input() const override { return current_audio_input; }
91 void set_video_mode_no_restart(uint32_t video_mode_id);
93 std::atomic<int> refcount{1};
94 bool done_init = false;
95 std::string description;
96 uint16_t timecode = 0;
99 bool has_dequeue_callbacks = false;
100 std::function<void()> dequeue_init_callback = nullptr;
101 std::function<void()> dequeue_cleanup_callback = nullptr;
103 FrameAllocator *video_frame_allocator = nullptr;
104 FrameAllocator *audio_frame_allocator = nullptr;
105 frame_callback_t frame_callback = nullptr;
107 IDeckLinkConfiguration *config = nullptr;
109 IDeckLinkInput *input = nullptr;
110 BMDTimeValue frame_duration;
111 BMDTimeScale time_scale;
113 std::map<uint32_t, VideoMode> video_modes;
114 BMDDisplayMode current_video_mode;
116 std::map<uint32_t, std::string> video_inputs;
117 BMDVideoConnection current_video_input;
119 std::map<uint32_t, std::string> audio_inputs;
120 BMDAudioConnection current_audio_input;
123 #endif // !defined(_DECKLINK_CAPTURE_H)