#include <assert.h>
#include <errno.h>
#include <libusb.h>
+#include <unistd.h>
#include <netinet/in.h>
#include <sched.h>
#include <stdint.h>
using namespace std;
using namespace std::placeholders;
+#define USB_VENDOR_BLACKMAGIC 0x1edb
#define MIN_WIDTH 640
#define HEADER_SIZE 44
//#define HEADER_SIZE 0
#define FRAME_SIZE (8 << 20) // 8 MB.
#define USB_VIDEO_TRANSFER_SIZE (128 << 10) // 128 kB.
+card_connected_callback_t BMUSBCapture::card_connected_callback = nullptr;
+
namespace {
FILE *audiofp;
void BMUSBCapture::cb_xfr(struct libusb_transfer *xfr)
{
- if (xfr->status != LIBUSB_TRANSFER_COMPLETED) {
- fprintf(stderr, "transfer status %d\n", xfr->status);
+ if (xfr->status != LIBUSB_TRANSFER_COMPLETED &&
+ xfr->status != LIBUSB_TRANSFER_NO_DEVICE) {
+ fprintf(stderr, "error: transfer status %d\n", xfr->status);
libusb_free_transfer(xfr);
exit(3);
}
assert(xfr->user_data != nullptr);
BMUSBCapture *usb = static_cast<BMUSBCapture *>(xfr->user_data);
+ if (xfr->status == LIBUSB_TRANSFER_NO_DEVICE) {
+ if (!usb->disconnected) {
+ fprintf(stderr, "Device went away, stopping transfers.\n");
+ usb->disconnected = true;
+ if (usb->card_disconnected_callback) {
+ usb->card_disconnected_callback();
+ }
+ }
+ // Don't reschedule the transfer; the loop will stop by itself.
+ return;
+ }
+
if (xfr->type == LIBUSB_TRANSFER_TYPE_ISOCHRONOUS) {
if (xfr->endpoint == 0x84) {
decode_packs(xfr, "DeckLinkAudioResyncT", 20, &usb->current_audio_frame, "audio", bind(&BMUSBCapture::start_new_audio_block, usb, _1));
}
}
+int BMUSBCapture::cb_hotplug(libusb_context *ctx, libusb_device *dev, libusb_hotplug_event event, void *user_data)
+{
+ if (card_connected_callback != nullptr) {
+ libusb_device_descriptor desc;
+ if (libusb_get_device_descriptor(dev, &desc) < 0) {
+ fprintf(stderr, "Error getting device descriptor for hotplugged device %p, killing hotplug\n", dev);
+ libusb_unref_device(dev);
+ return 1;
+ }
+
+ if ((desc.idVendor == USB_VENDOR_BLACKMAGIC && desc.idProduct == 0xbd3b) ||
+ (desc.idVendor == USB_VENDOR_BLACKMAGIC && desc.idProduct == 0xbd4f)) {
+ card_connected_callback(dev); // Callback takes ownership.
+ return 0;
+ }
+ }
+ libusb_unref_device(dev);
+ return 0;
+}
+
void BMUSBCapture::usb_thread_func()
{
sched_param param;
libusb_device *device;
};
+const char *get_product_name(uint16_t product)
+{
+ if (product == 0xbd3b) {
+ return "Intensity Shuttle";
+ } else if (product == 0xbd4f) {
+ return "UltraStudio SDI";
+ } else {
+ assert(false);
+ return nullptr;
+ }
+}
+
+string get_card_description(int id, uint8_t bus, uint8_t port, uint16_t product)
+{
+ const char *product_name = get_product_name(product);
+
+ char buf[256];
+ snprintf(buf, sizeof(buf), "USB card %d: Bus %03u Device %03u %s",
+ id, bus, port, product_name);
+ return buf;
+}
+
libusb_device_handle *open_card(int card_index, string *description)
-{
+{
libusb_device **devices;
ssize_t num_devices = libusb_get_device_list(nullptr, &devices);
if (num_devices == -1) {
uint8_t bus = libusb_get_bus_number(devices[i]);
uint8_t port = libusb_get_port_number(devices[i]);
- if (!(desc.idVendor == 0x1edb && desc.idProduct == 0xbd3b) &&
- !(desc.idVendor == 0x1edb && desc.idProduct == 0xbd4f)) {
+ if (!(desc.idVendor == USB_VENDOR_BLACKMAGIC && desc.idProduct == 0xbd3b) &&
+ !(desc.idVendor == USB_VENDOR_BLACKMAGIC && desc.idProduct == 0xbd4f)) {
libusb_unref_device(devices[i]);
continue;
}
});
for (size_t i = 0; i < found_cards.size(); ++i) {
- const char *product_name = nullptr;
- if (found_cards[i].product == 0xbd3b) {
- product_name = "Intensity Shuttle";
- } else if (found_cards[i].product == 0xbd4f) {
- product_name = "UltraStudio SDI";
- } else {
- assert(false);
- }
-
- char buf[256];
- snprintf(buf, sizeof(buf), "Card %d: Bus %03u Device %03u %s",
- int(i), found_cards[i].bus, found_cards[i].port, product_name);
+ string tmp_description = get_card_description(i, found_cards[i].bus, found_cards[i].port, found_cards[i].product);
+ fprintf(stderr, "%s\n", tmp_description.c_str());
if (i == size_t(card_index)) {
- *description = buf;
+ *description = tmp_description;
}
- fprintf(stderr, "%s\n", buf);
}
if (size_t(card_index) >= found_cards.size()) {
return devh;
}
+libusb_device_handle *open_card(unsigned card_index, libusb_device *dev, string *description)
+{
+ uint8_t bus = libusb_get_bus_number(dev);
+ uint8_t port = libusb_get_port_number(dev);
+
+ libusb_device_descriptor desc;
+ if (libusb_get_device_descriptor(dev, &desc) < 0) {
+ fprintf(stderr, "Error getting device descriptor for device %p\n", dev);
+ exit(1);
+ }
+
+ *description = get_card_description(card_index, bus, port, desc.idProduct);
+
+ libusb_device_handle *devh;
+ int rc = libusb_open(dev, &devh);
+ if (rc < 0) {
+ fprintf(stderr, "Error opening card %p: %s\n", dev, libusb_error_name(rc));
+ exit(1);
+ }
+
+ return devh;
+}
+
void BMUSBCapture::configure_card()
{
if (video_frame_allocator == nullptr) {
- set_video_frame_allocator(new MallocFrameAllocator(FRAME_SIZE, NUM_QUEUED_VIDEO_FRAMES)); // FIXME: leak.
+ owned_video_frame_allocator.reset(new MallocFrameAllocator(FRAME_SIZE, NUM_QUEUED_VIDEO_FRAMES));
+ set_video_frame_allocator(owned_video_frame_allocator.get());
}
if (audio_frame_allocator == nullptr) {
- set_audio_frame_allocator(new MallocFrameAllocator(65536, NUM_QUEUED_AUDIO_FRAMES)); // FIXME: leak.
+ owned_audio_frame_allocator.reset(new MallocFrameAllocator(65536, NUM_QUEUED_AUDIO_FRAMES));
+ set_audio_frame_allocator(owned_audio_frame_allocator.get());
}
dequeue_thread_should_quit = false;
dequeue_thread = thread(&BMUSBCapture::dequeue_thread_func, this);
exit(1);
}
- libusb_device_handle *devh = open_card(card_index, &description);
+ if (dev == nullptr) {
+ devh = open_card(card_index, &description);
+ } else {
+ devh = open_card(card_index, dev, &description);
+ libusb_unref_device(dev);
+ }
if (!devh) {
fprintf(stderr, "Error finding USB device\n");
exit(1);
// 0x20 - 720p??
// 0x30 - 576p??
+ update_capture_mode();
+
struct ctrl {
int endpoint;
int request;
{ LIBUSB_ENDPOINT_IN, 214, 16, 0 },
{ LIBUSB_ENDPOINT_IN, 214, 0, 0 },
- // seems to capture on HDMI, clearing the 0x20000000 bit seems to activate 10-bit
- // capture (v210).
- // clearing the 0x08000000 bit seems to change the capture format (other source?)
- // 0x10000000 = analog audio instead of embedded audio, it seems
- // 0x3a000000 = component video? (analog audio)
- // 0x3c000000 = composite video? (analog audio)
- // 0x3e000000 = s-video? (analog audio)
- { LIBUSB_ENDPOINT_OUT, 215, 0, 0x29000000 },
//{ LIBUSB_ENDPOINT_OUT, 215, 0, 0x80000100 },
//{ LIBUSB_ENDPOINT_OUT, 215, 0, 0x09000000 },
{ LIBUSB_ENDPOINT_OUT, 215, 24, 0x73c60001 }, // latch for frame start?
xfr->user_data = this;
//libusb_submit_transfer(xfr);
- audiofp = fopen("audio.raw", "wb");
+ //audiofp = fopen("audio.raw", "wb");
// set up isochronous transfers for audio and video
for (int e = 3; e <= 4; ++e) {
//int num_transfers = (e == 3) ? 6 : 6;
- int num_transfers = 10;
+ int num_transfers = 6;
for (int i = 0; i < num_transfers; ++i) {
size_t buf_size;
int num_iso_pack, size;
num_iso_pack = 80;
buf_size = num_iso_pack * size;
}
- assert(size_t(num_iso_pack * size) <= buf_size);
- uint8_t *buf = new uint8_t[buf_size];
+ int num_bytes = num_iso_pack * size;
+ assert(size_t(num_bytes) <= buf_size);
+#if LIBUSB_API_VERSION >= 0x01000105
+ uint8_t *buf = libusb_dev_mem_alloc(devh, num_bytes);
+#else
+ uint8_t *buf = nullptr;
+#endif
+ if (buf == nullptr) {
+ fprintf(stderr, "Failed to allocate persistent DMA memory ");
+#if LIBUSB_API_VERSION >= 0x01000105
+ fprintf(stderr, "(probably too old kernel; use 4.6.0 or newer).\n");
+#else
+ fprintf(stderr, "(compiled against too old libusb-1.0).\n");
+#endif
+ fprintf(stderr, "Will go slower, and likely fail due to memory fragmentation after a few hours.\n");
+ buf = new uint8_t[num_bytes];
+ }
xfr = libusb_alloc_transfer(num_iso_pack);
if (!xfr) {
void BMUSBCapture::start_bm_thread()
{
+ // Devices leaving are discovered by seeing the isochronous packets
+ // coming back with errors, so only care about devices joining.
+ if (card_connected_callback != nullptr) {
+ if (libusb_hotplug_register_callback(
+ nullptr, LIBUSB_HOTPLUG_EVENT_DEVICE_ARRIVED, LIBUSB_HOTPLUG_NO_FLAGS,
+ USB_VENDOR_BLACKMAGIC, LIBUSB_HOTPLUG_MATCH_ANY, LIBUSB_HOTPLUG_MATCH_ANY,
+ &BMUSBCapture::cb_hotplug, nullptr, nullptr) < 0) {
+ fprintf(stderr, "libusb_hotplug_register_callback() failed\n");
+ exit(1);
+ }
+ }
+
should_quit = false;
usb_thread = thread(&BMUSBCapture::usb_thread_func);
}
decoded_video_format->frame_rate_den = 1;
return false;
}
+
+map<uint32_t, VideoMode> BMUSBCapture::get_available_video_modes() const
+{
+ // The USB3 cards autodetect, and seem to have no provision for forcing modes.
+ VideoMode auto_mode;
+ auto_mode.name = "Autodetect";
+ auto_mode.autodetect = true;
+ return {{ 0, auto_mode }};
+}
+
+uint32_t BMUSBCapture::get_current_video_mode() const
+{
+ return 0; // Matches get_available_video_modes().
+}
+
+void BMUSBCapture::set_video_mode(uint32_t video_mode_id)
+{
+ assert(video_mode_id == 0); // Matches get_available_video_modes().
+}
+
+std::map<uint32_t, std::string> BMUSBCapture::get_available_video_inputs() const
+{
+ return {
+ { 0x00000000, "HDMI/SDI" },
+ { 0x02000000, "Component" },
+ { 0x04000000, "Composite" },
+ { 0x06000000, "S-video" }
+ };
+}
+
+void BMUSBCapture::set_video_input(uint32_t video_input_id)
+{
+ assert((video_input_id & ~0x06000000) == 0);
+ current_video_input = video_input_id;
+ update_capture_mode();
+}
+
+std::map<uint32_t, std::string> BMUSBCapture::get_available_audio_inputs() const
+{
+ return {
+ { 0x00000000, "Embedded" },
+ { 0x10000000, "Analog" }
+ };
+}
+
+void BMUSBCapture::set_audio_input(uint32_t audio_input_id)
+{
+ assert((audio_input_id & ~0x10000000) == 0);
+ current_audio_input = audio_input_id;
+ update_capture_mode();
+}
+
+void BMUSBCapture::update_capture_mode()
+{
+ // clearing the 0x20000000 bit seems to activate 10-bit capture (v210).
+ // clearing the 0x08000000 bit seems to change the capture format (other source?)
+ uint32_t mode = htonl(0x29000000 | current_video_input | current_audio_input);
+
+ int rc = libusb_control_transfer(devh, LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_OUT,
+ /*request=*/215, /*value=*/0, /*index=*/0, (unsigned char *)&mode, sizeof(mode), /*timeout=*/0);
+ if (rc < 0) {
+ fprintf(stderr, "Error on setting mode: %s\n", libusb_error_name(rc));
+ exit(1);
+ }
+}