Guest User

Untitled

a guest
Jan 18th, 2018
85
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.86 KB | None | 0 0
  1. macro_rules! spec_range { ($($t:ty)*) =>
  2. ($(
  3. impl Iterator for ops::Range<$t> {
  4. #[inline]
  5. default fn next(&mut self) -> Option<Self::Item> {
  6. if self.start < self.end && self.start < <$t>::max_value() {
  7. let temp = self.start;
  8. self.start += 1;
  9. Some(temp)
  10. } else {
  11. None
  12. }
  13. }
  14.  
  15. #[inline]
  16. default fn size_hint(&self) -> (usize, Option<usize>) {
  17. if self.start < self.end {
  18. if let Ok(hint) = usize::try_from(self.end - self.start) {
  19. return (hint, Some(hint))
  20. }
  21. }
  22. (0, None)
  23. }
  24.  
  25. #[inline]
  26. default fn nth(&mut self, n: usize) -> Option<Self::Item> {
  27. if let Ok(nconv) = <$t>::try_from(n) {
  28. if let Some(res) = self.start.checked_add(nconv) {
  29. if res < self.end {
  30. self.start = res + 1;
  31. return Some(res)
  32. }
  33. }
  34. }
  35. self.start = self.end;
  36. None
  37. }
  38. }
  39. )*)
  40. }
  41.  
  42. spec_range!(usize u8 u16 u32 u64);
  43. spec_range!(isize i8 i16 i32 i64);
  44.  
  45.  
  46. macro_rules! spec_range {
  47. ($($t:ty)*) => ($(
  48. #[unstable(feature = "iterator_step_by",
  49. reason = "unstable replacement of Range::step_by",
  50. issue = "27741")]
  51. impl Iterator for StepBy<ops::Range<$t>> {
  52. #[inline]
  53. fn next(&mut self) -> Option<Self::Item> {
  54. if self.iter.start < self.iter.end {
  55. self.first_take = false;
  56. if let Ok(n) = <$t>::try_from(self.step) {
  57. if let Some(step) = self.iter.start.checked_add(n) {
  58. let ret = self.iter.start;
  59. self.iter.start = step;
  60. return Some(ret)
  61. }
  62. }
  63. let ret = self.iter.start;
  64. self.iter.start = self.iter.end;
  65. Some(ret)
  66. } else {
  67. None
  68. }
  69. }
  70.  
  71. #[inline]
  72. fn size_hint(&self) -> (usize, Option<usize>) {
  73. let inner_hint = self.iter.size_hint();
  74.  
  75. if self.first_take {
  76. let f = |n| if n == 0 { 0 } else { 1 + (n-1)/(self.step+1) };
  77. (f(inner_hint.0), inner_hint.1.map(f))
  78. } else {
  79. let f = |n| n / (self.step+1);
  80. (f(inner_hint.0), inner_hint.1.map(f))
  81. }
  82. }
  83. }
  84. )*)
  85. }
  86.  
  87. spec_range!(usize u8 u16 u32 u64 isize i8 i16 i32 i64);
Add Comment
Please, Sign In to add comment