flower/memory-stream.cc

   1 /*
   2   memory-stream.cc --  implement Memory_out_stream
   3
   4   source file of the GNU LilyPond music typesetter
   5
   6   (c) 2005 Han-Wen Nienhuys <hanwen@xs4all.nl>
   7
   8 */
   9
  10 #include <assert.h>
  11 #include <string.h>
  12 #include <stdlib.h>
  13
  14 #include "memory-stream.hh"
  15
  16
  17 /*
  18   TODO: add read support as well.
  19 */
  20 const int Memory_out_stream::block_size_ = 1024;
  21
  22 cookie_io_functions_t
  23 Memory_out_stream::functions_ = {
  24   Memory_out_stream::reader,
  25   Memory_out_stream::writer,
  26   Memory_out_stream::seeker,
  27   Memory_out_stream::cleaner
  28 };
  29
  30 int
  31 Memory_out_stream::cleaner (void *cookie)
  32 {
  33   Memory_out_stream *stream = (Memory_out_stream*) cookie;
  34
  35   stream->file_ = 0;
  36   return 0;
  37 }
  38
  39 Memory_out_stream::Memory_out_stream ()
  40 {
  41   size_ = 0;
  42   buffer_ = 0;
  43   buffer_blocks_ = 0;
  44   file_ = fopencookie ((void*) this, "w", functions_);
  45 }
  46
  47 Memory_out_stream::~Memory_out_stream ()
  48 {
  49   if (file_)
  50     fclose (file_);
  51
  52   free (buffer_);
  53 }
  54
  55 FILE *
  56 Memory_out_stream::get_file () const
  57 {
  58   return file_;
  59 }
  60
  61 ssize_t
  62 Memory_out_stream::get_length () const
  63 {
  64   return size_;
  65 }
  66
  67 const char *
  68 Memory_out_stream::get_string () const
  69 {
  70   return buffer_;
  71 }
  72
  73 ssize_t
  74 Memory_out_stream::writer (void *cookie,
  75                            const char *buffer,
  76                            size_t size)
  77 {
  78   Memory_out_stream *stream = (Memory_out_stream*) cookie;
  79
  80   ssize_t newsize = stream->size_ + size;
  81
  82   bool change = false;
  83   while (newsize > stream->buffer_blocks_ * block_size_)
  84     {
  85       stream->buffer_blocks_ *= 2;
  86       stream->buffer_blocks_ += 1;
  87       change = true;
  88     }
  89
  90   if (change)
  91     stream->buffer_ = (char*) realloc (stream->buffer_,
  92                                        stream->buffer_blocks_ * block_size_);
  93
  94   memcpy (stream->buffer_ + stream->size_, buffer, size);
  95   stream->size_ = newsize;
  96
  97   return size;
  98 }
  99
 100 ssize_t
 101 Memory_out_stream::reader (void *cookie,
 102                            char *buffer,
 103                            size_t size)
 104 {
 105   (void) cookie;
 106   (void) buffer;
 107   (void) size;
 108
 109   assert (false);
 110   return 0;
 111 }
 112
 113 int
 114 Memory_out_stream::seeker (void *, off64_t *, int whence)
 115 {
 116   (void) whence;
 117
 118   assert (false);
 119   return 0;
 120 }