Use the new Sizzle matches functionality.
[jquery.git] / src / traversing.js
1 (function( jQuery ) {
2
3 var runtil = /Until$/,
4         rparentsprev = /^(?:parents|prevUntil|prevAll)/,
5         // Note: This RegExp should be improved, or likely pulled from Sizzle
6         rmultiselector = /,/,
7         rchild = /^\s*>/,
8         isSimple = /^.[^:#\[\.,]*$/,
9         slice = Array.prototype.slice;
10
11 var POS = jQuery.expr.match.POS;
12
13 jQuery.fn.extend({
14         find: function( selector ) {
15                 // Handle "> div" child selectors and pass them to .children()
16                 if ( typeof selector === "string" && rchild.test( selector ) ) {
17                         return this.children( selector.replace( rchild, "" ) );
18                 }
19
20                 var ret = this.pushStack( "", "find", selector ), length = 0;
21
22                 for ( var i = 0, l = this.length; i < l; i++ ) {
23                         length = ret.length;
24                         jQuery.find( selector, this[i], ret );
25
26                         if ( i > 0 ) {
27                                 // Make sure that the results are unique
28                                 for ( var n = length; n < ret.length; n++ ) {
29                                         for ( var r = 0; r < length; r++ ) {
30                                                 if ( ret[r] === ret[n] ) {
31                                                         ret.splice(n--, 1);
32                                                         break;
33                                                 }
34                                         }
35                                 }
36                         }
37                 }
38
39                 return ret;
40         },
41
42         has: function( target ) {
43                 var targets = jQuery( target );
44                 return this.filter(function() {
45                         for ( var i = 0, l = targets.length; i < l; i++ ) {
46                                 if ( jQuery.contains( this, targets[i] ) ) {
47                                         return true;
48                                 }
49                         }
50                 });
51         },
52
53         not: function( selector ) {
54                 return this.pushStack( winnow(this, selector, false), "not", selector);
55         },
56
57         filter: function( selector ) {
58                 return this.pushStack( winnow(this, selector, true), "filter", selector );
59         },
60         
61         is: function( selector ) {
62                 return !!selector && jQuery.filter( selector, this ).length > 0;
63         },
64
65         closest: function( selectors, context ) {
66                 var ret;
67
68                 if ( jQuery.isArray( selectors ) ) {
69                         var cur = this[0], match, matches = {}, selector, level = 1;
70                         ret = [];
71
72                         if ( cur && selectors.length ) {
73                                 for ( var i = 0, l = selectors.length; i < l; i++ ) {
74                                         selector = selectors[i];
75
76                                         if ( !matches[selector] ) {
77                                                 matches[selector] = jQuery.expr.match.POS.test( selector ) ? 
78                                                         jQuery( selector, context || this.context ) :
79                                                         selector;
80                                         }
81                                 }
82
83                                 while ( cur && cur.ownerDocument && cur !== context ) {
84                                         for ( selector in matches ) {
85                                                 match = matches[selector];
86
87                                                 if ( match.jquery ? match.index(cur) > -1 : jQuery(cur).is(match) ) {
88                                                         ret.push({ selector: selector, elem: cur, level: level });
89                                                 }
90                                         }
91
92                                         cur = cur.parentNode;
93                                         level++;
94                                 }
95                         }
96
97                         return ret;
98                 }
99
100                 var pos = POS.test( selectors ) ? 
101                         jQuery( selectors, context || this.context ) : null;
102
103                 var ret = [];
104
105                 for ( var i = 0, j = this.length; i < j; i++ ) {
106                         var cur = this[i];
107
108                         while ( cur ) {
109                                 if ( pos ? pos.index(cur) > -1 : jQuery.find.matches(selectors, cur) ) {
110                                         ret.push( cur );
111                                         break;
112
113                                 } else {
114                                         cur = cur.parentNode;
115                                         if ( !cur.ownerDocument || cur === context ) {
116                                                 break;
117                                         }
118                                 }
119                         }
120                 }
121
122                 ret = ret.length > 1 ? jQuery.unique(ret) : ret;
123                 
124                 return this.pushStack( ret, "closest", selectors );
125         },
126         
127         // Determine the position of an element within
128         // the matched set of elements
129         index: function( elem ) {
130                 if ( !elem || typeof elem === "string" ) {
131                         return jQuery.inArray( this[0],
132                                 // If it receives a string, the selector is used
133                                 // If it receives nothing, the siblings are used
134                                 elem ? jQuery( elem ) : this.parent().children() );
135                 }
136                 // Locate the position of the desired element
137                 return jQuery.inArray(
138                         // If it receives a jQuery object, the first element is used
139                         elem.jquery ? elem[0] : elem, this );
140         },
141
142         add: function( selector, context ) {
143                 var set = typeof selector === "string" ?
144                                 jQuery( selector, context || this.context ) :
145                                 jQuery.makeArray( selector ),
146                         all = jQuery.merge( this.get(), set );
147
148                 return this.pushStack( isDisconnected( set[0] ) || isDisconnected( all[0] ) ?
149                         all :
150                         jQuery.unique( all ) );
151         },
152
153         andSelf: function() {
154                 return this.add( this.prevObject );
155         }
156 });
157
158 // A painfully simple check to see if an element is disconnected
159 // from a document (should be improved, where feasible).
160 function isDisconnected( node ) {
161         return !node || !node.parentNode || node.parentNode.nodeType === 11;
162 }
163
164 jQuery.each({
165         parent: function( elem ) {
166                 var parent = elem.parentNode;
167                 return parent && parent.nodeType !== 11 ? parent : null;
168         },
169         parents: function( elem ) {
170                 return jQuery.dir( elem, "parentNode" );
171         },
172         parentsUntil: function( elem, i, until ) {
173                 return jQuery.dir( elem, "parentNode", until );
174         },
175         next: function( elem ) {
176                 return jQuery.nth( elem, 2, "nextSibling" );
177         },
178         prev: function( elem ) {
179                 return jQuery.nth( elem, 2, "previousSibling" );
180         },
181         nextAll: function( elem ) {
182                 return jQuery.dir( elem, "nextSibling" );
183         },
184         prevAll: function( elem ) {
185                 return jQuery.dir( elem, "previousSibling" );
186         },
187         nextUntil: function( elem, i, until ) {
188                 return jQuery.dir( elem, "nextSibling", until );
189         },
190         prevUntil: function( elem, i, until ) {
191                 return jQuery.dir( elem, "previousSibling", until );
192         },
193         siblings: function( elem ) {
194                 return jQuery.sibling( elem.parentNode.firstChild, elem );
195         },
196         children: function( elem ) {
197                 return jQuery.sibling( elem.firstChild );
198         },
199         contents: function( elem ) {
200                 return jQuery.nodeName( elem, "iframe" ) ?
201                         elem.contentDocument || elem.contentWindow.document :
202                         jQuery.makeArray( elem.childNodes );
203         }
204 }, function( name, fn ) {
205         jQuery.fn[ name ] = function( until, selector ) {
206                 var ret = jQuery.map( this, fn, until );
207                 
208                 if ( !runtil.test( name ) ) {
209                         selector = until;
210                 }
211
212                 if ( selector && typeof selector === "string" ) {
213                         ret = jQuery.filter( selector, ret );
214                 }
215
216                 ret = this.length > 1 ? jQuery.unique( ret ) : ret;
217
218                 if ( (this.length > 1 || rmultiselector.test( selector )) && rparentsprev.test( name ) ) {
219                         ret = ret.reverse();
220                 }
221
222                 return this.pushStack( ret, name, slice.call(arguments).join(",") );
223         };
224 });
225
226 jQuery.extend({
227         filter: function( expr, elems, not ) {
228                 if ( not ) {
229                         expr = ":not(" + expr + ")";
230                 }
231
232                 return jQuery.find.matches(expr, elems);
233         },
234         
235         dir: function( elem, dir, until ) {
236                 var matched = [], cur = elem[dir];
237                 while ( cur && cur.nodeType !== 9 && (until === undefined || cur.nodeType !== 1 || !jQuery( cur ).is( until )) ) {
238                         if ( cur.nodeType === 1 ) {
239                                 matched.push( cur );
240                         }
241                         cur = cur[dir];
242                 }
243                 return matched;
244         },
245
246         nth: function( cur, result, dir, elem ) {
247                 result = result || 1;
248                 var num = 0;
249
250                 for ( ; cur; cur = cur[dir] ) {
251                         if ( cur.nodeType === 1 && ++num === result ) {
252                                 break;
253                         }
254                 }
255
256                 return cur;
257         },
258
259         sibling: function( n, elem ) {
260                 var r = [];
261
262                 for ( ; n; n = n.nextSibling ) {
263                         if ( n.nodeType === 1 && n !== elem ) {
264                                 r.push( n );
265                         }
266                 }
267
268                 return r;
269         }
270 });
271
272 // Implement the identical functionality for filter and not
273 function winnow( elements, qualifier, keep ) {
274         if ( jQuery.isFunction( qualifier ) ) {
275                 return jQuery.grep(elements, function( elem, i ) {
276                         var retVal = !!qualifier.call( elem, i, elem );
277                         return retVal === keep;
278                 });
279
280         } else if ( qualifier.nodeType ) {
281                 return jQuery.grep(elements, function( elem, i ) {
282                         return (elem === qualifier) === keep;
283                 });
284
285         } else if ( typeof qualifier === "string" ) {
286                 var filtered = jQuery.grep(elements, function( elem ) {
287                         return elem.nodeType === 1;
288                 });
289
290                 if ( isSimple.test( qualifier ) ) {
291                         return jQuery.filter(qualifier, filtered, !keep);
292                 } else {
293                         qualifier = jQuery.filter( qualifier, filtered );
294                 }
295         }
296
297         return jQuery.grep(elements, function( elem, i ) {
298                 return (jQuery.inArray( elem, qualifier ) >= 0) === keep;
299         });
300 }
301
302 })( jQuery );