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