gst-libs/gst/video/video-info.c

   1 /* GStreamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  * Library       <2002> Ronald Bultje <rbultje@ronald.bitfreak.net>
   4  * Copyright (C) 2007 David A. Schleef <ds@schleef.org>
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Library General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2 of the License, or (at your option) any later version.
  10  *
  11  * This library 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  * Library General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Library General Public
  17  * License along with this library; if not, write to the
  18  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  19  * Boston, MA 02110-1301, USA.
  20  */
  21
  22 /**
  23  * SECTION:video-info
  24  * @title: GstVideoInfo
  25  * @short_description: Structures and enumerations to describe raw images
  26  */
  27
  28 #ifdef HAVE_CONFIG_H
  29 #  include "config.h"
  30 #endif
  31
  32 #include <string.h>
  33 #include <stdio.h>
  34
  35 #include "video-info.h"
  36 #include "video-tile.h"
  37
  38 #ifndef GST_DISABLE_GST_DEBUG
  39 #define GST_CAT_DEFAULT ensure_debug_category()
  40 static GstDebugCategory *
  41 ensure_debug_category (void)
  42 {
  43   static gsize cat_gonce = 0;
  44
  45   if (g_once_init_enter (&cat_gonce)) {
  46     gsize cat_done;
  47
  48     cat_done = (gsize) _gst_debug_category_new ("video-info", 0,
  49         "video-info structure");
  50
  51     g_once_init_leave (&cat_gonce, cat_done);
  52   }
  53
  54   return (GstDebugCategory *) cat_gonce;
  55 }
  56 #else
  57 #define ensure_debug_category() /* NOOP */
  58 #endif /* GST_DISABLE_GST_DEBUG */
  59
  60 /**
  61  * gst_video_info_copy:
  62  * @info: a #GstVideoInfo
  63  *
  64  * Copy a GstVideoInfo structure.
  65  *
  66  * Returns: a new #GstVideoInfo. free with gst_video_info_free.
  67  *
  68  * Since: 1.6
  69  */
  70 GstVideoInfo *
  71 gst_video_info_copy (const GstVideoInfo * info)
  72 {
  73   return g_slice_dup (GstVideoInfo, info);
  74 }
  75
  76 /**
  77  * gst_video_info_free:
  78  * @info: a #GstVideoInfo
  79  *
  80  * Free a GstVideoInfo structure previously allocated with gst_video_info_new()
  81  * or gst_video_info_copy().
  82  *
  83  * Since: 1.6
  84  */
  85 void
  86 gst_video_info_free (GstVideoInfo * info)
  87 {
  88   g_slice_free (GstVideoInfo, info);
  89 }
  90
  91 G_DEFINE_BOXED_TYPE (GstVideoInfo, gst_video_info,
  92     (GBoxedCopyFunc) gst_video_info_copy, (GBoxedFreeFunc) gst_video_info_free);
  93
  94 /**
  95  * gst_video_info_new:
  96  *
  97  * Allocate a new #GstVideoInfo that is also initialized with
  98  * gst_video_info_init().
  99  *
 100  * Returns: a new #GstVideoInfo. free with gst_video_info_free().
 101  *
 102  * Since: 1.6
 103  */
 104 GstVideoInfo *
 105 gst_video_info_new (void)
 106 {
 107   GstVideoInfo *info;
 108
 109   info = g_slice_new (GstVideoInfo);
 110   gst_video_info_init (info);
 111
 112   return info;
 113 }
 114
 115 static gboolean fill_planes (GstVideoInfo * info,
 116     gsize plane_size[GST_VIDEO_MAX_PLANES]);
 117
 118 /**
 119  * gst_video_info_init:
 120  * @info: (out caller-allocates): a #GstVideoInfo
 121  *
 122  * Initialize @info with default values.
 123  */
 124 void
 125 gst_video_info_init (GstVideoInfo * info)
 126 {
 127   g_return_if_fail (info != NULL);
 128
 129   memset (info, 0, sizeof (GstVideoInfo));
 130
 131   info->finfo = gst_video_format_get_info (GST_VIDEO_FORMAT_UNKNOWN);
 132
 133   info->views = 1;
 134   /* arrange for sensible defaults, e.g. if turned into caps */
 135   info->fps_n = 0;
 136   info->fps_d = 1;
 137   info->par_n = 1;
 138   info->par_d = 1;
 139   GST_VIDEO_INFO_MULTIVIEW_MODE (info) = GST_VIDEO_MULTIVIEW_MODE_NONE;
 140   GST_VIDEO_INFO_MULTIVIEW_FLAGS (info) = GST_VIDEO_MULTIVIEW_FLAGS_NONE;
 141   GST_VIDEO_INFO_FIELD_ORDER (info) = GST_VIDEO_FIELD_ORDER_UNKNOWN;
 142 }
 143
 144 #define MAKE_COLORIMETRY(r,m,t,p) {  \
 145   GST_VIDEO_COLOR_RANGE ##r, GST_VIDEO_COLOR_MATRIX_ ##m, \
 146   GST_VIDEO_TRANSFER_ ##t, GST_VIDEO_COLOR_PRIMARIES_ ##p }
 147
 148 #define DEFAULT_YUV_SD  0
 149 #define DEFAULT_YUV_HD  1
 150 #define DEFAULT_RGB     2
 151 #define DEFAULT_GRAY    3
 152 #define DEFAULT_UNKNOWN 4
 153 #define DEFAULT_YUV_UHD 5
 154
 155 static const GstVideoColorimetry default_color[] = {
 156   MAKE_COLORIMETRY (_16_235, BT601, BT601, SMPTE170M),
 157   MAKE_COLORIMETRY (_16_235, BT709, BT709, BT709),
 158   MAKE_COLORIMETRY (_0_255, RGB, SRGB, BT709),
 159   MAKE_COLORIMETRY (_0_255, BT601, UNKNOWN, UNKNOWN),
 160   MAKE_COLORIMETRY (_UNKNOWN, UNKNOWN, UNKNOWN, UNKNOWN),
 161   MAKE_COLORIMETRY (_16_235, BT2020, BT2020_12, BT2020),
 162 };
 163
 164 static void
 165 set_default_colorimetry (GstVideoInfo * info)
 166 {
 167   const GstVideoFormatInfo *finfo = info->finfo;
 168
 169   if (GST_VIDEO_FORMAT_INFO_IS_YUV (finfo)) {
 170     if (info->height >= 2160) {
 171       info->chroma_site = GST_VIDEO_CHROMA_SITE_H_COSITED;
 172       info->colorimetry = default_color[DEFAULT_YUV_UHD];
 173     } else if (info->height > 576) {
 174       info->chroma_site = GST_VIDEO_CHROMA_SITE_H_COSITED;
 175       info->colorimetry = default_color[DEFAULT_YUV_HD];
 176     } else {
 177       info->chroma_site = GST_VIDEO_CHROMA_SITE_NONE;
 178       info->colorimetry = default_color[DEFAULT_YUV_SD];
 179     }
 180   } else if (GST_VIDEO_FORMAT_INFO_IS_GRAY (finfo)) {
 181     info->colorimetry = default_color[DEFAULT_GRAY];
 182   } else if (GST_VIDEO_FORMAT_INFO_IS_RGB (finfo)) {
 183     info->colorimetry = default_color[DEFAULT_RGB];
 184   } else {
 185     info->colorimetry = default_color[DEFAULT_UNKNOWN];
 186   }
 187 }
 188
 189 static gboolean
 190 validate_colorimetry (GstVideoInfo * info)
 191 {
 192   const GstVideoFormatInfo *finfo = info->finfo;
 193
 194   if (!GST_VIDEO_FORMAT_INFO_IS_RGB (finfo) &&
 195       info->colorimetry.matrix == GST_VIDEO_COLOR_MATRIX_RGB) {
 196     GST_WARNING
 197         ("color matrix RGB is only supported with RGB format, %s is not",
 198         finfo->name);
 199     return FALSE;
 200   }
 201
 202   if (GST_VIDEO_FORMAT_INFO_IS_YUV (finfo) &&
 203       info->colorimetry.matrix == GST_VIDEO_COLOR_MATRIX_UNKNOWN) {
 204     GST_WARNING ("Need to specify a color matrix when using YUV format (%s)",
 205         finfo->name);
 206     return FALSE;
 207   }
 208
 209   return TRUE;
 210 }
 211
 212 static gboolean
 213 gst_video_info_set_format_common (GstVideoInfo * info, GstVideoFormat format,
 214     guint width, guint height)
 215 {
 216   g_return_val_if_fail (info != NULL, FALSE);
 217   g_return_val_if_fail (format != GST_VIDEO_FORMAT_UNKNOWN, FALSE);
 218
 219   if (width > G_MAXINT || height > G_MAXINT)
 220     return FALSE;
 221
 222   gst_video_info_init (info);
 223
 224   info->finfo = gst_video_format_get_info (format);
 225   info->width = width;
 226   info->height = height;
 227   info->views = 1;
 228
 229   set_default_colorimetry (info);
 230
 231   return TRUE;
 232 }
 233
 234 /**
 235  * gst_video_info_set_format:
 236  * @info: a #GstVideoInfo
 237  * @format: the format
 238  * @width: a width
 239  * @height: a height
 240  *
 241  * Set the default info for a video frame of @format and @width and @height.
 242  *
 243  * Note: This initializes @info first, no values are preserved. This function
 244  * does not set the offsets correctly for interlaced vertically
 245  * subsampled formats.
 246  *
 247  * Returns: %FALSE if the returned video info is invalid, e.g. because the
 248  *   size of a frame can't be represented as a 32 bit integer (Since: 1.12)
 249  */
 250 gboolean
 251 gst_video_info_set_format (GstVideoInfo * info, GstVideoFormat format,
 252     guint width, guint height)
 253 {
 254   if (!gst_video_info_set_format_common (info, format, width, height))
 255     return FALSE;
 256
 257   return fill_planes (info, NULL);
 258 }
 259
 260 /**
 261  * gst_video_info_set_interlaced_format:
 262  * @info: a #GstVideoInfo
 263  * @format: the format
 264  * @mode: a #GstVideoInterlaceMode
 265  * @width: a width
 266  * @height: a height
 267  *
 268  * Same as #gst_video_info_set_format but also allowing to set the interlaced
 269  * mode.
 270  *
 271  * Returns: %FALSE if the returned video info is invalid, e.g. because the
 272  *   size of a frame can't be represented as a 32 bit integer.
 273  *
 274  * Since: 1.16
 275  */
 276 gboolean
 277 gst_video_info_set_interlaced_format (GstVideoInfo * info,
 278     GstVideoFormat format, GstVideoInterlaceMode mode, guint width,
 279     guint height)
 280 {
 281   if (!gst_video_info_set_format_common (info, format, width, height))
 282     return FALSE;
 283
 284   GST_VIDEO_INFO_INTERLACE_MODE (info) = mode;
 285   return fill_planes (info, NULL);
 286 }
 287
 288 static const gchar *interlace_mode[] = {
 289   "progressive",
 290   "interleaved",
 291   "mixed",
 292   "fields",
 293   "alternate"
 294 };
 295
 296 /**
 297  * gst_video_interlace_mode_to_string:
 298  * @mode: a #GstVideoInterlaceMode
 299  *
 300  * Convert @mode to its string representation.
 301  *
 302  * Returns: @mode as a string or NULL if @mode in invalid.
 303  *
 304  * Since: 1.6
 305  */
 306 const gchar *
 307 gst_video_interlace_mode_to_string (GstVideoInterlaceMode mode)
 308 {
 309   if (((guint) mode) >= G_N_ELEMENTS (interlace_mode))
 310     return NULL;
 311
 312   return interlace_mode[mode];
 313 }
 314
 315 /**
 316  * gst_video_interlace_mode_from_string:
 317  * @mode: a mode
 318  *
 319  * Convert @mode to a #GstVideoInterlaceMode
 320  *
 321  * Returns: the #GstVideoInterlaceMode of @mode or
 322  *    #GST_VIDEO_INTERLACE_MODE_PROGRESSIVE when @mode is not a valid
 323  *    string representation for a #GstVideoInterlaceMode.
 324  *
 325  * Since: 1.6
 326  */
 327 GstVideoInterlaceMode
 328 gst_video_interlace_mode_from_string (const gchar * mode)
 329 {
 330   gint i;
 331   for (i = 0; i < G_N_ELEMENTS (interlace_mode); i++) {
 332     if (g_str_equal (interlace_mode[i], mode))
 333       return i;
 334   }
 335   return GST_VIDEO_INTERLACE_MODE_PROGRESSIVE;
 336 }
 337
 338 static const gchar *field_order[] = {
 339   "unknown",
 340   "top-field-first",
 341   "bottom-field-first"
 342 };
 343
 344 /**
 345  * gst_video_field_order_to_string:
 346  * @order: a #GstVideoFieldOrder
 347  *
 348  * Convert @order to its string representation.
 349  *
 350  * Returns: @order as a string or NULL if @order in invalid.
 351  *
 352  * Since: 1.12
 353  */
 354 const gchar *
 355 gst_video_field_order_to_string (GstVideoFieldOrder order)
 356 {
 357   if (((guint) order) >= G_N_ELEMENTS (field_order))
 358     return NULL;
 359
 360   return field_order[order];
 361 }
 362
 363 /**
 364  * gst_video_field_order_from_string:
 365  * @order: a field order
 366  *
 367  * Convert @order to a #GstVideoFieldOrder
 368  *
 369  * Returns: the #GstVideoFieldOrder of @order or
 370  *    #GST_VIDEO_FIELD_ORDER_UNKNOWN when @order is not a valid
 371  *    string representation for a #GstVideoFieldOrder.
 372  *
 373  * Since: 1.12
 374  */
 375 GstVideoFieldOrder
 376 gst_video_field_order_from_string (const gchar * order)
 377 {
 378   gint i;
 379   for (i = 0; i < G_N_ELEMENTS (field_order); i++) {
 380     if (g_str_equal (field_order[i], order))
 381       return i;
 382   }
 383   return GST_VIDEO_FIELD_ORDER_UNKNOWN;
 384 }
 385
 386 /**
 387  * gst_video_info_from_caps:
 388  * @info: (out caller-allocates): #GstVideoInfo
 389  * @caps: a #GstCaps
 390  *
 391  * Parse @caps and update @info.
 392  *
 393  * Returns: TRUE if @caps could be parsed
 394  */
 395 gboolean
 396 gst_video_info_from_caps (GstVideoInfo * info, const GstCaps * caps)
 397 {
 398   GstStructure *structure;
 399   const gchar *s;
 400   GstVideoFormat format = GST_VIDEO_FORMAT_UNKNOWN;
 401   gint width = 0, height = 0;
 402   gint fps_n, fps_d;
 403   gint par_n, par_d;
 404   guint multiview_flags;
 405
 406   g_return_val_if_fail (info != NULL, FALSE);
 407   g_return_val_if_fail (caps != NULL, FALSE);
 408   g_return_val_if_fail (gst_caps_is_fixed (caps), FALSE);
 409
 410   GST_DEBUG ("parsing caps %" GST_PTR_FORMAT, caps);
 411
 412   structure = gst_caps_get_structure (caps, 0);
 413
 414   if (gst_structure_has_name (structure, "video/x-raw")) {
 415     if (!(s = gst_structure_get_string (structure, "format")))
 416       goto no_format;
 417
 418     format = gst_video_format_from_string (s);
 419     if (format == GST_VIDEO_FORMAT_UNKNOWN)
 420       goto unknown_format;
 421
 422   } else if (g_str_has_prefix (gst_structure_get_name (structure), "video/") ||
 423       g_str_has_prefix (gst_structure_get_name (structure), "image/")) {
 424     format = GST_VIDEO_FORMAT_ENCODED;
 425   } else {
 426     goto wrong_name;
 427   }
 428
 429   /* width and height are mandatory, except for non-raw-formats */
 430   if (!gst_structure_get_int (structure, "width", &width) &&
 431       format != GST_VIDEO_FORMAT_ENCODED)
 432     goto no_width;
 433   if (!gst_structure_get_int (structure, "height", &height) &&
 434       format != GST_VIDEO_FORMAT_ENCODED)
 435     goto no_height;
 436
 437   gst_video_info_init (info);
 438
 439   info->finfo = gst_video_format_get_info (format);
 440   info->width = width;
 441   info->height = height;
 442
 443   if (gst_structure_get_fraction (structure, "framerate", &fps_n, &fps_d)) {
 444     if (fps_n == 0) {
 445       /* variable framerate */
 446       info->flags |= GST_VIDEO_FLAG_VARIABLE_FPS;
 447       /* see if we have a max-framerate */
 448       gst_structure_get_fraction (structure, "max-framerate", &fps_n, &fps_d);
 449     }
 450     info->fps_n = fps_n;
 451     info->fps_d = fps_d;
 452   } else {
 453     /* unspecified is variable framerate */
 454     info->fps_n = 0;
 455     info->fps_d = 1;
 456   }
 457
 458   if (gst_structure_get_fraction (structure, "pixel-aspect-ratio",
 459           &par_n, &par_d)) {
 460     info->par_n = par_n;
 461     info->par_d = par_d;
 462   } else {
 463     info->par_n = 1;
 464     info->par_d = 1;
 465   }
 466
 467   if ((s = gst_structure_get_string (structure, "interlace-mode")))
 468     info->interlace_mode = gst_video_interlace_mode_from_string (s);
 469   else
 470     info->interlace_mode = GST_VIDEO_INTERLACE_MODE_PROGRESSIVE;
 471
 472   /* Interlaced feature is mandatory for raw alternate streams */
 473   if (info->interlace_mode == GST_VIDEO_INTERLACE_MODE_ALTERNATE &&
 474       format != GST_VIDEO_FORMAT_ENCODED) {
 475     GstCapsFeatures *f;
 476
 477     f = gst_caps_get_features (caps, 0);
 478     if (!f
 479         || !gst_caps_features_contains (f, GST_CAPS_FEATURE_FORMAT_INTERLACED))
 480       goto alternate_no_feature;
 481   }
 482
 483   if (GST_VIDEO_INFO_IS_INTERLACED (info) &&
 484       (s = gst_structure_get_string (structure, "field-order"))) {
 485     GST_VIDEO_INFO_FIELD_ORDER (info) = gst_video_field_order_from_string (s);
 486   } else {
 487     GST_VIDEO_INFO_FIELD_ORDER (info) = GST_VIDEO_FIELD_ORDER_UNKNOWN;
 488   }
 489
 490   {
 491     if ((s = gst_structure_get_string (structure, "multiview-mode")))
 492       GST_VIDEO_INFO_MULTIVIEW_MODE (info) =
 493           gst_video_multiview_mode_from_caps_string (s);
 494     else
 495       GST_VIDEO_INFO_MULTIVIEW_MODE (info) = GST_VIDEO_MULTIVIEW_MODE_NONE;
 496
 497     if (gst_structure_get_flagset (structure, "multiview-flags",
 498             &multiview_flags, NULL))
 499       GST_VIDEO_INFO_MULTIVIEW_FLAGS (info) = multiview_flags;
 500
 501     if (!gst_structure_get_int (structure, "views", &info->views))
 502       info->views = 1;
 503
 504     /* At one point, I tried normalising the half-aspect flag here,
 505      * but it behaves weird for GstVideoInfo operations other than
 506      * directly converting to/from caps - sometimes causing the
 507      * PAR to be doubled/halved too many times */
 508   }
 509
 510   if ((s = gst_structure_get_string (structure, "chroma-site")))
 511     info->chroma_site = gst_video_chroma_site_from_string (s);
 512   else
 513     info->chroma_site = GST_VIDEO_CHROMA_SITE_UNKNOWN;
 514
 515   if ((s = gst_structure_get_string (structure, "colorimetry"))) {
 516     if (!gst_video_colorimetry_from_string (&info->colorimetry, s)) {
 517       GST_WARNING ("unparsable colorimetry, using default");
 518       set_default_colorimetry (info);
 519     } else if (!validate_colorimetry (info)) {
 520       GST_WARNING ("invalid colorimetry, using default");
 521       set_default_colorimetry (info);
 522     } else {
 523       /* force RGB matrix for RGB formats */
 524       if (GST_VIDEO_FORMAT_INFO_IS_RGB (info->finfo) &&
 525           info->colorimetry.matrix != GST_VIDEO_COLOR_MATRIX_RGB) {
 526         GST_WARNING ("invalid matrix %d for RGB format, using RGB",
 527             info->colorimetry.matrix);
 528         info->colorimetry.matrix = GST_VIDEO_COLOR_MATRIX_RGB;
 529       }
 530     }
 531   } else {
 532     GST_DEBUG ("no colorimetry, using default");
 533     set_default_colorimetry (info);
 534   }
 535
 536   if (!fill_planes (info, NULL))
 537     return FALSE;
 538
 539   return TRUE;
 540
 541   /* ERROR */
 542 wrong_name:
 543   {
 544     GST_ERROR ("wrong name '%s', expected video/ or image/",
 545         gst_structure_get_name (structure));
 546     return FALSE;
 547   }
 548 no_format:
 549   {
 550     GST_ERROR ("no format given");
 551     return FALSE;
 552   }
 553 unknown_format:
 554   {
 555     GST_ERROR ("unknown format '%s' given", s);
 556     return FALSE;
 557   }
 558 no_width:
 559   {
 560     GST_ERROR ("no width property given");
 561     return FALSE;
 562   }
 563 no_height:
 564   {
 565     GST_ERROR ("no height property given");
 566     return FALSE;
 567   }
 568 alternate_no_feature:
 569   {
 570     GST_ERROR
 571         ("caps has 'interlace-mode=alternate' but doesn't have the Interlaced feature");
 572     return FALSE;
 573   }
 574 }
 575
 576 /**
 577  * gst_video_info_is_equal:
 578  * @info: a #GstVideoInfo
 579  * @other: a #GstVideoInfo
 580  *
 581  * Compares two #GstVideoInfo and returns whether they are equal or not
 582  *
 583  * Returns: %TRUE if @info and @other are equal, else %FALSE.
 584  */
 585 gboolean
 586 gst_video_info_is_equal (const GstVideoInfo * info, const GstVideoInfo * other)
 587 {
 588   gint i;
 589
 590   if (GST_VIDEO_INFO_FORMAT (info) != GST_VIDEO_INFO_FORMAT (other))
 591     return FALSE;
 592   if (GST_VIDEO_INFO_INTERLACE_MODE (info) !=
 593       GST_VIDEO_INFO_INTERLACE_MODE (other))
 594     return FALSE;
 595   if (GST_VIDEO_INFO_FLAGS (info) != GST_VIDEO_INFO_FLAGS (other))
 596     return FALSE;
 597   if (GST_VIDEO_INFO_WIDTH (info) != GST_VIDEO_INFO_WIDTH (other))
 598     return FALSE;
 599   if (GST_VIDEO_INFO_HEIGHT (info) != GST_VIDEO_INFO_HEIGHT (other))
 600     return FALSE;
 601   if (GST_VIDEO_INFO_SIZE (info) != GST_VIDEO_INFO_SIZE (other))
 602     return FALSE;
 603   if (GST_VIDEO_INFO_PAR_N (info) != GST_VIDEO_INFO_PAR_N (other))
 604     return FALSE;
 605   if (GST_VIDEO_INFO_PAR_D (info) != GST_VIDEO_INFO_PAR_D (other))
 606     return FALSE;
 607   if (GST_VIDEO_INFO_FPS_N (info) != GST_VIDEO_INFO_FPS_N (other))
 608     return FALSE;
 609   if (GST_VIDEO_INFO_FPS_D (info) != GST_VIDEO_INFO_FPS_D (other))
 610     return FALSE;
 611   if (!gst_video_colorimetry_is_equal (&GST_VIDEO_INFO_COLORIMETRY (info),
 612           &GST_VIDEO_INFO_COLORIMETRY (other)))
 613     return FALSE;
 614   if (GST_VIDEO_INFO_CHROMA_SITE (info) != GST_VIDEO_INFO_CHROMA_SITE (other))
 615     return FALSE;
 616   if (GST_VIDEO_INFO_MULTIVIEW_MODE (info) !=
 617       GST_VIDEO_INFO_MULTIVIEW_MODE (other))
 618     return FALSE;
 619   if (GST_VIDEO_INFO_MULTIVIEW_FLAGS (info) !=
 620       GST_VIDEO_INFO_MULTIVIEW_FLAGS (other))
 621     return FALSE;
 622   if (GST_VIDEO_INFO_VIEWS (info) != GST_VIDEO_INFO_VIEWS (other))
 623     return FALSE;
 624
 625   for (i = 0; i < info->finfo->n_planes; i++) {
 626     if (info->stride[i] != other->stride[i])
 627       return FALSE;
 628     if (info->offset[i] != other->offset[i])
 629       return FALSE;
 630   }
 631
 632   return TRUE;
 633 }
 634
 635 /**
 636  * gst_video_info_to_caps:
 637  * @info: a #GstVideoInfo
 638  *
 639  * Convert the values of @info into a #GstCaps.
 640  *
 641  * Returns: a new #GstCaps containing the info of @info.
 642  */
 643 GstCaps *
 644 gst_video_info_to_caps (const GstVideoInfo * info)
 645 {
 646   GstCaps *caps;
 647   const gchar *format;
 648   gchar *color;
 649   gint par_n, par_d;
 650   GstVideoColorimetry colorimetry;
 651
 652   g_return_val_if_fail (info != NULL, NULL);
 653   g_return_val_if_fail (info->finfo != NULL, NULL);
 654   g_return_val_if_fail (info->finfo->format != GST_VIDEO_FORMAT_UNKNOWN, NULL);
 655
 656   format = gst_video_format_to_string (info->finfo->format);
 657   g_return_val_if_fail (format != NULL, NULL);
 658
 659   caps = gst_caps_new_simple ("video/x-raw",
 660       "format", G_TYPE_STRING, format,
 661       "width", G_TYPE_INT, info->width,
 662       "height", G_TYPE_INT, info->height, NULL);
 663
 664   par_n = info->par_n;
 665   par_d = info->par_d;
 666
 667   gst_caps_set_simple (caps, "interlace-mode", G_TYPE_STRING,
 668       gst_video_interlace_mode_to_string (info->interlace_mode), NULL);
 669
 670   if ((info->interlace_mode == GST_VIDEO_INTERLACE_MODE_INTERLEAVED ||
 671           info->interlace_mode == GST_VIDEO_INTERLACE_MODE_ALTERNATE) &&
 672       GST_VIDEO_INFO_FIELD_ORDER (info) != GST_VIDEO_FIELD_ORDER_UNKNOWN) {
 673     gst_caps_set_simple (caps, "field-order", G_TYPE_STRING,
 674         gst_video_field_order_to_string (GST_VIDEO_INFO_FIELD_ORDER (info)),
 675         NULL);
 676   }
 677
 678   if (info->interlace_mode == GST_VIDEO_INTERLACE_MODE_ALTERNATE) {
 679     /* 'alternate' mode must always be accompanied by interlaced caps feature.
 680      */
 681     GstCapsFeatures *features;
 682
 683     features = gst_caps_features_new (GST_CAPS_FEATURE_FORMAT_INTERLACED, NULL);
 684     gst_caps_set_features (caps, 0, features);
 685   }
 686
 687   if (GST_VIDEO_INFO_MULTIVIEW_MODE (info) != GST_VIDEO_MULTIVIEW_MODE_NONE) {
 688     const gchar *caps_str = NULL;
 689     GstVideoMultiviewFlags multiview_flags =
 690         GST_VIDEO_INFO_MULTIVIEW_FLAGS (info);
 691
 692     /* If the half-aspect flag is set, applying it into the PAR of the
 693      * resulting caps now seems safe, and helps with automatic behaviour
 694      * in elements that aren't explicitly multiview aware */
 695     if (multiview_flags & GST_VIDEO_MULTIVIEW_FLAGS_HALF_ASPECT) {
 696       multiview_flags &= ~GST_VIDEO_MULTIVIEW_FLAGS_HALF_ASPECT;
 697       switch (GST_VIDEO_INFO_MULTIVIEW_MODE (info)) {
 698         case GST_VIDEO_MULTIVIEW_MODE_SIDE_BY_SIDE:
 699         case GST_VIDEO_MULTIVIEW_MODE_SIDE_BY_SIDE_QUINCUNX:
 700         case GST_VIDEO_MULTIVIEW_MODE_COLUMN_INTERLEAVED:
 701         case GST_VIDEO_MULTIVIEW_MODE_CHECKERBOARD:
 702           par_n *= 2;           /* double the width / half the height */
 703           break;
 704         case GST_VIDEO_MULTIVIEW_MODE_ROW_INTERLEAVED:
 705         case GST_VIDEO_MULTIVIEW_MODE_TOP_BOTTOM:
 706           par_d *= 2;           /* half the width / double the height */
 707           break;
 708         default:
 709           break;
 710       }
 711     }
 712
 713     caps_str =
 714         gst_video_multiview_mode_to_caps_string (GST_VIDEO_INFO_MULTIVIEW_MODE
 715         (info));
 716     if (caps_str != NULL) {
 717       gst_caps_set_simple (caps, "multiview-mode", G_TYPE_STRING,
 718           caps_str, "multiview-flags", GST_TYPE_VIDEO_MULTIVIEW_FLAGSET,
 719           multiview_flags, GST_FLAG_SET_MASK_EXACT, NULL);
 720     }
 721   }
 722
 723   gst_caps_set_simple (caps, "pixel-aspect-ratio",
 724       GST_TYPE_FRACTION, par_n, par_d, NULL);
 725
 726   if (info->chroma_site != GST_VIDEO_CHROMA_SITE_UNKNOWN) {
 727     gchar *chroma_site = gst_video_chroma_site_to_string (info->chroma_site);
 728
 729     if (!chroma_site) {
 730       GST_WARNING ("Couldn't convert chroma-site 0x%x to string",
 731           info->chroma_site);
 732     } else {
 733       gst_caps_set_simple (caps,
 734           "chroma-site", G_TYPE_STRING, chroma_site, NULL);
 735       g_free (chroma_site);
 736     }
 737   }
 738
 739   /* make sure we set the RGB matrix for RGB formats */
 740   colorimetry = info->colorimetry;
 741   if (GST_VIDEO_FORMAT_INFO_IS_RGB (info->finfo) &&
 742       colorimetry.matrix != GST_VIDEO_COLOR_MATRIX_RGB) {
 743     GST_WARNING ("invalid matrix %d for RGB format, using RGB",
 744         colorimetry.matrix);
 745     colorimetry.matrix = GST_VIDEO_COLOR_MATRIX_RGB;
 746   }
 747   if ((color = gst_video_colorimetry_to_string (&colorimetry))) {
 748     gst_caps_set_simple (caps, "colorimetry", G_TYPE_STRING, color, NULL);
 749     g_free (color);
 750   }
 751
 752   if (info->views > 1)
 753     gst_caps_set_simple (caps, "views", G_TYPE_INT, info->views, NULL);
 754
 755   if (info->flags & GST_VIDEO_FLAG_VARIABLE_FPS && info->fps_n != 0) {
 756     /* variable fps with a max-framerate */
 757     gst_caps_set_simple (caps, "framerate", GST_TYPE_FRACTION, 0, 1,
 758         "max-framerate", GST_TYPE_FRACTION, info->fps_n, info->fps_d, NULL);
 759   } else {
 760     /* no variable fps or no max-framerate */
 761     gst_caps_set_simple (caps, "framerate", GST_TYPE_FRACTION,
 762         info->fps_n, info->fps_d, NULL);
 763   }
 764
 765   return caps;
 766 }
 767
 768 static gboolean
 769 fill_planes (GstVideoInfo * info, gsize plane_size[GST_VIDEO_MAX_PLANES])
 770 {
 771   gsize width, height, cr_h;
 772   gint bpp = 0, i;
 773
 774   width = (gsize) info->width;
 775   height = (gsize) GST_VIDEO_INFO_FIELD_HEIGHT (info);
 776
 777   /* Sanity check the resulting frame size for overflows */
 778   for (i = 0; i < GST_VIDEO_INFO_N_COMPONENTS (info); i++)
 779     bpp += GST_VIDEO_INFO_COMP_DEPTH (info, i);
 780   bpp = GST_ROUND_UP_8 (bpp) / 8;
 781   if (bpp > 0 && GST_ROUND_UP_128 ((guint64) width) * ((guint64) height) >=
 782       G_MAXUINT / bpp) {
 783     GST_ERROR ("Frame size %ux%u would overflow", info->width, info->height);
 784     return FALSE;
 785   }
 786
 787   switch (info->finfo->format) {
 788     case GST_VIDEO_FORMAT_YUY2:
 789     case GST_VIDEO_FORMAT_YVYU:
 790     case GST_VIDEO_FORMAT_UYVY:
 791     case GST_VIDEO_FORMAT_VYUY:
 792       info->stride[0] = GST_ROUND_UP_4 (width * 2);
 793       info->offset[0] = 0;
 794       info->size = info->stride[0] * height;
 795       break;
 796     case GST_VIDEO_FORMAT_AYUV:
 797     case GST_VIDEO_FORMAT_RGBx:
 798     case GST_VIDEO_FORMAT_RGBA:
 799     case GST_VIDEO_FORMAT_BGRx:
 800     case GST_VIDEO_FORMAT_BGRA:
 801     case GST_VIDEO_FORMAT_xRGB:
 802     case GST_VIDEO_FORMAT_ARGB:
 803     case GST_VIDEO_FORMAT_xBGR:
 804     case GST_VIDEO_FORMAT_ABGR:
 805     case GST_VIDEO_FORMAT_r210:
 806     case GST_VIDEO_FORMAT_Y410:
 807     case GST_VIDEO_FORMAT_VUYA:
 808     case GST_VIDEO_FORMAT_BGR10A2_LE:
 809     case GST_VIDEO_FORMAT_RGB10A2_LE:
 810       info->stride[0] = width * 4;
 811       info->offset[0] = 0;
 812       info->size = info->stride[0] * height;
 813       break;
 814     case GST_VIDEO_FORMAT_RGB16:
 815     case GST_VIDEO_FORMAT_BGR16:
 816     case GST_VIDEO_FORMAT_RGB15:
 817     case GST_VIDEO_FORMAT_BGR15:
 818       info->stride[0] = GST_ROUND_UP_4 (width * 2);
 819       info->offset[0] = 0;
 820       info->size = info->stride[0] * height;
 821       break;
 822     case GST_VIDEO_FORMAT_RGB:
 823     case GST_VIDEO_FORMAT_BGR:
 824     case GST_VIDEO_FORMAT_v308:
 825     case GST_VIDEO_FORMAT_IYU2:
 826       info->stride[0] = GST_ROUND_UP_4 (width * 3);
 827       info->offset[0] = 0;
 828       info->size = info->stride[0] * height;
 829       break;
 830     case GST_VIDEO_FORMAT_v210:
 831       info->stride[0] = ((width + 47) / 48) * 128;
 832       info->offset[0] = 0;
 833       info->size = info->stride[0] * height;
 834       break;
 835     case GST_VIDEO_FORMAT_v216:
 836     case GST_VIDEO_FORMAT_Y210:
 837     case GST_VIDEO_FORMAT_Y212_BE:
 838     case GST_VIDEO_FORMAT_Y212_LE:
 839       info->stride[0] = GST_ROUND_UP_8 (width * 4);
 840       info->offset[0] = 0;
 841       info->size = info->stride[0] * height;
 842       break;
 843     case GST_VIDEO_FORMAT_GRAY8:
 844       info->stride[0] = GST_ROUND_UP_4 (width);
 845       info->offset[0] = 0;
 846       info->size = info->stride[0] * height;
 847       break;
 848     case GST_VIDEO_FORMAT_GRAY16_BE:
 849     case GST_VIDEO_FORMAT_GRAY16_LE:
 850       info->stride[0] = GST_ROUND_UP_4 (width * 2);
 851       info->offset[0] = 0;
 852       info->size = info->stride[0] * height;
 853       break;
 854     case GST_VIDEO_FORMAT_UYVP:
 855       info->stride[0] = GST_ROUND_UP_4 ((width * 2 * 5 + 3) / 4);
 856       info->offset[0] = 0;
 857       info->size = info->stride[0] * height;
 858       break;
 859     case GST_VIDEO_FORMAT_RGB8P:
 860       info->stride[0] = GST_ROUND_UP_4 (width);
 861       info->stride[1] = 4;
 862       info->offset[0] = 0;
 863       info->offset[1] = info->stride[0] * height;
 864       info->size = info->offset[1] + (4 * 256);
 865       break;
 866     case GST_VIDEO_FORMAT_IYU1:
 867       info->stride[0] = GST_ROUND_UP_4 (GST_ROUND_UP_4 (width) +
 868           GST_ROUND_UP_4 (width) / 2);
 869       info->offset[0] = 0;
 870       info->size = info->stride[0] * height;
 871       break;
 872     case GST_VIDEO_FORMAT_ARGB64:
 873     case GST_VIDEO_FORMAT_AYUV64:
 874     case GST_VIDEO_FORMAT_Y412_BE:
 875     case GST_VIDEO_FORMAT_Y412_LE:
 876       info->stride[0] = width * 8;
 877       info->offset[0] = 0;
 878       info->size = info->stride[0] * height;
 879       break;
 880     case GST_VIDEO_FORMAT_I420:
 881     case GST_VIDEO_FORMAT_YV12:        /* same as I420, but plane 1+2 swapped */
 882       info->stride[0] = GST_ROUND_UP_4 (width);
 883       info->stride[1] = GST_ROUND_UP_4 (GST_ROUND_UP_2 (width) / 2);
 884       info->stride[2] = info->stride[1];
 885       info->offset[0] = 0;
 886       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
 887       cr_h = GST_ROUND_UP_2 (height) / 2;
 888       if (GST_VIDEO_INFO_IS_INTERLACED (info))
 889         cr_h = GST_ROUND_UP_2 (cr_h);
 890       info->offset[2] = info->offset[1] + info->stride[1] * cr_h;
 891       info->size = info->offset[2] + info->stride[2] * cr_h;
 892       break;
 893     case GST_VIDEO_FORMAT_Y41B:
 894       info->stride[0] = GST_ROUND_UP_4 (width);
 895       info->stride[1] = GST_ROUND_UP_16 (width) / 4;
 896       info->stride[2] = info->stride[1];
 897       info->offset[0] = 0;
 898       info->offset[1] = info->stride[0] * height;
 899       info->offset[2] = info->offset[1] + info->stride[1] * height;
 900       /* simplification of ROUNDUP4(w)*h + 2*((ROUNDUP16(w)/4)*h */
 901       info->size = (info->stride[0] + (GST_ROUND_UP_16 (width) / 2)) * height;
 902       break;
 903     case GST_VIDEO_FORMAT_Y42B:
 904       info->stride[0] = GST_ROUND_UP_4 (width);
 905       info->stride[1] = GST_ROUND_UP_8 (width) / 2;
 906       info->stride[2] = info->stride[1];
 907       info->offset[0] = 0;
 908       info->offset[1] = info->stride[0] * height;
 909       info->offset[2] = info->offset[1] + info->stride[1] * height;
 910       /* simplification of ROUNDUP4(w)*h + 2*(ROUNDUP8(w)/2)*h */
 911       info->size = (info->stride[0] + GST_ROUND_UP_8 (width)) * height;
 912       break;
 913     case GST_VIDEO_FORMAT_Y444:
 914     case GST_VIDEO_FORMAT_GBR:
 915     case GST_VIDEO_FORMAT_RGBP:
 916     case GST_VIDEO_FORMAT_BGRP:
 917       info->stride[0] = GST_ROUND_UP_4 (width);
 918       info->stride[1] = info->stride[0];
 919       info->stride[2] = info->stride[0];
 920       info->offset[0] = 0;
 921       info->offset[1] = info->stride[0] * height;
 922       info->offset[2] = info->offset[1] * 2;
 923       info->size = info->stride[0] * height * 3;
 924       break;
 925     case GST_VIDEO_FORMAT_GBRA:
 926       info->stride[0] = GST_ROUND_UP_4 (width);
 927       info->stride[1] = info->stride[0];
 928       info->stride[2] = info->stride[0];
 929       info->stride[3] = info->stride[0];
 930       info->offset[0] = 0;
 931       info->offset[1] = info->stride[0] * height;
 932       info->offset[2] = info->offset[1] * 2;
 933       info->offset[3] = info->offset[1] * 3;
 934       info->size = info->stride[0] * height * 4;
 935       break;
 936     case GST_VIDEO_FORMAT_NV12:
 937     case GST_VIDEO_FORMAT_NV21:
 938       info->stride[0] = GST_ROUND_UP_4 (width);
 939       info->stride[1] = info->stride[0];
 940       info->offset[0] = 0;
 941       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
 942       cr_h = GST_ROUND_UP_2 (height) / 2;
 943       if (GST_VIDEO_INFO_IS_INTERLACED (info))
 944         cr_h = GST_ROUND_UP_2 (cr_h);
 945       info->size = info->offset[1] + info->stride[0] * cr_h;
 946       break;
 947     case GST_VIDEO_FORMAT_NV16:
 948     case GST_VIDEO_FORMAT_NV61:
 949       info->stride[0] = GST_ROUND_UP_4 (width);
 950       info->stride[1] = info->stride[0];
 951       info->offset[0] = 0;
 952       info->offset[1] = info->stride[0] * height;
 953       info->size = info->stride[0] * height * 2;
 954       break;
 955     case GST_VIDEO_FORMAT_NV24:
 956       info->stride[0] = GST_ROUND_UP_4 (width);
 957       info->stride[1] = GST_ROUND_UP_4 (width * 2);
 958       info->offset[0] = 0;
 959       info->offset[1] = info->stride[0] * height;
 960       info->size = info->stride[0] * height + info->stride[1] * height;
 961       break;
 962     case GST_VIDEO_FORMAT_A420:
 963       info->stride[0] = GST_ROUND_UP_4 (width);
 964       info->stride[1] = GST_ROUND_UP_4 (GST_ROUND_UP_2 (width) / 2);
 965       info->stride[2] = info->stride[1];
 966       info->stride[3] = info->stride[0];
 967       info->offset[0] = 0;
 968       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
 969       cr_h = GST_ROUND_UP_2 (height) / 2;
 970       if (GST_VIDEO_INFO_IS_INTERLACED (info))
 971         cr_h = GST_ROUND_UP_2 (cr_h);
 972       info->offset[2] = info->offset[1] + info->stride[1] * cr_h;
 973       info->offset[3] = info->offset[2] + info->stride[2] * cr_h;
 974       info->size = info->offset[3] + info->stride[0] * GST_ROUND_UP_2 (height);
 975       break;
 976     case GST_VIDEO_FORMAT_YUV9:
 977     case GST_VIDEO_FORMAT_YVU9:
 978       info->stride[0] = GST_ROUND_UP_4 (width);
 979       info->stride[1] = GST_ROUND_UP_4 (GST_ROUND_UP_4 (width) / 4);
 980       info->stride[2] = info->stride[1];
 981       info->offset[0] = 0;
 982       info->offset[1] = info->stride[0] * height;
 983       cr_h = GST_ROUND_UP_4 (height) / 4;
 984       if (GST_VIDEO_INFO_IS_INTERLACED (info))
 985         cr_h = GST_ROUND_UP_2 (cr_h);
 986       info->offset[2] = info->offset[1] + info->stride[1] * cr_h;
 987       info->size = info->offset[2] + info->stride[2] * cr_h;
 988       break;
 989     case GST_VIDEO_FORMAT_I420_10LE:
 990     case GST_VIDEO_FORMAT_I420_10BE:
 991     case GST_VIDEO_FORMAT_I420_12LE:
 992     case GST_VIDEO_FORMAT_I420_12BE:
 993       info->stride[0] = GST_ROUND_UP_4 (width * 2);
 994       info->stride[1] = GST_ROUND_UP_4 (width);
 995       info->stride[2] = info->stride[1];
 996       info->offset[0] = 0;
 997       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
 998       cr_h = GST_ROUND_UP_2 (height) / 2;
 999       if (GST_VIDEO_INFO_IS_INTERLACED (info))
1000         cr_h = GST_ROUND_UP_2 (cr_h);
1001       info->offset[2] = info->offset[1] + info->stride[1] * cr_h;
1002       info->size = info->offset[2] + info->stride[2] * cr_h;
1003       break;
1004     case GST_VIDEO_FORMAT_I422_10LE:
1005     case GST_VIDEO_FORMAT_I422_10BE:
1006     case GST_VIDEO_FORMAT_I422_12LE:
1007     case GST_VIDEO_FORMAT_I422_12BE:
1008       info->stride[0] = GST_ROUND_UP_4 (width * 2);
1009       info->stride[1] = GST_ROUND_UP_4 (width);
1010       info->stride[2] = info->stride[1];
1011       info->offset[0] = 0;
1012       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
1013       info->offset[2] = info->offset[1] +
1014           info->stride[1] * GST_ROUND_UP_2 (height);
1015       info->size = info->offset[2] + info->stride[2] * GST_ROUND_UP_2 (height);
1016       break;
1017     case GST_VIDEO_FORMAT_Y444_10LE:
1018     case GST_VIDEO_FORMAT_Y444_10BE:
1019     case GST_VIDEO_FORMAT_Y444_12LE:
1020     case GST_VIDEO_FORMAT_Y444_12BE:
1021     case GST_VIDEO_FORMAT_GBR_10LE:
1022     case GST_VIDEO_FORMAT_GBR_10BE:
1023     case GST_VIDEO_FORMAT_GBR_12LE:
1024     case GST_VIDEO_FORMAT_GBR_12BE:
1025     case GST_VIDEO_FORMAT_Y444_16LE:
1026     case GST_VIDEO_FORMAT_Y444_16BE:
1027       info->stride[0] = GST_ROUND_UP_4 (width * 2);
1028       info->stride[1] = info->stride[0];
1029       info->stride[2] = info->stride[0];
1030       info->offset[0] = 0;
1031       info->offset[1] = info->stride[0] * height;
1032       info->offset[2] = info->offset[1] * 2;
1033       info->size = info->stride[0] * height * 3;
1034       break;
1035     case GST_VIDEO_FORMAT_GBRA_10LE:
1036     case GST_VIDEO_FORMAT_GBRA_10BE:
1037     case GST_VIDEO_FORMAT_GBRA_12LE:
1038     case GST_VIDEO_FORMAT_GBRA_12BE:
1039       info->stride[0] = GST_ROUND_UP_4 (width * 2);
1040       info->stride[1] = info->stride[0];
1041       info->stride[2] = info->stride[0];
1042       info->stride[3] = info->stride[0];
1043       info->offset[0] = 0;
1044       info->offset[1] = info->stride[0] * height;
1045       info->offset[2] = info->offset[1] * 2;
1046       info->offset[3] = info->offset[1] * 3;
1047       info->size = info->stride[0] * height * 4;
1048       break;
1049     case GST_VIDEO_FORMAT_NV12_64Z32:
1050       info->stride[0] =
1051           GST_VIDEO_TILE_MAKE_STRIDE (GST_ROUND_UP_128 (width) / 64,
1052           GST_ROUND_UP_32 (height) / 32);
1053       info->stride[1] =
1054           GST_VIDEO_TILE_MAKE_STRIDE (GST_ROUND_UP_128 (width) / 64,
1055           GST_ROUND_UP_64 (height) / 64);
1056       info->offset[0] = 0;
1057       info->offset[1] = GST_ROUND_UP_128 (width) * GST_ROUND_UP_32 (height);
1058       info->size = info->offset[1] +
1059           GST_ROUND_UP_128 (width) * (GST_ROUND_UP_64 (height) / 2);
1060       break;
1061     case GST_VIDEO_FORMAT_NV12_4L4:
1062     case GST_VIDEO_FORMAT_NV12_32L32:
1063     {
1064       gint ws = GST_VIDEO_FORMAT_INFO_TILE_WS (info->finfo);
1065       gint hs = GST_VIDEO_FORMAT_INFO_TILE_HS (info->finfo);
1066       info->stride[0] =
1067           GST_VIDEO_TILE_MAKE_STRIDE (GST_ROUND_UP_N (width, 1 << ws) >> ws,
1068           GST_ROUND_UP_N (height, 1 << hs) >> hs);
1069       info->stride[1] =
1070           GST_VIDEO_TILE_MAKE_STRIDE (GST_ROUND_UP_N (width, 1 << ws) >> ws,
1071           GST_ROUND_UP_N (height, 1 << (hs + 1)) >> (hs + 1));
1072       info->offset[0] = 0;
1073       info->offset[1] =
1074           GST_ROUND_UP_N (width, 1 << ws) * GST_ROUND_UP_N (height, 1 << hs);
1075       info->size = info->offset[1] +
1076           GST_ROUND_UP_N (width, 1 << ws) *
1077           (GST_ROUND_UP_N (height, 1 << (hs + 1)) / 2);
1078       break;
1079     }
1080     case GST_VIDEO_FORMAT_A420_10LE:
1081     case GST_VIDEO_FORMAT_A420_10BE:
1082       info->stride[0] = GST_ROUND_UP_4 (width * 2);
1083       info->stride[1] = GST_ROUND_UP_4 (width);
1084       info->stride[2] = info->stride[1];
1085       info->stride[3] = info->stride[0];
1086       info->offset[0] = 0;
1087       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
1088       cr_h = GST_ROUND_UP_2 (height) / 2;
1089       if (GST_VIDEO_INFO_IS_INTERLACED (info))
1090         cr_h = GST_ROUND_UP_2 (cr_h);
1091       info->offset[2] = info->offset[1] + info->stride[1] * cr_h;
1092       info->offset[3] = info->offset[2] + info->stride[2] * cr_h;
1093       info->size = info->offset[3] + info->stride[0] * GST_ROUND_UP_2 (height);
1094       break;
1095     case GST_VIDEO_FORMAT_A422_10LE:
1096     case GST_VIDEO_FORMAT_A422_10BE:
1097       info->stride[0] = GST_ROUND_UP_4 (width * 2);
1098       info->stride[1] = GST_ROUND_UP_4 (width);
1099       info->stride[2] = info->stride[1];
1100       info->stride[3] = info->stride[0];
1101       info->offset[0] = 0;
1102       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
1103       info->offset[2] = info->offset[1] +
1104           info->stride[1] * GST_ROUND_UP_2 (height);
1105       info->offset[3] =
1106           info->offset[2] + info->stride[2] * GST_ROUND_UP_2 (height);
1107       info->size = info->offset[3] + info->stride[0] * GST_ROUND_UP_2 (height);
1108       break;
1109     case GST_VIDEO_FORMAT_A444_10LE:
1110     case GST_VIDEO_FORMAT_A444_10BE:
1111       info->stride[0] = GST_ROUND_UP_4 (width * 2);
1112       info->stride[1] = info->stride[0];
1113       info->stride[2] = info->stride[0];
1114       info->stride[3] = info->stride[0];
1115       info->offset[0] = 0;
1116       info->offset[1] = info->stride[0] * height;
1117       info->offset[2] = info->offset[1] * 2;
1118       info->offset[3] = info->offset[1] * 3;
1119       info->size = info->stride[0] * height * 4;
1120       break;
1121     case GST_VIDEO_FORMAT_P010_10LE:
1122     case GST_VIDEO_FORMAT_P010_10BE:
1123     case GST_VIDEO_FORMAT_P016_LE:
1124     case GST_VIDEO_FORMAT_P016_BE:
1125     case GST_VIDEO_FORMAT_P012_LE:
1126     case GST_VIDEO_FORMAT_P012_BE:
1127       info->stride[0] = GST_ROUND_UP_4 (width * 2);
1128       info->stride[1] = info->stride[0];
1129       info->offset[0] = 0;
1130       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
1131       cr_h = GST_ROUND_UP_2 (height) / 2;
1132       info->size = info->offset[1] + info->stride[0] * cr_h;
1133       break;
1134     case GST_VIDEO_FORMAT_GRAY10_LE32:
1135       info->stride[0] = (width + 2) / 3 * 4;
1136       info->offset[0] = 0;
1137       info->size = info->stride[0] * GST_ROUND_UP_2 (height);
1138       break;
1139     case GST_VIDEO_FORMAT_NV12_10LE32:
1140       info->stride[0] = (width + 2) / 3 * 4;
1141       info->stride[1] = info->stride[0];
1142       info->offset[0] = 0;
1143       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
1144       cr_h = GST_ROUND_UP_2 (height) / 2;
1145       if (GST_VIDEO_INFO_IS_INTERLACED (info))
1146         cr_h = GST_ROUND_UP_2 (cr_h);
1147       info->size = info->offset[1] + info->stride[0] * cr_h;
1148       break;
1149     case GST_VIDEO_FORMAT_NV16_10LE32:
1150       info->stride[0] = (width + 2) / 3 * 4;
1151       info->stride[1] = info->stride[0];
1152       info->offset[0] = 0;
1153       info->offset[1] = info->stride[0] * height;
1154       info->size = info->stride[0] * height * 2;
1155       break;
1156     case GST_VIDEO_FORMAT_NV12_10LE40:
1157       info->stride[0] = ((width * 5 >> 2) + 4) / 5 * 5;
1158       info->stride[1] = info->stride[0];
1159       info->offset[0] = 0;
1160       info->offset[1] = info->stride[0] * GST_ROUND_UP_2 (height);
1161       cr_h = GST_ROUND_UP_2 (height) / 2;
1162       if (GST_VIDEO_INFO_IS_INTERLACED (info))
1163         cr_h = GST_ROUND_UP_2 (cr_h);
1164       info->size = info->offset[1] + info->stride[0] * cr_h;
1165       break;
1166
1167     case GST_VIDEO_FORMAT_ENCODED:
1168       break;
1169     case GST_VIDEO_FORMAT_UNKNOWN:
1170       GST_ERROR ("invalid format");
1171       g_warning ("invalid format");
1172       return FALSE;
1173       break;
1174   }
1175
1176   if (plane_size) {
1177     for (i = 0; i < GST_VIDEO_MAX_PLANES; i++) {
1178       if (i < GST_VIDEO_INFO_N_PLANES (info)) {
1179         gint comps[GST_VIDEO_MAX_COMPONENTS];
1180         guint plane_height;
1181
1182         /* Convert plane index to component index */
1183         gst_video_format_info_component (info->finfo, i, comps);
1184         plane_height =
1185             GST_VIDEO_FORMAT_INFO_SCALE_HEIGHT (info->finfo, comps[0],
1186             GST_VIDEO_INFO_FIELD_HEIGHT (info));
1187         plane_size[i] = plane_height * GST_VIDEO_INFO_PLANE_STRIDE (info, i);
1188       } else {
1189         plane_size[i] = 0;
1190       }
1191     }
1192   }
1193
1194   return TRUE;
1195 }
1196
1197 /**
1198  * gst_video_info_convert:
1199  * @info: a #GstVideoInfo
1200  * @src_format: #GstFormat of the @src_value
1201  * @src_value: value to convert
1202  * @dest_format: #GstFormat of the @dest_value
1203  * @dest_value: (out): pointer to destination value
1204  *
1205  * Converts among various #GstFormat types.  This function handles
1206  * GST_FORMAT_BYTES, GST_FORMAT_TIME, and GST_FORMAT_DEFAULT.  For
1207  * raw video, GST_FORMAT_DEFAULT corresponds to video frames.  This
1208  * function can be used to handle pad queries of the type GST_QUERY_CONVERT.
1209  *
1210  * Returns: TRUE if the conversion was successful.
1211  */
1212 gboolean
1213 gst_video_info_convert (const GstVideoInfo * info,
1214     GstFormat src_format, gint64 src_value,
1215     GstFormat dest_format, gint64 * dest_value)
1216 {
1217   gboolean ret = FALSE;
1218   int fps_n, fps_d;
1219   gsize size;
1220
1221   g_return_val_if_fail (info != NULL, 0);
1222   g_return_val_if_fail (info->finfo != NULL, 0);
1223   g_return_val_if_fail (info->finfo->format != GST_VIDEO_FORMAT_UNKNOWN, 0);
1224   g_return_val_if_fail (info->size > 0, 0);
1225
1226   size = info->size;
1227   fps_n = info->fps_n;
1228   fps_d = info->fps_d;
1229
1230   GST_DEBUG ("converting value %" G_GINT64_FORMAT " from %s to %s",
1231       src_value, gst_format_get_name (src_format),
1232       gst_format_get_name (dest_format));
1233
1234   if (src_format == dest_format) {
1235     *dest_value = src_value;
1236     ret = TRUE;
1237     goto done;
1238   }
1239
1240   if (src_value == -1) {
1241     *dest_value = -1;
1242     ret = TRUE;
1243     goto done;
1244   }
1245
1246   /* bytes to frames */
1247   if (src_format == GST_FORMAT_BYTES && dest_format == GST_FORMAT_DEFAULT) {
1248     if (size != 0) {
1249       *dest_value = gst_util_uint64_scale (src_value, 1, size);
1250     } else {
1251       GST_ERROR ("blocksize is 0");
1252       *dest_value = 0;
1253     }
1254     ret = TRUE;
1255     goto done;
1256   }
1257
1258   /* frames to bytes */
1259   if (src_format == GST_FORMAT_DEFAULT && dest_format == GST_FORMAT_BYTES) {
1260     *dest_value = gst_util_uint64_scale (src_value, size, 1);
1261     ret = TRUE;
1262     goto done;
1263   }
1264
1265   /* time to frames */
1266   if (src_format == GST_FORMAT_TIME && dest_format == GST_FORMAT_DEFAULT) {
1267     if (fps_d != 0) {
1268       *dest_value = gst_util_uint64_scale (src_value,
1269           fps_n, GST_SECOND * fps_d);
1270     } else {
1271       GST_ERROR ("framerate denominator is 0");
1272       *dest_value = 0;
1273     }
1274     ret = TRUE;
1275     goto done;
1276   }
1277
1278   /* frames to time */
1279   if (src_format == GST_FORMAT_DEFAULT && dest_format == GST_FORMAT_TIME) {
1280     if (fps_n != 0) {
1281       *dest_value = gst_util_uint64_scale (src_value,
1282           GST_SECOND * fps_d, fps_n);
1283     } else {
1284       GST_ERROR ("framerate numerator is 0");
1285       *dest_value = 0;
1286     }
1287     ret = TRUE;
1288     goto done;
1289   }
1290
1291   /* time to bytes */
1292   if (src_format == GST_FORMAT_TIME && dest_format == GST_FORMAT_BYTES) {
1293     if (fps_d != 0) {
1294       *dest_value = gst_util_uint64_scale (src_value,
1295           fps_n * size, GST_SECOND * fps_d);
1296     } else {
1297       GST_ERROR ("framerate denominator is 0");
1298       *dest_value = 0;
1299     }
1300     ret = TRUE;
1301     goto done;
1302   }
1303
1304   /* bytes to time */
1305   if (src_format == GST_FORMAT_BYTES && dest_format == GST_FORMAT_TIME) {
1306     if (fps_n != 0 && size != 0) {
1307       *dest_value = gst_util_uint64_scale (src_value,
1308           GST_SECOND * fps_d, fps_n * size);
1309     } else {
1310       GST_ERROR ("framerate denominator and/or blocksize is 0");
1311       *dest_value = 0;
1312     }
1313     ret = TRUE;
1314   }
1315
1316 done:
1317
1318   GST_DEBUG ("ret=%d result %" G_GINT64_FORMAT, ret, *dest_value);
1319
1320   return ret;
1321 }
1322
1323 /**
1324  * gst_video_info_align_full:
1325  * @info: a #GstVideoInfo
1326  * @align: alignment parameters
1327  * @plane_size: (out) (allow-none): array used to store the plane sizes
1328  *
1329  * Extra padding will be added to the right side when stride alignment padding
1330  * is required and @align will be updated with the new padding values.
1331  *
1332  * This variant of gst_video_info_align() provides the updated size, in bytes,
1333  * of each video plane after the alignment, including all horizontal and vertical
1334  * paddings.
1335  *
1336  * In case of GST_VIDEO_INTERLACE_MODE_ALTERNATE info, the returned sizes are the
1337  * ones used to hold a single field, not the full frame.
1338  *
1339  * Returns: %FALSE if alignment could not be applied, e.g. because the
1340  *   size of a frame can't be represented as a 32 bit integer
1341  *
1342  * Since: 1.18
1343  */
1344 gboolean
1345 gst_video_info_align_full (GstVideoInfo * info, GstVideoAlignment * align,
1346     gsize plane_size[GST_VIDEO_MAX_PLANES])
1347 {
1348   const GstVideoFormatInfo *vinfo = info->finfo;
1349   gint width, height;
1350   gint padded_width, padded_height;
1351   gint i, n_planes;
1352   gboolean aligned;
1353
1354   width = GST_VIDEO_INFO_WIDTH (info);
1355   height = GST_VIDEO_INFO_HEIGHT (info);
1356
1357   GST_LOG ("padding %u-%ux%u-%u", align->padding_top,
1358       align->padding_left, align->padding_right, align->padding_bottom);
1359
1360   n_planes = GST_VIDEO_INFO_N_PLANES (info);
1361
1362   if (GST_VIDEO_FORMAT_INFO_HAS_PALETTE (vinfo))
1363     n_planes--;
1364
1365   /* first make sure the left padding does not cause alignment problems later */
1366   do {
1367     GST_LOG ("left padding %u", align->padding_left);
1368     aligned = TRUE;
1369     for (i = 0; i < n_planes; i++) {
1370       gint hedge;
1371
1372       /* this is the amount of pixels to add as left padding */
1373       hedge = GST_VIDEO_FORMAT_INFO_SCALE_WIDTH (vinfo, i, align->padding_left);
1374       hedge *= GST_VIDEO_FORMAT_INFO_PSTRIDE (vinfo, i);
1375
1376       GST_LOG ("plane %d, padding %d, alignment %u", i, hedge,
1377           align->stride_align[i]);
1378       aligned &= (hedge & align->stride_align[i]) == 0;
1379     }
1380     if (aligned)
1381       break;
1382
1383     GST_LOG ("unaligned padding, increasing padding");
1384     /* increase padded_width */
1385     align->padding_left += align->padding_left & ~(align->padding_left - 1);
1386   } while (!aligned);
1387
1388   /* add the padding */
1389   padded_width = width + align->padding_left + align->padding_right;
1390   padded_height = height + align->padding_top + align->padding_bottom;
1391
1392   do {
1393     GST_LOG ("padded dimension %u-%u", padded_width, padded_height);
1394
1395     info->width = padded_width;
1396     info->height = padded_height;
1397
1398     if (!fill_planes (info, plane_size))
1399       return FALSE;
1400
1401     /* check alignment */
1402     aligned = TRUE;
1403     for (i = 0; i < n_planes; i++) {
1404       GST_LOG ("plane %d, stride %d, alignment %u", i, info->stride[i],
1405           align->stride_align[i]);
1406       aligned &= (info->stride[i] & align->stride_align[i]) == 0;
1407     }
1408     if (aligned)
1409       break;
1410
1411     GST_LOG ("unaligned strides, increasing dimension");
1412     /* increase padded_width */
1413     padded_width += padded_width & ~(padded_width - 1);
1414   } while (!aligned);
1415
1416   align->padding_right = padded_width - width - align->padding_left;
1417
1418   info->width = width;
1419   info->height = height;
1420
1421   for (i = 0; i < n_planes; i++) {
1422     gint vedge, hedge, comp;
1423
1424     /* Find the component for this plane, FIXME, we assume the plane number and
1425      * component number is the same for now, for scaling the dimensions this is
1426      * currently true for all formats but it might not be when adding new
1427      * formats. We might need to add a plane subsamling in the format info to
1428      * make this more generic or maybe use a plane -> component mapping. */
1429     comp = i;
1430
1431     hedge =
1432         GST_VIDEO_FORMAT_INFO_SCALE_WIDTH (vinfo, comp, align->padding_left);
1433     vedge =
1434         GST_VIDEO_FORMAT_INFO_SCALE_HEIGHT (vinfo, comp, align->padding_top);
1435
1436     GST_DEBUG ("plane %d: comp: %d, hedge %d vedge %d align %d stride %d", i,
1437         comp, hedge, vedge, align->stride_align[i], info->stride[i]);
1438
1439     info->offset[i] += (vedge * info->stride[i]) +
1440         (hedge * GST_VIDEO_FORMAT_INFO_PSTRIDE (vinfo, comp));
1441   }
1442
1443   return TRUE;
1444 }
1445
1446 /**
1447  * gst_video_info_align:
1448  * @info: a #GstVideoInfo
1449  * @align: alignment parameters
1450  *
1451  * Adjust the offset and stride fields in @info so that the padding and
1452  * stride alignment in @align is respected.
1453  *
1454  * Extra padding will be added to the right side when stride alignment padding
1455  * is required and @align will be updated with the new padding values.
1456  *
1457  * Returns: %FALSE if alignment could not be applied, e.g. because the
1458  *   size of a frame can't be represented as a 32 bit integer (Since: 1.12)
1459  */
1460 gboolean
1461 gst_video_info_align (GstVideoInfo * info, GstVideoAlignment * align)
1462 {
1463   return gst_video_info_align_full (info, align, NULL);
1464 }