]> git.lizzy.rs Git - rust.git/blob - src/libcore/option.rs
Auto merge of #45623 - mneumann:dragonfly-ci, r=alexcrichton
[rust.git] / src / libcore / option.rs
1 // Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
4 //
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
10
11 //! Optional values.
12 //!
13 //! Type [`Option`] represents an optional value: every [`Option`]
14 //! is either [`Some`] and contains a value, or [`None`], and
15 //! does not. [`Option`] types are very common in Rust code, as
16 //! they have a number of uses:
17 //!
18 //! * Initial values
19 //! * Return values for functions that are not defined
20 //!   over their entire input range (partial functions)
21 //! * Return value for otherwise reporting simple errors, where `None` is
22 //!   returned on error
23 //! * Optional struct fields
24 //! * Struct fields that can be loaned or "taken"
25 //! * Optional function arguments
26 //! * Nullable pointers
27 //! * Swapping things out of difficult situations
28 //!
29 //! [`Option`]s are commonly paired with pattern matching to query the presence
30 //! of a value and take action, always accounting for the [`None`] case.
31 //!
32 //! ```
33 //! fn divide(numerator: f64, denominator: f64) -> Option<f64> {
34 //!     if denominator == 0.0 {
35 //!         None
36 //!     } else {
37 //!         Some(numerator / denominator)
38 //!     }
39 //! }
40 //!
41 //! // The return value of the function is an option
42 //! let result = divide(2.0, 3.0);
43 //!
44 //! // Pattern match to retrieve the value
45 //! match result {
46 //!     // The division was valid
47 //!     Some(x) => println!("Result: {}", x),
48 //!     // The division was invalid
49 //!     None    => println!("Cannot divide by 0"),
50 //! }
51 //! ```
52 //!
53 //
54 // FIXME: Show how `Option` is used in practice, with lots of methods
55 //
56 //! # Options and pointers ("nullable" pointers)
57 //!
58 //! Rust's pointer types must always point to a valid location; there are
59 //! no "null" pointers. Instead, Rust has *optional* pointers, like
60 //! the optional owned box, [`Option`]`<`[`Box<T>`]`>`.
61 //!
62 //! The following example uses [`Option`] to create an optional box of
63 //! [`i32`]. Notice that in order to use the inner [`i32`] value first, the
64 //! `check_optional` function needs to use pattern matching to
65 //! determine whether the box has a value (i.e. it is [`Some(...)`][`Some`]) or
66 //! not ([`None`]).
67 //!
68 //! ```
69 //! let optional = None;
70 //! check_optional(optional);
71 //!
72 //! let optional = Some(Box::new(9000));
73 //! check_optional(optional);
74 //!
75 //! fn check_optional(optional: Option<Box<i32>>) {
76 //!     match optional {
77 //!         Some(ref p) => println!("has value {}", p),
78 //!         None => println!("has no value"),
79 //!     }
80 //! }
81 //! ```
82 //!
83 //! This usage of [`Option`] to create safe nullable pointers is so
84 //! common that Rust does special optimizations to make the
85 //! representation of [`Option`]`<`[`Box<T>`]`>` a single pointer. Optional pointers
86 //! in Rust are stored as efficiently as any other pointer type.
87 //!
88 //! # Examples
89 //!
90 //! Basic pattern matching on [`Option`]:
91 //!
92 //! ```
93 //! let msg = Some("howdy");
94 //!
95 //! // Take a reference to the contained string
96 //! if let Some(ref m) = msg {
97 //!     println!("{}", *m);
98 //! }
99 //!
100 //! // Remove the contained string, destroying the Option
101 //! let unwrapped_msg = msg.unwrap_or("default message");
102 //! ```
103 //!
104 //! Initialize a result to [`None`] before a loop:
105 //!
106 //! ```
107 //! enum Kingdom { Plant(u32, &'static str), Animal(u32, &'static str) }
108 //!
109 //! // A list of data to search through.
110 //! let all_the_big_things = [
111 //!     Kingdom::Plant(250, "redwood"),
112 //!     Kingdom::Plant(230, "noble fir"),
113 //!     Kingdom::Plant(229, "sugar pine"),
114 //!     Kingdom::Animal(25, "blue whale"),
115 //!     Kingdom::Animal(19, "fin whale"),
116 //!     Kingdom::Animal(15, "north pacific right whale"),
117 //! ];
118 //!
119 //! // We're going to search for the name of the biggest animal,
120 //! // but to start with we've just got `None`.
121 //! let mut name_of_biggest_animal = None;
122 //! let mut size_of_biggest_animal = 0;
123 //! for big_thing in &all_the_big_things {
124 //!     match *big_thing {
125 //!         Kingdom::Animal(size, name) if size > size_of_biggest_animal => {
126 //!             // Now we've found the name of some big animal
127 //!             size_of_biggest_animal = size;
128 //!             name_of_biggest_animal = Some(name);
129 //!         }
130 //!         Kingdom::Animal(..) | Kingdom::Plant(..) => ()
131 //!     }
132 //! }
133 //!
134 //! match name_of_biggest_animal {
135 //!     Some(name) => println!("the biggest animal is {}", name),
136 //!     None => println!("there are no animals :("),
137 //! }
138 //! ```
139 //!
140 //! [`Option`]: enum.Option.html
141 //! [`Some`]: enum.Option.html#variant.Some
142 //! [`None`]: enum.Option.html#variant.None
143 //! [`Box<T>`]: ../../std/boxed/struct.Box.html
144 //! [`i32`]: ../../std/primitive.i32.html
145
146 #![stable(feature = "rust1", since = "1.0.0")]
147
148 use iter::{FromIterator, FusedIterator, TrustedLen};
149 use {mem, ops};
150
151 // Note that this is not a lang item per se, but it has a hidden dependency on
152 // `Iterator`, which is one. The compiler assumes that the `next` method of
153 // `Iterator` is an enumeration with one type parameter and two variants,
154 // which basically means it must be `Option`.
155
156 /// The `Option` type. See [the module level documentation](index.html) for more.
157 #[derive(Clone, Copy, PartialEq, PartialOrd, Eq, Ord, Debug, Hash)]
158 #[stable(feature = "rust1", since = "1.0.0")]
159 pub enum Option<T> {
160     /// No value
161     #[stable(feature = "rust1", since = "1.0.0")]
162     None,
163     /// Some value `T`
164     #[stable(feature = "rust1", since = "1.0.0")]
165     Some(#[stable(feature = "rust1", since = "1.0.0")] T),
166 }
167
168 /////////////////////////////////////////////////////////////////////////////
169 // Type implementation
170 /////////////////////////////////////////////////////////////////////////////
171
172 impl<T> Option<T> {
173     /////////////////////////////////////////////////////////////////////////
174     // Querying the contained values
175     /////////////////////////////////////////////////////////////////////////
176
177     /// Returns `true` if the option is a [`Some`] value.
178     ///
179     /// # Examples
180     ///
181     /// ```
182     /// let x: Option<u32> = Some(2);
183     /// assert_eq!(x.is_some(), true);
184     ///
185     /// let x: Option<u32> = None;
186     /// assert_eq!(x.is_some(), false);
187     /// ```
188     ///
189     /// [`Some`]: #variant.Some
190     #[inline]
191     #[stable(feature = "rust1", since = "1.0.0")]
192     pub fn is_some(&self) -> bool {
193         match *self {
194             Some(_) => true,
195             None => false,
196         }
197     }
198
199     /// Returns `true` if the option is a [`None`] value.
200     ///
201     /// # Examples
202     ///
203     /// ```
204     /// let x: Option<u32> = Some(2);
205     /// assert_eq!(x.is_none(), false);
206     ///
207     /// let x: Option<u32> = None;
208     /// assert_eq!(x.is_none(), true);
209     /// ```
210     ///
211     /// [`None`]: #variant.None
212     #[inline]
213     #[stable(feature = "rust1", since = "1.0.0")]
214     pub fn is_none(&self) -> bool {
215         !self.is_some()
216     }
217
218     /////////////////////////////////////////////////////////////////////////
219     // Adapter for working with references
220     /////////////////////////////////////////////////////////////////////////
221
222     /// Converts from `Option<T>` to `Option<&T>`.
223     ///
224     /// # Examples
225     ///
226     /// Convert an `Option<`[`String`]`>` into an `Option<`[`usize`]`>`, preserving the original.
227     /// The [`map`] method takes the `self` argument by value, consuming the original,
228     /// so this technique uses `as_ref` to first take an `Option` to a reference
229     /// to the value inside the original.
230     ///
231     /// [`map`]: enum.Option.html#method.map
232     /// [`String`]: ../../std/string/struct.String.html
233     /// [`usize`]: ../../std/primitive.usize.html
234     ///
235     /// ```
236     /// let num_as_str: Option<String> = Some("10".to_string());
237     /// // First, cast `Option<String>` to `Option<&String>` with `as_ref`,
238     /// // then consume *that* with `map`, leaving `num_as_str` on the stack.
239     /// let num_as_int: Option<usize> = num_as_str.as_ref().map(|n| n.len());
240     /// println!("still can print num_as_str: {:?}", num_as_str);
241     /// ```
242     #[inline]
243     #[stable(feature = "rust1", since = "1.0.0")]
244     pub fn as_ref(&self) -> Option<&T> {
245         match *self {
246             Some(ref x) => Some(x),
247             None => None,
248         }
249     }
250
251     /// Converts from `Option<T>` to `Option<&mut T>`.
252     ///
253     /// # Examples
254     ///
255     /// ```
256     /// let mut x = Some(2);
257     /// match x.as_mut() {
258     ///     Some(v) => *v = 42,
259     ///     None => {},
260     /// }
261     /// assert_eq!(x, Some(42));
262     /// ```
263     #[inline]
264     #[stable(feature = "rust1", since = "1.0.0")]
265     pub fn as_mut(&mut self) -> Option<&mut T> {
266         match *self {
267             Some(ref mut x) => Some(x),
268             None => None,
269         }
270     }
271
272     /////////////////////////////////////////////////////////////////////////
273     // Getting to contained values
274     /////////////////////////////////////////////////////////////////////////
275
276     /// Unwraps an option, yielding the content of a [`Some`].
277     ///
278     /// # Panics
279     ///
280     /// Panics if the value is a [`None`] with a custom panic message provided by
281     /// `msg`.
282     ///
283     /// [`Some`]: #variant.Some
284     /// [`None`]: #variant.None
285     ///
286     /// # Examples
287     ///
288     /// ```
289     /// let x = Some("value");
290     /// assert_eq!(x.expect("the world is ending"), "value");
291     /// ```
292     ///
293     /// ```{.should_panic}
294     /// let x: Option<&str> = None;
295     /// x.expect("the world is ending"); // panics with `the world is ending`
296     /// ```
297     #[inline]
298     #[stable(feature = "rust1", since = "1.0.0")]
299     pub fn expect(self, msg: &str) -> T {
300         match self {
301             Some(val) => val,
302             None => expect_failed(msg),
303         }
304     }
305
306     /// Moves the value `v` out of the `Option<T>` if it is [`Some(v)`].
307     ///
308     /// In general, because this function may panic, its use is discouraged.
309     /// Instead, prefer to use pattern matching and handle the [`None`]
310     /// case explicitly.
311     ///
312     /// # Panics
313     ///
314     /// Panics if the self value equals [`None`].
315     ///
316     /// [`Some(v)`]: #variant.Some
317     /// [`None`]: #variant.None
318     ///
319     /// # Examples
320     ///
321     /// ```
322     /// let x = Some("air");
323     /// assert_eq!(x.unwrap(), "air");
324     /// ```
325     ///
326     /// ```{.should_panic}
327     /// let x: Option<&str> = None;
328     /// assert_eq!(x.unwrap(), "air"); // fails
329     /// ```
330     #[inline]
331     #[stable(feature = "rust1", since = "1.0.0")]
332     pub fn unwrap(self) -> T {
333         match self {
334             Some(val) => val,
335             None => panic!("called `Option::unwrap()` on a `None` value"),
336         }
337     }
338
339     /// Returns the contained value or a default.
340     ///
341     /// # Examples
342     ///
343     /// ```
344     /// assert_eq!(Some("car").unwrap_or("bike"), "car");
345     /// assert_eq!(None.unwrap_or("bike"), "bike");
346     /// ```
347     #[inline]
348     #[stable(feature = "rust1", since = "1.0.0")]
349     pub fn unwrap_or(self, def: T) -> T {
350         match self {
351             Some(x) => x,
352             None => def,
353         }
354     }
355
356     /// Returns the contained value or computes it from a closure.
357     ///
358     /// # Examples
359     ///
360     /// ```
361     /// let k = 10;
362     /// assert_eq!(Some(4).unwrap_or_else(|| 2 * k), 4);
363     /// assert_eq!(None.unwrap_or_else(|| 2 * k), 20);
364     /// ```
365     #[inline]
366     #[stable(feature = "rust1", since = "1.0.0")]
367     pub fn unwrap_or_else<F: FnOnce() -> T>(self, f: F) -> T {
368         match self {
369             Some(x) => x,
370             None => f(),
371         }
372     }
373
374     /////////////////////////////////////////////////////////////////////////
375     // Transforming contained values
376     /////////////////////////////////////////////////////////////////////////
377
378     /// Maps an `Option<T>` to `Option<U>` by applying a function to a contained value.
379     ///
380     /// # Examples
381     ///
382     /// Convert an `Option<`[`String`]`>` into an `Option<`[`usize`]`>`, consuming the original:
383     ///
384     /// [`String`]: ../../std/string/struct.String.html
385     /// [`usize`]: ../../std/primitive.usize.html
386     ///
387     /// ```
388     /// let maybe_some_string = Some(String::from("Hello, World!"));
389     /// // `Option::map` takes self *by value*, consuming `maybe_some_string`
390     /// let maybe_some_len = maybe_some_string.map(|s| s.len());
391     ///
392     /// assert_eq!(maybe_some_len, Some(13));
393     /// ```
394     #[inline]
395     #[stable(feature = "rust1", since = "1.0.0")]
396     pub fn map<U, F: FnOnce(T) -> U>(self, f: F) -> Option<U> {
397         match self {
398             Some(x) => Some(f(x)),
399             None => None,
400         }
401     }
402
403     /// Applies a function to the contained value (if any),
404     /// or returns a [`default`][] (if not).
405     ///
406     /// [`default`]: ../default/trait.Default.html#tymethod.default
407     ///
408     /// # Examples
409     ///
410     /// ```
411     /// let x = Some("foo");
412     /// assert_eq!(x.map_or(42, |v| v.len()), 3);
413     ///
414     /// let x: Option<&str> = None;
415     /// assert_eq!(x.map_or(42, |v| v.len()), 42);
416     /// ```
417     #[inline]
418     #[stable(feature = "rust1", since = "1.0.0")]
419     pub fn map_or<U, F: FnOnce(T) -> U>(self, default: U, f: F) -> U {
420         match self {
421             Some(t) => f(t),
422             None => default,
423         }
424     }
425
426     /// Applies a function to the contained value (if any),
427     /// or computes a [`default`][] (if not).
428     ///
429     /// [`default`]: ../default/trait.Default.html#tymethod.default
430     ///
431     /// # Examples
432     ///
433     /// ```
434     /// let k = 21;
435     ///
436     /// let x = Some("foo");
437     /// assert_eq!(x.map_or_else(|| 2 * k, |v| v.len()), 3);
438     ///
439     /// let x: Option<&str> = None;
440     /// assert_eq!(x.map_or_else(|| 2 * k, |v| v.len()), 42);
441     /// ```
442     #[inline]
443     #[stable(feature = "rust1", since = "1.0.0")]
444     pub fn map_or_else<U, D: FnOnce() -> U, F: FnOnce(T) -> U>(self, default: D, f: F) -> U {
445         match self {
446             Some(t) => f(t),
447             None => default(),
448         }
449     }
450
451     /// Transforms the `Option<T>` into a [`Result<T, E>`], mapping [`Some(v)`] to
452     /// [`Ok(v)`] and [`None`] to [`Err(err)`].
453     ///
454     /// [`Result<T, E>`]: ../../std/result/enum.Result.html
455     /// [`Ok(v)`]: ../../std/result/enum.Result.html#variant.Ok
456     /// [`Err(err)`]: ../../std/result/enum.Result.html#variant.Err
457     /// [`None`]: #variant.None
458     /// [`Some(v)`]: #variant.Some
459     ///
460     /// # Examples
461     ///
462     /// ```
463     /// let x = Some("foo");
464     /// assert_eq!(x.ok_or(0), Ok("foo"));
465     ///
466     /// let x: Option<&str> = None;
467     /// assert_eq!(x.ok_or(0), Err(0));
468     /// ```
469     #[inline]
470     #[stable(feature = "rust1", since = "1.0.0")]
471     pub fn ok_or<E>(self, err: E) -> Result<T, E> {
472         match self {
473             Some(v) => Ok(v),
474             None => Err(err),
475         }
476     }
477
478     /// Transforms the `Option<T>` into a [`Result<T, E>`], mapping [`Some(v)`] to
479     /// [`Ok(v)`] and [`None`] to [`Err(err())`].
480     ///
481     /// [`Result<T, E>`]: ../../std/result/enum.Result.html
482     /// [`Ok(v)`]: ../../std/result/enum.Result.html#variant.Ok
483     /// [`Err(err())`]: ../../std/result/enum.Result.html#variant.Err
484     /// [`None`]: #variant.None
485     /// [`Some(v)`]: #variant.Some
486     ///
487     /// # Examples
488     ///
489     /// ```
490     /// let x = Some("foo");
491     /// assert_eq!(x.ok_or_else(|| 0), Ok("foo"));
492     ///
493     /// let x: Option<&str> = None;
494     /// assert_eq!(x.ok_or_else(|| 0), Err(0));
495     /// ```
496     #[inline]
497     #[stable(feature = "rust1", since = "1.0.0")]
498     pub fn ok_or_else<E, F: FnOnce() -> E>(self, err: F) -> Result<T, E> {
499         match self {
500             Some(v) => Ok(v),
501             None => Err(err()),
502         }
503     }
504
505     /////////////////////////////////////////////////////////////////////////
506     // Iterator constructors
507     /////////////////////////////////////////////////////////////////////////
508
509     /// Returns an iterator over the possibly contained value.
510     ///
511     /// # Examples
512     ///
513     /// ```
514     /// let x = Some(4);
515     /// assert_eq!(x.iter().next(), Some(&4));
516     ///
517     /// let x: Option<u32> = None;
518     /// assert_eq!(x.iter().next(), None);
519     /// ```
520     #[inline]
521     #[stable(feature = "rust1", since = "1.0.0")]
522     pub fn iter(&self) -> Iter<T> {
523         Iter { inner: Item { opt: self.as_ref() } }
524     }
525
526     /// Returns a mutable iterator over the possibly contained value.
527     ///
528     /// # Examples
529     ///
530     /// ```
531     /// let mut x = Some(4);
532     /// match x.iter_mut().next() {
533     ///     Some(v) => *v = 42,
534     ///     None => {},
535     /// }
536     /// assert_eq!(x, Some(42));
537     ///
538     /// let mut x: Option<u32> = None;
539     /// assert_eq!(x.iter_mut().next(), None);
540     /// ```
541     #[inline]
542     #[stable(feature = "rust1", since = "1.0.0")]
543     pub fn iter_mut(&mut self) -> IterMut<T> {
544         IterMut { inner: Item { opt: self.as_mut() } }
545     }
546
547     /////////////////////////////////////////////////////////////////////////
548     // Boolean operations on the values, eager and lazy
549     /////////////////////////////////////////////////////////////////////////
550
551     /// Returns [`None`] if the option is [`None`], otherwise returns `optb`.
552     ///
553     /// [`None`]: #variant.None
554     ///
555     /// # Examples
556     ///
557     /// ```
558     /// let x = Some(2);
559     /// let y: Option<&str> = None;
560     /// assert_eq!(x.and(y), None);
561     ///
562     /// let x: Option<u32> = None;
563     /// let y = Some("foo");
564     /// assert_eq!(x.and(y), None);
565     ///
566     /// let x = Some(2);
567     /// let y = Some("foo");
568     /// assert_eq!(x.and(y), Some("foo"));
569     ///
570     /// let x: Option<u32> = None;
571     /// let y: Option<&str> = None;
572     /// assert_eq!(x.and(y), None);
573     /// ```
574     #[inline]
575     #[stable(feature = "rust1", since = "1.0.0")]
576     pub fn and<U>(self, optb: Option<U>) -> Option<U> {
577         match self {
578             Some(_) => optb,
579             None => None,
580         }
581     }
582
583     /// Returns [`None`] if the option is [`None`], otherwise calls `f` with the
584     /// wrapped value and returns the result.
585     ///
586     /// Some languages call this operation flatmap.
587     ///
588     /// [`None`]: #variant.None
589     ///
590     /// # Examples
591     ///
592     /// ```
593     /// fn sq(x: u32) -> Option<u32> { Some(x * x) }
594     /// fn nope(_: u32) -> Option<u32> { None }
595     ///
596     /// assert_eq!(Some(2).and_then(sq).and_then(sq), Some(16));
597     /// assert_eq!(Some(2).and_then(sq).and_then(nope), None);
598     /// assert_eq!(Some(2).and_then(nope).and_then(sq), None);
599     /// assert_eq!(None.and_then(sq).and_then(sq), None);
600     /// ```
601     #[inline]
602     #[stable(feature = "rust1", since = "1.0.0")]
603     pub fn and_then<U, F: FnOnce(T) -> Option<U>>(self, f: F) -> Option<U> {
604         match self {
605             Some(x) => f(x),
606             None => None,
607         }
608     }
609
610     /// Returns `None` if the option is `None`, otherwise calls `predicate`
611     /// with the wrapped value and returns:
612     ///
613     /// - `Some(t)` if `predicate` returns `true` (where `t` is the wrapped
614     ///   value), and
615     /// - `None` if `predicate` returns `false`.
616     ///
617     /// This function works similar to `Iterator::filter()`. You can imagine
618     /// the `Option<T>` being an iterator over one or zero elements. `filter()`
619     /// lets you decide which elements to keep.
620     ///
621     /// # Examples
622     ///
623     /// ```rust
624     /// #![feature(option_filter)]
625     ///
626     /// fn is_even(n: &i32) -> bool {
627     ///     n % 2 == 0
628     /// }
629     ///
630     /// assert_eq!(None.filter(is_even), None);
631     /// assert_eq!(Some(3).filter(is_even), None);
632     /// assert_eq!(Some(4).filter(is_even), Some(4));
633     /// ```
634     #[inline]
635     #[unstable(feature = "option_filter", issue = "45860")]
636     pub fn filter<P: FnOnce(&T) -> bool>(self, predicate: P) -> Self {
637         match self {
638             Some(x) => {
639                 if predicate(&x) {
640                     Some(x)
641                 } else {
642                     None
643                 }
644             }
645             None => None,
646         }
647     }
648
649     /// Returns the option if it contains a value, otherwise returns `optb`.
650     ///
651     /// # Examples
652     ///
653     /// ```
654     /// let x = Some(2);
655     /// let y = None;
656     /// assert_eq!(x.or(y), Some(2));
657     ///
658     /// let x = None;
659     /// let y = Some(100);
660     /// assert_eq!(x.or(y), Some(100));
661     ///
662     /// let x = Some(2);
663     /// let y = Some(100);
664     /// assert_eq!(x.or(y), Some(2));
665     ///
666     /// let x: Option<u32> = None;
667     /// let y = None;
668     /// assert_eq!(x.or(y), None);
669     /// ```
670     #[inline]
671     #[stable(feature = "rust1", since = "1.0.0")]
672     pub fn or(self, optb: Option<T>) -> Option<T> {
673         match self {
674             Some(_) => self,
675             None => optb,
676         }
677     }
678
679     /// Returns the option if it contains a value, otherwise calls `f` and
680     /// returns the result.
681     ///
682     /// # Examples
683     ///
684     /// ```
685     /// fn nobody() -> Option<&'static str> { None }
686     /// fn vikings() -> Option<&'static str> { Some("vikings") }
687     ///
688     /// assert_eq!(Some("barbarians").or_else(vikings), Some("barbarians"));
689     /// assert_eq!(None.or_else(vikings), Some("vikings"));
690     /// assert_eq!(None.or_else(nobody), None);
691     /// ```
692     #[inline]
693     #[stable(feature = "rust1", since = "1.0.0")]
694     pub fn or_else<F: FnOnce() -> Option<T>>(self, f: F) -> Option<T> {
695         match self {
696             Some(_) => self,
697             None => f(),
698         }
699     }
700
701     /////////////////////////////////////////////////////////////////////////
702     // Entry-like operations to insert if None and return a reference
703     /////////////////////////////////////////////////////////////////////////
704
705     /// Inserts `v` into the option if it is [`None`], then
706     /// returns a mutable reference to the contained value.
707     ///
708     /// [`None`]: #variant.None
709     ///
710     /// # Examples
711     ///
712     /// ```
713     /// let mut x = None;
714     ///
715     /// {
716     ///     let y: &mut u32 = x.get_or_insert(5);
717     ///     assert_eq!(y, &5);
718     ///
719     ///     *y = 7;
720     /// }
721     ///
722     /// assert_eq!(x, Some(7));
723     /// ```
724     #[inline]
725     #[stable(feature = "option_entry", since = "1.20.0")]
726     pub fn get_or_insert(&mut self, v: T) -> &mut T {
727         match *self {
728             None => *self = Some(v),
729             _ => (),
730         }
731
732         match *self {
733             Some(ref mut v) => v,
734             _ => unreachable!(),
735         }
736     }
737
738     /// Inserts a value computed from `f` into the option if it is [`None`], then
739     /// returns a mutable reference to the contained value.
740     ///
741     /// [`None`]: #variant.None
742     ///
743     /// # Examples
744     ///
745     /// ```
746     /// let mut x = None;
747     ///
748     /// {
749     ///     let y: &mut u32 = x.get_or_insert_with(|| 5);
750     ///     assert_eq!(y, &5);
751     ///
752     ///     *y = 7;
753     /// }
754     ///
755     /// assert_eq!(x, Some(7));
756     /// ```
757     #[inline]
758     #[stable(feature = "option_entry", since = "1.20.0")]
759     pub fn get_or_insert_with<F: FnOnce() -> T>(&mut self, f: F) -> &mut T {
760         match *self {
761             None => *self = Some(f()),
762             _ => (),
763         }
764
765         match *self {
766             Some(ref mut v) => v,
767             _ => unreachable!(),
768         }
769     }
770
771     /////////////////////////////////////////////////////////////////////////
772     // Misc
773     /////////////////////////////////////////////////////////////////////////
774
775     /// Takes the value out of the option, leaving a [`None`] in its place.
776     ///
777     /// [`None`]: #variant.None
778     ///
779     /// # Examples
780     ///
781     /// ```
782     /// let mut x = Some(2);
783     /// x.take();
784     /// assert_eq!(x, None);
785     ///
786     /// let mut x: Option<u32> = None;
787     /// x.take();
788     /// assert_eq!(x, None);
789     /// ```
790     #[inline]
791     #[stable(feature = "rust1", since = "1.0.0")]
792     pub fn take(&mut self) -> Option<T> {
793         mem::replace(self, None)
794     }
795 }
796
797 impl<'a, T: Clone> Option<&'a T> {
798     /// Maps an `Option<&T>` to an `Option<T>` by cloning the contents of the
799     /// option.
800     ///
801     /// # Examples
802     ///
803     /// ```
804     /// let x = 12;
805     /// let opt_x = Some(&x);
806     /// assert_eq!(opt_x, Some(&12));
807     /// let cloned = opt_x.cloned();
808     /// assert_eq!(cloned, Some(12));
809     /// ```
810     #[stable(feature = "rust1", since = "1.0.0")]
811     pub fn cloned(self) -> Option<T> {
812         self.map(|t| t.clone())
813     }
814 }
815
816 impl<'a, T: Clone> Option<&'a mut T> {
817     /// Maps an `Option<&mut T>` to an `Option<T>` by cloning the contents of the
818     /// option.
819     ///
820     /// # Examples
821     ///
822     /// ```
823     /// #![feature(option_ref_mut_cloned)]
824     /// let mut x = 12;
825     /// let opt_x = Some(&mut x);
826     /// assert_eq!(opt_x, Some(&mut 12));
827     /// let cloned = opt_x.cloned();
828     /// assert_eq!(cloned, Some(12));
829     /// ```
830     #[unstable(feature = "option_ref_mut_cloned", issue = "43738")]
831     pub fn cloned(self) -> Option<T> {
832         self.map(|t| t.clone())
833     }
834 }
835
836 impl<T: Default> Option<T> {
837     /// Returns the contained value or a default
838     ///
839     /// Consumes the `self` argument then, if [`Some`], returns the contained
840     /// value, otherwise if [`None`], returns the default value for that
841     /// type.
842     ///
843     /// # Examples
844     ///
845     /// Convert a string to an integer, turning poorly-formed strings
846     /// into 0 (the default value for integers). [`parse`] converts
847     /// a string to any other type that implements [`FromStr`], returning
848     /// [`None`] on error.
849     ///
850     /// ```
851     /// let good_year_from_input = "1909";
852     /// let bad_year_from_input = "190blarg";
853     /// let good_year = good_year_from_input.parse().ok().unwrap_or_default();
854     /// let bad_year = bad_year_from_input.parse().ok().unwrap_or_default();
855     ///
856     /// assert_eq!(1909, good_year);
857     /// assert_eq!(0, bad_year);
858     /// ```
859     ///
860     /// [`Some`]: #variant.Some
861     /// [`None`]: #variant.None
862     /// [`parse`]: ../../std/primitive.str.html#method.parse
863     /// [`FromStr`]: ../../std/str/trait.FromStr.html
864     #[inline]
865     #[stable(feature = "rust1", since = "1.0.0")]
866     pub fn unwrap_or_default(self) -> T {
867         match self {
868             Some(x) => x,
869             None => Default::default(),
870         }
871     }
872 }
873
874 // This is a separate function to reduce the code size of .expect() itself.
875 #[inline(never)]
876 #[cold]
877 fn expect_failed(msg: &str) -> ! {
878     panic!("{}", msg)
879 }
880
881
882 /////////////////////////////////////////////////////////////////////////////
883 // Trait implementations
884 /////////////////////////////////////////////////////////////////////////////
885
886 #[stable(feature = "rust1", since = "1.0.0")]
887 impl<T> Default for Option<T> {
888     /// Returns [`None`].
889     ///
890     /// [`None`]: #variant.None
891     #[inline]
892     fn default() -> Option<T> { None }
893 }
894
895 #[stable(feature = "rust1", since = "1.0.0")]
896 impl<T> IntoIterator for Option<T> {
897     type Item = T;
898     type IntoIter = IntoIter<T>;
899
900     /// Returns a consuming iterator over the possibly contained value.
901     ///
902     /// # Examples
903     ///
904     /// ```
905     /// let x = Some("string");
906     /// let v: Vec<&str> = x.into_iter().collect();
907     /// assert_eq!(v, ["string"]);
908     ///
909     /// let x = None;
910     /// let v: Vec<&str> = x.into_iter().collect();
911     /// assert!(v.is_empty());
912     /// ```
913     #[inline]
914     fn into_iter(self) -> IntoIter<T> {
915         IntoIter { inner: Item { opt: self } }
916     }
917 }
918
919 #[stable(since = "1.4.0", feature = "option_iter")]
920 impl<'a, T> IntoIterator for &'a Option<T> {
921     type Item = &'a T;
922     type IntoIter = Iter<'a, T>;
923
924     fn into_iter(self) -> Iter<'a, T> {
925         self.iter()
926     }
927 }
928
929 #[stable(since = "1.4.0", feature = "option_iter")]
930 impl<'a, T> IntoIterator for &'a mut Option<T> {
931     type Item = &'a mut T;
932     type IntoIter = IterMut<'a, T>;
933
934     fn into_iter(self) -> IterMut<'a, T> {
935         self.iter_mut()
936     }
937 }
938
939 #[stable(since = "1.12.0", feature = "option_from")]
940 impl<T> From<T> for Option<T> {
941     fn from(val: T) -> Option<T> {
942         Some(val)
943     }
944 }
945
946 /////////////////////////////////////////////////////////////////////////////
947 // The Option Iterators
948 /////////////////////////////////////////////////////////////////////////////
949
950 #[derive(Clone, Debug)]
951 struct Item<A> {
952     opt: Option<A>
953 }
954
955 impl<A> Iterator for Item<A> {
956     type Item = A;
957
958     #[inline]
959     fn next(&mut self) -> Option<A> {
960         self.opt.take()
961     }
962
963     #[inline]
964     fn size_hint(&self) -> (usize, Option<usize>) {
965         match self.opt {
966             Some(_) => (1, Some(1)),
967             None => (0, Some(0)),
968         }
969     }
970 }
971
972 impl<A> DoubleEndedIterator for Item<A> {
973     #[inline]
974     fn next_back(&mut self) -> Option<A> {
975         self.opt.take()
976     }
977 }
978
979 impl<A> ExactSizeIterator for Item<A> {}
980 impl<A> FusedIterator for Item<A> {}
981 unsafe impl<A> TrustedLen for Item<A> {}
982
983 /// An iterator over a reference to the [`Some`] variant of an [`Option`].
984 ///
985 /// The iterator yields one value if the [`Option`] is a [`Some`], otherwise none.
986 ///
987 /// This `struct` is created by the [`Option::iter`] function.
988 ///
989 /// [`Option`]: enum.Option.html
990 /// [`Some`]: enum.Option.html#variant.Some
991 /// [`Option::iter`]: enum.Option.html#method.iter
992 #[stable(feature = "rust1", since = "1.0.0")]
993 #[derive(Debug)]
994 pub struct Iter<'a, A: 'a> { inner: Item<&'a A> }
995
996 #[stable(feature = "rust1", since = "1.0.0")]
997 impl<'a, A> Iterator for Iter<'a, A> {
998     type Item = &'a A;
999
1000     #[inline]
1001     fn next(&mut self) -> Option<&'a A> { self.inner.next() }
1002     #[inline]
1003     fn size_hint(&self) -> (usize, Option<usize>) { self.inner.size_hint() }
1004 }
1005
1006 #[stable(feature = "rust1", since = "1.0.0")]
1007 impl<'a, A> DoubleEndedIterator for Iter<'a, A> {
1008     #[inline]
1009     fn next_back(&mut self) -> Option<&'a A> { self.inner.next_back() }
1010 }
1011
1012 #[stable(feature = "rust1", since = "1.0.0")]
1013 impl<'a, A> ExactSizeIterator for Iter<'a, A> {}
1014
1015 #[unstable(feature = "fused", issue = "35602")]
1016 impl<'a, A> FusedIterator for Iter<'a, A> {}
1017
1018 #[unstable(feature = "trusted_len", issue = "37572")]
1019 unsafe impl<'a, A> TrustedLen for Iter<'a, A> {}
1020
1021 #[stable(feature = "rust1", since = "1.0.0")]
1022 impl<'a, A> Clone for Iter<'a, A> {
1023     fn clone(&self) -> Iter<'a, A> {
1024         Iter { inner: self.inner.clone() }
1025     }
1026 }
1027
1028 /// An iterator over a mutable reference to the [`Some`] variant of an [`Option`].
1029 ///
1030 /// The iterator yields one value if the [`Option`] is a [`Some`], otherwise none.
1031 ///
1032 /// This `struct` is created by the [`Option::iter_mut`] function.
1033 ///
1034 /// [`Option`]: enum.Option.html
1035 /// [`Some`]: enum.Option.html#variant.Some
1036 /// [`Option::iter_mut`]: enum.Option.html#method.iter_mut
1037 #[stable(feature = "rust1", since = "1.0.0")]
1038 #[derive(Debug)]
1039 pub struct IterMut<'a, A: 'a> { inner: Item<&'a mut A> }
1040
1041 #[stable(feature = "rust1", since = "1.0.0")]
1042 impl<'a, A> Iterator for IterMut<'a, A> {
1043     type Item = &'a mut A;
1044
1045     #[inline]
1046     fn next(&mut self) -> Option<&'a mut A> { self.inner.next() }
1047     #[inline]
1048     fn size_hint(&self) -> (usize, Option<usize>) { self.inner.size_hint() }
1049 }
1050
1051 #[stable(feature = "rust1", since = "1.0.0")]
1052 impl<'a, A> DoubleEndedIterator for IterMut<'a, A> {
1053     #[inline]
1054     fn next_back(&mut self) -> Option<&'a mut A> { self.inner.next_back() }
1055 }
1056
1057 #[stable(feature = "rust1", since = "1.0.0")]
1058 impl<'a, A> ExactSizeIterator for IterMut<'a, A> {}
1059
1060 #[unstable(feature = "fused", issue = "35602")]
1061 impl<'a, A> FusedIterator for IterMut<'a, A> {}
1062 #[unstable(feature = "trusted_len", issue = "37572")]
1063 unsafe impl<'a, A> TrustedLen for IterMut<'a, A> {}
1064
1065 /// An iterator over the value in [`Some`] variant of an [`Option`].
1066 ///
1067 /// The iterator yields one value if the [`Option`] is a [`Some`], otherwise none.
1068 ///
1069 /// This `struct` is created by the [`Option::into_iter`] function.
1070 ///
1071 /// [`Option`]: enum.Option.html
1072 /// [`Some`]: enum.Option.html#variant.Some
1073 /// [`Option::into_iter`]: enum.Option.html#method.into_iter
1074 #[derive(Clone, Debug)]
1075 #[stable(feature = "rust1", since = "1.0.0")]
1076 pub struct IntoIter<A> { inner: Item<A> }
1077
1078 #[stable(feature = "rust1", since = "1.0.0")]
1079 impl<A> Iterator for IntoIter<A> {
1080     type Item = A;
1081
1082     #[inline]
1083     fn next(&mut self) -> Option<A> { self.inner.next() }
1084     #[inline]
1085     fn size_hint(&self) -> (usize, Option<usize>) { self.inner.size_hint() }
1086 }
1087
1088 #[stable(feature = "rust1", since = "1.0.0")]
1089 impl<A> DoubleEndedIterator for IntoIter<A> {
1090     #[inline]
1091     fn next_back(&mut self) -> Option<A> { self.inner.next_back() }
1092 }
1093
1094 #[stable(feature = "rust1", since = "1.0.0")]
1095 impl<A> ExactSizeIterator for IntoIter<A> {}
1096
1097 #[unstable(feature = "fused", issue = "35602")]
1098 impl<A> FusedIterator for IntoIter<A> {}
1099
1100 #[unstable(feature = "trusted_len", issue = "37572")]
1101 unsafe impl<A> TrustedLen for IntoIter<A> {}
1102
1103 /////////////////////////////////////////////////////////////////////////////
1104 // FromIterator
1105 /////////////////////////////////////////////////////////////////////////////
1106
1107 #[stable(feature = "rust1", since = "1.0.0")]
1108 impl<A, V: FromIterator<A>> FromIterator<Option<A>> for Option<V> {
1109     /// Takes each element in the [`Iterator`]: if it is [`None`], no further
1110     /// elements are taken, and the [`None`] is returned. Should no [`None`] occur, a
1111     /// container with the values of each `Option` is returned.
1112     ///
1113     /// Here is an example which increments every integer in a vector,
1114     /// checking for overflow:
1115     ///
1116     /// ```
1117     /// use std::u16;
1118     ///
1119     /// let v = vec![1, 2];
1120     /// let res: Option<Vec<u16>> = v.iter().map(|&x: &u16|
1121     ///     if x == u16::MAX { None }
1122     ///     else { Some(x + 1) }
1123     /// ).collect();
1124     /// assert!(res == Some(vec![2, 3]));
1125     /// ```
1126     ///
1127     /// [`Iterator`]: ../iter/trait.Iterator.html
1128     /// [`None`]: enum.Option.html#variant.None
1129     #[inline]
1130     fn from_iter<I: IntoIterator<Item=Option<A>>>(iter: I) -> Option<V> {
1131         // FIXME(#11084): This could be replaced with Iterator::scan when this
1132         // performance bug is closed.
1133
1134         struct Adapter<Iter> {
1135             iter: Iter,
1136             found_none: bool,
1137         }
1138
1139         impl<T, Iter: Iterator<Item=Option<T>>> Iterator for Adapter<Iter> {
1140             type Item = T;
1141
1142             #[inline]
1143             fn next(&mut self) -> Option<T> {
1144                 match self.iter.next() {
1145                     Some(Some(value)) => Some(value),
1146                     Some(None) => {
1147                         self.found_none = true;
1148                         None
1149                     }
1150                     None => None,
1151                 }
1152             }
1153         }
1154
1155         let mut adapter = Adapter { iter: iter.into_iter(), found_none: false };
1156         let v: V = FromIterator::from_iter(adapter.by_ref());
1157
1158         if adapter.found_none {
1159             None
1160         } else {
1161             Some(v)
1162         }
1163     }
1164 }
1165
1166 /// The error type that results from applying the try operator (`?`) to a `None` value. If you wish
1167 /// to allow `x?` (where `x` is an `Option<T>`) to be converted into your error type, you can
1168 /// implement `impl From<NoneError>` for `YourErrorType`. In that case, `x?` within a function that
1169 /// returns `Result<_, YourErrorType>` will translate a `None` value into an `Err` result.
1170 #[unstable(feature = "try_trait", issue = "42327")]
1171 #[derive(Clone, Copy, PartialEq, PartialOrd, Eq, Ord, Debug, Hash)]
1172 pub struct NoneError;
1173
1174 #[unstable(feature = "try_trait", issue = "42327")]
1175 impl<T> ops::Try for Option<T> {
1176     type Ok = T;
1177     type Error = NoneError;
1178
1179     fn into_result(self) -> Result<T, NoneError> {
1180         self.ok_or(NoneError)
1181     }
1182
1183     fn from_ok(v: T) -> Self {
1184         Some(v)
1185     }
1186
1187     fn from_error(_: NoneError) -> Self {
1188         None
1189     }
1190 }