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