mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
91 lines
3.3 KiB
C++
91 lines
3.3 KiB
C++
//===--- DefaultArgumentKind.cpp - Default Argument Implementation --------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2016 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements utilities associated with default arguments.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#include "swift/AST/DefaultArgumentKind.h"
|
|
#include "swift/AST/ASTContext.h"
|
|
#include "swift/AST/Decl.h"
|
|
#include "swift/AST/Expr.h"
|
|
using namespace swift;
|
|
|
|
StringRef swift::getDefaultArgumentSpelling(DefaultArgumentKind kind) {
|
|
switch (kind) {
|
|
case DefaultArgumentKind::None:
|
|
case DefaultArgumentKind::Normal:
|
|
case DefaultArgumentKind::Inherited:
|
|
return StringRef();
|
|
case DefaultArgumentKind::File: return "#file";
|
|
case DefaultArgumentKind::Line: return "#line";
|
|
case DefaultArgumentKind::Column: return "#column";
|
|
case DefaultArgumentKind::Function: return "#function";
|
|
case DefaultArgumentKind::DSOHandle: return "#dsohandle";
|
|
case DefaultArgumentKind::Nil: return "nil";
|
|
case DefaultArgumentKind::EmptyArray: return "[]";
|
|
case DefaultArgumentKind::EmptyDictionary: return "[:]";
|
|
}
|
|
}
|
|
|
|
DefaultArgumentKind swift::inferDefaultArgumentKind(Expr *expr) {
|
|
if (auto call = dyn_cast<CallExpr>(expr)) {
|
|
if (auto ctorRefCall = dyn_cast<ConstructorRefCallExpr>(call->getFn())) {
|
|
if (auto ctorRef = dyn_cast<DeclRefExpr>(ctorRefCall->getFn())) {
|
|
if (auto ctor = dyn_cast<ConstructorDecl>(ctorRef->getDecl())) {
|
|
auto ctorArg = call->getArg()->getSemanticsProvidingExpr();
|
|
|
|
// #file, #line, #column, #function, #dsohandle.
|
|
if (auto magic = dyn_cast<MagicIdentifierLiteralExpr>(ctorArg)) {
|
|
switch (magic->getKind()) {
|
|
case MagicIdentifierLiteralExpr::File:
|
|
return DefaultArgumentKind::File;
|
|
case MagicIdentifierLiteralExpr::Line:
|
|
return DefaultArgumentKind::Line;
|
|
case MagicIdentifierLiteralExpr::Column:
|
|
return DefaultArgumentKind::Column;
|
|
case MagicIdentifierLiteralExpr::Function:
|
|
return DefaultArgumentKind::Function;
|
|
case MagicIdentifierLiteralExpr::DSOHandle:
|
|
return DefaultArgumentKind::DSOHandle;
|
|
}
|
|
}
|
|
|
|
// nil.
|
|
if (ctor->getFullName().getArgumentNames().size() == 1 &&
|
|
ctor->getFullName().getArgumentNames()[0]
|
|
== ctor->getASTContext().Id_nilLiteral)
|
|
return DefaultArgumentKind::Nil;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Empty array literals, [].
|
|
if (auto arrayExpr = dyn_cast<ArrayExpr>(expr)) {
|
|
if (arrayExpr->getElements().empty())
|
|
return DefaultArgumentKind::EmptyArray;
|
|
|
|
return DefaultArgumentKind::None;
|
|
}
|
|
|
|
// Empty dictionary literals, [:].
|
|
if (auto dictionaryExpr = dyn_cast<DictionaryExpr>(expr)) {
|
|
if (dictionaryExpr->getElements().empty())
|
|
return DefaultArgumentKind::EmptyDictionary;
|
|
|
|
return DefaultArgumentKind::None;
|
|
}
|
|
|
|
return DefaultArgumentKind::None;
|
|
}
|
|
|