mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
106 lines
3.2 KiB
C++
106 lines
3.2 KiB
C++
//===--- SILLocation.cpp - Location information for SIL nodes -------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2015 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See http://swift.org/LICENSE.txt for license information
|
|
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/SIL/SILLocation.h"
|
|
#include "swift/AST/Decl.h"
|
|
#include "swift/AST/Expr.h"
|
|
#include "swift/AST/Pattern.h"
|
|
#include "swift/AST/Stmt.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace swift;
|
|
|
|
SourceLoc SILLocation::getSourceLoc() const {
|
|
if (ASTNode.isNull())
|
|
return SILFileSourceLoc;
|
|
|
|
if (alwaysPointsToStart())
|
|
return getStartSourceLoc();
|
|
|
|
// Use the end location for the CleanupKind.
|
|
if (getKind() == CleanupKind) {
|
|
return getEndSourceLoc();
|
|
}
|
|
|
|
// Use the end location for the ImplicitReturnKind.
|
|
if (getKind() == ImplicitReturnKind) {
|
|
return getEndSourceLoc();
|
|
}
|
|
|
|
// Use the start location for the ReturnKind.
|
|
if (getKind() == ReturnKind) {
|
|
return getStartSourceLoc();
|
|
}
|
|
|
|
if (auto decl = ASTNode.dyn_cast<Decl*>())
|
|
return decl->getLoc();
|
|
if (auto expr = ASTNode.dyn_cast<Expr*>())
|
|
return expr->getLoc();
|
|
if (auto stmt = ASTNode.dyn_cast<Stmt*>())
|
|
return stmt->getStartLoc();
|
|
if (auto patt = ASTNode.dyn_cast<Pattern*>())
|
|
return patt->getStartLoc();
|
|
llvm_unreachable("impossible SILLocation");
|
|
}
|
|
|
|
SourceLoc SILLocation::getStartSourceLoc() const {
|
|
if (ASTNode.isNull())
|
|
return SILFileSourceLoc;
|
|
if (auto decl = ASTNode.dyn_cast<Decl*>())
|
|
return decl->getStartLoc();
|
|
if (auto expr = ASTNode.dyn_cast<Expr*>())
|
|
return expr->getStartLoc();
|
|
if (auto stmt = ASTNode.dyn_cast<Stmt*>())
|
|
return stmt->getStartLoc();
|
|
if (auto patt = ASTNode.dyn_cast<Pattern*>())
|
|
return patt->getStartLoc();
|
|
llvm_unreachable("impossible SILLocation");
|
|
}
|
|
|
|
SourceLoc SILLocation::getEndSourceLoc() const {
|
|
if (ASTNode.isNull())
|
|
return SILFileSourceLoc;
|
|
if (auto decl = ASTNode.dyn_cast<Decl*>())
|
|
return decl->getEndLoc();
|
|
if (auto expr = ASTNode.dyn_cast<Expr*>())
|
|
return expr->getEndLoc();
|
|
if (auto stmt = ASTNode.dyn_cast<Stmt*>())
|
|
return stmt->getEndLoc();
|
|
if (auto patt = ASTNode.dyn_cast<Pattern*>())
|
|
return patt->getEndLoc();
|
|
llvm_unreachable("impossible SILLocation");
|
|
}
|
|
|
|
void SILLocation::dump(const SourceManager &SM) const {
|
|
print(llvm::errs(), SM);
|
|
}
|
|
void SILLocation::print(raw_ostream &OS, const SourceManager &SM) const {
|
|
if (isNull())
|
|
OS << "<no loc>";
|
|
getSourceLoc().print(OS, SM);
|
|
}
|
|
|
|
CleanupLocation CleanupLocation::getCleanupLocation(SILLocation L) {
|
|
if (Expr *E = L.getAsASTNode<Expr>())
|
|
return CleanupLocation(E);
|
|
if (Stmt *S = L.getAsASTNode<Stmt>())
|
|
return CleanupLocation(S);
|
|
if (Pattern *P = L.getAsASTNode<Pattern>())
|
|
return CleanupLocation(P);
|
|
if (Decl *D = L.getAsASTNode<Decl>())
|
|
return CleanupLocation(D);
|
|
if (L.isNull())
|
|
return CleanupLocation();
|
|
llvm_unreachable("Cannot construct Cleanup loc from the "
|
|
"given location.");
|
|
}
|