//===- SourceLocation.h - Compact identifier for Source Files ---*- C++ -*-===//

//

// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.

// See https://llvm.org/LICENSE.txt for license information.

// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

//

//===----------------------------------------------------------------------===//

//

/// \file

/// Defines the clang::SourceLocation class and associated facilities.

//

//===----------------------------------------------------------------------===//

#ifndef LLVM_CLANG_BASIC_SOURCELOCATION_H

#define LLVM_CLANG_BASIC_SOURCELOCATION_H

#include "clang/Basic/LLVM.h"

#include "llvm/ADT/StringRef.h"

#include <cassert>

#include <cstdint>

#include <string>

#include <utility>

namespace llvm {

class FoldingSetNodeID;

template <typename T> struct FoldingSetTrait;

} // namespace llvm

namespace clang {

class SourceManager;

/// An opaque identifier used by SourceManager which refers to a

/// source file (MemoryBuffer) along with its \#include path and \#line data.

///

class FileID {

  /// A mostly-opaque identifier, where 0 is "invalid", >0 is

  /// this module, and <-1 is something loaded from another module.

  int ID = 0;

public:

  bool isValid() const { return ID != 0; }

  bool isInvalid() const { return ID == 0; }

  bool operator==(const FileID &RHS) const { return ID == RHS.ID; }

  bool operator<(const FileID &RHS) const { return ID < RHS.ID; }

  bool operator<=(const FileID &RHS) const { return ID <= RHS.ID; }

  bool operator!=(const FileID &RHS) const { return !(*this == RHS); }

  bool operator>(const FileID &RHS) const { return RHS < *this; }

  bool operator>=(const FileID &RHS) const { return RHS <= *this; }

  static FileID getSentinel() { return get(-1); }

  unsigned getHashValue() const { return static_cast<unsigned>(ID); }

private:

  friend class ASTWriter;

  friend class ASTReader;

  friend class SourceManager;

  static FileID get(int V) {

    FileID F;

    F.ID = V;

    return F;

  }

  int getOpaqueValue() const { return ID; }

};

/// Encodes a location in the source. The SourceManager can decode this

/// to get at the full include stack, line and column information.

///

/// Technically, a source location is simply an offset into the manager's view

/// of the input source, which is all input buffers (including macro

/// expansions) concatenated in an effectively arbitrary order. The manager

/// actually maintains two blocks of input buffers. One, starting at offset

/// 0 and growing upwards, contains all buffers from this module. The other,

/// starting at the highest possible offset and growing downwards, contains

/// buffers of loaded modules.

///

/// In addition, one bit of SourceLocation is used for quick access to the

/// information whether the location is in a file or a macro expansion.

///

/// It is important that this type remains small. It is currently 32 bits wide.

class SourceLocation {

  friend class ASTReader;

  friend class ASTWriter;

  friend class SourceManager;

  friend struct llvm::FoldingSetTrait<SourceLocation>;

public:

  using UIntTy = uint32_t;

  using IntTy = int32_t;

private:

  UIntTy ID = 0;

  enum : UIntTy { MacroIDBit = 1ULL << (8 * sizeof(UIntTy) - 1) };

public:

  bool isFileID() const  { return (ID & MacroIDBit) == 0; }

  bool isMacroID() const { return (ID & MacroIDBit) != 0; }

  /// Return true if this is a valid SourceLocation object.

  ///

  /// Invalid SourceLocations are often used when events have no corresponding

  /// location in the source (e.g. a diagnostic is required for a command line

  /// option).

  bool isValid() const { return ID != 0; }

  bool isInvalid() const { return ID == 0; }

private:

  /// Return the offset into the manager's global input view.

  UIntTy getOffset() const { return ID & ~MacroIDBit; }

  static SourceLocation getFileLoc(UIntTy ID) {

    assert((ID & MacroIDBit) == 0 && "Ran out of source locations!");

    SourceLocation L;

    L.ID = ID;

    return L;

  }

  static SourceLocation getMacroLoc(UIntTy ID) {

    assert((ID & MacroIDBit) == 0 && "Ran out of source locations!");

    SourceLocation L;

    L.ID = MacroIDBit | ID;

    return L;

  }

public:

  /// Return a source location with the specified offset from this

  /// SourceLocation.

  SourceLocation getLocWithOffset(IntTy Offset) const {

    assert(((getOffset()+Offset) & MacroIDBit) == 0 && "offset overflow");

    SourceLocation L;

    L.ID = ID+Offset;

    return L;

  }

