Samples - updates & fixes (#47)

* Samples - updates & fixes

* use std::cerr

samples/jpegDecode/jpegdecode.cpp

@@ -91,7 +91,15 @@ int main(int argc, char **argv) {
         std::cout << "Input file name: " << base_file_name << std::endl;
         std::cout << "Input image resolution: " << widths[0] << "x" << heights[0] << std::endl;
         std::cout << "Chroma subsampling: " + chroma_sub_sampling  << std::endl;
-        if (subsampling == ROCJPEG_CSS_411) {
+        if (widths[0] < 64 || heights[0] < 64) {
+            std::cerr << "The image resolution is not supported by VCN Hardware" << std::endl;
+            if (is_dir) {
+                std::cout << std::endl;
+                continue;
+            } else
+                return EXIT_FAILURE;
+        }
+        if (subsampling == ROCJPEG_CSS_411 || subsampling == ROCJPEG_CSS_UNKNOWN) {
             std::cerr << "The chroma sub-sampling is not supported by VCN Hardware" << std::endl;
             if (is_dir) {
                 std::cout << std::endl;

samples/jpegDecodeBatched/jpegdecodebatched.cpp

@@ -104,7 +104,16 @@ int main(int argc, char **argv) {
             CHECK_ROCJPEG(rocJpegGetImageInfo(rocjpeg_handle, rocjpeg_stream_handles[index], &num_components, &subsamplings[index], widths[index].data(), heights[index].data()));
 
             rocjpeg_utils.GetChromaSubsamplingStr(subsamplings[index], chroma_sub_sampling);
-            if (subsamplings[index] == ROCJPEG_CSS_411) {
+            if (widths[index][0] < 64 || heights[index][0] < 64) {
+                std::cerr << "The image resolution is not supported by VCN Hardware" << std::endl;
+                if (is_dir) {
+                    std::cout << std::endl;
+                    continue;
+                } else
+                    return EXIT_FAILURE;
+            }
+
+            if (subsamplings[index] == ROCJPEG_CSS_411 || subsamplings[index] == ROCJPEG_CSS_UNKNOWN) {
                 std::cerr << "The chroma sub-sampling is not supported by VCN Hardware" << std::endl;
                 if (is_dir) {
                     std::cout << std::endl;

samples/jpegDecodeMultiThreads/jpegdecodemultithreads.cpp

@@ -71,15 +71,21 @@ void ThreadFunction(std::vector<std::string>& jpegFiles, RocJpegHandle rocjpeg_h
 
         CHECK_ROCJPEG(rocJpegStreamParse(reinterpret_cast<uint8_t *>(file_data.data()), file_size, rocjpeg_stream));
         CHECK_ROCJPEG(rocJpegGetImageInfo(rocjpeg_handle, rocjpeg_stream, &num_components, &subsampling, widths, heights));
-        if (subsampling == ROCJPEG_CSS_411) {
-            std::cout << "The chroma sub-sampling is not supported by VCN Hardware" << std::endl;
+        if (widths[0] < 64 || heights[0] < 64) {
+            std::cerr << "The image resolution is not supported by VCN Hardware" << std::endl;
+            std::cout << "Skipping decoding file " << base_file_name << std::endl;
+            return;
+        }
+
+        if (subsampling == ROCJPEG_CSS_411 || subsampling == ROCJPEG_CSS_UNKNOWN) {
+            std::cerr << "The chroma sub-sampling is not supported by VCN Hardware" << std::endl;
             std::cout << "Skipping decoding file " << base_file_name << std::endl;
             return;
         }
 
         if (rocjpeg_util.GetChannelPitchAndSizes(decode_params, subsampling, widths, heights, num_channels, *output_image, channel_sizes)) {
-                std::cerr << "ERROR: Failed to get the channel pitch and sizes" << std::endl;
-                return;
+            std::cerr << "ERROR: Failed to get the channel pitch and sizes" << std::endl;
+            return;
         }
 
         // allocate memory for each channel