• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===--- ArgumentsAdjusters.cpp - Command line arguments adjuster ---------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file contains definitions of classes which implement ArgumentsAdjuster
11 // interface.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #include "clang/Tooling/ArgumentsAdjusters.h"
16 
17 namespace clang {
18 namespace tooling {
19 
20 /// Add -fsyntax-only option to the commnand line arguments.
getClangSyntaxOnlyAdjuster()21 ArgumentsAdjuster getClangSyntaxOnlyAdjuster() {
22   return [](const CommandLineArguments &Args, StringRef /*unused*/) {
23     CommandLineArguments AdjustedArgs;
24     for (size_t i = 0, e = Args.size(); i != e; ++i) {
25       StringRef Arg = Args[i];
26       // FIXME: Remove options that generate output.
27       if (!Arg.startswith("-fcolor-diagnostics") &&
28           !Arg.startswith("-fdiagnostics-color"))
29         AdjustedArgs.push_back(Args[i]);
30     }
31     AdjustedArgs.push_back("-fsyntax-only");
32     return AdjustedArgs;
33   };
34 }
35 
getClangStripOutputAdjuster()36 ArgumentsAdjuster getClangStripOutputAdjuster() {
37   return [](const CommandLineArguments &Args, StringRef /*unused*/) {
38     CommandLineArguments AdjustedArgs;
39     for (size_t i = 0, e = Args.size(); i < e; ++i) {
40       StringRef Arg = Args[i];
41       if (!Arg.startswith("-o"))
42         AdjustedArgs.push_back(Args[i]);
43 
44       if (Arg == "-o") {
45         // Output is specified as -o foo. Skip the next argument also.
46         ++i;
47       }
48       // Else, the output is specified as -ofoo. Just do nothing.
49     }
50     return AdjustedArgs;
51   };
52 }
53 
getInsertArgumentAdjuster(const CommandLineArguments & Extra,ArgumentInsertPosition Pos)54 ArgumentsAdjuster getInsertArgumentAdjuster(const CommandLineArguments &Extra,
55                                             ArgumentInsertPosition Pos) {
56   return [Extra, Pos](const CommandLineArguments &Args, StringRef /*unused*/) {
57     CommandLineArguments Return(Args);
58 
59     CommandLineArguments::iterator I;
60     if (Pos == ArgumentInsertPosition::END) {
61       I = Return.end();
62     } else {
63       I = Return.begin();
64       ++I; // To leave the program name in place
65     }
66 
67     Return.insert(I, Extra.begin(), Extra.end());
68     return Return;
69   };
70 }
71 
getInsertArgumentAdjuster(const char * Extra,ArgumentInsertPosition Pos)72 ArgumentsAdjuster getInsertArgumentAdjuster(const char *Extra,
73                                             ArgumentInsertPosition Pos) {
74   return getInsertArgumentAdjuster(CommandLineArguments(1, Extra), Pos);
75 }
76 
combineAdjusters(ArgumentsAdjuster First,ArgumentsAdjuster Second)77 ArgumentsAdjuster combineAdjusters(ArgumentsAdjuster First,
78                                    ArgumentsAdjuster Second) {
79   return [First, Second](const CommandLineArguments &Args, StringRef File) {
80     return Second(First(Args, File), File);
81   };
82 }
83 
84 } // end namespace tooling
85 } // end namespace clang
86 
87