NTest_pixbuf.lua 9.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288
  1. local N = ...
  2. N = (N or require "NTest")("pixbuf")
  3. local function initBuffer(buf, ...)
  4. for i,v in ipairs({...}) do
  5. buf:set(i, v, v*2, v*3, v*4)
  6. end
  7. return buf
  8. end
  9. N.test('initialize a buffer', function()
  10. local buffer = pixbuf.newBuffer(9, 3)
  11. nok(buffer == nil)
  12. ok(eq(buffer:size(), 9), "check size")
  13. ok(eq(buffer:dump(), string.char(0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0)), "initialize with 0")
  14. fail(function() pixbuf.newBuffer(9, -1) end, "should be a positive integer")
  15. fail(function() pixbuf.newBuffer(0, 3) end, "should be a positive integer")
  16. fail(function() pixbuf.newBuffer(-1, 3) end, "should be a positive integer")
  17. end)
  18. N.test('have correct size', function()
  19. local buffer = pixbuf.newBuffer(9, 3)
  20. ok(eq(buffer:size(), 9), "check size")
  21. buffer = pixbuf.newBuffer(9, 4)
  22. ok(eq(buffer:size(), 9), "check size")
  23. end)
  24. N.test('fill a buffer with one color', function()
  25. local buffer = pixbuf.newBuffer(3, 3)
  26. buffer:fill(1,222,55)
  27. ok(eq(buffer:dump(), string.char(1,222,55,1,222,55,1,222,55)), "RGB")
  28. buffer = pixbuf.newBuffer(3, 4)
  29. buffer:fill(1,222,55,77)
  30. ok(eq(buffer:dump(), string.char(1,222,55,77,1,222,55,77,1,222,55,77)), "RGBW")
  31. end)
  32. N.test('replace correctly', function()
  33. local buffer = pixbuf.newBuffer(5, 3)
  34. buffer:replace(string.char(3,255,165,33,0,244,12,87,255))
  35. ok(eq(buffer:dump(), string.char(3,255,165,33,0,244,12,87,255,0,0,0,0,0,0)), "RGBW")
  36. buffer = pixbuf.newBuffer(5, 3)
  37. buffer:replace(string.char(3,255,165,33,0,244,12,87,255), 2)
  38. ok(eq(buffer:dump(), string.char(0,0,0,3,255,165,33,0,244,12,87,255,0,0,0)), "RGBW")
  39. buffer = pixbuf.newBuffer(5, 3)
  40. buffer:replace(string.char(3,255,165,33,0,244,12,87,255), -5)
  41. ok(eq(buffer:dump(), string.char(3,255,165,33,0,244,12,87,255,0,0,0,0,0,0)), "RGBW")
  42. fail(function() buffer:replace(string.char(3,255,165,33,0,244,12,87,255), 4) end,
  43. "does not fit into destination")
  44. end)
  45. N.test('replace correctly issue #2921', function()
  46. local buffer = pixbuf.newBuffer(5, 3)
  47. buffer:replace(string.char(3,255,165,33,0,244,12,87,255), -7)
  48. ok(eq(buffer:dump(), string.char(3,255,165,33,0,244,12,87,255,0,0,0,0,0,0)), "RGBW")
  49. end)
  50. N.test('get/set correctly', function()
  51. local buffer = pixbuf.newBuffer(3, 4)
  52. buffer:fill(1,222,55,13)
  53. ok(eq({buffer:get(2)},{1,222,55,13}))
  54. buffer:set(2, 4,53,99,0)
  55. ok(eq({buffer:get(1)},{1,222,55,13}))
  56. ok(eq({buffer:get(2)},{4,53,99,0}))
  57. ok(eq(buffer:dump(), string.char(1,222,55,13,4,53,99,0,1,222,55,13)), "RGBW")
  58. fail(function() buffer:get(0) end, "index out of range")
  59. fail(function() buffer:get(4) end, "index out of range")
  60. fail(function() buffer:set(0,1,2,3,4) end, "index out of range")
  61. fail(function() buffer:set(4,1,2,3,4) end, "index out of range")
  62. fail(function() buffer:set(2,1,2,3) end, "number expected, got no value")
  63. fail(function() buffer:set(2,1,2,3,4,5) end, "extra values given")
  64. end)
  65. N.test('get/set multiple with string', function()
  66. -- verify that :set does indeed return its input
  67. local buffer = pixbuf.newBuffer(4, 3):set(1,"ABCDEF")
  68. buffer:set(3,"LMNOPQ")
  69. ok(eq(buffer:dump(), "ABCDEFLMNOPQ"))
  70. fail(function() buffer:set(4,"AAAAAA") end, "string size will exceed strip length")
  71. fail(function() buffer:set(2,"AAAAA") end, "string does not contain whole LEDs")
  72. end)
  73. N.test('fade correctly', function()
  74. local buffer = pixbuf.newBuffer(1, 3)
  75. buffer:fill(1,222,55)
  76. buffer:fade(2)
  77. ok(buffer:dump() == string.char(0,111,27), "RGB")
  78. buffer:fill(1,222,55)
  79. buffer:fade(3, pixbuf.FADE_OUT)
  80. ok(buffer:dump() == string.char(0,math.floor(222/3),math.floor(55/3)), "RGB")
  81. buffer:fill(1,222,55)
  82. buffer:fade(3, pixbuf.FADE_IN)
  83. ok(buffer:dump() == string.char(3,255,165), "RGB")
  84. buffer = pixbuf.newBuffer(1, 4)
  85. buffer:fill(1,222,55, 77)
  86. buffer:fade(2, pixbuf.FADE_OUT)
  87. ok(eq(buffer:dump(), string.char(0,111,27,38)), "RGBW")
  88. end)
  89. N.test('mix correctly issue #1736', function()
  90. local buffer1 = pixbuf.newBuffer(1, 3)
  91. local buffer2 = pixbuf.newBuffer(1, 3)
  92. buffer1:fill(10,22,54)
  93. buffer2:fill(10,27,55)
  94. buffer1:mix(256/8*7,buffer1,256/8,buffer2)
  95. ok(eq({buffer1:get(1)}, {10,23,54}))
  96. end)
  97. N.test('mix saturation correctly ', function()
  98. local buffer1 = pixbuf.newBuffer(1, 3)
  99. local buffer2 = pixbuf.newBuffer(1, 3)
  100. buffer1:fill(10,22,54)
  101. buffer2:fill(10,27,55)
  102. buffer1:mix(256/2,buffer1,-256,buffer2)
  103. ok(eq({buffer1:get(1)}, {0,0,0}))
  104. buffer1:fill(10,22,54)
  105. buffer2:fill(10,27,55)
  106. buffer1:mix(25600,buffer1,256/8,buffer2)
  107. ok(eq({buffer1:get(1)}, {255,255,255}))
  108. buffer1:fill(10,22,54)
  109. buffer2:fill(10,27,55)
  110. buffer1:mix(-257,buffer1,255,buffer2)
  111. ok(eq({buffer1:get(1)}, {0,5,1}))
  112. end)
  113. N.test('power', function()
  114. local buffer = pixbuf.newBuffer(2, 4)
  115. buffer:fill(10,22,54,234)
  116. ok(eq(buffer:power(), 2*(10+22+54+234)))
  117. end)
  118. N.test('shift LOGICAL', function()
  119. local buffer1 = pixbuf.newBuffer(4, 4)
  120. local buffer2 = pixbuf.newBuffer(4, 4)
  121. initBuffer(buffer1,7,8,9,12)
  122. initBuffer(buffer2,0,0,7,8)
  123. ok(buffer1 ~= buffer2, "disequality pre shift")
  124. buffer1:shift(2)
  125. ok(buffer1 == buffer2, "shift right")
  126. initBuffer(buffer1,7,8,9,12)
  127. initBuffer(buffer2,9,12,0,0)
  128. buffer1:shift(-2)
  129. ok(buffer1 == buffer2, "shift left")
  130. initBuffer(buffer1,7,8,9,12)
  131. initBuffer(buffer2,7,0,8,12)
  132. buffer1:shift(1, nil, 2,3)
  133. ok(buffer1 == buffer2, "shift middle right")
  134. initBuffer(buffer1,7,8,9,12)
  135. initBuffer(buffer2,7,9,0,12)
  136. buffer1:shift(-1, nil, 2,3)
  137. ok(buffer1 == buffer2, "shift middle left")
  138. -- bounds checks, handle gracefully as string:sub does
  139. initBuffer(buffer1,7,8,9,12)
  140. initBuffer(buffer2,8,9,12,0)
  141. buffer1:shift(-1, pixbuf.SHIFT_LOGICAL, 0,5)
  142. ok(buffer1 == buffer2, "shift left out of bound")
  143. initBuffer(buffer1,7,8,9,12)
  144. initBuffer(buffer2,0,7,8,9)
  145. buffer1:shift(1, pixbuf.SHIFT_LOGICAL, 0,5)
  146. ok(buffer1 == buffer2, "shift right out of bound")
  147. end)
  148. N.test('shift LOGICAL issue #2946', function()
  149. local buffer1 = pixbuf.newBuffer(4, 4)
  150. local buffer2 = pixbuf.newBuffer(4, 4)
  151. initBuffer(buffer1,7,8,9,12)
  152. initBuffer(buffer2,0,0,0,0)
  153. buffer1:shift(4)
  154. ok(buffer1 == buffer2, "shift all right")
  155. initBuffer(buffer1,7,8,9,12)
  156. initBuffer(buffer2,0,0,0,0)
  157. buffer1:shift(-4)
  158. ok(buffer1 == buffer2, "shift all left")
  159. fail(function() buffer1:shift(10) end, "shifting more elements than buffer size")
  160. fail(function() buffer1:shift(-6) end, "shifting more elements than buffer size")
  161. end)
  162. N.test('shift CIRCULAR', function()
  163. local buffer1 = pixbuf.newBuffer(4, 4)
  164. local buffer2 = pixbuf.newBuffer(4, 4)
  165. initBuffer(buffer1,7,8,9,12)
  166. initBuffer(buffer2,9,12,7,8)
  167. buffer1:shift(2, pixbuf.SHIFT_CIRCULAR)
  168. ok(buffer1 == buffer2, "shift right")
  169. initBuffer(buffer1,7,8,9,12)
  170. initBuffer(buffer2,9,12,7,8)
  171. buffer1:shift(-2, pixbuf.SHIFT_CIRCULAR)
  172. ok(buffer1 == buffer2, "shift left")
  173. initBuffer(buffer1,7,8,9,12)
  174. initBuffer(buffer2,7,9,8,12)
  175. buffer1:shift(1, pixbuf.SHIFT_CIRCULAR, 2,3)
  176. ok(buffer1 == buffer2, "shift middle right")
  177. initBuffer(buffer1,7,8,9,12)
  178. initBuffer(buffer2,7,9,8,12)
  179. buffer1:shift(-1, pixbuf.SHIFT_CIRCULAR, 2,3)
  180. ok(buffer1 == buffer2, "shift middle left")
  181. -- bounds checks, handle gracefully as string:sub does
  182. initBuffer(buffer1,7,8,9,12)
  183. initBuffer(buffer2,8,9,12,7)
  184. buffer1:shift(-1, pixbuf.SHIFT_CIRCULAR, 0,5)
  185. ok(buffer1 == buffer2, "shift left out of bound")
  186. initBuffer(buffer1,7,8,9,12)
  187. initBuffer(buffer2,12,7,8,9)
  188. buffer1:shift(1, pixbuf.SHIFT_CIRCULAR, 0,5)
  189. ok(buffer1 == buffer2, "shift right out of bound")
  190. initBuffer(buffer1,7,8,9,12)
  191. initBuffer(buffer2,12,7,8,9)
  192. buffer1:shift(1, pixbuf.SHIFT_CIRCULAR, -12,12)
  193. ok(buffer1 == buffer2, "shift right way out of bound")
  194. end)
  195. N.test('sub', function()
  196. local buffer1 = pixbuf.newBuffer(4, 4)
  197. initBuffer(buffer1,7,8,9,12)
  198. buffer1 = buffer1:sub(4,3)
  199. ok(eq(buffer1:size(), 0), "sub empty")
  200. local buffer2 = pixbuf.newBuffer(2, 4)
  201. buffer1 = pixbuf.newBuffer(4, 4)
  202. initBuffer(buffer1,7,8,9,12)
  203. initBuffer(buffer2,9,12)
  204. buffer1 = buffer1:sub(3,4)
  205. ok(buffer1 == buffer2, "sub")
  206. buffer1 = pixbuf.newBuffer(4, 4)
  207. buffer2 = pixbuf.newBuffer(4, 4)
  208. initBuffer(buffer1,7,8,9,12)
  209. initBuffer(buffer2,7,8,9,12)
  210. buffer1 = buffer1:sub(-12,33)
  211. ok(buffer1 == buffer2, "out of bounds")
  212. end)
  213. N.test('map', function()
  214. local buffer1 = pixbuf.newBuffer(4, 4)
  215. buffer1:fill(65,66,67,68)
  216. buffer1:map(function(a,b,c,d) return b,a,c,d end)
  217. ok(eq("BACDBACDBACDBACD", buffer1:dump()), "swizzle")
  218. local buffer2 = pixbuf.newBuffer(4, 1)
  219. buffer2:map(function(b,a,c,d) return c end, buffer1) -- luacheck: ignore
  220. ok(eq("CCCC", buffer2:dump()), "projection")
  221. local buffer3 = pixbuf.newBuffer(4, 3)
  222. buffer3:map(function(b,a,c,d) return a,b,d end, buffer1) -- luacheck: ignore
  223. ok(eq("ABDABDABDABD", buffer3:dump()), "projection 2")
  224. buffer1:fill(70,71,72,73)
  225. buffer1:map(function(c,a,b,d) return a,b,c,d end, buffer2, nil, nil, buffer3)
  226. ok(eq("ABCDABCDABCDABCD", buffer1:dump()), "zip")
  227. buffer1 = pixbuf.newBuffer(2, 4)
  228. buffer1:fill(70,71,72,73)
  229. buffer2:set(1,"ABCD")
  230. buffer3:set(1,"EFGHIJKLM")
  231. buffer1:map(function(c,a,b,d) return a,b,c,d end, buffer2, 1, 2, buffer3, 2)
  232. ok(eq("HIAJKLBM", buffer1:dump()), "partial zip")
  233. end)
  234. --[[
  235. pixbuf.buffer:__concat()
  236. --]]