| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613 |
- // This file is part of OpenCV project.
- // It is subject to the license terms in the LICENSE file found in the top-level
- // directory of this distribution and at http://opencv.org/license.html
- #include <string>
- #include <vector>
- #include "test_precomp.hpp"
- namespace opencv_test { namespace {
- static Mat makeCirclesImage(Size size, int type, int nbits)
- {
- Mat img(size, type);
- img.setTo(Scalar::all(0));
- RNG& rng = theRNG();
- int maxval = (int)(1 << nbits);
- for (int i = 0; i < 100; i++) {
- int x = rng.uniform(0, img.cols);
- int y = rng.uniform(0, img.rows);
- int radius = rng.uniform(5, std::min(img.cols, img.rows) / 5);
- int b = rng.uniform(0, maxval);
- int g = rng.uniform(0, maxval);
- int r = rng.uniform(0, maxval);
- circle(img, Point(x, y), radius, Scalar(b, g, r), -1, LINE_AA);
- }
- return img;
- }
- static std::vector<uchar> getSampleExifData() {
- return {
- 'M', 'M', 0, '*', 0, 0, 0, 8, 0, 10, 1, 0, 0, 4, 0, 0, 0, 1, 0, 0, 5,
- 0, 1, 1, 0, 4, 0, 0, 0, 1, 0, 0, 2, 208, 1, 2, 0, 3, 0, 0, 0, 1,
- 0, 10, 0, 0, 1, 18, 0, 3, 0, 0, 0, 1, 0, 1, 0, 0, 1, 14, 0, 2, 0, 0,
- 0, '"', 0, 0, 0, 176, 1, '1', 0, 2, 0, 0, 0, 7, 0, 0, 0, 210, 1, 26,
- 0, 5, 0, 0, 0, 1, 0, 0, 0, 218, 1, 27, 0, 5, 0, 0, 0, 1, 0, 0, 0,
- 226, 1, '(', 0, 3, 0, 0, 0, 1, 0, 2, 0, 0, 135, 'i', 0, 4, 0, 0, 0,
- 1, 0, 0, 0, 134, 0, 0, 0, 0, 0, 3, 144, 0, 0, 7, 0, 0, 0, 4, '0', '2',
- '2', '1', 160, 2, 0, 4, 0, 0, 0, 1, 0, 0, 5, 0, 160, 3, 0, 4, 0, 0,
- 0, 1, 0, 0, 2, 208, 0, 0, 0, 0, 'S', 'a', 'm', 'p', 'l', 'e', ' ', '1', '0',
- '-', 'b', 'i', 't', ' ', 'i', 'm', 'a', 'g', 'e', ' ', 'w', 'i', 't', 'h', ' ',
- 'm', 'e', 't', 'a', 'd', 'a', 't', 'a', 0, 'O', 'p', 'e', 'n', 'C', 'V', 0, 0,
- 0, 0, 0, 'H', 0, 0, 0, 1, 0, 0, 0, 'H', 0, 0, 0, 1
- };
- }
- static std::vector<uchar> getSampleXmpData() {
- return {
- '<','x',':','x','m','p','m','e','t','a',' ','x','m','l','n','s',':','x','=',
- '"','a','d','o','b','e',':','x','m','p','"','>',
- '<','x','m','p',':','C','r','e','a','t','o','r','T','o','o','l','>',
- 'O','p','e','n','C','V',
- '<','/','x','m','p',':','C','r','e','a','t','o','r','T','o','o','l','>',
- '<','/','x',':','x','m','p','m','e','t','a','>',0
- };
- }
- // Returns a Minimal ICC profile data (Generated with help from ChatGPT)
- static std::vector<uchar> getSampleIccpData() {
- std::vector<uchar> iccp_data(192, 0);
- iccp_data[3] = 192; // Profile size: 192 bytes
- iccp_data[12] = 'm';
- iccp_data[13] = 'n';
- iccp_data[14] = 't';
- iccp_data[15] = 'r';
- iccp_data[16] = 'R';
- iccp_data[17] = 'G';
- iccp_data[18] = 'B';
- iccp_data[19] = ' ';
- iccp_data[20] = 'X';
- iccp_data[21] = 'Y';
- iccp_data[22] = 'Z';
- iccp_data[23] = ' ';
- // File signature 'acsp' at offset 36 (0x24)
- iccp_data[36] = 'a';
- iccp_data[37] = 'c';
- iccp_data[38] = 's';
- iccp_data[39] = 'p';
- // Illuminant D50 at offset 68 (0x44), example values:
- iccp_data[68] = 0x00;
- iccp_data[69] = 0x00;
- iccp_data[70] = 0xF6;
- iccp_data[71] = 0xD6; // 0.9642
- iccp_data[72] = 0x00;
- iccp_data[73] = 0x01;
- iccp_data[74] = 0x00;
- iccp_data[75] = 0x00; // 1.0
- iccp_data[76] = 0x00;
- iccp_data[77] = 0x00;
- iccp_data[78] = 0xD3;
- iccp_data[79] = 0x2D; // 0.8249
- // Tag count at offset 128 (0x80) = 1
- iccp_data[131] = 1;
- // Tag record at offset 132 (0x84): signature 'desc', offset 128, size 64
- iccp_data[132] = 'd';
- iccp_data[133] = 'e';
- iccp_data[134] = 's';
- iccp_data[135] = 'c';
- iccp_data[139] = 128; // offset
- iccp_data[143] = 64; // size
- // Tag data 'desc' at offset 128 (start of tag data)
- // Set type 'desc' etc. here, for simplicity fill zeros
- iccp_data[144] = 'd';
- iccp_data[145] = 'e';
- iccp_data[146] = 's';
- iccp_data[147] = 'c';
- // ASCII string length at offset 156
- iccp_data[156] = 20; // length
- // ASCII string "Minimal ICC Profile" starting at offset 160
- iccp_data[160] = 'M';
- iccp_data[161] = 'i';
- iccp_data[162] = 'n';
- iccp_data[163] = 'i';
- iccp_data[164] = 'm';
- iccp_data[165] = 'a';
- iccp_data[166] = 'l';
- iccp_data[167] = ' ';
- iccp_data[168] = 'I';
- iccp_data[169] = 'C';
- iccp_data[170] = 'C';
- iccp_data[171] = ' ';
- iccp_data[172] = 'P';
- iccp_data[173] = 'r';
- iccp_data[174] = 'o';
- iccp_data[175] = 'f';
- iccp_data[176] = 'i';
- iccp_data[177] = 'l';
- iccp_data[178] = 'e';
- return iccp_data;
- }
- /**
- * Test to check whether the EXIF orientation tag was processed successfully or not.
- * The test uses a set of 8 images named testExifOrientation_{1 to 8}.(extension).
- * Each test image is a 10x10 square, divided into four smaller sub-squares:
- * (R corresponds to Red, G to Green, B to Blue, W to White)
- * --------- ---------
- * | R | G | | G | R |
- * |-------| - (tag 1) |-------| - (tag 2)
- * | B | W | | W | B |
- * --------- ---------
- *
- * --------- ---------
- * | W | B | | B | W |
- * |-------| - (tag 3) |-------| - (tag 4)
- * | G | R | | R | G |
- * --------- ---------
- *
- * --------- ---------
- * | R | B | | G | W |
- * |-------| - (tag 5) |-------| - (tag 6)
- * | G | W | | R | B |
- * --------- ---------
- *
- * --------- ---------
- * | W | G | | B | R |
- * |-------| - (tag 7) |-------| - (tag 8)
- * | B | R | | W | G |
- * --------- ---------
- *
- *
- * Each image contains an EXIF field with an orientation tag (0x112).
- * After reading each image and applying the orientation tag,
- * the resulting image should be:
- * ---------
- * | R | G |
- * |-------|
- * | B | W |
- * ---------
- *
- * Note:
- * The flags parameter of the imread function is set as IMREAD_COLOR | IMREAD_ANYCOLOR | IMREAD_ANYDEPTH.
- * Using this combination is an undocumented trick to load images similarly to the IMREAD_UNCHANGED flag,
- * preserving the alpha channel (if present) while also applying the orientation.
- */
- typedef testing::TestWithParam<string> Exif;
- TEST_P(Exif, exif_orientation)
- {
- const string root = cvtest::TS::ptr()->get_data_path();
- const string filename = root + GetParam();
- const int colorThresholdHigh = 250;
- const int colorThresholdLow = 5;
- // Refer to the note in the explanation above.
- Mat m_img = imread(filename, IMREAD_COLOR | IMREAD_ANYCOLOR | IMREAD_ANYDEPTH);
- ASSERT_FALSE(m_img.empty());
- if (m_img.channels() == 3)
- {
- Vec3b vec;
- //Checking the first quadrant (with supposed red)
- vec = m_img.at<Vec3b>(2, 2); //some point inside the square
- EXPECT_LE(vec.val[0], colorThresholdLow);
- EXPECT_LE(vec.val[1], colorThresholdLow);
- EXPECT_GE(vec.val[2], colorThresholdHigh);
- //Checking the second quadrant (with supposed green)
- vec = m_img.at<Vec3b>(2, 7); //some point inside the square
- EXPECT_LE(vec.val[0], colorThresholdLow);
- EXPECT_GE(vec.val[1], colorThresholdHigh);
- EXPECT_LE(vec.val[2], colorThresholdLow);
- //Checking the third quadrant (with supposed blue)
- vec = m_img.at<Vec3b>(7, 2); //some point inside the square
- EXPECT_GE(vec.val[0], colorThresholdHigh);
- EXPECT_LE(vec.val[1], colorThresholdLow);
- EXPECT_LE(vec.val[2], colorThresholdLow);
- }
- else
- {
- Vec4b vec;
- //Checking the first quadrant (with supposed red)
- vec = m_img.at<Vec4b>(2, 2); //some point inside the square
- EXPECT_LE(vec.val[0], colorThresholdLow);
- EXPECT_LE(vec.val[1], colorThresholdLow);
- EXPECT_GE(vec.val[2], colorThresholdHigh);
- //Checking the second quadrant (with supposed green)
- vec = m_img.at<Vec4b>(2, 7); //some point inside the square
- EXPECT_LE(vec.val[0], colorThresholdLow);
- EXPECT_GE(vec.val[1], colorThresholdHigh);
- EXPECT_LE(vec.val[2], colorThresholdLow);
- //Checking the third quadrant (with supposed blue)
- vec = m_img.at<Vec4b>(7, 2); //some point inside the square
- EXPECT_GE(vec.val[0], colorThresholdHigh);
- EXPECT_LE(vec.val[1], colorThresholdLow);
- EXPECT_LE(vec.val[2], colorThresholdLow);
- }
- }
- const std::vector<std::string> exif_files
- {
- #ifdef HAVE_JPEG
- "readwrite/testExifOrientation_1.jpg",
- "readwrite/testExifOrientation_2.jpg",
- "readwrite/testExifOrientation_3.jpg",
- "readwrite/testExifOrientation_4.jpg",
- "readwrite/testExifOrientation_5.jpg",
- "readwrite/testExifOrientation_6.jpg",
- "readwrite/testExifOrientation_7.jpg",
- "readwrite/testExifOrientation_8.jpg",
- #endif
- #ifdef OPENCV_IMGCODECS_PNG_WITH_EXIF
- "readwrite/testExifOrientation_1.png",
- "readwrite/testExifOrientation_2.png",
- "readwrite/testExifOrientation_3.png",
- "readwrite/testExifOrientation_4.png",
- "readwrite/testExifOrientation_5.png",
- "readwrite/testExifOrientation_6.png",
- "readwrite/testExifOrientation_7.png",
- "readwrite/testExifOrientation_8.png",
- #endif
- #ifdef HAVE_AVIF
- "readwrite/testExifOrientation_1.avif",
- "readwrite/testExifOrientation_2.avif",
- "readwrite/testExifOrientation_3.avif",
- "readwrite/testExifOrientation_4.avif",
- "readwrite/testExifOrientation_5.avif",
- "readwrite/testExifOrientation_6.avif",
- "readwrite/testExifOrientation_7.avif",
- "readwrite/testExifOrientation_8.avif",
- #endif
- #ifdef HAVE_WEBP
- "readwrite/testExifOrientation_1.webp",
- "readwrite/testExifOrientation_2.webp",
- "readwrite/testExifOrientation_3.webp",
- "readwrite/testExifOrientation_4.webp",
- "readwrite/testExifOrientation_5.webp",
- "readwrite/testExifOrientation_6.webp",
- "readwrite/testExifOrientation_7.webp",
- "readwrite/testExifOrientation_8.webp",
- #endif
- };
- INSTANTIATE_TEST_CASE_P(Imgcodecs, Exif,
- testing::ValuesIn(exif_files));
- #ifdef HAVE_AVIF
- typedef testing::TestWithParam<int> MatChannels;
- TEST_P(MatChannels, Imgcodecs_Avif_ReadWriteWithExif)
- {
- int avif_nbits = 10;
- int avif_speed = 10;
- int avif_quality = 85;
- int imgdepth = avif_nbits > 8 ? CV_16U : CV_8U;
- int imgtype = CV_MAKETYPE(imgdepth, GetParam());
- const string outputname = cv::tempfile(".avif");
- Mat img = makeCirclesImage(Size(1280, 720), imgtype, avif_nbits);
- std::vector<int> metadata_types = {IMAGE_METADATA_EXIF};
- std::vector<std::vector<uchar>> metadata = {
- getSampleExifData() };
- std::vector<int> write_params = {
- IMWRITE_AVIF_DEPTH, avif_nbits,
- IMWRITE_AVIF_SPEED, avif_speed,
- IMWRITE_AVIF_QUALITY, avif_quality
- };
- imwriteWithMetadata(outputname, img, metadata_types, metadata, write_params);
- std::vector<uchar> compressed;
- imencodeWithMetadata(outputname, img, metadata_types, metadata, compressed, write_params);
- std::vector<int> read_metadata_types, read_metadata_types2;
- std::vector<std::vector<uchar> > read_metadata, read_metadata2;
- Mat img2 = imreadWithMetadata(outputname, read_metadata_types, read_metadata, IMREAD_UNCHANGED);
- Mat img3 = imdecodeWithMetadata(compressed, read_metadata_types2, read_metadata2, IMREAD_UNCHANGED);
- EXPECT_EQ(img2.cols, img.cols);
- EXPECT_EQ(img2.rows, img.rows);
- EXPECT_EQ(img2.type(), imgtype);
- EXPECT_EQ(read_metadata_types, read_metadata_types2);
- ASSERT_GE(read_metadata_types.size(), 1u);
- EXPECT_EQ(read_metadata, read_metadata2);
- EXPECT_EQ(read_metadata_types[0], IMAGE_METADATA_EXIF);
- EXPECT_EQ(read_metadata_types.size(), read_metadata.size());
- EXPECT_EQ(read_metadata[0], metadata[0]);
- EXPECT_EQ(cv::norm(img2, img3, NORM_INF), 0.);
- double mse = cv::norm(img, img2, NORM_L2SQR)/(img.rows*img.cols);
- EXPECT_LT(mse, 1500);
- remove(outputname.c_str());
- }
- INSTANTIATE_TEST_CASE_P(Imgcodecs, MatChannels,
- testing::Values(1,3,4));
- #endif // HAVE_AVIF
- #ifdef HAVE_WEBP
- TEST(Imgcodecs_WebP, Read_Write_With_Exif_Xmp_Iccp)
- {
- int imgtype = CV_MAKETYPE(CV_8U, 3);
- const std::string outputname = cv::tempfile(".webp");
- cv::Mat img = makeCirclesImage(cv::Size(160, 120), imgtype, 8);
- std::vector<int> metadata_types = {IMAGE_METADATA_EXIF, IMAGE_METADATA_XMP, IMAGE_METADATA_ICCP};
- std::vector<std::vector<uchar>> metadata = {
- getSampleExifData(),
- getSampleXmpData(),
- getSampleIccpData()
- };
- int webp_quality = 101; // 101 is lossless compression
- std::vector<int> write_params = {IMWRITE_WEBP_QUALITY, webp_quality};
- imwriteWithMetadata(outputname, img, metadata_types, metadata, write_params);
- std::vector<uchar> compressed;
- imencodeWithMetadata(outputname, img, metadata_types, metadata, compressed, write_params);
- std::vector<int> read_metadata_types, read_metadata_types2;
- std::vector<std::vector<uchar>> read_metadata, read_metadata2;
- cv::Mat img2 = imreadWithMetadata(outputname, read_metadata_types, read_metadata, cv::IMREAD_UNCHANGED);
- cv::Mat img3 = imdecodeWithMetadata(compressed, read_metadata_types2, read_metadata2, cv::IMREAD_UNCHANGED);
- EXPECT_EQ(img2.cols, img.cols);
- EXPECT_EQ(img2.rows, img.rows);
- EXPECT_EQ(img2.type(), imgtype);
- EXPECT_EQ(read_metadata_types, read_metadata_types2);
- EXPECT_EQ(read_metadata_types.size(), 3u);
- EXPECT_EQ(read_metadata, read_metadata2);
- EXPECT_EQ(read_metadata, metadata);
- EXPECT_EQ(cv::norm(img2, img3, cv::NORM_INF), 0.0);
- double mse = cv::norm(img, img2, cv::NORM_L2SQR) / (img.rows * img.cols);
- EXPECT_EQ(mse, 0);
- remove(outputname.c_str());
- }
- #endif // HAVE_WEBP
- TEST(Imgcodecs_Jpeg, Read_Write_With_Exif)
- {
- int jpeg_quality = 95;
- int imgtype = CV_MAKETYPE(CV_8U, 3);
- const string outputname = cv::tempfile(".jpeg");
- Mat img = makeCirclesImage(Size(1280, 720), imgtype, 8);
- std::vector<int> metadata_types = {IMAGE_METADATA_EXIF};
- std::vector<std::vector<uchar>> metadata = {
- getSampleExifData() };
- std::vector<int> write_params = {
- IMWRITE_JPEG_QUALITY, jpeg_quality
- };
- imwriteWithMetadata(outputname, img, metadata_types, metadata, write_params);
- std::vector<uchar> compressed;
- imencodeWithMetadata(outputname, img, metadata_types, metadata, compressed, write_params);
- std::vector<int> read_metadata_types, read_metadata_types2;
- std::vector<std::vector<uchar> > read_metadata, read_metadata2;
- Mat img2 = imreadWithMetadata(outputname, read_metadata_types, read_metadata, IMREAD_UNCHANGED);
- Mat img3 = imdecodeWithMetadata(compressed, read_metadata_types2, read_metadata2, IMREAD_UNCHANGED);
- EXPECT_EQ(img2.cols, img.cols);
- EXPECT_EQ(img2.rows, img.rows);
- EXPECT_EQ(img2.type(), imgtype);
- EXPECT_EQ(read_metadata_types, read_metadata_types2);
- EXPECT_GE(read_metadata_types.size(), 1u);
- EXPECT_EQ(read_metadata, read_metadata2);
- EXPECT_EQ(read_metadata_types[0], IMAGE_METADATA_EXIF);
- EXPECT_EQ(read_metadata_types.size(), read_metadata.size());
- EXPECT_EQ(read_metadata[0], metadata[0]);
- EXPECT_EQ(cv::norm(img2, img3, NORM_INF), 0.);
- double mse = cv::norm(img, img2, NORM_L2SQR)/(img.rows*img.cols);
- EXPECT_LT(mse, 80);
- remove(outputname.c_str());
- }
- TEST(Imgcodecs_Png, Read_Write_With_Exif)
- {
- int png_compression = 3;
- int imgtype = CV_MAKETYPE(CV_8U, 3);
- const string outputname = cv::tempfile(".png");
- Mat img = makeCirclesImage(Size(160, 120), imgtype, 8);
- std::vector<int> metadata_types = {IMAGE_METADATA_EXIF};
- std::vector<std::vector<uchar>> metadata = {
- getSampleExifData() };
- std::vector<int> write_params = {
- IMWRITE_PNG_COMPRESSION, png_compression
- };
- imwriteWithMetadata(outputname, img, metadata_types, metadata, write_params);
- std::vector<uchar> compressed;
- imencodeWithMetadata(outputname, img, metadata_types, metadata, compressed, write_params);
- std::vector<int> read_metadata_types, read_metadata_types2;
- std::vector<std::vector<uchar> > read_metadata, read_metadata2;
- Mat img2 = imreadWithMetadata(outputname, read_metadata_types, read_metadata, IMREAD_UNCHANGED);
- Mat img3 = imdecodeWithMetadata(compressed, read_metadata_types2, read_metadata2, IMREAD_UNCHANGED);
- EXPECT_EQ(img2.cols, img.cols);
- EXPECT_EQ(img2.rows, img.rows);
- EXPECT_EQ(img2.type(), imgtype);
- EXPECT_EQ(read_metadata_types, read_metadata_types2);
- ASSERT_GE(read_metadata_types.size(), 1u);
- EXPECT_EQ(read_metadata, read_metadata2);
- EXPECT_EQ(read_metadata_types[0], IMAGE_METADATA_EXIF);
- EXPECT_EQ(read_metadata_types.size(), read_metadata.size());
- EXPECT_EQ(read_metadata[0], metadata[0]);
- EXPECT_EQ(cv::norm(img2, img3, NORM_INF), 0.);
- double mse = cv::norm(img, img2, NORM_L2SQR)/(img.rows*img.cols);
- EXPECT_EQ(mse, 0); // png is lossless
- remove(outputname.c_str());
- }
- TEST(Imgcodecs_Png, Read_Write_With_Exif_Xmp_Iccp)
- {
- int png_compression = 3;
- int imgtype = CV_MAKETYPE(CV_8U, 3);
- const string outputname = cv::tempfile(".png");
- Mat img = makeCirclesImage(Size(160, 120), imgtype, 8);
- std::vector<int> metadata_types = { IMAGE_METADATA_EXIF, IMAGE_METADATA_XMP, IMAGE_METADATA_ICCP };
- std::vector<std::vector<uchar>> metadata = {
- getSampleExifData(),
- getSampleXmpData(),
- getSampleIccpData(),
- };
- std::vector<int> write_params = {
- IMWRITE_PNG_COMPRESSION, png_compression
- };
- imwriteWithMetadata(outputname, img, metadata_types, metadata, write_params);
- std::vector<uchar> compressed;
- imencodeWithMetadata(outputname, img, metadata_types, metadata, compressed, write_params);
- std::vector<int> read_metadata_types, read_metadata_types2;
- std::vector<std::vector<uchar> > read_metadata, read_metadata2;
- Mat img2 = imreadWithMetadata(outputname, read_metadata_types, read_metadata, IMREAD_UNCHANGED);
- Mat img3 = imdecodeWithMetadata(compressed, read_metadata_types2, read_metadata2, IMREAD_UNCHANGED);
- EXPECT_EQ(img2.cols, img.cols);
- EXPECT_EQ(img2.rows, img.rows);
- EXPECT_EQ(img2.type(), imgtype);
- EXPECT_EQ(metadata_types, read_metadata_types);
- EXPECT_EQ(read_metadata_types, read_metadata_types2);
- EXPECT_EQ(metadata, read_metadata);
- remove(outputname.c_str());
- }
- TEST(Imgcodecs_Png, Read_Exif_From_Text)
- {
- const string root = cvtest::TS::ptr()->get_data_path();
- const string filename = root + "../perf/320x260.png";
- const string dst_file = cv::tempfile(".png");
- std::vector<uchar> exif_data =
- { 'M' , 'M' , 0, '*' , 0, 0, 0, 8, 0, 4, 1,
- 26, 0, 5, 0, 0, 0, 1, 0, 0, 0, 62, 1, 27, 0, 5, 0, 0, 0, 1, 0, 0, 0,
- 70, 1, 40, 0, 3, 0, 0, 0, 1, 0, 2, 0, 0, 1, 49, 0, 2, 0, 0, 0, 18, 0,
- 0, 0, 78, 0, 0, 0, 0, 0, 0, 0, 96, 0, 0, 0, 1, 0, 0, 0, 96, 0, 0, 0,
- 1, 80, 97, 105, 110, 116, 46, 78, 69, 84, 32, 118, 51, 46, 53, 46, 49, 48, 0
- };
- std::vector<int> read_metadata_types;
- std::vector<std::vector<uchar> > read_metadata;
- Mat img = imreadWithMetadata(filename, read_metadata_types, read_metadata, IMREAD_GRAYSCALE);
- std::vector<int> metadata_types = { IMAGE_METADATA_EXIF };
- EXPECT_EQ(read_metadata_types, metadata_types);
- EXPECT_EQ(read_metadata[0], exif_data);
- }
- static size_t locateString(const uchar* exif, size_t exif_size, const std::string& pattern)
- {
- size_t plen = pattern.size();
- for (size_t i = 0; i + plen <= exif_size; i++) {
- if (exif[i] == pattern[0] && memcmp(&exif[i], pattern.c_str(), plen) == 0)
- return i;
- }
- return 0xFFFFFFFFu;
- }
- typedef std::tuple<std::string, size_t, std::string, size_t, size_t, size_t> ReadExif_Sanity_Params;
- typedef testing::TestWithParam<ReadExif_Sanity_Params> ReadExif_Sanity;
- TEST_P(ReadExif_Sanity, Check)
- {
- std::string filename = get<0>(GetParam());
- size_t exif_size = get<1>(GetParam());
- std::string pattern = get<2>(GetParam());
- size_t ploc = get<3>(GetParam());
- size_t expected_xmp_size = get<4>(GetParam());
- size_t expected_iccp_size = get<5>(GetParam());
- const string root = cvtest::TS::ptr()->get_data_path();
- filename = root + filename;
- std::vector<int> metadata_types, metadata_types2;
- std::vector<std::vector<uchar> > metadata, metadata2;
- Mat img = imreadWithMetadata(filename, metadata_types, metadata);
- std::vector<uchar> compressed;
- imencodeWithMetadata(".jpg", img, metadata_types, metadata, compressed);
- img = imdecodeWithMetadata(compressed, metadata_types2, metadata2);
- EXPECT_EQ(metadata_types, metadata_types2);
- EXPECT_EQ(metadata, metadata2);
- EXPECT_EQ(img.type(), CV_8UC3);
- ASSERT_GE(metadata_types.size(), 1u);
- EXPECT_EQ(metadata_types.size(), metadata.size());
- const Mat exif = Mat(metadata[IMAGE_METADATA_EXIF]);
- EXPECT_EQ(exif.type(), CV_8U);
- EXPECT_EQ(exif.total(), exif_size);
- ASSERT_GE(exif_size, 26u); // minimal exif should take at least 26 bytes
- // (the header + IDF0 with at least 1 entry).
- EXPECT_TRUE(exif.data[0] == 'I' || exif.data[0] == 'M');
- EXPECT_EQ(exif.data[0], exif.data[1]);
- EXPECT_EQ(locateString(exif.data, exif_size, pattern), ploc);
- if (metadata_types.size() > IMAGE_METADATA_XMP)
- {
- const Mat xmp = Mat(metadata[IMAGE_METADATA_XMP]);
- EXPECT_EQ(xmp.type(), CV_8U);
- EXPECT_GT(xmp.total(), 0u);
- size_t xmp_size = xmp.total() * xmp.elemSize();
- EXPECT_EQ(expected_xmp_size, xmp_size);
- }
- if (metadata_types.size() > IMAGE_METADATA_ICCP)
- {
- const Mat iccp = Mat(metadata[IMAGE_METADATA_ICCP]);
- EXPECT_EQ(iccp.type(), CV_8U);
- EXPECT_GT(iccp.total(), 0u);
- size_t iccp_size = iccp.total() * iccp.elemSize();
- EXPECT_EQ(expected_iccp_size, iccp_size);
- }
- }
- static const std::vector<ReadExif_Sanity_Params> exif_sanity_params
- {
- #ifdef HAVE_JPEG
- ReadExif_Sanity_Params("readwrite/testExifOrientation_3.jpg", 916, "Photoshop", 120, 3597, 940),
- #endif
- #ifdef OPENCV_IMGCODECS_PNG_WITH_EXIF
- ReadExif_Sanity_Params("readwrite/testExifOrientation_5.png", 112, "ExifTool", 102, 505, 0),
- #endif
- #ifdef HAVE_AVIF
- ReadExif_Sanity_Params("readwrite/testExifOrientation_7.avif", 913, "Photoshop", 120, 3597, 940),
- #endif
- };
- INSTANTIATE_TEST_CASE_P(Imgcodecs, ReadExif_Sanity,
- testing::ValuesIn(exif_sanity_params));
- }}
|