1 /*
2 * Copyright 2024 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <gtest/gtest.h>
18 #include <vector>
19
20 #include "ultrahdr/gainmapmetadata.h"
21
22 namespace ultrahdr {
23
24 class GainMapMetadataTest : public testing::Test {
25 public:
26 GainMapMetadataTest();
27 ~GainMapMetadataTest();
28
29 protected:
30 virtual void SetUp();
31 virtual void TearDown();
32 };
33
GainMapMetadataTest()34 GainMapMetadataTest::GainMapMetadataTest() {}
35
~GainMapMetadataTest()36 GainMapMetadataTest::~GainMapMetadataTest() {}
37
SetUp()38 void GainMapMetadataTest::SetUp() {}
39
TearDown()40 void GainMapMetadataTest::TearDown() {}
41
42 const std::string kIso = "urn:iso:std:iso:ts:21496:-1";
43
TEST_F(GainMapMetadataTest,encodeMetadataThenDecode)44 TEST_F(GainMapMetadataTest, encodeMetadataThenDecode) {
45 uhdr_gainmap_metadata_ext_t expected("1.0");
46 for (int i = 0; i < 3; i++) {
47 expected.max_content_boost[i] = 100.5f + i;
48 expected.min_content_boost[i] = 1.5f + i * 0.1f;
49 expected.gamma[i] = 1.0f + i * 0.01f;
50 expected.offset_sdr[i] = 0.0625f + i * 0.025f;
51 expected.offset_hdr[i] = 0.0625f + i * 0.025f;
52 }
53 expected.hdr_capacity_min = 1.0f;
54 expected.hdr_capacity_max = 10000.0f / 203.0f;
55 expected.use_base_cg = false;
56
57 uhdr_gainmap_metadata_frac metadata;
58 EXPECT_EQ(
59 uhdr_gainmap_metadata_frac::gainmapMetadataFloatToFraction(&expected, &metadata).error_code,
60 UHDR_CODEC_OK);
61 // metadata.dump();
62
63 std::vector<uint8_t> data;
64 EXPECT_EQ(uhdr_gainmap_metadata_frac::encodeGainmapMetadata(&metadata, data).error_code,
65 UHDR_CODEC_OK);
66
67 uhdr_gainmap_metadata_frac decodedMetadata;
68 EXPECT_EQ(uhdr_gainmap_metadata_frac::decodeGainmapMetadata(data, &decodedMetadata).error_code,
69 UHDR_CODEC_OK);
70
71 uhdr_gainmap_metadata_ext_t decodedUHdrMetadata;
72 EXPECT_EQ(uhdr_gainmap_metadata_frac::gainmapMetadataFractionToFloat(&decodedMetadata,
73 &decodedUHdrMetadata)
74 .error_code,
75 UHDR_CODEC_OK);
76
77 for (int i = 0; i < 3; i++) {
78 EXPECT_FLOAT_EQ(expected.max_content_boost[i], decodedUHdrMetadata.max_content_boost[i]);
79 EXPECT_FLOAT_EQ(expected.min_content_boost[i], decodedUHdrMetadata.min_content_boost[i]);
80 EXPECT_FLOAT_EQ(expected.gamma[i], decodedUHdrMetadata.gamma[i]);
81 EXPECT_FLOAT_EQ(expected.offset_sdr[i], decodedUHdrMetadata.offset_sdr[i]);
82 EXPECT_FLOAT_EQ(expected.offset_hdr[i], decodedUHdrMetadata.offset_hdr[i]);
83 }
84 EXPECT_FLOAT_EQ(expected.hdr_capacity_min, decodedUHdrMetadata.hdr_capacity_min);
85 EXPECT_FLOAT_EQ(expected.hdr_capacity_max, decodedUHdrMetadata.hdr_capacity_max);
86 EXPECT_EQ(expected.use_base_cg, decodedUHdrMetadata.use_base_cg);
87
88 data.clear();
89 for (int i = 0; i < 3; i++) {
90 expected.min_content_boost[i] = 0.000578369f + i * 0.001f;
91 expected.offset_sdr[i] = -0.0625f + i * 0.001f;
92 expected.offset_hdr[i] = -0.0625f + i * 0.001f;
93 }
94 expected.hdr_capacity_max = 1000.0f / 203.0f;
95 expected.use_base_cg = true;
96
97 EXPECT_EQ(
98 uhdr_gainmap_metadata_frac::gainmapMetadataFloatToFraction(&expected, &metadata).error_code,
99 UHDR_CODEC_OK);
100 EXPECT_EQ(uhdr_gainmap_metadata_frac::encodeGainmapMetadata(&metadata, data).error_code,
101 UHDR_CODEC_OK);
102 EXPECT_EQ(uhdr_gainmap_metadata_frac::decodeGainmapMetadata(data, &decodedMetadata).error_code,
103 UHDR_CODEC_OK);
104 EXPECT_EQ(uhdr_gainmap_metadata_frac::gainmapMetadataFractionToFloat(&decodedMetadata,
105 &decodedUHdrMetadata)
106 .error_code,
107 UHDR_CODEC_OK);
108
109 for (int i = 0; i < 3; i++) {
110 EXPECT_FLOAT_EQ(expected.max_content_boost[i], decodedUHdrMetadata.max_content_boost[i]);
111 EXPECT_FLOAT_EQ(expected.min_content_boost[i], decodedUHdrMetadata.min_content_boost[i]);
112 EXPECT_FLOAT_EQ(expected.gamma[i], decodedUHdrMetadata.gamma[i]);
113 EXPECT_FLOAT_EQ(expected.offset_sdr[i], decodedUHdrMetadata.offset_sdr[i]);
114 EXPECT_FLOAT_EQ(expected.offset_hdr[i], decodedUHdrMetadata.offset_hdr[i]);
115 }
116 EXPECT_FLOAT_EQ(expected.hdr_capacity_min, decodedUHdrMetadata.hdr_capacity_min);
117 EXPECT_FLOAT_EQ(expected.hdr_capacity_max, decodedUHdrMetadata.hdr_capacity_max);
118 EXPECT_EQ(expected.use_base_cg, decodedUHdrMetadata.use_base_cg);
119 }
120
121 } // namespace ultrahdr
122