Subversion Repositories Kolibri OS

Rev

Go to most recent revision | Blame | Last modification | View Log | RSS feed

  1. /*
  2.  * Copyright (c) 2013 Clément Bœsch
  3.  *
  4.  * This file is part of FFmpeg.
  5.  *
  6.  * FFmpeg is free software; you can redistribute it and/or
  7.  * modify it under the terms of the GNU Lesser General Public
  8.  * License as published by the Free Software Foundation; either
  9.  * version 2.1 of the License, or (at your option) any later version.
  10.  *
  11.  * FFmpeg is distributed in the hope that it will be useful,
  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14.  * Lesser General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU Lesser General Public
  17.  * License along with FFmpeg; if not, write to the Free Software
  18.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19.  */
  20.  
  21. #include <quvi/quvi.h>
  22.  
  23. #include "libavformat/avformat.h"
  24. #include "libavformat/internal.h"
  25. #include "libavutil/opt.h"
  26.  
  27. typedef struct {
  28.     const AVClass *class;
  29.     char *format;
  30.     AVFormatContext *fmtctx;
  31. } LibQuviContext;
  32.  
  33. #define OFFSET(x) offsetof(LibQuviContext, x)
  34. #define FLAGS AV_OPT_FLAG_DECODING_PARAM
  35. static const AVOption libquvi_options[] = {
  36.     { "format", "request specific format", OFFSET(format), AV_OPT_TYPE_STRING, {.str="best"}, .flags = FLAGS },
  37.     { NULL }
  38. };
  39.  
  40. static const AVClass libquvi_context_class = {
  41.     .class_name     = "libquvi",
  42.     .item_name      = av_default_item_name,
  43.     .option         = libquvi_options,
  44.     .version        = LIBAVUTIL_VERSION_INT,
  45. };
  46.  
  47. static int libquvi_close(AVFormatContext *s)
  48. {
  49.     LibQuviContext *qc = s->priv_data;
  50.     if (qc->fmtctx)
  51.         avformat_close_input(&qc->fmtctx);
  52.     return 0;
  53. }
  54.  
  55. static int libquvi_read_header(AVFormatContext *s)
  56. {
  57.     int i, ret;
  58.     quvi_t q;
  59.     quvi_media_t m;
  60.     QUVIcode rc;
  61.     LibQuviContext *qc = s->priv_data;
  62.     char *media_url, *pagetitle;
  63.  
  64.     rc = quvi_init(&q);
  65.     if (rc != QUVI_OK)
  66.         goto quvi_fail;
  67.  
  68.     quvi_setopt(q, QUVIOPT_FORMAT, qc->format);
  69.  
  70.     rc = quvi_parse(q, s->filename, &m);
  71.     if (rc != QUVI_OK)
  72.         goto quvi_fail;
  73.  
  74.     rc = quvi_getprop(m, QUVIPROP_MEDIAURL, &media_url);
  75.     if (rc != QUVI_OK)
  76.         goto quvi_fail;
  77.  
  78.     ret = avformat_open_input(&qc->fmtctx, media_url, NULL, NULL);
  79.     if (ret < 0)
  80.         goto end;
  81.  
  82.     rc = quvi_getprop(m, QUVIPROP_PAGETITLE, &pagetitle);
  83.     if (rc == QUVI_OK)
  84.         av_dict_set(&s->metadata, "title", pagetitle, 0);
  85.  
  86.     for (i = 0; i < qc->fmtctx->nb_streams; i++) {
  87.         AVStream *st = avformat_new_stream(s, NULL);
  88.         AVStream *ist = qc->fmtctx->streams[i];
  89.         if (!st) {
  90.             ret = AVERROR(ENOMEM);
  91.             goto end;
  92.         }
  93.         avpriv_set_pts_info(st, ist->pts_wrap_bits, ist->time_base.num, ist->time_base.den);
  94.         avcodec_copy_context(st->codec, qc->fmtctx->streams[i]->codec);
  95.     }
  96.  
  97.     return 0;
  98.  
  99. quvi_fail:
  100.     av_log(s, AV_LOG_ERROR, "%s\n", quvi_strerror(q, rc));
  101.     ret = AVERROR_EXTERNAL;
  102.  
  103. end:
  104.     quvi_parse_close(&m);
  105.     quvi_close(&q);
  106.     return ret;
  107. }
  108.  
  109. static int libquvi_read_packet(AVFormatContext *s, AVPacket *pkt)
  110. {
  111.     LibQuviContext *qc = s->priv_data;
  112.     return av_read_frame(qc->fmtctx, pkt);
  113. }
  114.  
  115. static int libquvi_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
  116. {
  117.     LibQuviContext *qc = s->priv_data;
  118.     return av_seek_frame(qc->fmtctx, stream_index, timestamp, flags);
  119. }
  120.  
  121. static int libquvi_probe(AVProbeData *p)
  122. {
  123.     int score;
  124.     quvi_t q;
  125.     QUVIcode rc;
  126.  
  127.     rc = quvi_init(&q);
  128.     if (rc != QUVI_OK)
  129.         return AVERROR(ENOMEM);
  130.     score = quvi_supported(q, (char *)p->filename) == QUVI_OK ? AVPROBE_SCORE_EXTENSION : 0;
  131.     quvi_close(&q);
  132.     return score;
  133. }
  134.  
  135. AVInputFormat ff_libquvi_demuxer = {
  136.     .name           = "libquvi",
  137.     .long_name      = NULL_IF_CONFIG_SMALL("libquvi demuxer"),
  138.     .priv_data_size = sizeof(LibQuviContext),
  139.     .read_probe     = libquvi_probe,
  140.     .read_header    = libquvi_read_header,
  141.     .read_packet    = libquvi_read_packet,
  142.     .read_close     = libquvi_close,
  143.     .read_seek      = libquvi_read_seek,
  144.     .priv_class     = &libquvi_context_class,
  145.     .flags          = AVFMT_NOFILE,
  146. };
  147.