fn_document 0.9.2

data/lib/fn/pdf/node/end_page_ext.rb

@@ -0,0 +1,22 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      
+      def EndPageExt(number)
+        FN::Node::Base("end_page_ext", :pagenumber => number).extend(EndPageExt)
+      end
+      
+      module EndPageExt 
+        include FN::Node::Base
+      
+        def visit(struct)
+          has_no_children
+          struct.resume_page(attributes.to_h)
+          struct.end_page_ext("")
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/fit_image.rb

@@ -0,0 +1,28 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      def FitImage(image, x, y, opts = {})
+        if opts.delete(:inverted)
+          opts[:orientate] = "south"
+        end
+        FN::Node::Base("fit_image", opts.merge(:image => image, :x => x, :y => y)).extend(FitImage)
+      end
+      
+      module FitImage 
+        include FN::Node::Base
+      
+        def visit(struct)
+          has_no_children
+          h = attributes.to_h
+          img = struct[h.delete("image")]
+          x = h.delete("x").to_i
+          y = h.delete("y").to_i
+          h["scale"] = 0.99 # if h["scale"] == "1.0"  
+          struct.fit_image(img, x, y, h)
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/fit_pdi_page.rb

@@ -0,0 +1,26 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      def FitPdiPage(var)
+        FN::Node::Base("fit_pdi_page", :page => var)
+      end
+      
+      module FitPdiPage 
+        include FN::Node::Base
+        
+        def visit(struct)
+          @logger = Logger.new("#{RAILS_ROOT}/log/pdf_writer_logs/#{Time.now.strftime('pdf_writer_log_%Y_%m_%d')}.log")
+          # @logger.info "HELP "*88 
+          # @logger.info "self[:page] => #{self[:page]}" # yields {page}
+          # @logger.info "struct[self[:page]] => #{struct[self[:page]]}" # yields -1
+          # @logger.info "struct => #{struct.inspect}"
+          has_no_children
+          struct.fit_pdi_page(struct[self[:page]], 0, struct[CURRENT_PAGE_HEIGHT], "")
+          
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/fit_textflow.rb

@@ -0,0 +1,27 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      def FitTextflow(flow, block)
+        FN::Node::Base("fit_textflow", :flow => "{#{flow && flow.flow_name}}", 
+              :x  => block["x"], 
+              :y  => block["y"],
+              :x2 => block["x"].to_f + block["width"].to_f,
+              :y2 => block["y"].to_f + block["height"].to_f
+            ).extend(FitTextflow)
+      end
+      
+      module FitTextflow 
+        include FN::Node::Base
+      
+        def visit(struct)
+          has_no_children
+          struct.fit_textflow struct[self["flow"]], 
+              self["x"].to_f,  self["y"].to_f, 
+              self["x2"].to_f, self["y2"].to_f, ""
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/invert.rb

@@ -0,0 +1,36 @@
+# if block['flip'] == "yes"
+#   @pdf.save
+#   @pdf.translate block["x"].to_f + block["x2"].to_f, block["y"].to_f + block["y2"].to_f
+#   @pdf.rotate 180
+#   @pdf.fit_textflow flow, block["x"].to_f, block["y"].to_f, block["x2"].to_f, block["y2"].to_f, ""
+#   @pdf.restore
+# else
+
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      def Invert(block)
+        FN::Node::Base("invert", 
+              :x  => block["x"], 
+              :y  => block["y"],
+              :x2 => block["x"].to_f + (block["boxWidth"] || block["width"]).to_f,
+              :y2 => block["y"].to_f + (block["boxHeight"] || block["height"]).to_f
+            ).extend(Invert)
+      end
+      
+      module Invert 
+        include FN::Node::Base
+      
+        def visit(struct)
+          struct.save
+          struct.translate self["x"].to_f + self["x2"].to_f, self["y"].to_f + self["y2"].to_f
+          struct.rotate 180
+          visit_children struct
+          struct.restore
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/load_image.rb

@@ -0,0 +1,21 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      def LoadImage(file, var)
+        FN::Node::Base("load_image", :file => file, :assigns => var).extend(LoadImage)
+      end
+      
+      module LoadImage 
+        include FN::Node::Base
+        
+        def visit(struct)
+          has_no_children
+          img = struct.load_image("auto", self[:file], "")
+          struct.assigns self, img
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/open_pdi.rb

