Landing Ben Alman's patch to add nextUntil, prevUntil, and parentsUntil. Also adds...
[jquery.git] / src / traversing.js
1 var runtil = /Until$/,
2         rparentsprev = /^(?:parents|prevUntil|prevAll)/,
3         // Note: This RegExp should be improved, or likely pulled from Sizzle
4         rmultiselector = /,/,
5         slice = Array.prototype.slice,
6         join = Array.prototype.join;
7
8 // Implement the identical functionality for filter and not
9 var winnow = function( elements, qualifier, keep ) {
10         if ( jQuery.isFunction( qualifier ) ) {
11                 return jQuery.grep(elements, function(elem, i) {
12                         return !!qualifier.call( elem, i ) === keep;
13                 });
14
15         } else if ( qualifier.nodeType ) {
16                 return jQuery.grep(elements, function(elem, i) {
17                         return (elem === qualifier) === keep;
18                 });
19
20         } else if ( typeof qualifier === "string" ) {
21                 var filtered = jQuery.grep(elements, function(elem) {
22                         return elem.nodeType === 1;
23                 });
24
25                 if ( isSimple.test( qualifier ) ) {
26                         return jQuery.filter(qualifier, filtered, !keep);
27                 } else {
28                         qualifier = jQuery.filter( qualifier, elements );
29                 }
30         }
31
32         return jQuery.grep(elements, function(elem, i) {
33                 return (jQuery.inArray( elem, qualifier ) >= 0) === keep;
34         });
35 };
36
37 jQuery.fn.extend({
38         find: function( selector ) {
39                 var ret = this.pushStack( "", "find", selector ), length = 0;
40
41                 for ( var i = 0, l = this.length; i < l; i++ ) {
42                         length = ret.length;
43                         jQuery.find( selector, this[i], ret );
44
45                         if ( i > 0 ) {
46                                 // Make sure that the results are unique
47                                 for ( var n = length; n < ret.length; n++ ) {
48                                         for ( var r = 0; r < length; r++ ) {
49                                                 if ( ret[r] === ret[n] ) {
50                                                         ret.splice(n--, 1);
51                                                         break;
52                                                 }
53                                         }
54                                 }
55                         }
56                 }
57
58                 return ret;
59         },
60
61         not: function( selector ) {
62                 return this.pushStack( winnow(this, selector, false), "not", selector);
63         },
64
65         filter: function( selector ) {
66                 return this.pushStack( winnow(this, selector, true), "filter", selector );
67         },
68
69         closest: function( selectors, context ) {
70                 if ( jQuery.isArray( selectors ) ) {
71                         var ret = [], cur = this[0], match, matches = {}, selector;
72
73                         if ( cur && selectors.length ) {
74                                 for ( var i = 0, l = selectors.length; i < l; i++ ) {
75                                         selector = selectors[i];
76
77                                         if ( !matches[selector] ) {
78                                                 matches[selector] = jQuery.expr.match.POS.test( selector ) ? 
79                                                         jQuery( selector, context || this.context ) :
80                                                         selector;
81                                         }
82                                 }
83
84                                 while ( cur && cur.ownerDocument && cur !== context ) {
85                                         for ( selector in matches ) {
86                                                 match = matches[selector];
87
88                                                 if ( match.jquery ? match.index(cur) > -1 : jQuery(cur).is(match) ) {
89                                                         ret.push({ selector: selector, elem: cur });
90                                                         delete matches[selector];
91                                                 }
92                                         }
93                                         cur = cur.parentNode;
94                                 }
95                         }
96
97                         return ret;
98                 }
99
100                 var pos = jQuery.expr.match.POS.test( selectors ) ? 
101                         jQuery( selectors, context || this.context ) : null;
102
103                 return this.map(function(i, cur){
104                         while ( cur && cur.ownerDocument && cur !== context ) {
105                                 if ( pos ? pos.index(cur) > -1 : jQuery(cur).is(selectors) ) {
106                                         return cur;
107                                 }
108                                 cur = cur.parentNode;
109                         }
110                         return null;
111                 });
112         },
113
114         add: function( selector, context ) {
115                 var set = typeof selector === "string" ?
116                                 jQuery( selector, context || this.context ) :
117                                 jQuery.makeArray( selector ),
118                         all = jQuery.merge( this.get(), set );
119
120                 return this.pushStack( set[0] && (set[0].setInterval || set[0].nodeType === 9 || (set[0].parentNode && set[0].parentNode.nodeType !== 11)) ?
121                         jQuery.unique( all ) :
122                         all );
123         },
124
125         eq: function( i ) {
126                 return i === -1 ?
127                         this.slice( i ) :
128                         this.slice( i, +i + 1 );
129         },
130
131         first: function() {
132                 return this.eq( 0 );
133         },
134
135         last: function() {
136                 return this.eq( -1 );
137         },
138
139         slice: function() {
140                 return this.pushStack( slice.apply( this, arguments ),
141                         "slice", join.call(arguments, ",") );
142         },
143
144         map: function( callback ) {
145                 return this.pushStack( jQuery.map(this, function(elem, i){
146                         return callback.call( elem, i, elem );
147                 }));
148         },
149
150         andSelf: function() {
151                 return this.add( this.prevObject );
152         },
153
154         end: function() {
155                 return this.prevObject || jQuery(null);
156         }
157 });
158
159 jQuery.each({
160         parent: function(elem){return elem.parentNode;},
161         parents: function(elem){return jQuery.dir(elem,"parentNode");},
162         parentsUntil: function(elem,i,until){return jQuery.dir(elem,"parentNode",until);},
163         next: function(elem){return jQuery.nth(elem,2,"nextSibling");},
164         prev: function(elem){return jQuery.nth(elem,2,"previousSibling");},
165         nextAll: function(elem){return jQuery.dir(elem,"nextSibling");},
166         prevAll: function(elem){return jQuery.dir(elem,"previousSibling");},
167         nextUntil: function(elem,i,until){return jQuery.dir(elem,"nextSibling",until);},
168         prevUntil: function(elem,i,until){return jQuery.dir(elem,"previousSibling",until);},
169         siblings: function(elem){return jQuery.sibling(elem.parentNode.firstChild,elem);},
170         children: function(elem){return jQuery.sibling(elem.firstChild);},
171         contents: function(elem){return jQuery.nodeName(elem,"iframe")?elem.contentDocument||elem.contentWindow.document:jQuery.makeArray(elem.childNodes);}
172 }, function(name, fn){
173         jQuery.fn[ name ] = function( until, selector ) {
174                 var ret = jQuery.map( this, fn, until );
175                 
176                 if ( !runtil.test( name ) ) {
177                         selector = until;
178                 }
179
180                 if ( selector && typeof selector === "string" ) {
181                         ret = jQuery.filter( selector, ret );
182                 }
183
184                 ret = this.length > 1 ? jQuery.unique( ret ) : ret;
185
186                 if ( (this.length > 1 || rmultiselector.test( selector )) && rparentsprev.test( name ) ) {
187                         ret = ret.reverse();
188                 }
189
190                 return this.pushStack( ret, name, join.call(arguments, ",") );
191         };
192 });
193
194 jQuery.extend({
195         filter: function( expr, elems, not ) {
196                 if ( not ) {
197                         expr = ":not(" + expr + ")";
198                 }
199
200                 return jQuery.find.matches(expr, elems);
201         },
202         
203         dir: function( elem, dir, until ) {
204                 var matched = [], cur = elem[dir];
205                 while ( cur && cur.nodeType !== 9 && (until === undefined || !jQuery( cur ).is( until )) ) {
206                         if ( cur.nodeType === 1 ) {
207                                 matched.push( cur );
208                         }
209                         cur = cur[dir];
210                 }
211                 return matched;
212         },
213
214         nth: function( cur, result, dir, elem ) {
215                 result = result || 1;
216                 var num = 0;
217
218                 for ( ; cur; cur = cur[dir] ) {
219                         if ( cur.nodeType === 1 && ++num === result ) {
220                                 break;
221                         }
222                 }
223
224                 return cur;
225         },
226
227         sibling: function( n, elem ) {
228                 var r = [];
229
230                 for ( ; n; n = n.nextSibling ) {
231                         if ( n.nodeType === 1 && n !== elem ) {
232                                 r.push( n );
233                         }
234                 }
235
236                 return r;
237         }
238 });