adjust_test.go 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504
  1. package imaging
  2. import (
  3. "image"
  4. "testing"
  5. )
  6. func TestGrayscale(t *testing.T) {
  7. td := []struct {
  8. desc string
  9. src image.Image
  10. want *image.NRGBA
  11. }{
  12. {
  13. "Grayscale 3x3",
  14. &image.NRGBA{
  15. Rect: image.Rect(-1, -1, 2, 2),
  16. Stride: 3 * 4,
  17. Pix: []uint8{
  18. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  19. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  20. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  21. },
  22. },
  23. &image.NRGBA{
  24. Rect: image.Rect(0, 0, 3, 3),
  25. Stride: 3 * 4,
  26. Pix: []uint8{
  27. 0x3d, 0x3d, 0x3d, 0x01, 0x78, 0x78, 0x78, 0x02, 0x17, 0x17, 0x17, 0x03,
  28. 0x1f, 0x1f, 0x1f, 0xff, 0x25, 0x25, 0x25, 0xff, 0x66, 0x66, 0x66, 0xff,
  29. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  30. },
  31. },
  32. },
  33. }
  34. for _, d := range td {
  35. got := Grayscale(d.src)
  36. want := d.want
  37. if !compareNRGBA(got, want, 0) {
  38. t.Errorf("test [%s] failed: %#v", d.desc, got)
  39. }
  40. }
  41. }
  42. func TestInvert(t *testing.T) {
  43. td := []struct {
  44. desc string
  45. src image.Image
  46. want *image.NRGBA
  47. }{
  48. {
  49. "Invert 3x3",
  50. &image.NRGBA{
  51. Rect: image.Rect(-1, -1, 2, 2),
  52. Stride: 3 * 4,
  53. Pix: []uint8{
  54. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  55. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  56. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  57. },
  58. },
  59. &image.NRGBA{
  60. Rect: image.Rect(0, 0, 3, 3),
  61. Stride: 3 * 4,
  62. Pix: []uint8{
  63. 0x33, 0xff, 0xff, 0x01, 0xff, 0x33, 0xff, 0x02, 0xff, 0xff, 0x33, 0x03,
  64. 0xee, 0xdd, 0xcc, 0xff, 0xcc, 0xdd, 0xee, 0xff, 0x55, 0xcc, 0x44, 0xff,
  65. 0xff, 0xff, 0xff, 0xff, 0xcc, 0xcc, 0xcc, 0xff, 0x00, 0x00, 0x00, 0xff,
  66. },
  67. },
  68. },
  69. }
  70. for _, d := range td {
  71. got := Invert(d.src)
  72. want := d.want
  73. if !compareNRGBA(got, want, 0) {
  74. t.Errorf("test [%s] failed: %#v", d.desc, got)
  75. }
  76. }
  77. }
  78. func TestAdjustContrast(t *testing.T) {
  79. td := []struct {
  80. desc string
  81. src image.Image
  82. p float64
  83. want *image.NRGBA
  84. }{
  85. {
  86. "AdjustContrast 3x3 10",
  87. &image.NRGBA{
  88. Rect: image.Rect(-1, -1, 2, 2),
  89. Stride: 3 * 4,
  90. Pix: []uint8{
  91. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  92. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  93. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  94. },
  95. },
  96. 10,
  97. &image.NRGBA{
  98. Rect: image.Rect(0, 0, 3, 3),
  99. Stride: 3 * 4,
  100. Pix: []uint8{
  101. 0xd5, 0x00, 0x00, 0x01, 0x00, 0xd5, 0x00, 0x02, 0x00, 0x00, 0xd5, 0x03,
  102. 0x05, 0x18, 0x2b, 0xff, 0x2b, 0x18, 0x05, 0xff, 0xaf, 0x2b, 0xc2, 0xff,
  103. 0x00, 0x00, 0x00, 0xff, 0x2b, 0x2b, 0x2b, 0xff, 0xff, 0xff, 0xff, 0xff,
  104. },
  105. },
  106. },
  107. {
  108. "AdjustContrast 3x3 100",
  109. &image.NRGBA{
  110. Rect: image.Rect(-1, -1, 2, 2),
  111. Stride: 3 * 4,
  112. Pix: []uint8{
  113. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  114. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  115. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  116. },
  117. },
  118. 100,
  119. &image.NRGBA{
  120. Rect: image.Rect(0, 0, 3, 3),
  121. Stride: 3 * 4,
  122. Pix: []uint8{
  123. 0xff, 0x00, 0x00, 0x01, 0x00, 0xff, 0x00, 0x02, 0x00, 0x00, 0xff, 0x03,
  124. 0x00, 0x00, 0x00, 0xff, 0x00, 0x00, 0x00, 0xff, 0xff, 0x00, 0xff, 0xff,
  125. 0x00, 0x00, 0x00, 0xff, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff,
  126. },
  127. },
  128. },
  129. {
  130. "AdjustContrast 3x3 -10",
  131. &image.NRGBA{
  132. Rect: image.Rect(-1, -1, 2, 2),
  133. Stride: 3 * 4,
  134. Pix: []uint8{
  135. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  136. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  137. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  138. },
  139. },
  140. -10,
  141. &image.NRGBA{
  142. Rect: image.Rect(0, 0, 3, 3),
  143. Stride: 3 * 4,
  144. Pix: []uint8{
  145. 0xc4, 0x0d, 0x0d, 0x01, 0x0d, 0xc4, 0x0d, 0x02, 0x0d, 0x0d, 0xc4, 0x03,
  146. 0x1c, 0x2b, 0x3b, 0xff, 0x3b, 0x2b, 0x1c, 0xff, 0xa6, 0x3b, 0xb5, 0xff,
  147. 0x0d, 0x0d, 0x0d, 0xff, 0x3b, 0x3b, 0x3b, 0xff, 0xf2, 0xf2, 0xf2, 0xff,
  148. },
  149. },
  150. },
  151. {
  152. "AdjustContrast 3x3 -100",
  153. &image.NRGBA{
  154. Rect: image.Rect(-1, -1, 2, 2),
  155. Stride: 3 * 4,
  156. Pix: []uint8{
  157. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  158. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  159. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  160. },
  161. },
  162. -100,
  163. &image.NRGBA{
  164. Rect: image.Rect(0, 0, 3, 3),
  165. Stride: 3 * 4,
  166. Pix: []uint8{
  167. 0x80, 0x80, 0x80, 0x01, 0x80, 0x80, 0x80, 0x02, 0x80, 0x80, 0x80, 0x03,
  168. 0x80, 0x80, 0x80, 0xff, 0x80, 0x80, 0x80, 0xff, 0x80, 0x80, 0x80, 0xff,
  169. 0x80, 0x80, 0x80, 0xff, 0x80, 0x80, 0x80, 0xff, 0x80, 0x80, 0x80, 0xff,
  170. },
  171. },
  172. },
  173. {
  174. "AdjustContrast 3x3 0",
  175. &image.NRGBA{
  176. Rect: image.Rect(-1, -1, 2, 2),
  177. Stride: 3 * 4,
  178. Pix: []uint8{
  179. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  180. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  181. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  182. },
  183. },
  184. 0,
  185. &image.NRGBA{
  186. Rect: image.Rect(0, 0, 3, 3),
  187. Stride: 3 * 4,
  188. Pix: []uint8{
  189. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  190. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  191. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  192. },
  193. },
  194. },
  195. }
  196. for _, d := range td {
  197. got := AdjustContrast(d.src, d.p)
  198. want := d.want
  199. if !compareNRGBA(got, want, 0) {
  200. t.Errorf("test [%s] failed: %#v", d.desc, got)
  201. }
  202. }
  203. }
  204. func TestAdjustBrightness(t *testing.T) {
  205. td := []struct {
  206. desc string
  207. src image.Image
  208. p float64
  209. want *image.NRGBA
  210. }{
  211. {
  212. "AdjustBrightness 3x3 10",
  213. &image.NRGBA{
  214. Rect: image.Rect(-1, -1, 2, 2),
  215. Stride: 3 * 4,
  216. Pix: []uint8{
  217. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  218. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  219. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  220. },
  221. },
  222. 10,
  223. &image.NRGBA{
  224. Rect: image.Rect(0, 0, 3, 3),
  225. Stride: 3 * 4,
  226. Pix: []uint8{
  227. 0xe6, 0x1a, 0x1a, 0x01, 0x1a, 0xe6, 0x1a, 0x02, 0x1a, 0x1a, 0xe6, 0x03,
  228. 0x2b, 0x3c, 0x4d, 0xff, 0x4d, 0x3c, 0x2b, 0xff, 0xc4, 0x4d, 0xd5, 0xff,
  229. 0x1a, 0x1a, 0x1a, 0xff, 0x4d, 0x4d, 0x4d, 0xff, 0xff, 0xff, 0xff, 0xff,
  230. },
  231. },
  232. },
  233. {
  234. "AdjustBrightness 3x3 100",
  235. &image.NRGBA{
  236. Rect: image.Rect(-1, -1, 2, 2),
  237. Stride: 3 * 4,
  238. Pix: []uint8{
  239. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  240. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  241. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  242. },
  243. },
  244. 100,
  245. &image.NRGBA{
  246. Rect: image.Rect(0, 0, 3, 3),
  247. Stride: 3 * 4,
  248. Pix: []uint8{
  249. 0xff, 0xff, 0xff, 0x01, 0xff, 0xff, 0xff, 0x02, 0xff, 0xff, 0xff, 0x03,
  250. 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
  251. 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
  252. },
  253. },
  254. },
  255. {
  256. "AdjustBrightness 3x3 -10",
  257. &image.NRGBA{
  258. Rect: image.Rect(-1, -1, 2, 2),
  259. Stride: 3 * 4,
  260. Pix: []uint8{
  261. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  262. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  263. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  264. },
  265. },
  266. -10,
  267. &image.NRGBA{
  268. Rect: image.Rect(0, 0, 3, 3),
  269. Stride: 3 * 4,
  270. Pix: []uint8{
  271. 0xb3, 0x00, 0x00, 0x01, 0x00, 0xb3, 0x00, 0x02, 0x00, 0x00, 0xb3, 0x03,
  272. 0x00, 0x09, 0x1a, 0xff, 0x1a, 0x09, 0x00, 0xff, 0x91, 0x1a, 0xa2, 0xff,
  273. 0x00, 0x00, 0x00, 0xff, 0x1a, 0x1a, 0x1a, 0xff, 0xe6, 0xe6, 0xe6, 0xff,
  274. },
  275. },
  276. },
  277. {
  278. "AdjustBrightness 3x3 -100",
  279. &image.NRGBA{
  280. Rect: image.Rect(-1, -1, 2, 2),
  281. Stride: 3 * 4,
  282. Pix: []uint8{
  283. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  284. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  285. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  286. },
  287. },
  288. -100,
  289. &image.NRGBA{
  290. Rect: image.Rect(0, 0, 3, 3),
  291. Stride: 3 * 4,
  292. Pix: []uint8{
  293. 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x03,
  294. 0x00, 0x00, 0x00, 0xff, 0x00, 0x00, 0x00, 0xff, 0x00, 0x00, 0x00, 0xff,
  295. 0x00, 0x00, 0x00, 0xff, 0x00, 0x00, 0x00, 0xff, 0x00, 0x00, 0x00, 0xff,
  296. },
  297. },
  298. },
  299. {
  300. "AdjustBrightness 3x3 0",
  301. &image.NRGBA{
  302. Rect: image.Rect(-1, -1, 2, 2),
  303. Stride: 3 * 4,
  304. Pix: []uint8{
  305. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  306. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  307. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  308. },
  309. },
  310. 0,
  311. &image.NRGBA{
  312. Rect: image.Rect(0, 0, 3, 3),
  313. Stride: 3 * 4,
  314. Pix: []uint8{
  315. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  316. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  317. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  318. },
  319. },
  320. },
  321. }
  322. for _, d := range td {
  323. got := AdjustBrightness(d.src, d.p)
  324. want := d.want
  325. if !compareNRGBA(got, want, 0) {
  326. t.Errorf("test [%s] failed: %#v", d.desc, got)
  327. }
  328. }
  329. }
  330. func TestAdjustGamma(t *testing.T) {
  331. td := []struct {
  332. desc string
  333. src image.Image
  334. p float64
  335. want *image.NRGBA
  336. }{
  337. {
  338. "AdjustGamma 3x3 0.75",
  339. &image.NRGBA{
  340. Rect: image.Rect(-1, -1, 2, 2),
  341. Stride: 3 * 4,
  342. Pix: []uint8{
  343. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  344. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  345. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  346. },
  347. },
  348. 0.75,
  349. &image.NRGBA{
  350. Rect: image.Rect(0, 0, 3, 3),
  351. Stride: 3 * 4,
  352. Pix: []uint8{
  353. 0xbd, 0x00, 0x00, 0x01, 0x00, 0xbd, 0x00, 0x02, 0x00, 0x00, 0xbd, 0x03,
  354. 0x07, 0x11, 0x1e, 0xff, 0x1e, 0x11, 0x07, 0xff, 0x95, 0x1e, 0xa9, 0xff,
  355. 0x00, 0x00, 0x00, 0xff, 0x1e, 0x1e, 0x1e, 0xff, 0xff, 0xff, 0xff, 0xff,
  356. },
  357. },
  358. },
  359. {
  360. "AdjustGamma 3x3 1.5",
  361. &image.NRGBA{
  362. Rect: image.Rect(-1, -1, 2, 2),
  363. Stride: 3 * 4,
  364. Pix: []uint8{
  365. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  366. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  367. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  368. },
  369. },
  370. 1.5,
  371. &image.NRGBA{
  372. Rect: image.Rect(0, 0, 3, 3),
  373. Stride: 3 * 4,
  374. Pix: []uint8{
  375. 0xdc, 0x00, 0x00, 0x01, 0x00, 0xdc, 0x00, 0x02, 0x00, 0x00, 0xdc, 0x03,
  376. 0x2a, 0x43, 0x57, 0xff, 0x57, 0x43, 0x2a, 0xff, 0xc3, 0x57, 0xcf, 0xff,
  377. 0x00, 0x00, 0x00, 0xff, 0x57, 0x57, 0x57, 0xff, 0xff, 0xff, 0xff, 0xff,
  378. },
  379. },
  380. },
  381. {
  382. "AdjustGamma 3x3 1.0",
  383. &image.NRGBA{
  384. Rect: image.Rect(-1, -1, 2, 2),
  385. Stride: 3 * 4,
  386. Pix: []uint8{
  387. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  388. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  389. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  390. },
  391. },
  392. 1.0,
  393. &image.NRGBA{
  394. Rect: image.Rect(0, 0, 3, 3),
  395. Stride: 3 * 4,
  396. Pix: []uint8{
  397. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  398. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  399. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  400. },
  401. },
  402. },
  403. }
  404. for _, d := range td {
  405. got := AdjustGamma(d.src, d.p)
  406. want := d.want
  407. if !compareNRGBA(got, want, 0) {
  408. t.Errorf("test [%s] failed: %#v", d.desc, got)
  409. }
  410. }
  411. }
  412. func TestAdjustSigmoid(t *testing.T) {
  413. td := []struct {
  414. desc string
  415. src image.Image
  416. m float64
  417. p float64
  418. want *image.NRGBA
  419. }{
  420. {
  421. "AdjustSigmoid 3x3 0.5 3.0",
  422. &image.NRGBA{
  423. Rect: image.Rect(-1, -1, 2, 2),
  424. Stride: 3 * 4,
  425. Pix: []uint8{
  426. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  427. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  428. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  429. },
  430. },
  431. 0.5,
  432. 3.0,
  433. &image.NRGBA{
  434. Rect: image.Rect(0, 0, 3, 3),
  435. Stride: 3 * 4,
  436. Pix: []uint8{
  437. 0xd4, 0x00, 0x00, 0x01, 0x00, 0xd4, 0x00, 0x02, 0x00, 0x00, 0xd4, 0x03,
  438. 0x0d, 0x1b, 0x2b, 0xff, 0x2b, 0x1b, 0x0d, 0xff, 0xb1, 0x2b, 0xc3, 0xff,
  439. 0x00, 0x00, 0x00, 0xff, 0x2b, 0x2b, 0x2b, 0xff, 0xff, 0xff, 0xff, 0xff,
  440. },
  441. },
  442. },
  443. {
  444. "AdjustSigmoid 3x3 0.5 -3.0",
  445. &image.NRGBA{
  446. Rect: image.Rect(-1, -1, 2, 2),
  447. Stride: 3 * 4,
  448. Pix: []uint8{
  449. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  450. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  451. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  452. },
  453. },
  454. 0.5,
  455. -3.0,
  456. &image.NRGBA{
  457. Rect: image.Rect(0, 0, 3, 3),
  458. Stride: 3 * 4,
  459. Pix: []uint8{
  460. 0xc4, 0x00, 0x00, 0x01, 0x00, 0xc4, 0x00, 0x02, 0x00, 0x00, 0xc4, 0x03,
  461. 0x16, 0x2a, 0x3b, 0xff, 0x3b, 0x2a, 0x16, 0xff, 0xa4, 0x3b, 0xb3, 0xff,
  462. 0x00, 0x00, 0x00, 0xff, 0x3b, 0x3b, 0x3b, 0xff, 0xff, 0xff, 0xff, 0xff,
  463. },
  464. },
  465. },
  466. {
  467. "AdjustSigmoid 3x3 0.5 0.0",
  468. &image.NRGBA{
  469. Rect: image.Rect(-1, -1, 2, 2),
  470. Stride: 3 * 4,
  471. Pix: []uint8{
  472. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  473. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  474. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  475. },
  476. },
  477. 0.5,
  478. 0.0,
  479. &image.NRGBA{
  480. Rect: image.Rect(0, 0, 3, 3),
  481. Stride: 3 * 4,
  482. Pix: []uint8{
  483. 0xcc, 0x00, 0x00, 0x01, 0x00, 0xcc, 0x00, 0x02, 0x00, 0x00, 0xcc, 0x03,
  484. 0x11, 0x22, 0x33, 0xff, 0x33, 0x22, 0x11, 0xff, 0xaa, 0x33, 0xbb, 0xff,
  485. 0x00, 0x00, 0x00, 0xff, 0x33, 0x33, 0x33, 0xff, 0xff, 0xff, 0xff, 0xff,
  486. },
  487. },
  488. },
  489. }
  490. for _, d := range td {
  491. got := AdjustSigmoid(d.src, d.m, d.p)
  492. want := d.want
  493. if !compareNRGBA(got, want, 0) {
  494. t.Errorf("test [%s] failed: %#v", d.desc, got)
  495. }
  496. }
  497. }