forked from mirrors/gecko-dev
95 lines
2.5 KiB
C++
95 lines
2.5 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.
|
|
|
|
#ifndef LIB_JXL_DCT_UTIL_H_
|
|
#define LIB_JXL_DCT_UTIL_H_
|
|
|
|
#include <jxl/memory_manager.h>
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <memory>
|
|
#include <type_traits>
|
|
|
|
#include "lib/jxl/base/common.h"
|
|
#include "lib/jxl/base/status.h"
|
|
#include "lib/jxl/image.h"
|
|
#include "lib/jxl/image_ops.h"
|
|
|
|
namespace jxl {
|
|
|
|
union ACPtr {
|
|
int32_t* ptr32;
|
|
int16_t* ptr16;
|
|
ACPtr() = default;
|
|
explicit ACPtr(int16_t* p) : ptr16(p) {}
|
|
explicit ACPtr(int32_t* p) : ptr32(p) {}
|
|
};
|
|
|
|
union ConstACPtr {
|
|
const int32_t* ptr32;
|
|
const int16_t* ptr16;
|
|
ConstACPtr() = default;
|
|
explicit ConstACPtr(const int16_t* p) : ptr16(p) {}
|
|
explicit ConstACPtr(const int32_t* p) : ptr32(p) {}
|
|
};
|
|
|
|
enum class ACType { k16 = 0, k32 = 1 };
|
|
|
|
class ACImage {
|
|
public:
|
|
virtual ~ACImage() = default;
|
|
virtual ACType Type() const = 0;
|
|
virtual ACPtr PlaneRow(size_t c, size_t y, size_t xbase) = 0;
|
|
virtual ConstACPtr PlaneRow(size_t c, size_t y, size_t xbase) const = 0;
|
|
virtual size_t PixelsPerRow() const = 0;
|
|
virtual void ZeroFill() = 0;
|
|
virtual void ZeroFillPlane(size_t c) = 0;
|
|
virtual bool IsEmpty() const = 0;
|
|
};
|
|
|
|
template <typename T>
|
|
class ACImageT final : public ACImage {
|
|
public:
|
|
ACImageT() = default;
|
|
|
|
static StatusOr<std::unique_ptr<ACImageT>> Make(
|
|
JxlMemoryManager* memory_manager, size_t xsize, size_t ysize) {
|
|
static_assert(
|
|
std::is_same<T, int16_t>::value || std::is_same<T, int32_t>::value,
|
|
"ACImage must be either 32- or 16- bit");
|
|
std::unique_ptr<ACImageT> result = jxl::make_unique<ACImageT>();
|
|
JXL_ASSIGN_OR_RETURN(result->img_,
|
|
Image3<T>::Create(memory_manager, xsize, ysize));
|
|
return result;
|
|
}
|
|
|
|
ACType Type() const override {
|
|
return sizeof(T) == 2 ? ACType::k16 : ACType::k32;
|
|
}
|
|
ACPtr PlaneRow(size_t c, size_t y, size_t xbase) override {
|
|
return ACPtr(img_.PlaneRow(c, y) + xbase);
|
|
}
|
|
ConstACPtr PlaneRow(size_t c, size_t y, size_t xbase) const override {
|
|
return ConstACPtr(img_.PlaneRow(c, y) + xbase);
|
|
}
|
|
|
|
size_t PixelsPerRow() const override { return img_.PixelsPerRow(); }
|
|
|
|
void ZeroFill() override { ZeroFillImage(&img_); }
|
|
|
|
void ZeroFillPlane(size_t c) override { ZeroFillImage(&img_.Plane(c)); }
|
|
|
|
bool IsEmpty() const override {
|
|
return img_.xsize() == 0 || img_.ysize() == 0;
|
|
}
|
|
|
|
private:
|
|
Image3<T> img_;
|
|
};
|
|
|
|
} // namespace jxl
|
|
|
|
#endif // LIB_JXL_DCT_UTIL_H_
|