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