range.h 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. // ======================================================================== //
  2. // Copyright 2009-2017 Intel Corporation //
  3. // //
  4. // Licensed under the Apache License, Version 2.0 (the "License"); //
  5. // you may not use this file except in compliance with the License. //
  6. // You may obtain a copy of the License at //
  7. // //
  8. // http://www.apache.org/licenses/LICENSE-2.0 //
  9. // //
  10. // Unless required by applicable law or agreed to in writing, software //
  11. // distributed under the License is distributed on an "AS IS" BASIS, //
  12. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. //
  13. // See the License for the specific language governing permissions and //
  14. // limitations under the License. //
  15. // ======================================================================== //
  16. #pragma once
  17. #include "../sys/platform.h"
  18. #include "../math/math.h"
  19. namespace embree
  20. {
  21. template<typename Ty>
  22. struct range
  23. {
  24. __forceinline range() {}
  25. __forceinline range(const Ty& begin)
  26. : _begin(begin), _end(begin+1) {}
  27. __forceinline range(const Ty& begin, const Ty& end)
  28. : _begin(begin), _end(end) {}
  29. __forceinline range(const range& other)
  30. : _begin(other._begin), _end(other._end) {}
  31. template<typename T1>
  32. __forceinline range(const range<T1>& other)
  33. : _begin(Ty(other._begin)), _end(Ty(other._end)) {}
  34. template<typename T1>
  35. __forceinline range& operator =(const range<T1>& other) {
  36. _begin = other._begin;
  37. _end = other._end;
  38. return *this;
  39. }
  40. __forceinline Ty begin() const {
  41. return _begin;
  42. }
  43. __forceinline Ty end() const {
  44. return _end;
  45. }
  46. __forceinline range intersect(const range& r) const {
  47. return range (max(_begin,r._begin),min(_end,r._end));
  48. }
  49. __forceinline Ty size() const {
  50. return _end - _begin;
  51. }
  52. __forceinline bool empty() const {
  53. return _end <= _begin;
  54. }
  55. __forceinline Ty center() const {
  56. return (_begin + _end)/2;
  57. }
  58. __forceinline std::pair<range,range> split() const
  59. {
  60. const Ty _center = center();
  61. return std::make_pair(range(_begin,_center),range(_center,_end));
  62. }
  63. __forceinline void split(range& left_o, range& right_o) const
  64. {
  65. const Ty _center = center();
  66. left_o = range(_begin,_center);
  67. right_o = range(_center,_end);
  68. }
  69. __forceinline friend bool operator< (const range& r0, const range& r1) {
  70. return r0.size() < r1.size();
  71. }
  72. friend std::ostream& operator<<(std::ostream& cout, const range& r) {
  73. return cout << "range [" << r.begin() << ", " << r.end() << "]";
  74. }
  75. Ty _begin, _end;
  76. };
  77. template<typename Ty>
  78. range<Ty> make_range(const Ty& begin, const Ty& end) {
  79. return range<Ty>(begin,end);
  80. }
  81. template<typename Ty>
  82. struct extended_range
  83. {
  84. __forceinline extended_range () {}
  85. __forceinline extended_range (const Ty& begin)
  86. : _begin(begin), _end(begin+1), _ext_end(begin+1) {}
  87. __forceinline extended_range (const Ty& begin, const Ty& end)
  88. : _begin(begin), _end(end), _ext_end(end) {}
  89. __forceinline extended_range (const Ty& begin, const Ty& end, const Ty& ext_end)
  90. : _begin(begin), _end(end), _ext_end(ext_end) {}
  91. __forceinline Ty begin() const {
  92. return _begin;
  93. }
  94. __forceinline Ty end() const {
  95. return _end;
  96. }
  97. __forceinline Ty ext_end() const {
  98. return _ext_end;
  99. }
  100. __forceinline Ty size() const {
  101. return _end - _begin;
  102. }
  103. __forceinline Ty ext_size() const {
  104. return _ext_end - _begin;
  105. }
  106. __forceinline Ty ext_range_size() const {
  107. return _ext_end - _end;
  108. }
  109. __forceinline bool has_ext_range() const {
  110. assert(_ext_end >= _end);
  111. return (_ext_end - _end) > 0;
  112. }
  113. __forceinline void set_ext_range(const size_t ext_end){
  114. assert(ext_end >= _end);
  115. _ext_end = ext_end;
  116. }
  117. __forceinline void move_right(const size_t plus){
  118. _begin += plus;
  119. _end += plus;
  120. _ext_end += plus;
  121. }
  122. friend std::ostream& operator<<(std::ostream& cout, const extended_range& r) {
  123. return cout << "extended_range [" << r.begin() << ", " << r.end() << " (" << r.ext_end() << ")]";
  124. }
  125. Ty _begin, _end, _ext_end;
  126. };
  127. }