  /// When a SourceLocation itself cannot be used, this returns

  /// an (opaque) 32-bit integer encoding for it.

  ///

  /// This should only be passed to SourceLocation::getFromRawEncoding, it

  /// should not be inspected directly.

  UIntTy getRawEncoding() const { return ID; }

  /// Turn a raw encoding of a SourceLocation object into

  /// a real SourceLocation.

  ///

  /// \see getRawEncoding.

  static SourceLocation getFromRawEncoding(UIntTy Encoding) {

    SourceLocation X;

    X.ID = Encoding;

    return X;

  }

  /// When a SourceLocation itself cannot be used, this returns

  /// an (opaque) pointer encoding for it.

  ///

  /// This should only be passed to SourceLocation::getFromPtrEncoding, it

  /// should not be inspected directly.

  void* getPtrEncoding() const {

    // Double cast to avoid a warning "cast to pointer from integer of different

    // size".

    return (void*)(uintptr_t)getRawEncoding();

  }

  /// Turn a pointer encoding of a SourceLocation object back

  /// into a real SourceLocation.

  static SourceLocation getFromPtrEncoding(const void *Encoding) {

    return getFromRawEncoding((SourceLocation::UIntTy)(uintptr_t)Encoding);

  }

  static bool isPairOfFileLocations(SourceLocation Start, SourceLocation End) {

    return Start.isValid() && Start.isFileID() && End.isValid() &&

           End.isFileID();

  }

  unsigned getHashValue() const;

  void print(raw_ostream &OS, const SourceManager &SM) const;

  std::string printToString(const SourceManager &SM) const;

  void dump(const SourceManager &SM) const;

};

inline bool operator==(const SourceLocation &LHS, const SourceLocation &RHS) {

  return LHS.getRawEncoding() == RHS.getRawEncoding();

}

inline bool operator!=(const SourceLocation &LHS, const SourceLocation &RHS) {

  return !(LHS == RHS);

}

// Ordering is meaningful only if LHS and RHS have the same FileID!

// Otherwise use SourceManager::isBeforeInTranslationUnit().

inline bool operator<(const SourceLocation &LHS, const SourceLocation &RHS) {

  return LHS.getRawEncoding() < RHS.getRawEncoding();

}

inline bool operator>(const SourceLocation &LHS, const SourceLocation &RHS) {

  return LHS.getRawEncoding() > RHS.getRawEncoding();

}

inline bool operator<=(const SourceLocation &LHS, const SourceLocation &RHS) {

  return LHS.getRawEncoding() <= RHS.getRawEncoding();

}

inline bool operator>=(const SourceLocation &LHS, const SourceLocation &RHS) {

  return LHS.getRawEncoding() >= RHS.getRawEncoding();

}

/// A trivial tuple used to represent a source range.

class SourceRange {

  SourceLocation B;

  SourceLocation E;

public:

  SourceRange() = default;

  SourceRange(SourceLocation loc) : B(loc), E(loc) {}

  SourceRange(SourceLocation begin, SourceLocation end) : B(begin), E(end) {}

  SourceLocation getBegin() const { return B; }

  SourceLocation getEnd() const { return E; }

  void setBegin(SourceLocation b) { B = b; }

  void setEnd(SourceLocation e) { E = e; }

  bool isValid() const { return B.isValid() && 
E.isValid()61.0M
; }

  bool isInvalid() const { return !isValid(); }

  bool operator==(const SourceRange &X) const {

    return B == X.B && 
E == X.E13.2M
;

  }

  bool operator!=(const SourceRange &X) const {

    return B != X.B || E != X.E;

  }

  // Returns true iff other is wholly contained within this range.

  bool fullyContains(const SourceRange &other) const {

    return B <= other.B && E >= other.E;

  }

  void print(raw_ostream &OS, const SourceManager &SM) const;

  std::string printToString(const SourceManager &SM) const;

  void dump(const SourceManager &SM) const;

};

/// Represents a character-granular source range.

///

/// The underlying SourceRange can either specify the starting/ending character

/// of the range, or it can specify the start of the range and the start of the

/// last token of the range (a "token range").  In the token range case, the

/// size of the last token must be measured to determine the actual end of the

/// range.

class CharSourceRange {

  SourceRange Range;

  bool IsTokenRange = false;

public:

  CharSourceRange() = default;

  CharSourceRange(SourceRange R, bool ITR) : Range(R), IsTokenRange(ITR) {}

