mirror of
https://github.com/RGBCube/serenity
synced 2025-06-01 09:28:13 +00:00
514559f074
The class that was previously named Decoder handled section 6.X.X of the spec, which actually deals with parsing out the syntax of the data, not the actual decoding logic which is specified in section 8.X.X. The new Decoder class will be in charge of owning and running the Parser, as well as implementing all of the decoding behavior.
89 lines
2.4 KiB
C++
89 lines
2.4 KiB
C++
/*
|
|
* Copyright (c) 2021, Hunter Salyer <thefalsehonesty@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "BitStream.h"
|
|
#include "Enums.h"
|
|
#include "ProbabilityTables.h"
|
|
#include "SyntaxElementCounter.h"
|
|
|
|
namespace Video::VP9 {
|
|
|
|
class Parser;
|
|
|
|
class TreeParser {
|
|
public:
|
|
explicit TreeParser(Parser& decoder)
|
|
: m_decoder(decoder)
|
|
{
|
|
}
|
|
|
|
class TreeSelection {
|
|
public:
|
|
union TreeSelectionValue {
|
|
int const* m_tree;
|
|
int m_value;
|
|
};
|
|
|
|
TreeSelection(int const* values);
|
|
TreeSelection(int value);
|
|
|
|
bool is_single_value() const { return m_is_single_value; }
|
|
int get_single_value() const { return m_value.m_value; }
|
|
int const* get_tree_value() const { return m_value.m_tree; }
|
|
|
|
private:
|
|
bool m_is_single_value;
|
|
TreeSelectionValue m_value;
|
|
};
|
|
|
|
/* (9.3.3) */
|
|
template<typename T = int>
|
|
T parse_tree(SyntaxElementType type);
|
|
/* (9.3.1) */
|
|
TreeSelection select_tree(SyntaxElementType type);
|
|
/* (9.3.2) */
|
|
u8 select_tree_probability(SyntaxElementType type, u8 node);
|
|
/* (9.3.4) */
|
|
void count_syntax_element(SyntaxElementType type, int value);
|
|
|
|
void set_default_intra_mode_variables(u8 idx, u8 idy)
|
|
{
|
|
m_idx = idx;
|
|
m_idy = idy;
|
|
}
|
|
|
|
private:
|
|
u8 calculate_partition_probability(u8 node);
|
|
u8 calculate_default_intra_mode_probability(u8 node);
|
|
u8 calculate_default_uv_mode_probability(u8 node);
|
|
u8 calculate_intra_mode_probability(u8 node);
|
|
u8 calculate_sub_intra_mode_probability(u8 node);
|
|
u8 calculate_uv_mode_probability(u8 node);
|
|
u8 calculate_segment_id_probability(u8 node);
|
|
u8 calculate_skip_probability();
|
|
u8 calculate_seg_id_predicted_probability();
|
|
u8 calculate_is_inter_probability();
|
|
u8 calculate_comp_mode_probability();
|
|
u8 calculate_comp_ref_probability();
|
|
u8 calculate_single_ref_p1_probability();
|
|
u8 calculate_single_ref_p2_probability();
|
|
u8 calculate_tx_size_probability(u8 node);
|
|
u8 calculate_inter_mode_probability(u8 node);
|
|
u8 calculate_interp_filter_probability(u8 node);
|
|
|
|
Parser& m_decoder;
|
|
// m_ctx is a member variable because it is required for syntax element counting (section 9.3.4)
|
|
u8 m_ctx { 0 };
|
|
|
|
// These are variables necessary for parsing tree data, but aren't useful otherwise, so they're
|
|
// not stored in the Decoder itself.
|
|
u8 m_idx { 0 };
|
|
u8 m_idy { 0 };
|
|
};
|
|
|
|
}
|