1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
/*
* Copyright 2020 The libgav1 Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef LIBGAV1_SRC_UTILS_REFERENCE_INFO_H_
#define LIBGAV1_SRC_UTILS_REFERENCE_INFO_H_
#include <array>
#include <cstdint>
#include "src/utils/array_2d.h"
#include "src/utils/constants.h"
#include "src/utils/types.h"
namespace libgav1 {
// This struct collects some members related to reference frames in one place to
// make it easier to pass them as parameters to some dsp functions.
struct ReferenceInfo {
// Initialize |motion_field_reference_frame| so that
// Tile::StoreMotionFieldMvsIntoCurrentFrame() can skip some updates when
// the updates are the same as the initialized value.
// Set to kReferenceFrameIntra instead of kReferenceFrameNone to simplify
// branch conditions in motion field projection.
// The following memory initialization of contiguous memory is very fast. It
// is not recommended to make the initialization multi-threaded, unless the
// memory which needs to be initialized in each thread is still contiguous.
LIBGAV1_MUST_USE_RESULT bool Reset(int rows, int columns) {
return motion_field_reference_frame.Reset(rows, columns,
/*zero_initialize=*/true) &&
motion_field_mv.Reset(
rows, columns,
#if LIBGAV1_MSAN
// It is set in Tile::StoreMotionFieldMvsIntoCurrentFrame() only
// for qualified blocks. In MotionFieldProjectionKernel() dsp
// optimizations, it is read no matter it was set or not.
/*zero_initialize=*/true
#else
/*zero_initialize=*/false
#endif
);
}
// All members are used by inter frames only.
// For intra frames, they are not initialized.
std::array<uint8_t, kNumReferenceFrameTypes> order_hint;
// An example when |relative_distance_from| does not equal
// -|relative_distance_to|:
// |relative_distance_from| = GetRelativeDistance(7, 71, 25) = -64
// -|relative_distance_to| = -GetRelativeDistance(71, 7, 25) = 64
// This is why we need both |relative_distance_from| and
// |relative_distance_to|.
// |relative_distance_from|: Relative distances from reference frames to this
// frame.
std::array<int8_t, kNumReferenceFrameTypes> relative_distance_from;
// |relative_distance_to|: Relative distances to reference frames.
std::array<int8_t, kNumReferenceFrameTypes> relative_distance_to;
// Skip motion field projection of specific types of frames if their
// |relative_distance_to| is negative or too large.
std::array<bool, kNumReferenceFrameTypes> skip_references;
// Lookup table to get motion field projection division multiplier of specific
// types of frames. Derived from kProjectionMvDivisionLookup.
std::array<int16_t, kNumReferenceFrameTypes> projection_divisions;
// The current frame's |motion_field_reference_frame| and |motion_field_mv_|
// are guaranteed to be allocated only when refresh_frame_flags is not 0.
// Array of size (rows4x4 / 2) x (columns4x4 / 2). Entry at i, j corresponds
// to MfRefFrames[i * 2 + 1][j * 2 + 1] in the spec.
Array2D<ReferenceFrameType> motion_field_reference_frame;
// Array of size (rows4x4 / 2) x (columns4x4 / 2). Entry at i, j corresponds
// to MfMvs[i * 2 + 1][j * 2 + 1] in the spec.
Array2D<MotionVector> motion_field_mv;
};
} // namespace libgav1
#endif // LIBGAV1_SRC_UTILS_REFERENCE_INFO_H_
|
