1//===--- SelectorLocationsKind.cpp - Kind of selector locations -*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// Describes whether the identifier locations for a selector are "standard"
10// or not.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/AST/SelectorLocationsKind.h"
15#include "clang/AST/Expr.h"
16
17using namespace clang;
18
19static SourceLocation getStandardSelLoc(unsigned Index,
20 Selector Sel,
21 bool WithArgSpace,
22 SourceLocation ArgLoc,
23 SourceLocation EndLoc) {
24 unsigned NumSelArgs = Sel.getNumArgs();
25 if (NumSelArgs == 0) {
26 assert(Index == 0);
27 if (EndLoc.isInvalid())
28 return SourceLocation();
29 const IdentifierInfo *II = Sel.getIdentifierInfoForSlot(argIndex: 0);
30 unsigned Len = II ? II->getLength() : 0;
31 return EndLoc.getLocWithOffset(Offset: -Len);
32 }
33
34 assert(Index < NumSelArgs);
35 if (ArgLoc.isInvalid())
36 return SourceLocation();
37 const IdentifierInfo *II = Sel.getIdentifierInfoForSlot(argIndex: Index);
38 unsigned Len = /* selector id */ (II ? II->getLength() : 0) + /* ':' */ 1;
39 if (WithArgSpace)
40 ++Len;
41 return ArgLoc.getLocWithOffset(Offset: -Len);
42}
43
44namespace {
45
46template <typename T>
47SourceLocation getArgLoc(T* Arg);
48
49template <>
50SourceLocation getArgLoc<Expr>(Expr *Arg) {
51 return Arg->getBeginLoc();
52}
53
54template <>
55SourceLocation getArgLoc<ParmVarDecl>(ParmVarDecl *Arg) {
56 SourceLocation Loc = Arg->getBeginLoc();
57 if (Loc.isInvalid())
58 return Loc;
59 // -1 to point to left paren of the method parameter's type.
60 return Loc.getLocWithOffset(Offset: -1);
61}
62
63template <typename T>
64SourceLocation getArgLoc(unsigned Index, ArrayRef<T*> Args) {
65 return Index < Args.size() ? getArgLoc(Args[Index]) : SourceLocation();
66}
67
68template <typename T>
69SelectorLocationsKind hasStandardSelLocs(Selector Sel,
70 ArrayRef<SourceLocation> SelLocs,
71 ArrayRef<T *> Args,
72 SourceLocation EndLoc) {
73 // Are selector locations in standard position with no space between args ?
74 unsigned i;
75 for (i = 0; i != SelLocs.size(); ++i) {
76 if (SelLocs[i] != getStandardSelectorLoc(i, Sel, /*WithArgSpace=*/false,
77 Args, EndLoc))
78 break;
79 }
80 if (i == SelLocs.size())
81 return SelLoc_StandardNoSpace;
82
83 // Are selector locations in standard position with space between args ?
84 for (i = 0; i != SelLocs.size(); ++i) {
85 if (SelLocs[i] != getStandardSelectorLoc(i, Sel, /*WithArgSpace=*/true,
86 Args, EndLoc))
87 return SelLoc_NonStandard;
88 }
89
90 return SelLoc_StandardWithSpace;
91}
92
93} // anonymous namespace
94
95SelectorLocationsKind
96clang::hasStandardSelectorLocs(Selector Sel,
97 ArrayRef<SourceLocation> SelLocs,
98 ArrayRef<Expr *> Args,
99 SourceLocation EndLoc) {
100 return hasStandardSelLocs(Sel, SelLocs, Args, EndLoc);
101}
102
103SourceLocation clang::getStandardSelectorLoc(unsigned Index,
104 Selector Sel,
105 bool WithArgSpace,
106 ArrayRef<Expr *> Args,
107 SourceLocation EndLoc) {
108 return getStandardSelLoc(Index, Sel, WithArgSpace,
109 ArgLoc: getArgLoc(Index, Args), EndLoc);
110}
111
112SelectorLocationsKind
113clang::hasStandardSelectorLocs(Selector Sel,
114 ArrayRef<SourceLocation> SelLocs,
115 ArrayRef<ParmVarDecl *> Args,
116 SourceLocation EndLoc) {
117 return hasStandardSelLocs(Sel, SelLocs, Args, EndLoc);
118}
119
120SourceLocation clang::getStandardSelectorLoc(unsigned Index,
121 Selector Sel,
122 bool WithArgSpace,
123 ArrayRef<ParmVarDecl *> Args,
124 SourceLocation EndLoc) {
125 return getStandardSelLoc(Index, Sel, WithArgSpace,
126 ArgLoc: getArgLoc(Index, Args), EndLoc);
127}
128