AI/rocm-systems
2
0
Garpu 0
Kode Masalah Tarik Permintaan Actions Packages Projects Rilis Wiki Kegiatan

Use the standard .yuv and .rgb file extensions when saving the decoded jpeg outputs (#34)

Telusuri Sumber 
* Use the standard .yuv and .rgb file extensions when saving the decoded jpeg outputs

* modify the file description for saving
This commit is contained in:
Aryan Salmanpour
2024-06-26 09:00:45 -04:00
committed by GitHub
orang tua c660aeab43
melakukan 44f74c4958
4 mengubah file dengan 33 tambahan dan 8 penghapusan
Download Patch File Download Diff File Expand all files Collapse all files
samples/jpegDecode/jpegdecode.cpp
+1 -1
Melihat File
@@ -127,7 +127,7 @@ int main(int argc, char **argv) {
if (save_images) {
std::string image_save_path = output_file_path;
if (is_dir) {
rocjpeg_utils.GetOutputFileExt(decode_params.output_format, base_file_name, widths[0], heights[0], image_save_path);
rocjpeg_utils.GetOutputFileExt(decode_params.output_format, base_file_name, widths[0], heights[0], subsampling, image_save_path);
}
rocjpeg_utils.SaveImage(image_save_path, &output_image, widths[0], heights[0], subsampling, decode_params.output_format);
}
samples/jpegDecodeBatched/jpegdecodebatched.cpp
+1 -1
Melihat File
@@ -148,7 +148,7 @@ int main(int argc, char **argv) {
for (int b = 0; b < current_batch_size; b++) {
std::string image_save_path = output_file_path;
if (is_dir) {
rocjpeg_utils.GetOutputFileExt(decode_params.output_format, base_file_names[b], widths[b][0], heights[b][0], image_save_path);
rocjpeg_utils.GetOutputFileExt(decode_params.output_format, base_file_names[b], widths[b][0], heights[b][0], subsamplings[b], image_save_path);
}
rocjpeg_utils.SaveImage(image_save_path, &output_images[b], widths[b][0], heights[b][0], subsamplings[b], decode_params.output_format);
}
samples/jpegDecodeMultiThreads/jpegdecodemultithreads.cpp
+1 -1
Melihat File
@@ -99,7 +99,7 @@ void ThreadFunction(std::vector<std::string>& jpegFiles, RocJpegHandle rocjpeg_h
if (save_images) {
std::string image_save_path = output_file_path;
rocjpeg_util.GetOutputFileExt(decode_params.output_format, base_file_name, widths[0], heights[0], image_save_path);
rocjpeg_util.GetOutputFileExt(decode_params.output_format, base_file_name, widths[0], heights[0], subsampling, image_save_path);
rocjpeg_util.SaveImage(image_save_path, output_image, widths[0], heights[0], subsampling, decode_params.output_format);
}
samples/rocjpeg_samples_utils.h
+30 -5
Melihat File
@@ -360,32 +360,57 @@ public:
* @param image_height The image height.
* @param file_name_for_saving The string to store the file name for saving.
*/
void GetOutputFileExt(RocJpegOutputFormat output_format, std::string &base_file_name, uint32_t image_width, uint32_t image_height, std::string &file_name_for_saving) {
void GetOutputFileExt(RocJpegOutputFormat output_format, std::string &base_file_name, uint32_t image_width, uint32_t image_height, RocJpegChromaSubsampling subsampling, std::string &file_name_for_saving) {
std::string file_extension;
std::string::size_type const p(base_file_name.find_last_of('.'));
std::string file_name_no_ext = base_file_name.substr(0, p);
std::string format_description = "";
switch (output_format) {
case ROCJPEG_OUTPUT_NATIVE:
file_extension = "native";
file_extension = "yuv";
switch (subsampling) {
case ROCJPEG_CSS_444:
format_description = "444";
break;
case ROCJPEG_CSS_440:
format_description = "440";
break;
case ROCJPEG_CSS_422:
format_description = "422_yuyv";
break;
case ROCJPEG_CSS_420:
format_description = "nv12";
break;
case ROCJPEG_CSS_400:
format_description = "400";
break;
default:
std::cout << "Unknown chroma subsampling!" << std::endl;
return;
}
break;
case ROCJPEG_OUTPUT_YUV_PLANAR:
file_extension = "yuv";
format_description = "planar";
break;
case ROCJPEG_OUTPUT_Y:
file_extension = "y";
file_extension = "yuv";
format_description = "400";
break;
case ROCJPEG_OUTPUT_RGB:
file_extension = "rgb";
format_description = "packed";
break;
case ROCJPEG_OUTPUT_RGB_PLANAR:
file_extension = "rgb_planar";
file_extension = "rgb";
format_description = "planar";
break;
default:
file_extension = "";
break;
}
file_name_for_saving += "//" + file_name_no_ext + "_" + std::to_string(image_width) + "x"
+ std::to_string(image_height) + "." + file_extension;
+ std::to_string(image_height) + "_" + format_description + "." + file_extension;
}
/**