15c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)#!/usr/bin/perl -w 25c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 35c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Copyright 2009 Apple Inc. All rights reserved. 45c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 55c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Redistribution and use in source and binary forms, with or without 65c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# modification, are permitted provided that the following conditions 75c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# are met: 85c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 1. Redistributions of source code must retain the above copyright 95c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# notice, this list of conditions and the following disclaimer. 105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 2. Redistributions in binary form must reproduce the above copyright 115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# notice, this list of conditions and the following disclaimer in the 125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# documentation and/or other materials provided with the distribution. 135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY 155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR 185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 24fff8884795cb540f87cf6e6d67b629519b00eb8bBen Murdoch# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 255c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Helper script to add includes to source files. 275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)use strict; 295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)my $headerPattern = '[\"<][A-Za-z][A-Za-z0-9_/]+(\.h)?[\">]'; # " Make Xcode formatter happy. 315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)my $headerToAdd = shift @ARGV or die; 335c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)$headerToAdd =~ /^([A-Za-z][A-Za-z0-9]+)\.h$/ or die "Header to add must be a .h file: $headerToAdd.\n"; 345c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 355c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)sub includesParagraph; 365c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 375c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)FILE: for my $filename (@ARGV) { 385c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unless ($filename =~ /(\w+)\.cpp$/) { print STDERR "Command line args must be .cpp files: $filename.\n"; next FILE; } 395c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 405c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $base = $1; 415c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 425c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $sawConfig = 0; 435c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $sawSelfInclude = 0; 445c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 455c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $pastIncludes = 0; 465c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my %includes; 475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $beforeIncludes = ""; 495c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $afterIncludes = ""; 505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 515c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $currentCondition = ""; 525c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 535c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $entireFileCondition = ""; 545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 555c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unless (open INPUT, "<", $filename) { print STDERR "File does not exist: $filename\n"; next FILE; } 565c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) while (my $line = <INPUT>) { 575c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if ($line =~ /^\s*#(include|import)\s*($headerPattern)\s*\n/) { 585c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $include = $2; 595c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if ($pastIncludes) { print STDERR "Saw more includes after include section in $filename, line $.\n"; next FILE; } 605c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if ($include eq "\"config.h\"") { 615c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $sawConfig = 1; 625c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } else { 635c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unless ($sawConfig) { print STDERR "First include must be config.h in $filename, line $.\n"; next FILE; } 645c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if ($include eq "\"$base.h\"") { 655c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $sawSelfInclude = 1; 665c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } else { 675c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unless ($sawSelfInclude) { print STDERR "Second include must be $base.h in $filename, line $.\n"; next FILE; } 685c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $includes{$currentCondition}{$include} = 1; 695c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 705c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 715c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } else { 725c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if ($sawConfig && !$pastIncludes) { 735c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if ($line =~ /^\s*#\s*if\s+(.+?)\s*$/) { 745c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $condition = $1; 755c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (!$sawSelfInclude) { 765c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $entireFileCondition = $1; 775c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) next; 785c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 795c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unless ($currentCondition eq "") { print STDERR "Nested #if in include section in $filename, line $.\n"; next FILE; } 805c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $currentCondition = $condition; 815c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) next; 825c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 835c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if ($line =~ /^\s*#\s*endif\s*$/) { 845c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unless ($currentCondition ne "") { print STDERR "Extra #endif in include section in $filename, line $.\n"; next FILE; } 855c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $currentCondition = ""; 865c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) next; 875c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 885c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 895c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (!$sawConfig) { 905c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $beforeIncludes .= $line; 915c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } else { 925c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $pastIncludes = 1 if $line !~ /^\s*$/; 935c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if ($pastIncludes) { 945c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unless ($currentCondition eq "") { print STDERR "Unterminated #if in include section in $filename, line $.\n"; next FILE; } 955c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $afterIncludes .= $line; 965c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 975c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 985c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 995c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1005c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) close INPUT or die; 1015c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1025c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $includes{""}{"\"$headerToAdd\""} = 1; 1035c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1045c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $beforeIncludes =~ s/\n+$//; 1055c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $afterIncludes =~ s/^\n+//; 1065c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1075c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $contents = $beforeIncludes; 1085c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $contents .= "\n\n#include \"config.h\"\n"; 1095c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $contents .= "\n#if $entireFileCondition\n" if $entireFileCondition ne ""; 1105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $contents .= "#include \"$base.h\"\n\n"; 1115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) for my $condition (sort keys %includes) { 1125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $contents .= "#if $condition\n" unless $condition eq ""; 1135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $contents .= includesParagraph($includes{$condition}); 1145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $contents .= "#endif\n" unless $condition eq ""; 1155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $contents .= "\n"; 1165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $contents .= $afterIncludes; 1185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unless (open OUTPUT, ">", $filename) { print STDERR "Could not open file for writing: $filename\n"; next FILE; }; 1205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) print OUTPUT $contents; 1215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) close OUTPUT or die; 1225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 1235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1245c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)sub includesParagraph() 1255c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 1265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my ($includes) = @_; 1275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) my $paragraph = ""; 1295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) for my $include (sort keys %{$includes}) { 1315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) $paragraph .= "#include $include\n"; 1325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1335c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1345c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return $paragraph; 1355c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 136