1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
|
//===--- MoveOnlyBorrowToDestructure.h ------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2022 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_MANDATORY_MOVEONLYBORROWTODESTRUCTURE_H
#define SWIFT_SILOPTIMIZER_MANDATORY_MOVEONLYBORROWTODESTRUCTURE_H
#include "swift/Basic/FrozenMultiMap.h"
#include "swift/SIL/FieldSensitivePrunedLiveness.h"
#include "swift/SIL/StackList.h"
#include "swift/SILOptimizer/Analysis/PostOrderAnalysis.h"
#include "llvm/ADT/IntervalMap.h"
namespace swift {
namespace siloptimizer {
class DiagnosticEmitter;
namespace borrowtodestructure {
class IntervalMapAllocator {
public:
using Map = llvm::IntervalMap<
unsigned, SILValue,
llvm::IntervalMapImpl::NodeSizer<unsigned, SILValue>::LeafSize,
llvm::IntervalMapHalfOpenInfo<unsigned>>;
using Allocator = Map::Allocator;
private:
/// Lazily initialized allocator.
std::optional<Allocator> allocator;
public:
Allocator &get() {
if (!allocator)
allocator.emplace();
return *allocator;
}
};
struct Implementation;
} // namespace borrowtodestructure
class BorrowToDestructureTransform {
friend borrowtodestructure::Implementation;
using IntervalMapAllocator = borrowtodestructure::IntervalMapAllocator;
IntervalMapAllocator &allocator;
MarkUnresolvedNonCopyableValueInst *mmci;
SILValue rootValue;
DiagnosticEmitter &diagnosticEmitter;
PostOrderAnalysis *poa;
PostOrderFunctionInfo *pofi = nullptr;
SmallVector<SILInstruction *, 8> createdDestructures;
SmallVector<SILPhiArgument *, 8> createdPhiArguments;
public:
BorrowToDestructureTransform(IntervalMapAllocator &allocator,
MarkUnresolvedNonCopyableValueInst *mmci,
SILValue rootValue,
DiagnosticEmitter &diagnosticEmitter,
PostOrderAnalysis *poa)
: allocator(allocator), mmci(mmci), rootValue(rootValue),
diagnosticEmitter(diagnosticEmitter), poa(poa) {}
bool transform();
private:
PostOrderFunctionInfo *getPostOrderFunctionInfo() {
if (!pofi)
pofi = poa->get(mmci->getFunction());
return pofi;
}
};
} // namespace siloptimizer
} // namespace swift
#endif
|