enc_quant_weights.h

Enable keyboard shortcuts

// 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_QUANT_WEIGHTS_H_

#define LIB_JXL_ENC_QUANT_WEIGHTS_H_

#include <jxl/memory_manager.h>

#include <cstdint>

#include <vector>

#include "lib/jxl/base/status.h"

#include "lib/jxl/quant_weights.h"

namespace jxl {

struct AuxOut;

enum class LayerType : uint8_t;

struct BitWriter;

Status DequantMatricesEncode(

    JxlMemoryManager* memory_manager, const DequantMatrices& matrices,

    BitWriter* writer, LayerType layer, AuxOut* aux_out,

    ModularFrameEncoder* modular_frame_encoder = nullptr);

Status DequantMatricesEncodeDC(const DequantMatrices& matrices,

                               BitWriter* writer, LayerType layer,

                               AuxOut* aux_out);

// For consistency with QuantEncoding, higher values correspond to more

// precision.

Status DequantMatricesSetCustomDC(JxlMemoryManager* memory_manager,

                                  DequantMatrices* matrices, const float* dc);

Status DequantMatricesScaleDC(JxlMemoryManager* memory_manager,

                              DequantMatrices* matrices, float scale);

Status DequantMatricesSetCustom(DequantMatrices* matrices,

                                const std::vector<QuantEncoding>& encodings,

                                ModularFrameEncoder* encoder);

// Roundtrip encode/decode the matrices to ensure same values as decoder.

Status DequantMatricesRoundtrip(JxlMemoryManager* memory_manager,

                                DequantMatrices* matrices);

}  // namespace jxl

#endif  // LIB_JXL_ENC_QUANT_WEIGHTS_H_