X-Git-Url: http://git.asbjorn.biz/?a=blobdiff_plain;f=spec%2Fspec_helper.rb;h=1ee0b39da6c1c7fedc964fe8b3527b67f7504c63;hb=5c336b4f7017fb121fc2cbe2a02c1fa56986b560;hp=d1c145817379d44c9fb65311e2d533bd5e7898cb;hpb=f42e0edd3094808c319a89594ae6f55843a0a3bf;p=swftools.git

diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
index d1c1458..1ee0b39 100644
--- a/spec/spec_helper.rb
+++ b/spec/spec_helper.rb
@@ -41,16 +41,20 @@ class ConversionFailed < Exception
 end
 
 class Area
-    def initialize(x1,y1,x2,y2,data)
-	@x1,@y1,@x2,@y2 = x1,y1,x2,y2
-	@rgb = Array.new(data.size/3) do |i| data.slice(i*3,3) end
+    def initialize(x1,y1,x2,y2,file)
+	@x1,@y1,@x2,@y2,@file = x1,y1,x2,y2,file
     end
     def should_be_plain_colored
+	@rgb = @file.get_area(@x1,@y1,@x2,@y2) unless @rgb
 	@rgb.minmax == [@rgb[0],@rgb[0]] or raise AreaError.new(self,"is not plain colored")
     end
     def should_not_be_plain_colored
+	@rgb = @file.get_area(@x1,@y1,@x2,@y2) unless @rgb
 	@rgb.minmax != [@rgb[0],@rgb[0]] or raise AreaError.new(self,"is plain colored")
     end
+    def should_contain_text(text)
+	@file.get_text(@x1,@y1,@x2,@y2) == text or raise AreaError.new(self, "doesn't contain text #{text}")
+    end
     def to_s
 	"(#{@x1},#{@y1},#{@x2},#{@y2})"
     end
@@ -92,15 +96,21 @@ class DocFile
 	@filename = filename
 	@page = page
     end
-    def load()
+    def convert()
+	return if @swfname
 	@swfname = @filename.gsub(/.pdf$/i,"")+".swf"
+	`pdfinfo #{@filename}` =~ /Page size:\s*([0-9]+) x ([0-9]+) pts/
+	width,height = $1,$2
+	dpi = (72.0 * 612 / width.to_i).to_i
+	output = `pdf2swf --flatten -s zoom=#{dpi} -p #{@page} #{@filename} -o #{@swfname} 2>&1`
+	#output = `pdf2swf -s zoom=#{dpi} --flatten -p #{@page} #{@filename} -o #{@swfname} 2>&1`
+	raise ConversionFailed.new(output,@swfname) unless File.exists?(@swfname)
+    end
+    def render()
+	return if @img
+	convert()
 	@pngname = @filename.gsub(/.pdf$/i,"")+".png"
 	begin
-	    `pdfinfo #{@filename}` =~ /Page size:\s*([0-9]+) x ([0-9]+) pts/
-	    width,height = $1,$2
-	    dpi = (72.0 * 612 / width.to_i).to_i
-	    output = `pdf2swf -s zoom=#{dpi} --flatten -p #{@page} #{@filename} -o #{@swfname} 2>&1`
-	    raise ConversionFailed.new(output,@swfname) unless File.exists?(@swfname)
 	    output = `swfrender --legacy #{@swfname} -o #{@pngname} 2>&1`
 	    raise ConversionFailed.new(output,@pngname) unless File.exists?(@pngname)
 	    @img = Magick::Image.read(@pngname).first
@@ -109,21 +119,30 @@ class DocFile
 	    `rm -f #{@pngname}`
 	end
     end
-    def area_at(x1,y1,x2,y2)
-	self.load()
+    def get_text(x1,y1,x2,y2)
+	self.convert()
+	puts "swfstrings -x #{x1} -y #{y1} -W #{x2-x1} -H #{y2-y1} #{@swfname}"
+	puts `swfstrings -x #{x1} -y #{y1} -W #{x2-x1} -H #{y2-y1} #{@swfname}`
+	`swfstrings -x #{x1} -y #{y1} -W #{x2-x1} -H #{y2-y1} #{@swfname}`
+    end
+    def get_area(x1,y1,x2,y2)
+	self.render()
 	data = @img.export_pixels(x1, y1, x2-x1, y2-y1, "RGB")
-	return Area.new(x1,y1,x2,y2,data)
+	Array.new(data.size/3) do |i| data.slice(i*3,3) end
+    end
+    def area_at(x1,y1,x2,y2)
+	return Area.new(x1,y1,x2,y2,self)
     end
     def width()
-	self.load()
+	self.render()
 	return @img.columns
     end
     def height()
-	self.load()
+	self.render()
 	return @img.rows
     end
     def pixel_at(x,y)
-	self.load()
+	self.render()
 	data = @img.export_pixels(x, y, 1, 1, "RGB")
 	return Pixel.new(x,y,data)
     end