scottk/UltraGrid
1
0
Fork 0
mirror of https://github.com/outbackdingo/UltraGrid.git synced 2026-03-21 20:40:27 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
467cb49bd03196691ab0e032f5af9676a6b40dd1
UltraGrid/src/main.cpp
Martin Pulec 7db96528cf Name threads
2019-05-30 11:38:05 +02:00

1349 lines
58 KiB
C++
Raw Blame History

/*
* FILE: main.cpp
* AUTHORS: Colin Perkins <csp@csperkins.org>
* Ladan Gharai <ladan@isi.edu>
* Martin Benes <martinbenesh@gmail.com>
* Lukas Hejtmanek <xhejtman@ics.muni.cz>
* Petr Holub <hopet@ics.muni.cz>
* Milos Liska <xliska@fi.muni.cz>
* Jiri Matela <matela@ics.muni.cz>
* Dalibor Matura <255899@mail.muni.cz>
* Ian Wesley-Smith <iwsmith@cct.lsu.edu>
* David Cassany <david.cassany@i2cat.net>
* Ignacio Contreras <ignacio.contreras@i2cat.net>
* Gerard Castillo <gerard.castillo@i2cat.net>
* Martin Pulec <pulec@cesnet.cz>
*
* Copyright (c) 2005-2014 Fundació i2CAT, Internet I Innovació Digital a Catalunya
* Copyright (c) 2005-2018 CESNET z.s.p.o.
* Copyright (c) 2001-2004 University of Southern California
* Copyright (c) 2003-2004 University of Glasgow
*
* Redistribution and use in source and binary forms, with or without
* modification, is permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
*
* This product includes software developed by the University of Southern
* California Information Sciences Institute. This product also includes
* software developed by CESNET z.s.p.o.
*
* 4. Neither the name of the University nor of the Institute may be used
* to endorse or promote products derived from this software without
* specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING,
* BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
* AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
* EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
* OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
* EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#include "config_unix.h"
#include "config_win32.h"
#endif // HAVE_CONFIG_H
#include <string.h>
#include <stdlib.h>
#include <getopt.h>
#include <pthread.h>
#include <rang.hpp>
#include "control_socket.h"
#include "debug.h"
#include "host.h"
#include "keyboard_control.h"
#include "lib_common.h"
#include "messaging.h"
#include "module.h"
#include "rtp/rtp.h"
#include "rtsp/rtsp_utils.h"
#include "ug_runtime_error.h"
#include "utils/misc.h"
#include "utils/net.h"
#include "utils/thread.h"
#include "utils/wait_obj.h"
#include "video.h"
#include "video_capture.h"
#include "video_capture/import.h"
#include "video_display.h"
#include "video_compress.h"
#include "export.h"
#include "video_rxtx.h"
#include "audio/audio.h"
#include "audio/audio_capture.h"
#include "audio/codec.h"
#include "audio/utils.h"
#include <iostream>
#include <memory>
#include <string>
#define PORT_BASE 5004
#define DEFAULT_AUDIO_FEC "none"
static constexpr const char *DEFAULT_VIDEO_COMPRESSION = "none";
static constexpr const char *DEFAULT_AUDIO_CODEC = "PCM";
#define OPT_AUDIO_CAPTURE_CHANNELS (('a' << 8) | 'c')
#define OPT_AUDIO_CAPTURE_FORMAT (('C' << 8) | 'F')
#define OPT_AUDIO_CHANNEL_MAP (('a' << 8) | 'm')
#define OPT_AUDIO_CODEC (('A' << 8) | 'C')
#define OPT_AUDIO_DELAY (('A' << 8) | 'D')
#define OPT_AUDIO_PROTOCOL (('A' << 8) | 'P')
#define OPT_AUDIO_SCALE (('a' << 8) | 's')
#define OPT_CAPABILITIES (('C' << 8) | 'C')
#define OPT_CAPTURE_FILTER (('O' << 8) | 'F')
#define OPT_CONTROL_PORT (('C' << 8) | 'P')
#define OPT_CUDA_DEVICE (('C' << 8) | 'D')
#define OPT_ECHO_CANCELLATION (('E' << 8) | 'C')
#define OPT_ENCRYPTION (('E' << 8) | 'N')
#define OPT_FULLHELP (('F' << 8u) | 'H')
#define OPT_EXPORT (('E' << 8) | 'X')
#define OPT_IMPORT (('I' << 8) | 'M')
#define OPT_LIST_MODULES (('L' << 8) | 'M')
#define OPT_MCAST_IF (('M' << 8) | 'I')
#define OPT_PARAM (('O' << 8) | 'P')
#define OPT_PROTOCOL (('P' << 8) | 'R')
#define OPT_START_PAUSED (('S' << 8) | 'P')
#define OPT_VERBOSE (('V' << 8) | 'E')
#define OPT_VIDEO_PROTOCOL (('V' << 8) | 'P')
#define OPT_WINDOW_TITLE (('W' << 8) | 'T')
#define MAX_CAPTURE_COUNT 17
using namespace std;
struct state_uv {
state_uv() : capture_device{}, display_device{}, audio{}, state_video_rxtx{} {
module_init_default(&root_module);
root_module.cls = MODULE_CLASS_ROOT;
root_module.priv_data = this;
}
~state_uv() {
module_done(&root_module);
}
struct vidcap *capture_device;
struct display *display_device;
struct state_audio *audio;
struct module root_module;
video_rxtx *state_video_rxtx;
};
static int exit_status = EXIT_SUCCESS;
static struct state_uv *uv_state;
static void signal_handler(int signal)
{
if (log_level >= LOG_LEVEL_DEBUG) {
char msg[] = "Caught signal ";
char buf[128];
char *ptr = buf;
for (size_t i = 0; i < sizeof msg - 1; ++i) {
*ptr++ = msg[i];
}
if (signal / 10) {
*ptr++ = '0' + signal/10;
}
*ptr++ = '0' + signal%10;
*ptr++ = '\n';
size_t bytes = ptr - buf;
ptr = buf;
do {
ssize_t written = write(STDERR_FILENO, ptr, bytes);
if (written < 0) {
break;
}
bytes -= written;
ptr += written;
} while (bytes > 0);
}
exit_uv(0);
}
static void crash_signal_handler(int sig)
{
char buf[1024];
char *ptr = buf;
*ptr++ = '\n';
const char message1[] = " has crashed";
for (size_t i = 0; i < sizeof PACKAGE_NAME - 1; ++i) {
*ptr++ = PACKAGE_NAME[i];
}
for (size_t i = 0; i < sizeof message1 - 1; ++i) {
*ptr++ = message1[i];
}
#ifndef WIN32
*ptr++ = ' '; *ptr++ = '(';
for (size_t i = 0; i < sizeof sys_siglist[sig] - 1; ++i) {
if (sys_siglist[sig][i] == '\0') {
break;
}
*ptr++ = sys_siglist[sig][i];
}
*ptr++ = ')';
#endif
const char message2[] = ".\n\nPlease send a bug report to address ";
for (size_t i = 0; i < sizeof message2 - 1; ++i) {
*ptr++ = message2[i];
}
for (size_t i = 0; i < sizeof PACKAGE_BUGREPORT - 1; ++i) {
*ptr++ = PACKAGE_BUGREPORT[i];
}
*ptr++ = '.'; *ptr++ = '\n';
const char message3[] = "You may find some tips how to report bugs in file REPORTING-BUGS distributed with ";
for (size_t i = 0; i < sizeof message3 - 1; ++i) {
*ptr++ = message3[i];
}
for (size_t i = 0; i < sizeof PACKAGE_NAME - 1; ++i) {
*ptr++ = PACKAGE_NAME[i];
}
*ptr++ = '.'; *ptr++ = '\n';
size_t bytes = ptr - buf;
ptr = buf;
do {
ssize_t written = write(STDERR_FILENO, ptr, bytes);
if (written < 0) {
break;
}
bytes -= written;
ptr += written;
} while (bytes > 0);
signal(SIGABRT, SIG_DFL);
signal(SIGSEGV, SIG_DFL);
raise(sig);
}
void exit_uv(int status) {
exit_status = status;
should_exit = true;
}
static void usage(const char *exec_path, bool full = false)
{
printf("\nUsage: %s [options] address\n\n", exec_path ? exec_path : "<executable_path>");
printf("Options:\n\n");
printf
("\t-h|--fullhelp \tshow usage (basic/full)\n");
printf("\n");
printf
("\t-d <display_device> \tselect display device, use '-d help'\n");
printf("\t \tto get list of supported devices\n");
printf("\n");
printf
("\t-t <capture_device> \tselect capture device, use '-t help'\n");
printf("\t \tto get list of supported devices\n");
printf("\n");
printf("\t-c <cfg> \tvideo compression (see '-c help')\n");
printf("\n");
printf("\t-r <playback_device> \taudio playback device (see '-r help')\n");
printf("\n");
printf("\t-s <capture_device> \taudio capture device (see '-s help')\n");
printf("\n");
if (full) {
printf("\t--verbose[=<level>] \tprint verbose messages (optinaly specify level [0-%d])\n", LOG_LEVEL_MAX);
printf("\n");
printf("\t--list-modules \tprints list of modules\n");
printf("\n");
printf("\t--control-port <port>[:0|1] \tset control port (default port: 5054)\n");
printf("\t \tconnection types: 0- Server (default), 1- Client\n");
printf("\n");
printf("\t--video-protocol <proto> \ttransmission protocol, see '--video-protocol help'\n");
printf("\t \tfor list. Use --video-protocol rtsp for RTSP server\n");
printf("\t \t(see --video-protocol rtsp:help for usage)\n");
printf("\n");
printf("\t--audio-protocol <proto>[:<settings>]\t<proto> can be " AUDIO_PROTOCOLS "\n");
printf("\n");
printf("\t--protocol <proto> \tshortcut for '--audio-protocol <proto> --video-protocol <proto>'\n");
printf("\n");
#ifdef HAVE_IPv6
printf("\t-4/-6 \tforce IPv4/IPv6 resolving\n");
printf("\n");
#endif // HAVE_IPv6
printf("\t--mcast-if <iface> \tbind to specified interface for multicast\n");
printf("\n");
printf("\t-M <video_mode> \treceived video mode (eg tiled-4K, 3D,\n");
printf("\t \tdual-link)\n");
printf("\n");
printf("\t-p <postprocess> | help \tpostprocess module\n");
printf("\n");
}
printf("\t-f [A:|V:]<settings> \tFEC settings (audio or video) - use \"none\"\n"
"\t \t\"mult:<nr>\",\n");
printf("\t \t\"ldgm:<max_expected_loss>%%\" or\n");
printf("\t \t\"ldgm:<k>:<m>:<c>\"\n");
printf("\t \t\"rs:<k>:<n>\"\n");
printf("\n");
printf("\t-P <port> | <video_rx>:<video_tx>[:<audio_rx>:<audio_tx>]\n");
printf("\t \t<port> is base port number, also 3 subsequent\n");
printf("\t \tports can be used for RTCP and audio\n");
printf("\t \tstreams. Default: %d.\n", PORT_BASE);
printf("\t \tYou can also specify all two or four ports\n");
printf("\t \tdirectly.\n");
printf("\n");
printf("\t-l <limit_bitrate> | unlimited | auto\tlimit sending bitrate\n");
printf("\t \tto <limit_bitrate> (with optional k/M/G suffix)\n");
printf("\n");
if (full) {
printf("\t-A <address> \taudio destination address\n");
printf("\t \tIf not specified, will use same as for video\n");
printf("\n");
}
printf("\t--audio-capture-format <fmt> | help format of captured audio\n");
printf("\n");
if (full) {
printf("\t--audio-channel-map <mapping> | help\n");
printf("\n");
}
printf("\t--audio-codec <codec>[:sample_rate=<sr>][:bitrate=<br>] | help\taudio codec\n");
printf("\n");
if (full) {
printf("\t--audio-delay <delay_ms> \tamount of time audio should be delayed to video\n");
printf("\t \t(may be also negative to delay video)\n");
printf("\n");
printf("\t--audio-scale <factor> | <method> | help\n");
printf("\t \tscales received audio\n");
printf("\n");
}
#if 0
printf("\t--echo-cancellation \tapply acoustic echo cancellation to audio\n");
printf("\n");
#endif
printf("\t--cuda-device <index> | help\tuse specified CUDA device\n");
printf("\n");
printf("\t--encryption <passphrase>\tkey material for encryption\n");
printf("\n");
printf("\t--playback <directory> | help \treplays recorded audio and video\n");
printf("\n");
printf("\t--record[=<directory>] \trecord captured audio and video\n");
printf("\n");
if (full) {
printf("\t--capture-filter <filter> | help\n");
printf("\t \tcapture filter(s), must be given before capture device\n");
printf("\n");
}
if (full) {
printf("\t--param <params> | help \tadditional advanced parameters, use help for list\n");
printf("\n");
}
printf("\taddress \tdestination address\n");
printf("\n");
printf("\n");
}
/**
* This function captures video and possibly compresses it.
* It then delegates sending to another thread.
*
* @param[in] arg pointer to UltraGrid (root) module
*/
static void *capture_thread(void *arg)
{
struct module *uv_mod = (struct module *)arg;
struct state_uv *uv = (struct state_uv *) uv_mod->priv_data;
struct wait_obj *wait_obj;
wait_obj = wait_obj_init();
while (!should_exit) {
/* Capture and transmit video... */
struct audio_frame *audio;
struct video_frame *tx_frame = vidcap_grab(uv->capture_device, &audio);
if (tx_frame != NULL) {
if(audio) {
audio_sdi_send(uv->audio, audio);
}
//tx_frame = vf_get_copy(tx_frame);
bool wait_for_cur_uncompressed_frame;
shared_ptr<video_frame> frame;
if (!tx_frame->callbacks.dispose) {
wait_obj_reset(wait_obj);
wait_for_cur_uncompressed_frame = true;
frame = shared_ptr<video_frame>(tx_frame, [wait_obj](struct video_frame *) {
wait_obj_notify(wait_obj);
});
} else {
wait_for_cur_uncompressed_frame = false;
frame = shared_ptr<video_frame>(tx_frame, tx_frame->callbacks.dispose);
}
uv->state_video_rxtx->send(move(frame)); // std::move really important here (!)
// wait for frame frame to be processed, eg. by compress
// or sender (uncompressed video). Grab invalidates previous frame
// (if not defined dispose function).
if (wait_for_cur_uncompressed_frame) {
wait_obj_wait(wait_obj);
tx_frame->callbacks.dispose = NULL;
tx_frame->callbacks.dispose_udata = NULL;
}
}
}
wait_obj_done(wait_obj);
return NULL;
}
static bool parse_audio_capture_format(const char *optarg)
{
if (strcmp(optarg, "help") == 0) {
printf("Usage:\n");
printf("\t--audio-capture-format {channels=<num>|bps=<bits_per_sample>|sample_rate=<rate>}*\n");
printf("\t\tmultiple options can be separated by a colon\n");
return false;
}
unique_ptr<char[]> arg_copy(new char[strlen(optarg) + 1]);
char *arg = arg_copy.get();
strcpy(arg, optarg);
char *item, *save_ptr, *tmp;
tmp = arg;
while ((item = strtok_r(tmp, ":", &save_ptr))) {
if (strncmp(item, "channels=", strlen("channels=")) == 0) {
audio_capture_channels = atoi(item + strlen("channels="));
if (audio_capture_channels < 1 || audio_capture_channels > MAX_AUDIO_CAPTURE_CHANNELS) {
log_msg(LOG_LEVEL_ERROR, "Invalid number of channels %d!\n", audio_capture_channels);
return false;
}
} else if (strncmp(item, "bps=", strlen("bps=")) == 0) {
int bps = atoi(item + strlen("bps="));
if (bps % 8 != 0 || (bps != 8 && bps != 16 && bps != 24 && bps != 32)) {
log_msg(LOG_LEVEL_ERROR, "Invalid bps %d!\n", bps);
log_msg(LOG_LEVEL_ERROR, "Supported values are 8, 16, 24, or 32 bits.\n");
return false;
}
audio_capture_bps = bps / 8;
} else if (strncmp(item, "sample_rate=", strlen("sample_rate=")) == 0) {
long long val = unit_evaluate(item + strlen("sample_rate="));
assert(val > 0 && val <= numeric_limits<decltype(audio_capture_sample_rate)>::max());
audio_capture_sample_rate = val;
} else {
log_msg(LOG_LEVEL_ERROR, "Unkonwn format for --audio-capture-format!\n");
return false;
}
tmp = NULL;
}
return true;
}
static bool parse_params(char *optarg)
{
if (optarg && strcmp(optarg, "help") == 0) {
puts("Params can be one or more (separated by comma) of following:");
print_param_doc();
return false;
}
char *item, *save_ptr;
while ((item = strtok_r(optarg, ",", &save_ptr))) {
char *key_cstr = item;
if (strchr(item, '=')) {
char *val_cstr = strchr(item, '=') + 1;
*strchr(item, '=') = '\0';
commandline_params[key_cstr] = val_cstr;
} else {
commandline_params[key_cstr] = string();
}
if (!validate_param(key_cstr)) {
log_msg(LOG_LEVEL_ERROR, "Unknown parameter: %s\n", key_cstr);
log_msg(LOG_LEVEL_INFO, "Type '%s --param help' for list.\n", uv_argv[0]);
return false;
}
optarg = NULL;
}
return true;
}
int main(int argc, char *argv[])
{
#if defined HAVE_SCHED_SETSCHEDULER && defined USE_RT
struct sched_param sp;
#endif
// NULL terminated array of capture devices
struct vidcap_params *vidcap_params_head = vidcap_params_allocate();
struct vidcap_params *vidcap_params_tail = vidcap_params_head;
const char *display_cfg = "";
const char *audio_recv = "none";
const char *audio_send = "none";
const char *requested_video_fec = "none";
const char *requested_audio_fec = DEFAULT_AUDIO_FEC;
char *audio_channel_map = NULL;
const char *audio_scale = "mixauto";
int port_base = PORT_BASE;
int video_rx_port = -1, video_tx_port = -1, audio_rx_port = -1, audio_tx_port = -1;
bool echo_cancellation = false;
bool should_export = false;
char *export_opts = NULL;
int control_port = 0;
int connection_type = 0;
struct control_state *control = NULL;
const char *audio_host = NULL;
enum video_mode decoder_mode = VIDEO_NORMAL;
const char *requested_compression = nullptr;
int force_ip_version = 0;
struct state_uv uv{};
int ch;
const char *audio_codec = nullptr;
pthread_t receiver_thread_id,
capture_thread_id;
bool receiver_thread_started = false,
capture_thread_started = false;
unsigned display_flags = 0;
int ret;
struct vidcap_params *audio_cap_dev;
const char *requested_mcast_if = NULL;
unsigned requested_mtu = 0;
const char *postprocess = NULL;
const char *requested_display = "none";
const char *requested_receiver = "localhost";
const char *requested_encryption = NULL;
struct exporter *exporter = NULL;
long long int bitrate = RATE_DEFAULT;
int audio_rxtx_mode = 0, video_rxtx_mode = 0;
const chrono::steady_clock::time_point start_time(chrono::steady_clock::now());
keyboard_control kc{};
bool print_capabilities_req = false;
bool start_paused = false;
static struct option getopt_options[] = {
{"display", required_argument, 0, 'd'},
{"capture", required_argument, 0, 't'},
{"mtu", required_argument, 0, 'm'},
{"mode", required_argument, 0, 'M'},
{"version", no_argument, 0, 'v'},
{"compress", required_argument, 0, 'c'},
{"receive", required_argument, 0, 'r'},
{"send", required_argument, 0, 's'},
{"help", no_argument, 0, 'h'},
{"fec", required_argument, 0, 'f'},
{"port", required_argument, 0, 'P'},
{"limit-bitrate", required_argument, 0, 'l'},
{"audio-channel-map", required_argument, 0, OPT_AUDIO_CHANNEL_MAP},
{"audio-scale", required_argument, 0, OPT_AUDIO_SCALE},
{"audio-capture-channels", required_argument, 0, OPT_AUDIO_CAPTURE_CHANNELS},
{"audio-capture-format", required_argument, 0, OPT_AUDIO_CAPTURE_FORMAT},
{"echo-cancellation", no_argument, 0, OPT_ECHO_CANCELLATION},
{"fullhelp", no_argument, 0, OPT_FULLHELP},
{"cuda-device", required_argument, 0, OPT_CUDA_DEVICE},
{"mcast-if", required_argument, 0, OPT_MCAST_IF},
{"record", optional_argument, 0, OPT_EXPORT},
{"playback", required_argument, 0, OPT_IMPORT},
{"audio-host", required_argument, 0, 'A'},
{"audio-codec", required_argument, 0, OPT_AUDIO_CODEC},
{"capture-filter", required_argument, 0, OPT_CAPTURE_FILTER},
{"control-port", required_argument, 0, OPT_CONTROL_PORT},
{"encryption", required_argument, 0, OPT_ENCRYPTION},
{"verbose", optional_argument, 0, OPT_VERBOSE},
{"window-title", required_argument, 0, OPT_WINDOW_TITLE},
{"capabilities", no_argument, 0, OPT_CAPABILITIES},
{"audio-delay", required_argument, 0, OPT_AUDIO_DELAY},
{"list-modules", no_argument, 0, OPT_LIST_MODULES},
{"start-paused", no_argument, 0, OPT_START_PAUSED},
{"audio-protocol", required_argument, 0, OPT_AUDIO_PROTOCOL},
{"video-protocol", required_argument, 0, OPT_VIDEO_PROTOCOL},
{"protocol", required_argument, 0, OPT_PROTOCOL},
{"rtsp-server", optional_argument, 0, 'H'},
{"param", required_argument, 0, OPT_PARAM},
{0, 0, 0, 0}
};
const char optstring[] = "d:t:m:r:s:v46c:hM:p:f:P:l:A:";
const char *audio_protocol = "ultragrid_rtp";
const char *audio_protocol_opts = "";
const char *video_protocol = "ultragrid_rtp";
const char *video_protocol_opts = "";
// First we need to set verbosity level prior to everything else.
// common_preinit() uses the verbosity level.
while ((ch =
getopt_long(argc, argv, optstring, getopt_options,
NULL)) != -1) {
switch (ch) {
case OPT_VERBOSE:
if (optarg) {
log_level = atoi(optarg);
} else {
log_level = LOG_LEVEL_VERBOSE;
}
break;
default:
break;
}
}
optind = 1;
uv_state = &uv;
if (!common_preinit(argc, argv)) {
log_msg(LOG_LEVEL_FATAL, "common_preinit() failed!\n");
return EXIT_FAILURE;
}
vidcap_params_set_device(vidcap_params_head, "none");
while ((ch =
getopt_long(argc, argv, optstring, getopt_options,
NULL)) != -1) {
switch (ch) {
case 'd':
if (!strcmp(optarg, "help")) {
list_video_display_devices();
return 0;
}
requested_display = optarg;
if(strchr(optarg, ':')) {
char *delim = strchr(optarg, ':');
*delim = '\0';
display_cfg = delim + 1;
}
break;
case 't':
if (!strcmp(optarg, "help")) {
list_video_capture_devices();
return 0;
}
vidcap_params_set_device(vidcap_params_tail, optarg);
vidcap_params_tail = vidcap_params_allocate_next(vidcap_params_tail);
break;
case 'm':
requested_mtu = atoi(optarg);
if (requested_mtu < 576 && optarg[strlen(optarg) - 1] != '!') {
log_msg(LOG_LEVEL_WARNING, "MTU %1$u seems to be too low, use \"%1$u!\" to force.\n", requested_mtu);
return EXIT_FAIL_USAGE;
}
break;
case 'M':
decoder_mode = get_video_mode_from_str(optarg);
if (decoder_mode == VIDEO_UNKNOWN) {
return strcasecmp(optarg, "help") == 0 ? EXIT_SUCCESS : EXIT_FAIL_USAGE;
}
break;
case 'p':
postprocess = optarg;
break;
case 'v':
print_version();
printf("\n");
print_configuration();
return EXIT_SUCCESS;
case 'c':
requested_compression = optarg;
break;
case 'H':
log_msg(LOG_LEVEL_WARNING, "Option \"--rtsp-server[=args]\" "
"is deprecated and will be removed in future.\n"
"Please use \"--video-protocol rtsp[:args]\"instead.\n");
video_protocol = "rtsp";
video_protocol_opts = optarg ? optarg : "";
break;
case OPT_AUDIO_PROTOCOL:
audio_protocol = optarg;
if (strchr(optarg, ':')) {
char *delim = strchr(optarg, ':');
*delim = '\0';
audio_protocol_opts = delim + 1;
}
if (strcmp(audio_protocol, "help") == 0) {
printf("Audio protocol can be one of: " AUDIO_PROTOCOLS "\n");
return EXIT_SUCCESS;
}
break;
case OPT_VIDEO_PROTOCOL:
video_protocol = optarg;
if (strchr(optarg, ':')) {
char *delim = strchr(optarg, ':');
*delim = '\0';
video_protocol_opts = delim + 1;
}
break;
case OPT_PROTOCOL:
if (strcmp(optarg, "help") == 0) {
cout << "Specify a " << rang::style::bold << "common" << rang::style::reset << " protocol for both audio and video.\n";
cout << "Audio protocol can be one of: " << rang::style::bold << AUDIO_PROTOCOLS "\n" << rang::style::reset;
video_rxtx::create("help", {});
return EXIT_SUCCESS;
}
audio_protocol = video_protocol = optarg;
if (strchr(optarg, ':')) {
char *delim = strchr(optarg, ':');
*delim = '\0';
audio_protocol_opts = video_protocol_opts = delim + 1;
}
break;
case 'r':
audio_recv = optarg;
break;
case 's':
audio_send = optarg;
break;
case 'f':
if(strlen(optarg) > 2 && optarg[1] == ':' &&
(toupper(optarg[0]) == 'A' || toupper(optarg[0]) == 'V')) {
if(toupper(optarg[0]) == 'A') {
requested_audio_fec = optarg + 2;
} else {
requested_video_fec = optarg + 2;
}
} else {
// there should be setting for both audio and video
// but we conservativelly expect that the user wants
// only vieo and let audio default until explicitly
// stated otehrwise
requested_video_fec = optarg;
}
break;
case 'h':
usage(uv_argv[0], false);
return 0;
case 'P':
if(strchr(optarg, ':')) {
char *save_ptr = NULL;
char *tok;
video_rx_port = atoi(strtok_r(optarg, ":", &save_ptr));
video_tx_port = atoi(strtok_r(NULL, ":", &save_ptr));
if((tok = strtok_r(NULL, ":", &save_ptr))) {
audio_rx_port = atoi(tok);
if((tok = strtok_r(NULL, ":", &save_ptr))) {
audio_tx_port = atoi(tok);
} else {
usage(uv_argv[0]);
return EXIT_FAIL_USAGE;
}
}
} else {
port_base = atoi(optarg);
}
break;
case 'l':
if(strcmp(optarg, "unlimited") == 0) {
bitrate = RATE_UNLIMITED;
} else if(strcmp(optarg, "auto") == 0) {
bitrate = RATE_AUTO;
} else {
bool force = false;
if (optarg[strlen(optarg) - 1] == '!') {
force = true;
optarg[strlen(optarg) - 1] = '\0';
}
bitrate = unit_evaluate(optarg);
if (bitrate <= 0) {
log_msg(LOG_LEVEL_ERROR, "Invalid bitrate %s!\n", optarg);
return EXIT_FAIL_USAGE;
}
if (bitrate < 10 * 1000 * 1000 && !force) {
log_msg(LOG_LEVEL_WARNING, "Bitrate %lld bps seems to be too low, use \"-l %s!\" to force if this is not a mistake.\n", bitrate, optarg);
return EXIT_FAIL_USAGE;
}
}
break;
case '4':
force_ip_version = 4;
break;
case '6':
force_ip_version = 6;
break;
case OPT_AUDIO_CHANNEL_MAP:
audio_channel_map = optarg;
break;
case OPT_AUDIO_SCALE:
audio_scale = optarg;
break;
case OPT_AUDIO_CAPTURE_CHANNELS:
log_msg(LOG_LEVEL_WARNING, "Parameter --audio-capture-channels is deprecated. "
"Use \"--audio-capture-format channels=<count>\" instead.\n");
audio_capture_channels = atoi(optarg);
if (audio_capture_channels < 1 || audio_capture_channels > MAX_AUDIO_CAPTURE_CHANNELS) {
log_msg(LOG_LEVEL_ERROR, "Invalid number of channels %d!\n", audio_capture_channels);
return EXIT_FAIL_USAGE;
}
break;
case OPT_AUDIO_CAPTURE_FORMAT:
if (!parse_audio_capture_format(optarg)) {
return EXIT_FAIL_USAGE;
}
break;
case OPT_ECHO_CANCELLATION:
echo_cancellation = true;
break;
case OPT_FULLHELP:
usage(uv_argv[0], true);
return EXIT_SUCCESS;
case OPT_CUDA_DEVICE:
#ifdef HAVE_JPEG
if(strcmp("help", optarg) == 0) {
struct compress_state *compression;
int ret = compress_init(&uv.root_module, "JPEG:list_devices", &compression);
if(ret >= 0) {
if(ret == 0) {
module_done(CAST_MODULE(compression));
}
return EXIT_SUCCESS;
} else {
return EXIT_FAILURE;
}
} else {
char *item, *save_ptr = NULL;
unsigned int i = 0;
while((item = strtok_r(optarg, ",", &save_ptr))) {
if(i >= MAX_CUDA_DEVICES) {
fprintf(stderr, "Maximal number of CUDA device exceeded.\n");
return EXIT_FAILURE;
}
cuda_devices[i] = atoi(item);
optarg = NULL;
++i;
}
cuda_devices_count = i;
}
break;
#else
fprintf(stderr, "CUDA support is not enabled!\n");
return EXIT_FAIL_USAGE;
#endif // HAVE_CUDA
case OPT_MCAST_IF:
requested_mcast_if = optarg;
break;
case 'A':
audio_host = optarg;
break;
case OPT_EXPORT:
should_export = true;
export_opts = optarg;
break;
case OPT_IMPORT:
if (import_has_audio(optarg)) {
audio_send = "embedded";
}
{
char dev_string[1024];
snprintf(dev_string, sizeof(dev_string), "import:%s", optarg);
vidcap_params_set_device(vidcap_params_tail, dev_string);
vidcap_params_tail = vidcap_params_allocate_next(vidcap_params_tail);
}
break;
case OPT_AUDIO_CODEC:
if(strcmp(optarg, "help") == 0) {
list_audio_codecs();
return EXIT_SUCCESS;
}
audio_codec = optarg;
if (!check_audio_codec(optarg)) {
return EXIT_FAIL_USAGE;
}
break;
case OPT_CAPTURE_FILTER:
vidcap_params_set_capture_filter(vidcap_params_tail, optarg);
break;
case OPT_ENCRYPTION:
requested_encryption = optarg;
break;
case OPT_CONTROL_PORT:
if (strchr(optarg, ':')) {
char *save_ptr = NULL;
char *tok;
control_port = atoi(strtok_r(optarg, ":", &save_ptr));
connection_type = atoi(strtok_r(NULL, ":", &save_ptr));
if(connection_type < 0 || connection_type > 1){
usage(uv_argv[0]);
return EXIT_FAIL_USAGE;
}
if ((tok = strtok_r(NULL, ":", &save_ptr))) {
usage(uv_argv[0]);
return EXIT_FAIL_USAGE;
}
} else {
control_port = atoi(optarg);
connection_type = 0;
}
break;
case OPT_VERBOSE:
break; // already handled earlier
case OPT_WINDOW_TITLE:
log_msg(LOG_LEVEL_WARNING, "Deprecated option used, please use "
"--param window-title=<title>\n");
commandline_params["window-title"] = optarg;
break;
case OPT_CAPABILITIES:
print_capabilities_req = true;
break;
case OPT_AUDIO_DELAY:
audio_offset = max(atoi(optarg), 0);
video_offset = atoi(optarg) < 0 ? abs(atoi(optarg)) : 0;
break;
case OPT_LIST_MODULES:
list_all_modules();
return EXIT_SUCCESS;
case OPT_START_PAUSED:
start_paused = true;
break;
case OPT_PARAM:
if (!parse_params(optarg)) {
return EXIT_SUCCESS;
}
break;
case '?':
default:
usage(uv_argv[0]);
return EXIT_FAIL_USAGE;
}
}
argc -= optind;
argv += optind;
if (argc > 1) {
log_msg(LOG_LEVEL_ERROR, "Multiple receivers given!\n");
usage(uv_argv[0]);
return EXIT_FAIL_USAGE;
}
if (argc > 0) {
requested_receiver = argv[0];
}
if (!audio_host) {
audio_host = requested_receiver;
}
if (!set_output_buffering()) {
log_msg(LOG_LEVEL_WARNING, "Cannot set console output buffering!\n");
}
// default values for different RXTX protocols
if (strcmp(video_protocol, "rtsp") == 0 || strcmp(video_protocol, "sdp") == 0) {
if (audio_codec == nullptr) {
audio_codec = "OPUS:sample_rate=48000";
}
if (requested_compression == nullptr) {
requested_compression = "none"; // will be set later
}
if (force_ip_version == 0 && strcmp(video_protocol, "rtsp") == 0) {
force_ip_version = 4;
}
} else {
if (requested_compression == nullptr) {
requested_compression = DEFAULT_VIDEO_COMPRESSION;
}
if (audio_codec == nullptr) {
audio_codec = DEFAULT_AUDIO_CODEC;
}
}
if (strcmp("none", audio_recv) != 0) {
audio_rxtx_mode |= MODE_RECEIVER;
}
if (strcmp("none", audio_send) != 0) {
audio_rxtx_mode |= MODE_SENDER;
}
if (strcmp("none", requested_display) != 0) {
video_rxtx_mode |= MODE_RECEIVER;
}
if (strcmp("none", vidcap_params_get_driver(vidcap_params_head)) != 0) {
video_rxtx_mode |= MODE_SENDER;
}
if (video_rx_port == -1) {
if ((video_rxtx_mode & MODE_RECEIVER) == 0) {
// do not occupy recv port if we are not receiving (note that this disables communication with
// our receiver, because RTCP ports are changed as well)
video_rx_port = 0;
} else {
video_rx_port = port_base;
}
}
if (video_tx_port == -1) {
if ((video_rxtx_mode & MODE_SENDER) == 0) {
video_tx_port = 0; // does not matter, we are receiver
} else {
video_tx_port = port_base;
}
}
if (audio_rx_port == -1) {
if ((audio_rxtx_mode & MODE_RECEIVER) == 0) {
// do not occupy recv port if we are not receiving (note that this disables communication with
// our receiver, because RTCP ports are changed as well)
audio_rx_port = 0;
} else {
audio_rx_port = video_rx_port ? video_rx_port + 2 : port_base + 2;
}
}
if (audio_tx_port == -1) {
if ((audio_rxtx_mode & MODE_SENDER) == 0) {
audio_tx_port = 0;
} else {
audio_tx_port = video_tx_port ? video_tx_port + 2 : port_base + 2;
}
}
// If we are sure that this UltraGrid is sending to itself we can optimize some parameters
// (aka "-m 9000 -l unlimited"). If ports weren't equal it is possibile that we are sending
// to a reflector, thats why we require equal ports (we are a receiver as well).
if (is_host_loopback(requested_receiver) && video_rx_port == video_tx_port &&
audio_rx_port == audio_tx_port) {
requested_mtu = requested_mtu == 0 ? requested_mtu = min(RTP_MAX_MTU, 65535) : requested_mtu;
bitrate = bitrate == RATE_DEFAULT ? RATE_UNLIMITED : bitrate;
} else {
requested_mtu = requested_mtu == 0 ? 1500 : requested_mtu;
bitrate = bitrate == RATE_DEFAULT ? RATE_AUTO : bitrate;
}
if((strcmp("none", audio_send) != 0 || strcmp("none", audio_recv) != 0) && strcmp(video_protocol, "rtsp") == 0){
//TODO: to implement a high level rxtx struct to manage different standards (i.e.:H264_STD, VP8_STD,...)
if (strcmp(audio_protocol, "rtsp") != 0) {
log_msg(LOG_LEVEL_WARNING, "Using RTSP for video but not for audio is not recommended. Consider adding '--audio-protocol rtsp'.\n");
}
}
if (strcmp(audio_protocol, "rtsp") == 0 && strcmp(video_protocol, "rtsp") != 0) {
log_msg(LOG_LEVEL_WARNING, "Using RTSP for audio but not for video is not recommended and might not work.\n");
}
print_version();
printf("\n");
auto fmt = rang::style::bold;
auto fmt_rst = rang::style::reset;
cout << fmt << "Display device : " << fmt_rst << requested_display << "\n";
cout << fmt << "Capture device : " << fmt_rst << vidcap_params_get_driver(vidcap_params_head) << "\n";
cout << fmt << "Audio capture : " << fmt_rst << audio_send << "\n";
cout << fmt << "Audio playback : " << fmt_rst << audio_recv << "\n";
cout << fmt << "MTU : " << fmt_rst << requested_mtu << " B\n";
cout << fmt << "Video compression: " << fmt_rst << requested_compression << "\n";
cout << fmt << "Audio codec : " << fmt_rst << get_name_to_audio_codec(get_audio_codec(audio_codec)) << "\n";
cout << fmt << "Network protocol : " << fmt_rst << video_rxtx::get_long_name(video_protocol) << "\n";
cout << fmt << "Audio FEC : " << fmt_rst << requested_audio_fec << "\n";
cout << fmt << "Video FEC : " << fmt_rst << requested_video_fec << "\n";
cout << "\n";
exporter = export_init(&uv.root_module, export_opts, should_export);
if (!exporter) {
log_msg(LOG_LEVEL_ERROR, "Export initialization failed.\n");
return EXIT_FAILURE;
}
if (control_init(control_port, connection_type, &control, &uv.root_module) != 0) {
fprintf(stderr, "Error: Unable to initialize remote control!\n");
return EXIT_FAIL_CONTROL_SOCK;
}
uv.audio = audio_cfg_init (&uv.root_module, audio_host, audio_rx_port,
audio_tx_port, audio_send, audio_recv,
audio_protocol, audio_protocol_opts,
requested_audio_fec, requested_encryption,
audio_channel_map,
audio_scale, echo_cancellation, force_ip_version, requested_mcast_if,
audio_codec, bitrate, &audio_offset, &start_time,
requested_mtu, exporter);
if(!uv.audio) {
exit_uv(EXIT_FAIL_AUDIO);
goto cleanup;
}
display_flags |= audio_get_display_flags(uv.audio);
// Display initialization should be prior to modules that may use graphic card (eg. GLSL) in order
// to initalize shared resource (X display) first
ret =
initialize_video_display(&uv.root_module, requested_display, display_cfg, display_flags, postprocess, &uv.display_device);
if (ret < 0) {
printf("Unable to open display device: %s\n",
requested_display);
exit_uv(EXIT_FAIL_DISPLAY);
goto cleanup;
} else if(ret > 0) {
exit_uv(EXIT_SUCCESS);
goto cleanup;
}
printf("Display initialized-%s\n", requested_display);
/* Pass embedded/analog/AESEBU flags to selected vidcap
* device. */
if (audio_capture_get_vidcap_flags(audio_send)) {
audio_cap_dev = vidcap_params_get_nth(
vidcap_params_head,
audio_capture_get_vidcap_index(audio_send));
if (audio_cap_dev != NULL) {
unsigned int orig_flags =
vidcap_params_get_flags(audio_cap_dev);
vidcap_params_set_flags(audio_cap_dev, orig_flags
| audio_capture_get_vidcap_flags(audio_send));
} else {
fprintf(stderr, "Entered index for non-existing vidcap (audio).\n");
exit_uv(EXIT_FAIL_CAPTURE);
goto cleanup;
}
}
ret = initialize_video_capture(&uv.root_module, vidcap_params_head, &uv.capture_device);
if (ret < 0) {
printf("Unable to open capture device: %s\n",
vidcap_params_get_driver(vidcap_params_head));
exit_uv(EXIT_FAIL_CAPTURE);
goto cleanup;
} else if(ret > 0) {
exit_uv(EXIT_SUCCESS);
goto cleanup;
}
printf("Video capture initialized-%s\n", vidcap_params_get_driver(vidcap_params_head));
signal(SIGINT, signal_handler);
signal(SIGTERM, signal_handler);
#ifndef WIN32
signal(SIGHUP, signal_handler);
#endif
signal(SIGABRT, crash_signal_handler);
signal(SIGSEGV, crash_signal_handler);
#ifdef USE_RT
#ifdef HAVE_SCHED_SETSCHEDULER
sp.sched_priority = sched_get_priority_max(SCHED_FIFO);
if (sched_setscheduler(0, SCHED_FIFO, &sp) != 0) {
printf("WARNING: Unable to set real-time scheduling\n");
}
#else
printf("WARNING: System does not support real-time scheduling\n");
#endif /* HAVE_SCHED_SETSCHEDULER */
#endif /* USE_RT */
control_start(control);
kc.start(&uv.root_module);
try {
map<string, param_u> params;
// common
params["parent"].ptr = &uv.root_module;
params["exporter"].ptr = exporter;
params["compression"].ptr = (void *) requested_compression;
params["rxtx_mode"].i = video_rxtx_mode;
params["paused"].b = start_paused;
// iHDTV
params["argc"].i = argc;
params["argv"].ptr = argv;
params["capture_device"].ptr = NULL;
params["display_device"].ptr = NULL;
if (video_rxtx_mode & MODE_SENDER)
params["capture_device"].ptr = uv.capture_device;
if (video_rxtx_mode & MODE_RECEIVER)
params["display_device"].ptr = uv.display_device;
//RTP
params["mtu"].i = requested_mtu;
params["receiver"].ptr = (void *) requested_receiver;
params["rx_port"].i = video_rx_port;
params["tx_port"].i = video_tx_port;
params["force_ip_version"].i = force_ip_version;
params["mcast_if"].ptr = (void *) requested_mcast_if;
params["mtu"].i = requested_mtu;
params["fec"].ptr = (void *) requested_video_fec;
params["encryption"].ptr = (void *) requested_encryption;
params["bitrate"].ll = bitrate;
params["start_time"].ptr = (void *) &start_time;
params["video_delay"].ptr = (void *) &video_offset;
// UltraGrid RTP
params["decoder_mode"].l = (long) decoder_mode;
params["display_device"].ptr = uv.display_device;
// SAGE + RTSP
params["opts"].ptr = (void *) video_protocol_opts;
// RTSP/SDP
params["audio_codec"].l = get_audio_codec(audio_codec);
params["audio_sample_rate"].i = get_audio_codec_sample_rate(audio_codec) ? get_audio_codec_sample_rate(audio_codec) : 48000;
params["audio_channels"].i = audio_capture_channels;
params["audio_bps"].i = 2;
params["a_rx_port"].i = audio_rx_port;
params["a_tx_port"].i = audio_tx_port;
if (strcmp(video_protocol, "rtsp") == 0) {
rtps_types_t avType;
if(strcmp("none", vidcap_params_get_driver(vidcap_params_head)) != 0 && (strcmp("none",audio_send) != 0)) avType = av; //AVStream
else if((strcmp("none",audio_send) != 0)) avType = audio; //AStream
else if(strcmp("none", vidcap_params_get_driver(vidcap_params_head))) avType = video; //VStream
else {
printf("[RTSP SERVER CHECK] no stream type... check capture devices input...\n");
avType = none;
}
params["avType"].l = (long) avType;
}
uv.state_video_rxtx = video_rxtx::create(video_protocol, params);
if (!uv.state_video_rxtx) {
if (strcmp(video_protocol, "help") != 0) {
throw string("Requested RX/TX cannot be created (missing library?)");
} else {
throw 0;
}
}
if (video_rxtx_mode & MODE_RECEIVER) {
if (!uv.state_video_rxtx->supports_receiving()) {
fprintf(stderr, "Selected RX/TX mode doesn't support receiving.\n");
exit_uv(EXIT_FAILURE);
goto cleanup;
}
// init module here so as it is capable of receiving messages
if (pthread_create
(&receiver_thread_id, NULL, video_rxtx::receiver_thread,
(void *) uv.state_video_rxtx) != 0) {
perror("Unable to create display thread!\n");
exit_uv(EXIT_FAILURE);
goto cleanup;
} else {
receiver_thread_started = true;
}
}
if (video_rxtx_mode & MODE_SENDER) {
if (pthread_create
(&capture_thread_id, NULL, capture_thread,
(void *) &uv.root_module) != 0) {
perror("Unable to create capture thread!\n");
exit_uv(EXIT_FAILURE);
goto cleanup;
} else {
capture_thread_started = true;
}
}
if(audio_get_display_flags(uv.audio)) {
audio_register_display_callbacks(uv.audio,
uv.display_device,
(void (*)(void *, struct audio_frame *)) display_put_audio_frame,
(int (*)(void *, int, int, int)) display_reconfigure_audio,
(int (*)(void *, int, void *, size_t *)) display_get_property);
}
audio_start(uv.audio);
// This has to be run after start of capture thread since it may request
// captured video format information.
if (print_capabilities_req) {
print_capabilities(&uv.root_module, strcmp("none", vidcap_params_get_driver(vidcap_params_head)) != 0);
exit_uv(EXIT_SUCCESS);
goto cleanup;
}
if (strcmp("none", requested_display) != 0) {
if (!mainloop) {
set_thread_name("display");
display_run(uv.display_device);
} else {
throw string("Cannot run display when "
"another mainloop registered!\n");
}
}
if (mainloop) {
set_thread_name("mainloop");
mainloop(mainloop_udata);
}
set_thread_name("main");
} catch (ug_runtime_error const &e) {
cerr << e.what() << endl;
exit_uv(e.get_code());
} catch (runtime_error const &e) {
cerr << e.what() << endl;
exit_uv(EXIT_FAILURE);
} catch (string const &str) {
cerr << str << endl;
exit_uv(EXIT_FAILURE);
} catch (int i) {
exit_uv(i);
}
cleanup:
if (strcmp("none", requested_display) != 0 &&
receiver_thread_started)
pthread_join(receiver_thread_id, NULL);
if (video_rxtx_mode & MODE_SENDER
&& capture_thread_started)
pthread_join(capture_thread_id, NULL);
/* also wait for audio threads */
audio_join(uv.audio);
if (uv.state_video_rxtx)
uv.state_video_rxtx->join();
if(uv.audio)
audio_done(uv.audio);
delete uv.state_video_rxtx;
if (uv.capture_device)
vidcap_done(uv.capture_device);
if (uv.display_device)
display_done(uv.display_device);
export_destroy(exporter);
kc.stop();
control_done(control);
while (vidcap_params_head) {
struct vidcap_params *next = vidcap_params_get_next(vidcap_params_head);
vidcap_params_free_struct(vidcap_params_head);
vidcap_params_head = next;
}
printf("Exit\n");
return exit_status;
}
/* vim: set expandtab sw=8: */
Reference in New Issue View Git Blame Copy Permalink