xref: /aosp_15_r20/external/skia/tests/SkJpegXmpTest.cpp (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1*c8dee2aaSAndroid Build Coastguard Worker /*
2*c8dee2aaSAndroid Build Coastguard Worker  * Copyright 2023 Google Inc.
3*c8dee2aaSAndroid Build Coastguard Worker  *
4*c8dee2aaSAndroid Build Coastguard Worker  * Use of this source code is governed by a BSD-style license that can be
5*c8dee2aaSAndroid Build Coastguard Worker  * found in the LICENSE file.
6*c8dee2aaSAndroid Build Coastguard Worker  */
7*c8dee2aaSAndroid Build Coastguard Worker 
8*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkData.h"
9*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/SkGainmapInfo.h"
10*c8dee2aaSAndroid Build Coastguard Worker #include "src/codec/SkJpegXmp.h"
11*c8dee2aaSAndroid Build Coastguard Worker #include "src/core/SkMD5.h"
12*c8dee2aaSAndroid Build Coastguard Worker #include "tests/Test.h"
13*c8dee2aaSAndroid Build Coastguard Worker 
14*c8dee2aaSAndroid Build Coastguard Worker #include <iomanip>
15*c8dee2aaSAndroid Build Coastguard Worker #include <iostream>
16*c8dee2aaSAndroid Build Coastguard Worker #include <regex>
17*c8dee2aaSAndroid Build Coastguard Worker #include <sstream>
18*c8dee2aaSAndroid Build Coastguard Worker 
DEF_TEST(SkJpegXmp_standardXmp,r)19*c8dee2aaSAndroid Build Coastguard Worker DEF_TEST(SkJpegXmp_standardXmp, r) {
20*c8dee2aaSAndroid Build Coastguard Worker     const char xmpData[] =
21*c8dee2aaSAndroid Build Coastguard Worker             "http://ns.adobe.com/xap/1.0/\0"
22*c8dee2aaSAndroid Build Coastguard Worker             R"(
23*c8dee2aaSAndroid Build Coastguard Worker             <x:xmpmeta xmlns:x="adobe:ns:meta/" x:xmptk="XMP Core 6.0.0">
24*c8dee2aaSAndroid Build Coastguard Worker                <rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
25*c8dee2aaSAndroid Build Coastguard Worker                         xmlns:hdrgm="http://ns.adobe.com/hdr-gain-map/1.0/">
26*c8dee2aaSAndroid Build Coastguard Worker                   <rdf:Description rdf:about="">
27*c8dee2aaSAndroid Build Coastguard Worker                      <hdrgm:Version>1.0</hdrgm:Version>
28*c8dee2aaSAndroid Build Coastguard Worker                      <hdrgm:GainMapMax>3</hdrgm:GainMapMax>
29*c8dee2aaSAndroid Build Coastguard Worker                      <hdrgm:HDRCapacityMax>4</hdrgm:HDRCapacityMax>
30*c8dee2aaSAndroid Build Coastguard Worker                   </rdf:Description>
31*c8dee2aaSAndroid Build Coastguard Worker                </rdf:RDF>
32*c8dee2aaSAndroid Build Coastguard Worker             </x:xmpmeta>)";
33*c8dee2aaSAndroid Build Coastguard Worker 
34*c8dee2aaSAndroid Build Coastguard Worker     std::vector<sk_sp<SkData>> app1Params;
35*c8dee2aaSAndroid Build Coastguard Worker     app1Params.push_back(SkData::MakeWithoutCopy(xmpData, sizeof(xmpData) - 1));
36*c8dee2aaSAndroid Build Coastguard Worker 
37*c8dee2aaSAndroid Build Coastguard Worker     auto xmp = SkJpegMakeXmp(app1Params);
38*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, xmp);
39*c8dee2aaSAndroid Build Coastguard Worker 
40*c8dee2aaSAndroid Build Coastguard Worker     SkGainmapInfo info;
41*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, xmp->getGainmapInfoHDRGM(&info));
42*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapRatioMax.fR == 8.f);
43*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fDisplayRatioHdr == 16.f);
44*c8dee2aaSAndroid Build Coastguard Worker }
45*c8dee2aaSAndroid Build Coastguard Worker 
46*c8dee2aaSAndroid Build Coastguard Worker DEF_TEST(SkJpegXmp_defaultValues, r) {
47*c8dee2aaSAndroid Build Coastguard Worker     const char xmpData[] =
48*c8dee2aaSAndroid Build Coastguard Worker             "http://ns.adobe.com/xap/1.0/\0"
49*c8dee2aaSAndroid Build Coastguard Worker             R"(
50*c8dee2aaSAndroid Build Coastguard Worker             <x:xmpmeta xmlns:x="adobe:ns:meta/" x:xmptk="XMP Core 6.0.0">
51*c8dee2aaSAndroid Build Coastguard Worker                <rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
52*c8dee2aaSAndroid Build Coastguard Worker                         xmlns:hdrgm="http://ns.adobe.com/hdr-gain-map/1.0/">
53*c8dee2aaSAndroid Build Coastguard Worker                   <rdf:Description rdf:about="" hdrgm:Version="1.0">
54*c8dee2aaSAndroid Build Coastguard Worker                   </rdf:Description>
55*c8dee2aaSAndroid Build Coastguard Worker                </rdf:RDF>
56*c8dee2aaSAndroid Build Coastguard Worker             </x:xmpmeta>)";
57*c8dee2aaSAndroid Build Coastguard Worker 
58*c8dee2aaSAndroid Build Coastguard Worker     std::vector<sk_sp<SkData>> app1Params;
59*c8dee2aaSAndroid Build Coastguard Worker     app1Params.push_back(SkData::MakeWithoutCopy(xmpData, sizeof(xmpData) - 1));
60*c8dee2aaSAndroid Build Coastguard Worker 
61*c8dee2aaSAndroid Build Coastguard Worker     auto xmp = SkJpegMakeXmp(app1Params);
62*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, xmp);
63*c8dee2aaSAndroid Build Coastguard Worker 
64*c8dee2aaSAndroid Build Coastguard Worker     SkGainmapInfo info;
65*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, xmp->getGainmapInfoHDRGM(&info));
66*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapRatioMin.fR == 1.f);
67*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapRatioMin.fG == 1.f);
68*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapRatioMin.fB == 1.f);
69*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapRatioMax.fR == 2.f);
70*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapRatioMax.fG == 2.f);
71*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapRatioMax.fB == 2.f);
72*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapGamma.fR == 1.f);
73*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapGamma.fG == 1.f);
74*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapGamma.fB == 1.f);
75*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fEpsilonSdr.fR == 1.f / 64.f);
76*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fEpsilonSdr.fG == 1.f / 64.f);
77*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fEpsilonSdr.fB == 1.f / 64.f);
78*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fEpsilonHdr.fG == 1.f / 64.f);
79*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fEpsilonHdr.fR == 1.f / 64.f);
80*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fEpsilonHdr.fB == 1.f / 64.f);
81*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fDisplayRatioSdr == 1.f);
82*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fDisplayRatioHdr == 2.f);
83*c8dee2aaSAndroid Build Coastguard Worker }
84*c8dee2aaSAndroid Build Coastguard Worker 
85*c8dee2aaSAndroid Build Coastguard Worker static std::string uint32_to_string(uint32_t v) {
86*c8dee2aaSAndroid Build Coastguard Worker     const char c[4] = {static_cast<char>((v >> 24) & 0xff),
87*c8dee2aaSAndroid Build Coastguard Worker                        static_cast<char>((v >> 16) & 0xff),
88*c8dee2aaSAndroid Build Coastguard Worker                        static_cast<char>((v >> 8) & 0xff),
89*c8dee2aaSAndroid Build Coastguard Worker                        static_cast<char>(v & 0xff)};
90*c8dee2aaSAndroid Build Coastguard Worker     return std::string(c, 4);
91*c8dee2aaSAndroid Build Coastguard Worker }
92*c8dee2aaSAndroid Build Coastguard Worker 
93*c8dee2aaSAndroid Build Coastguard Worker static std::string standard_xmp_with_header(const SkMD5::Digest& digest, const std::string& data) {
94*c8dee2aaSAndroid Build Coastguard Worker     const std::string guid = digest.toHexString().c_str();
95*c8dee2aaSAndroid Build Coastguard Worker     const std::string dataWithGuid = std::regex_replace(data, std::regex("\\$GUID"), guid);
96*c8dee2aaSAndroid Build Coastguard Worker     return std::string("http://ns.adobe.com/xap/1.0/") + '\0' + dataWithGuid;
97*c8dee2aaSAndroid Build Coastguard Worker }
98*c8dee2aaSAndroid Build Coastguard Worker 
extended_xmp_with_header(const SkMD5::Digest & digest,uint32_t size,uint32_t offset,const std::string & data)99*c8dee2aaSAndroid Build Coastguard Worker static std::string extended_xmp_with_header(const SkMD5::Digest& digest,
100*c8dee2aaSAndroid Build Coastguard Worker                                             uint32_t size,
101*c8dee2aaSAndroid Build Coastguard Worker                                             uint32_t offset,
102*c8dee2aaSAndroid Build Coastguard Worker                                             const std::string& data) {
103*c8dee2aaSAndroid Build Coastguard Worker     return std::string("http://ns.adobe.com/xmp/extension/") + '\0' + digest.toHexString().c_str() +
104*c8dee2aaSAndroid Build Coastguard Worker            uint32_to_string(size) + uint32_to_string(offset) + data;
105*c8dee2aaSAndroid Build Coastguard Worker }
106*c8dee2aaSAndroid Build Coastguard Worker 
DEF_TEST(SkJpegXmp_readExtendedXmp,r)107*c8dee2aaSAndroid Build Coastguard Worker DEF_TEST(SkJpegXmp_readExtendedXmp, r) {
108*c8dee2aaSAndroid Build Coastguard Worker     const std::string standardXmpData = R"(
109*c8dee2aaSAndroid Build Coastguard Worker             <x:xmpmeta xmlns:x="adobe:ns:meta/" x:xmptk="XMP Core 6.0.0">
110*c8dee2aaSAndroid Build Coastguard Worker                <rdf:RDF xmlns:xmpNote="http://ns.adobe.com/xmp/note/">
111*c8dee2aaSAndroid Build Coastguard Worker                   <rdf:Description rdf:about="">
112*c8dee2aaSAndroid Build Coastguard Worker                      <xmpNote:HasExtendedXMP>$GUID</xmpNote:HasExtendedXMP>
113*c8dee2aaSAndroid Build Coastguard Worker                   </rdf:Description>
114*c8dee2aaSAndroid Build Coastguard Worker                </rdf:RDF>
115*c8dee2aaSAndroid Build Coastguard Worker             </x:xmpmeta>)";
116*c8dee2aaSAndroid Build Coastguard Worker 
117*c8dee2aaSAndroid Build Coastguard Worker     const std::string extendedXmpData1 = R"(
118*c8dee2aaSAndroid Build Coastguard Worker         <x:xmpmeta xmlns:x="adobe:ns:meta/" x:xmptk="XMP Core 6.0.0">
119*c8dee2aaSAndroid Build Coastguard Worker             <rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
120*c8dee2aaSAndroid Build Coastguard Worker                     xmlns:hdrgm="http://ns.adobe.com/hdr-gain-map/1.0/">
121*c8dee2aaSAndroid Build Coastguard Worker                 <rdf:Description rdf:about="">
122*c8dee2aaSAndroid Build Coastguard Worker                      <hdrgm:Version>1.0</hdrgm:Version>
123*c8dee2aaSAndroid Build Coastguard Worker                      <hdrgm:GainMapMax>3</hdrgm:GainMapMax>
124*c8dee2aaSAndroid Build Coastguard Worker                     <hdrgm:HDRCapacityMax>4</hdrgm:HDRCapacityMax>)";
125*c8dee2aaSAndroid Build Coastguard Worker     const std::string extendedXmpData2 = R"(
126*c8dee2aaSAndroid Build Coastguard Worker                 </rdf:Description>
127*c8dee2aaSAndroid Build Coastguard Worker             </rdf:RDF>
128*c8dee2aaSAndroid Build Coastguard Worker         </x:xmpmeta>)";
129*c8dee2aaSAndroid Build Coastguard Worker 
130*c8dee2aaSAndroid Build Coastguard Worker     const uint32_t totalExtendedXmpSize = extendedXmpData1.size() + extendedXmpData2.size();
131*c8dee2aaSAndroid Build Coastguard Worker     SkMD5 md5;
132*c8dee2aaSAndroid Build Coastguard Worker     md5.write(extendedXmpData1.data(), extendedXmpData1.length());
133*c8dee2aaSAndroid Build Coastguard Worker     md5.write(extendedXmpData2.data(), extendedXmpData2.length());
134*c8dee2aaSAndroid Build Coastguard Worker     const SkMD5::Digest digest = md5.finish();
135*c8dee2aaSAndroid Build Coastguard Worker 
136*c8dee2aaSAndroid Build Coastguard Worker     const std::string standardXmpDataWithHeader = standard_xmp_with_header(digest, standardXmpData);
137*c8dee2aaSAndroid Build Coastguard Worker 
138*c8dee2aaSAndroid Build Coastguard Worker     const uint32_t offset1 = 0;
139*c8dee2aaSAndroid Build Coastguard Worker     const std::string extendedXmpData1WithHeader =
140*c8dee2aaSAndroid Build Coastguard Worker             extended_xmp_with_header(digest, totalExtendedXmpSize, offset1, extendedXmpData1);
141*c8dee2aaSAndroid Build Coastguard Worker 
142*c8dee2aaSAndroid Build Coastguard Worker     const uint32_t offset2 = extendedXmpData1.size();
143*c8dee2aaSAndroid Build Coastguard Worker     const std::string extendedXmpData2WithHeader =
144*c8dee2aaSAndroid Build Coastguard Worker             extended_xmp_with_header(digest, totalExtendedXmpSize, offset2, extendedXmpData2);
145*c8dee2aaSAndroid Build Coastguard Worker 
146*c8dee2aaSAndroid Build Coastguard Worker     std::vector<sk_sp<SkData>> app1Params;
147*c8dee2aaSAndroid Build Coastguard Worker     app1Params.push_back(SkData::MakeWithoutCopy(standardXmpDataWithHeader.data(),
148*c8dee2aaSAndroid Build Coastguard Worker                                                  standardXmpDataWithHeader.length()));
149*c8dee2aaSAndroid Build Coastguard Worker     app1Params.push_back(SkData::MakeWithoutCopy(extendedXmpData1WithHeader.data(),
150*c8dee2aaSAndroid Build Coastguard Worker                                                  extendedXmpData1WithHeader.length()));
151*c8dee2aaSAndroid Build Coastguard Worker     app1Params.push_back(SkData::MakeWithoutCopy(extendedXmpData2WithHeader.data(),
152*c8dee2aaSAndroid Build Coastguard Worker                                                  extendedXmpData2WithHeader.length()));
153*c8dee2aaSAndroid Build Coastguard Worker 
154*c8dee2aaSAndroid Build Coastguard Worker     auto xmp = SkJpegMakeXmp(app1Params);
155*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, xmp);
156*c8dee2aaSAndroid Build Coastguard Worker 
157*c8dee2aaSAndroid Build Coastguard Worker     SkGainmapInfo info;
158*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, xmp->getGainmapInfoHDRGM(&info));
159*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fGainmapRatioMax.fR == 8.f);
160*c8dee2aaSAndroid Build Coastguard Worker     REPORTER_ASSERT(r, info.fDisplayRatioHdr == 16.f);
161*c8dee2aaSAndroid Build Coastguard Worker }
162