78 lines
2.3 KiB
C++
78 lines
2.3 KiB
C++
// 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 <cstddef>
|
|
#include <vector>
|
|
|
|
#include "benchmark/benchmark.h"
|
|
#include "lib/jxl/base/rect.h"
|
|
#include "lib/jxl/base/status.h"
|
|
#include "lib/jxl/chroma_from_luma.h"
|
|
#include "lib/jxl/image_ops.h"
|
|
#include "lib/jxl/splines.h"
|
|
#include "tools/no_memory_manager.h"
|
|
|
|
namespace jxl {
|
|
namespace {
|
|
|
|
#define QUIT(M) \
|
|
state.SkipWithError(M); \
|
|
return;
|
|
|
|
#define BM_CHECK(C) \
|
|
if (!(C)) { \
|
|
QUIT(#C) \
|
|
}
|
|
|
|
constexpr int kQuantizationAdjustment = 0;
|
|
const ColorCorrelation color_correlation{};
|
|
const float kYToX = color_correlation.YtoXRatio(0);
|
|
const float kYToB = color_correlation.YtoBRatio(0);
|
|
|
|
void BM_Splines(benchmark::State& state) {
|
|
const size_t n = state.range();
|
|
|
|
Spline spline1{
|
|
/*control_points=*/{
|
|
{9, 54}, {118, 159}, {97, 3}, {10, 40}, {150, 25}, {120, 300}},
|
|
/*color_dct=*/
|
|
{Dct32{0.03125f, 0.00625f, 0.003125f}, Dct32{1.f, 0.321875f},
|
|
Dct32{1.f, 0.24375f}},
|
|
/*sigma_dct=*/{0.3125f, 0.f, 0.f, 0.0625f}};
|
|
std::vector<Spline> spline_data = {spline1};
|
|
std::vector<QuantizedSpline> quantized_splines;
|
|
std::vector<Spline::Point> starting_points;
|
|
for (const Spline& spline : spline_data) {
|
|
JXL_ASSIGN_OR_QUIT(
|
|
QuantizedSpline qspline,
|
|
QuantizedSpline::Create(spline, kQuantizationAdjustment, kYToX, kYToB),
|
|
"Failed to create spline.");
|
|
quantized_splines.emplace_back(std::move(qspline));
|
|
starting_points.push_back(spline.control_points.front());
|
|
}
|
|
Splines splines(kQuantizationAdjustment, std::move(quantized_splines),
|
|
std::move(starting_points));
|
|
|
|
JXL_ASSIGN_OR_QUIT(
|
|
Image3F drawing_area,
|
|
Image3F::Create(jpegxl::tools::NoMemoryManager(), 320, 320),
|
|
"Failed to allocate drawing plane.");
|
|
ZeroFillImage(&drawing_area);
|
|
for (auto _ : state) {
|
|
(void)_;
|
|
for (size_t i = 0; i < n; ++i) {
|
|
BM_CHECK(splines.InitializeDrawCache(
|
|
drawing_area.xsize(), drawing_area.ysize(), color_correlation));
|
|
splines.AddTo(&drawing_area, Rect(drawing_area));
|
|
}
|
|
}
|
|
|
|
state.SetItemsProcessed(n * state.iterations());
|
|
}
|
|
|
|
BENCHMARK(BM_Splines)->Range(1, 1 << 10);
|
|
|
|
} // namespace
|
|
} // namespace jxl
|