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_STATIC_CAPS (
  66         "dv1394src",
  67         "video/dv",
  68 /*
  69         gst_props_new (
  70           "format", GST_PROPS_LIST (
  71                         G_TYPE_STRING ("NTSC"),
  72                         G_TYPE_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
 171   dv1394src->buf = NULL;
 172   dv1394src->frame = NULL;
 173   dv1394src->frameSequence = 0;
 174 }
 175
 176 static void
 177 gst_dv1394src_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
 178 {
 179   GstDV1394Src *filter;
 180
 181   /* it's not null if we got it, but it might not be ours */
 182   g_return_if_fail(GST_IS_DV1394SRC(object));
 183   filter = GST_DV1394SRC(object);
 184
 185   switch (prop_id) {
 186     case ARG_SKIP:
 187         filter->skip = g_value_get_int(value);
 188         break;
 189     case ARG_CONSECUTIVE:
 190         filter->consecutive = g_value_get_int(value);
 191         break;
 192     case ARG_DROP_INCOMPLETE:
 193         filter->drop_incomplete = g_value_get_boolean(value);
 194         break;
 195     default:
 196       break;
 197   }
 198 }
 199
 200 static void
 201 gst_dv1394src_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
 202 {
 203   GstDV1394Src *filter;
 204
 205   /* it's not null if we got it, but it might not be ours */
 206   g_return_if_fail(GST_IS_DV1394SRC(object));
 207   filter = GST_DV1394SRC(object);
 208
 209   switch (prop_id) {
 210     case ARG_SKIP:
 211         g_value_set_int( value, filter->skip );
 212         break;
 213     case ARG_CONSECUTIVE:
 214         g_value_set_int( value, filter->consecutive );
 215         break;
 216     case ARG_DROP_INCOMPLETE:
 217         g_value_set_boolean( value, filter->drop_incomplete );
 218         break;
 219     default:
 220       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 221       break;
 222   }
 223 }
 224
 225 static
 226 int gst_dv1394src_iso_receive(raw1394handle_t handle,int channel,size_t len,quadlet_t *data) {
 227   GstDV1394Src *dv1394src = GST_DV1394SRC (raw1394_get_userdata(handle));
 228
 229   if (len > 16) {
 230         /*
 231                  the following code taken from kino-0.51 (Dan Dennedy/Charles Yates)
 232         */
 233         unsigned char *p = (unsigned char*) & data[3];
 234         int section_type = p[0] >> 5;           /* section type is in bits 5 - 7 */
 235         int dif_sequence = p[1] >> 4;           /* dif sequence number is in bits 4 - 7 */
 236         int dif_block = p[2];
 237
 238         /* if we are at the beginning of a frame,
 239           we set buf=frame, and alloc a new buffer for frame
 240         */
 241
 242         if (section_type == 0 && dif_sequence == 0) {   // dif header
 243
 244           if( !dv1394src->negotiated) {
 245             // figure format (NTSC/PAL)
 246             if( p[3] & 0x80 ) {
 247               // PAL
 248               dv1394src->frameSize = PAL_FRAMESIZE;
 249               GST_DEBUG ("PAL data");
 250               if (gst_pad_try_set_caps (dv1394src->srcpad,
 251                     gst_caps_new_simple ("video/dv",
 252                       "format", G_TYPE_STRING, "PAL", NULL)) <= 0) {
 253                 GST_ELEMENT_ERROR (dv1394src, CORE, NEGOTIATION, (NULL),
 254                                    ("Could not set source caps for PAL"));
 255                 return 0;
 256               }
 257             } else {
 258               // NTSC (untested)
 259               dv1394src->frameSize = NTSC_FRAMESIZE;
 260               GST_DEBUG ("NTSC data [untested] - please report success/failure to <dan@f3c.com>");
 261               if (gst_pad_try_set_caps (dv1394src->srcpad,
 262                     gst_caps_new_simple ("video/dv",
 263                       "format", G_TYPE_STRING, "NTSC", NULL)) <= 0) {
 264                 GST_ELEMENT_ERROR (dv1394src, CORE, NEGOTIATION, (NULL),
 265                                    ("Could not set source caps for NTSC"));
 266                 return 0;
 267               }
 268             }
 269             dv1394src->negotiated = TRUE;
 270           }
 271
 272           // drop last frame when not complete
 273           if( !dv1394src->drop_incomplete || dv1394src->bytesInFrame == dv1394src->frameSize ) {
 274             dv1394src->buf = dv1394src->frame;
 275           } else {
 276             GST_INFO_OBJECT (GST_ELEMENT(dv1394src), "incomplete frame dropped");
 277           }
 278           dv1394src->frame = NULL;
 279
 280           dv1394src->frameSequence++;
 281
 282           if( dv1394src->frameSequence % (dv1394src->skip+dv1394src->consecutive) < dv1394src->consecutive ) {
 283             dv1394src->frame = gst_buffer_new_and_alloc (dv1394src->frameSize);
 284             }
 285             dv1394src->bytesInFrame = 0;
 286           }
 287
 288           if (dv1394src->frame != NULL) {
 289             void *data = GST_BUFFER_DATA( dv1394src->frame );
 290
 291
 292             switch (section_type) {
 293             case 0: /* 1 Header block */
 294                 /* p[3] |= 0x80; // hack to force PAL data */
 295                 memcpy(data + dif_sequence * 150 * 80, p, 480);
 296                 break;
 297
 298             case 1: /* 2 Subcode blocks */
 299                 memcpy(data + dif_sequence * 150 * 80 + (1 + dif_block) * 80, p, 480);
 300                 break;
 301
 302             case 2: /* 3 VAUX blocks */
 303                 memcpy(data + dif_sequence * 150 * 80 + (3 + dif_block) * 80, p, 480);
 304                 break;
 305
 306             case 3: /* 9 Audio blocks interleaved with video */
 307                 memcpy(data + dif_sequence * 150 * 80 + (6 + dif_block * 16) * 80, p, 480);
 308                 break;
 309
 310             case 4: /* 135 Video blocks interleaved with audio */
 311                 memcpy(data + dif_sequence * 150 * 80 + (7 + (dif_block / 15) + dif_block) * 80, p, 480);
 312                 break;
 313
 314             default: /* we can´t handle any other data */
 315                 break;
 316             }
 317             dv1394src->bytesInFrame += 480;
 318         }
 319   }
 320
 321   return 0;
 322 }
 323
 324 static
 325 int gst_dv1394src_bus_reset(raw1394handle_t handle,
 326                             unsigned int generation) {
 327   GST_INFO_OBJECT (GST_DV1394SRC(raw1394_get_userdata(handle)),"have bus reset");
 328   return 0;
 329 }
 330
 331 static GstData *
 332 gst_dv1394src_get (GstPad *pad)
 333 {
 334   GstDV1394Src *dv1394src = GST_DV1394SRC (GST_PAD_PARENT(pad));
 335
 336   dv1394src->buf = NULL;
 337   while (dv1394src->buf == NULL)
 338     raw1394_loop_iterate(dv1394src->handle);
 339
 340   return GST_DATA(dv1394src->buf);
 341 }
 342
 343 static GstElementStateReturn
 344 gst_dv1394src_change_state (GstElement *element)
 345 {
 346   GstDV1394Src *dv1394src;
 347
 348   g_return_val_if_fail (GST_IS_DV1394SRC (element), GST_STATE_FAILURE);
 349   dv1394src = GST_DV1394SRC (element);
 350
 351   switch (GST_STATE_TRANSITION (element)) {
 352     case GST_STATE_NULL_TO_READY:
 353       if ((dv1394src->handle = raw1394_new_handle()) == NULL) {
 354         GST_INFO_OBJECT (dv1394src,"can't get raw1394 handle");
 355         return GST_STATE_FAILURE;
 356       }
 357       raw1394_set_userdata(dv1394src->handle,dv1394src);
 358       dv1394src->numcards = raw1394_get_port_info(dv1394src->handle,dv1394src->pinfo,16);
 359       if (dv1394src->numcards == 0) {
 360         GST_INFO_OBJECT (dv1394src,"no cards available for raw1394");
 361         return GST_STATE_FAILURE;
 362       }
 363       if (dv1394src->pinfo[dv1394src->card].nodes <= 1) {
 364         GST_INFO_OBJECT (dv1394src,"there are no nodes on the 1394 bus");
 365         return GST_STATE_FAILURE;
 366       }
 367       if (raw1394_set_port(dv1394src->handle,dv1394src->port) < 0) {
 368         GST_INFO_OBJECT (dv1394src,"can't set 1394 port %d",dv1394src->port);
 369         return GST_STATE_FAILURE;
 370       }
 371       raw1394_set_iso_handler(dv1394src->handle,dv1394src->channel,gst_dv1394src_iso_receive);
 372       raw1394_set_bus_reset_handler(dv1394src->handle,gst_dv1394src_bus_reset);
 373       dv1394src->started = FALSE;
 374       GST_DEBUG ("successfully opened up 1394 connection");
 375       break;
 376     case GST_STATE_PAUSED_TO_PLAYING:
 377       if (raw1394_start_iso_rcv(dv1394src->handle,dv1394src->channel) < 0) {
 378         GST_INFO_OBJECT (dv1394src,"can't start 1394 iso receive");
 379         return GST_STATE_FAILURE;
 380       }
 381       break;
 382     case GST_STATE_PLAYING_TO_PAUSED:
 383       raw1394_stop_iso_rcv(dv1394src->handle, dv1394src->channel);
 384       break;
 385     case GST_STATE_READY_TO_NULL:
 386       raw1394_destroy_handle(dv1394src->handle);
 387       break;
 388     default:
 389       break;
 390   }
 391
 392   /* if we haven't failed already, give the parent class a chance to ;-) */
 393   if (GST_ELEMENT_CLASS (parent_class)->change_state)
 394     return GST_ELEMENT_CLASS (parent_class)->change_state (element);
 395
 396   return GST_STATE_SUCCESS;
 397 }