Source code
Revision control
Copy as Markdown
Other Tools
// 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