1 eval "exec perl -S $0 $*"
2 if $running_under_some_random_shell;
4 # reads CPP output and turns #line things into appropriate Haskell
7 # considered to be GHC-project specific
10 # OPTIONALLY processes GENERATE_SPECS pragmas
11 # when give flag -genSPECS
15 # {-# GENERATE_SPECS a b #-}
21 # {-# SPECIALIZE fn :: type[ a/a,u1/b] #-}
22 # {-# SPECIALIZE fn :: type[ a/a,u2/b] #-}
23 # {-# SPECIALIZE fn :: type[u1/a, b/b] #-}
24 # {-# SPECIALIZE fn :: type[u1/a,u1/b] #-}
25 # {-# SPECIALIZE fn :: type[u1/a,u2/b] #-}
26 # {-# SPECIALIZE fn :: type[u2/a, b/b] #-}
27 # {-# SPECIALIZE fn :: type[u2/a,u1/b] #-}
28 # {-# SPECIALIZE fn :: type[u2/a,u2/b] #-}
30 # where the u's are extracted from a predetermined
31 # set of unboxed types $SpecingString
33 # The types to substitute can be specified explicitly in { }s following
38 # {-# GENERATE_SPECS a{ty1,ty2...} b{+,ty1,ty2...} c{~,ty1,ty2,...} d{~,+,ty1,ty2,...} #-}
42 # ~ indicates that no specialisations are to be left polymorhphic in this type variable
43 # (this is required for overloaded tyvars which must have ground specialisations)
44 # + indicates that the predetermined types are to be added to the list
46 # Note: There must be no white space between { }s
47 # Use ( )s around type names when separation is required
52 $DoGenSpecsUnboxed = 0;
55 while ($#ARGV >= 0 && $ARGV[0] eq '-v' || $ARGV[0] =~ /^-genSPECS/) {
56 if ($ARGV[0] eq '-v') {
58 } elsif ($ARGV[0] eq '-genSPECS') {
60 } elsif ($ARGV[0] eq '-genSPECSunboxed') {
62 $DoGenSpecsUnboxed = 1;
63 $SpecingString = "Char#,Int#,Double#";
64 @SpecingTypes = split(/,/, $SpecingString);
66 die "hscpp: unrecognised argument: $$ARGV[0]\n";
70 #ToDo: print a version number ?
72 $OrigCpp = '$(RAWCPP)';
74 if ( $OrigCpp =~ /(\S+)\s+(.*)/ ) {
77 if ( -x $cmd ) { # cool
79 } else { # oops; try to guess
80 $GccV = `gcc -v 2>&1`;
81 if ( $GccV =~ /Reading specs from (.*)\/specs/ ) {
82 $Cpp = "$1/cpp $rest";
84 die "hscpp: don't know how to run cpp: $OrigCpp\n";
91 print STDERR "hscpp:CPP invoked: $Cpp @ARGV\n" if $Verbose;
93 open(INPIPE, "$Cpp @ARGV |") || die "Can't open C pre-processor pipe\n";
97 # line directives come in flavo[u]rs:
98 # s/^#\s*line\s+\d+$/\{\-# LINE \-\}/; IGNORE THIS ONE FOR NOW
99 s/^#\s*line\s+(\d+)\s+(\".+\")$/\{\-# LINE \1 \2 \-\}/;
100 s/^#\s*(\d+)\s+(\".*\").*/\{\-# LINE \1 \2 \-\}/;
102 # genSPEC processing:
103 if ( /^\s*\{-#\s*GENERATE_SPECS/ ) {
111 if ( /^\s*\{-#\s*GENERATE_SPECS\s+(data)\s+(\S.*)\s*::(.*)#-\}\s*$/ ) {
116 } elsif ( /^\s*\{-#\s*GENERATE_SPECS\s+(instance)\s+(\S.*)\s*::(.*)#-\}\s*$/ ) {
121 } elsif ( /^(\s*)\{-#\s*GENERATE_SPECS\s+(\S+)\s+(.*)\s*#-\}\s*$/ ) {
127 while ( $tysig =~ /^\s*$/ ) {
131 $funpat = $fun; # quote non alphanumeric characters in pattern
132 $funpat =~ s/(\W)/\\\1/g;
133 $tysig =~ /^\s*$funpat\s*::(.*)$/ || die "Error: GENERATE_SPECS not followed by type signature for $fun:\n$line$tysig\n";
135 $type =~ s/^(.*)=>//; # remove context from type
137 die "Error: invlaid GENERATE_SPECS pragma:\n $_";
140 @tyvars = split(/\s+/, $vars);
142 foreach $var (@tyvars) {
145 if ( $var =~ /^(\w+)\{(.*)\}$/ ) {
147 @specing_types = &split_types($2);
148 if ($specing_types[0] eq '~') {
149 shift(@specing_types);
150 @tospec = (); # remove specs polymorphic in this tyvar
153 if ($specing_types[0] eq '+') {
154 shift(@specing_types);
155 unshift(@specing_types, @SpecingTypes);
158 @specing_types = @SpecingTypes;
161 # If not $DoGenSpecsUnboxed we remove any unboxed types
162 # (i.e. types containing a #) from the specing_types
164 @specing_types = grep(!/#/, @specing_types) if ! $DoGenSpecsUnboxed;
166 foreach $uty (@specing_types) {
168 foreach $i (0..$#speced) {
169 $speced[$i] =~ s/\b$var\b/$uty/g ;
171 push(@tospec, @speced);
174 shift(@tospec) if $remove_poly; # remove fully polymorphic spec
177 $specty = shift(@tospec);
178 print ($data_or_inst ? "\{-# SPECIALIZE $data_inst_str $specty #-\}" : "$space\{-# SPECIALIZE $fun :: $specty");
179 while ($#tospec >= 0) {
180 $specty = shift(@tospec);
181 print ($data_or_inst ? "; \{-# SPECIALIZE $data_inst_str $specty #-\}" : ", $specty");
183 print ($data_or_inst ? "\n" : " #-}\n");
185 print "\{- NO_SPECS_GENERATED ", $data_or_inst ? $specty : $fun, " -\}\n";
186 print STDERR "Warning: No specs for GENERATE_SPECS pre-processing pragma:\n $_";
188 print $tysig if ! $data_or_inst;
190 print STDERR "Warning: GENERATE_SPECS pre-processing pragma ignored:\n $_";
198 close(INPIPE) || exit(1); # exit is so we reflect any errors.
203 # splits a list of types seperated by ,s but allowing ,s within ()s
207 local($type_str) = @_;
209 local(@chars) = split(//,$type_str);
217 while ($char = $chars[$cur]) {
218 if ($char eq ',' && $depth == 0) {
219 push(@types, join('', @chars[$start .. $cur-1])) if ($start < $cur);
221 } elsif ($char eq '(') {
223 } elsif ($char eq ')') {
229 push(@types, join('', @chars[$start .. $cur-1])) if ($start < $cur);
231 print STDERR "Error: GENERATE_SPECS pre-processing pragma has unbalanced ( )s\n$line\n";