ext/dvdread/dvdreadsrc.c

   1 /* GStreamer DVD title source
   2  * Copyright (C) 1999 Erik Walthinsen <omega@cse.ogi.edu>
   3  * Copyright (C) 2001 Billy Biggs <vektor@dumbterm.net>.
   4  * Copyright (C) 2006 Tim-Philipp Müller <tim centricular net>
   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., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include "config.h"
  24 #endif
  25
  26 #include "_stdint.h"
  27
  28 #include <stdio.h>
  29 #include <stdlib.h>
  30 #include <unistd.h>
  31 #include <string.h>
  32 #include <errno.h>
  33
  34 #include "dvdreadsrc.h"
  35
  36 #include <gst/gst-i18n-plugin.h>
  37
  38 GST_DEBUG_CATEGORY_STATIC (gstgst_dvd_read_src_debug);
  39 #define GST_CAT_DEFAULT (gstgst_dvd_read_src_debug)
  40
  41 static void gst_dvd_read_src_do_init (GType dvdreadsrc_type);
  42
  43 enum
  44 {
  45   ARG_0,
  46   ARG_DEVICE,
  47   ARG_TITLE,
  48   ARG_CHAPTER,
  49   ARG_ANGLE
  50 };
  51
  52 static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
  53     GST_PAD_SRC,
  54     GST_PAD_ALWAYS,
  55     GST_STATIC_CAPS ("video/mpeg, mpegversion=2, systemstream=(boolean)true"));
  56
  57 static GstFormat title_format;
  58 static GstFormat angle_format;
  59 static GstFormat sector_format;
  60 static GstFormat chapter_format;
  61
  62 static gboolean gst_dvd_read_src_start (GstBaseSrc * basesrc);
  63 static gboolean gst_dvd_read_src_stop (GstBaseSrc * basesrc);
  64 static GstFlowReturn gst_dvd_read_src_create (GstPushSrc * pushsrc,
  65     GstBuffer ** buf);
  66 static gboolean gst_dvd_read_src_src_query (GstBaseSrc * basesrc,
  67     GstQuery * query);
  68 static gboolean gst_dvd_read_src_src_event (GstBaseSrc * basesrc,
  69     GstEvent * event);
  70 static gboolean gst_dvd_read_src_goto_title (GstDvdReadSrc * src, gint title,
  71     gint angle);
  72 static gboolean gst_dvd_read_src_goto_chapter (GstDvdReadSrc * src,
  73     gint chapter);
  74 static gboolean gst_dvd_read_src_goto_sector (GstDvdReadSrc * src, gint angle);
  75 static void gst_dvd_read_src_set_property (GObject * object, guint prop_id,
  76     const GValue * value, GParamSpec * pspec);
  77 static void gst_dvd_read_src_get_property (GObject * object, guint prop_id,
  78     GValue * value, GParamSpec * pspec);
  79 static GstEvent *gst_dvd_read_src_make_clut_change_event (GstDvdReadSrc * src,
  80     const guint * clut);
  81 static gboolean gst_dvd_read_src_get_size (GstDvdReadSrc * src, gint64 * size);
  82 static gboolean gst_dvd_read_src_do_seek (GstBaseSrc * src, GstSegment * s);
  83 static gint64 gst_dvd_read_src_convert_timecode (dvd_time_t * time);
  84 static gint gst_dvd_read_src_get_next_cell (GstDvdReadSrc * src,
  85     pgc_t * pgc, gint cell);
  86 static GstClockTime gst_dvd_read_src_get_time_for_sector (GstDvdReadSrc * src,
  87     guint sector);
  88 static gint gst_dvd_read_src_get_sector_from_time (GstDvdReadSrc * src,
  89     GstClockTime ts);
  90
  91 GST_BOILERPLATE_FULL (GstDvdReadSrc, gst_dvd_read_src, GstPushSrc,
  92     GST_TYPE_PUSH_SRC, gst_dvd_read_src_do_init);
  93
  94 static void
  95 gst_dvd_read_src_base_init (gpointer g_class)
  96 {
  97   GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
  98
  99   gst_element_class_add_pad_template (element_class,
 100       gst_static_pad_template_get (&srctemplate));
 101
 102   gst_element_class_set_details_simple (element_class, "DVD Source",
 103       "Source/File/DVD",
 104       "Access a DVD title/chapter/angle using libdvdread",
 105       "Erik Walthinsen <omega@cse.ogi.edu>");
 106 }
 107
 108 static void
 109 gst_dvd_read_src_finalize (GObject * object)
 110 {
 111   GstDvdReadSrc *src = GST_DVD_READ_SRC (object);
 112
 113   g_free (src->location);
 114   g_free (src->last_uri);
 115
 116   G_OBJECT_CLASS (parent_class)->finalize (object);
 117 }
 118
 119 static void
 120 gst_dvd_read_src_init (GstDvdReadSrc * src, GstDvdReadSrcClass * klass)
 121 {
 122   src->dvd = NULL;
 123   src->vts_file = NULL;
 124   src->vmg_file = NULL;
 125   src->dvd_title = NULL;
 126
 127   src->location = g_strdup ("/dev/dvd");
 128   src->last_uri = NULL;
 129   src->new_seek = TRUE;
 130   src->new_cell = TRUE;
 131   src->change_cell = FALSE;
 132   src->uri_title = 1;
 133   src->uri_chapter = 1;
 134   src->uri_angle = 1;
 135
 136   src->title_lang_event_pending = NULL;
 137   src->pending_clut_event = NULL;
 138
 139   gst_pad_use_fixed_caps (GST_BASE_SRC_PAD (src));
 140   gst_pad_set_caps (GST_BASE_SRC_PAD (src),
 141       gst_static_pad_template_get_caps (&srctemplate));
 142 }
 143
 144 static gboolean
 145 gst_dvd_read_src_is_seekable (GstBaseSrc * src)
 146 {
 147   return TRUE;
 148 }
 149
 150 static void
 151 gst_dvd_read_src_class_init (GstDvdReadSrcClass * klass)
 152 {
 153   GstPushSrcClass *gstpushsrc_class = GST_PUSH_SRC_CLASS (klass);
 154   GstBaseSrcClass *gstbasesrc_class = GST_BASE_SRC_CLASS (klass);
 155   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
 156
 157   gobject_class->finalize = gst_dvd_read_src_finalize;
 158   gobject_class->set_property = gst_dvd_read_src_set_property;
 159   gobject_class->get_property = gst_dvd_read_src_get_property;
 160
 161   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_DEVICE,
 162       g_param_spec_string ("device", "Device",
 163           "DVD device location", NULL, G_PARAM_READWRITE));
 164   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_TITLE,
 165       g_param_spec_int ("title", "title", "title",
 166           1, 999, 1, G_PARAM_READWRITE));
 167   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_CHAPTER,
 168       g_param_spec_int ("chapter", "chapter", "chapter",
 169           1, 999, 1, G_PARAM_READWRITE));
 170   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_ANGLE,
 171       g_param_spec_int ("angle", "angle", "angle",
 172           1, 999, 1, G_PARAM_READWRITE));
 173
 174   gstbasesrc_class->start = GST_DEBUG_FUNCPTR (gst_dvd_read_src_start);
 175   gstbasesrc_class->stop = GST_DEBUG_FUNCPTR (gst_dvd_read_src_stop);
 176   gstbasesrc_class->query = GST_DEBUG_FUNCPTR (gst_dvd_read_src_src_query);
 177   gstbasesrc_class->event = GST_DEBUG_FUNCPTR (gst_dvd_read_src_src_event);
 178   gstbasesrc_class->do_seek = GST_DEBUG_FUNCPTR (gst_dvd_read_src_do_seek);
 179   gstbasesrc_class->is_seekable =
 180       GST_DEBUG_FUNCPTR (gst_dvd_read_src_is_seekable);
 181
 182   gstpushsrc_class->create = GST_DEBUG_FUNCPTR (gst_dvd_read_src_create);
 183 }
 184
 185 static gboolean
 186 gst_dvd_read_src_start (GstBaseSrc * basesrc)
 187 {
 188   GstDvdReadSrc *src = GST_DVD_READ_SRC (basesrc);
 189
 190   g_return_val_if_fail (src->location != NULL, FALSE);
 191
 192   GST_DEBUG_OBJECT (src, "Opening DVD '%s'", src->location);
 193
 194   if ((src->dvd = DVDOpen (src->location)) == NULL)
 195     goto open_failed;
 196
 197   /* Load the video manager to find out the information about the titles */
 198   GST_DEBUG_OBJECT (src, "Loading VMG info");
 199
 200   if (!(src->vmg_file = ifoOpen (src->dvd, 0)))
 201     goto ifo_open_failed;
 202
 203   src->tt_srpt = src->vmg_file->tt_srpt;
 204
 205   src->title = src->uri_title - 1;
 206   src->chapter = src->uri_chapter - 1;
 207   src->angle = src->uri_angle - 1;
 208
 209   if (!gst_dvd_read_src_goto_title (src, src->title, src->angle))
 210     goto title_open_failed;
 211
 212   if (!gst_dvd_read_src_goto_chapter (src, src->chapter))
 213     goto chapter_open_failed;
 214
 215   src->new_seek = FALSE;
 216   src->change_cell = TRUE;
 217
 218   return TRUE;
 219
 220   /* ERRORS */
 221 open_failed:
 222   {
 223     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
 224         (_("Could not open DVD")),
 225         ("DVDOpen(%s) failed: %s", src->location, g_strerror (errno)));
 226     return FALSE;
 227   }
 228 ifo_open_failed:
 229   {
 230     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
 231         (_("Could not open DVD")),
 232         ("ifoOpen() failed: %s", g_strerror (errno)));
 233     return FALSE;
 234   }
 235 title_open_failed:
 236   {
 237     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
 238         (_("Could not open DVD title %d"), src->uri_title), (NULL));
 239     return FALSE;
 240   }
 241 chapter_open_failed:
 242   {
 243     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
 244         (_("Failed to go to chapter %d of DVD title %d"),
 245             src->uri_chapter, src->uri_title), (NULL));
 246     return FALSE;
 247   }
 248 }
 249
 250 static gboolean
 251 gst_dvd_read_src_stop (GstBaseSrc * basesrc)
 252 {
 253   GstDvdReadSrc *src = GST_DVD_READ_SRC (basesrc);
 254
 255   if (src->vts_file) {
 256     ifoClose (src->vts_file);
 257     src->vts_file = NULL;
 258   }
 259   if (src->vmg_file) {
 260     ifoClose (src->vmg_file);
 261     src->vmg_file = NULL;
 262   }
 263   if (src->dvd_title) {
 264     DVDCloseFile (src->dvd_title);
 265     src->dvd_title = NULL;
 266   }
 267   if (src->dvd) {
 268     DVDClose (src->dvd);
 269     src->dvd = NULL;
 270   }
 271   src->new_cell = TRUE;
 272   src->new_seek = TRUE;
 273   src->change_cell = FALSE;
 274   src->chapter = 0;
 275   src->title = 0;
 276   src->need_newsegment = TRUE;
 277   src->vts_tmapt = NULL;
 278   if (src->title_lang_event_pending) {
 279     gst_event_unref (src->title_lang_event_pending);
 280     src->title_lang_event_pending = NULL;
 281   }
 282   if (src->pending_clut_event) {
 283     gst_event_unref (src->pending_clut_event);
 284     src->pending_clut_event = NULL;
 285   }
 286   if (src->chapter_starts) {
 287     g_free (src->chapter_starts);
 288     src->chapter_starts = NULL;
 289   }
 290
 291   GST_LOG_OBJECT (src, "closed DVD");
 292
 293   return TRUE;
 294 }
 295
 296 static void
 297 cur_title_get_chapter_pgc (GstDvdReadSrc * src, gint chapter, gint * p_pgn,
 298     gint * p_pgc_id, pgc_t ** p_pgc)
 299 {
 300   pgc_t *pgc;
 301   gint pgn, pgc_id;
 302
 303   g_assert (chapter >= 0 && chapter < src->num_chapters);
 304
 305   pgc_id = src->vts_ptt_srpt->title[src->ttn - 1].ptt[chapter].pgcn;
 306   pgn = src->vts_ptt_srpt->title[src->ttn - 1].ptt[chapter].pgn;
 307   pgc = src->vts_file->vts_pgcit->pgci_srp[pgc_id - 1].pgc;
 308
 309   *p_pgn = pgn;
 310   *p_pgc_id = pgc_id;
 311   *p_pgc = pgc;
 312 }
 313
 314 static void
 315 cur_title_get_chapter_bounds (GstDvdReadSrc * src, gint chapter,
 316     gint * p_first_cell, gint * p_last_cell)
 317 {
 318   pgc_t *pgc;
 319   gint pgn, pgc_id, pgn_next_ch;
 320
 321   g_assert (chapter >= 0 && chapter < src->num_chapters);
 322
 323   cur_title_get_chapter_pgc (src, chapter, &pgn, &pgc_id, &pgc);
 324
 325   *p_first_cell = pgc->program_map[pgn - 1] - 1;
 326
 327   if (chapter == (src->num_chapters - 1)) {
 328     *p_last_cell = pgc->nr_of_cells;
 329   } else {
 330     pgn_next_ch = src->vts_ptt_srpt->title[src->ttn - 1].ptt[chapter + 1].pgn;
 331     *p_last_cell = pgc->program_map[pgn_next_ch - 1] - 1;
 332   }
 333 }
 334
 335 static gboolean
 336 gst_dvd_read_src_goto_chapter (GstDvdReadSrc * src, gint chapter)
 337 {
 338   gint i;
 339
 340   /* make sure the chapter number is valid for this title */
 341   if (chapter < 0 || chapter >= src->num_chapters) {
 342     GST_WARNING_OBJECT (src, "invalid chapter %d (only %d available)",
 343         chapter, src->num_chapters);
 344     chapter = CLAMP (chapter, 0, src->num_chapters - 1);
 345   }
 346
 347   /* determine which program chain we want to watch. This is
 348    * based on the chapter number */
 349   cur_title_get_chapter_pgc (src, chapter, &src->pgn, &src->pgc_id,
 350       &src->cur_pgc);
 351   cur_title_get_chapter_bounds (src, chapter, &src->start_cell,
 352       &src->last_cell);
 353
 354   GST_LOG_OBJECT (src, "Opened chapter %d - cell %d-%d", chapter + 1,
 355       src->start_cell, src->last_cell);
 356
 357   /* retrieve position */
 358   src->cur_pack = 0;
 359   for (i = 0; i < chapter; i++) {
 360     gint c1, c2;
 361
 362     cur_title_get_chapter_bounds (src, i, &c1, &c2);
 363
 364     while (c1 < c2) {
 365       src->cur_pack +=
 366           src->cur_pgc->cell_playback[c1].last_sector -
 367           src->cur_pgc->cell_playback[c1].first_sector;
 368       ++c1;
 369     }
 370   }
 371
 372   /* prepare reading for new cell */
 373   src->new_cell = TRUE;
 374   src->next_cell = src->start_cell;
 375
 376   src->chapter = chapter;
 377
 378   if (src->pending_clut_event)
 379     gst_event_unref (src->pending_clut_event);
 380
 381   src->pending_clut_event =
 382       gst_dvd_read_src_make_clut_change_event (src, src->cur_pgc->palette);
 383
 384   return TRUE;
 385 }
 386
 387 static void
 388 gst_dvd_read_src_get_chapter_starts (GstDvdReadSrc * src)
 389 {
 390   GstClockTime uptohere;
 391   guint c;
 392
 393   g_free (src->chapter_starts);
 394   src->chapter_starts = g_new (GstClockTime, src->num_chapters);
 395
 396   uptohere = (GstClockTime) 0;
 397   for (c = 0; c < src->num_chapters; ++c) {
 398     GstClockTime chapter_duration = 0;
 399     gint cell_start, cell_end, cell;
 400     gint pgn, pgc_id;
 401     pgc_t *pgc;
 402
 403     cur_title_get_chapter_pgc (src, c, &pgn, &pgc_id, &pgc);
 404     cur_title_get_chapter_bounds (src, c, &cell_start, &cell_end);
 405
 406     cell = cell_start;
 407     while (cell < cell_end) {
 408       dvd_time_t *cell_duration;
 409
 410       cell_duration = &pgc->cell_playback[cell].playback_time;
 411       chapter_duration += gst_dvd_read_src_convert_timecode (cell_duration);
 412       cell = gst_dvd_read_src_get_next_cell (src, pgc, cell);
 413     }
 414
 415     src->chapter_starts[c] = uptohere;
 416
 417     GST_INFO_OBJECT (src, "[%02u] Chapter %02u starts at %" GST_TIME_FORMAT
 418         ", dur = %" GST_TIME_FORMAT ", cells %d-%d", src->title + 1, c + 1,
 419         GST_TIME_ARGS (uptohere), GST_TIME_ARGS (chapter_duration),
 420         cell_start, cell_end);
 421
 422     uptohere += chapter_duration;
 423   }
 424 }
 425
 426 static gboolean
 427 gst_dvd_read_src_goto_title (GstDvdReadSrc * src, gint title, gint angle)
 428 {
 429   GstStructure *s;
 430   gchar lang_code[3] = { '\0', '\0', '\0' }, *t;
 431   pgc_t *pgc0;
 432   gint title_set_nr;
 433   gint num_titles;
 434   gint pgn0, pgc0_id;
 435   gint i;
 436
 437   /* make sure our title number is valid */
 438   num_titles = src->tt_srpt->nr_of_srpts;
 439   GST_INFO_OBJECT (src, "There are %d titles on this DVD", num_titles);
 440   if (title < 0 || title >= num_titles)
 441     goto invalid_title;
 442
 443   src->num_chapters = src->tt_srpt->title[title].nr_of_ptts;
 444   GST_INFO_OBJECT (src, "Title %d has %d chapters", title + 1,
 445       src->num_chapters);
 446
 447   /* make sure the angle number is valid for this title */
 448   src->num_angles = src->tt_srpt->title[title].nr_of_angles;
 449   GST_LOG_OBJECT (src, "Title %d has %d angles", title + 1, src->num_angles);
 450   if (angle < 0 || angle >= src->num_angles) {
 451     GST_WARNING_OBJECT (src, "Invalid angle %d (only %d available)",
 452         angle, src->num_angles);
 453     angle = CLAMP (angle, 0, src->num_angles - 1);
 454   }
 455
 456   /* load the VTS information for the title set our title is in */
 457   title_set_nr = src->tt_srpt->title[title].title_set_nr;
 458   src->vts_file = ifoOpen (src->dvd, title_set_nr);
 459   if (src->vts_file == NULL)
 460     goto ifo_open_failed;
 461
 462   src->ttn = src->tt_srpt->title[title].vts_ttn;
 463   src->vts_ptt_srpt = src->vts_file->vts_ptt_srpt;
 464
 465   /* interactive title? */
 466   if (src->num_chapters > 0 &&
 467       src->vts_ptt_srpt->title[src->ttn - 1].ptt[0].pgn == 0) {
 468     goto commands_only_pgc;
 469   }
 470
 471   /* we've got enough info, time to open the title set data */
 472   src->dvd_title = DVDOpenFile (src->dvd, title_set_nr, DVD_READ_TITLE_VOBS);
 473   if (src->dvd_title == NULL)
 474     goto title_open_failed;
 475
 476   GST_INFO_OBJECT (src, "Opened title %d, angle %d", title + 1, angle);
 477   src->title = title;
 478   src->angle = angle;
 479
 480   /* build event */
 481
 482   if (src->title_lang_event_pending) {
 483     gst_event_unref (src->title_lang_event_pending);
 484     src->title_lang_event_pending = NULL;
 485   }
 486
 487   s = gst_structure_new ("application/x-gst-dvd",
 488       "event", G_TYPE_STRING, "dvd-lang-codes", NULL);
 489
 490   /* so we can filter out invalid/unused streams (same for all chapters) */
 491   cur_title_get_chapter_pgc (src, 0, &pgn0, &pgc0_id, &pgc0);
 492
 493   /* audio */
 494   for (i = 0; i < src->vts_file->vtsi_mat->nr_of_vts_audio_streams; i++) {
 495     const audio_attr_t *a;
 496
 497     /* audio stream present? */
 498     if (pgc0 != NULL && (pgc0->audio_control[i] & 0x8000) == 0)
 499       continue;
 500
 501     a = &src->vts_file->vtsi_mat->vts_audio_attr[i];
 502
 503     t = g_strdup_printf ("audio-%d-format", i);
 504     gst_structure_set (s, t, G_TYPE_INT, (int) a->audio_format, NULL);
 505     g_free (t);
 506
 507     if (a->lang_type) {
 508       t = g_strdup_printf ("audio-%d-language", i);
 509       lang_code[0] = (a->lang_code >> 8) & 0xff;
 510       lang_code[1] = a->lang_code & 0xff;
 511       gst_structure_set (s, t, G_TYPE_STRING, lang_code, NULL);
 512       g_free (t);
 513     } else {
 514       lang_code[0] = '\0';
 515     }
 516
 517     GST_INFO_OBJECT (src, "[%02d] Audio    %02d: lang='%s', format=%d",
 518         src->title + 1, i, lang_code, (gint) a->audio_format);
 519   }
 520
 521   /* subtitle */
 522   for (i = 0; i < src->vts_file->vtsi_mat->nr_of_vts_subp_streams; i++) {
 523     const subp_attr_t *u;
 524
 525     /* subpicture stream present? */
 526     if (pgc0 != NULL && (pgc0->subp_control[i] & 0x80000000) == 0)
 527       continue;
 528
 529     u = &src->vts_file->vtsi_mat->vts_subp_attr[i];
 530
 531     if (u->type) {
 532       t = g_strdup_printf ("subtitle-%d-language", i);
 533       lang_code[0] = (u->lang_code >> 8) & 0xff;
 534       lang_code[1] = u->lang_code & 0xff;
 535       gst_structure_set (s, t, G_TYPE_STRING, lang_code, NULL);
 536       g_free (t);
 537     } else {
 538       lang_code[0] = '\0';
 539     }
 540
 541     GST_INFO_OBJECT (src, "[%02d] Subtitle %02d: lang='%s', type=%d",
 542         src->title + 1, i, lang_code, u->type);
 543   }
 544
 545   src->title_lang_event_pending =
 546       gst_event_new_custom (GST_EVENT_CUSTOM_DOWNSTREAM, s);
 547
 548   /* dump seek tables */
 549   src->vts_tmapt = src->vts_file->vts_tmapt;
 550   if (src->vts_tmapt) {
 551     gint i, j;
 552
 553     GST_LOG_OBJECT (src, "nr_of_tmaps = %d", src->vts_tmapt->nr_of_tmaps);
 554     for (i = 0; i < src->vts_tmapt->nr_of_tmaps; ++i) {
 555       GST_LOG_OBJECT (src, "======= Table %d ===================", i);
 556       GST_LOG_OBJECT (src, "Offset relative to VTS_TMAPTI: %d",
 557           src->vts_tmapt->tmap_offset[i]);
 558       GST_LOG_OBJECT (src, "Time unit (seconds)          : %d",
 559           src->vts_tmapt->tmap[i].tmu);
 560       GST_LOG_OBJECT (src, "Number of entries            : %d",
 561           src->vts_tmapt->tmap[i].nr_of_entries);
 562       for (j = 0; j < src->vts_tmapt->tmap[i].nr_of_entries; j++) {
 563         guint64 time;
 564
 565         time = src->vts_tmapt->tmap[i].tmu * (j + 1) * GST_SECOND;
 566         GST_LOG_OBJECT (src, "Time: %" GST_TIME_FORMAT " VOBU "
 567             "Sector: 0x%08x %s", GST_TIME_ARGS (time),
 568             src->vts_tmapt->tmap[i].map_ent[j] & 0x7fffffff,
 569             (src->vts_tmapt->tmap[i].map_ent[j] >> 31) ? "discontinuity" : "");
 570       }
 571     }
 572   } else {
 573     GST_WARNING_OBJECT (src, "no vts_tmapt - seeking will suck");
 574   }
 575
 576   gst_dvd_read_src_get_chapter_starts (src);
 577
 578   return TRUE;
 579
 580   /* ERRORS */
 581 invalid_title:
 582   {
 583     GST_WARNING_OBJECT (src, "Invalid title %d (only %d available)",
 584         title, num_titles);
 585     return FALSE;
 586   }
 587 ifo_open_failed:
 588   {
 589     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
 590         (_("Could not open DVD title %d"), title_set_nr),
 591         ("ifoOpen(%d) failed: %s", title_set_nr, g_strerror (errno)));
 592     return FALSE;
 593   }
 594 title_open_failed:
 595   {
 596     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
 597         (_("Could not open DVD title %d"), title_set_nr),
 598         ("Can't open title VOBS (VTS_%02d_1.VOB)", title_set_nr));
 599     return FALSE;
 600   }
 601 commands_only_pgc:
 602   {
 603     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
 604         (_("Could not open DVD title %d. Interactive titles are not supported "
 605                 "by this element"), title_set_nr),
 606         ("Commands-only PGC, not supported, use rsndvdbin"));
 607     return FALSE;
 608   }
 609 }
 610
 611 /* FIXME: double-check this function, compare against original */
 612 static gint
 613 gst_dvd_read_src_get_next_cell (GstDvdReadSrc * src, pgc_t * pgc, gint cell)
 614 {
 615   /* Check if we're entering an angle block. */
 616   if (pgc->cell_playback[cell].block_type != BLOCK_TYPE_ANGLE_BLOCK)
 617     return (cell + 1);
 618
 619   while (pgc->cell_playback[cell].block_mode == BLOCK_MODE_LAST_CELL)
 620     ++cell;
 621
 622   return cell + 1;              /* really +1? (tpm) */
 623 }
 624
 625 /* Returns true if the pack is a NAV pack */
 626 static gboolean
 627 gst_dvd_read_src_is_nav_pack (const guint8 * data, gint lbn, dsi_t * dsi_pack)
 628 {
 629   if (GST_READ_UINT32_BE (data + 0x26) != 0x000001BF)
 630     return FALSE;
 631
 632   /* Check that this is substream 0 (PCI) */
 633   if (data[0x2c] != 0)
 634     return FALSE;
 635
 636   if (GST_READ_UINT32_BE (data + 0x400) != 0x000001BF)
 637     return FALSE;
 638
 639   /* Check that this is substream 1 (DSI) */
 640   if (data[0x406] != 1)
 641     return FALSE;
 642
 643   /* Check sizes of PCI and DSI packets */
 644   if (GST_READ_UINT16_BE (data + 0x2a) != 0x03d4)
 645     return FALSE;
 646
 647   if (GST_READ_UINT16_BE (data + 0x404) != 0x03fa)
 648     return FALSE;
 649
 650   /* Read the DSI packet into the provided struct and check it */
 651   navRead_DSI (dsi_pack, (unsigned char *) data + DSI_START_BYTE);
 652   if (lbn != dsi_pack->dsi_gi.nv_pck_lbn)
 653     return FALSE;
 654
 655   return TRUE;
 656 }
 657
 658 /* find time for sector from index, returns NONE if there is no exact match */
 659 static GstClockTime
 660 gst_dvd_read_src_get_time_for_sector (GstDvdReadSrc * src, guint sector)
 661 {
 662   gint i, j;
 663
 664   if (src->vts_tmapt == NULL || src->vts_tmapt->nr_of_tmaps == 0)
 665     return GST_CLOCK_TIME_NONE;
 666
 667   for (i = 0; i < src->vts_tmapt->nr_of_tmaps; ++i) {
 668     for (j = 0; j < src->vts_tmapt->tmap[i].nr_of_entries; ++j) {
 669       if ((src->vts_tmapt->tmap[i].map_ent[j] & 0x7fffffff) == sector)
 670         return src->vts_tmapt->tmap[i].tmu * (j + 1) * GST_SECOND;
 671     }
 672   }
 673
 674   if (sector == 0)
 675     return (GstClockTime) 0;
 676
 677   return GST_CLOCK_TIME_NONE;
 678 }
 679
 680 /* returns the sector in the index at (or before) the given time, or -1 */
 681 static gint
 682 gst_dvd_read_src_get_sector_from_time (GstDvdReadSrc * src, GstClockTime ts)
 683 {
 684   gint sector, i, j;
 685
 686   if (src->vts_tmapt == NULL || src->vts_tmapt->nr_of_tmaps == 0)
 687     return -1;
 688
 689   sector = 0;
 690   for (i = 0; i < src->vts_tmapt->nr_of_tmaps; ++i) {
 691     for (j = 0; j < src->vts_tmapt->tmap[i].nr_of_entries; ++j) {
 692       GstClockTime entry_time;
 693
 694       entry_time = src->vts_tmapt->tmap[i].tmu * (j + 1) * GST_SECOND;
 695       if (entry_time <= ts) {
 696         sector = src->vts_tmapt->tmap[i].map_ent[j] & 0x7fffffff;
 697       }
 698       if (entry_time >= ts) {
 699         return sector;
 700       }
 701     }
 702   }
 703
 704   if (ts == 0)
 705     return 0;
 706
 707   return -1;
 708 }
 709
 710 typedef enum
 711 {
 712   GST_DVD_READ_OK = 0,
 713   GST_DVD_READ_ERROR = -1,
 714   GST_DVD_READ_EOS = -2,
 715   GST_DVD_READ_AGAIN = -3
 716 } GstDvdReadReturn;
 717
 718 static GstDvdReadReturn
 719 gst_dvd_read_src_read (GstDvdReadSrc * src, gint angle, gint new_seek,
 720     GstBuffer ** p_buf)
 721 {
 722   GstBuffer *buf;
 723   GstSegment *seg;
 724   guint8 oneblock[DVD_VIDEO_LB_LEN];
 725   dsi_t dsi_pack;
 726   guint next_vobu, next_ilvu_start, cur_output_size;
 727   gint len;
 728   gint retries;
 729   gint64 next_time;
 730
 731   seg = &(GST_BASE_SRC (src)->segment);
 732
 733   /* playback by cell in this pgc, starting at the cell for our chapter */
 734   if (new_seek)
 735     src->cur_cell = src->start_cell;
 736
 737 again:
 738
 739   if (src->cur_cell >= src->last_cell) {
 740     /* advance to next chapter */
 741     if (src->chapter == (src->num_chapters - 1) ||
 742         (seg->format == chapter_format && seg->stop != -1 &&
 743             src->chapter == (seg->stop - 1))) {
 744       GST_DEBUG_OBJECT (src, "end of chapter segment");
 745       goto eos;
 746     }
 747
 748     GST_INFO_OBJECT (src, "end of chapter %d, switch to next",
 749         src->chapter + 1);
 750
 751     ++src->chapter;
 752     gst_dvd_read_src_goto_chapter (src, src->chapter);
 753
 754     return GST_DVD_READ_AGAIN;
 755   }
 756
 757   if (src->new_cell || new_seek) {
 758     if (!new_seek) {
 759       src->cur_cell = src->next_cell;
 760       if (src->cur_cell >= src->last_cell) {
 761         GST_LOG_OBJECT (src, "last cell in chapter");
 762         goto again;
 763       }
 764     }
 765
 766     /* take angle into account */
 767     if (src->cur_pgc->cell_playback[src->cur_cell].block_type
 768         == BLOCK_TYPE_ANGLE_BLOCK)
 769       src->cur_cell += angle;
 770
 771     /* calculate next cell */
 772     src->next_cell =
 773         gst_dvd_read_src_get_next_cell (src, src->cur_pgc, src->cur_cell);
 774
 775     /* we loop until we're out of this cell */
 776     src->cur_pack = src->cur_pgc->cell_playback[src->cur_cell].first_sector;
 777     src->new_cell = FALSE;
 778     GST_DEBUG_OBJECT (src, "Starting new cell %d @ pack %d", src->cur_cell,
 779         src->cur_pack);
 780   }
 781
 782   if (src->cur_pack >= src->cur_pgc->cell_playback[src->cur_cell].last_sector) {
 783     src->new_cell = TRUE;
 784     GST_LOG_OBJECT (src, "Beyond last sector for cell %d, going to next cell",
 785         src->cur_cell);
 786     return GST_DVD_READ_AGAIN;
 787   }
 788
 789   /* read NAV packet */
 790   retries = 0;
 791 nav_retry:
 792   retries++;
 793
 794   len = DVDReadBlocks (src->dvd_title, src->cur_pack, 1, oneblock);
 795   if (len != 1)
 796     goto read_error;
 797
 798   if (!gst_dvd_read_src_is_nav_pack (oneblock, src->cur_pack, &dsi_pack)) {
 799     GST_LOG_OBJECT (src, "Skipping nav packet @ pack %d", src->cur_pack);
 800     src->cur_pack++;
 801
 802     if (retries < 2000) {
 803       goto nav_retry;
 804     } else {
 805       GST_LOG_OBJECT (src, "No nav packet @ pack %d after 2000 blocks",
 806           src->cur_pack);
 807       goto read_error;
 808     }
 809   }
 810
 811   /* determine where we go next. These values are the ones we
 812    * mostly care about */
 813   next_ilvu_start = src->cur_pack + dsi_pack.sml_agli.data[angle].address;
 814   cur_output_size = dsi_pack.dsi_gi.vobu_ea + 1;
 815
 816   /* If we're not at the end of this cell, we can determine the next
 817    * VOBU to display using the VOBU_SRI information section of the
 818    * DSI.  Using this value correctly follows the current angle,
 819    * avoiding the doubled scenes in The Matrix, and makes our life
 820    * really happy.
 821    *
 822    * Otherwise, we set our next address past the end of this cell to
 823    * force the code above to go to the next cell in the program. */
 824   if (dsi_pack.vobu_sri.next_vobu != SRI_END_OF_CELL) {
 825     next_vobu = src->cur_pack + (dsi_pack.vobu_sri.next_vobu & 0x7fffffff);
 826   } else {
 827     next_vobu = src->cur_pgc->cell_playback[src->cur_cell].last_sector + 1;
 828   }
 829
 830   g_assert (cur_output_size < 1024);
 831
 832   /* create the buffer (TODO: use buffer pool?) */
 833   buf = gst_buffer_new_and_alloc (cur_output_size * DVD_VIDEO_LB_LEN);
 834
 835   GST_LOG_OBJECT (src, "Going to read %u sectors @ pack %d", cur_output_size,
 836       src->cur_pack);
 837
 838   /* read in and output cursize packs */
 839   len = DVDReadBlocks (src->dvd_title, src->cur_pack, cur_output_size,
 840       GST_BUFFER_DATA (buf));
 841
 842   if (len != cur_output_size)
 843     goto block_read_error;
 844
 845   GST_BUFFER_SIZE (buf) = cur_output_size * DVD_VIDEO_LB_LEN;
 846   /* GST_BUFFER_OFFSET (buf) = priv->cur_pack * DVD_VIDEO_LB_LEN; */
 847   GST_BUFFER_TIMESTAMP (buf) =
 848       gst_dvd_read_src_get_time_for_sector (src, src->cur_pack);
 849
 850   gst_buffer_set_caps (buf, GST_PAD_CAPS (GST_BASE_SRC_PAD (src)));
 851
 852   *p_buf = buf;
 853
 854   GST_LOG_OBJECT (src, "Read %u sectors", cur_output_size);
 855
 856   src->cur_pack = next_vobu;
 857
 858   next_time = GST_BUFFER_TIMESTAMP (buf);
 859   if (GST_CLOCK_TIME_IS_VALID (next_time) && seg->format == GST_FORMAT_TIME &&
 860       GST_CLOCK_TIME_IS_VALID (seg->stop) &&
 861       next_time > seg->stop + 5 * GST_SECOND) {
 862     GST_DEBUG_OBJECT (src, "end of TIME segment");
 863     goto eos;
 864   }
 865
 866   return GST_DVD_READ_OK;
 867
 868   /* ERRORS */
 869 eos:
 870   {
 871     GST_INFO_OBJECT (src, "Reached end-of-segment/stream - EOS");
 872     return GST_DVD_READ_EOS;
 873   }
 874 read_error:
 875   {
 876     GST_ERROR_OBJECT (src, "Read failed for block %d", src->cur_pack);
 877     return GST_DVD_READ_ERROR;
 878   }
 879 block_read_error:
 880   {
 881     GST_ERROR_OBJECT (src, "Read failed for %d blocks at %d",
 882         cur_output_size, src->cur_pack);
 883     gst_buffer_unref (buf);
 884     return GST_DVD_READ_ERROR;
 885   }
 886 }
 887
 888 static GstFlowReturn
 889 gst_dvd_read_src_create (GstPushSrc * pushsrc, GstBuffer ** p_buf)
 890 {
 891   GstDvdReadSrc *src = GST_DVD_READ_SRC (pushsrc);
 892   GstPad *srcpad;
 893   gint res;
 894
 895   g_return_val_if_fail (src->dvd != NULL, GST_FLOW_ERROR);
 896
 897   srcpad = GST_BASE_SRC (src)->srcpad;
 898
 899   if (src->need_newsegment) {
 900     gst_pad_push_event (srcpad,
 901         gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES,
 902             (gint64) src->cur_pack * DVD_VIDEO_LB_LEN, -1, 0));
 903     src->need_newsegment = FALSE;
 904   }
 905
 906   if (src->new_seek) {
 907     gst_dvd_read_src_goto_title (src, src->title, src->angle);
 908     gst_dvd_read_src_goto_chapter (src, src->chapter);
 909
 910     src->new_seek = FALSE;
 911     src->change_cell = TRUE;
 912   }
 913
 914   if (src->title_lang_event_pending) {
 915     gst_pad_push_event (srcpad, src->title_lang_event_pending);
 916     src->title_lang_event_pending = NULL;
 917   }
 918
 919   if (src->pending_clut_event) {
 920     gst_pad_push_event (srcpad, src->pending_clut_event);
 921     src->pending_clut_event = NULL;
 922   }
 923
 924   /* read it in */
 925   do {
 926     res = gst_dvd_read_src_read (src, src->angle, src->change_cell, p_buf);
 927   } while (res == GST_DVD_READ_AGAIN);
 928
 929   switch (res) {
 930     case GST_DVD_READ_ERROR:{
 931       GST_ELEMENT_ERROR (src, RESOURCE, READ, (NULL), (NULL));
 932       return GST_FLOW_ERROR;
 933     }
 934     case GST_DVD_READ_EOS:{
 935       return GST_FLOW_UNEXPECTED;
 936     }
 937     case GST_DVD_READ_OK:{
 938       src->change_cell = FALSE;
 939       return GST_FLOW_OK;
 940     }
 941     default:
 942       break;
 943   }
 944
 945   g_return_val_if_reached (GST_FLOW_UNEXPECTED);
 946 }
 947
 948 static void
 949 gst_dvd_read_src_set_property (GObject * object, guint prop_id,
 950     const GValue * value, GParamSpec * pspec)
 951 {
 952   GstDvdReadSrc *src = GST_DVD_READ_SRC (object);
 953   gboolean started;
 954
 955   GST_OBJECT_LOCK (src);
 956   started = GST_OBJECT_FLAG_IS_SET (src, GST_BASE_SRC_STARTED);
 957
 958   switch (prop_id) {
 959     case ARG_DEVICE:{
 960       if (started) {
 961         g_warning ("%s: property '%s' needs to be set before the device is "
 962             "opened", GST_ELEMENT_NAME (src), pspec->name);
 963         break;;
 964       }
 965
 966       g_free (src->location);
 967       /* clear the filename if we get a NULL (is that possible?) */
 968       if (g_value_get_string (value) == NULL) {
 969         src->location = g_strdup ("/dev/dvd");
 970       } else {
 971         src->location = g_strdup (g_value_get_string (value));
 972       }
 973       break;
 974     }
 975     case ARG_TITLE:
 976       src->uri_title = g_value_get_int (value);
 977       if (started) {
 978         src->title = src->uri_title - 1;
 979         src->new_seek = TRUE;
 980       }
 981       break;
 982     case ARG_CHAPTER:
 983       src->uri_chapter = g_value_get_int (value);
 984       if (started) {
 985         src->chapter = src->uri_chapter - 1;
 986         src->new_seek = TRUE;
 987       }
 988       break;
 989     case ARG_ANGLE:
 990       src->uri_angle = g_value_get_int (value);
 991       if (started) {
 992         src->angle = src->uri_angle - 1;
 993       }
 994       break;
 995     default:
 996       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 997       break;
 998   }
 999
1000   GST_OBJECT_UNLOCK (src);
1001 }
1002
1003 static void
1004 gst_dvd_read_src_get_property (GObject * object, guint prop_id, GValue * value,
1005     GParamSpec * pspec)
1006 {
1007   GstDvdReadSrc *src = GST_DVD_READ_SRC (object);
1008
1009   GST_OBJECT_LOCK (src);
1010
1011   switch (prop_id) {
1012     case ARG_DEVICE:
1013       g_value_set_string (value, src->location);
1014       break;
1015     case ARG_TITLE:
1016       g_value_set_int (value, src->uri_title);
1017       break;
1018     case ARG_CHAPTER:
1019       g_value_set_int (value, src->uri_chapter);
1020       break;
1021     case ARG_ANGLE:
1022       g_value_set_int (value, src->uri_angle);
1023       break;
1024     default:
1025       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
1026       break;
1027   }
1028
1029   GST_OBJECT_UNLOCK (src);
1030 }
1031
1032 static gboolean
1033 gst_dvd_read_src_get_size (GstDvdReadSrc * src, gint64 * size)
1034 {
1035   gboolean ret = FALSE;
1036
1037   if (src->dvd_title) {
1038     gsize blocks;
1039
1040     blocks = DVDFileSize (src->dvd_title);
1041     if (blocks >= 0) {
1042       *size = (gint64) blocks *DVD_VIDEO_LB_LEN;
1043
1044       ret = TRUE;
1045     } else {
1046       GST_WARNING_OBJECT (src, "DVDFileSize(%p) failed!", src->dvd_title);
1047     }
1048   }
1049
1050   return ret;
1051 }
1052
1053 /*** Querying and seeking ***/
1054
1055 static gboolean
1056 gst_dvd_read_src_handle_seek_event (GstDvdReadSrc * src, GstEvent * event)
1057 {
1058   GstSeekFlags flags;
1059   GstSeekType cur_type, end_type;
1060   gint64 new_off, total;
1061   GstFormat format;
1062   GstPad *srcpad;
1063   gboolean query_ok;
1064   gdouble rate;
1065
1066   gst_event_parse_seek (event, &rate, &format, &flags, &cur_type, &new_off,
1067       &end_type, NULL);
1068
1069   if (rate <= 0.0) {
1070     GST_DEBUG_OBJECT (src, "cannot do backwards playback yet");
1071     return FALSE;
1072   }
1073
1074   if (end_type != GST_SEEK_TYPE_NONE) {
1075     if ((format != chapter_format && format != GST_FORMAT_TIME) ||
1076         end_type != GST_SEEK_TYPE_SET) {
1077       GST_DEBUG_OBJECT (src, "end seek type not supported");
1078       return FALSE;
1079     }
1080   }
1081
1082   if (cur_type != GST_SEEK_TYPE_SET) {
1083     GST_DEBUG_OBJECT (src, "only SEEK_TYPE_SET is supported");
1084     return FALSE;
1085   }
1086
1087   if (format == angle_format) {
1088     GST_OBJECT_LOCK (src);
1089     if (new_off < 0 || new_off >= src->num_angles) {
1090       GST_OBJECT_UNLOCK (src);
1091       GST_DEBUG_OBJECT (src, "invalid angle %d, only %d available",
1092           src->num_angles, src->num_angles);
1093       return FALSE;
1094     }
1095     src->angle = (gint) new_off;
1096     GST_OBJECT_UNLOCK (src);
1097     GST_DEBUG_OBJECT (src, "switched to angle %d", (gint) new_off + 1);
1098     return TRUE;
1099   }
1100
1101   if (format != chapter_format && format != title_format &&
1102       format != GST_FORMAT_BYTES && format != GST_FORMAT_TIME) {
1103     GST_DEBUG_OBJECT (src, "unsupported seek format %d (%s)", format,
1104         gst_format_get_name (format));
1105     return FALSE;
1106   }
1107
1108   if (format == GST_FORMAT_BYTES) {
1109     GST_DEBUG_OBJECT (src, "Requested seek to byte %" G_GUINT64_FORMAT,
1110         new_off);
1111   } else if (format == GST_FORMAT_TIME) {
1112     GST_DEBUG_OBJECT (src, "Requested seek to time %" GST_TIME_FORMAT,
1113         GST_TIME_ARGS (new_off));
1114     if (gst_dvd_read_src_get_sector_from_time (src, new_off) < 0) {
1115       GST_DEBUG_OBJECT (src, "Can't find sector for requested time");
1116       return FALSE;
1117     }
1118   }
1119
1120   srcpad = GST_BASE_SRC_PAD (src);
1121
1122   /* check whether the seek looks reasonable (ie within possible range) */
1123   if (format == GST_FORMAT_BYTES) {
1124     GST_OBJECT_LOCK (src);
1125     query_ok = gst_dvd_read_src_get_size (src, &total);
1126     GST_OBJECT_UNLOCK (src);
1127   } else {
1128     query_ok = gst_pad_query_duration (srcpad, &format, &total);
1129   }
1130
1131   if (!query_ok) {
1132     GST_DEBUG_OBJECT (src, "Failed to query duration in format %s",
1133         gst_format_get_name (format));
1134     return FALSE;
1135   }
1136
1137   GST_DEBUG_OBJECT (src, "Total      %s: %12" G_GINT64_FORMAT,
1138       gst_format_get_name (format), total);
1139   GST_DEBUG_OBJECT (src, "Seek to    %s: %12" G_GINT64_FORMAT,
1140       gst_format_get_name (format), new_off);
1141
1142   if (new_off >= total) {
1143     GST_DEBUG_OBJECT (src, "Seek position out of range");
1144     return FALSE;
1145   }
1146
1147   /* set segment to seek format; this allows us to use the do_seek
1148    * virtual function and let the base source handle all the tricky
1149    * stuff for us. We don't use the segment internally anyway */
1150   /* FIXME: can't take the stream lock here - what to do? */
1151   GST_OBJECT_LOCK (src);
1152   GST_BASE_SRC (src)->segment.format = format;
1153   GST_BASE_SRC (src)->segment.start = 0;
1154   GST_BASE_SRC (src)->segment.stop = total;
1155   GST_BASE_SRC (src)->segment.duration = total;
1156   GST_OBJECT_UNLOCK (src);
1157
1158   return GST_BASE_SRC_CLASS (parent_class)->event (GST_BASE_SRC (src), event);
1159 }
1160
1161 static gboolean
1162 gst_dvd_read_src_do_seek (GstBaseSrc * basesrc, GstSegment * s)
1163 {
1164   GstDvdReadSrc *src;
1165
1166   src = GST_DVD_READ_SRC (basesrc);
1167
1168   GST_DEBUG_OBJECT (src, "Seeking to %s: %12" G_GINT64_FORMAT,
1169       gst_format_get_name (s->format), s->last_stop);
1170
1171   if (s->format == sector_format || s->format == GST_FORMAT_BYTES
1172       || s->format == GST_FORMAT_TIME) {
1173     guint old;
1174
1175     old = src->cur_pack;
1176
1177     if (s->format == sector_format) {
1178       src->cur_pack = s->last_stop;
1179     } else if (s->format == GST_FORMAT_TIME) {
1180       gint sector;
1181
1182       sector = gst_dvd_read_src_get_sector_from_time (src, s->last_stop);
1183
1184       GST_DEBUG_OBJECT (src, "Time %" GST_TIME_FORMAT " => sector %d",
1185           GST_TIME_ARGS (s->last_stop), sector);
1186
1187       /* really shouldn't happen, we've checked this earlier ... */
1188       g_return_val_if_fail (sector >= 0, FALSE);
1189
1190       src->cur_pack = sector;
1191     } else {
1192       /* byte format */
1193       src->cur_pack = s->last_stop / DVD_VIDEO_LB_LEN;
1194       if (((gint64) src->cur_pack * DVD_VIDEO_LB_LEN) != s->last_stop) {
1195         GST_LOG_OBJECT (src, "rounded down offset %" G_GINT64_FORMAT " => %"
1196             G_GINT64_FORMAT, s->last_stop,
1197             (gint64) src->cur_pack * DVD_VIDEO_LB_LEN);
1198       }
1199     }
1200
1201     if (!gst_dvd_read_src_goto_sector (src, src->angle)) {
1202       GST_DEBUG_OBJECT (src, "seek to sector 0x%08x failed", src->cur_pack);
1203       src->cur_pack = old;
1204       return FALSE;
1205     }
1206
1207     GST_LOG_OBJECT (src, "seek to sector 0x%08x ok", src->cur_pack);
1208   } else if (s->format == chapter_format) {
1209     if (!gst_dvd_read_src_goto_chapter (src, (gint) s->last_stop)) {
1210       GST_DEBUG_OBJECT (src, "seek to chapter %d failed",
1211           (gint) s->last_stop + 1);
1212       return FALSE;
1213     }
1214     GST_INFO_OBJECT (src, "seek to chapter %d ok", (gint) s->last_stop + 1);
1215     src->chapter = s->last_stop;
1216   } else if (s->format == title_format) {
1217     if (!gst_dvd_read_src_goto_title (src, (gint) s->last_stop, src->angle) ||
1218         !gst_dvd_read_src_goto_chapter (src, 0)) {
1219       GST_DEBUG_OBJECT (src, "seek to title %d failed", (gint) s->last_stop);
1220       return FALSE;
1221     }
1222     src->title = (gint) s->last_stop;
1223     src->chapter = 0;
1224     GST_INFO_OBJECT (src, "seek to title %d ok", src->title + 1);
1225   } else {
1226     g_return_val_if_reached (FALSE);
1227   }
1228
1229   src->need_newsegment = TRUE;
1230   return TRUE;
1231 }
1232
1233 static gboolean
1234 gst_dvd_read_src_src_event (GstBaseSrc * basesrc, GstEvent * event)
1235 {
1236   GstDvdReadSrc *src = GST_DVD_READ_SRC (basesrc);
1237   gboolean res;
1238
1239   GST_LOG_OBJECT (src, "handling %s event", GST_EVENT_TYPE_NAME (event));
1240
1241   switch (GST_EVENT_TYPE (event)) {
1242     case GST_EVENT_SEEK:
1243       res = gst_dvd_read_src_handle_seek_event (src, event);
1244       break;
1245     default:
1246       res = GST_BASE_SRC_CLASS (parent_class)->event (basesrc, event);
1247       break;
1248   }
1249
1250   return res;
1251 }
1252
1253 static GstEvent *
1254 gst_dvd_read_src_make_clut_change_event (GstDvdReadSrc * src,
1255     const guint * clut)
1256 {
1257   GstStructure *structure;
1258   gchar name[16];
1259   gint i;
1260
1261   structure = gst_structure_new ("application/x-gst-dvd",
1262       "event", G_TYPE_STRING, "dvd-spu-clut-change", NULL);
1263
1264   /* Create a separate field for each value in the table. */
1265   for (i = 0; i < 16; i++) {
1266     g_snprintf (name, sizeof (name), "clut%02d", i);
1267     gst_structure_set (structure, name, G_TYPE_INT, (int) clut[i], NULL);
1268   }
1269
1270   /* Create the DVD event and put the structure into it. */
1271   return gst_event_new_custom (GST_EVENT_CUSTOM_DOWNSTREAM, structure);
1272 }
1273
1274 static gint64
1275 gst_dvd_read_src_convert_timecode (dvd_time_t * time)
1276 {
1277   gint64 ret_time;
1278   const gint64 one_hour = 3600 * GST_SECOND;
1279   const gint64 one_min = 60 * GST_SECOND;
1280
1281   g_return_val_if_fail ((time->hour >> 4) < 0xa
1282       && (time->hour & 0xf) < 0xa, -1);
1283   g_return_val_if_fail ((time->minute >> 4) < 0x7
1284       && (time->minute & 0xf) < 0xa, -1);
1285   g_return_val_if_fail ((time->second >> 4) < 0x7
1286       && (time->second & 0xf) < 0xa, -1);
1287
1288   ret_time = ((time->hour >> 4) * 10 + (time->hour & 0xf)) * one_hour;
1289   ret_time += ((time->minute >> 4) * 10 + (time->minute & 0xf)) * one_min;
1290   ret_time += ((time->second >> 4) * 10 + (time->second & 0xf)) * GST_SECOND;
1291
1292   return ret_time;
1293 }
1294
1295 static gboolean
1296 gst_dvd_read_src_do_duration_query (GstDvdReadSrc * src, GstQuery * query)
1297 {
1298   GstFormat format;
1299   gint64 val;
1300
1301   gst_query_parse_duration (query, &format, NULL);
1302
1303   switch (format) {
1304     case GST_FORMAT_TIME:{
1305       if (src->cur_pgc == NULL)
1306         return FALSE;
1307       val = gst_dvd_read_src_convert_timecode (&src->cur_pgc->playback_time);
1308       if (val < 0)
1309         return FALSE;
1310       break;
1311     }
1312     case GST_FORMAT_BYTES:{
1313       if (!gst_dvd_read_src_get_size (src, &val))
1314         return FALSE;
1315       break;
1316     }
1317     default:{
1318       if (format == sector_format) {
1319         val = DVDFileSize (src->dvd_title);
1320       } else if (format == title_format) {
1321         val = src->tt_srpt->nr_of_srpts;
1322       } else if (format == chapter_format) {
1323         val = src->num_chapters;
1324       } else if (format == angle_format) {
1325         val = src->tt_srpt->title[src->title].nr_of_angles;
1326       } else {
1327         GST_DEBUG_OBJECT (src, "Don't know how to handle format %d (%s)",
1328             format, gst_format_get_name (format));
1329         return FALSE;
1330       }
1331       break;
1332     }
1333   }
1334
1335   GST_LOG_OBJECT (src, "duration = %" G_GINT64_FORMAT " %s", val,
1336       gst_format_get_name (format));
1337
1338   gst_query_set_duration (query, format, val);
1339   return TRUE;
1340 }
1341
1342 static gboolean
1343 gst_dvd_read_src_do_position_query (GstDvdReadSrc * src, GstQuery * query)
1344 {
1345   GstFormat format;
1346   gint64 val;
1347
1348   gst_query_parse_position (query, &format, NULL);
1349
1350   switch (format) {
1351     case GST_FORMAT_BYTES:{
1352       val = (gint64) src->cur_pack * DVD_VIDEO_LB_LEN;
1353       break;
1354     }
1355     default:{
1356       if (format == sector_format) {
1357         val = src->cur_pack;
1358       } else if (format == title_format) {
1359         val = src->title;
1360       } else if (format == chapter_format) {
1361         val = src->chapter;
1362       } else if (format == angle_format) {
1363         val = src->angle;
1364       } else {
1365         GST_DEBUG_OBJECT (src, "Don't know how to handle format %d (%s)",
1366             format, gst_format_get_name (format));
1367         return FALSE;
1368       }
1369       break;
1370     }
1371   }
1372
1373   GST_LOG_OBJECT (src, "position = %" G_GINT64_FORMAT " %s", val,
1374       gst_format_get_name (format));
1375
1376   gst_query_set_position (query, format, val);
1377   return TRUE;
1378 }
1379
1380 static gboolean
1381 gst_dvd_read_src_do_convert_query (GstDvdReadSrc * src, GstQuery * query)
1382 {
1383   GstFormat src_format, dest_format;
1384   gboolean ret = FALSE;
1385   gint64 src_val, dest_val = -1;
1386
1387   gst_query_parse_convert (query, &src_format, &src_val, &dest_format, NULL);
1388
1389   if (src_format == dest_format) {
1390     dest_val = src_val;
1391     ret = TRUE;
1392     goto done;
1393   }
1394
1395   /* Formats to consider: TIME, DEFAULT, BYTES, title, chapter, sector.
1396    * Note: title and chapter are counted as starting from 0 here, just like
1397    * in the context of seek events. Another note: DEFAULT format is undefined */
1398
1399   if (src_format == GST_FORMAT_BYTES) {
1400     src_format = sector_format;
1401     src_val /= DVD_VIDEO_LB_LEN;
1402   }
1403
1404   if (src_format == sector_format) {
1405     /* SECTOR => xyz */
1406     if (dest_format == GST_FORMAT_TIME && src_val < G_MAXUINT) {
1407       dest_val = gst_dvd_read_src_get_time_for_sector (src, (guint) src_val);
1408       ret = (dest_val >= 0);
1409     } else if (dest_format == GST_FORMAT_BYTES) {
1410       dest_val = src_val * DVD_VIDEO_LB_LEN;
1411       ret = TRUE;
1412     } else {
1413       ret = FALSE;
1414     }
1415   } else if (src_format == title_format) {
1416     /* TITLE => xyz */
1417     if (dest_format == GST_FORMAT_TIME) {
1418       /* not really true, but we use this to trick the base source into
1419        * handling seeks in title-format for us (the source won't know that
1420        * we changed the title in this case) (changing titles should really
1421        * be done with an interface rather than a seek, but for now we're
1422        * stuck with this mechanism. Fix in 0.11) */
1423       dest_val = (GstClockTime) 0;
1424       ret = TRUE;
1425     } else {
1426       ret = FALSE;
1427     }
1428   } else if (src_format == chapter_format) {
1429     /* CHAPTER => xyz */
1430     if (dest_format == GST_FORMAT_TIME) {
1431       if (src->num_chapters >= 0 && src_val < src->num_chapters) {
1432         dest_val = src->chapter_starts[src_val];
1433         ret = TRUE;
1434       }
1435     } else if (dest_format == sector_format) {
1436     } else {
1437       ret = FALSE;
1438     }
1439   } else if (src_format == GST_FORMAT_TIME) {
1440     /* TIME => xyz */
1441     if (dest_format == sector_format || dest_format == GST_FORMAT_BYTES) {
1442       dest_val = gst_dvd_read_src_get_sector_from_time (src, src_val);
1443       ret = (dest_val >= 0);
1444       if (dest_format == GST_FORMAT_BYTES)
1445         dest_val *= DVD_VIDEO_LB_LEN;
1446     } else if (dest_format == chapter_format) {
1447       if (src->chapter_starts != NULL) {
1448         gint i;
1449
1450         for (i = src->num_chapters - 1; i >= 0; --i) {
1451           if (src->chapter_starts && src->chapter_starts[i] >= src_val) {
1452             dest_val = i;
1453             ret = TRUE;
1454             break;
1455           }
1456         }
1457       } else {
1458         ret = FALSE;
1459       }
1460     } else {
1461       ret = FALSE;
1462     }
1463   } else {
1464     ret = FALSE;
1465   }
1466
1467 done:
1468
1469   if (ret) {
1470     gst_query_set_convert (query, src_format, src_val, dest_format, dest_val);
1471   }
1472
1473   return ret;
1474 }
1475
1476 static gboolean
1477 gst_dvd_read_src_src_query (GstBaseSrc * basesrc, GstQuery * query)
1478 {
1479   GstDvdReadSrc *src = GST_DVD_READ_SRC (basesrc);
1480   gboolean started;
1481   gboolean res = TRUE;
1482
1483   GST_LOG_OBJECT (src, "handling %s query",
1484       gst_query_type_get_name (GST_QUERY_TYPE (query)));
1485
1486   GST_OBJECT_LOCK (src);
1487   started = (GST_OBJECT_FLAG_IS_SET (src, GST_BASE_SRC_STARTED));
1488   GST_OBJECT_UNLOCK (src);
1489
1490   if (!started) {
1491     GST_DEBUG_OBJECT (src, "query failed: not started");
1492     return FALSE;
1493   }
1494
1495   switch (GST_QUERY_TYPE (query)) {
1496     case GST_QUERY_DURATION:
1497       GST_OBJECT_LOCK (src);
1498       res = gst_dvd_read_src_do_duration_query (src, query);
1499       GST_OBJECT_UNLOCK (src);
1500       break;
1501     case GST_QUERY_POSITION:
1502       GST_OBJECT_LOCK (src);
1503       res = gst_dvd_read_src_do_position_query (src, query);
1504       GST_OBJECT_UNLOCK (src);
1505       break;
1506     case GST_QUERY_CONVERT:
1507       GST_OBJECT_LOCK (src);
1508       res = gst_dvd_read_src_do_convert_query (src, query);
1509       GST_OBJECT_UNLOCK (src);
1510       break;
1511     default:
1512       res = GST_BASE_SRC_CLASS (parent_class)->query (basesrc, query);
1513       break;
1514   }
1515
1516   return res;
1517 }
1518
1519 static gboolean
1520 gst_dvd_read_src_goto_sector (GstDvdReadSrc * src, int angle)
1521 {
1522   gint seek_to = src->cur_pack;
1523   gint chapter, sectors, next, cur, i;
1524
1525   /* retrieve position */
1526   src->cur_pack = 0;
1527   for (i = 0; i < src->num_chapters; i++) {
1528     gint c1, c2;
1529
1530     cur_title_get_chapter_bounds (src, i, &c1, &c2);
1531
1532     for (next = cur = c1; cur < c2;) {
1533       if (next != cur) {
1534         sectors =
1535             src->cur_pgc->cell_playback[cur].last_sector -
1536             src->cur_pgc->cell_playback[cur].first_sector;
1537         if (src->cur_pack + sectors > seek_to) {
1538           chapter = i;
1539           goto done;
1540         }
1541         src->cur_pack += sectors;
1542       }
1543       cur = next;
1544       if (src->cur_pgc->cell_playback[cur].block_type == BLOCK_TYPE_ANGLE_BLOCK)
1545         cur += angle;
1546       next = gst_dvd_read_src_get_next_cell (src, src->cur_pgc, cur);
1547     }
1548   }
1549
1550   GST_DEBUG_OBJECT (src, "Seek to sector %u failed", seek_to);
1551
1552   return FALSE;
1553
1554 done:
1555   {
1556     /* so chapter $chapter and cell $cur contain our sector
1557      * of interest. Let's go there! */
1558     GST_INFO_OBJECT (src, "Seek succeeded, going to chapter %u, cell %u",
1559         chapter + 1, cur);
1560
1561     gst_dvd_read_src_goto_chapter (src, chapter);
1562     src->cur_cell = cur;
1563     src->next_cell = next;
1564     src->new_cell = FALSE;
1565     src->cur_pack = seek_to;
1566
1567     return TRUE;
1568   }
1569 }
1570
1571
1572 /*** URI interface ***/
1573
1574 static GstURIType
1575 gst_dvd_read_src_uri_get_type (void)
1576 {
1577   return GST_URI_SRC;
1578 }
1579
1580 static gchar **
1581 gst_dvd_read_src_uri_get_protocols (void)
1582 {
1583   static gchar *protocols[] = { "dvd", NULL };
1584
1585   return protocols;
1586 }
1587
1588 static const gchar *
1589 gst_dvd_read_src_uri_get_uri (GstURIHandler * handler)
1590 {
1591   GstDvdReadSrc *src = GST_DVD_READ_SRC (handler);
1592
1593   GST_OBJECT_LOCK (src);
1594
1595   g_free (src->last_uri);
1596   src->last_uri = g_strdup_printf ("dvd://%d,%d,%d", src->uri_title,
1597       src->uri_chapter, src->uri_angle);
1598
1599   GST_OBJECT_UNLOCK (src);
1600
1601   return src->last_uri;
1602 }
1603
1604 static gboolean
1605 gst_dvd_read_src_uri_set_uri (GstURIHandler * handler, const gchar * uri)
1606 {
1607   GstDvdReadSrc *src = GST_DVD_READ_SRC (handler);
1608   gboolean ret;
1609   gchar *protocol;
1610
1611   protocol = gst_uri_get_protocol (uri);
1612   ret = (protocol != NULL && g_str_equal (protocol, "dvd"));
1613   g_free (protocol);
1614   protocol = NULL;
1615
1616   if (!ret)
1617     return ret;
1618
1619   /* parse out the new t/c/a and seek to them */
1620   {
1621     gchar *location = NULL;
1622     gchar **strs;
1623     gchar **strcur;
1624     gint pos = 0;
1625
1626     location = gst_uri_get_location (uri);
1627
1628     if (!location)
1629       return ret;
1630
1631     GST_OBJECT_LOCK (src);
1632
1633     src->uri_title = 1;
1634     src->uri_chapter = 1;
1635     src->uri_angle = 1;
1636
1637     strcur = strs = g_strsplit (location, ",", 0);
1638     while (strcur && *strcur) {
1639       gint val;
1640
1641       if (!sscanf (*strcur, "%d", &val))
1642         break;
1643
1644       if (val <= 0) {
1645         g_warning ("Invalid value %d in URI '%s'. Must be 1 or greater",
1646             val, location);
1647         break;
1648       }
1649
1650       switch (pos) {
1651         case 0:
1652           src->uri_title = val;
1653           break;
1654         case 1:
1655           src->uri_chapter = val;
1656           break;
1657         case 2:
1658           src->uri_angle = val;
1659           break;
1660       }
1661
1662       strcur++;
1663       pos++;
1664     }
1665
1666     if (pos > 0 && GST_OBJECT_FLAG_IS_SET (src, GST_BASE_SRC_STARTED)) {
1667       src->title = src->uri_title - 1;
1668       src->chapter = src->uri_chapter - 1;
1669       src->angle = src->uri_angle - 1;
1670       src->new_seek = TRUE;
1671     }
1672
1673     GST_OBJECT_UNLOCK (src);
1674
1675     g_strfreev (strs);
1676     g_free (location);
1677   }
1678
1679   return ret;
1680 }
1681
1682 static void
1683 gst_dvd_read_src_uri_handler_init (gpointer g_iface, gpointer iface_data)
1684 {
1685   GstURIHandlerInterface *iface = (GstURIHandlerInterface *) g_iface;
1686
1687   iface->get_type = gst_dvd_read_src_uri_get_type;
1688   iface->get_protocols = gst_dvd_read_src_uri_get_protocols;
1689   iface->get_uri = gst_dvd_read_src_uri_get_uri;
1690   iface->set_uri = gst_dvd_read_src_uri_set_uri;
1691 }
1692
1693 static void
1694 gst_dvd_read_src_do_init (GType dvdreadsrc_type)
1695 {
1696   static const GInterfaceInfo urihandler_info = {
1697     gst_dvd_read_src_uri_handler_init,
1698     NULL,
1699     NULL
1700   };
1701
1702   g_type_add_interface_static (dvdreadsrc_type, GST_TYPE_URI_HANDLER,
1703       &urihandler_info);
1704
1705   title_format = gst_format_register ("title", "DVD title");
1706   angle_format = gst_format_register ("angle", "DVD angle");
1707   sector_format = gst_format_register ("sector", "DVD sector");
1708   chapter_format = gst_format_register ("chapter", "DVD chapter");
1709 }
1710
1711 static gboolean
1712 plugin_init (GstPlugin * plugin)
1713 {
1714   GST_DEBUG_CATEGORY_INIT (gstgst_dvd_read_src_debug, "dvdreadsrc", 0,
1715       "DVD reader element based on dvdreadsrc");
1716
1717 #ifdef ENABLE_NLS
1718   GST_DEBUG ("binding text domain %s to locale dir %s", GETTEXT_PACKAGE,
1719       LOCALEDIR);
1720   bindtextdomain (GETTEXT_PACKAGE, LOCALEDIR);
1721   bind_textdomain_codeset (GETTEXT_PACKAGE, "UTF-8");
1722 #endif /* ENABLE_NLS */
1723
1724   if (!gst_element_register (plugin, "dvdreadsrc", GST_RANK_SECONDARY,
1725           GST_TYPE_DVD_READ_SRC)) {
1726     return FALSE;
1727   }
1728
1729   return TRUE;
1730 }
1731
1732 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
1733     GST_VERSION_MINOR,
1734     "dvdread",
1735     "Access a DVD with dvdread",
1736     plugin_init, VERSION, "GPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);