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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
|
/*
* Copyright (c) 2001-2016, Alliance for Open Media. All rights reserved
*
* This source code is subject to the terms of the BSD 2 Clause License and
* the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
* was not distributed with this source code in the LICENSE file, you can
* obtain it at www.aomedia.org/license/software. If the Alliance for Open
* Media Patent License 1.0 was not distributed with this source code in the
* PATENTS file, you can obtain it at www.aomedia.org/license/patent.
*/
#ifndef DAV1D_SRC_REF_MVS_H
#define DAV1D_SRC_REF_MVS_H
#include <stddef.h>
#include "src/levels.h"
#include "src/tables.h"
typedef struct refmvs {
mv mv[2];
int8_t ref[2]; // [0] = 0: intra=1, [1] = -1: comp=0
int8_t mode, sb_type;
} refmvs;
typedef struct candidate_mv {
mv this_mv;
mv comp_mv;
int weight;
} candidate_mv;
typedef struct AV1_COMMON AV1_COMMON;
// call once per frame thread
AV1_COMMON *av1_alloc_ref_mv_common(void);
void av1_free_ref_mv_common(AV1_COMMON *cm);
// call once per frame
int av1_init_ref_mv_common(AV1_COMMON *cm,
int w8, int h8,
ptrdiff_t stride,
int allow_sb128,
refmvs *cur,
refmvs *ref_mvs[7],
unsigned cur_poc,
const unsigned ref_poc[7],
const unsigned ref_ref_poc[7][7],
const Dav1dWarpedMotionParams gmv[7],
int allow_hp, int force_int_mv,
int allow_ref_frame_mvs, int order_hint);
// call for start of each sbrow per tile
void av1_init_ref_mv_tile_row(AV1_COMMON *cm,
int tile_col_start4, int tile_col_end4,
int row_start4, int row_end4);
// call for each block
void av1_find_ref_mvs(candidate_mv *mvstack, int *cnt, mv (*mvlist)[2],
int *ctx, int refidx[2], int w4, int h4,
enum BlockSize bs, enum BlockPartition bp,
int by4, int bx4, int tile_col_start4,
int tile_col_end4, int tile_row_start4,
int tile_row_end4, AV1_COMMON *cm);
extern const uint8_t bs_to_sbtype[];
extern const uint8_t sbtype_to_bs[];
static inline void splat_oneref_mv(refmvs *r, const ptrdiff_t stride,
const int by4, const int bx4,
const enum BlockSize bs,
const enum InterPredMode mode,
const int ref, const mv mv,
const int is_interintra)
{
const int bw4 = dav1d_block_dimensions[bs][0];
int bh4 = dav1d_block_dimensions[bs][1];
r += by4 * stride + bx4;
const refmvs tmpl = (refmvs) {
.ref = { ref + 1, is_interintra ? 0 : -1 },
.mv = { mv },
.sb_type = bs_to_sbtype[bs],
.mode = N_INTRA_PRED_MODES + mode,
};
do {
for (int x = 0; x < bw4; x++)
r[x] = tmpl;
r += stride;
} while (--bh4);
}
static inline void splat_intrabc_mv(refmvs *r, const ptrdiff_t stride,
const int by4, const int bx4,
const enum BlockSize bs, const mv mv)
{
const int bw4 = dav1d_block_dimensions[bs][0];
int bh4 = dav1d_block_dimensions[bs][1];
r += by4 * stride + bx4;
const refmvs tmpl = (refmvs) {
.ref = { 0, -1 },
.mv = { mv },
.sb_type = bs_to_sbtype[bs],
.mode = DC_PRED,
};
do {
for (int x = 0; x < bw4; x++)
r[x] = tmpl;
r += stride;
} while (--bh4);
}
static inline void splat_tworef_mv(refmvs *r, const ptrdiff_t stride,
const int by4, const int bx4,
const enum BlockSize bs,
const enum CompInterPredMode mode,
const int ref1, const int ref2,
const mv mv1, const mv mv2)
{
const int bw4 = dav1d_block_dimensions[bs][0];
int bh4 = dav1d_block_dimensions[bs][1];
r += by4 * stride + bx4;
const refmvs tmpl = (refmvs) {
.ref = { ref1 + 1, ref2 + 1 },
.mv = { mv1, mv2 },
.sb_type = bs_to_sbtype[bs],
.mode = N_INTRA_PRED_MODES + N_INTER_PRED_MODES + mode,
};
do {
for (int x = 0; x < bw4; x++)
r[x] = tmpl;
r += stride;
} while (--bh4);
}
static inline void splat_intraref(refmvs *r, const ptrdiff_t stride,
const int by4, const int bx4,
const enum BlockSize bs,
const enum IntraPredMode mode)
{
const int bw4 = dav1d_block_dimensions[bs][0];
int bh4 = dav1d_block_dimensions[bs][1];
r += by4 * stride + bx4;
do {
int x;
for (x = 0; x < bw4; x++)
r[x] = (refmvs) {
.ref = { 0, -1 },
.mv = { [0] = { .y = -0x8000, .x = -0x8000 }, },
.sb_type = bs_to_sbtype[bs],
.mode = mode,
};
r += stride;
} while (--bh4);
}
static inline void fix_mv_precision(const Dav1dFrameHeader *const hdr,
mv *const mv)
{
if (hdr->force_integer_mv) {
const int xmod = mv->x & 7;
mv->x &= ~7;
mv->x += (xmod > 4 - (mv->x < 0)) << 3;
const int ymod = mv->y & 7;
mv->y &= ~7;
mv->y += (ymod > 4 - (mv->y < 0)) << 3;
} else if (!hdr->hp) {
if (mv->x & 1) {
if (mv->x < 0) mv->x++;
else mv->x--;
}
if (mv->y & 1) {
if (mv->y < 0) mv->y++;
else mv->y--;
}
}
}
#endif /* DAV1D_SRC_REF_MVS_H */
|
