ladybird/Userland/Libraries/LibVideo/VP9/SyntaxElementCounter.h
Zaggy1024 1fcac52e77 LibVideo/VP9: Count syntax elements in TileContext, and sum at the end
Syntax element counters were previously accessed across tiles, which
would cause a race condition updating the counts in a tile-threaded
mode.
2023-04-23 23:14:30 +02:00

49 lines
1.6 KiB
C++

/*
* 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 {
class SyntaxElementCounter final {
public:
SyntaxElementCounter();
/* (8.3) Clear Counts Process */
void clear_counts();
u32 m_counts_intra_mode[BLOCK_SIZE_GROUPS][INTRA_MODES];
u32 m_counts_uv_mode[INTRA_MODES][INTRA_MODES];
u32 m_counts_partition[PARTITION_CONTEXTS][PARTITION_TYPES];
u32 m_counts_interp_filter[INTERP_FILTER_CONTEXTS][SWITCHABLE_FILTERS];
u32 m_counts_inter_mode[INTER_MODE_CONTEXTS][INTER_MODES];
u32 m_counts_tx_size[TX_SIZES][TX_SIZE_CONTEXTS][TX_SIZES];
u32 m_counts_is_inter[IS_INTER_CONTEXTS][2];
u32 m_counts_comp_mode[COMP_MODE_CONTEXTS][2];
u32 m_counts_single_ref[REF_CONTEXTS][2][2];
u32 m_counts_comp_ref[REF_CONTEXTS][2];
u32 m_counts_skip[SKIP_CONTEXTS][2];
u32 m_counts_mv_joint[MV_JOINTS];
u32 m_counts_mv_sign[2][2];
u32 m_counts_mv_class[2][MV_CLASSES];
u32 m_counts_mv_class0_bit[2][CLASS0_SIZE];
u32 m_counts_mv_class0_fr[2][CLASS0_SIZE][MV_FR_SIZE];
u32 m_counts_mv_class0_hp[2][2];
u32 m_counts_mv_bits[2][MV_OFFSET_BITS][2];
u32 m_counts_mv_fr[2][MV_FR_SIZE];
u32 m_counts_mv_hp[2][2];
u32 m_counts_token[TX_SIZES][BLOCK_TYPES][REF_TYPES][COEF_BANDS][PREV_COEF_CONTEXTS][UNCONSTRAINED_NODES];
u32 m_counts_more_coefs[TX_SIZES][BLOCK_TYPES][REF_TYPES][COEF_BANDS][PREV_COEF_CONTEXTS][2];
SyntaxElementCounter operator+(SyntaxElementCounter const&) const;
SyntaxElementCounter& operator+=(SyntaxElementCounter const&);
};
}