Add a GIcon implementation that can add an emblem to another icon.
[glib.git] / gio / gbufferedoutputstream.c
blob92b5f43b9dd8198d228ba9c473790579d69fb1cd
1 /* GIO - GLib Input, Output and Streaming Library
2 *
3 * Copyright (C) 2006-2007 Red Hat, Inc.
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser 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.
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 * Lesser General Public License for more details.
15 * You should have received a copy of the GNU Lesser General
16 * Public 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.
20 * Author: Christian Kellner <gicmo@gnome.org>
23 #include "config.h"
24 #include "gbufferedoutputstream.h"
25 #include "goutputstream.h"
26 #include "gsimpleasyncresult.h"
27 #include "string.h"
28 #include "glibintl.h"
30 #include <gioalias.h>
32 /**
33 * SECTION:gbufferedoutputstream
34 * @short_description: Buffered Output Stream
35 * @include: gio/gio.h
36 * @see_also: #GFilterOutputStream, #GOutputStream
38 * Buffered output stream implements #GFilterOutputStream and provides
39 * for buffered writes.
41 * By default, #GBufferedOutputStream's buffer size is set at 4 kilobytes.
43 * To create a buffered output stream, use g_buffered_output_stream_new(),
44 * or g_buffered_output_stream_new_sized() to specify the buffer's size
45 * at construction.
47 * To get the size of a buffer within a buffered input stream, use
48 * g_buffered_output_stream_get_buffer_size(). To change the size of a
49 * buffered output stream's buffer, use
50 * g_buffered_output_stream_set_buffer_size(). Note that the buffer's
51 * size cannot be reduced below the size of the data within the buffer.
52 **/
54 #define DEFAULT_BUFFER_SIZE 4096
56 struct _GBufferedOutputStreamPrivate {
57 guint8 *buffer;
58 gsize len;
59 goffset pos;
60 gboolean auto_grow;
63 enum {
64 PROP_0,
65 PROP_BUFSIZE,
66 PROP_AUTO_GROW
69 static void g_buffered_output_stream_set_property (GObject *object,
70 guint prop_id,
71 const GValue *value,
72 GParamSpec *pspec);
74 static void g_buffered_output_stream_get_property (GObject *object,
75 guint prop_id,
76 GValue *value,
77 GParamSpec *pspec);
78 static void g_buffered_output_stream_finalize (GObject *object);
81 static gssize g_buffered_output_stream_write (GOutputStream *stream,
82 const void *buffer,
83 gsize count,
84 GCancellable *cancellable,
85 GError **error);
86 static gboolean g_buffered_output_stream_flush (GOutputStream *stream,
87 GCancellable *cancellable,
88 GError **error);
89 static gboolean g_buffered_output_stream_close (GOutputStream *stream,
90 GCancellable *cancellable,
91 GError **error);
93 static void g_buffered_output_stream_write_async (GOutputStream *stream,
94 const void *buffer,
95 gsize count,
96 int io_priority,
97 GCancellable *cancellable,
98 GAsyncReadyCallback callback,
99 gpointer data);
100 static gssize g_buffered_output_stream_write_finish (GOutputStream *stream,
101 GAsyncResult *result,
102 GError **error);
103 static void g_buffered_output_stream_flush_async (GOutputStream *stream,
104 int io_priority,
105 GCancellable *cancellable,
106 GAsyncReadyCallback callback,
107 gpointer data);
108 static gboolean g_buffered_output_stream_flush_finish (GOutputStream *stream,
109 GAsyncResult *result,
110 GError **error);
111 static void g_buffered_output_stream_close_async (GOutputStream *stream,
112 int io_priority,
113 GCancellable *cancellable,
114 GAsyncReadyCallback callback,
115 gpointer data);
116 static gboolean g_buffered_output_stream_close_finish (GOutputStream *stream,
117 GAsyncResult *result,
118 GError **error);
120 G_DEFINE_TYPE (GBufferedOutputStream,
121 g_buffered_output_stream,
122 G_TYPE_FILTER_OUTPUT_STREAM)
125 static void
126 g_buffered_output_stream_class_init (GBufferedOutputStreamClass *klass)
128 GObjectClass *object_class;
129 GOutputStreamClass *ostream_class;
131 g_type_class_add_private (klass, sizeof (GBufferedOutputStreamPrivate));
133 object_class = G_OBJECT_CLASS (klass);
134 object_class->get_property = g_buffered_output_stream_get_property;
135 object_class->set_property = g_buffered_output_stream_set_property;
136 object_class->finalize = g_buffered_output_stream_finalize;
138 ostream_class = G_OUTPUT_STREAM_CLASS (klass);
139 ostream_class->write_fn = g_buffered_output_stream_write;
140 ostream_class->flush = g_buffered_output_stream_flush;
141 ostream_class->close_fn = g_buffered_output_stream_close;
142 ostream_class->write_async = g_buffered_output_stream_write_async;
143 ostream_class->write_finish = g_buffered_output_stream_write_finish;
144 ostream_class->flush_async = g_buffered_output_stream_flush_async;
145 ostream_class->flush_finish = g_buffered_output_stream_flush_finish;
146 ostream_class->close_async = g_buffered_output_stream_close_async;
147 ostream_class->close_finish = g_buffered_output_stream_close_finish;
149 g_object_class_install_property (object_class,
150 PROP_BUFSIZE,
151 g_param_spec_uint ("buffer-size",
152 P_("Buffer Size"),
153 P_("The size of the backend buffer"),
155 G_MAXUINT,
156 DEFAULT_BUFFER_SIZE,
157 G_PARAM_READWRITE|G_PARAM_CONSTRUCT|
158 G_PARAM_STATIC_NAME|G_PARAM_STATIC_NICK|G_PARAM_STATIC_BLURB));
160 g_object_class_install_property (object_class,
161 PROP_AUTO_GROW,
162 g_param_spec_boolean ("auto-grow",
163 P_("Auto-grow"),
164 P_("Whether the buffer should automatically grow"),
165 FALSE,
166 G_PARAM_READWRITE|
167 G_PARAM_STATIC_NAME|G_PARAM_STATIC_NICK|G_PARAM_STATIC_BLURB));
172 * g_buffered_output_stream_get_buffer_size:
173 * @stream: a #GBufferedOutputStream.
175 * Gets the size of the buffer in the @stream.
177 * Returns: the current size of the buffer.
179 gsize
180 g_buffered_output_stream_get_buffer_size (GBufferedOutputStream *stream)
182 g_return_val_if_fail (G_IS_BUFFERED_OUTPUT_STREAM (stream), -1);
184 return stream->priv->len;
188 * g_buffered_output_stream_set_buffer_size:
189 * @stream: a #GBufferedOutputStream.
190 * @size: a #gsize.
192 * Sets the size of the internal buffer to @size.
193 **/
194 void
195 g_buffered_output_stream_set_buffer_size (GBufferedOutputStream *stream,
196 gsize size)
198 GBufferedOutputStreamPrivate *priv;
199 guint8 *buffer;
201 g_return_if_fail (G_IS_BUFFERED_OUTPUT_STREAM (stream));
203 priv = stream->priv;
205 if (size == priv->len)
206 return;
208 if (priv->buffer)
210 size = MAX (size, priv->pos);
212 buffer = g_malloc (size);
213 memcpy (buffer, priv->buffer, priv->pos);
214 g_free (priv->buffer);
215 priv->buffer = buffer;
216 priv->len = size;
217 /* Keep old pos */
219 else
221 priv->buffer = g_malloc (size);
222 priv->len = size;
223 priv->pos = 0;
226 g_object_notify (G_OBJECT (stream), "buffer-size");
230 * g_buffered_output_stream_get_auto_grow:
231 * @stream: a #GBufferedOutputStream.
233 * Checks if the buffer automatically grows as data is added.
235 * Returns: %TRUE if the @stream's buffer automatically grows,
236 * %FALSE otherwise.
237 **/
238 gboolean
239 g_buffered_output_stream_get_auto_grow (GBufferedOutputStream *stream)
241 g_return_val_if_fail (G_IS_BUFFERED_OUTPUT_STREAM (stream), FALSE);
243 return stream->priv->auto_grow;
247 * g_buffered_output_stream_set_auto_grow:
248 * @stream: a #GBufferedOutputStream.
249 * @auto_grow: a #gboolean.
251 * Sets whether or not the @stream's buffer should automatically grow.
252 * If @auto_grow is true, then each write will just make the buffer
253 * larger, and you must manually flush the buffer to actually write out
254 * the data to the underlying stream.
256 void
257 g_buffered_output_stream_set_auto_grow (GBufferedOutputStream *stream,
258 gboolean auto_grow)
260 GBufferedOutputStreamPrivate *priv;
261 g_return_if_fail (G_IS_BUFFERED_OUTPUT_STREAM (stream));
262 priv = stream->priv;
263 auto_grow = auto_grow != FALSE;
264 if (priv->auto_grow != auto_grow)
266 priv->auto_grow = auto_grow;
267 g_object_notify (G_OBJECT (stream), "auto-grow");
271 static void
272 g_buffered_output_stream_set_property (GObject *object,
273 guint prop_id,
274 const GValue *value,
275 GParamSpec *pspec)
277 GBufferedOutputStream *stream;
279 stream = G_BUFFERED_OUTPUT_STREAM (object);
281 switch (prop_id)
283 case PROP_BUFSIZE:
284 g_buffered_output_stream_set_buffer_size (stream, g_value_get_uint (value));
285 break;
287 case PROP_AUTO_GROW:
288 g_buffered_output_stream_set_auto_grow (stream, g_value_get_boolean (value));
289 break;
291 default:
292 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
293 break;
298 static void
299 g_buffered_output_stream_get_property (GObject *object,
300 guint prop_id,
301 GValue *value,
302 GParamSpec *pspec)
304 GBufferedOutputStream *buffered_stream;
305 GBufferedOutputStreamPrivate *priv;
307 buffered_stream = G_BUFFERED_OUTPUT_STREAM (object);
308 priv = buffered_stream->priv;
310 switch (prop_id)
312 case PROP_BUFSIZE:
313 g_value_set_uint (value, priv->len);
314 break;
316 case PROP_AUTO_GROW:
317 g_value_set_boolean (value, priv->auto_grow);
318 break;
320 default:
321 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
322 break;
327 static void
328 g_buffered_output_stream_finalize (GObject *object)
330 GBufferedOutputStream *stream;
331 GBufferedOutputStreamPrivate *priv;
333 stream = G_BUFFERED_OUTPUT_STREAM (object);
334 priv = stream->priv;
336 g_free (priv->buffer);
338 G_OBJECT_CLASS (g_buffered_output_stream_parent_class)->finalize (object);
341 static void
342 g_buffered_output_stream_init (GBufferedOutputStream *stream)
344 stream->priv = G_TYPE_INSTANCE_GET_PRIVATE (stream,
345 G_TYPE_BUFFERED_OUTPUT_STREAM,
346 GBufferedOutputStreamPrivate);
351 * g_buffered_output_stream_new:
352 * @base_stream: a #GOutputStream.
354 * Creates a new buffered output stream for a base stream.
356 * Returns: a #GOutputStream for the given @base_stream.
357 **/
358 GOutputStream *
359 g_buffered_output_stream_new (GOutputStream *base_stream)
361 GOutputStream *stream;
363 g_return_val_if_fail (G_IS_OUTPUT_STREAM (base_stream), NULL);
365 stream = g_object_new (G_TYPE_BUFFERED_OUTPUT_STREAM,
366 "base-stream", base_stream,
367 NULL);
369 return stream;
373 * g_buffered_output_stream_new_sized:
374 * @base_stream: a #GOutputStream.
375 * @size: a #gsize.
377 * Creates a new buffered output stream with a given buffer size.
379 * Returns: a #GOutputStream with an internal buffer set to @size.
380 **/
381 GOutputStream *
382 g_buffered_output_stream_new_sized (GOutputStream *base_stream,
383 gsize size)
385 GOutputStream *stream;
387 g_return_val_if_fail (G_IS_OUTPUT_STREAM (base_stream), NULL);
389 stream = g_object_new (G_TYPE_BUFFERED_OUTPUT_STREAM,
390 "base-stream", base_stream,
391 "buffer-size", size,
392 NULL);
394 return stream;
397 static gboolean
398 flush_buffer (GBufferedOutputStream *stream,
399 GCancellable *cancellable,
400 GError **error)
402 GBufferedOutputStreamPrivate *priv;
403 GOutputStream *base_stream;
404 gboolean res;
405 gsize bytes_written;
406 gsize count;
408 priv = stream->priv;
409 bytes_written = 0;
410 base_stream = G_FILTER_OUTPUT_STREAM (stream)->base_stream;
412 g_return_val_if_fail (G_IS_OUTPUT_STREAM (base_stream), FALSE);
414 res = g_output_stream_write_all (base_stream,
415 priv->buffer,
416 priv->pos,
417 &bytes_written,
418 cancellable,
419 error);
421 count = priv->pos - bytes_written;
423 if (count > 0)
424 g_memmove (priv->buffer, priv->buffer + bytes_written, count);
426 priv->pos -= bytes_written;
428 return res;
431 static gssize
432 g_buffered_output_stream_write (GOutputStream *stream,
433 const void *buffer,
434 gsize count,
435 GCancellable *cancellable,
436 GError **error)
438 GBufferedOutputStream *bstream;
439 GBufferedOutputStreamPrivate *priv;
440 gboolean res;
441 gsize n;
442 gsize new_size;
444 bstream = G_BUFFERED_OUTPUT_STREAM (stream);
445 priv = bstream->priv;
447 n = priv->len - priv->pos;
449 if (priv->auto_grow && n < count)
451 new_size = MAX (priv->len * 2, priv->len + count);
452 g_buffered_output_stream_set_buffer_size (bstream, new_size);
454 else if (n == 0)
456 res = flush_buffer (bstream, cancellable, error);
458 if (res == FALSE)
459 return -1;
462 n = priv->len - priv->pos;
464 count = MIN (count, n);
465 memcpy (priv->buffer + priv->pos, buffer, count);
466 priv->pos += count;
468 return count;
471 static gboolean
472 g_buffered_output_stream_flush (GOutputStream *stream,
473 GCancellable *cancellable,
474 GError **error)
476 GBufferedOutputStream *bstream;
477 GBufferedOutputStreamPrivate *priv;
478 GOutputStream *base_stream;
479 gboolean res;
481 bstream = G_BUFFERED_OUTPUT_STREAM (stream);
482 priv = bstream->priv;
483 base_stream = G_FILTER_OUTPUT_STREAM (stream)->base_stream;
485 res = flush_buffer (bstream, cancellable, error);
487 if (res == FALSE)
488 return FALSE;
490 res = g_output_stream_flush (base_stream, cancellable, error);
492 return res;
495 static gboolean
496 g_buffered_output_stream_close (GOutputStream *stream,
497 GCancellable *cancellable,
498 GError **error)
500 GBufferedOutputStream *bstream;
501 GBufferedOutputStreamPrivate *priv;
502 GOutputStream *base_stream;
503 gboolean res;
505 bstream = G_BUFFERED_OUTPUT_STREAM (stream);
506 priv = bstream->priv;
507 base_stream = G_FILTER_OUTPUT_STREAM (bstream)->base_stream;
509 res = flush_buffer (bstream, cancellable, error);
511 /* report the first error but still close the stream */
512 if (res)
513 res = g_output_stream_close (base_stream, cancellable, error);
514 else
515 g_output_stream_close (base_stream, cancellable, NULL);
517 return res;
520 /* ************************** */
521 /* Async stuff implementation */
522 /* ************************** */
524 /* TODO: This should be using the base class async ops, not threads */
526 typedef struct {
528 guint flush_stream : 1;
529 guint close_stream : 1;
531 } FlushData;
533 static void
534 free_flush_data (gpointer data)
536 g_slice_free (FlushData, data);
539 /* This function is used by all three (i.e.
540 * _write, _flush, _close) functions since
541 * all of them will need to flush the buffer
542 * and so closing and writing is just a special
543 * case of flushing + some addition stuff */
544 static void
545 flush_buffer_thread (GSimpleAsyncResult *result,
546 GObject *object,
547 GCancellable *cancellable)
549 GBufferedOutputStream *stream;
550 GOutputStream *base_stream;
551 FlushData *fdata;
552 gboolean res;
553 GError *error = NULL;
555 stream = G_BUFFERED_OUTPUT_STREAM (object);
556 fdata = g_simple_async_result_get_op_res_gpointer (result);
557 base_stream = G_FILTER_OUTPUT_STREAM (stream)->base_stream;
559 res = flush_buffer (stream, cancellable, &error);
561 /* if flushing the buffer didn't work don't even bother
562 * to flush the stream but just report that error */
563 if (res && fdata->flush_stream)
564 res = g_output_stream_flush (base_stream, cancellable, &error);
566 if (fdata->close_stream)
569 /* if flushing the buffer or the stream returned
570 * an error report that first error but still try
571 * close the stream */
572 if (res == FALSE)
573 g_output_stream_close (base_stream, cancellable, NULL);
574 else
575 res = g_output_stream_close (base_stream, cancellable, &error);
578 if (res == FALSE)
580 g_simple_async_result_set_from_error (result, error);
581 g_error_free (error);
585 typedef struct {
587 FlushData fdata;
589 gsize count;
590 const void *buffer;
592 } WriteData;
594 static void
595 free_write_data (gpointer data)
597 g_slice_free (WriteData, data);
600 static void
601 g_buffered_output_stream_write_async (GOutputStream *stream,
602 const void *buffer,
603 gsize count,
604 int io_priority,
605 GCancellable *cancellable,
606 GAsyncReadyCallback callback,
607 gpointer data)
609 GBufferedOutputStream *buffered_stream;
610 GBufferedOutputStreamPrivate *priv;
611 GSimpleAsyncResult *res;
612 WriteData *wdata;
614 buffered_stream = G_BUFFERED_OUTPUT_STREAM (stream);
615 priv = buffered_stream->priv;
617 wdata = g_slice_new (WriteData);
618 wdata->count = count;
619 wdata->buffer = buffer;
621 res = g_simple_async_result_new (G_OBJECT (stream),
622 callback,
623 data,
624 g_buffered_output_stream_write_async);
626 g_simple_async_result_set_op_res_gpointer (res, wdata, free_write_data);
628 /* if we have space left directly call the
629 * callback (from idle) otherwise schedule a buffer
630 * flush in the thread. In both cases the actual
631 * copying of the data to the buffer will be done in
632 * the write_finish () func since that should
633 * be fast enough */
634 if (priv->len - priv->pos > 0)
636 g_simple_async_result_complete_in_idle (res);
638 else
640 wdata->fdata.flush_stream = FALSE;
641 wdata->fdata.close_stream = FALSE;
642 g_simple_async_result_run_in_thread (res,
643 flush_buffer_thread,
644 io_priority,
645 cancellable);
646 g_object_unref (res);
650 static gssize
651 g_buffered_output_stream_write_finish (GOutputStream *stream,
652 GAsyncResult *result,
653 GError **error)
655 GBufferedOutputStreamPrivate *priv;
656 GBufferedOutputStream *buffered_stream;
657 GSimpleAsyncResult *simple;
658 WriteData *wdata;
659 gssize count;
661 simple = G_SIMPLE_ASYNC_RESULT (result);
662 buffered_stream = G_BUFFERED_OUTPUT_STREAM (stream);
663 priv = buffered_stream->priv;
665 g_warn_if_fail (g_simple_async_result_get_source_tag (simple) ==
666 g_buffered_output_stream_write_async);
668 wdata = g_simple_async_result_get_op_res_gpointer (simple);
670 /* Now do the real copying of data to the buffer */
671 count = priv->len - priv->pos;
672 count = MIN (wdata->count, count);
674 memcpy (priv->buffer + priv->pos, wdata->buffer, count);
676 priv->pos += count;
678 return count;
681 static void
682 g_buffered_output_stream_flush_async (GOutputStream *stream,
683 int io_priority,
684 GCancellable *cancellable,
685 GAsyncReadyCallback callback,
686 gpointer data)
688 GSimpleAsyncResult *res;
689 FlushData *fdata;
691 fdata = g_slice_new (FlushData);
692 fdata->flush_stream = TRUE;
693 fdata->close_stream = FALSE;
695 res = g_simple_async_result_new (G_OBJECT (stream),
696 callback,
697 data,
698 g_buffered_output_stream_flush_async);
700 g_simple_async_result_set_op_res_gpointer (res, fdata, free_flush_data);
702 g_simple_async_result_run_in_thread (res,
703 flush_buffer_thread,
704 io_priority,
705 cancellable);
706 g_object_unref (res);
709 static gboolean
710 g_buffered_output_stream_flush_finish (GOutputStream *stream,
711 GAsyncResult *result,
712 GError **error)
714 GSimpleAsyncResult *simple;
716 simple = G_SIMPLE_ASYNC_RESULT (result);
718 g_warn_if_fail (g_simple_async_result_get_source_tag (simple) ==
719 g_buffered_output_stream_flush_async);
721 return TRUE;
724 static void
725 g_buffered_output_stream_close_async (GOutputStream *stream,
726 int io_priority,
727 GCancellable *cancellable,
728 GAsyncReadyCallback callback,
729 gpointer data)
731 GSimpleAsyncResult *res;
732 FlushData *fdata;
734 fdata = g_slice_new (FlushData);
735 fdata->close_stream = TRUE;
737 res = g_simple_async_result_new (G_OBJECT (stream),
738 callback,
739 data,
740 g_buffered_output_stream_close_async);
742 g_simple_async_result_set_op_res_gpointer (res, fdata, free_flush_data);
744 g_simple_async_result_run_in_thread (res,
745 flush_buffer_thread,
746 io_priority,
747 cancellable);
748 g_object_unref (res);
751 static gboolean
752 g_buffered_output_stream_close_finish (GOutputStream *stream,
753 GAsyncResult *result,
754 GError **error)
756 GSimpleAsyncResult *simple;
758 simple = G_SIMPLE_ASYNC_RESULT (result);
760 g_warn_if_fail (g_simple_async_result_get_source_tag (simple) ==
761 g_buffered_output_stream_flush_async);
763 return TRUE;
766 #define __G_BUFFERED_OUTPUT_STREAM_C__
767 #include "gioaliasdef.c"