RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-27 10:57:35 +00:00
Code Issues Activity Actions

LibVideo/VP9: Implement syntax element counter

Browse source
This commit is contained in:
FalseHonesty 2021-06-05 16:28:08 -04:00 • committed by Ali Mohammad Pur
parent 8dd9d1bbdb
commit 18759ff56d
3 changed files with 112 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
Userland/Libraries/LibVideo/CMakeLists.txt
View file

@ -4,6 +4,7 @@ set(SOURCES
VP9/Enums.h
VP9/ProbabilityTables.cpp
VP9/Symbols.h
VP9/SyntaxElementCounter.cpp
)
serenity_lib(LibVideo video)

37
Userland/Libraries/LibVideo/VP9/SyntaxElementCounter.cpp Normal file
View file

@ -0,0 +1,37 @@
/*
* Copyright (c) 2021, Hunter Salyer <thefalsehonesty@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include "SyntaxElementCounter.h"
namespace Video::VP9 {
void SyntaxElementCounter::clear_counts()
{
__builtin_memset(counts_intra_mode, 0, BLOCK_SIZE_GROUPS * INTRA_MODES);
__builtin_memset(counts_uv_mode, 0, INTRA_MODES * INTRA_MODES);
__builtin_memset(counts_partition, 0, PARTITION_CONTEXTS * PARTITION_TYPES);
__builtin_memset(counts_interp_filter, 0, INTERP_FILTER_CONTEXTS * SWITCHABLE_FILTERS);
__builtin_memset(counts_inter_mode, 0, INTER_MODE_CONTEXTS * INTER_MODES);
__builtin_memset(counts_tx_size, 0, TX_SIZES * TX_SIZE_CONTEXTS * TX_SIZES);
__builtin_memset(counts_is_inter, 0, IS_INTER_CONTEXTS * 2);
__builtin_memset(counts_comp_mode, 0, COMP_MODE_CONTEXTS * 2);
__builtin_memset(counts_single_ref, 0, REF_CONTEXTS * 2 * 2);
__builtin_memset(counts_comp_ref, 0, REF_CONTEXTS * 2);
__builtin_memset(counts_skip, 0, SKIP_CONTEXTS * 2);
__builtin_memset(counts_mv_joint, 0, MV_JOINTS);
__builtin_memset(counts_mv_sign, 0, 2 * 2);
__builtin_memset(counts_mv_class, 0, 2 * MV_CLASSES);
__builtin_memset(counts_mv_class0_bit, 0, 2 * CLASS0_SIZE);
__builtin_memset(counts_mv_class0_fr, 0, 2 * CLASS0_SIZE * MV_FR_SIZE);
__builtin_memset(counts_mv_class0_hp, 0, 2 * 2);
__builtin_memset(counts_mv_bits, 0, 2 * MV_OFFSET_BITS * 2);
__builtin_memset(counts_mv_fr, 0, 2 * MV_FR_SIZE);
__builtin_memset(counts_mv_hp, 0, 2 * 2);
__builtin_memset(counts_token, 0, TX_SIZES * BLOCK_TYPES * REF_TYPES * COEF_BANDS * PREV_COEF_CONTEXTS * UNCONSTRAINED_NODES);
__builtin_memset(counts_more_coefs, 0, TX_SIZES * BLOCK_TYPES * REF_TYPES * COEF_BANDS * PREV_COEF_CONTEXTS * 2);
}
}

74
Userland/Libraries/LibVideo/VP9/SyntaxElementCounter.h Normal file
View file

@ -0,0 +1,74 @@
/*
* Copyright (c) 2021, Hunter Salyer <thefalsehonesty@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#pragma once
#include "Symbols.h"
#include <AK/Types.h>
namespace Video::VP9 {
enum class SyntaxElementType {
Partition,
DefaultIntraMode,
DefaultUVMode,
IntraMode,
SubIntraMode,
UVMode,
SegmentID,
Skip,
SegIDPredicted,
IsInter,
CompMode,
CompRef,
SingleRefP1,
SingleRefP2,
MVSign,
MVClass0Bit,
MVBit,
TXSize,
InterMode,
InterpFilter,
MVJoint,
MVClass,
MVClass0FR,
MVClass0HP,
MVFR,
MVHP,
Token,
MoreCoefs,
};
class SyntaxElementCounter final {
public:
void clear_counts();
private:
u8 counts_intra_mode[BLOCK_SIZE_GROUPS][INTRA_MODES];
u8 counts_uv_mode[INTRA_MODES][INTRA_MODES];
u8 counts_partition[PARTITION_CONTEXTS][PARTITION_TYPES];
u8 counts_interp_filter[INTERP_FILTER_CONTEXTS][SWITCHABLE_FILTERS];
u8 counts_inter_mode[INTER_MODE_CONTEXTS][INTER_MODES];
u8 counts_tx_size[TX_SIZES][TX_SIZE_CONTEXTS][TX_SIZES];
u8 counts_is_inter[IS_INTER_CONTEXTS][2];
u8 counts_comp_mode[COMP_MODE_CONTEXTS][2];
u8 counts_single_ref[REF_CONTEXTS][2][2];
u8 counts_comp_ref[REF_CONTEXTS][2];
u8 counts_skip[SKIP_CONTEXTS][2];
u8 counts_mv_joint[MV_JOINTS];
u8 counts_mv_sign[2][2];
u8 counts_mv_class[2][MV_CLASSES];
u8 counts_mv_class0_bit[2][CLASS0_SIZE];
u8 counts_mv_class0_fr[2][CLASS0_SIZE][MV_FR_SIZE];
u8 counts_mv_class0_hp[2][2];
u8 counts_mv_bits[2][MV_OFFSET_BITS][2];
u8 counts_mv_fr[2][MV_FR_SIZE];
u8 counts_mv_hp[2][2];
u8 counts_token[TX_SIZES][BLOCK_TYPES][REF_TYPES][COEF_BANDS][PREV_COEF_CONTEXTS][UNCONSTRAINED_NODES];
u8 counts_more_coefs[TX_SIZES][BLOCK_TYPES][REF_TYPES][COEF_BANDS][PREV_COEF_CONTEXTS][2];
};
}