ImfCheckedArithmetic.h 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. ///////////////////////////////////////////////////////////////////////////
  2. //
  3. // Copyright (c) 2009, Industrial Light & Magic, a division of Lucas
  4. // Digital Ltd. LLC
  5. //
  6. // All rights reserved.
  7. //
  8. // Redistribution and use in source and binary forms, with or without
  9. // modification, are permitted provided that the following conditions are
  10. // met:
  11. // * Redistributions of source code must retain the above copyright
  12. // notice, this list of conditions and the following disclaimer.
  13. // * Redistributions in binary form must reproduce the above
  14. // copyright notice, this list of conditions and the following disclaimer
  15. // in the documentation and/or other materials provided with the
  16. // distribution.
  17. // * Neither the name of Industrial Light & Magic nor the names of
  18. // its contributors may be used to endorse or promote products derived
  19. // from this software without specific prior written permission.
  20. //
  21. // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22. // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23. // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24. // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25. // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26. // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27. // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28. // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29. // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30. // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31. // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32. //
  33. ///////////////////////////////////////////////////////////////////////////
  34. #ifndef INCLUDED_IMF_CHECKED_ARITHMETIC_H
  35. #define INCLUDED_IMF_CHECKED_ARITHMETIC_H
  36. //-----------------------------------------------------------------------------
  37. //
  38. // Integer arithmetic operations that throw exceptions
  39. // on overflow, underflow or division by zero.
  40. //
  41. //-----------------------------------------------------------------------------
  42. #include <limits>
  43. #include "IexMathExc.h"
  44. #include "ImfNamespace.h"
  45. OPENEXR_IMF_INTERNAL_NAMESPACE_HEADER_ENTER
  46. template <bool b> struct StaticAssertionFailed;
  47. template <> struct StaticAssertionFailed <true> {};
  48. #define IMF_STATIC_ASSERT(x) \
  49. do {StaticAssertionFailed <x> staticAssertionFailed; ((void) staticAssertionFailed);} while (false)
  50. template <class T>
  51. T
  52. uiMult (T a, T b)
  53. {
  54. //
  55. // Unsigned integer multiplication
  56. //
  57. IMF_STATIC_ASSERT (!std::numeric_limits<T>::is_signed &&
  58. std::numeric_limits<T>::is_integer);
  59. if (a > 0 && b > std::numeric_limits<T>::max() / a)
  60. throw IEX_NAMESPACE::OverflowExc ("Integer multiplication overflow.");
  61. return a * b;
  62. }
  63. template <class T>
  64. T
  65. uiDiv (T a, T b)
  66. {
  67. //
  68. // Unsigned integer division
  69. //
  70. IMF_STATIC_ASSERT (!std::numeric_limits<T>::is_signed &&
  71. std::numeric_limits<T>::is_integer);
  72. if (b == 0)
  73. throw IEX_NAMESPACE::DivzeroExc ("Integer division by zero.");
  74. return a / b;
  75. }
  76. template <class T>
  77. T
  78. uiAdd (T a, T b)
  79. {
  80. //
  81. // Unsigned integer addition
  82. //
  83. IMF_STATIC_ASSERT (!std::numeric_limits<T>::is_signed &&
  84. std::numeric_limits<T>::is_integer);
  85. if (a > std::numeric_limits<T>::max() - b)
  86. throw IEX_NAMESPACE::OverflowExc ("Integer addition overflow.");
  87. return a + b;
  88. }
  89. template <class T>
  90. T
  91. uiSub (T a, T b)
  92. {
  93. //
  94. // Unsigned integer subtraction
  95. //
  96. IMF_STATIC_ASSERT (!std::numeric_limits<T>::is_signed &&
  97. std::numeric_limits<T>::is_integer);
  98. if (a < b)
  99. throw IEX_NAMESPACE::UnderflowExc ("Integer subtraction underflow.");
  100. return a - b;
  101. }
  102. template <class T>
  103. size_t
  104. checkArraySize (T n, size_t s)
  105. {
  106. //
  107. // Verify that the size, in bytes, of an array with n elements
  108. // of size s can be computed without overflowing:
  109. //
  110. // If computing
  111. //
  112. // size_t (n) * s
  113. //
  114. // would overflow, then throw an IEX_NAMESPACE::OverflowExc exception.
  115. // Otherwise return
  116. //
  117. // size_t (n).
  118. //
  119. IMF_STATIC_ASSERT (!std::numeric_limits<T>::is_signed &&
  120. std::numeric_limits<T>::is_integer);
  121. IMF_STATIC_ASSERT (sizeof (T) <= sizeof (size_t));
  122. if (size_t (n) > std::numeric_limits<size_t>::max() / s)
  123. throw IEX_NAMESPACE::OverflowExc ("Integer multiplication overflow.");
  124. return size_t (n);
  125. }
  126. OPENEXR_IMF_INTERNAL_NAMESPACE_HEADER_EXIT
  127. #endif