yading@10: /*
yading@10:  * Copyright (c) 2012 Georg Lippitsch <georg.lippitsch@gmx.at>
yading@10:  *
yading@10:  * This file is part of FFmpeg.
yading@10:  *
yading@10:  * FFmpeg is free software; you can redistribute it and/or
yading@10:  * modify it under the terms of the GNU Lesser General Public
yading@10:  * License as published by the Free Software Foundation; either
yading@10:  * version 2.1 of the License, or (at your option) any later version.
yading@10:  *
yading@10:  * FFmpeg is distributed in the hope that it will be useful,
yading@10:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
yading@10:  * Lesser General Public License for more details.
yading@10:  *
yading@10:  * You should have received a copy of the GNU Lesser General Public
yading@10:  * License along with FFmpeg; if not, write to the Free Software
yading@10:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10:  */
yading@10: 
yading@10: /**
yading@10:  * @file
yading@10:  * libiec61883 interface
yading@10:  */
yading@10: 
yading@10: #include <sys/poll.h>
yading@10: #include <libraw1394/raw1394.h>
yading@10: #include <libavc1394/avc1394.h>
yading@10: #include <libavc1394/rom1394.h>
yading@10: #include <libiec61883/iec61883.h>
yading@10: #include "libavformat/dv.h"
yading@10: #include "libavformat/mpegts.h"
yading@10: #include "libavutil/opt.h"
yading@10: #include "avdevice.h"
yading@10: 
yading@10: #define THREADS HAVE_PTHREADS
yading@10: 
yading@10: #if THREADS
yading@10: #include <pthread.h>
yading@10: #endif
yading@10: 
yading@10: #define MOTDCT_SPEC_ID      0x00005068
yading@10: #define IEC61883_AUTO       0
yading@10: #define IEC61883_DV         1
yading@10: #define IEC61883_HDV        2
yading@10: 
yading@10: /**
yading@10:  * For DV, one packet corresponds exactly to one frame.
yading@10:  * For HDV, these are MPEG2 transport stream packets.
yading@10:  * The queue is implemented as linked list.
yading@10:  */
yading@10: typedef struct DVPacket {
yading@10:     uint8_t *buf;                       ///< actual buffer data
yading@10:     int len;                            ///< size of buffer allocated
yading@10:     struct DVPacket *next;              ///< next DVPacket
yading@10: } DVPacket;
yading@10: 
yading@10: struct iec61883_data {
yading@10:     AVClass *class;
yading@10:     raw1394handle_t raw1394;            ///< handle for libraw1394
yading@10:     iec61883_dv_fb_t iec61883_dv;       ///< handle for libiec61883 when used with DV
yading@10:     iec61883_mpeg2_t iec61883_mpeg2;    ///< handle for libiec61883 when used with HDV
yading@10: 
yading@10:     DVDemuxContext *dv_demux;           ///< generic DV muxing/demuxing context
yading@10:     MpegTSContext *mpeg_demux;          ///< generic HDV muxing/demuxing context
yading@10: 
yading@10:     DVPacket *queue_first;              ///< first element of packet queue
yading@10:     DVPacket *queue_last;               ///< last element of packet queue
yading@10: 
yading@10:     char *device_guid;                  ///< to select one of multiple DV devices
yading@10: 
yading@10:     int packets;                        ///< Number of packets queued
yading@10:     int max_packets;                    ///< Max. number of packets in queue
yading@10: 
yading@10:     int bandwidth;                      ///< returned by libiec61883
yading@10:     int channel;                        ///< returned by libiec61883
yading@10:     int input_port;                     ///< returned by libiec61883
yading@10:     int type;                           ///< Stream type, to distinguish DV/HDV
yading@10:     int node;                           ///< returned by libiec61883
yading@10:     int output_port;                    ///< returned by libiec61883
yading@10:     int thread_loop;                    ///< Condition for thread while-loop
yading@10:     int receiving;                      ///< True as soon data from device available
yading@10:     int receive_error;                  ///< Set in receive task in case of error
yading@10:     int eof;                            ///< True as soon as no more data available
yading@10: 
yading@10:     struct pollfd raw1394_poll;         ///< to poll for new data from libraw1394
yading@10: 
yading@10:     /** Parse function for DV/HDV differs, so this is set before packets arrive */
yading@10:     int (*parse_queue)(struct iec61883_data *dv, AVPacket *pkt);
yading@10: 
yading@10: #if THREADS
yading@10:     pthread_t receive_task_thread;
yading@10:     pthread_mutex_t mutex;
yading@10:     pthread_cond_t cond;
yading@10: #endif
yading@10: };
yading@10: 
yading@10: static int iec61883_callback(unsigned char *data, int length,
yading@10:                              int complete, void *callback_data)
yading@10: {
yading@10:     struct iec61883_data *dv = callback_data;
yading@10:     DVPacket *packet;
yading@10:     int ret;
yading@10: 
yading@10: #ifdef THREADS
yading@10:     pthread_mutex_lock(&dv->mutex);
yading@10: #endif
yading@10: 
yading@10:     if (dv->packets >= dv->max_packets) {
yading@10:         av_log(NULL, AV_LOG_ERROR, "DV packet queue overrun, dropping.\n");
yading@10:         ret = 0;
yading@10:         goto exit;
yading@10:     }
yading@10: 
yading@10:     packet = av_mallocz(sizeof(*packet));
yading@10:     if (!packet) {
yading@10:         ret = -1;
yading@10:         goto exit;
yading@10:     }
yading@10: 
yading@10:     packet->buf = av_malloc(length);
yading@10:     if (!packet->buf) {
yading@10:         ret = -1;
yading@10:         goto exit;
yading@10:     }
yading@10:     packet->len = length;
yading@10: 
yading@10:     memcpy(packet->buf, data, length);
yading@10: 
yading@10:     if (dv->queue_first) {
yading@10:         dv->queue_last->next = packet;
yading@10:         dv->queue_last = packet;
yading@10:     } else {
yading@10:         dv->queue_first = packet;
yading@10:         dv->queue_last = packet;
yading@10:     }
yading@10:     dv->packets++;
yading@10: 
yading@10:     ret = 0;
yading@10: 
yading@10: exit:
yading@10: #ifdef THREADS
yading@10:     pthread_cond_broadcast(&dv->cond);
yading@10:     pthread_mutex_unlock(&dv->mutex);
yading@10: #endif
yading@10:     return ret;
yading@10: }
yading@10: 
yading@10: static void *iec61883_receive_task(void *opaque)
yading@10: {
yading@10:     struct iec61883_data *dv = (struct iec61883_data *)opaque;
yading@10:     int result;
yading@10: 
yading@10: #ifdef THREADS
yading@10:     while (dv->thread_loop)
yading@10: #endif
yading@10:     {
yading@10:         while ((result = poll(&dv->raw1394_poll, 1, 200)) < 0) {
yading@10:             if (!(errno == EAGAIN || errno == EINTR)) {
yading@10:                 av_log(NULL, AV_LOG_ERROR, "Raw1394 poll error occurred.\n");
yading@10:                 dv->receive_error = AVERROR(EIO);
yading@10:                 return NULL;
yading@10:             }
yading@10:         }
yading@10:         if (result > 0 && ((dv->raw1394_poll.revents & POLLIN)
yading@10:                            || (dv->raw1394_poll.revents & POLLPRI))) {
yading@10:             dv->receiving = 1;
yading@10:             raw1394_loop_iterate(dv->raw1394);
yading@10:         } else if (dv->receiving) {
yading@10:             av_log(NULL, AV_LOG_ERROR, "No more input data available\n");
yading@10: #ifdef THREADS
yading@10:             pthread_mutex_lock(&dv->mutex);
yading@10:             dv->eof = 1;
yading@10:             pthread_cond_broadcast(&dv->cond);
yading@10:             pthread_mutex_unlock(&dv->mutex);
yading@10: #else
yading@10:             dv->eof = 1;
yading@10: #endif
yading@10:             return NULL;
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     return NULL;
yading@10: }
yading@10: 
yading@10: static int iec61883_parse_queue_dv(struct iec61883_data *dv, AVPacket *pkt)
yading@10: {
yading@10:     DVPacket *packet;
yading@10:     int size;
yading@10: 
yading@10:     size = avpriv_dv_get_packet(dv->dv_demux, pkt);
yading@10:     if (size > 0)
yading@10:         return size;
yading@10: 
yading@10:     packet = dv->queue_first;
yading@10:     if (!packet)
yading@10:         return -1;
yading@10: 
yading@10:     size = avpriv_dv_produce_packet(dv->dv_demux, pkt,
yading@10:                                     packet->buf, packet->len, -1);
yading@10:     pkt->destruct = av_destruct_packet;
yading@10:     dv->queue_first = packet->next;
yading@10:     av_free(packet);
yading@10:     dv->packets--;
yading@10: 
yading@10:     if (size > 0)
yading@10:         return size;
yading@10: 
yading@10:     return -1;
yading@10: }
yading@10: 
yading@10: static int iec61883_parse_queue_hdv(struct iec61883_data *dv, AVPacket *pkt)
yading@10: {
yading@10:     DVPacket *packet;
yading@10:     int size;
yading@10: 
yading@10:     while (dv->queue_first) {
yading@10:         packet = dv->queue_first;
yading@10:         size = ff_mpegts_parse_packet(dv->mpeg_demux, pkt, packet->buf,
yading@10:                                       packet->len);
yading@10:         dv->queue_first = packet->next;
yading@10:         av_free(packet->buf);
yading@10:         av_free(packet);
yading@10:         dv->packets--;
yading@10: 
yading@10:         if (size > 0)
yading@10:             return size;
yading@10:     }
yading@10: 
yading@10:     return -1;
yading@10: }
yading@10: 
yading@10: static int iec61883_read_header(AVFormatContext *context)
yading@10: {
yading@10:     struct iec61883_data *dv = context->priv_data;
yading@10:     struct raw1394_portinfo pinf[16];
yading@10:     rom1394_directory rom_dir;
yading@10:     char *endptr;
yading@10:     int inport;
yading@10:     int nb_ports;
yading@10:     int port = -1;
yading@10:     int response;
yading@10:     int i, j = 0;
yading@10:     uint64_t guid = 0;
yading@10: 
yading@10:     dv->input_port = -1;
yading@10:     dv->output_port = -1;
yading@10:     dv->channel = -1;
yading@10: 
yading@10:     dv->raw1394 = raw1394_new_handle();
yading@10: 
yading@10:     if (!dv->raw1394) {
yading@10:         av_log(context, AV_LOG_ERROR, "Failed to open IEEE1394 interface.\n");
yading@10:         return AVERROR(EIO);
yading@10:     }
yading@10: 
yading@10:     if ((nb_ports = raw1394_get_port_info(dv->raw1394, pinf, 16)) < 0) {
yading@10:         av_log(context, AV_LOG_ERROR, "Failed to get number of IEEE1394 ports.\n");
yading@10:         goto fail;
yading@10:     }
yading@10: 
yading@10:     inport = strtol(context->filename, &endptr, 10);
yading@10:     if (endptr != context->filename && *endptr == '\0') {
yading@10:         av_log(context, AV_LOG_INFO, "Selecting IEEE1394 port: %d\n", inport);
yading@10:         j = inport;
yading@10:         nb_ports = inport + 1;
yading@10:     } else if (strcmp(context->filename, "auto")) {
yading@10:         av_log(context, AV_LOG_ERROR, "Invalid input \"%s\", you should specify "
yading@10:                "\"auto\" for auto-detection, or the port number.\n", context->filename);
yading@10:         goto fail;
yading@10:     }
yading@10: 
yading@10:     if (dv->device_guid) {
yading@10:         if (sscanf(dv->device_guid, "%llx", (long long unsigned int *)&guid) != 1) {
yading@10:             av_log(context, AV_LOG_INFO, "Invalid dvguid parameter: %s\n",
yading@10:                    dv->device_guid);
yading@10:             goto fail;
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     for (; j < nb_ports && port==-1; ++j) {
yading@10:         raw1394_destroy_handle(dv->raw1394);
yading@10: 
yading@10:         if (!(dv->raw1394 = raw1394_new_handle_on_port(j))) {
yading@10:             av_log(context, AV_LOG_ERROR, "Failed setting IEEE1394 port.\n");
yading@10:             goto fail;
yading@10:         }
yading@10: 
yading@10:         for (i=0; i<raw1394_get_nodecount(dv->raw1394); ++i) {
yading@10: 
yading@10:             /* Select device explicitly by GUID */
yading@10: 
yading@10:             if (guid > 1) {
yading@10:                 if (guid == rom1394_get_guid(dv->raw1394, i)) {
yading@10:                     dv->node = i;
yading@10:                     port = j;
yading@10:                     break;
yading@10:                 }
yading@10:             } else {
yading@10: 
yading@10:                 /* Select first AV/C tape recorder player node */
yading@10: 
yading@10:                 if (rom1394_get_directory(dv->raw1394, i, &rom_dir) < 0)
yading@10:                     continue;
yading@10:                 if (((rom1394_get_node_type(&rom_dir) == ROM1394_NODE_TYPE_AVC) &&
yading@10:                      avc1394_check_subunit_type(dv->raw1394, i, AVC1394_SUBUNIT_TYPE_VCR)) ||
yading@10:                     (rom_dir.unit_spec_id == MOTDCT_SPEC_ID)) {
yading@10:                     rom1394_free_directory(&rom_dir);
yading@10:                     dv->node = i;
yading@10:                     port = j;
yading@10:                     break;
yading@10:                 }
yading@10:                 rom1394_free_directory(&rom_dir);
yading@10:             }
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     if (port == -1) {
yading@10:         av_log(context, AV_LOG_ERROR, "No AV/C devices found.\n");
yading@10:         goto fail;
yading@10:     }
yading@10: 
yading@10:     /* Provide bus sanity for multiple connections */
yading@10: 
yading@10:     iec61883_cmp_normalize_output(dv->raw1394, 0xffc0 | dv->node);
yading@10: 
yading@10:     /* Find out if device is DV or HDV */
yading@10: 
yading@10:     if (dv->type == IEC61883_AUTO) {
yading@10:         response = avc1394_transaction(dv->raw1394, dv->node,
yading@10:                                        AVC1394_CTYPE_STATUS |
yading@10:                                        AVC1394_SUBUNIT_TYPE_TAPE_RECORDER |
yading@10:                                        AVC1394_SUBUNIT_ID_0 |
yading@10:                                        AVC1394_VCR_COMMAND_OUTPUT_SIGNAL_MODE |
yading@10:                                        0xFF, 2);
yading@10:         response = AVC1394_GET_OPERAND0(response);
yading@10:         dv->type = (response == 0x10 || response == 0x90 || response == 0x1A || response == 0x9A) ?
yading@10:             IEC61883_HDV : IEC61883_DV;
yading@10:     }
yading@10: 
yading@10:     /* Connect to device, and do initialization */
yading@10: 
yading@10:     dv->channel = iec61883_cmp_connect(dv->raw1394, dv->node, &dv->output_port,
yading@10:                                        raw1394_get_local_id(dv->raw1394),
yading@10:                                        &dv->input_port, &dv->bandwidth);
yading@10: 
yading@10:     if (dv->channel < 0)
yading@10:         dv->channel = 63;
yading@10: 
yading@10:     if (!dv->max_packets)
yading@10:         dv->max_packets = 100;
yading@10: 
yading@10:     if (dv->type == IEC61883_HDV) {
yading@10: 
yading@10:         /* Init HDV receive */
yading@10: 
yading@10:         avformat_new_stream(context, NULL);
yading@10: 
yading@10:         dv->mpeg_demux = ff_mpegts_parse_open(context);
yading@10:         if (!dv->mpeg_demux)
yading@10:             goto fail;
yading@10: 
yading@10:         dv->parse_queue = iec61883_parse_queue_hdv;
yading@10: 
yading@10:         dv->iec61883_mpeg2 = iec61883_mpeg2_recv_init(dv->raw1394,
yading@10:                                                       (iec61883_mpeg2_recv_t)iec61883_callback,
yading@10:                                                       dv);
yading@10: 
yading@10:         dv->max_packets *= 766;
yading@10:     } else {
yading@10: 
yading@10:         /* Init DV receive */
yading@10: 
yading@10:         dv->dv_demux = avpriv_dv_init_demux(context);
yading@10:         if (!dv->dv_demux)
yading@10:             goto fail;
yading@10: 
yading@10:         dv->parse_queue = iec61883_parse_queue_dv;
yading@10: 
yading@10:         dv->iec61883_dv = iec61883_dv_fb_init(dv->raw1394, iec61883_callback, dv);
yading@10:     }
yading@10: 
yading@10:     dv->raw1394_poll.fd = raw1394_get_fd(dv->raw1394);
yading@10:     dv->raw1394_poll.events = POLLIN | POLLERR | POLLHUP | POLLPRI;
yading@10: 
yading@10:     /* Actually start receiving */
yading@10: 
yading@10:     if (dv->type == IEC61883_HDV)
yading@10:         iec61883_mpeg2_recv_start(dv->iec61883_mpeg2, dv->channel);
yading@10:     else
yading@10:         iec61883_dv_fb_start(dv->iec61883_dv, dv->channel);
yading@10: 
yading@10: #if THREADS
yading@10:     dv->thread_loop = 1;
yading@10:     pthread_mutex_init(&dv->mutex, NULL);
yading@10:     pthread_cond_init(&dv->cond, NULL);
yading@10:     pthread_create(&dv->receive_task_thread, NULL, iec61883_receive_task, dv);
yading@10: #endif
yading@10: 
yading@10:     return 0;
yading@10: 
yading@10: fail:
yading@10:     raw1394_destroy_handle(dv->raw1394);
yading@10:     return AVERROR(EIO);
yading@10: }
yading@10: 
yading@10: static int iec61883_read_packet(AVFormatContext *context, AVPacket *pkt)
yading@10: {
yading@10:     struct iec61883_data *dv = context->priv_data;
yading@10:     int size;
yading@10: 
yading@10:     /**
yading@10:      * Try to parse frames from queue
yading@10:      */
yading@10: 
yading@10: #ifdef THREADS
yading@10:     pthread_mutex_lock(&dv->mutex);
yading@10:     while ((size = dv->parse_queue(dv, pkt)) == -1)
yading@10:         if (!dv->eof)
yading@10:             pthread_cond_wait(&dv->cond, &dv->mutex);
yading@10:         else
yading@10:             break;
yading@10:     pthread_mutex_unlock(&dv->mutex);
yading@10: #else
yading@10:     int result;
yading@10:     while ((size = dv->parse_queue(dv, pkt)) == -1) {
yading@10:         iec61883_receive_task((void *)dv);
yading@10:         if (dv->receive_error)
yading@10:             return dv->receive_error;
yading@10:     }
yading@10: #endif
yading@10: 
yading@10:     return size;
yading@10: }
yading@10: 
yading@10: static int iec61883_close(AVFormatContext *context)
yading@10: {
yading@10:     struct iec61883_data *dv = context->priv_data;
yading@10: 
yading@10: #if THREADS
yading@10:     dv->thread_loop = 0;
yading@10:     pthread_join(dv->receive_task_thread, NULL);
yading@10:     pthread_cond_destroy(&dv->cond);
yading@10:     pthread_mutex_destroy(&dv->mutex);
yading@10: #endif
yading@10: 
yading@10:     if (dv->type == IEC61883_HDV) {
yading@10:         iec61883_mpeg2_recv_stop(dv->iec61883_mpeg2);
yading@10:         iec61883_mpeg2_close(dv->iec61883_mpeg2);
yading@10:         ff_mpegts_parse_close(dv->mpeg_demux);
yading@10:     } else {
yading@10:         iec61883_dv_fb_stop(dv->iec61883_dv);
yading@10:         iec61883_dv_fb_close(dv->iec61883_dv);
yading@10:     }
yading@10:     while (dv->queue_first) {
yading@10:         DVPacket *packet = dv->queue_first;
yading@10:         dv->queue_first = packet->next;
yading@10:         av_free(packet->buf);
yading@10:         av_free(packet);
yading@10:     }
yading@10: 
yading@10:     iec61883_cmp_disconnect(dv->raw1394, dv->node, dv->output_port,
yading@10:                             raw1394_get_local_id(dv->raw1394),
yading@10:                             dv->input_port, dv->channel, dv->bandwidth);
yading@10: 
yading@10:     raw1394_destroy_handle(dv->raw1394);
yading@10: 
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static const AVOption options[] = {
yading@10:     { "dvtype", "override autodetection of DV/HDV", offsetof(struct iec61883_data, type), AV_OPT_TYPE_INT, {.i64 = IEC61883_AUTO}, IEC61883_AUTO, IEC61883_HDV, AV_OPT_FLAG_DECODING_PARAM, "dvtype" },
yading@10:     { "auto",   "auto detect DV/HDV", 0, AV_OPT_TYPE_CONST, {.i64 = IEC61883_AUTO}, 0, 0, AV_OPT_FLAG_DECODING_PARAM, "dvtype" },
yading@10:     { "dv",     "force device being treated as DV device", 0, AV_OPT_TYPE_CONST, {.i64 = IEC61883_DV},   0, 0, AV_OPT_FLAG_DECODING_PARAM, "dvtype" },
yading@10:     { "hdv" ,   "force device being treated as HDV device", 0, AV_OPT_TYPE_CONST, {.i64 = IEC61883_HDV},  0, 0, AV_OPT_FLAG_DECODING_PARAM, "dvtype" },
yading@10:     { "dvbuffer", "set queue buffer size (in packets)", offsetof(struct iec61883_data, max_packets), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
yading@10:     { "dvguid", "select one of multiple DV devices by its GUID", offsetof(struct iec61883_data, device_guid), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, AV_OPT_FLAG_DECODING_PARAM },
yading@10:     { NULL },
yading@10: };
yading@10: 
yading@10: static const AVClass iec61883_class = {
yading@10:     .class_name = "iec61883 indev",
yading@10:     .item_name  = av_default_item_name,
yading@10:     .option     = options,
yading@10:     .version    = LIBAVUTIL_VERSION_INT,
yading@10: };
yading@10: 
yading@10: AVInputFormat ff_iec61883_demuxer = {
yading@10:     .name           = "iec61883",
yading@10:     .long_name      = NULL_IF_CONFIG_SMALL("libiec61883 (new DV1394) A/V input device"),
yading@10:     .priv_data_size = sizeof(struct iec61883_data),
yading@10:     .read_header    = iec61883_read_header,
yading@10:     .read_packet    = iec61883_read_packet,
yading@10:     .read_close     = iec61883_close,
yading@10:     .flags          = AVFMT_NOFILE,
yading@10:     .priv_class     = &iec61883_class,
yading@10: };