@@ -0,0 +1,22 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      def OpenPdi(file, var)
+        FN::Node::Base("open_pdi", :file => file, :assigns => var).extend(OpenPdi)
+      end
+      
+      module OpenPdi 
+        include FN::Node::Base
+      
+        def visit(struct)
+          pdi = struct.open_pdi(self[:file], "", 0)
+          struct.assigns(self, pdi)
+          visit_children struct
+          struct.close_pdi(pdi)
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/open_pdi_page.rb

@@ -0,0 +1,31 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+
+      def OpenPdiPage(pdi_var, page_number, page_var)
+        FN::Node::Base("open_pdi_page", :pdi => pdi_var, :number => page_number, :assigns => page_var).extend(OpenPdiPage)
+      end
+      
+      module OpenPdiPage 
+        include FN::Node::Base
+      
+        def visit(struct)
+          @logger = Logger.new("#{RAILS_ROOT}/log/pdf_writer_logs/#{Time.now.strftime('pdf_writer_log_%Y_%m_%d')}.log")
+          # @logger.info "DOGS "*88 
+          # @logger.info "struct.inspect: #{struct.inspect}"
+          # @logger.info "struct[self[:pdi]]: #{struct[self[:pdi]]}"
+          # @logger.info "self[:number].to_i: #{self[:number].to_i}"
+          pg = struct.open_pdi_page(struct[self[:pdi]], self[:number].to_i, "")
+          # @logger.info "self.inspect: #{self.inspect}" 
+          # @logger.info "pg.inspect: #{pg.inspect}"
+          # @logger.info "CATS "*88 
+          struct.assigns self, pg
+          visit_children struct
+          struct.close_pdi_page(pg)
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/resume_page.rb

@@ -0,0 +1,21 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      def ResumePage(number)
+        FN::Node::Base("resume_page", :pagenumber => number).extend(ResumePage)
+      end
+      
+      module ResumePage 
+        include FN::Node::Base
+      
+        def visit(struct)
+          struct.resume_page(attributes.to_h)
+          visit_children(struct)
+          struct.suspend_page("")
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/set_parameter.rb

@@ -0,0 +1,22 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      
+      def SetParameter(key, value)
+        FN::Node::Base("set_parameter", :key => key, :value => value).extend(SetParameter)
+      end
+      
+      module SetParameter 
+        include FN::Node::Base
+      
+        def visit(struct)
+          has_no_children
+          struct[self[:key]] = self[:value]
+          struct.set_parameter(self[:key], self[:value])
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/node/watermark.rb

@@ -0,0 +1,24 @@
+require File.dirname(__FILE__) + "/../../node/base"
+require "PDFlib"
+module FN
+  module PDF
+    module Node
+      
+      def Watermark(text)
+        FN::Node::Base("watermark", :text => text).extend(Watermark)
+      end
+      
+      module Watermark
+        include FN::Node::Base
+      
+        def visit(struct)
+          has_no_children
+          w = struct[CURRENT_PAGE_WIDTH]
+          h = struct[CURRENT_PAGE_HEIGHT]
+          font = struct.load_font("Arial,Bold", "unicode", "")
+          struct.fit_textline(self["text"], 0, h, "font #{font} fontsize 30 boxsize {#{w} #{h}} fitmethod meet rotate 0 textrendering 1 position {0 50}")
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/struct.rb

