added support for hardware-gpu & accelerated-blit header flags
[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,data)
45         @x1,@y1,@x2,@y2 = x1,y1,x2,y2
46         @rgb = Array.new(data.size/3) do |i| data.slice(i*3,3) end
47     end
48     def should_be_plain_colored
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.minmax != [@rgb[0],@rgb[0]] or raise AreaError.new(self,"is plain colored")
53     end
54     def to_s
55         "(#{@x1},#{@y1},#{@x2},#{@y2})"
56     end
57 end
58
59 def rgb_to_int(rgb)
60     # ImageMagick rgb triples are 16 bit
61     (rgb.reverse+[0]).map {|c| c>>8}.pack("CCCC").unpack("i")[0]
62 end
63
64 class Pixel
65     attr :rgb
66     def initialize(x,y,rgb)
67         @x,@y,@rgb = x,y,rgb
68     end
69     def should_be_of_color(color2)
70         color1 = rgb_to_int(@rgb)
71         color1 == color2 or raise WrongColor.new(self)
72     end
73     def should_be_brighter_than(pixel)
74         gray1 = @rgb.inject(0) {|sum,e| sum+e}
75         gray2 = pixel.rgb.inject(0) {|sum,e| sum+e}
76         gray1 > gray2 or raise PixelError.new(self,"is not brighter than",pixel)
77     end
78     def should_be_darker_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 less bright than",pixel)
82     end
83     def should_be_the_same_as(pixel)
84         @rgb == pixel.rgb or raise PixelError.new(self,"is not the same as",pixel)
85     end
86     def to_s
87         "(#{@x},#{@y})"
88     end
89 end
90 class DocFile
91     def initialize(filename, page)
92         @filename = filename
93         @page = page
94     end
95     def load()
96         @swfname = @filename.gsub(/.pdf$/i,"")+".swf"
97         @pngname = @filename.gsub(/.pdf$/i,"")+".png"
98         begin
99             output = `pdf2swf --flatten -p #{@page} #{@filename} -o #{@swfname} 2>&1`
100             raise ConversionFailed.new(output,@swfname) unless File.exists?(@swfname)
101             output = `swfrender --legacy #{@swfname} -o #{@pngname} 2>&1`
102             raise ConversionFailed.new(output,@pngname) unless File.exists?(@pngname)
103             @img = Magick::Image.read(@pngname).first
104         ensure
105             `rm -f #{@swfname}`
106             `rm -f #{@pngname}`
107         end
108     end
109     def area_at(x1,y1,x2,y2)
110         self.load()
111         data = @img.export_pixels(x1, y1, x2-x1, y2-y1, "RGB")
112         return Area.new(x1,y1,x2,y2,data)
113     end
114     def width()
115         self.load()
116         return @img.columns
117     end
118     def height()
119         self.load()
120         return @img.rows
121     end
122     def pixel_at(x,y)
123         self.load()
124         data = @img.export_pixels(x, y, 1, 1, "RGB")
125         return Pixel.new(x,y,data)
126     end
127 end
128
129 module Spec
130     module Example
131         module ExampleMethods
132            def area_at(x1,y1,x2,y2)
133                @file.area_at(x1,y1,x2,y2)
134            end
135            def width
136                @file.width
137            end
138            def height
139                @file.height
140            end
141            def pixel_at(x,y)
142                @file.pixel_at(x,y)
143            end
144            def initialize(proxy,&impl)
145
146                # overriding the initialize() function saves us from having to
147                # set up the document in the test. The bad news, however, is that
148                # we have to be very careful not to raise exceptions here-
149                # rspec would miss those.
150
151                @_proxy = proxy
152                @_implementation = impl
153                @_backtrace = caller
154                @file = DocFile.new(proxy.description, (proxy.options[:page] or 1))
155            end
156         end
157         module ExampleGroupMethods
158             alias :convert_file :example
159             def page(nr)
160                 nr
161             end
162         end
163     end
164 end
165
166