arm_compute v18.05

[platform/upstream/armcl.git] / examples / neon_scale.cpp

diff --git a/examples/neon_scale.cpp b/examples/neon_scale.cpp
index db255f4..bec277d 100644 (file)
--- a/examples/neon_scale.cpp
+++ b/examples/neon_scale.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2016, 2017 ARM Limited.
+ * Copyright (c) 2016, 2018 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -24,66 +24,80 @@
 #include "arm_compute/runtime/NEON/NEFunctions.h"
 
 #include "arm_compute/core/Types.h"
-#include "test_helpers/Utils.h"
+#include "utils/Utils.h"
 
 using namespace arm_compute;
-using namespace test_helpers;
+using namespace utils;
 
-void main_neon_scale(int argc, const char **argv)
+class NEONScaleExample : public Example
 {
-    PPMLoader ppm;
-    Image     src, dst;
-
-    if(argc < 2)
-    {
-        // Print help
-        std::cout << "Usage: ./build/neon_scale[input_image.ppm]\n\n";
-        std::cout << "No input_image provided, creating a dummy 640x480 image\n";
-        // Create an empty grayscale 640x480 image
-        src.allocator()->init(TensorInfo(640, 480, Format::U8));
-    }
-    else
+public:
+    void do_setup(int argc, char **argv) override
     {
-        ppm.open(argv[1]);
-        ppm.init_image(src, Format::U8);
-    }
+        PPMLoader ppm;
 
-    constexpr int scale_factor = 2;
+        if(argc < 2)
+        {
+            // Print help
+            std::cout << "Usage: ./build/neon_scale[input_image.ppm]\n\n";
+            std::cout << "No input_image provided, creating a dummy 640x480 image\n";
+            // Create an empty grayscale 640x480 image
+            src.allocator()->init(TensorInfo(640, 480, Format::U8));
+        }
+        else
+        {
+            ppm.open(argv[1]);
+            ppm.init_image(src, Format::U8);
+        }
 
-    TensorInfo dst_tensor_info(src.info()->dimension(0) / scale_factor, src.info()->dimension(1) / scale_factor, Format::U8);
+        constexpr int scale_factor = 2;
 
-    // Configure the destination image
-    dst.allocator()->init(dst_tensor_info);
+        TensorInfo dst_tensor_info(src.info()->dimension(0) / scale_factor, src.info()->dimension(1) / scale_factor,
+                                   Format::U8);
 
-    // Create and initialize a Scale function object:
-    NEScale scale;
-    scale.configure(&src, &dst, InterpolationPolicy::NEAREST_NEIGHBOR, BorderMode::UNDEFINED);
+        // Configure the destination image
+        dst.allocator()->init(dst_tensor_info);
 
-    // Allocate all the images
-    src.allocator()->allocate();
-    dst.allocator()->allocate();
-    // Fill the input image with the content of the PPM image if a filename was provided:
-    if(ppm.is_open())
-    {
-        ppm.fill_image(src);
-    }
+        // Configure Scale function object:
+        scale.configure(&src, &dst, InterpolationPolicy::NEAREST_NEIGHBOR, BorderMode::UNDEFINED);
 
-    // Run the scale operation:
-    scale.run();
+        // Allocate all the images
+        src.allocator()->allocate();
+        dst.allocator()->allocate();
 
-    // Save the result to file:
-    if(ppm.is_open())
+        // Fill the input image with the content of the PPM image if a filename was provided:
+        if(ppm.is_open())
+        {
+            ppm.fill_image(src);
+            output_filename = std::string(argv[1]) + "_out.ppm";
+        }
+    }
+    void do_run() override
     {
-        const std::string output_filename = std::string(argv[1]) + "_out.ppm";
-        save_to_ppm(dst, output_filename);
+        // Run the scale operation:
+        scale.run();
     }
-}
+    void do_teardown() override
+    {
+        // Save the result to file:
+        if(!output_filename.empty())
+        {
+            save_to_ppm(dst, output_filename); // save_to_ppm maps and unmaps the image to store as PPM
+        }
+    }
+
+private:
+    Image       src{}, dst{};
+    NEScale     scale{};
+    std::string output_filename{};
+};
 
 /** Main program for convolution test
+ *
  * @param[in] argc Number of arguments
  * @param[in] argv Arguments ( [optional] Path to PPM image to process )
  */
-int main(int argc, const char **argv)
+int main(int argc, char **argv)
 {
-    return test_helpers::run_example(argc, argv, main_neon_scale);
+    return utils::run_example<NEONScaleExample>(argc, argv);
 }