@@ -0,0 +1,40 @@
+require "PDFlib"
+module FN
+  module PDF
+    class Struct < Hash
+      attr_reader :pdf
+      
+      def initialize(debug = false)
+        @pdf = PDFlib.new
+        @debug = debug
+      end  
+      
+      def assigns(node, value)
+        self["{#{node[:assigns]}}"] = value
+      end
+      
+      def method_missing(*args, &block)
+        @logger = Logger.new("#{RAILS_ROOT}/log/pdf_writer_logs/#{Time.now.strftime('pdf_writer_log_%Y_%m_%d')}.log")
+        # @logger.info("This is what args looks like when we first get it: #{a.inspect}")
+        args.map! do |elem|
+          case elem
+          when Hash:
+            elem.inject([]) {|m, (key, value)| 
+              m << "#{key}={#{value}}"
+            }.join(" ")
+          else
+            elem
+          end
+        end
+        begin
+          # @logger.info("Calling #{args.inspect}, this struct: #{self.inspect}")
+          @pdf.send(*args, &block)
+        rescue Exception => error
+          # @logger.info("FAILURE FAILURE FAILURE FAILURE FAILURE FAILURE")
+          # @logger.info("Exception: #{error}")
+          raise error
+        end
+      end
+    end
+  end
+end

data/lib/fn/pdf/writer.rb

