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
|
// 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.
#ifndef LIB_JXL_ENC_BUTTERAUGLI_COMPARATOR_H_
#define LIB_JXL_ENC_BUTTERAUGLI_COMPARATOR_H_
#include <jxl/cms_interface.h>
#include <stddef.h>
#include <memory>
#include <vector>
#include "lib/jxl/base/data_parallel.h"
#include "lib/jxl/base/status.h"
#include "lib/jxl/butteraugli/butteraugli.h"
#include "lib/jxl/enc_comparator.h"
#include "lib/jxl/image.h"
#include "lib/jxl/image_bundle.h"
namespace jxl {
class JxlButteraugliComparator : public Comparator {
public:
explicit JxlButteraugliComparator(const ButteraugliParams& params,
const JxlCmsInterface& cms);
Status SetReferenceImage(const ImageBundle& ref) override;
Status SetLinearReferenceImage(const Image3F& linear);
Status CompareWith(const ImageBundle& actual, ImageF* diffmap,
float* score) override;
float GoodQualityScore() const override;
float BadQualityScore() const override;
private:
ButteraugliParams params_;
JxlCmsInterface cms_;
std::unique_ptr<ButteraugliComparator> comparator_;
size_t xsize_ = 0;
size_t ysize_ = 0;
};
// Returns the butteraugli distance between rgb0 and rgb1.
// If distmap is not null, it must be the same size as rgb0 and rgb1.
float ButteraugliDistance(const ImageBundle& rgb0, const ImageBundle& rgb1,
const ButteraugliParams& params,
const JxlCmsInterface& cms, ImageF* distmap = nullptr,
ThreadPool* pool = nullptr,
bool ignore_alpha = false);
float ButteraugliDistance(const std::vector<ImageBundle>& frames0,
const std::vector<ImageBundle>& frames1,
const ButteraugliParams& params,
const JxlCmsInterface& cms, ImageF* distmap = nullptr,
ThreadPool* pool = nullptr);
} // namespace jxl
#endif // LIB_JXL_ENC_BUTTERAUGLI_COMPARATOR_H_
|