Make generate_offset_extractor.rb architectures argument more robust
[WebKit-https.git] / Source / JavaScriptCore / offlineasm / generate_offset_extractor.rb
1 #!/usr/bin/env ruby
2
3 # Copyright (C) 2011 Apple Inc. All rights reserved.
4 #
5 # Redistribution and use in source and binary forms, with or without
6 # modification, are permitted provided that the following conditions
7 # are met:
8 # 1. Redistributions of source code must retain the above copyright
9 #    notice, this list of conditions and the following disclaimer.
10 # 2. Redistributions in binary form must reproduce the above copyright
11 #    notice, this list of conditions and the following disclaimer in the
12 #    documentation and/or other materials provided with the distribution.
13 #
14 # THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
15 # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
16 # THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
18 # BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
19 # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
20 # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
21 # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
22 # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
23 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
24 # THE POSSIBILITY OF SUCH DAMAGE.
25
26 $: << File.dirname(__FILE__)
27
28 require "config"
29 require "backends"
30 require "digest/sha1"
31 require "offsets"
32 require "parser"
33 require "self_hash"
34 require "settings"
35 require "transform"
36
37 IncludeFile.processIncludeOptions()
38
39 inputFlnm = ARGV.shift
40 outputFlnm = ARGV.shift
41
42 validBackends = canonicalizeBackendNames(ARGV.shift.split(/[,\s]+/))
43 $stderr.puts "Only dealing with backends: #{validBackends}"
44 includeOnlyBackends(validBackends)
45
46 def emitMagicNumber
47     OFFSET_MAGIC_NUMBERS.each {
48         | number |
49         $output.puts "unsigned(#{number}),"
50     }
51 end
52
53 inputHash = "// offlineasm input hash: #{parseHash(inputFlnm)} #{selfHash}"
54
55 if FileTest.exist? outputFlnm
56     File.open(outputFlnm, "r") {
57         | inp |
58         firstLine = inp.gets
59         if firstLine and firstLine.chomp == inputHash
60             $stderr.puts "offlineasm: Nothing changed."
61             exit 0
62         end
63     }
64 end
65
66 originalAST = parse(inputFlnm)
67
68 #
69 # Optimize the AST to make configuration extraction faster. This reduces the AST to a form
70 # that only contains the things that matter for our purposes: offsets, sizes, and if
71 # statements.
72 #
73
74 class Node
75     def offsetsPruneTo(sequence)
76         children.each {
77             | child |
78             child.offsetsPruneTo(sequence)
79         }
80     end
81     
82     def offsetsPrune
83         result = Sequence.new(codeOrigin, [])
84         offsetsPruneTo(result)
85         result
86     end
87 end
88
89 class IfThenElse
90     def offsetsPruneTo(sequence)
91         ifThenElse = IfThenElse.new(codeOrigin, predicate, thenCase.offsetsPrune)
92         ifThenElse.elseCase = elseCase.offsetsPrune
93         sequence.list << ifThenElse
94     end
95 end
96
97 class StructOffset
98     def offsetsPruneTo(sequence)
99         sequence.list << self
100     end
101 end
102
103 class Sizeof
104     def offsetsPruneTo(sequence)
105         sequence.list << self
106     end
107 end
108
109 class ConstExpr
110     def offsetsPruneTo(sequence)
111         sequence.list << self
112     end
113 end
114
115 prunedAST = originalAST.offsetsPrune
116
117 File.open(outputFlnm, "w") {
118     | outp |
119     $output = outp
120     outp.puts inputHash
121     length = 0
122
123     emitCodeInAllConfigurations(prunedAST) {
124         | settings, ast, backend, index |
125         constsList = ast.filter(ConstExpr).uniq.sort
126
127         constsList.each_with_index {
128             | const, index |
129             outp.puts "constexpr int64_t constValue#{index} = static_cast<int64_t>(#{const.value});"
130         }
131     }
132
133     emitCodeInAllConfigurations(prunedAST) {
134         | settings, ast, backend, index |
135         offsetsList = ast.filter(StructOffset).uniq.sort
136         sizesList = ast.filter(Sizeof).uniq.sort
137         constsList = ast.filter(ConstExpr).uniq.sort
138         length += OFFSET_HEADER_MAGIC_NUMBERS.size + (OFFSET_MAGIC_NUMBERS.size + 1) * (1 + offsetsList.size + sizesList.size + constsList.size)
139     }
140     outp.puts "static const int64_t extractorTable[#{length}] = {"
141     emitCodeInAllConfigurations(prunedAST) {
142         | settings, ast, backend, index |
143         OFFSET_HEADER_MAGIC_NUMBERS.each {
144             | number |
145             $output.puts "unsigned(#{number}),"
146         }
147
148         offsetsList = ast.filter(StructOffset).uniq.sort
149         sizesList = ast.filter(Sizeof).uniq.sort
150         constsList = ast.filter(ConstExpr).uniq.sort
151
152         emitMagicNumber
153         outp.puts "#{index},"
154         offsetsList.each {
155             | offset |
156             emitMagicNumber
157             outp.puts "OFFLINE_ASM_OFFSETOF(#{offset.struct}, #{offset.field}),"
158         }
159         sizesList.each {
160             | sizeof |
161             emitMagicNumber
162             outp.puts "sizeof(#{sizeof.struct}),"
163         }
164         constsList.each_index {
165             | index |
166             emitMagicNumber
167             outp.puts "constValue#{index},"
168         }
169     }
170     outp.puts "};"
171
172 }