  static CharSourceRange getTokenRange(SourceRange R) {

    return CharSourceRange(R, true);

  }

  static CharSourceRange getCharRange(SourceRange R) {

    return CharSourceRange(R, false);

  }

  static CharSourceRange getTokenRange(SourceLocation B, SourceLocation E) {

    return getTokenRange(SourceRange(B, E));

  }

  static CharSourceRange getCharRange(SourceLocation B, SourceLocation E) {

    return getCharRange(SourceRange(B, E));

  }

  /// Return true if the end of this range specifies the start of

  /// the last token.  Return false if the end of this range specifies the last

  /// character in the range.

  bool isTokenRange() const { return IsTokenRange; }

  bool isCharRange() const { return !IsTokenRange; }

  SourceLocation getBegin() const { return Range.getBegin(); }

  SourceLocation getEnd() const { return Range.getEnd(); }

  SourceRange getAsRange() const { return Range; }

  void setBegin(SourceLocation b) { Range.setBegin(b); }

  void setEnd(SourceLocation e) { Range.setEnd(e); }

  void setTokenRange(bool TR) { IsTokenRange = TR; }

  bool isValid() const { return Range.isValid(); }

  bool isInvalid() const { return !isValid(); }

};

/// Represents an unpacked "presumed" location which can be presented

/// to the user.

///

/// A 'presumed' location can be modified by \#line and GNU line marker

/// directives and is always the expansion point of a normal location.

///

/// You can get a PresumedLoc from a SourceLocation with SourceManager.

class PresumedLoc {

  const char *Filename = nullptr;

  FileID ID;

  unsigned Line, Col;

  SourceLocation IncludeLoc;

public:

  PresumedLoc() = default;

  PresumedLoc(const char *FN, FileID FID, unsigned Ln, unsigned Co,

              SourceLocation IL)

      : Filename(FN), ID(FID), Line(Ln), Col(Co), IncludeLoc(IL) {}

  /// Return true if this object is invalid or uninitialized.

  ///

  /// This occurs when created with invalid source locations or when walking

  /// off the top of a \#include stack.

  bool isInvalid() const { return Filename == nullptr; }

  bool isValid() const { return Filename != nullptr; }

  /// Return the presumed filename of this location.

  ///

  /// This can be affected by \#line etc.

  const char *getFilename() const {

    assert(isValid());

    return Filename;

  }

  FileID getFileID() const {

    assert(isValid());

    return ID;

  }

  /// Return the presumed line number of this location.

  ///

  /// This can be affected by \#line etc.

  unsigned getLine() const {

    assert(isValid());

    return Line;

  }

  /// Return the presumed column number of this location.

  ///

  /// This cannot be affected by \#line, but is packaged here for convenience.

  unsigned getColumn() const {

    assert(isValid());

    return Col;

  }

  /// Return the presumed include location of this location.

  ///

  /// This can be affected by GNU linemarker directives.

  SourceLocation getIncludeLoc() const {

    assert(isValid());

    return IncludeLoc;

  }

};

class FileEntry;

/// A SourceLocation and its associated SourceManager.

///

/// This is useful for argument passing to functions that expect both objects.

///

/// This class does not guarantee the presence of either the SourceManager or

/// a valid SourceLocation. Clients should use `isValid()` and `hasManager()`

/// before calling the member functions.

class FullSourceLoc : public SourceLocation {

  const SourceManager *SrcMgr = nullptr;

public:

  /// Creates a FullSourceLoc where isValid() returns \c false.

  FullSourceLoc() = default;

  explicit FullSourceLoc(SourceLocation Loc, const SourceManager &SM)

      : SourceLocation(Loc), SrcMgr(&SM) {}

  /// Checks whether the SourceManager is present.

  bool hasManager() const { return SrcMgr != nullptr; }

  /// \pre hasManager()

  const SourceManager &getManager() const {

    assert(SrcMgr && "SourceManager is NULL.");

    return *SrcMgr;

  }

  FileID getFileID() const;

  FullSourceLoc getExpansionLoc() const;

  FullSourceLoc getSpellingLoc() const;

  FullSourceLoc getFileLoc() const;

  PresumedLoc getPresumedLoc(bool UseLineDirectives = true) const;

  bool isMacroArgExpansion(FullSourceLoc *StartLoc = nullptr) const;

  FullSourceLoc getImmediateMacroCallerLoc() const;

