mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
68 lines
1.7 KiB
C++
68 lines
1.7 KiB
C++
//===--- CaptureInfo.cpp - Data Structure for Capture Lists ---------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/AST/CaptureInfo.h"
|
|
#include "swift/AST/Decl.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace swift;
|
|
|
|
bool CaptureInfo::hasLocalCaptures() const {
|
|
for (auto capture : getCaptures())
|
|
if (capture.getDecl()->getDeclContext()->isLocalContext())
|
|
return true;
|
|
return false;
|
|
}
|
|
|
|
|
|
void CaptureInfo::
|
|
getLocalCaptures(SmallVectorImpl<CapturedValue> &Result) const {
|
|
if (!hasLocalCaptures()) return;
|
|
|
|
Result.reserve(Count);
|
|
|
|
// Filter out global variables.
|
|
for (auto capture : getCaptures()) {
|
|
if (!capture.getDecl()->getDeclContext()->isLocalContext())
|
|
continue;
|
|
|
|
Result.push_back(capture);
|
|
}
|
|
}
|
|
|
|
void CaptureInfo::dump() const {
|
|
print(llvm::errs());
|
|
llvm::errs() << '\n';
|
|
}
|
|
|
|
void CaptureInfo::print(raw_ostream &OS) const {
|
|
OS << "captures=(";
|
|
|
|
if (hasGenericParamCaptures())
|
|
OS << "<generic> ";
|
|
if (hasDynamicSelfCapture())
|
|
OS << "<dynamic_self> ";
|
|
|
|
interleave(getCaptures(),
|
|
[&](const CapturedValue &capture) {
|
|
OS << capture.getDecl()->getName();
|
|
|
|
if (capture.isDirect())
|
|
OS << "<direct>";
|
|
if (capture.isNoEscape())
|
|
OS << "<noescape>";
|
|
},
|
|
[&] { OS << ", "; });
|
|
OS << ')';
|
|
}
|
|
|