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         gst_props_new (
  69           "format", GST_PROPS_LIST (
  70                         G_TYPE_STRING ("NTSC"),
  71                         G_TYPE_STRING ("PAL")
  72                         ),
  73           NULL)
  74         ),
  75       NULL);
  76   }
  77   return template;
  78 }
  79 #endif
  80
  81 static void             gst_dv1394src_base_init         (gpointer g_class);
  82 static void             gst_dv1394src_class_init                (GstDV1394SrcClass *klass);
  83 static void             gst_dv1394src_init              (GstDV1394Src *filter);
  84
  85 static void             gst_dv1394src_set_property              (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec);
  86 static void             gst_dv1394src_get_property              (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec);
  87
  88 static GstElementStateReturn    gst_dv1394src_change_state      (GstElement *element);
  89
  90 static GstData *        gst_dv1394src_get                       (GstPad *pad);
  91
  92 static GstElementClass *parent_class = NULL;
  93 /*static guint gst_filter_signals[LAST_SIGNAL] = { 0 }; */
  94
  95 GType
  96 gst_dv1394src_get_type(void) {
  97   static GType gst_dv1394src_type = 0;
  98
  99   if (!gst_dv1394src_type) {
 100     static const GTypeInfo gst_dv1394src_info = {
 101       sizeof(GstDV1394Src),
 102       gst_dv1394src_base_init,
 103       NULL,
 104       (GClassInitFunc)gst_dv1394src_class_init,
 105       NULL,
 106       NULL,
 107       sizeof(GstDV1394Src),
 108       0,
 109       (GInstanceInitFunc)gst_dv1394src_init,
 110     };
 111     gst_dv1394src_type = g_type_register_static(GST_TYPE_ELEMENT, "DV1394Src", &gst_dv1394src_info, 0);
 112   }
 113   return gst_dv1394src_type;
 114 }
 115
 116 static void
 117 gst_dv1394src_base_init (gpointer g_class)
 118 {
 119   GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
 120
 121   gst_element_class_set_details (element_class, &gst_dv1394src_details);
 122 }
 123
 124 static void
 125 gst_dv1394src_class_init (GstDV1394SrcClass *klass)
 126 {
 127   GObjectClass *gobject_class;
 128   GstElementClass *gstelement_class;
 129
 130   gobject_class = (GObjectClass*)klass;
 131   gstelement_class = (GstElementClass*)klass;
 132
 133   g_object_class_install_property( G_OBJECT_CLASS(klass), ARG_CONSECUTIVE,
 134         g_param_spec_int("consecutive","consecutive frames","send n consecutive frames after skipping",
 135                         1, G_MAXINT,1,G_PARAM_READWRITE));
 136   g_object_class_install_property( G_OBJECT_CLASS(klass), ARG_SKIP,
 137         g_param_spec_int("skip","skip frames","skip n frames",
 138                         0, G_MAXINT,1,G_PARAM_READWRITE));
 139   g_object_class_install_property( G_OBJECT_CLASS(klass), ARG_DROP_INCOMPLETE,
 140         g_param_spec_boolean("drop_incomplete","drop_incomplete","drop incomplete frames",
 141                         TRUE, G_PARAM_READWRITE));
 142
 143   parent_class = g_type_class_ref(GST_TYPE_ELEMENT);
 144
 145   gobject_class->set_property = gst_dv1394src_set_property;
 146   gobject_class->get_property = gst_dv1394src_get_property;
 147
 148   gstelement_class->change_state = gst_dv1394src_change_state;
 149 }
 150
 151 static void
 152 gst_dv1394src_init (GstDV1394Src *dv1394src)
 153 {
 154   dv1394src->srcpad = gst_pad_new ("src", GST_PAD_SRC);
 155   gst_pad_set_get_function (dv1394src->srcpad, gst_dv1394src_get);
 156   gst_element_add_pad (GST_ELEMENT (dv1394src), dv1394src->srcpad);
 157
 158   dv1394src->card = 0;
 159   dv1394src->port = 0;
 160   dv1394src->channel = 63;
 161
 162   dv1394src->consecutive = 1;
 163   dv1394src->skip = 0;
 164   dv1394src->drop_incomplete = TRUE;
 165
 166   /* initialized when first header received */
 167   dv1394src->frameSize=0;
 168
 169   dv1394src->buf = NULL;
 170   dv1394src->frame = NULL;
 171   dv1394src->frameSequence = 0;
 172 }
 173
 174 static void
 175 gst_dv1394src_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
 176 {
 177   GstDV1394Src *filter;
 178
 179   /* it's not null if we got it, but it might not be ours */
 180   g_return_if_fail(GST_IS_DV1394SRC(object));
 181   filter = GST_DV1394SRC(object);
 182
 183   switch (prop_id) {
 184     case ARG_SKIP:
 185         filter->skip = g_value_get_int(value);
 186         break;
 187     case ARG_CONSECUTIVE:
 188         filter->consecutive = g_value_get_int(value);
 189         break;
 190     case ARG_DROP_INCOMPLETE:
 191         filter->drop_incomplete = g_value_get_boolean(value);
 192         break;
 193     default:
 194       break;
 195   }
 196 }
 197
 198 static void
 199 gst_dv1394src_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
 200 {
 201   GstDV1394Src *filter;
 202
 203   /* it's not null if we got it, but it might not be ours */
 204   g_return_if_fail(GST_IS_DV1394SRC(object));
 205   filter = GST_DV1394SRC(object);
 206
 207   switch (prop_id) {
 208     case ARG_SKIP:
 209         g_value_set_int( value, filter->skip );
 210         break;
 211     case ARG_CONSECUTIVE:
 212         g_value_set_int( value, filter->consecutive );
 213         break;
 214     case ARG_DROP_INCOMPLETE:
 215         g_value_set_boolean( value, filter->drop_incomplete );
 216         break;
 217     default:
 218       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 219       break;
 220   }
 221 }
 222
 223 static
 224 int gst_dv1394src_iso_receive(raw1394handle_t handle,int channel,size_t len,quadlet_t *data) {
 225   GstDV1394Src *dv1394src = GST_DV1394SRC (raw1394_get_userdata(handle));
 226
 227   if (len > 16) {
 228         /*
 229                  the following code taken from kino-0.51 (Dan Dennedy/Charles Yates)
 230         */
 231         unsigned char *p = (unsigned char*) & data[3];
 232         int section_type = p[0] >> 5;           /* section type is in bits 5 - 7 */
 233         int dif_sequence = p[1] >> 4;           /* dif sequence number is in bits 4 - 7 */
 234         int dif_block = p[2];
 235
 236         /* if we are at the beginning of a frame,
 237           we set buf=frame, and alloc a new buffer for frame
 238         */
 239
 240         if (section_type == 0 && dif_sequence == 0) {   // dif header
 241
 242           if( !dv1394src->negotiated) {
 243             // figure format (NTSC/PAL)
 244             if( p[3] & 0x80 ) {
 245               // PAL
 246               dv1394src->frameSize = PAL_FRAMESIZE;
 247               GST_DEBUG ("PAL data");
 248               if (gst_pad_try_set_caps (dv1394src->srcpad,
 249                     gst_caps_new_simple ("video/dv",
 250                       "format", G_TYPE_STRING, "PAL", NULL)) <= 0) {
 251                 GST_ELEMENT_ERROR (dv1394src, CORE, NEGOTIATION, (NULL),
 252                                    ("Could not set source caps for PAL"));
 253                 return 0;
 254               }
 255             } else {
 256               // NTSC (untested)
 257               dv1394src->frameSize = NTSC_FRAMESIZE;
 258               GST_DEBUG ("NTSC data [untested] - please report success/failure to <dan@f3c.com>");
 259               if (gst_pad_try_set_caps (dv1394src->srcpad,
 260                     gst_caps_new_simple ("video/dv",
 261                       "format", G_TYPE_STRING, "NTSC", NULL)) <= 0) {
 262                 GST_ELEMENT_ERROR (dv1394src, CORE, NEGOTIATION, (NULL),
 263                                    ("Could not set source caps for NTSC"));
 264                 return 0;
 265               }
 266             }
 267             dv1394src->negotiated = TRUE;
 268           }
 269
 270           // drop last frame when not complete
 271           if( !dv1394src->drop_incomplete || dv1394src->bytesInFrame == dv1394src->frameSize ) {
 272             dv1394src->buf = dv1394src->frame;
 273           } else {
 274             GST_INFO_OBJECT (GST_ELEMENT(dv1394src), "incomplete frame dropped");
 275           }
 276           dv1394src->frame = NULL;
 277
 278           dv1394src->frameSequence++;
 279
 280           if( dv1394src->frameSequence % (dv1394src->skip+dv1394src->consecutive) < dv1394src->consecutive ) {
 281             dv1394src->frame = gst_buffer_new_and_alloc (dv1394src->frameSize);
 282             }
 283             dv1394src->bytesInFrame = 0;
 284           }
 285
 286           if (dv1394src->frame != NULL) {
 287             void *data = GST_BUFFER_DATA( dv1394src->frame );
 288
 289
 290             switch (section_type) {
 291             case 0: /* 1 Header block */
 292                 /* p[3] |= 0x80; // hack to force PAL data */
 293                 memcpy(data + dif_sequence * 150 * 80, p, 480);
 294                 break;
 295
 296             case 1: /* 2 Subcode blocks */
 297                 memcpy(data + dif_sequence * 150 * 80 + (1 + dif_block) * 80, p, 480);
 298                 break;
 299
 300             case 2: /* 3 VAUX blocks */
 301                 memcpy(data + dif_sequence * 150 * 80 + (3 + dif_block) * 80, p, 480);
 302                 break;
 303
 304             case 3: /* 9 Audio blocks interleaved with video */
 305                 memcpy(data + dif_sequence * 150 * 80 + (6 + dif_block * 16) * 80, p, 480);
 306                 break;
 307
 308             case 4: /* 135 Video blocks interleaved with audio */
 309                 memcpy(data + dif_sequence * 150 * 80 + (7 + (dif_block / 15) + dif_block) * 80, p, 480);
 310                 break;
 311
 312             default: /* we can´t handle any other data */
 313                 break;
 314             }
 315             dv1394src->bytesInFrame += 480;
 316         }
 317   }
 318
 319   return 0;
 320 }
 321
 322 static
 323 int gst_dv1394src_bus_reset(raw1394handle_t handle,
 324                             unsigned int generation) {
 325   GST_INFO_OBJECT (GST_DV1394SRC(raw1394_get_userdata(handle)),"have bus reset");
 326   return 0;
 327 }
 328
 329 static GstData *
 330 gst_dv1394src_get (GstPad *pad)
 331 {
 332   GstDV1394Src *dv1394src = GST_DV1394SRC (GST_PAD_PARENT(pad));
 333
 334   dv1394src->buf = NULL;
 335   while (dv1394src->buf == NULL)
 336     raw1394_loop_iterate(dv1394src->handle);
 337
 338   return GST_DATA(dv1394src->buf);
 339 }
 340
 341 static GstElementStateReturn
 342 gst_dv1394src_change_state (GstElement *element)
 343 {
 344   GstDV1394Src *dv1394src;
 345
 346   g_return_val_if_fail (GST_IS_DV1394SRC (element), GST_STATE_FAILURE);
 347   dv1394src = GST_DV1394SRC (element);
 348
 349   switch (GST_STATE_TRANSITION (element)) {
 350     case GST_STATE_NULL_TO_READY:
 351       if ((dv1394src->handle = raw1394_new_handle()) == NULL) {
 352         GST_INFO_OBJECT (dv1394src,"can't get raw1394 handle");
 353         return GST_STATE_FAILURE;
 354       }
 355       raw1394_set_userdata(dv1394src->handle,dv1394src);
 356       dv1394src->numcards = raw1394_get_port_info(dv1394src->handle,dv1394src->pinfo,16);
 357       if (dv1394src->numcards == 0) {
 358         GST_INFO_OBJECT (dv1394src,"no cards available for raw1394");
 359         return GST_STATE_FAILURE;
 360       }
 361       if (dv1394src->pinfo[dv1394src->card].nodes <= 1) {
 362         GST_INFO_OBJECT (dv1394src,"there are no nodes on the 1394 bus");
 363         return GST_STATE_FAILURE;
 364       }
 365       if (raw1394_set_port(dv1394src->handle,dv1394src->port) < 0) {
 366         GST_INFO_OBJECT (dv1394src,"can't set 1394 port %d",dv1394src->port);
 367         return GST_STATE_FAILURE;
 368       }
 369       raw1394_set_iso_handler(dv1394src->handle,dv1394src->channel,gst_dv1394src_iso_receive);
 370       raw1394_set_bus_reset_handler(dv1394src->handle,gst_dv1394src_bus_reset);
 371       dv1394src->started = FALSE;
 372       GST_DEBUG ("successfully opened up 1394 connection");
 373       break;
 374     case GST_STATE_PAUSED_TO_PLAYING:
 375       if (raw1394_start_iso_rcv(dv1394src->handle,dv1394src->channel) < 0) {
 376         GST_INFO_OBJECT (dv1394src,"can't start 1394 iso receive");
 377         return GST_STATE_FAILURE;
 378       }
 379       break;
 380     case GST_STATE_PLAYING_TO_PAUSED:
 381       raw1394_stop_iso_rcv(dv1394src->handle, dv1394src->channel);
 382       break;
 383     case GST_STATE_READY_TO_NULL:
 384       raw1394_destroy_handle(dv1394src->handle);
 385       break;
 386     default:
 387       break;
 388   }
 389
 390   /* if we haven't failed already, give the parent class a chance to ;-) */
 391   if (GST_ELEMENT_CLASS (parent_class)->change_state)
 392     return GST_ELEMENT_CLASS (parent_class)->change_state (element);
 393
 394   return GST_STATE_SUCCESS;
 395 }