Home | History | Annotate | Download | only in Driver
      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 
     46   static types::ID GetActionType(const Action *A) {
     47     return A != nullptr ? A->getType() : types::TY_Nothing;
     48   }
     49 
     50 public:
     51   InputInfo() : InputInfo(nullptr, nullptr) {}
     52   InputInfo(const Action *A, const char *_BaseInput)
     53       : Kind(Nothing), Act(A), Type(GetActionType(A)), BaseInput(_BaseInput) {}
     54 
     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   }
     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 
     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   }
     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 
     75   bool isNothing() const { return Kind == Nothing; }
     76   bool isFilename() const { return Kind == Filename; }
     77   bool isInputArg() const { return Kind == InputArg; }
     78   types::ID getType() const { return Type; }
     79   const char *getBaseInput() const { return BaseInput; }
     80   /// The action for which this InputInfo was created.  May be null.
     81   const Action *getAction() const { return Act; }
     82   void setAction(const Action *A) { Act = A; }
     83 
     84   const char *getFilename() const {
     85     assert(isFilename() && "Invalid accessor.");
     86     return Data.Filename;
     87   }
     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.
     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