Леонтьев Никита. Задача 3. Вариант 30. Повышение контраста полутоново…

…го изображения посредством линейной растяжки гистограммы (#796) Последовательная реализация: Преобразуем изображение в вектор, затем для каждого пикселя вычислим его яркость по формуле среднего арифм. (R+G+B)/3. Линейным растяжением получим новую яркость для каждого пикселя. Результирующий вектор - вектор с повышенным контрастом. MPI реализация: Разделим все пиксели на N, где N - число процессов. Остаток распределим между процессами, используется broadcast, all_reduce, gatherv, scatterv. Далее используется алгоритм из посл. реализации.
learning-process · Dec 29, 2024 · f2d251e · f2d251e
1 parent 2459e8d
commit f2d251e
Show file tree

Hide file tree

Showing 8 changed files with 926 additions and 0 deletions.
diff --git a/tasks/mpi/leontev_n_image_enhancement/func_tests/main.cpp b/tasks/mpi/leontev_n_image_enhancement/func_tests/main.cpp
@@ -0,0 +1,220 @@
+#include <gtest/gtest.h>
+
+#include <algorithm>
+#include <boost/mpi/communicator.hpp>
+#include <boost/mpi/environment.hpp>
+#include <functional>
+#include <random>
+#include <vector>
+
+#include "mpi/leontev_n_image_enhancement/include/ops_mpi.hpp"
+
+namespace leontev_n_image_enhancement_mpi {
+std::vector<int> getRandomImage(int sz) {
+  std::random_device dev;
+  std::mt19937 gen(dev());
+  std::uniform_int_distribution<int> dist(0, 255);
+  std::vector<int> vec(sz);
+  for (int i = 0; i < sz; i++) {
+    vec[i] = dist(gen);
+  }
+  return vec;
+}
+}  // namespace leontev_n_image_enhancement_mpi
+
+inline void taskEmplacement(std::shared_ptr<ppc::core::TaskData> &taskDataPar, std::vector<int> &global_vec,
+                            std::vector<int> &global_sum) {
+  taskDataPar->inputs.emplace_back(reinterpret_cast<uint8_t *>(global_vec.data()));
+  taskDataPar->inputs_count.emplace_back(global_vec.size());
+  taskDataPar->outputs.emplace_back(reinterpret_cast<uint8_t *>(global_sum.data()));
+  taskDataPar->outputs_count.emplace_back(global_sum.size());
+}
+
+TEST(leontev_n_image_enhancement_mpi, test_image_imin_imax) {
+  boost::mpi::communicator world;
+
+  const int vector_size = 6;
+  std::vector<int> in_vec = {255, 0, 255, 255, 0, 255};
+  std::vector<int> out_vec_par(vector_size, 0);
+
+  std::shared_ptr<ppc::core::TaskData> taskDataPar = std::make_shared<ppc::core::TaskData>();
+
+  if (world.rank() == 0) {
+    taskEmplacement(taskDataPar, in_vec, out_vec_par);
+  }
+
+  leontev_n_image_enhancement_mpi::MPIImgEnhancementParallel MPIImgEnhancementParallel(taskDataPar);
+  ASSERT_EQ(MPIImgEnhancementParallel.validation(), true);
+  MPIImgEnhancementParallel.pre_processing();
+  MPIImgEnhancementParallel.run();
+  MPIImgEnhancementParallel.post_processing();
+
+  if (world.rank() == 0) {
+    // Create data
+    std::vector<int> out_vec_seq(vector_size, 0);
+
+    // Create TaskData
+    std::shared_ptr<ppc::core::TaskData> taskDataSeq = std::make_shared<ppc::core::TaskData>();
+    taskEmplacement(taskDataSeq, in_vec, out_vec_seq);
+
+    // Create Task
+    leontev_n_image_enhancement_mpi::MPIImgEnhancementSequential MPIImgEnhancementSequential(taskDataSeq);
+    ASSERT_EQ(MPIImgEnhancementSequential.validation(), true);
+    MPIImgEnhancementSequential.pre_processing();
+    MPIImgEnhancementSequential.run();
+    MPIImgEnhancementSequential.post_processing();
+
+    ASSERT_EQ(out_vec_par, out_vec_seq);
+  }
+}
+
+TEST(leontev_n_image_enhancement_mpi, test_image_1) {
+  boost::mpi::communicator world;
+
+  const int width = 300;
+  const int height = 300;
+  const int vector_size = width * height * 3;
+  std::vector<int> in_vec = leontev_n_image_enhancement_mpi::getRandomImage(vector_size);
+  std::vector<int> out_vec_par(vector_size, 0);
+
+  std::shared_ptr<ppc::core::TaskData> taskDataPar = std::make_shared<ppc::core::TaskData>();
+
+  if (world.rank() == 0) {
+    taskEmplacement(taskDataPar, in_vec, out_vec_par);
+  }
+
+  leontev_n_image_enhancement_mpi::MPIImgEnhancementParallel MPIImgEnhancementParallel(taskDataPar);
+  ASSERT_EQ(MPIImgEnhancementParallel.validation(), true);
+  MPIImgEnhancementParallel.pre_processing();
+  MPIImgEnhancementParallel.run();
+  MPIImgEnhancementParallel.post_processing();
+
+  if (world.rank() == 0) {
+    // Create data
+    std::vector<int> out_vec_seq(vector_size, 0);
+
+    // Create TaskData
+    std::shared_ptr<ppc::core::TaskData> taskDataSeq = std::make_shared<ppc::core::TaskData>();
+    taskEmplacement(taskDataSeq, in_vec, out_vec_seq);
+
+    // Create Task
+    leontev_n_image_enhancement_mpi::MPIImgEnhancementSequential MPIImgEnhancementSequential(taskDataSeq);
+    ASSERT_EQ(MPIImgEnhancementSequential.validation(), true);
+    MPIImgEnhancementSequential.pre_processing();
+    MPIImgEnhancementSequential.run();
+    MPIImgEnhancementSequential.post_processing();
+
+    ASSERT_EQ(out_vec_par, out_vec_seq);
+  }
+}
+
+TEST(leontev_n_image_enhancement_mpi, test_image_2) {
+  boost::mpi::communicator world;
+
+  const int width = 600;
+  const int height = 600;
+  const int vector_size = width * height * 3;
+  std::vector<int> in_vec = leontev_n_image_enhancement_mpi::getRandomImage(vector_size);
+  std::vector<int> out_vec_par(vector_size, 0);
+
+  std::shared_ptr<ppc::core::TaskData> taskDataPar = std::make_shared<ppc::core::TaskData>();
+
+  if (world.rank() == 0) {
+    taskEmplacement(taskDataPar, in_vec, out_vec_par);
+  }
+
+  leontev_n_image_enhancement_mpi::MPIImgEnhancementParallel MPIImgEnhancementParallel(taskDataPar);
+  ASSERT_EQ(MPIImgEnhancementParallel.validation(), true);
+  MPIImgEnhancementParallel.pre_processing();
+  MPIImgEnhancementParallel.run();
+  MPIImgEnhancementParallel.post_processing();
+
+  if (world.rank() == 0) {
+    // Create data
+    std::vector<int> out_vec_seq(vector_size, 0);
+
+    // Create TaskData
+    std::shared_ptr<ppc::core::TaskData> taskDataSeq = std::make_shared<ppc::core::TaskData>();
+    taskEmplacement(taskDataSeq, in_vec, out_vec_seq);
+
+    // Create Task
+    leontev_n_image_enhancement_mpi::MPIImgEnhancementSequential MPIImgEnhancementSequential(taskDataSeq);
+    ASSERT_EQ(MPIImgEnhancementSequential.validation(), true);
+    MPIImgEnhancementSequential.pre_processing();
+    MPIImgEnhancementSequential.run();
+    MPIImgEnhancementSequential.post_processing();
+
+    ASSERT_EQ(out_vec_par, out_vec_seq);
+  }
+}
+
+TEST(leontev_n_image_enhancement_mpi, test_incorrect_image_size) {
+  boost::mpi::communicator world;
+
+  const int vector_size = 5;
+  std::vector<int> in_vec = {0, 10, 10, 10, 60};
+  std::vector<int> out_vec_par(vector_size, 0);
+
+  std::shared_ptr<ppc::core::TaskData> taskDataPar = std::make_shared<ppc::core::TaskData>();
+
+  if (world.rank() == 0) {
+    taskEmplacement(taskDataPar, in_vec, out_vec_par);
+    leontev_n_image_enhancement_mpi::MPIImgEnhancementParallel MPIImgEnhancementParallel(taskDataPar);
+    ASSERT_EQ(MPIImgEnhancementParallel.validation(), false);
+  }
+}
+
+TEST(leontev_n_image_enhancement_mpi, test_incorrect_rgb_image) {
+  boost::mpi::communicator world;
+
+  const int width = 4;
+  const int height = 4;
+  const int vector_size = width * height * 3;
+  std::vector<int> in_vec = leontev_n_image_enhancement_mpi::getRandomImage(vector_size);
+  in_vec[0] = -100;
+  std::vector<int> out_vec_par(vector_size, 0);
+
+  std::shared_ptr<ppc::core::TaskData> taskDataPar = std::make_shared<ppc::core::TaskData>();
+
+  if (world.rank() == 0) {
+    taskEmplacement(taskDataPar, in_vec, out_vec_par);
+    leontev_n_image_enhancement_mpi::MPIImgEnhancementParallel MPIImgEnhancementParallel(taskDataPar);
+    ASSERT_EQ(MPIImgEnhancementParallel.validation(), false);
+  }
+}
+
+TEST(leontev_n_image_enhancement_mpi, test_incorrect_rgb_image2) {
+  boost::mpi::communicator world;
+
+  const int width = 2;
+  const int height = 3;
+  const int vector_size = width * height * 3;
+  std::vector<int> in_vec = {-2, -10, -20, 0, 555, -25, 50, 265, 0, 0, 0, 0, -22, 0, 1, 4, 105, 90};
+  std::vector<int> out_vec_par(vector_size, 0);
+
+  std::shared_ptr<ppc::core::TaskData> taskDataPar = std::make_shared<ppc::core::TaskData>();
+
+  if (world.rank() == 0) {
+    taskEmplacement(taskDataPar, in_vec, out_vec_par);
+    leontev_n_image_enhancement_mpi::MPIImgEnhancementParallel MPIImgEnhancementParallel(taskDataPar);
+    ASSERT_EQ(MPIImgEnhancementParallel.validation(), false);
+  }
+}
+
+TEST(leontev_n_image_enhancement_mpi, test_incorrect_empty_image) {
+  boost::mpi::communicator world;
+
+  const int width = 0;
+  const int height = 0;
+  const int vector_size = width * height * 3;
+  std::vector<int> in_vec = {};
+  std::vector<int> out_vec_par(vector_size, 0);
+
+  std::shared_ptr<ppc::core::TaskData> taskDataPar = std::make_shared<ppc::core::TaskData>();
+
+  if (world.rank() == 0) {
+    taskEmplacement(taskDataPar, in_vec, out_vec_par);
+    leontev_n_image_enhancement_mpi::MPIImgEnhancementParallel MPIImgEnhancementParallel(taskDataPar);
+    ASSERT_EQ(MPIImgEnhancementParallel.validation(), false);
+  }
+}
diff --git a/tasks/mpi/leontev_n_image_enhancement/include/ops_mpi.hpp b/tasks/mpi/leontev_n_image_enhancement/include/ops_mpi.hpp
@@ -0,0 +1,44 @@
+#pragma once
+
+#include <gtest/gtest.h>
+
+#include <boost/mpi/collectives.hpp>
+#include <boost/mpi/communicator.hpp>
+#include <memory>
+#include <numeric>
+#include <string>
+#include <utility>
+#include <vector>
+
+#include "core/task/include/task.hpp"
+
+namespace leontev_n_image_enhancement_mpi {
+
+class MPIImgEnhancementSequential : public ppc::core::Task {
+ public:
+  explicit MPIImgEnhancementSequential(std::shared_ptr<ppc::core::TaskData> taskData_) : Task(std::move(taskData_)) {}
+  bool pre_processing() override;
+  bool validation() override;
+  bool run() override;
+  bool post_processing() override;
+
+ private:
+  std::vector<int> I;
+  std::vector<int> image_input;
+  std::vector<int> image_output;
+};
+
+class MPIImgEnhancementParallel : public ppc::core::Task {
+ public:
+  explicit MPIImgEnhancementParallel(std::shared_ptr<ppc::core::TaskData> taskData_) : Task(std::move(taskData_)) {}
+  bool pre_processing() override;
+  bool validation() override;
+  bool run() override;
+  bool post_processing() override;
+
+ private:
+  std::vector<int> image_input;
+  std::vector<int> image_output;
+  boost::mpi::communicator world;
+};
+}  // namespace leontev_n_image_enhancement_mpi