• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===--- InputInfo.h - Input Source & Type Information ----------*- C++ -*-===//
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 #ifndef LLVM_CLANG_LIB_DRIVER_INPUTINFO_H
11 #define LLVM_CLANG_LIB_DRIVER_INPUTINFO_H
12 
13 #include "clang/Driver/Action.h"
14 #include "clang/Driver/Types.h"
15 #include "llvm/Option/Arg.h"
16 #include <cassert>
17 #include <string>
18 
19 namespace clang {
20 namespace driver {
21 
22 /// InputInfo - Wrapper for information about an input source.
23 class InputInfo {
24   // FIXME: The distinction between filenames and inputarg here is
25   // gross; we should probably drop the idea of a "linker
26   // input". Doing so means tweaking pipelining to still create link
27   // steps when it sees linker inputs (but not treat them as
28   // arguments), and making sure that arguments get rendered
29   // correctly.
30   enum Class {
31     Nothing,
32     Filename,
33     InputArg,
34     Pipe
35   };
36 
37   union {
38     const char *Filename;
39     const llvm::opt::Arg *InputArg;
40   } Data;
41   Class Kind;
42   const Action* Act;
43   types::ID Type;
44   const char *BaseInput;
45 
GetActionType(const Action * A)46   static types::ID GetActionType(const Action *A) {
47     return A != nullptr ? A->getType() : types::TY_Nothing;
48   }
49 
50 public:
InputInfo()51   InputInfo() : InputInfo(nullptr, nullptr) {}
InputInfo(const Action * A,const char * _BaseInput)52   InputInfo(const Action *A, const char *_BaseInput)
53       : Kind(Nothing), Act(A), Type(GetActionType(A)), BaseInput(_BaseInput) {}
54 
InputInfo(types::ID _Type,const char * _Filename,const char * _BaseInput)55   InputInfo(types::ID _Type, const char *_Filename, const char *_BaseInput)
56       : Kind(Filename), Act(nullptr), Type(_Type), BaseInput(_BaseInput) {
57     Data.Filename = _Filename;
58   }
InputInfo(const Action * A,const char * _Filename,const char * _BaseInput)59   InputInfo(const Action *A, const char *_Filename, const char *_BaseInput)
60       : Kind(Filename), Act(A), Type(GetActionType(A)), BaseInput(_BaseInput) {
61     Data.Filename = _Filename;
62   }
63 
InputInfo(types::ID _Type,const llvm::opt::Arg * _InputArg,const char * _BaseInput)64   InputInfo(types::ID _Type, const llvm::opt::Arg *_InputArg,
65             const char *_BaseInput)
66       : Kind(InputArg), Act(nullptr), Type(_Type), BaseInput(_BaseInput) {
67     Data.InputArg = _InputArg;
68   }
InputInfo(const Action * A,const llvm::opt::Arg * _InputArg,const char * _BaseInput)69   InputInfo(const Action *A, const llvm::opt::Arg *_InputArg,
70             const char *_BaseInput)
71       : Kind(InputArg), Act(A), Type(GetActionType(A)), BaseInput(_BaseInput) {
72     Data.InputArg = _InputArg;
73   }
74 
isNothing()75   bool isNothing() const { return Kind == Nothing; }
isFilename()76   bool isFilename() const { return Kind == Filename; }
isInputArg()77   bool isInputArg() const { return Kind == InputArg; }
getType()78   types::ID getType() const { return Type; }
getBaseInput()79   const char *getBaseInput() const { return BaseInput; }
80   /// The action for which this InputInfo was created.  May be null.
getAction()81   const Action *getAction() const { return Act; }
setAction(const Action * A)82   void setAction(const Action *A) { Act = A; }
83 
getFilename()84   const char *getFilename() const {
85     assert(isFilename() && "Invalid accessor.");
86     return Data.Filename;
87   }
getInputArg()88   const llvm::opt::Arg &getInputArg() const {
89     assert(isInputArg() && "Invalid accessor.");
90     return *Data.InputArg;
91   }
92 
93   /// getAsString - Return a string name for this input, for
94   /// debugging.
getAsString()95   std::string getAsString() const {
96     if (isFilename())
97       return std::string("\"") + getFilename() + '"';
98     else if (isInputArg())
99       return "(input arg)";
100     else
101       return "(nothing)";
102   }
103 };
104 
105 } // end namespace driver
106 } // end namespace clang
107 
108 #endif
109