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;
 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 the option if it contains a value, otherwise returns `optb`.
 611     ///
 612     /// # Examples
 613     ///
 614     /// ```
 615     /// let x = Some(2);
 616     /// let y = None;
 617     /// assert_eq!(x.or(y), Some(2));
 618     ///
 619     /// let x = None;
 620     /// let y = Some(100);
 621     /// assert_eq!(x.or(y), Some(100));
 622     ///
 623     /// let x = Some(2);
 624     /// let y = Some(100);
 625     /// assert_eq!(x.or(y), Some(2));
 626     ///
 627     /// let x: Option<u32> = None;
 628     /// let y = None;
 629     /// assert_eq!(x.or(y), None);
 630     /// ```
 631     #[inline]
 632     #[stable(feature = "rust1", since = "1.0.0")]
 633     pub fn or(self, optb: Option<T>) -> Option<T> {
 634         match self {
 635             Some(_) => self,
 636             None => optb,
 637         }
 638     }
 639
 640     /// Returns the option if it contains a value, otherwise calls `f` and
 641     /// returns the result.
 642     ///
 643     /// # Examples
 644     ///
 645     /// ```
 646     /// fn nobody() -> Option<&'static str> { None }
 647     /// fn vikings() -> Option<&'static str> { Some("vikings") }
 648     ///
 649     /// assert_eq!(Some("barbarians").or_else(vikings), Some("barbarians"));
 650     /// assert_eq!(None.or_else(vikings), Some("vikings"));
 651     /// assert_eq!(None.or_else(nobody), None);
 652     /// ```
 653     #[inline]
 654     #[stable(feature = "rust1", since = "1.0.0")]
 655     pub fn or_else<F: FnOnce() -> Option<T>>(self, f: F) -> Option<T> {
 656         match self {
 657             Some(_) => self,
 658             None => f(),
 659         }
 660     }
 661
 662     /////////////////////////////////////////////////////////////////////////
 663     // Entry-like operations to insert if None and return a reference
 664     /////////////////////////////////////////////////////////////////////////
 665
 666     /// Inserts `v` into the option if it is [`None`], then
 667     /// returns a mutable reference to the contained value.
 668     ///
 669     /// [`None`]: #variant.None
 670     ///
 671     /// # Examples
 672     ///
 673     /// ```
 674     /// let mut x = None;
 675     ///
 676     /// {
 677     ///     let y: &mut u32 = x.get_or_insert(5);
 678     ///     assert_eq!(y, &5);
 679     ///
 680     ///     *y = 7;
 681     /// }
 682     ///
 683     /// assert_eq!(x, Some(7));
 684     /// ```
 685     #[inline]
 686     #[stable(feature = "option_entry", since = "1.20.0")]
 687     pub fn get_or_insert(&mut self, v: T) -> &mut T {
 688         match *self {
 689             None => *self = Some(v),
 690             _ => (),
 691         }
 692
 693         match *self {
 694             Some(ref mut v) => v,
 695             _ => unreachable!(),
 696         }
 697     }
 698
 699     /// Inserts a value computed from `f` into the option if it is [`None`], then
 700     /// returns a mutable reference to the contained value.
 701     ///
 702     /// [`None`]: #variant.None
 703     ///
 704     /// # Examples
 705     ///
 706     /// ```
 707     /// let mut x = None;
 708     ///
 709     /// {
 710     ///     let y: &mut u32 = x.get_or_insert_with(|| 5);
 711     ///     assert_eq!(y, &5);
 712     ///
 713     ///     *y = 7;
 714     /// }
 715     ///
 716     /// assert_eq!(x, Some(7));
 717     /// ```
 718     #[inline]
 719     #[stable(feature = "option_entry", since = "1.20.0")]
 720     pub fn get_or_insert_with<F: FnOnce() -> T>(&mut self, f: F) -> &mut T {
 721         match *self {
 722             None => *self = Some(f()),
 723             _ => (),
 724         }
 725
 726         match *self {
 727             Some(ref mut v) => v,
 728             _ => unreachable!(),
 729         }
 730     }
 731
 732     /////////////////////////////////////////////////////////////////////////
 733     // Misc
 734     /////////////////////////////////////////////////////////////////////////
 735
 736     /// Takes the value out of the option, leaving a [`None`] in its place.
 737     ///
 738     /// [`None`]: #variant.None
 739     ///
 740     /// # Examples
 741     ///
 742     /// ```
 743     /// let mut x = Some(2);
 744     /// x.take();
 745     /// assert_eq!(x, None);
 746     ///
 747     /// let mut x: Option<u32> = None;
 748     /// x.take();
 749     /// assert_eq!(x, None);
 750     /// ```
 751     #[inline]
 752     #[stable(feature = "rust1", since = "1.0.0")]
 753     pub fn take(&mut self) -> Option<T> {
 754         mem::replace(self, None)
 755     }
 756 }
 757
 758 impl<'a, T: Clone> Option<&'a T> {
 759     /// Maps an `Option<&T>` to an `Option<T>` by cloning the contents of the
 760     /// option.
 761     ///
 762     /// # Examples
 763     ///
 764     /// ```
 765     /// let x = 12;
 766     /// let opt_x = Some(&x);
 767     /// assert_eq!(opt_x, Some(&12));
 768     /// let cloned = opt_x.cloned();
 769     /// assert_eq!(cloned, Some(12));
 770     /// ```
 771     #[stable(feature = "rust1", since = "1.0.0")]
 772     pub fn cloned(self) -> Option<T> {
 773         self.map(|t| t.clone())
 774     }
 775 }
 776
 777 impl<T: Default> Option<T> {
 778     /// Returns the contained value or a default
 779     ///
 780     /// Consumes the `self` argument then, if [`Some`], returns the contained
 781     /// value, otherwise if [`None`], returns the default value for that
 782     /// type.
 783     ///
 784     /// # Examples
 785     ///
 786     /// Convert a string to an integer, turning poorly-formed strings
 787     /// into 0 (the default value for integers). [`parse`] converts
 788     /// a string to any other type that implements [`FromStr`], returning
 789     /// [`None`] on error.
 790     ///
 791     /// ```
 792     /// let good_year_from_input = "1909";
 793     /// let bad_year_from_input = "190blarg";
 794     /// let good_year = good_year_from_input.parse().ok().unwrap_or_default();
 795     /// let bad_year = bad_year_from_input.parse().ok().unwrap_or_default();
 796     ///
 797     /// assert_eq!(1909, good_year);
 798     /// assert_eq!(0, bad_year);
 799     /// ```
 800     ///
 801     /// [`Some`]: #variant.Some
 802     /// [`None`]: #variant.None
 803     /// [`parse`]: ../../std/primitive.str.html#method.parse
 804     /// [`FromStr`]: ../../std/str/trait.FromStr.html
 805     #[inline]
 806     #[stable(feature = "rust1", since = "1.0.0")]
 807     pub fn unwrap_or_default(self) -> T {
 808         match self {
 809             Some(x) => x,
 810             None => Default::default(),
 811         }
 812     }
 813 }
 814
 815 // This is a separate function to reduce the code size of .expect() itself.
 816 #[inline(never)]
 817 #[cold]
 818 fn expect_failed(msg: &str) -> ! {
 819     panic!("{}", msg)
 820 }
 821
 822
 823 /////////////////////////////////////////////////////////////////////////////
 824 // Trait implementations
 825 /////////////////////////////////////////////////////////////////////////////
 826
 827 #[stable(feature = "rust1", since = "1.0.0")]
 828 impl<T> Default for Option<T> {
 829     /// Returns [`None`].
 830     ///
 831     /// [`None`]: #variant.None
 832     #[inline]
 833     fn default() -> Option<T> { None }
 834 }
 835
 836 #[stable(feature = "rust1", since = "1.0.0")]
 837 impl<T> IntoIterator for Option<T> {
 838     type Item = T;
 839     type IntoIter = IntoIter<T>;
 840
 841     /// Returns a consuming iterator over the possibly contained value.
 842     ///
 843     /// # Examples
 844     ///
 845     /// ```
 846     /// let x = Some("string");
 847     /// let v: Vec<&str> = x.into_iter().collect();
 848     /// assert_eq!(v, ["string"]);
 849     ///
 850     /// let x = None;
 851     /// let v: Vec<&str> = x.into_iter().collect();
 852     /// assert!(v.is_empty());
 853     /// ```
 854     #[inline]
 855     fn into_iter(self) -> IntoIter<T> {
 856         IntoIter { inner: Item { opt: self } }
 857     }
 858 }
 859
 860 #[stable(since = "1.4.0", feature = "option_iter")]
 861 impl<'a, T> IntoIterator for &'a Option<T> {
 862     type Item = &'a T;
 863     type IntoIter = Iter<'a, T>;
 864
 865     fn into_iter(self) -> Iter<'a, T> {
 866         self.iter()
 867     }
 868 }
 869
 870 #[stable(since = "1.4.0", feature = "option_iter")]
 871 impl<'a, T> IntoIterator for &'a mut Option<T> {
 872     type Item = &'a mut T;
 873     type IntoIter = IterMut<'a, T>;
 874
 875     fn into_iter(mut self) -> IterMut<'a, T> {
 876         self.iter_mut()
 877     }
 878 }
 879
 880 #[stable(since = "1.12.0", feature = "option_from")]
 881 impl<T> From<T> for Option<T> {
 882     fn from(val: T) -> Option<T> {
 883         Some(val)
 884     }
 885 }
 886
 887 /////////////////////////////////////////////////////////////////////////////
 888 // The Option Iterators
 889 /////////////////////////////////////////////////////////////////////////////
 890
 891 #[derive(Clone, Debug)]
 892 struct Item<A> {
 893     opt: Option<A>
 894 }
 895
 896 impl<A> Iterator for Item<A> {
 897     type Item = A;
 898
 899     #[inline]
 900     fn next(&mut self) -> Option<A> {
 901         self.opt.take()
 902     }
 903
 904     #[inline]
 905     fn size_hint(&self) -> (usize, Option<usize>) {
 906         match self.opt {
 907             Some(_) => (1, Some(1)),
 908             None => (0, Some(0)),
 909         }
 910     }
 911 }
 912
 913 impl<A> DoubleEndedIterator for Item<A> {
 914     #[inline]
 915     fn next_back(&mut self) -> Option<A> {
 916         self.opt.take()
 917     }
 918 }
 919
 920 impl<A> ExactSizeIterator for Item<A> {}
 921 impl<A> FusedIterator for Item<A> {}
 922 unsafe impl<A> TrustedLen for Item<A> {}
 923
 924 /// An iterator over a reference to the [`Some`] variant of an [`Option`].
 925 ///
 926 /// The iterator yields one value if the [`Option`] is a [`Some`], otherwise none.
 927 ///
 928 /// This `struct` is created by the [`Option::iter`] function.
 929 ///
 930 /// [`Option`]: enum.Option.html
 931 /// [`Some`]: enum.Option.html#variant.Some
 932 /// [`Option::iter`]: enum.Option.html#method.iter
 933 #[stable(feature = "rust1", since = "1.0.0")]
 934 #[derive(Debug)]
 935 pub struct Iter<'a, A: 'a> { inner: Item<&'a A> }
 936
 937 #[stable(feature = "rust1", since = "1.0.0")]
 938 impl<'a, A> Iterator for Iter<'a, A> {
 939     type Item = &'a A;
 940
 941     #[inline]
 942     fn next(&mut self) -> Option<&'a A> { self.inner.next() }
 943     #[inline]
 944     fn size_hint(&self) -> (usize, Option<usize>) { self.inner.size_hint() }
 945 }
 946
 947 #[stable(feature = "rust1", since = "1.0.0")]
 948 impl<'a, A> DoubleEndedIterator for Iter<'a, A> {
 949     #[inline]
 950     fn next_back(&mut self) -> Option<&'a A> { self.inner.next_back() }
 951 }
 952
 953 #[stable(feature = "rust1", since = "1.0.0")]
 954 impl<'a, A> ExactSizeIterator for Iter<'a, A> {}
 955
 956 #[unstable(feature = "fused", issue = "35602")]
 957 impl<'a, A> FusedIterator for Iter<'a, A> {}
 958
 959 #[unstable(feature = "trusted_len", issue = "37572")]
 960 unsafe impl<'a, A> TrustedLen for Iter<'a, A> {}
 961
 962 #[stable(feature = "rust1", since = "1.0.0")]
 963 impl<'a, A> Clone for Iter<'a, A> {
 964     fn clone(&self) -> Iter<'a, A> {
 965         Iter { inner: self.inner.clone() }
 966     }
 967 }
 968
 969 /// An iterator over a mutable reference to the [`Some`] variant of an [`Option`].
 970 ///
 971 /// The iterator yields one value if the [`Option`] is a [`Some`], otherwise none.
 972 ///
 973 /// This `struct` is created by the [`Option::iter_mut`] function.
 974 ///
 975 /// [`Option`]: enum.Option.html
 976 /// [`Some`]: enum.Option.html#variant.Some
 977 /// [`Option::iter_mut`]: enum.Option.html#method.iter_mut
 978 #[stable(feature = "rust1", since = "1.0.0")]
 979 #[derive(Debug)]
 980 pub struct IterMut<'a, A: 'a> { inner: Item<&'a mut A> }
 981
 982 #[stable(feature = "rust1", since = "1.0.0")]
 983 impl<'a, A> Iterator for IterMut<'a, A> {
 984     type Item = &'a mut A;
 985
 986     #[inline]
 987     fn next(&mut self) -> Option<&'a mut A> { self.inner.next() }
 988     #[inline]
 989     fn size_hint(&self) -> (usize, Option<usize>) { self.inner.size_hint() }
 990 }
 991
 992 #[stable(feature = "rust1", since = "1.0.0")]
 993 impl<'a, A> DoubleEndedIterator for IterMut<'a, A> {
 994     #[inline]
 995     fn next_back(&mut self) -> Option<&'a mut A> { self.inner.next_back() }
 996 }
 997
 998 #[stable(feature = "rust1", since = "1.0.0")]
 999 impl<'a, A> ExactSizeIterator for IterMut<'a, A> {}
1000
1001 #[unstable(feature = "fused", issue = "35602")]
1002 impl<'a, A> FusedIterator for IterMut<'a, A> {}
1003 #[unstable(feature = "trusted_len", issue = "37572")]
1004 unsafe impl<'a, A> TrustedLen for IterMut<'a, A> {}
1005
1006 /// An iterator over the value in [`Some`] variant of an [`Option`].
1007 ///
1008 /// The iterator yields one value if the [`Option`] is a [`Some`], otherwise none.
1009 ///
1010 /// This `struct` is created by the [`Option::into_iter`] function.
1011 ///
1012 /// [`Option`]: enum.Option.html
1013 /// [`Some`]: enum.Option.html#variant.Some
1014 /// [`Option::into_iter`]: enum.Option.html#method.into_iter
1015 #[derive(Clone, Debug)]
1016 #[stable(feature = "rust1", since = "1.0.0")]
1017 pub struct IntoIter<A> { inner: Item<A> }
1018
1019 #[stable(feature = "rust1", since = "1.0.0")]
1020 impl<A> Iterator for IntoIter<A> {
1021     type Item = A;
1022
1023     #[inline]
1024     fn next(&mut self) -> Option<A> { self.inner.next() }
1025     #[inline]
1026     fn size_hint(&self) -> (usize, Option<usize>) { self.inner.size_hint() }
1027 }
1028
1029 #[stable(feature = "rust1", since = "1.0.0")]
1030 impl<A> DoubleEndedIterator for IntoIter<A> {
1031     #[inline]
1032     fn next_back(&mut self) -> Option<A> { self.inner.next_back() }
1033 }
1034
1035 #[stable(feature = "rust1", since = "1.0.0")]
1036 impl<A> ExactSizeIterator for IntoIter<A> {}
1037
1038 #[unstable(feature = "fused", issue = "35602")]
1039 impl<A> FusedIterator for IntoIter<A> {}
1040
1041 #[unstable(feature = "trusted_len", issue = "37572")]
1042 unsafe impl<A> TrustedLen for IntoIter<A> {}
1043
1044 /////////////////////////////////////////////////////////////////////////////
1045 // FromIterator
1046 /////////////////////////////////////////////////////////////////////////////
1047
1048 #[stable(feature = "rust1", since = "1.0.0")]
1049 impl<A, V: FromIterator<A>> FromIterator<Option<A>> for Option<V> {
1050     /// Takes each element in the [`Iterator`]: if it is [`None`], no further
1051     /// elements are taken, and the [`None`] is returned. Should no [`None`] occur, a
1052     /// container with the values of each `Option` is returned.
1053     ///
1054     /// Here is an example which increments every integer in a vector,
1055     /// checking for overflow:
1056     ///
1057     /// ```
1058     /// use std::u16;
1059     ///
1060     /// let v = vec![1, 2];
1061     /// let res: Option<Vec<u16>> = v.iter().map(|&x: &u16|
1062     ///     if x == u16::MAX { None }
1063     ///     else { Some(x + 1) }
1064     /// ).collect();
1065     /// assert!(res == Some(vec![2, 3]));
1066     /// ```
1067     ///
1068     /// [`Iterator`]: ../iter/trait.Iterator.html
1069     /// [`None`]: enum.Option.html#variant.None
1070     #[inline]
1071     fn from_iter<I: IntoIterator<Item=Option<A>>>(iter: I) -> Option<V> {
1072         // FIXME(#11084): This could be replaced with Iterator::scan when this
1073         // performance bug is closed.
1074
1075         struct Adapter<Iter> {
1076             iter: Iter,
1077             found_none: bool,
1078         }
1079
1080         impl<T, Iter: Iterator<Item=Option<T>>> Iterator for Adapter<Iter> {
1081             type Item = T;
1082
1083             #[inline]
1084             fn next(&mut self) -> Option<T> {
1085                 match self.iter.next() {
1086                     Some(Some(value)) => Some(value),
1087                     Some(None) => {
1088                         self.found_none = true;
1089                         None
1090                     }
1091                     None => None,
1092                 }
1093             }
1094         }
1095
1096         let mut adapter = Adapter { iter: iter.into_iter(), found_none: false };
1097         let v: V = FromIterator::from_iter(adapter.by_ref());
1098
1099         if adapter.found_none {
1100             None
1101         } else {
1102             Some(v)
1103         }
1104     }
1105 }