ext/aalib/gstaasink.c

   1 /* GStreamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include "config.h"
  22 #endif
  23
  24 #include <string.h>
  25 #include <sys/time.h>
  26
  27 #include "gstaasink.h"
  28 #include <gst/video/video.h>
  29
  30 /* elementfactory information */
  31 static GstElementDetails gst_aasink_details = {
  32   "AA sink",
  33   "Sink/Video",
  34   "An ASCII art videosink",
  35   "Wim Taymans <wim.taymans@chello.be>"
  36 };
  37
  38 /* aasink signals and args */
  39 enum
  40 {
  41   SIGNAL_FRAME_DISPLAYED,
  42   SIGNAL_HAVE_SIZE,
  43   LAST_SIGNAL
  44 };
  45
  46
  47 enum
  48 {
  49   ARG_0,
  50   ARG_WIDTH,
  51   ARG_HEIGHT,
  52   ARG_DRIVER,
  53   ARG_DITHER,
  54   ARG_BRIGHTNESS,
  55   ARG_CONTRAST,
  56   ARG_GAMMA,
  57   ARG_INVERSION,
  58   ARG_RANDOMVAL,
  59   ARG_FRAMES_DISPLAYED,
  60   ARG_FRAME_TIME,
  61 };
  62
  63 static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
  64     GST_PAD_SINK,
  65     GST_PAD_ALWAYS,
  66     GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420"))
  67     );
  68
  69 static void gst_aasink_base_init (gpointer g_class);
  70 static void gst_aasink_class_init (GstAASinkClass * klass);
  71 static void gst_aasink_init (GstAASink * aasink);
  72
  73 static void gst_aasink_set_clock (GstElement * element, GstClock * clock);
  74 static void gst_aasink_chain (GstPad * pad, GstData * _data);
  75
  76 static void gst_aasink_set_property (GObject * object, guint prop_id,
  77     const GValue * value, GParamSpec * pspec);
  78 static void gst_aasink_get_property (GObject * object, guint prop_id,
  79     GValue * value, GParamSpec * pspec);
  80
  81 static GstElementStateReturn gst_aasink_change_state (GstElement * element);
  82
  83 static GstElementClass *parent_class = NULL;
  84 static guint gst_aasink_signals[LAST_SIGNAL] = { 0 };
  85
  86 GType
  87 gst_aasink_get_type (void)
  88 {
  89   static GType aasink_type = 0;
  90
  91   if (!aasink_type) {
  92     static const GTypeInfo aasink_info = {
  93       sizeof (GstAASinkClass),
  94       gst_aasink_base_init,
  95       NULL,
  96       (GClassInitFunc) gst_aasink_class_init,
  97       NULL,
  98       NULL,
  99       sizeof (GstAASink),
 100       0,
 101       (GInstanceInitFunc) gst_aasink_init,
 102     };
 103     aasink_type =
 104         g_type_register_static (GST_TYPE_ELEMENT, "GstAASink", &aasink_info, 0);
 105   }
 106   return aasink_type;
 107 }
 108
 109 #define GST_TYPE_AADRIVERS (gst_aasink_drivers_get_type())
 110 static GType
 111 gst_aasink_drivers_get_type (void)
 112 {
 113   static GType driver_type = 0;
 114
 115   if (!driver_type) {
 116     GEnumValue *drivers;
 117     const struct aa_driver *driver;
 118     gint n_drivers;
 119     gint i;
 120
 121     for (n_drivers = 0; aa_drivers[n_drivers]; n_drivers++) {
 122       /* count number of drivers */
 123     }
 124
 125     drivers = g_new0 (GEnumValue, n_drivers + 1);
 126
 127     for (i = 0; i < n_drivers; i++) {
 128       driver = aa_drivers[i];
 129       drivers[i].value = i;
 130       drivers[i].value_name = g_strdup (driver->shortname);
 131       drivers[i].value_nick = g_strdup (driver->name);
 132     }
 133     drivers[i].value = 0;
 134     drivers[i].value_name = NULL;
 135     drivers[i].value_nick = NULL;
 136
 137     driver_type = g_enum_register_static ("GstAASinkDrivers", drivers);
 138   }
 139   return driver_type;
 140 }
 141
 142 #define GST_TYPE_AADITHER (gst_aasink_dither_get_type())
 143 static GType
 144 gst_aasink_dither_get_type (void)
 145 {
 146   static GType dither_type = 0;
 147
 148   if (!dither_type) {
 149     GEnumValue *ditherers;
 150     gint n_ditherers;
 151     gint i;
 152
 153     for (n_ditherers = 0; aa_dithernames[n_ditherers]; n_ditherers++) {
 154       /* count number of ditherers */
 155     }
 156
 157     ditherers = g_new0 (GEnumValue, n_ditherers + 1);
 158
 159     for (i = 0; i < n_ditherers; i++) {
 160       ditherers[i].value = i;
 161       ditherers[i].value_name = g_strdup (aa_dithernames[i]);
 162       ditherers[i].value_nick = g_strdup (aa_dithernames[i]);
 163     }
 164     ditherers[i].value = 0;
 165     ditherers[i].value_name = NULL;
 166     ditherers[i].value_nick = NULL;
 167
 168     dither_type = g_enum_register_static ("GstAASinkDitherers", ditherers);
 169   }
 170   return dither_type;
 171 }
 172
 173 static void
 174 gst_aasink_base_init (gpointer g_class)
 175 {
 176   GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
 177
 178   gst_element_class_add_pad_template (element_class,
 179       gst_static_pad_template_get (&sink_template));
 180   gst_element_class_set_details (element_class, &gst_aasink_details);
 181 }
 182
 183 static void
 184 gst_aasink_class_init (GstAASinkClass * klass)
 185 {
 186   GObjectClass *gobject_class;
 187   GstElementClass *gstelement_class;
 188
 189   gobject_class = (GObjectClass *) klass;
 190   gstelement_class = (GstElementClass *) klass;
 191
 192   parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
 193
 194   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_WIDTH, g_param_spec_int ("width", "width", "width", G_MININT, G_MAXINT, 0, G_PARAM_READWRITE));  /* CHECKME */
 195   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_HEIGHT, g_param_spec_int ("height", "height", "height", G_MININT, G_MAXINT, 0, G_PARAM_READWRITE));      /* CHECKME */
 196   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_DRIVER, g_param_spec_enum ("driver", "driver", "driver", GST_TYPE_AADRIVERS, 0, G_PARAM_READWRITE));     /* CHECKME! */
 197   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_DITHER, g_param_spec_enum ("dither", "dither", "dither", GST_TYPE_AADITHER, 0, G_PARAM_READWRITE));      /* CHECKME! */
 198   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_BRIGHTNESS, g_param_spec_int ("brightness", "brightness", "brightness", G_MININT, G_MAXINT, 0, G_PARAM_READWRITE));      /* CHECKME */
 199   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_CONTRAST, g_param_spec_int ("contrast", "contrast", "contrast", G_MININT, G_MAXINT, 0, G_PARAM_READWRITE));      /* CHECKME */
 200   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_GAMMA, g_param_spec_float ("gamma", "gamma", "gamma", 0.0, 5.0, 1.0, G_PARAM_READWRITE));        /* CHECKME */
 201   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_INVERSION, g_param_spec_boolean ("inversion", "inversion", "inversion", TRUE, G_PARAM_READWRITE));       /* CHECKME */
 202   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_RANDOMVAL, g_param_spec_int ("randomval", "randomval", "randomval", G_MININT, G_MAXINT, 0, G_PARAM_READWRITE));  /* CHECKME */
 203   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_FRAMES_DISPLAYED, g_param_spec_int ("frames_displayed", "frames_displayed", "frames_displayed", G_MININT, G_MAXINT, 0, G_PARAM_READABLE));       /* CHECKME */
 204   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_FRAME_TIME, g_param_spec_int ("frame_time", "frame_time", "frame_time", G_MININT, G_MAXINT, 0, G_PARAM_READABLE));       /* CHECKME */
 205
 206   gobject_class->set_property = gst_aasink_set_property;
 207   gobject_class->get_property = gst_aasink_get_property;
 208
 209   gst_aasink_signals[SIGNAL_FRAME_DISPLAYED] =
 210       g_signal_new ("frame-displayed", G_TYPE_FROM_CLASS (klass),
 211       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstAASinkClass, frame_displayed),
 212       NULL, NULL, g_cclosure_marshal_VOID__VOID, G_TYPE_NONE, 0);
 213   gst_aasink_signals[SIGNAL_HAVE_SIZE] =
 214       g_signal_new ("have-size", G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
 215       G_STRUCT_OFFSET (GstAASinkClass, have_size), NULL, NULL,
 216       gst_marshal_VOID__INT_INT, G_TYPE_NONE, 2, G_TYPE_UINT, G_TYPE_UINT);
 217
 218   gstelement_class->change_state = gst_aasink_change_state;
 219   gstelement_class->set_clock = gst_aasink_set_clock;
 220 }
 221
 222 static GstCaps *
 223 gst_aasink_fixate (GstPad * pad, const GstCaps * caps)
 224 {
 225   GstStructure *structure;
 226   GstCaps *newcaps;
 227
 228   if (gst_caps_get_size (caps) > 1)
 229     return NULL;
 230
 231   newcaps = gst_caps_copy (caps);
 232   structure = gst_caps_get_structure (newcaps, 0);
 233
 234   if (gst_caps_structure_fixate_field_nearest_int (structure, "width", 320)) {
 235     return newcaps;
 236   }
 237   if (gst_caps_structure_fixate_field_nearest_int (structure, "height", 240)) {
 238     return newcaps;
 239   }
 240   if (gst_caps_structure_fixate_field_nearest_double (structure, "framerate",
 241           30.0)) {
 242     return newcaps;
 243   }
 244
 245   gst_caps_free (newcaps);
 246   return NULL;
 247 }
 248
 249 static GstPadLinkReturn
 250 gst_aasink_sinkconnect (GstPad * pad, const GstCaps * caps)
 251 {
 252   GstAASink *aasink;
 253   GstStructure *structure;
 254
 255   aasink = GST_AASINK (gst_pad_get_parent (pad));
 256
 257   structure = gst_caps_get_structure (caps, 0);
 258   gst_structure_get_int (structure, "width", &aasink->width);
 259   gst_structure_get_int (structure, "height", &aasink->height);
 260
 261   /* FIXME aasink->format is never set */
 262
 263   GST_DEBUG ("aasink: setting %08lx (" GST_FOURCC_FORMAT ")",
 264       aasink->format, GST_FOURCC_ARGS (aasink->format));
 265
 266   g_signal_emit (G_OBJECT (aasink), gst_aasink_signals[SIGNAL_HAVE_SIZE], 0,
 267       aasink->width, aasink->height);
 268
 269   return GST_PAD_LINK_OK;
 270 }
 271
 272 static void
 273 gst_aasink_set_clock (GstElement * element, GstClock * clock)
 274 {
 275   GstAASink *aasink = GST_AASINK (element);
 276
 277   aasink->clock = clock;
 278 }
 279
 280 static void
 281 gst_aasink_init (GstAASink * aasink)
 282 {
 283   aasink->sinkpad =
 284       gst_pad_new_from_template (gst_element_get_pad_template (GST_ELEMENT
 285           (aasink), "sink"), "sink");
 286   gst_element_add_pad (GST_ELEMENT (aasink), aasink->sinkpad);
 287   gst_pad_set_chain_function (aasink->sinkpad, gst_aasink_chain);
 288   gst_pad_set_link_function (aasink->sinkpad, gst_aasink_sinkconnect);
 289   gst_pad_set_fixate_function (aasink->sinkpad, gst_aasink_fixate);
 290
 291   memcpy (&aasink->ascii_surf, &aa_defparams,
 292       sizeof (struct aa_hardware_params));
 293   aasink->ascii_parms.bright = 0;
 294   aasink->ascii_parms.contrast = 16;
 295   aasink->ascii_parms.gamma = 1.0;
 296   aasink->ascii_parms.dither = 0;
 297   aasink->ascii_parms.inversion = 0;
 298   aasink->ascii_parms.randomval = 0;
 299   aasink->aa_driver = 0;
 300
 301   aasink->width = -1;
 302   aasink->height = -1;
 303
 304   aasink->clock = NULL;
 305
 306   GST_FLAG_SET (aasink, GST_ELEMENT_THREAD_SUGGESTED);
 307 }
 308
 309 static void
 310 gst_aasink_scale (GstAASink * aasink, gchar * src, gchar * dest,
 311     gint sw, gint sh, gint dw, gint dh)
 312 {
 313   gint ypos, yinc, y;
 314   gint xpos, xinc, x;
 315
 316   g_return_if_fail ((dw != 0) && (dh != 0));
 317
 318   ypos = 0x10000;
 319   yinc = (sh << 16) / dh;
 320   xinc = (sw << 16) / dw;
 321
 322   for (y = dh; y; y--) {
 323     while (ypos > 0x10000) {
 324       ypos -= 0x10000;
 325       src += sw;
 326     }
 327     xpos = 0x10000;
 328     {
 329       guchar *destp = dest;
 330       guchar *srcp = src;
 331
 332       for (x = dw; x; x--) {
 333         while (xpos >= 0x10000L) {
 334           srcp++;
 335           xpos -= 0x10000L;
 336         }
 337         *destp++ = *srcp;
 338         xpos += xinc;
 339       }
 340     }
 341     dest += dw;
 342     ypos += yinc;
 343   }
 344 }
 345
 346 static void
 347 gst_aasink_chain (GstPad * pad, GstData * _data)
 348 {
 349   GstBuffer *buf = GST_BUFFER (_data);
 350   GstAASink *aasink;
 351
 352   g_return_if_fail (pad != NULL);
 353   g_return_if_fail (GST_IS_PAD (pad));
 354   g_return_if_fail (buf != NULL);
 355
 356   aasink = GST_AASINK (gst_pad_get_parent (pad));
 357
 358   gst_aasink_scale (aasink, GST_BUFFER_DATA (buf),      /* src */
 359       aa_image (aasink->context),       /* dest */
 360       aasink->width,            /* sw */
 361       aasink->height,           /* sh */
 362       aa_imgwidth (aasink->context),    /* dw */
 363       aa_imgheight (aasink->context));  /* dh */
 364
 365   GST_DEBUG ("videosink: clock wait: %" G_GUINT64_FORMAT,
 366       GST_BUFFER_TIMESTAMP (buf));
 367
 368   if (aasink->clock && GST_BUFFER_TIMESTAMP_IS_VALID (buf)) {
 369     gst_element_wait (GST_ELEMENT (aasink), GST_BUFFER_TIMESTAMP (buf));
 370   }
 371
 372   aa_render (aasink->context, &aasink->ascii_parms,
 373       0, 0, aa_imgwidth (aasink->context), aa_imgheight (aasink->context));
 374   aa_flush (aasink->context);
 375   aa_getevent (aasink->context, FALSE);
 376
 377   g_signal_emit (G_OBJECT (aasink), gst_aasink_signals[SIGNAL_FRAME_DISPLAYED],
 378       0);
 379
 380   gst_buffer_unref (buf);
 381 }
 382
 383
 384 static void
 385 gst_aasink_set_property (GObject * object, guint prop_id, const GValue * value,
 386     GParamSpec * pspec)
 387 {
 388   GstAASink *aasink;
 389
 390   /* it's not null if we got it, but it might not be ours */
 391   g_return_if_fail (GST_IS_AASINK (object));
 392
 393   aasink = GST_AASINK (object);
 394
 395   switch (prop_id) {
 396     case ARG_WIDTH:
 397       aasink->ascii_surf.width = g_value_get_int (value);
 398       break;
 399     case ARG_HEIGHT:
 400       aasink->ascii_surf.height = g_value_get_int (value);
 401       break;
 402     case ARG_DRIVER:{
 403       aasink->aa_driver = g_value_get_enum (value);
 404       break;
 405     }
 406     case ARG_DITHER:{
 407       aasink->ascii_parms.dither = g_value_get_enum (value);
 408       break;
 409     }
 410     case ARG_BRIGHTNESS:{
 411       aasink->ascii_parms.bright = g_value_get_int (value);
 412       break;
 413     }
 414     case ARG_CONTRAST:{
 415       aasink->ascii_parms.contrast = g_value_get_int (value);
 416       break;
 417     }
 418     case ARG_GAMMA:{
 419       aasink->ascii_parms.gamma = g_value_get_float (value);
 420       break;
 421     }
 422     case ARG_INVERSION:{
 423       aasink->ascii_parms.inversion = g_value_get_boolean (value);
 424       break;
 425     }
 426     case ARG_RANDOMVAL:{
 427       aasink->ascii_parms.randomval = g_value_get_int (value);
 428       break;
 429     }
 430     default:
 431       break;
 432   }
 433 }
 434
 435 static void
 436 gst_aasink_get_property (GObject * object, guint prop_id, GValue * value,
 437     GParamSpec * pspec)
 438 {
 439   GstAASink *aasink;
 440
 441   /* it's not null if we got it, but it might not be ours */
 442   aasink = GST_AASINK (object);
 443
 444   switch (prop_id) {
 445     case ARG_WIDTH:{
 446       g_value_set_int (value, aasink->ascii_surf.width);
 447       break;
 448     }
 449     case ARG_HEIGHT:{
 450       g_value_set_int (value, aasink->ascii_surf.height);
 451       break;
 452     }
 453     case ARG_DRIVER:{
 454       g_value_set_enum (value, aasink->aa_driver);
 455       break;
 456     }
 457     case ARG_DITHER:{
 458       g_value_set_enum (value, aasink->ascii_parms.dither);
 459       break;
 460     }
 461     case ARG_BRIGHTNESS:{
 462       g_value_set_int (value, aasink->ascii_parms.bright);
 463       break;
 464     }
 465     case ARG_CONTRAST:{
 466       g_value_set_int (value, aasink->ascii_parms.contrast);
 467       break;
 468     }
 469     case ARG_GAMMA:{
 470       g_value_set_float (value, aasink->ascii_parms.gamma);
 471       break;
 472     }
 473     case ARG_INVERSION:{
 474       g_value_set_boolean (value, aasink->ascii_parms.inversion);
 475       break;
 476     }
 477     case ARG_RANDOMVAL:{
 478       g_value_set_int (value, aasink->ascii_parms.randomval);
 479       break;
 480     }
 481     case ARG_FRAMES_DISPLAYED:{
 482       g_value_set_int (value, aasink->frames_displayed);
 483       break;
 484     }
 485     case ARG_FRAME_TIME:{
 486       g_value_set_int (value, aasink->frame_time / 1000000);
 487       break;
 488     }
 489     default:{
 490       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 491       break;
 492     }
 493   }
 494 }
 495
 496 static gboolean
 497 gst_aasink_open (GstAASink * aasink)
 498 {
 499   g_return_val_if_fail (!GST_FLAG_IS_SET (aasink, GST_AASINK_OPEN), FALSE);
 500
 501   aa_recommendhidisplay (aa_drivers[aasink->aa_driver]->shortname);
 502
 503   aasink->context = aa_autoinit (&aasink->ascii_surf);
 504   if (aasink->context == NULL) {
 505     GST_ELEMENT_ERROR (GST_ELEMENT (aasink), LIBRARY, TOO_LAZY, (NULL),
 506         ("error opening aalib context"));
 507     return FALSE;
 508   }
 509   aa_autoinitkbd (aasink->context, 0);
 510   aa_resizehandler (aasink->context, (void *) aa_resize);
 511
 512   GST_FLAG_SET (aasink, GST_AASINK_OPEN);
 513
 514   return TRUE;
 515 }
 516
 517 static void
 518 gst_aasink_close (GstAASink * aasink)
 519 {
 520   g_return_if_fail (GST_FLAG_IS_SET (aasink, GST_AASINK_OPEN));
 521
 522   aa_close (aasink->context);
 523
 524   GST_FLAG_UNSET (aasink, GST_AASINK_OPEN);
 525 }
 526
 527 static GstElementStateReturn
 528 gst_aasink_change_state (GstElement * element)
 529 {
 530   g_return_val_if_fail (GST_IS_AASINK (element), GST_STATE_FAILURE);
 531
 532   if (GST_STATE_PENDING (element) == GST_STATE_NULL) {
 533     if (GST_FLAG_IS_SET (element, GST_AASINK_OPEN))
 534       gst_aasink_close (GST_AASINK (element));
 535   } else {
 536     if (!GST_FLAG_IS_SET (element, GST_AASINK_OPEN)) {
 537       if (!gst_aasink_open (GST_AASINK (element)))
 538         return GST_STATE_FAILURE;
 539     }
 540   }
 541
 542   if (GST_ELEMENT_CLASS (parent_class)->change_state)
 543     return GST_ELEMENT_CLASS (parent_class)->change_state (element);
 544
 545   return GST_STATE_SUCCESS;
 546 }
 547
 548 static gboolean
 549 plugin_init (GstPlugin * plugin)
 550 {
 551   if (!gst_element_register (plugin, "aasink", GST_RANK_NONE, GST_TYPE_AASINK))
 552     return FALSE;
 553
 554   return TRUE;
 555 }
 556
 557 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
 558     GST_VERSION_MINOR,
 559     "aasink",
 560     "ASCII Art video sink",
 561     plugin_init, VERSION, "GPL", GST_PACKAGE, GST_ORIGIN)