ext/smoothstreaming/gstmssmanifest.c

   1 /* GStreamer
   2  * Copyright (C) 2012 Smart TV Alliance
   3  *  Author: Thiago Sousa Santos <thiago.sousa.santos@collabora.com>, Collabora Ltd.
   4  *
   5  * gstmssmanifest.c:
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Library General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Library General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Library General Public
  18  * License along with this library; if not, write to the
  19  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  20  * Boston, MA 02111-1307, USA.
  21  */
  22
  23 #include <glib.h>
  24 #include <string.h>
  25 #include <libxml/parser.h>
  26 #include <libxml/tree.h>
  27
  28 #include "gstmssmanifest.h"
  29
  30 struct _GstMssManifestStream
  31 {
  32   xmlNodePtr xmlnode;
  33
  34   gint selectedQualityIndex;
  35 };
  36
  37 struct _GstMssManifest
  38 {
  39   xmlDocPtr xml;
  40   xmlNodePtr xmlrootnode;
  41
  42   GSList *streams;
  43 };
  44
  45 GstMssManifest *
  46 gst_mss_manifest_new (const GstBuffer * data)
  47 {
  48   GstMssManifest *manifest;
  49   xmlNodePtr root;
  50   xmlNodePtr nodeiter;
  51
  52   manifest = g_malloc0 (sizeof (GstMssManifest));
  53
  54   manifest->xml = xmlReadMemory ((const gchar *) GST_BUFFER_DATA (data),
  55       GST_BUFFER_SIZE (data), "manifest", NULL, 0);
  56   root = manifest->xmlrootnode = xmlDocGetRootElement (manifest->xml);
  57
  58   for (nodeiter = root->children; nodeiter; nodeiter = nodeiter->next) {
  59     if (nodeiter->type == XML_ELEMENT_NODE
  60         && (strcmp ((const char *) nodeiter->name, "StreamIndex") == 0)) {
  61       GstMssManifestStream *stream = g_new0 (GstMssManifestStream, 1);
  62
  63       manifest->streams = g_slist_append (manifest->streams, stream);
  64       stream->xmlnode = nodeiter;
  65     }
  66   }
  67
  68   return manifest;
  69 }
  70
  71 void
  72 gst_mss_manifest_free (GstMssManifest * manifest)
  73 {
  74   g_return_if_fail (manifest != NULL);
  75
  76   g_slist_free_full (manifest->streams, g_free);
  77
  78   xmlFreeDoc (manifest->xml);
  79   g_free (manifest);
  80 }
  81
  82 GSList *
  83 gst_mss_manifest_get_streams (GstMssManifest * manifest)
  84 {
  85   return manifest->streams;
  86 }
  87
  88 GstMssManifestStreamType
  89 gst_mss_manifest_stream_get_type (GstMssManifestStream * stream)
  90 {
  91   gchar *prop = (gchar *) xmlGetProp (stream->xmlnode, (xmlChar *) "Type");
  92   GstMssManifestStreamType ret = MSS_STREAM_TYPE_UNKNOWN;
  93
  94   if (strcmp (prop, "video") == 0) {
  95     ret = MSS_STREAM_TYPE_VIDEO;
  96   } else if (strcmp (prop, "audio") == 0) {
  97     ret = MSS_STREAM_TYPE_AUDIO;
  98   }
  99   xmlFree (prop);
 100   return ret;
 101 }
 102
 103 static GstCaps *
 104 _gst_mss_manifest_stream_video_caps_from_fourcc (gchar * fourcc)
 105 {
 106   if (!fourcc)
 107     return NULL;
 108
 109   if (strcmp (fourcc, "H264") == 0) {
 110     return gst_caps_new_simple ("video/x-h264", NULL);
 111   }
 112   return NULL;
 113 }
 114
 115 static GstCaps *
 116 _gst_mss_manifest_stream_audio_caps_from_fourcc (gchar * fourcc)
 117 {
 118   if (!fourcc)
 119     return NULL;
 120
 121   if (strcmp (fourcc, "AACL") == 0) {
 122     return gst_caps_new_simple ("audio/mpeg", "mpegversion", G_TYPE_INT, 4,
 123         NULL);
 124   }
 125   return NULL;
 126 }
 127
 128 static GstCaps *
 129 _gst_mss_manifest_stream_video_caps_from_qualitylevel_xml (xmlNodePtr node)
 130 {
 131   GstCaps *caps;
 132   GstStructure *structure;
 133   gchar *fourcc = (gchar *) xmlGetProp (node, (xmlChar *) "FourCC");
 134   gchar *max_width = (gchar *) xmlGetProp (node, (xmlChar *) "MaxWidth");
 135   gchar *max_height = (gchar *) xmlGetProp (node, (xmlChar *) "MaxHeight");
 136   gchar *codec_data =
 137       (gchar *) xmlGetProp (node, (xmlChar *) "CodecPrivateData");
 138
 139   caps = _gst_mss_manifest_stream_video_caps_from_fourcc (fourcc);
 140   if (!caps)
 141     goto end;
 142
 143   structure = gst_caps_get_structure (caps, 0);
 144
 145   if (max_width)
 146     gst_structure_set (structure, "width", G_TYPE_INT, atoi (max_width), NULL);
 147   if (max_height)
 148     gst_structure_set (structure, "height", G_TYPE_INT, atoi (max_height),
 149         NULL);
 150
 151   if (codec_data) {
 152     GValue *value = g_new0 (GValue, 1);
 153     g_value_init (value, GST_TYPE_BUFFER);
 154     gst_value_deserialize (value, (gchar *) codec_data);
 155     gst_structure_take_value (structure, "codec_data", value);
 156   }
 157
 158 end:
 159   g_free (fourcc);
 160   g_free (max_width);
 161   g_free (max_height);
 162   g_free (codec_data);
 163
 164   return caps;
 165 }
 166
 167 static GstCaps *
 168 _gst_mss_manifest_stream_audio_caps_from_qualitylevel_xml (xmlNodePtr node)
 169 {
 170   GstCaps *caps;
 171   GstStructure *structure;
 172   gchar *fourcc = (gchar *) xmlGetProp (node, (xmlChar *) "FourCC");
 173   gchar *channels = (gchar *) xmlGetProp (node, (xmlChar *) "Channels");
 174   gchar *rate = (gchar *) xmlGetProp (node, (xmlChar *) "SamplingRate");
 175   gchar *codec_data =
 176       (gchar *) xmlGetProp (node, (xmlChar *) "CodecPrivateData");
 177
 178   caps = _gst_mss_manifest_stream_audio_caps_from_fourcc (fourcc);
 179   if (!caps)
 180     goto end;
 181
 182   structure = gst_caps_get_structure (caps, 0);
 183
 184   if (channels)
 185     gst_structure_set (structure, "channels", G_TYPE_INT, atoi (channels),
 186         NULL);
 187   if (rate)
 188     gst_structure_set (structure, "rate", G_TYPE_INT, atoi (rate), NULL);
 189
 190   if (codec_data) {
 191     GValue *value = g_new0 (GValue, 1);
 192     g_value_init (value, GST_TYPE_BUFFER);
 193     gst_value_deserialize (value, (gchar *) codec_data);
 194     gst_structure_take_value (structure, "codec_data", value);
 195   }
 196
 197 end:
 198   g_free (fourcc);
 199   g_free (channels);
 200   g_free (rate);
 201   g_free (codec_data);
 202
 203   return caps;
 204 }
 205
 206 GstCaps *
 207 gst_mss_manifest_stream_get_caps (GstMssManifestStream * stream)
 208 {
 209   GstMssManifestStreamType streamtype =
 210       gst_mss_manifest_stream_get_type (stream);
 211
 212   /* TODO properly get the stream */
 213   xmlNodePtr qualitylevel = stream->xmlnode->children;
 214   while (strcmp ((gchar *) qualitylevel->name, "QualityLevel")) {
 215     qualitylevel = qualitylevel->next;
 216   }
 217
 218   if (streamtype == MSS_STREAM_TYPE_VIDEO)
 219     return
 220         _gst_mss_manifest_stream_video_caps_from_qualitylevel_xml
 221         (qualitylevel);
 222   else if (streamtype == MSS_STREAM_TYPE_AUDIO)
 223     return
 224         _gst_mss_manifest_stream_audio_caps_from_qualitylevel_xml
 225         (qualitylevel);
 226
 227   return NULL;
 228 }
 229
 230 const gchar *
 231 gst_mss_manifest_stream_type_name (GstMssManifestStreamType streamtype)
 232 {
 233   switch (streamtype) {
 234     case MSS_STREAM_TYPE_VIDEO:
 235       return "video";
 236     case MSS_STREAM_TYPE_AUDIO:
 237       return "audio";
 238     case MSS_STREAM_TYPE_UNKNOWN:
 239     default:
 240       return "unknown";
 241   }
 242 }