lib/mireru/widget/pdf.rb in mireru-0.2.1 vs lib/mireru/widget/pdf.rb in mireru-0.9.0

@@ -1,31 +1,49 @@
+require "gtk3"
 require "poppler"
 
 module Mireru
   module Widget
-    class PDF
-      class << self
-        def create(file)
-          drawing_area = Gtk::DrawingArea.new
+    class PDF < Gtk::DrawingArea
+      def initialize(file)
+        super()
+        document = Poppler::Document.new(file)
+        @page_index = 0
+        @page_max = document.size - 1
+        width, height = document.first.size
+        set_size_request(width, height)
 
-          document = Poppler::Document.new(file)
-          width, height = document.first.size
-          drawing_area.set_size_request(width, height * document.size)
-
-          drawing_area.signal_connect("draw") do |widget, event|
-            context = widget.window.create_cairo_context
-            document.each_with_index do |page, i|
-              context.save do
-                context.translate(0, height * i)
-                context.render_poppler_page(document[i])
-              end
-            end
-            context.show_page
-            true
+        signal_connect("draw") do |widget, event|
+          context = widget.window.create_cairo_context
+          window_width = widget.allocated_width
+          window_height = widget.allocated_height
+          width_scale = window_width.to_f / width
+          height_scale = window_height.to_f / height
+          scale = [width_scale, height_scale].min
+          begin
+            context.scale(scale, scale)
+          rescue => e
+            $stderr.puts("#{e.class}: #{e.message}")
+            $stderr.puts(e.backtrace)
           end
-
-          drawing_area
+          context.render_poppler_page(document[@page_index])
+          context.show_page
+          true
         end
+      end
+
+      def next
+        @page_index += 1
+        @page_index = @page_max if @page_index > @page_max
+        hide
+        show
+      end
+
+      def prev
+        @page_index -= 1
+        @page_index = 0 if @page_index < 0
+        hide
+        show
       end
     end
   end
 end