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/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   types::ID Type;
42   const char *BaseInput;
43 
44 public:
InputInfo()45   InputInfo() {}
InputInfo(types::ID _Type,const char * _BaseInput)46   InputInfo(types::ID _Type, const char *_BaseInput)
47     : Kind(Nothing), Type(_Type), BaseInput(_BaseInput) {
48   }
InputInfo(const char * _Filename,types::ID _Type,const char * _BaseInput)49   InputInfo(const char *_Filename, types::ID _Type, const char *_BaseInput)
50     : Kind(Filename), Type(_Type), BaseInput(_BaseInput) {
51     Data.Filename = _Filename;
52   }
InputInfo(const llvm::opt::Arg * _InputArg,types::ID _Type,const char * _BaseInput)53   InputInfo(const llvm::opt::Arg *_InputArg, types::ID _Type,
54             const char *_BaseInput)
55       : Kind(InputArg), Type(_Type), BaseInput(_BaseInput) {
56     Data.InputArg = _InputArg;
57   }
58 
isNothing()59   bool isNothing() const { return Kind == Nothing; }
isFilename()60   bool isFilename() const { return Kind == Filename; }
isInputArg()61   bool isInputArg() const { return Kind == InputArg; }
getType()62   types::ID getType() const { return Type; }
getBaseInput()63   const char *getBaseInput() const { return BaseInput; }
64 
getFilename()65   const char *getFilename() const {
66     assert(isFilename() && "Invalid accessor.");
67     return Data.Filename;
68   }
getInputArg()69   const llvm::opt::Arg &getInputArg() const {
70     assert(isInputArg() && "Invalid accessor.");
71     return *Data.InputArg;
72   }
73 
74   /// getAsString - Return a string name for this input, for
75   /// debugging.
getAsString()76   std::string getAsString() const {
77     if (isFilename())
78       return std::string("\"") + getFilename() + '"';
79     else if (isInputArg())
80       return "(input arg)";
81     else
82       return "(nothing)";
83   }
84 };
85 
86 } // end namespace driver
87 } // end namespace clang
88 
89 #endif
90