Use buffer instead of multiple writes

Author: CorvusCorrax

lib/document.coffee

@@ -27,13 +27,25 @@ class PDFDocument extends stream.Readable
     @_waiting = 0
     @_ended = false
     @_offset = 0
+    Pages = @ref
+      Type: 'Pages'
+      Count: 0
+      Kids: new Buffer('')
+
+    Pages.finalize = () ->
+      @offset = @document._offset;
+      @document._write(@id + " " + @gen + " obj");
+      @document._write('<<');
+      @document._write('/Type /Pages');
+      @document._write('/Count ' + @data.Count);
+      @document._write('/Kids [' + @data.Kids.slice(0,-1).toString() + ']');
+      @document._write('>>');
+      @document._write('endobj');
+      return @document._refEnd(@);
 
     @_root = @ref
       Type: 'Catalog'
-      Pages: @ref
-        Type: 'Pages'
-        Count: 0
-        Kids: []
+      Pages: Pages
 
     # The current page
     @page = null
@@ -88,7 +100,7 @@ class PDFDocument extends stream.Readable
 
     # add the page to the object store
     pages = @_root.data.Pages.data
-    pages.Kids.push @page.dictionary
+    pages.Kids = Buffer.concat([pages.Kids, new Buffer(@page.dictionary + ' ')]);
     pages.Count++
 
     # reset x and y coordinates

lib/reference.coffee

@@ -4,63 +4,50 @@ By Devon Govett
 ###
 
 zlib = require 'zlib'
-stream = require 'stream'
 
 class PDFReference
   constructor: (@document, @id, @data = {}) ->
     @gen = 0
-    @deflate = null
     @compress = @document.compress and not @data.Filter
     @uncompressedLength = 0
-    @chunks = []
+    @buffer = new Buffer('')
 
-  initDeflate: ->
-    @data.Filter = 'FlateDecode'
-
-    @deflate = zlib.createDeflate()
-    @deflate.on 'data', (chunk) =>
-      @chunks.push chunk
-      @data.Length += chunk.length
-
-    @deflate.on 'end', @finalize
-
-  write: (chunk, encoding, callback) ->
+  write: (chunk) ->
     unless Buffer.isBuffer(chunk)
       chunk = new Buffer(chunk + '\n', 'binary')
 
     @uncompressedLength += chunk.length
     @data.Length ?= 0
-
+    @buffer = Buffer.concat([@buffer, chunk])
+    @data.Length += chunk.length
     if @compress
-      @initDeflate() if not @deflate
-      @deflate.write chunk
-    else
-      @chunks.push chunk
-      @data.Length += chunk.length
+      @data.Filter = 'FlateDecode'
 
   end: (chunk) ->
-    if @deflate
-      @deflate.end()
-    else
-      @finalize()
+    if chunk
+      @write(chunk)
+    @finalize()
 
   finalize: =>
-    @offset = @document._offset
-
-    @document._write "#{@id} #{@gen} obj"
-    @document._write PDFObject.convert(@data)
+    setTimeout () =>
+      @offset = @document._offset
 
-    if @chunks.length
-      @document._write 'stream'
-      for chunk in @chunks
-        @document._write chunk
+      @document._write "#{@id} #{@gen} obj"
+      @document._write PDFObject.convert(@data)
 
-      @chunks.length = 0 # free up memory
-      @document._write '\nendstream'
+      if @buffer.length
+        if @compress
+          @buffer = zlib.deflateSync(@buffer)
+          @data.Length = @buffer.length
+        @document._write 'stream'
+        @document._write @buffer
 
-    @document._write 'endobj'
-    @document._refEnd(this)
+        @buffer.length = 0 # free up memory
+        @document._write '\nendstream'
 
+      @document._write 'endobj'
+      @document._refEnd(this)
+    , 0
   toString: ->
     return "#{@id} #{@gen} R"