fixed wrong argument in app.py
[swftools.git] / spec / spec_helper.rb
1 require 'spec'
2 require 'rubygems'
3 require 'RMagick'
4
5 class WrongColor < Exception
6     def initialize(pixel)
7         @pixel = pixel
8     end
9     def to_s
10         "Wrong color at #{@pixel}"
11     end
12 end
13 class AreaError < Exception
14     def initialize(area,problem)
15         @area,@problem = area,problem
16     end
17     def to_s
18         "Area at #{@area} #{@problem}"
19     end
20 end
21 class PixelError < Exception
22     def initialize(p1, relation,p2)
23         @p1,@p2,@relation = p1,p2,relation
24     end
25     def to_s
26         "Pixel #{@p1} #{@relation} #{@p2}"
27     end
28 end
29 class ConversionFailed < Exception
30     def initialize(output,file)
31         @output = output
32         @file = file
33         @exists = File.exists?(file)
34     end
35     def to_s
36         puts "-"*26+" Conversion failed "+"-"*27
37         (puts @output) if @output
38         puts "file #{@file} doesn't exist" if not @exists
39         puts "-"*72
40     end
41 end
42
43 class Area
44     def initialize(x1,y1,x2,y2,file)
45         @x1,@y1,@x2,@y2,@file = x1,y1,x2,y2,file
46     end
47     def should_be_plain_colored
48         @rgb = @file.get_area(@x1,@y1,@x2,@y2) unless @rgb
49         @rgb.minmax == [@rgb[0],@rgb[0]] or raise AreaError.new(self,"is not plain colored")
50     end
51     def should_not_be_plain_colored
52         @rgb = @file.get_area(@x1,@y1,@x2,@y2) unless @rgb
53         @rgb.minmax != [@rgb[0],@rgb[0]] or raise AreaError.new(self,"is plain colored")
54     end
55     def should_contain_text(text)
56         text2 = @file.get_text(@x1,@y1,@x2,@y2) 
57         text2 == text or raise AreaError.new(self, "doesn't contain text \"#{text}\" (found: \"#{text2}\")")
58     end
59     def to_s
60         "(#{@x1},#{@y1},#{@x2},#{@y2})"
61     end
62 end
63
64 def rgb_to_int(rgb)
65     # ImageMagick rgb triples are 16 bit
66     (rgb.reverse+[0]).map {|c| c>>8}.pack("CCCC").unpack("i")[0]
67 end
68
69 class Pixel
70     attr :rgb
71     def initialize(x,y,rgb)
72         @x,@y,@rgb = x,y,rgb
73     end
74     def should_be_of_color(color2)
75         color1 = rgb_to_int(@rgb)
76         color1 == color2 or raise WrongColor.new(self)
77     end
78     def should_be_brighter_than(pixel)
79         gray1 = @rgb.inject(0) {|sum,e| sum+e}
80         gray2 = pixel.rgb.inject(0) {|sum,e| sum+e}
81         gray1 > gray2 or raise PixelError.new(self,"is not brighter than",pixel)
82     end
83     def should_be_darker_than(pixel)
84         gray1 = @rgb.inject(0) {|sum,e| sum+e}
85         gray2 = pixel.rgb.inject(0) {|sum,e| sum+e}
86         gray1 < gray2 or raise PixelError.new(self,"is not less bright than",pixel)
87     end
88     def should_be_the_same_as(pixel)
89         @rgb == pixel.rgb or raise PixelError.new(self,"is not the same as",pixel)
90     end
91     def to_s
92         "(#{@x},#{@y})"
93     end
94 end
95
96 $tempfiles = []
97 Kernel.at_exit do
98     $tempfiles.each do |file|
99         `rm -f #{file}`
100     end
101 end
102
103 class DocFile
104     def initialize(filename, page)
105         @filename = filename
106         @page = page
107     end
108     def convert()
109         return if @swfname
110         @swfname = @filename.gsub(/.pdf$/i,"")+".swf"
111         $tempfiles += [@swfname]
112         `pdfinfo #{@filename}` =~ /Page size:\s*([0-9]+) x ([0-9]+) pts/
113         width,height = $1,$2
114         dpi = (72.0 * 612 / width.to_i).to_i
115         output = `pdf2swf -f --flatten -s zoom=#{dpi} -p #{@page} #{@filename} -o #{@swfname} 2>&1`
116         #output = `pdf2swf -s zoom=#{dpi} --flatten -p #{@page} #{@filename} -o #{@swfname} 2>&1`
117         raise ConversionFailed.new(output,@swfname) unless File.exists?(@swfname)
118     end
119     def render()
120         return if @img
121         convert()
122         @pngname = @filename.gsub(/.pdf$/i,"")+".png"
123         begin
124             output = `swfrender --legacy #{@swfname} -o #{@pngname} 2>&1`
125             raise ConversionFailed.new(output,@pngname) unless File.exists?(@pngname)
126             @img = Magick::Image.read(@pngname).first
127         ensure
128             `rm -f #{@pngname}`
129         end
130     end
131     def get_text(x1,y1,x2,y2)
132         self.convert()
133         #puts "swfstrings -x #{x1} -y #{y1} -W #{x2-x1} -H #{y2-y1} #{@swfname}"
134         #puts `swfstrings -x #{x1} -y #{y1} -W #{x2-x1} -H #{y2-y1} #{@swfname}`
135         `swfstrings -x #{x1} -y #{y1} -W #{x2-x1} -H #{y2-y1} #{@swfname}`.chomp
136     end
137     def get_area(x1,y1,x2,y2)
138         self.render()
139         data = @img.export_pixels(x1, y1, x2-x1, y2-y1, "RGB")
140         Array.new(data.size/3) do |i| data.slice(i*3,3) end
141     end
142     def area_at(x1,y1,x2,y2)
143         return Area.new(x1,y1,x2,y2,self)
144     end
145     def width()
146         self.render()
147         return @img.columns
148     end
149     def height()
150         self.render()
151         return @img.rows
152     end
153     def pixel_at(x,y)
154         self.render()
155         data = @img.export_pixels(x, y, 1, 1, "RGB")
156         return Pixel.new(x,y,data)
157     end
158 end
159
160 module Spec
161     module Example
162         module ExampleMethods
163            def area_at(x1,y1,x2,y2)
164                @file.area_at(x1,y1,x2,y2)
165            end
166            def width
167                @file.width
168            end
169            def height
170                @file.height
171            end
172            def pixel_at(x,y)
173                @file.pixel_at(x,y)
174            end
175            def initialize(proxy,&impl)
176
177                # overriding the initialize() function saves us from having to
178                # set up the document in the test. The bad news, however, is that
179                # we have to be very careful not to raise exceptions here-
180                # rspec would miss those.
181
182                @_proxy = proxy
183                @_implementation = impl
184                @_backtrace = caller
185                @file = DocFile.new(proxy.description, (proxy.options[:page] or 1))
186            end
187         end
188         module ExampleGroupMethods
189             alias :convert_file :example
190             def page(nr)
191                 nr
192             end
193         end
194     end
195 end
196
197