spacepaste

Paste details

reply | raw

  1.  
  2. #include <cmath>
  3. #include <cstdint>
  4. #include <cstdio>
  5. #include <functional>
  6. #include <memory>
  7. #include <random>
  8. #include <CImg.h>
  10. using cimg_library::CImg;
  11. using cimg_library::CImgDisplay;
  13. struct Rectangle {
  14. int x, y, w, h;
  15. };
  17. static double distance(const CImg<std::uint8_t> &original, const CImg<std::uint8_t> &img,
  18. Rectangle bounds)
  19. {
  20. int sum = 0;
  22. for (int c = 0; c < 3; ++c) {
  23. for (int y = 0; y < bounds.h; ++y) {
  24. for (int x = 0; x < bounds.w; ++x) {
  25. std::uint8_t orig_val = original(x + bounds.x, y + bounds.y, 0, c);
  26. std::uint8_t img_val = img(x + bounds.x, y + bounds.y, 0, c);
  28. std::int16_t diff = img_val - orig_val;
  29. sum += diff * diff;
  30. }
  31. }
  32. }
  34. return std::sqrt(sum) / bounds.w * bounds.h;
  35. }
  37. static double distance(const CImg<std::uint8_t> &original, std::uint8_t *colour,
  38. Rectangle bounds)
  39. {
  40. int sum = 0;
  42. for (int c = 0; c < 3; ++c) {
  43. for (int y = 0; y < bounds.h; ++y) {
  44. for (int x = 0; x < bounds.w; ++x) {
  45. std::uint8_t orig_val = original(x + bounds.x, y + bounds.y, 0, c);
  46. std::uint8_t img_val = colour[c];
  48. std::int16_t diff = img_val - orig_val;
  49. sum += diff * diff;
  50. }
  51. }
  52. }
  54. return std::sqrt(sum) / bounds.w * bounds.h;
  55. }
  57. int main()
  58. {
  59. std::random_device rd;
  60. std::mt19937 x_gen(rd());
  61. std::mt19937 y_gen(rd());
  62. std::mt19937 dim_gen(rd());
  64. CImg<std::uint8_t> original("test.png");
  65. CImg<std::uint8_t> img(original.width(), original.height(), 1, 3, 255);
  67. std::uniform_int_distribution<int> x_distribution(0, original.width() - 16);
  68. std::uniform_int_distribution<int> y_distribution(0, original.height() - 16);
  69. std::uniform_int_distribution<int> dim_distribution(5, 15);
  71. auto x_rand = std::bind(x_distribution, x_gen);
  72. auto y_rand = std::bind(y_distribution, y_gen);
  73. auto dim_rand = std::bind(dim_distribution, dim_gen);
  75. std::uint8_t colour[3];
  77. CImgDisplay disp(img, "randompic");
  79. unsigned long long iterations = 0;
  80. while (!disp.is_closed()) {
  81. for (int i = 0; i < 100000; ++i) {
  82. iterations++;
  83. Rectangle rect {x_rand(), y_rand(), dim_rand(), dim_rand()};
  84. colour[0] = original(x_rand(), y_rand(), 0, 0);
  85. colour[1] = original(x_rand(), y_rand(), 0, 1);
  86. colour[2] = original(x_rand(), y_rand(), 0, 2);
  88. if (distance(original, img, rect) > distance(original, colour, rect)) {
  89. img.draw_rectangle(rect.x, rect.y, rect.x + rect.w - 1, rect.y + rect.h - 1, colour);
  90. }
  91. }
  92. disp.display(img);
  93. std::printf("%llu\n", iterations);
  94. }
  96. return 0;
  97. }
  98.  
Powered by spacepaste. Paste removal.