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.
#ifndef LIB_JXL_BASE_FLOAT_H_
#define LIB_JXL_BASE_FLOAT_H_
#include <jxl/types.h>
#include <stddef.h>
#include <stdint.h>
#include <string.h>
#include "lib/jxl/base/byte_order.h"
#include "lib/jxl/base/compiler_specific.h"
#include "lib/jxl/base/status.h"
namespace jxl {
namespace detail {
// Based on highway scalar implementation, for testing
static JXL_INLINE float LoadFloat16(uint16_t bits16) {
const uint32_t sign = bits16 >> 15;
const uint32_t biased_exp = (bits16 >> 10) & 0x1F;
const uint32_t mantissa = bits16 & 0x3FF;
// Subnormal or zero
if (biased_exp == 0) {
const float subnormal =
(1.0f / 16384) * (static_cast<float>(mantissa) * (1.0f / 1024));
return sign ? -subnormal : subnormal;
}
// Normalized: convert the representation directly (faster than ldexp/tables).
const uint32_t biased_exp32 = biased_exp + (127 - 15);
const uint32_t mantissa32 = mantissa << (23 - 10);
const uint32_t bits32 = (sign << 31) | (biased_exp32 << 23) | mantissa32;
float result;
memcpy(&result, &bits32, 4);
return result;
}
} // namespace detail
template <typename SaveFloatAtFn>
static Status JXL_INLINE LoadFloatRow(const uint8_t* src, size_t count,
size_t stride, JxlDataType type,
bool little_endian, float scale,
SaveFloatAtFn callback) {
switch (type) {
case JXL_TYPE_FLOAT:
if (little_endian) {
for (size_t i = 0; i < count; ++i) {
callback(i, LoadLEFloat(src + stride * i));
}
} else {
for (size_t i = 0; i < count; ++i) {
callback(i, LoadBEFloat(src + stride * i));
}
}
return true;
case JXL_TYPE_UINT8:
for (size_t i = 0; i < count; ++i) {
callback(i, src[stride * i] * scale);
}
return true;
case JXL_TYPE_UINT16:
if (little_endian) {
for (size_t i = 0; i < count; ++i) {
callback(i, LoadLE16(src + stride * i) * scale);
}
} else {
for (size_t i = 0; i < count; ++i) {
callback(i, LoadBE16(src + stride * i) * scale);
}
}
return true;
case JXL_TYPE_FLOAT16:
if (little_endian) {
for (size_t i = 0; i < count; ++i) {
callback(i, detail::LoadFloat16(LoadLE16(src + stride * i)));
}
} else {
for (size_t i = 0; i < count; ++i) {
callback(i, detail::LoadFloat16(LoadBE16(src + stride * i)));
}
}
return true;
default:
return JXL_FAILURE("Unsupported sample format");
}
}
} // namespace jxl
#endif // LIB_JXL_BASE_FLOAT_H_