Utilities.h 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
  1. /*
  2. * Copyright (c) 2021, Hunter Salyer <thefalsehonesty@gmail.com>
  3. * Copyright (c) 2022, Gregory Bertilson <zaggy1024@gmail.com>
  4. *
  5. * SPDX-License-Identifier: BSD-2-Clause
  6. */
  7. #pragma once
  8. #include <AK/Types.h>
  9. #include <LibGfx/Size.h>
  10. #include "LookupTables.h"
  11. namespace Video::VP9 {
  12. // FIXME: Once everything is working, replace this with plain clamp
  13. // since parameter order is different
  14. template<typename T>
  15. T clip_3(T x, T y, T z)
  16. {
  17. return clamp(z, x, y);
  18. }
  19. template<typename T>
  20. u16 clip_1(u8 bit_depth, T x)
  21. {
  22. if (x < 0) {
  23. return 0u;
  24. }
  25. const T max = (1u << bit_depth) - 1u;
  26. if (x > max)
  27. return max;
  28. return x;
  29. }
  30. template<typename T, typename C>
  31. inline T brev(C bit_count, T value)
  32. {
  33. T result = 0;
  34. for (C i = 0; i < bit_count; i++) {
  35. auto bit = (value >> i) & 1;
  36. result |= bit << (bit_count - 1 - i);
  37. }
  38. return result;
  39. }
  40. inline BlockSubsize get_subsampled_block_size(BlockSubsize size, bool subsampling_x, bool subsampling_y)
  41. {
  42. return ss_size_lookup[size < Block_8x8 ? Block_8x8 : size][subsampling_x][subsampling_y];
  43. }
  44. inline Gfx::Size<u8> block_size_to_blocks(BlockSubsize size)
  45. {
  46. return Gfx::Size<u8>(num_8x8_blocks_wide_lookup[size], num_8x8_blocks_high_lookup[size]);
  47. }
  48. inline Gfx::Size<u8> block_size_to_sub_blocks(BlockSubsize size)
  49. {
  50. return Gfx::Size<u8>(num_4x4_blocks_wide_lookup[size], num_4x4_blocks_high_lookup[size]);
  51. }
  52. template<Integral T>
  53. inline T blocks_to_superblocks(T blocks)
  54. {
  55. return blocks >> 3;
  56. }
  57. template<Integral T>
  58. inline T superblocks_to_blocks(T superblocks)
  59. {
  60. return superblocks << 3;
  61. }
  62. template<Integral T>
  63. inline T blocks_ceiled_to_superblocks(T blocks)
  64. {
  65. return blocks_to_superblocks(blocks + 7);
  66. }
  67. template<Integral T>
  68. inline T blocks_to_sub_blocks(T blocks)
  69. {
  70. return blocks << 1;
  71. }
  72. template<Integral T>
  73. inline T sub_blocks_to_blocks(T sub_blocks)
  74. {
  75. return sub_blocks >> 1;
  76. }
  77. template<Integral T>
  78. inline T sub_blocks_to_pixels(T sub_blocks)
  79. {
  80. return sub_blocks << 2;
  81. }
  82. template<Integral T>
  83. inline T pixels_to_sub_blocks(T pixels)
  84. {
  85. return pixels >> 2;
  86. }
  87. template<Integral T>
  88. inline T blocks_to_pixels(T blocks)
  89. {
  90. return sub_blocks_to_pixels(blocks_to_sub_blocks(blocks));
  91. }
  92. template<Integral T>
  93. inline T pixels_to_blocks(T pixels)
  94. {
  95. return sub_blocks_to_blocks(pixels_to_sub_blocks(pixels));
  96. }
  97. inline u8 transform_size_to_sub_blocks(TransformSize transform_size)
  98. {
  99. return 1 << transform_size;
  100. }
  101. }