  std::pair<FullSourceLoc, StringRef> getModuleImportLoc() const;

  unsigned getFileOffset() const;

  unsigned getExpansionLineNumber(bool *Invalid = nullptr) const;

  unsigned getExpansionColumnNumber(bool *Invalid = nullptr) const;

  unsigned getSpellingLineNumber(bool *Invalid = nullptr) const;

  unsigned getSpellingColumnNumber(bool *Invalid = nullptr) const;

  const char *getCharacterData(bool *Invalid = nullptr) const;

  unsigned getLineNumber(bool *Invalid = nullptr) const;

  unsigned getColumnNumber(bool *Invalid = nullptr) const;

  const FileEntry *getFileEntry() const;

  /// Return a StringRef to the source buffer data for the

  /// specified FileID.

  StringRef getBufferData(bool *Invalid = nullptr) const;

  /// Decompose the specified location into a raw FileID + Offset pair.

  ///

  /// The first element is the FileID, the second is the offset from the

  /// start of the buffer of the location.

  std::pair<FileID, unsigned> getDecomposedLoc() const;

  bool isInSystemHeader() const;

  /// Determines the order of 2 source locations in the translation unit.

  ///

  /// \returns true if this source location comes before 'Loc', false otherwise.

  bool isBeforeInTranslationUnitThan(SourceLocation Loc) const;

  /// Determines the order of 2 source locations in the translation unit.

  ///

  /// \returns true if this source location comes before 'Loc', false otherwise.

  bool isBeforeInTranslationUnitThan(FullSourceLoc Loc) const {

    assert(Loc.isValid());

    assert(SrcMgr == Loc.SrcMgr && "Loc comes from another SourceManager!");

    return isBeforeInTranslationUnitThan((SourceLocation)Loc);

  }

  /// Comparison function class, useful for sorting FullSourceLocs.

  struct BeforeThanCompare {

    bool operator()(const FullSourceLoc& lhs, const FullSourceLoc& rhs) const {

      return lhs.isBeforeInTranslationUnitThan(rhs);

    }

  };

  /// Prints information about this FullSourceLoc to stderr.

  ///

  /// This is useful for debugging.

  void dump() const;

  friend bool

  operator==(const FullSourceLoc &LHS, const FullSourceLoc &RHS) {

    return LHS.getRawEncoding() == RHS.getRawEncoding() &&

          
LHS.SrcMgr == RHS.SrcMgr2.84k
;

  }

  friend bool

  operator!=(const FullSourceLoc &LHS, const FullSourceLoc &RHS) {

    return !(LHS == RHS);

  }

};

} // namespace clang

namespace llvm {

  /// Define DenseMapInfo so that FileID's can be used as keys in DenseMap and

  /// DenseSets.

  template <>

  struct DenseMapInfo<clang::FileID, void> {

    static clang::FileID getEmptyKey() {

      return {};

    }

    static clang::FileID getTombstoneKey() {

      return clang::FileID::getSentinel();

    }

    static unsigned getHashValue(clang::FileID S) {

      return S.getHashValue();

    }

    static bool isEqual(clang::FileID LHS, clang::FileID RHS) {

      return LHS == RHS;

    }

  };

  /// Define DenseMapInfo so that SourceLocation's can be used as keys in

  /// DenseMap and DenseSet. This trait class is eqivalent to

  /// DenseMapInfo<unsigned> which uses SourceLocation::ID is used as a key.

  template <> struct DenseMapInfo<clang::SourceLocation, void> {

    static clang::SourceLocation getEmptyKey() {

      constexpr clang::SourceLocation::UIntTy Zero = 0;

      return clang::SourceLocation::getFromRawEncoding(~Zero);

    }

    static clang::SourceLocation getTombstoneKey() {

      constexpr clang::SourceLocation::UIntTy Zero = 0;

      return clang::SourceLocation::getFromRawEncoding(~Zero - 1);

    }

    static unsigned getHashValue(clang::SourceLocation Loc) {

      return Loc.getHashValue();

    }

    static bool isEqual(clang::SourceLocation LHS, clang::SourceLocation RHS) {

      return LHS == RHS;

    }

  };

  // Allow calling FoldingSetNodeID::Add with SourceLocation object as parameter

  template <> struct FoldingSetTrait<clang::SourceLocation> {

    static void Profile(const clang::SourceLocation &X, FoldingSetNodeID &ID);

  };

} // namespace llvm

#endif // LLVM_CLANG_BASIC_SOURCELOCATION_H