mirror of
https://github.com/boykopovar/AnyPS5.git
synced 2026-10-22 08:33:12 -05:00
Federico Marras
d65c969df8
sceJpegDecParseHeader and sceJpegDecDecode threw for every down scale other than 1. SharpProspero's JpegDec.cs documents the factors 1, 2, 4 and 8 and output dimensions that account for the down scale. Decoder::Jpeg gains DecodeScaled, which decodes through libjpeg-turbo's scaled inverse DCT, and ScaledDimension, its rounded-up output size. sceJpegDecParseHeader reports the scaled output size; sceJpegDecDecode checks the pitch and image buffer against it, writes the scaled image and packs the output size into its result. A down scale of 1 still decodes as before. Decode modes other than 0, other down scales and a down-scaled CMYK or YCCK image still throw.
181 lines
9.3 KiB
C++
181 lines
9.3 KiB
C++
#include "Decoder/Jpeg.hpp"
|
|
|
|
#include <algorithm>
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <cstdlib>
|
|
#include <stdexcept>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
static void Require(bool value) { if (!value) std::abort(); }
|
|
|
|
template<typename TFunction>
|
|
static bool ThrowsInvalidArgument(TFunction function) {
|
|
try {
|
|
function();
|
|
} catch (const std::invalid_argument&) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool IsJpeg(const std::vector<std::uint8_t>& data) {
|
|
return data.size() > 4 && data[0] == 0xFF && data[1] == 0xD8 && data[data.size() - 2] == 0xFF && data.back() == 0xD9;
|
|
}
|
|
|
|
static int Difference(std::uint8_t left, std::uint8_t right) {
|
|
return left > right ? left - right : right - left;
|
|
}
|
|
|
|
struct FrameInfo { int components; std::uint8_t sampling; };
|
|
|
|
static FrameInfo ReadFrame(const std::vector<std::uint8_t>& jpeg) {
|
|
std::size_t offset = 2;
|
|
while (offset + 3 < jpeg.size()) {
|
|
Require(jpeg[offset++] == 0xFF);
|
|
while (offset < jpeg.size() && jpeg[offset] == 0xFF) ++offset;
|
|
Require(offset < jpeg.size());
|
|
const std::uint8_t marker = jpeg[offset++];
|
|
if (marker == 0xDA || marker == 0xD9) break;
|
|
Require(offset + 1 < jpeg.size());
|
|
const std::size_t length = (static_cast<std::size_t>(jpeg[offset]) << 8) | jpeg[offset + 1];
|
|
Require(length >= 2 && offset + length <= jpeg.size());
|
|
if (marker >= 0xC0 && marker <= 0xC3) return {jpeg[offset + 7], jpeg[offset + 9]};
|
|
offset += length;
|
|
}
|
|
std::abort();
|
|
}
|
|
|
|
int main() {
|
|
constexpr std::uint32_t width = 32;
|
|
constexpr std::uint32_t height = 24;
|
|
|
|
std::vector<std::uint8_t> rgb(width * height * 3);
|
|
for (std::uint32_t y = 0; y < height; ++y) {
|
|
for (std::uint32_t x = 0; x < width; ++x) {
|
|
std::uint8_t* pixel = &rgb[(y * width + x) * 3];
|
|
pixel[0] = static_cast<std::uint8_t>(x * 8);
|
|
pixel[1] = static_cast<std::uint8_t>(y * 10);
|
|
pixel[2] = 128;
|
|
}
|
|
}
|
|
|
|
const std::vector<std::uint8_t> rgbJpeg = Decoder::Jpeg::Encode(rgb, width, height, 3, 90);
|
|
Require(IsJpeg(rgbJpeg));
|
|
Require(ReadFrame(rgbJpeg).components == 3 && ReadFrame(rgbJpeg).sampling == 0x11);
|
|
const auto rgbImage = Decoder::Jpeg::Decode(rgbJpeg);
|
|
Require(rgbImage.has_value());
|
|
Require(rgbImage->width == width && rgbImage->height == height && rgbImage->channels == 3);
|
|
Require(rgbImage->pixels.size() == rgb.size());
|
|
long totalError = 0;
|
|
for (std::size_t i = 0; i < rgb.size(); ++i) totalError += Difference(rgb[i], rgbImage->pixels[i]);
|
|
Require(totalError / static_cast<long>(rgb.size()) < 4);
|
|
|
|
std::vector<std::uint8_t> gray(width * height);
|
|
for (std::uint32_t y = 0; y < height; ++y) {
|
|
for (std::uint32_t x = 0; x < width; ++x) gray[y * width + x] = static_cast<std::uint8_t>((x + y) * 4);
|
|
}
|
|
const std::vector<std::uint8_t> grayJpeg = Decoder::Jpeg::Encode(gray, width, height, 1, 90);
|
|
Require(IsJpeg(grayJpeg));
|
|
Require(ReadFrame(grayJpeg).components == 1);
|
|
const auto grayImage = Decoder::Jpeg::Decode(grayJpeg);
|
|
Require(grayImage.has_value());
|
|
Require(grayImage->width == width && grayImage->height == height);
|
|
totalError = 0;
|
|
for (std::size_t i = 0; i < gray.size(); ++i) totalError += Difference(gray[i], grayImage->pixels[i * grayImage->channels]);
|
|
Require(totalError / static_cast<long>(gray.size()) < 4);
|
|
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, width, height, 4, 90); }));
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, width, height, 3, 0); }));
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, width, height, 3, 101); }));
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, 0, height, 3, 90); }));
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, 0x10000, 1, 3, 90); }));
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, width, height + 1, 3, 90); }));
|
|
|
|
for (const auto [sampling, factor] : {std::pair{Decoder::Jpeg::Sampling::Yuv444, std::uint8_t{0x11}},
|
|
std::pair{Decoder::Jpeg::Sampling::Yuv422, std::uint8_t{0x21}},
|
|
std::pair{Decoder::Jpeg::Sampling::Yuv420, std::uint8_t{0x22}}}) {
|
|
const auto encoded = Decoder::Jpeg::Encode(rgb, 31, 23, 3, 1, sampling);
|
|
Require(ReadFrame(encoded).components == 3 && ReadFrame(encoded).sampling == factor);
|
|
const auto highQuality = Decoder::Jpeg::Encode(rgb, 31, 23, 3, 100, sampling);
|
|
Require(ReadFrame(highQuality).components == 3 && ReadFrame(highQuality).sampling == factor);
|
|
}
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, width, height, 3, 90, static_cast<Decoder::Jpeg::Sampling>(3)); }));
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, width, height, 3, 90, Decoder::Jpeg::Sampling::Yuv444, 1, 1); }));
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, width, height, 3, 90, Decoder::Jpeg::Sampling::Yuv444, 0x10000, 0); }));
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::Encode(rgb, width, height, 3, 90, Decoder::Jpeg::Sampling::Yuv444, 0, 0x10000); }));
|
|
|
|
Require(!Decoder::Jpeg::Decode({}).has_value());
|
|
const std::vector<std::uint8_t> garbage{1, 2, 3, 4, 5, 6, 7, 8};
|
|
Require(!Decoder::Jpeg::Decode(garbage).has_value());
|
|
|
|
Require(Decoder::Jpeg::ScaledDimension(7, 1) == 7 && Decoder::Jpeg::ScaledDimension(7, 2) == 4);
|
|
Require(Decoder::Jpeg::ScaledDimension(7, 4) == 2 && Decoder::Jpeg::ScaledDimension(7, 8) == 1);
|
|
Require(Decoder::Jpeg::ScaledDimension(32, 8) == 4 && Decoder::Jpeg::ScaledDimension(33, 8) == 5 && Decoder::Jpeg::ScaledDimension(1, 8) == 1);
|
|
Require(Decoder::Jpeg::ScaledDimension(0xFFFF, 2) == 0x8000);
|
|
for (const std::uint32_t denominator : {0u, 3u, 16u}) {
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::ScaledDimension(7, denominator); }));
|
|
Require(ThrowsInvalidArgument([&] { Decoder::Jpeg::DecodeScaled(rgbJpeg, denominator); }));
|
|
}
|
|
|
|
const auto boxError = [](const std::vector<std::uint8_t>& source, std::uint32_t sourceWidth, std::uint32_t sourceHeight,
|
|
std::uint32_t channels, const Decoder::Jpeg::Image& scaled, std::uint32_t denominator) {
|
|
long error = 0;
|
|
for (std::uint32_t y = 0; y < scaled.height; ++y) {
|
|
for (std::uint32_t x = 0; x < scaled.width; ++x) {
|
|
for (std::uint32_t c = 0; c < channels; ++c) {
|
|
long sum = 0;
|
|
long count = 0;
|
|
for (std::uint32_t sy = y * denominator; sy < std::min(sourceHeight, (y + 1) * denominator); ++sy) {
|
|
for (std::uint32_t sx = x * denominator; sx < std::min(sourceWidth, (x + 1) * denominator); ++sx) {
|
|
sum += source[(static_cast<std::size_t>(sy) * sourceWidth + sx) * channels + c];
|
|
++count;
|
|
}
|
|
}
|
|
const auto average = static_cast<std::uint8_t>((sum + count / 2) / count);
|
|
error += Difference(average, scaled.pixels[(static_cast<std::size_t>(y) * scaled.width + x) * channels + c]);
|
|
}
|
|
}
|
|
}
|
|
return error / static_cast<long>(static_cast<std::size_t>(scaled.width) * scaled.height * channels);
|
|
};
|
|
|
|
for (const std::uint32_t denominator : {1u, 2u, 4u, 8u}) {
|
|
const auto scaled = Decoder::Jpeg::DecodeScaled(rgbJpeg, denominator);
|
|
Require(scaled.has_value() && scaled->channels == 3);
|
|
Require(scaled->width == width / denominator && scaled->height == height / denominator);
|
|
Require(scaled->pixels.size() == static_cast<std::size_t>(scaled->width) * scaled->height * 3);
|
|
Require(boxError(rgb, width, height, 3, *scaled, denominator) < 4);
|
|
|
|
const auto grayScaled = Decoder::Jpeg::DecodeScaled(grayJpeg, denominator);
|
|
Require(grayScaled.has_value() && grayScaled->channels == 1);
|
|
Require(grayScaled->width == width / denominator && grayScaled->height == height / denominator);
|
|
Require(boxError(gray, width, height, 1, *grayScaled, denominator) < 4);
|
|
}
|
|
|
|
std::vector<std::uint8_t> odd(31 * 23 * 3);
|
|
for (std::uint32_t y = 0; y < 23; ++y) {
|
|
for (std::uint32_t x = 0; x < 31; ++x) {
|
|
std::uint8_t* pixel = &odd[(y * 31 + x) * 3];
|
|
pixel[0] = static_cast<std::uint8_t>(x * 8);
|
|
pixel[1] = static_cast<std::uint8_t>(y * 11);
|
|
pixel[2] = static_cast<std::uint8_t>(200 - x * 2);
|
|
}
|
|
}
|
|
const auto oddJpeg = Decoder::Jpeg::Encode(odd, 31, 23, 3, 95, Decoder::Jpeg::Sampling::Yuv420);
|
|
for (const std::uint32_t denominator : {2u, 4u, 8u}) {
|
|
const auto scaled = Decoder::Jpeg::DecodeScaled(oddJpeg, denominator);
|
|
Require(scaled.has_value() && scaled->channels == 3);
|
|
Require(scaled->width == Decoder::Jpeg::ScaledDimension(31, denominator) && scaled->height == Decoder::Jpeg::ScaledDimension(23, denominator));
|
|
Require(boxError(odd, 31, 23, 3, *scaled, denominator) < 8);
|
|
}
|
|
|
|
Require(!Decoder::Jpeg::DecodeScaled({}, 2).has_value());
|
|
Require(!Decoder::Jpeg::DecodeScaled(garbage, 2).has_value());
|
|
const std::vector<std::uint8_t> truncated(rgbJpeg.begin(), rgbJpeg.begin() + static_cast<std::ptrdiff_t>(rgbJpeg.size() / 2));
|
|
Require(!Decoder::Jpeg::DecodeScaled(truncated, 2).has_value());
|
|
|
|
return 0;
|
|
}
|