jpg.rb in prawn-0.13.0

- old
+ new

@@ -11,24 +11,29 @@
 module Prawn
   module Images
     # A convenience class that wraps the logic for extracting the parts
     # of a JPG image that we need to embed them in a PDF
     #
-    class JPG 
+    class JPG < Image
       attr_reader :width, :height, :bits, :channels
       attr_accessor :scaled_width, :scaled_height
-      
+
       JPEG_SOF_BLOCKS = %W(\xc0 \xc1 \xc2 \xc3 \xc5 \xc6 \xc7 \xc9 \xca \xcb \xcd \xce \xcf)
       JPEG_APP_BLOCKS = %W(\xe0 \xe1 \xe2 \xe3 \xe4 \xe5 \xe6 \xe7 \xe8 \xe9 \xea \xeb \xec \xed \xee \xef)
 
+      def self.can_render?(image_blob)
+        image_blob[0, 3].unpack("C*") == [255, 216, 255]
+      end
+
       # Process a new JPG image
       #
       # <tt>:data</tt>:: A binary string of JPEG data
       #
       def initialize(data)
-        @data = data
-        data = StringIO.new(data.dup)
+        @data = data.dup
+        ruby_19 { data.force_encoding("binary") }
+        data = StringIO.new(data)
 
         c_marker = "\xff" # Section marker.
         data.read(2)   # Skip the first two bytes of JPEG identifier.
         loop do
           marker, code, length = data.read(4).unpack('aan')
@@ -37,11 +42,11 @@
           if JPEG_SOF_BLOCKS.include?(code)
             @bits, @height, @width, @channels = data.read(6).unpack("CnnC")
             break
           end
 
-          buffer = data.read(length - 2)
+          data.read(length - 2)
         end
       end
 
       # Build a PDF object representing this image in +document+, and return
       # a Reference to it.
@@ -59,25 +64,24 @@
         end
 
         obj = document.ref!(
           :Type             => :XObject,
           :Subtype          => :Image,
-          :Filter           => :DCTDecode,
           :ColorSpace       => color_space,
           :BitsPerComponent => bits,
           :Width            => width,
-          :Height           => height,
-          :Length           => @data.size
-        ) 
+          :Height           => height
+        )
 
         # add extra decode params for CMYK images. By swapping the
         # min and max values from the default, we invert the colours. See
         # section 4.8.4 of the spec.
         if color_space == :DeviceCMYK
           obj.data[:Decode] = [ 1.0, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0, 0.0 ]
         end
 
-        obj << @data
+        obj.stream << @data
+        obj.stream.filters << :DCTDecode
         obj
       end
 
     end
   end