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