mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
133 lines
4.5 KiB
C++
133 lines
4.5 KiB
C++
//===--- SpecializationMangler.h - mangling of specializations --*- C++ -*-===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 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_SILOPTIMIZER_UTILS_SPECIALIZATIONMANGLER_H
|
|
#define SWIFT_SILOPTIMIZER_UTILS_SPECIALIZATIONMANGLER_H
|
|
|
|
#include "swift/SIL/GenericSpecializationMangler.h"
|
|
|
|
#include "swift/Demangling/Demangler.h"
|
|
#include "swift/Demangling/NamespaceMacros.h"
|
|
#include "swift/Basic/NullablePtr.h"
|
|
#include "swift/AST/ASTMangler.h"
|
|
#include "swift/SIL/SILLinkage.h"
|
|
#include "swift/SIL/SILFunction.h"
|
|
|
|
namespace swift {
|
|
namespace Mangle {
|
|
SWIFT_BEGIN_INLINE_NAMESPACE
|
|
|
|
class PartialSpecializationMangler : public SpecializationMangler {
|
|
|
|
CanSILFunctionType SpecializedFnTy;
|
|
bool isReAbstracted;
|
|
|
|
public:
|
|
PartialSpecializationMangler(ASTContext &Ctx, SILFunction *F,
|
|
CanSILFunctionType SpecializedFnTy,
|
|
swift::SerializedKind_t Serialized, bool isReAbstracted)
|
|
: SpecializationMangler(Ctx, SpecializationPass::GenericSpecializer,
|
|
Serialized, F),
|
|
SpecializedFnTy(SpecializedFnTy), isReAbstracted(isReAbstracted) {}
|
|
|
|
std::string mangle();
|
|
};
|
|
|
|
// The mangler for functions where arguments or effects are specialized.
|
|
class FunctionSignatureSpecializationMangler : public SpecializationMangler {
|
|
|
|
using ReturnValueModifierIntBase = uint16_t;
|
|
enum class ReturnValueModifier : ReturnValueModifierIntBase {
|
|
// Option Space 4 bits (i.e. 16 options).
|
|
Unmodified=0,
|
|
First_Option=0, Last_Option=31,
|
|
|
|
// Option Set Space. 12 bits (i.e. 12 option).
|
|
Dead=32,
|
|
OwnedToUnowned=64,
|
|
First_OptionSetEntry=32, LastOptionSetEntry=32768,
|
|
};
|
|
|
|
// We use this private typealias to make it easy to expand ArgumentModifier's
|
|
// size if we need to.
|
|
using ArgumentModifierIntBase = uint16_t;
|
|
enum class ArgumentModifier : ArgumentModifierIntBase {
|
|
// Option Space 4 bits (i.e. 16 options).
|
|
Unmodified = 0,
|
|
ConstantProp = 1,
|
|
ClosureProp = 2,
|
|
BoxToValue = 3,
|
|
BoxToStack = 4,
|
|
InOutToOut = 5,
|
|
|
|
First_Option = 0,
|
|
Last_Option = 31,
|
|
|
|
// Option Set Space. 12 bits (i.e. 12 option).
|
|
Dead = 32,
|
|
OwnedToGuaranteed = 64,
|
|
SROA = 128,
|
|
GuaranteedToOwned = 256,
|
|
ExistentialToGeneric = 512,
|
|
First_OptionSetEntry = 32,
|
|
LastOptionSetEntry = 32768,
|
|
};
|
|
|
|
using ArgInfo = std::pair<ArgumentModifierIntBase,
|
|
NullablePtr<SILInstruction>>;
|
|
// Information for each SIL argument in the original function before
|
|
// specialization. This includes SIL indirect result argument required for
|
|
// the original function type at the current stage of compilation.
|
|
llvm::SmallVector<ArgInfo, 8> OrigArgs;
|
|
|
|
ReturnValueModifierIntBase ReturnValue;
|
|
|
|
public:
|
|
FunctionSignatureSpecializationMangler(ASTContext &Ctx, SpecializationPass Pass,
|
|
swift::SerializedKind_t Serialized,
|
|
SILFunction *F);
|
|
// For arguments / return values
|
|
void setArgumentConstantProp(unsigned OrigArgIdx, SILInstruction *constInst);
|
|
void appendStringAsIdentifier(StringRef str);
|
|
|
|
void setArgumentClosureProp(unsigned OrigArgIdx, PartialApplyInst *PAI);
|
|
void setArgumentClosureProp(unsigned OrigArgIdx,
|
|
ThinToThickFunctionInst *TTTFI);
|
|
void setArgumentDead(unsigned OrigArgIdx);
|
|
void setArgumentOwnedToGuaranteed(unsigned OrigArgIdx);
|
|
void setArgumentGuaranteedToOwned(unsigned OrigArgIdx);
|
|
void setArgumentExistentialToGeneric(unsigned OrigArgIdx);
|
|
void setArgumentSROA(unsigned OrigArgIdx);
|
|
void setArgumentBoxToValue(unsigned OrigArgIdx);
|
|
void setArgumentBoxToStack(unsigned OrigArgIdx);
|
|
void setArgumentInOutToOut(unsigned OrigArgIdx);
|
|
void setReturnValueOwnedToUnowned();
|
|
|
|
// For effects
|
|
void setRemovedEffect(EffectKind effect);
|
|
|
|
std::string mangle();
|
|
|
|
private:
|
|
void mangleConstantProp(SILInstruction *constInst);
|
|
void mangleClosureProp(SILInstruction *Inst);
|
|
void mangleArgument(ArgumentModifierIntBase ArgMod,
|
|
NullablePtr<SILInstruction> Inst);
|
|
void mangleReturnValue(ReturnValueModifierIntBase RetMod);
|
|
};
|
|
|
|
SWIFT_END_INLINE_NAMESPACE
|
|
} // end namespace Mangle
|
|
} // end namespace swift
|
|
|
|
#endif
|