Read the title from the legend file.

gravitystorm · Aug 21, 2014 · 6eb4914 · 6eb4914
1 parent 40848fb
commit 6eb4914
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/lib/mapnik_legendary.rb b/lib/mapnik_legendary.rb
@@ -81,6 +81,7 @@ def self.generate_legend(legend_file, map_file, options)
     f.write(docs.to_html)
     f.close
 
-    docs.to_pdf(File.join(Dir.pwd, 'output', 'legend.pdf'))
+    title = legend.key?('title') ? legend['title'] : 'Legend'
+    docs.to_pdf(File.join(Dir.pwd, 'output', 'legend.pdf'), title)
   end
 end
diff --git a/lib/mapnik_legendary/docwriter.rb b/lib/mapnik_legendary/docwriter.rb
@@ -22,7 +22,7 @@ def to_html
       doc
     end
 
-    def to_pdf(filename)
+    def to_pdf(filename, title)
       entries = @entries
       Prawn::Document.generate(filename, page_size: 'A4') do
         font_families.update(
@@ -32,7 +32,8 @@ def to_pdf(filename)
         )
         font 'Ubuntu'
         font_size 12
-        text 'Legend', style: :bold, size: 24, align: :center
+        text title, style: :bold, size: 24, align: :center
+        move_down(40)
         data = Array.new
         entries.each do |entry|
           data << { image: File.join(File.dirname(filename), entry[0]),