ext/raw1394/gstdv1394src.c

   1 /* GStreamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  *               <2000> Daniel Fischer <dan@f3c.com>
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Library General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Library General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Library General Public
  16  * License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  */
  20
  21 #ifdef HAVE_CONFIG_H
  22 #include "config.h"
  23 #endif
  24 #include <gst/gst.h>
  25 #include <string.h>
  26 #include "gstdv1394src.h"
  27
  28 #define N_BUFFERS_IN_POOL 3
  29
  30 #define PAL_FRAMESIZE 144000
  31 #define NTSC_FRAMESIZE 120000
  32
  33 /* Filter signals and args */
  34 enum {
  35   /* FILL ME */
  36   LAST_SIGNAL
  37 };
  38
  39 enum {
  40   ARG_0,
  41   ARG_CONSECUTIVE,
  42   ARG_SKIP,
  43   ARG_DROP_INCOMPLETE,
  44 };
  45
  46 static GstElementDetails gst_dv1394src_details = GST_ELEMENT_DETAILS (
  47   "Firewire (1394) DV Source",
  48   "Source/Video",
  49   "Source for DV video data from firewire port",
  50   "Erik Walthinsen <omega@temple-baptist.com>\n"
  51   "Daniel Fischer <dan@f3c.com>"
  52 );
  53
  54 #if 0
  55 static GstPadTemplate*
  56 gst_dv1394src_factory (void)
  57 {
  58   static GstPadTemplate *template = NULL;
  59
  60   if (!template) {
  61     template = gst_pad_template_new (
  62       "src",
  63       GST_PAD_SRC,
  64       GST_PAD_ALWAYS,
  65       GST_CAPS_NEW (
  66         "dv1394src",
  67         "video/dv",
  68 /*
  69         gst_props_new (
  70           "format", GST_PROPS_LIST (
  71                         GST_PROPS_STRING ("NTSC"),
  72                         GST_PROPS_STRING ("PAL")
  73                         ),
  74           NULL)
  75         ),
  76 */
  77       NULL);
  78   }
  79   return template;
  80 }
  81 #endif
  82
  83 static void             gst_dv1394src_base_init         (gpointer g_class);
  84 static void             gst_dv1394src_class_init                (GstDV1394SrcClass *klass);
  85 static void             gst_dv1394src_init              (GstDV1394Src *filter);
  86
  87 static void             gst_dv1394src_set_property              (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec);
  88 static void             gst_dv1394src_get_property              (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec);
  89
  90 static GstElementStateReturn    gst_dv1394src_change_state      (GstElement *element);
  91
  92 static GstData *        gst_dv1394src_get                       (GstPad *pad);
  93
  94 static GstElementClass *parent_class = NULL;
  95 /*static guint gst_filter_signals[LAST_SIGNAL] = { 0 }; */
  96
  97 GType
  98 gst_dv1394src_get_type(void) {
  99   static GType gst_dv1394src_type = 0;
 100
 101   if (!gst_dv1394src_type) {
 102     static const GTypeInfo gst_dv1394src_info = {
 103       sizeof(GstDV1394Src),
 104       gst_dv1394src_base_init,
 105       NULL,
 106       (GClassInitFunc)gst_dv1394src_class_init,
 107       NULL,
 108       NULL,
 109       sizeof(GstDV1394Src),
 110       0,
 111       (GInstanceInitFunc)gst_dv1394src_init,
 112     };
 113     gst_dv1394src_type = g_type_register_static(GST_TYPE_ELEMENT, "DV1394Src", &gst_dv1394src_info, 0);
 114   }
 115   return gst_dv1394src_type;
 116 }
 117
 118 static void
 119 gst_dv1394src_base_init (gpointer g_class)
 120 {
 121   GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
 122
 123   gst_element_class_set_details (element_class, &gst_dv1394src_details);
 124 }
 125
 126 static void
 127 gst_dv1394src_class_init (GstDV1394SrcClass *klass)
 128 {
 129   GObjectClass *gobject_class;
 130   GstElementClass *gstelement_class;
 131
 132   gobject_class = (GObjectClass*)klass;
 133   gstelement_class = (GstElementClass*)klass;
 134
 135   g_object_class_install_property( G_OBJECT_CLASS(klass), ARG_CONSECUTIVE,
 136         g_param_spec_int("consecutive","consecutive frames","send n consecutive frames after skipping",
 137                         1, G_MAXINT,1,G_PARAM_READWRITE));
 138   g_object_class_install_property( G_OBJECT_CLASS(klass), ARG_SKIP,
 139         g_param_spec_int("skip","skip frames","skip n frames",
 140                         0, G_MAXINT,1,G_PARAM_READWRITE));
 141   g_object_class_install_property( G_OBJECT_CLASS(klass), ARG_DROP_INCOMPLETE,
 142         g_param_spec_boolean("drop_incomplete","drop_incomplete","drop incomplete frames",
 143                         TRUE, G_PARAM_READWRITE));
 144
 145   parent_class = g_type_class_ref(GST_TYPE_ELEMENT);
 146
 147   gobject_class->set_property = gst_dv1394src_set_property;
 148   gobject_class->get_property = gst_dv1394src_get_property;
 149
 150   gstelement_class->change_state = gst_dv1394src_change_state;
 151 }
 152
 153 static void
 154 gst_dv1394src_init (GstDV1394Src *dv1394src)
 155 {
 156   dv1394src->srcpad = gst_pad_new ("src", GST_PAD_SRC);
 157   gst_pad_set_get_function (dv1394src->srcpad, gst_dv1394src_get);
 158   gst_element_add_pad (GST_ELEMENT (dv1394src), dv1394src->srcpad);
 159
 160   dv1394src->card = 0;
 161   dv1394src->port = 0;
 162   dv1394src->channel = 63;
 163
 164   dv1394src->consecutive = 1;
 165   dv1394src->skip = 0;
 166   dv1394src->drop_incomplete = TRUE;
 167
 168   /* initialized when first header received */
 169   dv1394src->frameSize=0;
 170   dv1394src->pool = NULL;
 171
 172   dv1394src->buf = NULL;
 173   dv1394src->frame = NULL;
 174   dv1394src->frameSequence = 0;
 175 }
 176
 177 static void
 178 gst_dv1394src_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
 179 {
 180   GstDV1394Src *filter;
 181
 182   /* it's not null if we got it, but it might not be ours */
 183   g_return_if_fail(GST_IS_DV1394SRC(object));
 184   filter = GST_DV1394SRC(object);
 185
 186   switch (prop_id) {
 187     case ARG_SKIP:
 188         filter->skip = g_value_get_int(value);
 189         break;
 190     case ARG_CONSECUTIVE:
 191         filter->consecutive = g_value_get_int(value);
 192         break;
 193     case ARG_DROP_INCOMPLETE:
 194         filter->drop_incomplete = g_value_get_boolean(value);
 195         break;
 196     default:
 197       break;
 198   }
 199 }
 200
 201 static void
 202 gst_dv1394src_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
 203 {
 204   GstDV1394Src *filter;
 205
 206   /* it's not null if we got it, but it might not be ours */
 207   g_return_if_fail(GST_IS_DV1394SRC(object));
 208   filter = GST_DV1394SRC(object);
 209
 210   switch (prop_id) {
 211     case ARG_SKIP:
 212         g_value_set_int( value, filter->skip );
 213         break;
 214     case ARG_CONSECUTIVE:
 215         g_value_set_int( value, filter->consecutive );
 216         break;
 217     case ARG_DROP_INCOMPLETE:
 218         g_value_set_boolean( value, filter->drop_incomplete );
 219         break;
 220     default:
 221       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 222       break;
 223   }
 224 }
 225
 226 static
 227 int gst_dv1394src_iso_receive(raw1394handle_t handle,int channel,size_t len,quadlet_t *data) {
 228   GstDV1394Src *dv1394src = GST_DV1394SRC (raw1394_get_userdata(handle));
 229
 230   if (len > 16) {
 231         /*
 232                  the following code taken from kino-0.51 (Dan Dennedy/Charles Yates)
 233         */
 234         unsigned char *p = (unsigned char*) & data[3];
 235         int section_type = p[0] >> 5;           /* section type is in bits 5 - 7 */
 236         int dif_sequence = p[1] >> 4;           /* dif sequence number is in bits 4 - 7 */
 237         int dif_block = p[2];
 238
 239         /* if we are at the beginning of a frame,
 240           we set buf=frame, and get a new buffer from pool for frame
 241         */
 242
 243         if (section_type == 0 && dif_sequence == 0) {   // dif header
 244
 245           if( dv1394src->pool == NULL ) {
 246             // figure format (NTSC/PAL)
 247             if( p[3] & 0x80 ) {
 248               // PAL
 249               dv1394src->frameSize = PAL_FRAMESIZE;
 250               GST_DEBUG ("PAL data");
 251               if (gst_pad_try_set_caps (dv1394src->srcpad,
 252                         GST_CAPS_NEW ( "dv1394src", "video/dv",
 253                                       "format", GST_PROPS_STRING("PAL"),
 254                                       NULL)
 255               ) <= 0) {
 256                 gst_element_error (GST_ELEMENT(dv1394src), "Could not set source caps for PAL");
 257                 return 0;
 258               }
 259             } else {
 260               // NTSC (untested)
 261               dv1394src->frameSize = NTSC_FRAMESIZE;
 262               GST_DEBUG ("NTSC data [untested] - please report success/failure to <dan@f3c.com>");
 263               if (gst_pad_try_set_caps (dv1394src->srcpad,
 264                         GST_CAPS_NEW ( "dv1394src", "video/dv",
 265                                       "format", GST_PROPS_STRING ("NTSC"),
 266                                       NULL)
 267               ) <= 0) {
 268                 gst_element_error (GST_ELEMENT(dv1394src), "Could not set source caps for NTSC");
 269                 return 0;
 270               }
 271             }
 272
 273             dv1394src->pool = gst_buffer_pool_get_default( dv1394src->frameSize, N_BUFFERS_IN_POOL );
 274             if (dv1394src->pool == NULL) {
 275               gst_element_error (GST_ELEMENT(dv1394src), "gst_buffer_pool_get_default returned NULL");
 276             }
 277           }
 278
 279           // drop last frame when not complete
 280           if( !dv1394src->drop_incomplete || dv1394src->bytesInFrame == dv1394src->frameSize ) {
 281             dv1394src->buf = dv1394src->frame;
 282           } else {
 283             GST_INFO_OBJECT (GST_ELEMENT(dv1394src), "incomplete frame dropped");
 284           }
 285           dv1394src->frame = NULL;
 286
 287           dv1394src->frameSequence++;
 288
 289           if( dv1394src->frameSequence % (dv1394src->skip+dv1394src->consecutive) < dv1394src->consecutive ) {
 290             if( dv1394src->pool ) dv1394src->frame = gst_buffer_new_from_pool( dv1394src->pool, 0, dv1394src->frameSize );
 291               dv1394src->bytesInFrame = 0;
 292             }
 293           }
 294
 295           if (dv1394src->frame != NULL) {
 296             void *data = GST_BUFFER_DATA( dv1394src->frame );
 297
 298
 299             switch (section_type) {
 300             case 0: /* 1 Header block */
 301                 /* p[3] |= 0x80; // hack to force PAL data */
 302                 memcpy(data + dif_sequence * 150 * 80, p, 480);
 303                 break;
 304
 305             case 1: /* 2 Subcode blocks */
 306                 memcpy(data + dif_sequence * 150 * 80 + (1 + dif_block) * 80, p, 480);
 307                 break;
 308
 309             case 2: /* 3 VAUX blocks */
 310                 memcpy(data + dif_sequence * 150 * 80 + (3 + dif_block) * 80, p, 480);
 311                 break;
 312
 313             case 3: /* 9 Audio blocks interleaved with video */
 314                 memcpy(data + dif_sequence * 150 * 80 + (6 + dif_block * 16) * 80, p, 480);
 315                 break;
 316
 317             case 4: /* 135 Video blocks interleaved with audio */
 318                 memcpy(data + dif_sequence * 150 * 80 + (7 + (dif_block / 15) + dif_block) * 80, p, 480);
 319                 break;
 320
 321             default: /* we can´t handle any other data */
 322                 break;
 323             }
 324             dv1394src->bytesInFrame += 480;
 325         }
 326   }
 327
 328   return 0;
 329 }
 330
 331 static
 332 int gst_dv1394src_bus_reset(raw1394handle_t handle,
 333                             unsigned int generation) {
 334   GST_INFO_OBJECT (GST_DV1394SRC(raw1394_get_userdata(handle)),"have bus reset");
 335   return 0;
 336 }
 337
 338 static GstData *
 339 gst_dv1394src_get (GstPad *pad)
 340 {
 341   GstDV1394Src *dv1394src = GST_DV1394SRC (GST_PAD_PARENT(pad));
 342
 343   dv1394src->buf = NULL;
 344   while (dv1394src->buf == NULL)
 345     raw1394_loop_iterate(dv1394src->handle);
 346
 347   return GST_DATA(dv1394src->buf);
 348 }
 349
 350 static GstElementStateReturn
 351 gst_dv1394src_change_state (GstElement *element)
 352 {
 353   GstDV1394Src *dv1394src;
 354
 355   g_return_val_if_fail (GST_IS_DV1394SRC (element), GST_STATE_FAILURE);
 356   dv1394src = GST_DV1394SRC (element);
 357
 358   switch (GST_STATE_TRANSITION (element)) {
 359     case GST_STATE_NULL_TO_READY:
 360       if ((dv1394src->handle = raw1394_new_handle()) == NULL) {
 361         GST_INFO_OBJECT (dv1394src,"can't get raw1394 handle");
 362         return GST_STATE_FAILURE;
 363       }
 364       raw1394_set_userdata(dv1394src->handle,dv1394src);
 365       dv1394src->numcards = raw1394_get_port_info(dv1394src->handle,dv1394src->pinfo,16);
 366       if (dv1394src->numcards == 0) {
 367         GST_INFO_OBJECT (dv1394src,"no cards available for raw1394");
 368         return GST_STATE_FAILURE;
 369       }
 370       if (dv1394src->pinfo[dv1394src->card].nodes <= 1) {
 371         GST_INFO_OBJECT (dv1394src,"there are no nodes on the 1394 bus");
 372         return GST_STATE_FAILURE;
 373       }
 374       if (raw1394_set_port(dv1394src->handle,dv1394src->port) < 0) {
 375         GST_INFO_OBJECT (dv1394src,"can't set 1394 port %d",dv1394src->port);
 376         return GST_STATE_FAILURE;
 377       }
 378       raw1394_set_iso_handler(dv1394src->handle,dv1394src->channel,gst_dv1394src_iso_receive);
 379       raw1394_set_bus_reset_handler(dv1394src->handle,gst_dv1394src_bus_reset);
 380       dv1394src->started = FALSE;
 381       GST_DEBUG ("successfully opened up 1394 connection");
 382       break;
 383     case GST_STATE_PAUSED_TO_PLAYING:
 384       if (raw1394_start_iso_rcv(dv1394src->handle,dv1394src->channel) < 0) {
 385         GST_INFO_OBJECT (dv1394src,"can't start 1394 iso receive");
 386         return GST_STATE_FAILURE;
 387       }
 388       break;
 389     case GST_STATE_PLAYING_TO_PAUSED:
 390       raw1394_stop_iso_rcv(dv1394src->handle, dv1394src->channel);
 391       break;
 392     case GST_STATE_READY_TO_NULL:
 393       if (dv1394src->pool != NULL) {
 394         gst_buffer_pool_unref(dv1394src->pool);
 395         dv1394src->pool = NULL;
 396       }
 397       raw1394_destroy_handle(dv1394src->handle);
 398       break;
 399     default:
 400       break;
 401   }
 402
 403   /* if we haven't failed already, give the parent class a chance to ;-) */
 404   if (GST_ELEMENT_CLASS (parent_class)->change_state)
 405     return GST_ELEMENT_CLASS (parent_class)->change_state (element);
 406
 407   return GST_STATE_SUCCESS;
 408 }