mirror of
https://github.com/TelegramMessenger/Telegram-iOS.git
synced 2026-05-21 18:20:41 +00:00
Isaac
51 lines
1.5 KiB
C++
51 lines
1.5 KiB
C++
#include "sprite_encode.h"
|
|
#include <cassert>
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
#include <vector>
|
|
|
|
using namespace subcodec;
|
|
|
|
#define SPRITE_PX 64
|
|
|
|
static int test_alpha_encode() {
|
|
printf("test_alpha_encode...\n");
|
|
|
|
int padded = 96;
|
|
int half_chroma = padded / 2;
|
|
|
|
std::vector<uint8_t> y(padded * padded);
|
|
std::vector<uint8_t> cb(half_chroma * half_chroma, 128);
|
|
std::vector<uint8_t> cr(half_chroma * half_chroma, 128);
|
|
for (int r = 0; r < padded; r++)
|
|
for (int c = 0; c < padded; c++)
|
|
y[r * padded + c] = static_cast<uint8_t>((r + c) % 256);
|
|
|
|
std::vector<uint8_t> alpha(padded * padded, 255);
|
|
|
|
auto enc = SpriteEncoder::create({SPRITE_PX, SPRITE_PX, 26});
|
|
assert(enc.has_value());
|
|
|
|
auto result = enc->encode(y.data(), padded, cb.data(), half_chroma,
|
|
cr.data(), half_chroma, alpha.data(), padded, 0);
|
|
assert(result.has_value());
|
|
assert(result->color.size() == 36); // 6x6
|
|
assert(result->alpha.size() == 36);
|
|
|
|
auto result2 = enc->encode(y.data(), padded, cb.data(), half_chroma,
|
|
cr.data(), half_chroma, alpha.data(), padded, 1);
|
|
assert(result2.has_value());
|
|
assert(result2->color.size() == 36);
|
|
assert(result2->alpha.size() == 36);
|
|
|
|
for (const auto& mb : result2->alpha) {
|
|
if (mb.mb_type != MbType::SKIP)
|
|
assert(mb.cbp_chroma == 0);
|
|
}
|
|
|
|
printf(" PASS\n");
|
|
return 0;
|
|
}
|
|
|
|
int main() { return test_alpha_encode(); }
|