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 gboolean gst_aasink_setcaps (GstBaseSink * pad, GstCaps * caps);
  74 static void gst_aasink_get_times (GstBaseSink * sink, GstBuffer * buffer,
  75     GstClockTime * start, GstClockTime * end);
  76 static GstFlowReturn gst_aasink_render (GstBaseSink * basesink,
  77     GstBuffer * buffer);
  78
  79 static void gst_aasink_set_property (GObject * object, guint prop_id,
  80     const GValue * value, GParamSpec * pspec);
  81 static void gst_aasink_get_property (GObject * object, guint prop_id,
  82     GValue * value, GParamSpec * pspec);
  83
  84 static GstStateChangeReturn gst_aasink_change_state (GstElement * element,
  85     GstStateChange transition);
  86
  87 static GstElementClass *parent_class = NULL;
  88 static guint gst_aasink_signals[LAST_SIGNAL] = { 0 };
  89
  90 GType
  91 gst_aasink_get_type (void)
  92 {
  93   static GType aasink_type = 0;
  94
  95   if (!aasink_type) {
  96     static const GTypeInfo aasink_info = {
  97       sizeof (GstAASinkClass),
  98       gst_aasink_base_init,
  99       NULL,
 100       (GClassInitFunc) gst_aasink_class_init,
 101       NULL,
 102       NULL,
 103       sizeof (GstAASink),
 104       0,
 105       (GInstanceInitFunc) gst_aasink_init,
 106     };
 107
 108     aasink_type =
 109         g_type_register_static (GST_TYPE_BASE_SINK, "GstAASink", &aasink_info,
 110         0);
 111   }
 112   return aasink_type;
 113 }
 114
 115 #define GST_TYPE_AADRIVERS (gst_aasink_drivers_get_type())
 116 static GType
 117 gst_aasink_drivers_get_type (void)
 118 {
 119   static GType driver_type = 0;
 120
 121   if (!driver_type) {
 122     GEnumValue *drivers;
 123     const struct aa_driver *driver;
 124     gint n_drivers;
 125     gint i;
 126
 127     for (n_drivers = 0; aa_drivers[n_drivers]; n_drivers++) {
 128       /* count number of drivers */
 129     }
 130
 131     drivers = g_new0 (GEnumValue, n_drivers + 1);
 132
 133     for (i = 0; i < n_drivers; i++) {
 134       driver = aa_drivers[i];
 135       drivers[i].value = i;
 136       drivers[i].value_name = g_strdup (driver->name);
 137       drivers[i].value_nick = g_utf8_strdown (driver->shortname, -1);
 138     }
 139     drivers[i].value = 0;
 140     drivers[i].value_name = NULL;
 141     drivers[i].value_nick = NULL;
 142
 143     driver_type = g_enum_register_static ("GstAASinkDrivers", drivers);
 144   }
 145   return driver_type;
 146 }
 147
 148 #define GST_TYPE_AADITHER (gst_aasink_dither_get_type())
 149 static GType
 150 gst_aasink_dither_get_type (void)
 151 {
 152   static GType dither_type = 0;
 153
 154   if (!dither_type) {
 155     GEnumValue *ditherers;
 156     gint n_ditherers;
 157     gint i;
 158
 159     for (n_ditherers = 0; aa_dithernames[n_ditherers]; n_ditherers++) {
 160       /* count number of ditherers */
 161     }
 162
 163     ditherers = g_new0 (GEnumValue, n_ditherers + 1);
 164
 165     for (i = 0; i < n_ditherers; i++) {
 166       ditherers[i].value = i;
 167       ditherers[i].value_name = g_strdup (aa_dithernames[i]);
 168       ditherers[i].value_nick =
 169           g_strdelimit (g_strdup (aa_dithernames[i]), " _", '-');
 170     }
 171     ditherers[i].value = 0;
 172     ditherers[i].value_name = NULL;
 173     ditherers[i].value_nick = NULL;
 174
 175     dither_type = g_enum_register_static ("GstAASinkDitherers", ditherers);
 176   }
 177   return dither_type;
 178 }
 179
 180 static void
 181 gst_aasink_base_init (gpointer g_class)
 182 {
 183   GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
 184
 185   gst_element_class_add_pad_template (element_class,
 186       gst_static_pad_template_get (&sink_template));
 187   gst_element_class_set_details (element_class, &gst_aasink_details);
 188 }
 189
 190 static void
 191 gst_aasink_class_init (GstAASinkClass * klass)
 192 {
 193   GObjectClass *gobject_class;
 194   GstElementClass *gstelement_class;
 195   GstBaseSinkClass *gstbasesink_class;
 196
 197   gobject_class = (GObjectClass *) klass;
 198   gstelement_class = (GstElementClass *) klass;
 199   gstbasesink_class = (GstBaseSinkClass *) klass;
 200
 201   parent_class = g_type_class_ref (GST_TYPE_BASE_SINK);
 202
 203   gobject_class->set_property = gst_aasink_set_property;
 204   gobject_class->get_property = gst_aasink_get_property;
 205
 206   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 */
 207   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 */
 208   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! */
 209   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! */
 210   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 */
 211   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 */
 212   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 */
 213   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_INVERSION, g_param_spec_boolean ("inversion", "inversion", "inversion", TRUE, G_PARAM_READWRITE));       /* CHECKME */
 214   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 */
 215   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 */
 216   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 */
 217
 218   gst_aasink_signals[SIGNAL_FRAME_DISPLAYED] =
 219       g_signal_new ("frame-displayed", G_TYPE_FROM_CLASS (klass),
 220       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstAASinkClass, frame_displayed),
 221       NULL, NULL, g_cclosure_marshal_VOID__VOID, G_TYPE_NONE, 0);
 222   gst_aasink_signals[SIGNAL_HAVE_SIZE] =
 223       g_signal_new ("have-size", G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
 224       G_STRUCT_OFFSET (GstAASinkClass, have_size), NULL, NULL,
 225       gst_marshal_VOID__INT_INT, G_TYPE_NONE, 2, G_TYPE_UINT, G_TYPE_UINT);
 226
 227   gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_aasink_change_state);
 228
 229   gstbasesink_class->set_caps = GST_DEBUG_FUNCPTR (gst_aasink_setcaps);
 230   gstbasesink_class->get_times = GST_DEBUG_FUNCPTR (gst_aasink_get_times);
 231   gstbasesink_class->preroll = GST_DEBUG_FUNCPTR (gst_aasink_render);
 232   gstbasesink_class->render = GST_DEBUG_FUNCPTR (gst_aasink_render);
 233 }
 234
 235 static void
 236 gst_aasink_fixate (GstPad * pad, GstCaps * caps)
 237 {
 238   GstStructure *structure;
 239
 240   structure = gst_caps_get_structure (caps, 0);
 241
 242   gst_structure_fixate_field_nearest_int (structure, "width", 320);
 243   gst_structure_fixate_field_nearest_int (structure, "height", 240);
 244   gst_structure_fixate_field_nearest_double (structure, "framerate", 30.0);
 245 }
 246
 247 static gboolean
 248 gst_aasink_setcaps (GstBaseSink * basesink, GstCaps * caps)
 249 {
 250   GstAASink *aasink;
 251   GstStructure *structure;
 252
 253   aasink = GST_AASINK (basesink);
 254
 255   structure = gst_caps_get_structure (caps, 0);
 256   gst_structure_get_int (structure, "width", &aasink->width);
 257   gst_structure_get_int (structure, "height", &aasink->height);
 258
 259   /* FIXME aasink->format is never set */
 260   g_print ("%d %d\n", aasink->width, aasink->height);
 261
 262   GST_DEBUG ("aasink: setting %08lx (%" GST_FOURCC_FORMAT ")",
 263       aasink->format, GST_FOURCC_ARGS (aasink->format));
 264
 265   g_signal_emit (G_OBJECT (aasink), gst_aasink_signals[SIGNAL_HAVE_SIZE], 0,
 266       aasink->width, aasink->height);
 267
 268   return TRUE;
 269 }
 270
 271 static void
 272 gst_aasink_init (GstAASink * aasink)
 273 {
 274   GstPad *pad;
 275
 276   pad = GST_BASE_SINK_PAD (aasink);
 277   gst_pad_set_fixatecaps_function (pad, gst_aasink_fixate);
 278
 279   memcpy (&aasink->ascii_surf, &aa_defparams,
 280       sizeof (struct aa_hardware_params));
 281   aasink->ascii_parms.bright = 0;
 282   aasink->ascii_parms.contrast = 16;
 283   aasink->ascii_parms.gamma = 1.0;
 284   aasink->ascii_parms.dither = 0;
 285   aasink->ascii_parms.inversion = 0;
 286   aasink->ascii_parms.randomval = 0;
 287   aasink->aa_driver = 0;
 288
 289   aasink->width = -1;
 290   aasink->height = -1;
 291
 292 }
 293
 294 static void
 295 gst_aasink_scale (GstAASink * aasink, guchar * src, guchar * dest,
 296     gint sw, gint sh, gint dw, gint dh)
 297 {
 298   gint ypos, yinc, y;
 299   gint xpos, xinc, x;
 300
 301   g_return_if_fail ((dw != 0) && (dh != 0));
 302
 303   ypos = 0x10000;
 304   yinc = (sh << 16) / dh;
 305   xinc = (sw << 16) / dw;
 306
 307   for (y = dh; y; y--) {
 308     while (ypos > 0x10000) {
 309       ypos -= 0x10000;
 310       src += sw;
 311     }
 312     xpos = 0x10000;
 313     {
 314       guchar *destp = dest;
 315       guchar *srcp = src;
 316
 317       for (x = dw; x; x--) {
 318         while (xpos >= 0x10000L) {
 319           srcp++;
 320           xpos -= 0x10000L;
 321         }
 322         *destp++ = *srcp;
 323         xpos += xinc;
 324       }
 325     }
 326     dest += dw;
 327     ypos += yinc;
 328   }
 329 }
 330
 331 static void
 332 gst_aasink_get_times (GstBaseSink * sink, GstBuffer * buffer,
 333     GstClockTime * start, GstClockTime * end)
 334 {
 335   *start = GST_BUFFER_TIMESTAMP (buffer);
 336   *end = *start + GST_BUFFER_DURATION (buffer);
 337 }
 338
 339 static GstFlowReturn
 340 gst_aasink_render (GstBaseSink * basesink, GstBuffer * buffer)
 341 {
 342   GstAASink *aasink;
 343
 344   aasink = GST_AASINK (basesink);
 345
 346   GST_DEBUG ("render");
 347
 348   gst_aasink_scale (aasink, GST_BUFFER_DATA (buffer),   /* src */
 349       aa_image (aasink->context),       /* dest */
 350       aasink->width,            /* sw */
 351       aasink->height,           /* sh */
 352       aa_imgwidth (aasink->context),    /* dw */
 353       aa_imgheight (aasink->context));  /* dh */
 354
 355   aa_render (aasink->context, &aasink->ascii_parms,
 356       0, 0, aa_imgwidth (aasink->context), aa_imgheight (aasink->context));
 357   aa_flush (aasink->context);
 358   aa_getevent (aasink->context, FALSE);
 359
 360   return GST_FLOW_OK;
 361 }
 362
 363
 364 static void
 365 gst_aasink_set_property (GObject * object, guint prop_id, const GValue * value,
 366     GParamSpec * pspec)
 367 {
 368   GstAASink *aasink;
 369
 370   aasink = GST_AASINK (object);
 371
 372   switch (prop_id) {
 373     case ARG_WIDTH:
 374       aasink->ascii_surf.width = g_value_get_int (value);
 375       break;
 376     case ARG_HEIGHT:
 377       aasink->ascii_surf.height = g_value_get_int (value);
 378       break;
 379     case ARG_DRIVER:{
 380       aasink->aa_driver = g_value_get_enum (value);
 381       break;
 382     }
 383     case ARG_DITHER:{
 384       aasink->ascii_parms.dither = g_value_get_enum (value);
 385       break;
 386     }
 387     case ARG_BRIGHTNESS:{
 388       aasink->ascii_parms.bright = g_value_get_int (value);
 389       break;
 390     }
 391     case ARG_CONTRAST:{
 392       aasink->ascii_parms.contrast = g_value_get_int (value);
 393       break;
 394     }
 395     case ARG_GAMMA:{
 396       aasink->ascii_parms.gamma = g_value_get_float (value);
 397       break;
 398     }
 399     case ARG_INVERSION:{
 400       aasink->ascii_parms.inversion = g_value_get_boolean (value);
 401       break;
 402     }
 403     case ARG_RANDOMVAL:{
 404       aasink->ascii_parms.randomval = g_value_get_int (value);
 405       break;
 406     }
 407     default:
 408       break;
 409   }
 410 }
 411
 412 static void
 413 gst_aasink_get_property (GObject * object, guint prop_id, GValue * value,
 414     GParamSpec * pspec)
 415 {
 416   GstAASink *aasink;
 417
 418   aasink = GST_AASINK (object);
 419
 420   switch (prop_id) {
 421     case ARG_WIDTH:{
 422       g_value_set_int (value, aasink->ascii_surf.width);
 423       break;
 424     }
 425     case ARG_HEIGHT:{
 426       g_value_set_int (value, aasink->ascii_surf.height);
 427       break;
 428     }
 429     case ARG_DRIVER:{
 430       g_value_set_enum (value, aasink->aa_driver);
 431       break;
 432     }
 433     case ARG_DITHER:{
 434       g_value_set_enum (value, aasink->ascii_parms.dither);
 435       break;
 436     }
 437     case ARG_BRIGHTNESS:{
 438       g_value_set_int (value, aasink->ascii_parms.bright);
 439       break;
 440     }
 441     case ARG_CONTRAST:{
 442       g_value_set_int (value, aasink->ascii_parms.contrast);
 443       break;
 444     }
 445     case ARG_GAMMA:{
 446       g_value_set_float (value, aasink->ascii_parms.gamma);
 447       break;
 448     }
 449     case ARG_INVERSION:{
 450       g_value_set_boolean (value, aasink->ascii_parms.inversion);
 451       break;
 452     }
 453     case ARG_RANDOMVAL:{
 454       g_value_set_int (value, aasink->ascii_parms.randomval);
 455       break;
 456     }
 457     case ARG_FRAMES_DISPLAYED:{
 458       g_value_set_int (value, aasink->frames_displayed);
 459       break;
 460     }
 461     case ARG_FRAME_TIME:{
 462       g_value_set_int (value, aasink->frame_time / 1000000);
 463       break;
 464     }
 465     default:{
 466       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 467       break;
 468     }
 469   }
 470 }
 471
 472 static gboolean
 473 gst_aasink_open (GstAASink * aasink)
 474 {
 475   aa_recommendhidisplay (aa_drivers[aasink->aa_driver]->shortname);
 476
 477   aasink->context = aa_autoinit (&aasink->ascii_surf);
 478   if (aasink->context == NULL) {
 479     GST_ELEMENT_ERROR (GST_ELEMENT (aasink), LIBRARY, TOO_LAZY, (NULL),
 480         ("error opening aalib context"));
 481     return FALSE;
 482   }
 483   aa_autoinitkbd (aasink->context, 0);
 484   aa_resizehandler (aasink->context, (void *) aa_resize);
 485
 486   return TRUE;
 487 }
 488
 489 static gboolean
 490 gst_aasink_close (GstAASink * aasink)
 491 {
 492   aa_close (aasink->context);
 493
 494   return TRUE;
 495 }
 496
 497 static GstStateChangeReturn
 498 gst_aasink_change_state (GstElement * element, GstStateChange transition)
 499 {
 500   GstStateChangeReturn ret;
 501
 502
 503   switch (transition) {
 504     case GST_STATE_CHANGE_NULL_TO_READY:
 505       break;
 506     case GST_STATE_CHANGE_READY_TO_PAUSED:
 507       if (!gst_aasink_open (GST_AASINK (element)))
 508         goto open_failed;
 509       break;
 510     case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
 511       break;
 512     default:
 513       break;
 514   }
 515
 516   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
 517
 518   switch (transition) {
 519     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
 520       break;
 521     case GST_STATE_CHANGE_PAUSED_TO_READY:
 522       break;
 523     case GST_STATE_CHANGE_READY_TO_NULL:
 524       gst_aasink_close (GST_AASINK (element));
 525       break;
 526     default:
 527       break;
 528   }
 529
 530   return ret;
 531
 532 open_failed:
 533   {
 534     return GST_STATE_CHANGE_FAILURE;
 535   }
 536 }
 537
 538 static gboolean
 539 plugin_init (GstPlugin * plugin)
 540 {
 541   if (!gst_element_register (plugin, "aasink", GST_RANK_NONE, GST_TYPE_AASINK))
 542     return FALSE;
 543
 544   return TRUE;
 545 }
 546
 547 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
 548     GST_VERSION_MINOR,
 549     "aasink",
 550     "ASCII Art video sink",
 551     plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);