mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
b31133e67d
Due to the duality between the expression and declaration forms of freestanding macros, we could end up assigning two different discriminators to what is effectively the same freestanding macro expansion. Across different source files, this could lead to inconsistent discriminators in different translation units. Unify the storage of the discriminator to avoid this issue. Fixes rdar://116259748
133 lines
4.3 KiB
C++
133 lines
4.3 KiB
C++
//===--- FreestandingMacroExpansion.h ------------------------*- C++ -*-===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2023 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef SWIFT_AST_FREESTANDING_MACRO_EXPANSION_H
|
|
#define SWIFT_AST_FREESTANDING_MACRO_EXPANSION_H
|
|
|
|
#include "swift/AST/ASTAllocated.h"
|
|
#include "swift/AST/ASTNode.h"
|
|
#include "swift/AST/ConcreteDeclRef.h"
|
|
#include "swift/AST/DeclNameLoc.h"
|
|
#include "swift/AST/Identifier.h"
|
|
#include "swift/Basic/SourceLoc.h"
|
|
|
|
namespace swift {
|
|
class MacroExpansionDecl;
|
|
class MacroExpansionExpr;
|
|
class Expr;
|
|
class Decl;
|
|
class ArgumentList;
|
|
|
|
/// Information about a macro expansion that is common between macro
|
|
/// expansion declarations and expressions.
|
|
///
|
|
/// Instances of these types will be shared among paired macro expansion
|
|
/// declaration/expression nodes.
|
|
struct MacroExpansionInfo : ASTAllocated<MacroExpansionInfo> {
|
|
SourceLoc SigilLoc;
|
|
DeclNameRef ModuleName;
|
|
DeclNameLoc ModuleNameLoc;
|
|
DeclNameRef MacroName;
|
|
DeclNameLoc MacroNameLoc;
|
|
SourceLoc LeftAngleLoc, RightAngleLoc;
|
|
llvm::ArrayRef<TypeRepr *> GenericArgs;
|
|
ArgumentList *ArgList;
|
|
|
|
/// The referenced macro.
|
|
ConcreteDeclRef macroRef;
|
|
|
|
enum : unsigned { InvalidDiscriminator = 0xFFFF };
|
|
|
|
unsigned Discriminator = InvalidDiscriminator;
|
|
|
|
MacroExpansionInfo(SourceLoc sigilLoc, DeclNameRef moduleName,
|
|
DeclNameLoc moduleNameLoc, DeclNameRef macroName,
|
|
DeclNameLoc macroNameLoc, SourceLoc leftAngleLoc,
|
|
SourceLoc rightAngleLoc, ArrayRef<TypeRepr *> genericArgs,
|
|
ArgumentList *argList)
|
|
: SigilLoc(sigilLoc), ModuleName(moduleName),
|
|
ModuleNameLoc(moduleNameLoc), MacroName(macroName),
|
|
MacroNameLoc(macroNameLoc), LeftAngleLoc(leftAngleLoc),
|
|
RightAngleLoc(rightAngleLoc), GenericArgs(genericArgs),
|
|
ArgList(argList) {}
|
|
|
|
SourceLoc getLoc() const { return SigilLoc; }
|
|
SourceRange getGenericArgsRange() const {
|
|
return {LeftAngleLoc, RightAngleLoc};
|
|
}
|
|
SourceRange getSourceRange() const;
|
|
};
|
|
|
|
enum class FreestandingMacroKind {
|
|
Expr, // MacroExpansionExpr.
|
|
Decl, // MacroExpansionDecl.
|
|
};
|
|
|
|
/// A base class of either 'MacroExpansionExpr' or 'MacroExpansionDecl'.
|
|
class FreestandingMacroExpansion {
|
|
llvm::PointerIntPair<MacroExpansionInfo *, 1, FreestandingMacroKind>
|
|
infoAndKind;
|
|
|
|
protected:
|
|
FreestandingMacroExpansion(FreestandingMacroKind kind,
|
|
MacroExpansionInfo *info)
|
|
: infoAndKind(info, kind) {}
|
|
|
|
public:
|
|
MacroExpansionInfo *getExpansionInfo() const {
|
|
return infoAndKind.getPointer();
|
|
}
|
|
FreestandingMacroKind getFreestandingMacroKind() const {
|
|
return infoAndKind.getInt();
|
|
}
|
|
|
|
ASTNode getASTNode();
|
|
|
|
SourceLoc getPoundLoc() const { return getExpansionInfo()->SigilLoc; }
|
|
|
|
DeclNameLoc getModuleNameLoc() const {
|
|
return getExpansionInfo()->ModuleNameLoc;
|
|
}
|
|
DeclNameRef getModuleName() const { return getExpansionInfo()->ModuleName; }
|
|
DeclNameLoc getMacroNameLoc() const {
|
|
return getExpansionInfo()->MacroNameLoc;
|
|
}
|
|
DeclNameRef getMacroName() const { return getExpansionInfo()->MacroName; }
|
|
|
|
ArrayRef<TypeRepr *> getGenericArgs() const {
|
|
return getExpansionInfo()->GenericArgs;
|
|
}
|
|
SourceRange getGenericArgsRange() const {
|
|
return getExpansionInfo()->getGenericArgsRange();
|
|
}
|
|
|
|
ArgumentList *getArgs() const { return getExpansionInfo()->ArgList; }
|
|
void setArgs(ArgumentList *args) { getExpansionInfo()->ArgList = args; }
|
|
|
|
ConcreteDeclRef getMacroRef() const { return getExpansionInfo()->macroRef; }
|
|
void setMacroRef(ConcreteDeclRef ref) { getExpansionInfo()->macroRef = ref; }
|
|
|
|
DeclContext *getDeclContext() const;
|
|
SourceRange getSourceRange() const;
|
|
|
|
/// Returns a discriminator which determines this macro expansion's index
|
|
/// in the sequence of macro expansions within the current context.
|
|
unsigned getDiscriminator() const;
|
|
|
|
/// Returns the raw discriminator, for debugging purposes only.
|
|
unsigned getRawDiscriminator() const;
|
|
};
|
|
|
|
} // namespace swift
|
|
|
|
#endif // SWIFT_AST_FREESTANDING_MACRO_EXPANSION_H
|