mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
cd4ca76b6a
We have to work with selectors quite often, so provide an efficient representation for them. Switch ObjCAttr over to this representation, which has the nice property that it efficiently represents implicit @objc attributes with names and allows us to overwrite the Objective-C name without losing all source information. Addresses <rdar://problem/16478678>, and sets us up for dealing with selectors better. Swift SVN r16327
101 lines
2.7 KiB
C++
101 lines
2.7 KiB
C++
//===--- Identifier.cpp - Uniqued Identifier --------------------*- C++ -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the Identifier interface.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/AST/Identifier.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/Support/ConvertUTF.h"
|
|
using namespace swift;
|
|
|
|
|
|
raw_ostream &llvm::operator<<(raw_ostream &OS, Identifier I) {
|
|
if (I.get() == 0) return OS << "_";
|
|
return OS << I.get();
|
|
}
|
|
|
|
raw_ostream &llvm::operator<<(raw_ostream &OS, DeclName I) {
|
|
if (I.isSimpleName())
|
|
return OS << I.getBaseName();
|
|
|
|
OS << I.getBaseName() << "(";
|
|
for (auto c : I.getArgumentNames()) {
|
|
OS << c << ':';
|
|
}
|
|
OS << ")";
|
|
return OS;
|
|
}
|
|
|
|
raw_ostream &llvm::operator<<(raw_ostream &OS, swift::ObjCSelector S) {
|
|
unsigned n = S.getNumArgs();
|
|
if (n == 0) {
|
|
OS << S.getSelectorPieces()[0];
|
|
return OS;
|
|
}
|
|
|
|
for (auto piece : S.getSelectorPieces()) {
|
|
OS << piece << ':';
|
|
}
|
|
return OS;
|
|
}
|
|
|
|
bool Identifier::isOperatorSlow() const {
|
|
StringRef data = str();
|
|
auto *s = reinterpret_cast<UTF8 const *>(data.begin()),
|
|
*end = reinterpret_cast<UTF8 const *>(data.end());
|
|
UTF32 codePoint;
|
|
ConversionResult res = llvm::convertUTF8Sequence(&s, end, &codePoint,
|
|
strictConversion);
|
|
assert(res == conversionOK && "invalid UTF-8 in identifier?!");
|
|
(void)res;
|
|
return !empty() && isOperatorStartCodePoint(codePoint);
|
|
}
|
|
|
|
void DeclName::dump() {
|
|
llvm::errs() << *this << "\n";
|
|
}
|
|
|
|
ObjCSelector::ObjCSelector(ASTContext &ctx, unsigned numArgs,
|
|
ArrayRef<Identifier> pieces) {
|
|
if (numArgs == 0) {
|
|
assert(pieces.size() == 1 && "No-argument selector requires one piece");
|
|
Storage = DeclName(pieces[0]);
|
|
return;
|
|
}
|
|
|
|
assert(numArgs == pieces.size() && "Wrong number of selector pieces");
|
|
Storage = DeclName(ctx, Identifier(), pieces);
|
|
}
|
|
|
|
StringRef ObjCSelector::getString(llvm::SmallVectorImpl<char> &scratch) const {
|
|
// Fast path for zero-argument selectors.
|
|
if (getNumArgs() == 0) {
|
|
auto name = getSelectorPieces()[0];
|
|
if (name.empty())
|
|
return "_";
|
|
return name.str();
|
|
}
|
|
|
|
scratch.clear();
|
|
llvm::raw_svector_ostream os(scratch);
|
|
os << *this;
|
|
return os.str();
|
|
}
|
|
|
|
void ObjCSelector::dump() {
|
|
llvm::errs() << *this << "\n";
|
|
}
|
|
|
|
|