summaryrefslogtreecommitdiffstats
path: root/third_party/jpeg-xl/lib/extras/dec/pgx_test.cc
blob: 78ed689d07f7c76e76de0da274e4e3cda9a481a4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
// Copyright (c) the JPEG XL Project Authors. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

#include "lib/extras/dec/pgx.h"

#include "lib/extras/packed_image_convert.h"
#include "lib/jxl/image_bundle.h"
#include "lib/jxl/testing.h"

namespace jxl {
namespace extras {
namespace {

Span<const uint8_t> MakeSpan(const char* str) {
  return Span<const uint8_t>(reinterpret_cast<const uint8_t*>(str),
                             strlen(str));
}

TEST(CodecPGXTest, Test8bits) {
  std::string pgx = "PG ML + 8 2 3\npixels";

  PackedPixelFile ppf;
  ThreadPool* pool = nullptr;

  EXPECT_TRUE(DecodeImagePGX(MakeSpan(pgx.c_str()), ColorHints(), &ppf));
  CodecInOut io;
  EXPECT_TRUE(ConvertPackedPixelFileToCodecInOut(ppf, pool, &io));

  ScaleImage(255.f, io.Main().color());

  EXPECT_FALSE(io.metadata.m.bit_depth.floating_point_sample);
  EXPECT_EQ(8u, io.metadata.m.bit_depth.bits_per_sample);
  EXPECT_TRUE(io.metadata.m.color_encoding.IsGray());
  EXPECT_EQ(2u, io.xsize());
  EXPECT_EQ(3u, io.ysize());

  float eps = 1e-5;
  EXPECT_NEAR('p', io.Main().color()->Plane(0).Row(0)[0], eps);
  EXPECT_NEAR('i', io.Main().color()->Plane(0).Row(0)[1], eps);
  EXPECT_NEAR('x', io.Main().color()->Plane(0).Row(1)[0], eps);
  EXPECT_NEAR('e', io.Main().color()->Plane(0).Row(1)[1], eps);
  EXPECT_NEAR('l', io.Main().color()->Plane(0).Row(2)[0], eps);
  EXPECT_NEAR('s', io.Main().color()->Plane(0).Row(2)[1], eps);
}

TEST(CodecPGXTest, Test16bits) {
  std::string pgx = "PG ML + 16 2 3\np_i_x_e_l_s_";

  PackedPixelFile ppf;
  ThreadPool* pool = nullptr;

  EXPECT_TRUE(DecodeImagePGX(MakeSpan(pgx.c_str()), ColorHints(), &ppf));
  CodecInOut io;
  EXPECT_TRUE(ConvertPackedPixelFileToCodecInOut(ppf, pool, &io));

  ScaleImage(255.f, io.Main().color());

  EXPECT_FALSE(io.metadata.m.bit_depth.floating_point_sample);
  EXPECT_EQ(16u, io.metadata.m.bit_depth.bits_per_sample);
  EXPECT_TRUE(io.metadata.m.color_encoding.IsGray());
  EXPECT_EQ(2u, io.xsize());
  EXPECT_EQ(3u, io.ysize());

  // Comparing ~16-bit numbers in floats, only ~7 bits left.
  float eps = 1e-3;
  const auto& plane = io.Main().color()->Plane(0);
  EXPECT_NEAR(256.0f * 'p' + '_', plane.Row(0)[0] * 257, eps);
  EXPECT_NEAR(256.0f * 'i' + '_', plane.Row(0)[1] * 257, eps);
  EXPECT_NEAR(256.0f * 'x' + '_', plane.Row(1)[0] * 257, eps);
  EXPECT_NEAR(256.0f * 'e' + '_', plane.Row(1)[1] * 257, eps);
  EXPECT_NEAR(256.0f * 'l' + '_', plane.Row(2)[0] * 257, eps);
  EXPECT_NEAR(256.0f * 's' + '_', plane.Row(2)[1] * 257, eps);
}

}  // namespace
}  // namespace extras
}  // namespace jxl