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 CLANG_LIB_DRIVER_INPUTINFO_H_
11#define 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
18namespace clang {
19namespace driver {
20
21/// InputInfo - Wrapper for information about an input source.
22class 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
44public:
45  InputInfo() {}
46  InputInfo(types::ID _Type, const char *_BaseInput)
47    : Kind(Nothing), Type(_Type), BaseInput(_BaseInput) {
48  }
49  InputInfo(const char *_Filename, types::ID _Type, const char *_BaseInput)
50    : Kind(Filename), Type(_Type), BaseInput(_BaseInput) {
51    Data.Filename = _Filename;
52  }
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
59  bool isNothing() const { return Kind == Nothing; }
60  bool isFilename() const { return Kind == Filename; }
61  bool isInputArg() const { return Kind == InputArg; }
62  types::ID getType() const { return Type; }
63  const char *getBaseInput() const { return BaseInput; }
64
65  const char *getFilename() const {
66    assert(isFilename() && "Invalid accessor.");
67    return Data.Filename;
68  }
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.
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