Make .add() take an optional context and - if a context is specified in the root...
[jquery.git] / src / traversing.js
1 var winnow = function( elements, qualifier, keep ) {
2         if(jQuery.isFunction( qualifier )) {
3                 return jQuery.grep(elements, function(elem, i) {
4                         return !!qualifier.call( elem, i ) === keep;
5                 });
6         } else if( qualifier.nodeType ) {
7                 return jQuery.grep(elements, function(elem, i) {
8                         return (elem === qualifier) === keep;
9                 })
10         } else if( typeof qualifier === "string" ) {
11                 var filtered = jQuery.grep(elements, function(elem) { return elem.nodeType === 1 });
12
13                 if(isSimple.test( qualifier )) return jQuery.filter(qualifier, filtered, !keep);
14                 else qualifier = jQuery.filter( qualifier, elements );
15         }
16
17         return jQuery.grep(elements, function(elem, i) {
18                 return (jQuery.inArray( elem, qualifier ) >= 0) === keep;
19         });
20 }
21
22 jQuery.fn.extend({
23         find: function( selector ) {
24                 var ret = this.pushStack( "", "find", selector ), length = 0;
25
26                 for ( var i = 0, l = this.length; i < l; i++ ) {
27                         length = ret.length;
28                         jQuery.find( selector, this[i], ret );
29
30                         if ( i > 0 ) {
31                                 // Make sure that the results are unique
32                                 for ( var n = length; n < ret.length; n++ ) {
33                                         for ( var r = 0; r < length; r++ ) {
34                                                 if ( ret[r] === ret[n] ) {
35                                                         ret.splice(n--, 1);
36                                                         break;
37                                                 }
38                                         }
39                                 }
40                         }
41                 }
42
43                 return ret;
44         },
45
46         not: function( selector ) {
47                 return this.pushStack( winnow(this, selector, false), "not", selector);
48         },
49
50         filter: function( selector ) {
51                 return this.pushStack( winnow(this, selector, true), "filter", selector );
52         },
53
54         closest: function( selector, context ) {
55                 var pos = jQuery.expr.match.POS.test( selector ) ? jQuery(selector) : null,
56                         closer = 0;
57
58                 return this.map(function(){
59                         var cur = this;
60                         while ( cur && cur.ownerDocument && cur !== context ) {
61                                 if ( pos ? pos.index(cur) > -1 : jQuery(cur).is(selector) ) {
62                                         jQuery.data(cur, "closest", closer);
63                                         return cur;
64                                 }
65                                 cur = cur.parentNode;
66                                 closer++;
67                         }
68                 });
69         },
70
71         add: function( selector, context ) {
72                 var set = typeof selector === "string" ?
73                                 jQuery( selector, context || this.context ) :
74                                 jQuery.makeArray( selector ),
75                         all = jQuery.merge( this.get(), set );
76
77                 return this.pushStack( set[0] && (set[0].setInterval || set[0].nodeType === 9 || (set[0].parentNode && set[0].parentNode.nodeType !== 11)) ?
78                         jQuery.unique( all ) :
79                         all );
80         },
81
82         eq: function( i ) {
83                 return i === -1 ?
84                         this.slice( i ) :
85                         this.slice( i, +i + 1 );
86         },
87
88         first: function() {
89                 return this.eq( 0 );
90         },
91
92         last: function() {
93                 return this.eq( -1 );
94         },
95
96         slice: function() {
97                 return this.pushStack( Array.prototype.slice.apply( this, arguments ),
98                         "slice", Array.prototype.slice.call(arguments).join(",") );
99         },
100
101         map: function( callback ) {
102                 return this.pushStack( jQuery.map(this, function(elem, i){
103                         return callback.call( elem, i, elem );
104                 }));
105         },
106
107         andSelf: function() {
108                 return this.add( this.prevObject );
109         },
110
111         end: function() {
112                 return this.prevObject || jQuery(null);
113         }
114 });
115
116 jQuery.each({
117         parent: function(elem){return elem.parentNode;},
118         parents: function(elem){return jQuery.dir(elem,"parentNode");},
119         next: function(elem){return jQuery.nth(elem,2,"nextSibling");},
120         prev: function(elem){return jQuery.nth(elem,2,"previousSibling");},
121         nextAll: function(elem){return jQuery.dir(elem,"nextSibling");},
122         prevAll: function(elem){return jQuery.dir(elem,"previousSibling");},
123         siblings: function(elem){return jQuery.sibling(elem.parentNode.firstChild,elem);},
124         children: function(elem){return jQuery.sibling(elem.firstChild);},
125         contents: function(elem){return jQuery.nodeName(elem,"iframe")?elem.contentDocument||elem.contentWindow.document:jQuery.makeArray(elem.childNodes);}
126 }, function(name, fn){
127         jQuery.fn[ name ] = function( selector ) {
128                 var ret = jQuery.map( this, fn );
129
130                 if ( selector && typeof selector === "string" ) {
131                         ret = jQuery.filter( selector, ret );
132                 }
133
134                 ret = this.length > 1 ? jQuery.unique( ret ) : ret;
135
136                 if ( name === "parents" && this.length > 1 ) {
137                         ret = ret.reverse();
138                 }
139
140                 return this.pushStack( ret, name, selector );
141         };
142 });
143
144 jQuery.extend({
145         filter: function( expr, elems, not ) {
146                 if ( not ) {
147                         expr = ":not(" + expr + ")";
148                 }
149
150                 return jQuery.find.matches(expr, elems);
151         },
152         
153         dir: function( elem, dir ) {
154                 var matched = [], cur = elem[dir];
155                 while ( cur && cur.nodeType !== 9 ) {
156                         if ( cur.nodeType === 1 ) {
157                                 matched.push( cur );
158                         }
159                         cur = cur[dir];
160                 }
161                 return matched;
162         },
163
164         nth: function( cur, result, dir, elem ) {
165                 result = result || 1;
166                 var num = 0;
167
168                 for ( ; cur; cur = cur[dir] ) {
169                         if ( cur.nodeType === 1 && ++num === result ) {
170                                 break;
171                         }
172                 }
173
174                 return cur;
175         },
176
177         sibling: function( n, elem ) {
178                 var r = [];
179
180                 for ( ; n; n = n.nextSibling ) {
181                         if ( n.nodeType === 1 && n !== elem ) {
182                                 r.push( n );
183                         }
184                 }
185
186                 return r;
187         }
188 });