@@ -0,0 +1,205 @@
+require "rubygems"
+require "PDFlib"
+require "RMagick"
+require "tempfile"
+Dir[File.dirname(__FILE__) + "/node/*.rb"].each do |f|
+  require_dependency f.sub(/\.rb$/, '')
+end
+module FN
+  module PDF
+    class WriterError < RuntimeError; end
+    class Writer
+      include Node
+      SPACE = /[\s_]+/
+      
+      def self.encoding=(e)
+        @encoding = e
+      end
+      
+      def self.encoding
+        @encoding
+      end
+      
+      def write(doc, options = {})
+        options[:save_as] ||= (tmp = Tempfile.new("pdf"); tmp.close; tmp.path)
+        write_xml translate(doc, options), options[:save_as]
+      end
+      
+      def write_xml(xml, file)
+        xml.root.extend(FN::Node::Root)
+        xml.root.visit(FN::PDF::Struct.new)
+        return file
+      end
+      
+      def titlecase(str)
+        face = str.split(SPACE).map{|s| s.capitalize }.join("")
+      end
+      
+      def translate(doc, options = {})
+        raise "Not an FN Document" unless doc.is_a?(FN::Document)
+        @logger = Logger.new("#{RAILS_ROOT}/log/pdf_writer_logs/#{Time.now.strftime('pdf_writer_log_%Y_%m_%d')}.log")
+        
+        
+        debug         = options[:debug]
+        file          = options[:save_as]
+        bkg           = options[:background]
+        pdf_version   = options[:pdf_version] ||
+                        1.5
+        license       = options[:license]
+        resource_file = options[:font_list] ||
+                        File.join(File.dirname(__FILE__), "pdflib.upr")
+        self.class.encoding      = options[:encoding] || 
+                        "unicode"
+        textformat    = options[:textformat] || "utf8"
+        root          = options[:resource_root] ||
+                        ""
+        watermark     = options[:watermark]
+        
+        context = FN::Node::Context.new
+        
+        # @logger.info "===== Setting Parameters ====="
+        context.add SetParameter("license", license)
+        context.add SetParameter("resourcefile", resource_file)
+        context.add SetParameter("hypertextencoding", self.class.encoding)
+        context.add SetParameter("textformat", textformat)
+        
+        if bkg
+          # @logger.info "===== This Document Has a Background ====="
+          context << OpenPdi(bkg, assigns = "pdi")      
+        end
+        
+        # @logger.info "===== Uh... Begin Document ====="
+        context << BeginDocument(file, pdf_version)
+        
+        # @logger.info "===== Setting A Forgotten Parameter ====="
+        context.add SetParameter("topdown", true)
+        
+        # @logger.info "===== Adding #{doc.fonts.length} Fonts ====="
+        doc.fonts do |name|
+          LoadFont.add_all_variants_to context, name
+        end
+        
+        # @logger.info "===== Adding #{doc.textflows.length} Text Flows ====="
+        flows_by_name = {}
+        doc.textflows.each do |flow|
+          ctf = CreateTextflow(flow)
+          flows_by_name[ctf.flow_name] = ctf
+          context.add ctf
+        end
+        
+        
+        # @logger.info "===== Adding #{doc.pages.length} Pages ====="
+        doc.pages.each_with_index do |page,index|
+          context.retain_after do
+            # @logger.info "Page #{index+1}:"
+            # @logger.info "BeginPageExt(:width => #{page["width"]}, :height => #{page["height"]}, :number => #{page["number"]})"
+            context << BeginPageExt(page["width"], page["height"], page["number"])
+            if bkg
+              # @logger.info "This Document Has Background"
+              # @logger.info "OpenPdiPage(:pdi_var => \"{pdi}\", :page_number => #{page["number"]}, :page_var => \"page\")"
+              context <<  OpenPdiPage("{pdi}", page["number"], assigns = "page" )
+              # @logger.info "FitPdiPage(:var => \"{page}\")"
+              context.add FitPdiPage("{page}")
+            elsif page["background"]
+              # @logger.info "Page #{index+1}: This Document does NOT have a Background, but the page does"
+              bkg_image = doc.resource(page["background"]).path_from(root)
+              context.add LoadImage(bkg_image, "tmp")
+              context.add FitImage("{tmp}", 0, page["height"], 
+                                       :fitmethod => :meet, 
+                                       :boxsize => [page["width"], page["height"]])
+            else
+              # @logger.info "Page #{index+1}: Does not have a background at all"
+            end
+            context.add(Watermark(watermark)) if watermark
+          end
+        end
+        
+        # @logger.info "===== Adding #{doc.text_blocks.length} Text Blocks ====="
+        doc.text_blocks.each do |block|
+          flow = CreateTextflow(block)
+          
+          context.retain_after do
+            key = "flow-#{block.text}"
+            if flow.empty?
+              flow = flows_by_name[key]
+            else
+              flows_by_name.delete(key)
+              context.add(flow)
+            end
+            
+            if flow
+              context << ResumePage(block.page_number)
+              context << Invert(block) if block["flip"] == "yes"
+              context.add FitTextflow(flow, block)
+            end
+          end
+        end
+        
+        # @logger.info "===== Adding #{doc.photo_blocks.length} Photo Blocks ====="
+        doc.photo_blocks.each do |block|
+          # require "ruby-debug"
+          # debugger
+          image = nil
+          begin
+            context.inject_at_page(block.page_number) do 
+              image = doc.resource(block.src).path_from(root)
+              tmp = Magick::Image::read(image).first
+              dims = [tmp.columns.to_f, tmp.rows.to_f]
+              x, y, width, height = calculate(block, dims)
+              fi = FitImage("{tmp}", x, y, :fitmethod => "meet", :boxsize => [width, height], :inverted => block["flip"] == "yes")
+              context.add LoadImage(image, "tmp")
+              context.add(fi)
+            end
+          rescue Magick::ImageMagickError => e
+            STDERR.puts e.message
+            STDERR.puts e.backtrace.join("\n")
+            raise WriterError.new("Couldn't load '#{block.src}', given by #{doc.resource(block.src).node}")
+          end
+        end
+        
+        # @logger.info "===== Closing #{doc.pages.length} Pages ====="
+        doc.pages.each do |page|
+          context.add EndPageExt(page["number"])
+        end
+        
+        
+        # @logger.info "====================================================================="
+        # @logger.info "===== Somewhere after this point, the major churning begins     ====="
+        # @logger.info "===== And starts looping through all the context things we just ====="
+        # @logger.info "===== added and tries to execute them all                       ====="
+        # @logger.info "====================================================================="
+        
+        # @logger.info "================================== doc.xml_to_s ================================="
+        # @logger.info doc.xml_to_s
+        # @logger.info "================================== context.doc.to_s ============================="
+        # @logger.info context.doc.to_s
+        return context.doc
+      end
+      
+      
+      def calculate(block, dims)
+        x = bx = block["boxX"].to_i
+        y = by = block["boxY"].to_i
+        bw = block["boxWidth"].to_i
+        bh = block["boxHeight"].to_i
+        aw = block["width"].to_i
+        iw, ih = dims
+        ah =  aw * ih / iw
+        case block["align"]
+        when /right/:   x = bx + bw - aw
+        when /center/:  x = bx + (bw - aw) / 2
+        else            x = bx
+        end
+        
+        case block["align"]
+        when /bottom/:  y = by + bh 
+        when /middle/:  y = by + bh / 2
+        else            y = by
+        end
+        
+        return [x, y, aw, ah]
+      end
+      
+    end
+  end
+end