blob: 2e6aa2bd2077cbf1509f9b29ca7c94b60544061c [file] [log] [blame]
//===--- ApplySite.cpp - Wrapper around apply instructions ----------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2020 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
//
//===----------------------------------------------------------------------===//
#include "swift/SIL/ApplySite.h"
#include "swift/SIL/SILBuilder.h"
using namespace swift;
void FullApplySite::insertAfterInvocation(function_ref<void(SILBuilder &)> func) const {
SILBuilderWithScope::insertAfter(getInstruction(), func);
}
void FullApplySite::insertAfterFullEvaluation(
function_ref<void(SILBuilder &)> func) const {
switch (getKind()) {
case FullApplySiteKind::ApplyInst:
case FullApplySiteKind::TryApplyInst:
return insertAfterInvocation(func);
case FullApplySiteKind::BeginApplyInst:
SmallVector<EndApplyInst *, 2> endApplies;
SmallVector<AbortApplyInst *, 2> abortApplies;
auto *bai = cast<BeginApplyInst>(getInstruction());
bai->getCoroutineEndPoints(endApplies, abortApplies);
for (auto *eai : endApplies) {
SILBuilderWithScope builder(std::next(eai->getIterator()));
func(builder);
}
for (auto *aai : abortApplies) {
SILBuilderWithScope builder(std::next(aai->getIterator()));
func(builder);
}
return;
}
llvm_unreachable("covered switch isn't covered");
}