contrib/llvm-project/clang/lib/Driver/InputInfo.h

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