annotate motion_est_template.c @ 4253:b2c06cdd6154 libavcodec

remove pretty much useless skiping of some predictors this has pretty much no quality or speed effect except very small random changes
author michael
date Fri, 01 Dec 2006 20:38:54 +0000
parents 507ba4483ffc
children f775fd9f3b05
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1 /*
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
2 * Motion estimation
1739
07a484280a82 copyright year update of the files i touched and remembered, things look annoyingly unmaintained otherwise
michael
parents: 1708
diff changeset
3 * Copyright (c) 2002-2004 Michael Niedermayer
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
4 *
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3293
diff changeset
5 * This file is part of FFmpeg.
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3293
diff changeset
6 *
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3293
diff changeset
7 * FFmpeg is free software; you can redistribute it and/or
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
8 * modify it under the terms of the GNU Lesser General Public
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
9 * License as published by the Free Software Foundation; either
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3293
diff changeset
10 * version 2.1 of the License, or (at your option) any later version.
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
11 *
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3293
diff changeset
12 * FFmpeg is distributed in the hope that it will be useful,
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
15 * Lesser General Public License for more details.
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
16 *
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
17 * You should have received a copy of the GNU Lesser General Public
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3293
diff changeset
18 * License along with FFmpeg; if not, write to the Free Software
3036
0b546eab515d Update licensing information: The FSF changed postal address.
diego
parents: 2979
diff changeset
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
20 *
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
21 */
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
22
1106
1e39f273ecd6 per file doxy
michaelni
parents: 1013
diff changeset
23 /**
1e39f273ecd6 per file doxy
michaelni
parents: 1013
diff changeset
24 * @file motion_est_template.c
1e39f273ecd6 per file doxy
michaelni
parents: 1013
diff changeset
25 * Motion estimation template.
1e39f273ecd6 per file doxy
michaelni
parents: 1013
diff changeset
26 */
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
27
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
28 //lets hope gcc will remove the unused vars ...(gcc 3.2.2 seems to do it ...)
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
29 #define LOAD_COMMON\
2522
e25782262d7d kill warnings patch by (Mns Rullgrd <mru inprovide com>)
michael
parents: 2327
diff changeset
30 uint32_t attribute_unused * const score_map= c->score_map;\
e25782262d7d kill warnings patch by (Mns Rullgrd <mru inprovide com>)
michael
parents: 2327
diff changeset
31 const int attribute_unused xmin= c->xmin;\
e25782262d7d kill warnings patch by (Mns Rullgrd <mru inprovide com>)
michael
parents: 2327
diff changeset
32 const int attribute_unused ymin= c->ymin;\
e25782262d7d kill warnings patch by (Mns Rullgrd <mru inprovide com>)
michael
parents: 2327
diff changeset
33 const int attribute_unused xmax= c->xmax;\
e25782262d7d kill warnings patch by (Mns Rullgrd <mru inprovide com>)
michael
parents: 2327
diff changeset
34 const int attribute_unused ymax= c->ymax;\
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
35 uint8_t *mv_penalty= c->current_mv_penalty;\
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
36 const int pred_x= c->pred_x;\
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
37 const int pred_y= c->pred_y;\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
38
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
39 #define CHECK_HALF_MV(dx, dy, x, y)\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
40 {\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
41 const int hx= 2*(x)+(dx);\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
42 const int hy= 2*(y)+(dy);\
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
43 d= cmp(s, x, y, dx, dy, size, h, ref_index, src_index, cmp_sub, chroma_cmp_sub, flags);\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
44 d += (mv_penalty[hx - pred_x] + mv_penalty[hy - pred_y])*penalty_factor;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
45 COPY3_IF_LT(dmin, d, bx, hx, by, hy)\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
46 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
47
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
48 #if 0
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
49 static int hpel_motion_search)(MpegEncContext * s,
2979
bfabfdf9ce55 COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 2967
diff changeset
50 int *mx_ptr, int *my_ptr, int dmin,
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
51 uint8_t *ref_data[3],
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
52 int size)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
53 {
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
54 const int xx = 16 * s->mb_x + 8*(n&1);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
55 const int yy = 16 * s->mb_y + 8*(n>>1);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
56 const int mx = *mx_ptr;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
57 const int my = *my_ptr;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
58 const int penalty_factor= c->sub_penalty_factor;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
59
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
60 LOAD_COMMON
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
61
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
62 // INIT;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
63 //FIXME factorize
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
64 me_cmp_func cmp, chroma_cmp, cmp_sub, chroma_cmp_sub;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
65
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
66 if(s->no_rounding /*FIXME b_type*/){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
67 hpel_put= &s->dsp.put_no_rnd_pixels_tab[size];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
68 chroma_hpel_put= &s->dsp.put_no_rnd_pixels_tab[size+1];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
69 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
70 hpel_put=& s->dsp.put_pixels_tab[size];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
71 chroma_hpel_put= &s->dsp.put_pixels_tab[size+1];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
72 }
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
73 cmpf= s->dsp.me_cmp[size];
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
74 chroma_cmpf= s->dsp.me_cmp[size+1];
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
75 cmp_sub= s->dsp.me_sub_cmp[size];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
76 chroma_cmp_sub= s->dsp.me_sub_cmp[size+1];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
77
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
78 if(c->skip){ //FIXME somehow move up (benchmark)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
79 *mx_ptr = 0;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
80 *my_ptr = 0;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
81 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
82 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
83
2015
3ab8f3e2ae6a moving motion estimation specific variables from MpegEncContext -> MotionEstContext
michael
parents: 2014
diff changeset
84 if(c->avctx->me_cmp != c->avctx->me_sub_cmp){
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
85 CMP_HPEL(dmin, 0, 0, mx, my, size);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
86 if(mx || my)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
87 dmin += (mv_penalty[2*mx - pred_x] + mv_penalty[2*my - pred_y])*penalty_factor;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
88 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
89
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
90 if (mx > xmin && mx < xmax &&
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
91 my > ymin && my < ymax) {
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
92 int bx=2*mx, by=2*my;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
93 int d= dmin;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
94
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
95 CHECK_HALF_MV(1, 1, mx-1, my-1)
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
96 CHECK_HALF_MV(0, 1, mx , my-1)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
97 CHECK_HALF_MV(1, 1, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
98 CHECK_HALF_MV(1, 0, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
99 CHECK_HALF_MV(1, 0, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
100 CHECK_HALF_MV(1, 1, mx-1, my )
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
101 CHECK_HALF_MV(0, 1, mx , my )
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
102 CHECK_HALF_MV(1, 1, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
103
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
104 assert(bx >= xmin*2 || bx <= xmax*2 || by >= ymin*2 || by <= ymax*2);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
105
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
106 *mx_ptr = bx;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
107 *my_ptr = by;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
108 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
109 *mx_ptr =2*mx;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
110 *my_ptr =2*my;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
111 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
112
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
113 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
114 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
115
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
116 #else
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
117 static int hpel_motion_search(MpegEncContext * s,
2979
bfabfdf9ce55 COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 2967
diff changeset
118 int *mx_ptr, int *my_ptr, int dmin,
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
119 int src_index, int ref_index,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
120 int size, int h)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
121 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
122 MotionEstContext * const c= &s->me;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
123 const int mx = *mx_ptr;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
124 const int my = *my_ptr;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
125 const int penalty_factor= c->sub_penalty_factor;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
126 me_cmp_func cmp_sub, chroma_cmp_sub;
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
127 int bx=2*mx, by=2*my;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
128
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
129 LOAD_COMMON
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
130 int flags= c->sub_flags;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
131
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
132 //FIXME factorize
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
133
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
134 cmp_sub= s->dsp.me_sub_cmp[size];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
135 chroma_cmp_sub= s->dsp.me_sub_cmp[size+1];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
136
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
137 if(c->skip){ //FIXME move out of hpel?
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
138 *mx_ptr = 0;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
139 *my_ptr = 0;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
140 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
141 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
142
2015
3ab8f3e2ae6a moving motion estimation specific variables from MpegEncContext -> MotionEstContext
michael
parents: 2014
diff changeset
143 if(c->avctx->me_cmp != c->avctx->me_sub_cmp){
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
144 dmin= cmp(s, mx, my, 0, 0, size, h, ref_index, src_index, cmp_sub, chroma_cmp_sub, flags);
1011
michaelni
parents: 954
diff changeset
145 if(mx || my || size>0)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
146 dmin += (mv_penalty[2*mx - pred_x] + mv_penalty[2*my - pred_y])*penalty_factor;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
147 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
148
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
149 if (mx > xmin && mx < xmax &&
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
150 my > ymin && my < ymax) {
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
151 int d= dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
152 const int index= (my<<ME_MAP_SHIFT) + mx;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
153 const int t= score_map[(index-(1<<ME_MAP_SHIFT))&(ME_MAP_SIZE-1)]
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
154 + (mv_penalty[bx - pred_x] + mv_penalty[by-2 - pred_y])*c->penalty_factor;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
155 const int l= score_map[(index- 1 )&(ME_MAP_SIZE-1)]
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
156 + (mv_penalty[bx-2 - pred_x] + mv_penalty[by - pred_y])*c->penalty_factor;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
157 const int r= score_map[(index+ 1 )&(ME_MAP_SIZE-1)]
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
158 + (mv_penalty[bx+2 - pred_x] + mv_penalty[by - pred_y])*c->penalty_factor;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
159 const int b= score_map[(index+(1<<ME_MAP_SHIFT))&(ME_MAP_SIZE-1)]
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
160 + (mv_penalty[bx - pred_x] + mv_penalty[by+2 - pred_y])*c->penalty_factor;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
161
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
162 #if 1
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
163 int key;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
164 int map_generation= c->map_generation;
1419
a7a9df478e46 removed unused variable
bellard
parents: 1266
diff changeset
165 #ifndef NDEBUG
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
166 uint32_t *map= c->map;
1419
a7a9df478e46 removed unused variable
bellard
parents: 1266
diff changeset
167 #endif
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
168 key= ((my-1)<<ME_MAP_MV_BITS) + (mx) + map_generation;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
169 assert(map[(index-(1<<ME_MAP_SHIFT))&(ME_MAP_SIZE-1)] == key);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
170 key= ((my+1)<<ME_MAP_MV_BITS) + (mx) + map_generation;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
171 assert(map[(index+(1<<ME_MAP_SHIFT))&(ME_MAP_SIZE-1)] == key);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
172 key= ((my)<<ME_MAP_MV_BITS) + (mx+1) + map_generation;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
173 assert(map[(index+1)&(ME_MAP_SIZE-1)] == key);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
174 key= ((my)<<ME_MAP_MV_BITS) + (mx-1) + map_generation;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
175 assert(map[(index-1)&(ME_MAP_SIZE-1)] == key);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
176 #endif
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
177 if(t<=b){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
178 CHECK_HALF_MV(0, 1, mx ,my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
179 if(l<=r){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
180 CHECK_HALF_MV(1, 1, mx-1, my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
181 if(t+r<=b+l){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
182 CHECK_HALF_MV(1, 1, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
183 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
184 CHECK_HALF_MV(1, 1, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
185 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
186 CHECK_HALF_MV(1, 0, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
187 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
188 CHECK_HALF_MV(1, 1, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
189 if(t+l<=b+r){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
190 CHECK_HALF_MV(1, 1, mx-1, my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
191 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
192 CHECK_HALF_MV(1, 1, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
193 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
194 CHECK_HALF_MV(1, 0, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
195 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
196 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
197 if(l<=r){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
198 if(t+l<=b+r){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
199 CHECK_HALF_MV(1, 1, mx-1, my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
200 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
201 CHECK_HALF_MV(1, 1, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
202 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
203 CHECK_HALF_MV(1, 0, mx-1, my)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
204 CHECK_HALF_MV(1, 1, mx-1, my)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
205 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
206 if(t+r<=b+l){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
207 CHECK_HALF_MV(1, 1, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
208 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
209 CHECK_HALF_MV(1, 1, mx-1, my)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
210 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
211 CHECK_HALF_MV(1, 0, mx , my)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
212 CHECK_HALF_MV(1, 1, mx , my)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
213 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
214 CHECK_HALF_MV(0, 1, mx , my)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
215 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
216 assert(bx >= xmin*2 && bx <= xmax*2 && by >= ymin*2 && by <= ymax*2);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
217 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
218
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
219 *mx_ptr = bx;
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
220 *my_ptr = by;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
221
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
222 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
223 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
224 #endif
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
225
2327
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
226 static int no_sub_motion_search(MpegEncContext * s,
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
227 int *mx_ptr, int *my_ptr, int dmin,
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
228 int src_index, int ref_index,
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
229 int size, int h)
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
230 {
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
231 (*mx_ptr)<<=1;
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
232 (*my_ptr)<<=1;
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
233 return dmin;
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
234 }
5e5cf598a48b H.261 encoder by (Maarten Daniels <maarten dot daniels at luc dot ac dot be>)
michael
parents: 2226
diff changeset
235
3959
rathann
parents: 3947
diff changeset
236 inline int ff_get_mb_score(MpegEncContext * s, int mx, int my, int src_index,
2189
70b27300a496 quad tree based motion compensation (currently only 16x16 & 8x8 OBMC blocks, but can be extended to other block sizes easily)
michael
parents: 2184
diff changeset
237 int ref_index, int size, int h, int add_rate)
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
238 {
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
239 // const int check_luma= s->dsp.me_sub_cmp != s->dsp.mb_cmp;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
240 MotionEstContext * const c= &s->me;
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
241 const int penalty_factor= c->mb_penalty_factor;
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
242 const int flags= c->mb_flags;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
243 const int qpel= flags & FLAG_QPEL;
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
244 const int mask= 1+2*qpel;
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
245 me_cmp_func cmp_sub, chroma_cmp_sub;
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
246 int d;
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
247
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
248 LOAD_COMMON
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
249
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
250 //FIXME factorize
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
251
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
252 cmp_sub= s->dsp.mb_cmp[size];
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
253 chroma_cmp_sub= s->dsp.mb_cmp[size+1];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
254
2189
70b27300a496 quad tree based motion compensation (currently only 16x16 & 8x8 OBMC blocks, but can be extended to other block sizes easily)
michael
parents: 2184
diff changeset
255 // assert(!c->skip);
70b27300a496 quad tree based motion compensation (currently only 16x16 & 8x8 OBMC blocks, but can be extended to other block sizes easily)
michael
parents: 2184
diff changeset
256 // assert(c->avctx->me_sub_cmp != c->avctx->mb_cmp);
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
257
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
258 d= cmp(s, mx>>(qpel+1), my>>(qpel+1), mx&mask, my&mask, size, h, ref_index, src_index, cmp_sub, chroma_cmp_sub, flags);
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
259 //FIXME check cbp before adding penalty for (0,0) vector
2189
70b27300a496 quad tree based motion compensation (currently only 16x16 & 8x8 OBMC blocks, but can be extended to other block sizes easily)
michael
parents: 2184
diff changeset
260 if(add_rate && (mx || my || size>0))
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
261 d += (mv_penalty[mx - pred_x] + mv_penalty[my - pred_y])*penalty_factor;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
262
1013
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
263 return d;
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
264 }
5d4c95f323d0 finetuneing thresholds/factors
michaelni
parents: 1011
diff changeset
265
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
266 #define CHECK_QUARTER_MV(dx, dy, x, y)\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
267 {\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
268 const int hx= 4*(x)+(dx);\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
269 const int hy= 4*(y)+(dy);\
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
270 d= cmp(s, x, y, dx, dy, size, h, ref_index, src_index, cmpf, chroma_cmpf, flags);\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
271 d += (mv_penalty[hx - pred_x] + mv_penalty[hy - pred_y])*penalty_factor;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
272 COPY3_IF_LT(dmin, d, bx, hx, by, hy)\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
273 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
274
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
275 static int qpel_motion_search(MpegEncContext * s,
2979
bfabfdf9ce55 COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 2967
diff changeset
276 int *mx_ptr, int *my_ptr, int dmin,
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
277 int src_index, int ref_index,
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
278 int size, int h)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
279 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
280 MotionEstContext * const c= &s->me;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
281 const int mx = *mx_ptr;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
282 const int my = *my_ptr;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
283 const int penalty_factor= c->sub_penalty_factor;
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
284 const int map_generation= c->map_generation;
2015
3ab8f3e2ae6a moving motion estimation specific variables from MpegEncContext -> MotionEstContext
michael
parents: 2014
diff changeset
285 const int subpel_quality= c->avctx->me_subpel_quality;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
286 uint32_t *map= c->map;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
287 me_cmp_func cmpf, chroma_cmpf;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
288 me_cmp_func cmp_sub, chroma_cmp_sub;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
289
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
290 LOAD_COMMON
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
291 int flags= c->sub_flags;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
292
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
293 cmpf= s->dsp.me_cmp[size];
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
294 chroma_cmpf= s->dsp.me_cmp[size+1]; //factorize FIXME
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
295 //FIXME factorize
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
296
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
297 cmp_sub= s->dsp.me_sub_cmp[size];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
298 chroma_cmp_sub= s->dsp.me_sub_cmp[size+1];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
299
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
300 if(c->skip){ //FIXME somehow move up (benchmark)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
301 *mx_ptr = 0;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
302 *my_ptr = 0;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
303 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
304 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
305
2015
3ab8f3e2ae6a moving motion estimation specific variables from MpegEncContext -> MotionEstContext
michael
parents: 2014
diff changeset
306 if(c->avctx->me_cmp != c->avctx->me_sub_cmp){
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
307 dmin= cmp(s, mx, my, 0, 0, size, h, ref_index, src_index, cmp_sub, chroma_cmp_sub, flags);
1011
michaelni
parents: 954
diff changeset
308 if(mx || my || size>0)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
309 dmin += (mv_penalty[4*mx - pred_x] + mv_penalty[4*my - pred_y])*penalty_factor;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
310 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
311
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
312 if (mx > xmin && mx < xmax &&
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
313 my > ymin && my < ymax) {
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
314 int bx=4*mx, by=4*my;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
315 int d= dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
316 int i, nx, ny;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
317 const int index= (my<<ME_MAP_SHIFT) + mx;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
318 const int t= score_map[(index-(1<<ME_MAP_SHIFT) )&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
319 const int l= score_map[(index- 1 )&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
320 const int r= score_map[(index+ 1 )&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
321 const int b= score_map[(index+(1<<ME_MAP_SHIFT) )&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
322 const int c= score_map[(index )&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
323 int best[8];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
324 int best_pos[8][2];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
325
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
326 memset(best, 64, sizeof(int)*8);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
327 #if 1
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
328 if(s->me.dia_size>=2){
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
329 const int tl= score_map[(index-(1<<ME_MAP_SHIFT)-1)&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
330 const int bl= score_map[(index+(1<<ME_MAP_SHIFT)-1)&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
331 const int tr= score_map[(index-(1<<ME_MAP_SHIFT)+1)&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
332 const int br= score_map[(index+(1<<ME_MAP_SHIFT)+1)&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
333
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
334 for(ny= -3; ny <= 3; ny++){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
335 for(nx= -3; nx <= 3; nx++){
2079
75aae96e6b54 fewer overflows
michael
parents: 2072
diff changeset
336 //FIXME this could overflow (unlikely though)
75aae96e6b54 fewer overflows
michael
parents: 2072
diff changeset
337 const int64_t t2= nx*nx*(tr + tl - 2*t) + 4*nx*(tr-tl) + 32*t;
75aae96e6b54 fewer overflows
michael
parents: 2072
diff changeset
338 const int64_t c2= nx*nx*( r + l - 2*c) + 4*nx*( r- l) + 32*c;
75aae96e6b54 fewer overflows
michael
parents: 2072
diff changeset
339 const int64_t b2= nx*nx*(br + bl - 2*b) + 4*nx*(br-bl) + 32*b;
75aae96e6b54 fewer overflows
michael
parents: 2072
diff changeset
340 int score= (ny*ny*(b2 + t2 - 2*c2) + 4*ny*(b2 - t2) + 32*c2 + 512)>>10;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
341 int i;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
342
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
343 if((nx&3)==0 && (ny&3)==0) continue;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
344
2079
75aae96e6b54 fewer overflows
michael
parents: 2072
diff changeset
345 score += (mv_penalty[4*mx + nx - pred_x] + mv_penalty[4*my + ny - pred_y])*penalty_factor;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
346
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
347 // if(nx&1) score-=1024*c->penalty_factor;
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
348 // if(ny&1) score-=1024*c->penalty_factor;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
349
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
350 for(i=0; i<8; i++){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
351 if(score < best[i]){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
352 memmove(&best[i+1], &best[i], sizeof(int)*(7-i));
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
353 memmove(&best_pos[i+1][0], &best_pos[i][0], sizeof(int)*2*(7-i));
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
354 best[i]= score;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
355 best_pos[i][0]= nx + 4*mx;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
356 best_pos[i][1]= ny + 4*my;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
357 break;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
358 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
359 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
360 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
361 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
362 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
363 int tl;
2079
75aae96e6b54 fewer overflows
michael
parents: 2072
diff changeset
364 //FIXME this could overflow (unlikely though)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
365 const int cx = 4*(r - l);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
366 const int cx2= r + l - 2*c;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
367 const int cy = 4*(b - t);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
368 const int cy2= b + t - 2*c;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
369 int cxy;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
370
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
371 if(map[(index-(1<<ME_MAP_SHIFT)-1)&(ME_MAP_SIZE-1)] == (my<<ME_MAP_MV_BITS) + mx + map_generation && 0){ //FIXME
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
372 tl= score_map[(index-(1<<ME_MAP_SHIFT)-1)&(ME_MAP_SIZE-1)];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
373 }else{
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
374 tl= cmp(s, mx-1, my-1, 0, 0, size, h, ref_index, src_index, cmpf, chroma_cmpf, flags);//FIXME wrong if chroma me is different
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
375 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
376
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
377 cxy= 2*tl + (cx + cy)/4 - (cx2 + cy2) - 2*c;
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
378
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
379 assert(16*cx2 + 4*cx + 32*c == 32*r);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
380 assert(16*cx2 - 4*cx + 32*c == 32*l);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
381 assert(16*cy2 + 4*cy + 32*c == 32*b);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
382 assert(16*cy2 - 4*cy + 32*c == 32*t);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
383 assert(16*cxy + 16*cy2 + 16*cx2 - 4*cy - 4*cx + 32*c == 32*tl);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
384
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
385 for(ny= -3; ny <= 3; ny++){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
386 for(nx= -3; nx <= 3; nx++){
2079
75aae96e6b54 fewer overflows
michael
parents: 2072
diff changeset
387 //FIXME this could overflow (unlikely though)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
388 int score= ny*nx*cxy + nx*nx*cx2 + ny*ny*cy2 + nx*cx + ny*cy + 32*c; //FIXME factor
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
389 int i;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
390
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
391 if((nx&3)==0 && (ny&3)==0) continue;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
392
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
393 score += 32*(mv_penalty[4*mx + nx - pred_x] + mv_penalty[4*my + ny - pred_y])*penalty_factor;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
394 // if(nx&1) score-=32*c->penalty_factor;
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
395 // if(ny&1) score-=32*c->penalty_factor;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
396
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
397 for(i=0; i<8; i++){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
398 if(score < best[i]){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
399 memmove(&best[i+1], &best[i], sizeof(int)*(7-i));
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
400 memmove(&best_pos[i+1][0], &best_pos[i][0], sizeof(int)*2*(7-i));
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
401 best[i]= score;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
402 best_pos[i][0]= nx + 4*mx;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
403 best_pos[i][1]= ny + 4*my;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
404 break;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
405 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
406 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
407 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
408 }
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
409 }
954
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
410 for(i=0; i<subpel_quality; i++){
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
411 nx= best_pos[i][0];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
412 ny= best_pos[i][1];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
413 CHECK_QUARTER_MV(nx&3, ny&3, nx>>2, ny>>2)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
414 }
954
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
415
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
416 #if 0
954
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
417 const int tl= score_map[(index-(1<<ME_MAP_SHIFT)-1)&(ME_MAP_SIZE-1)];
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
418 const int bl= score_map[(index+(1<<ME_MAP_SHIFT)-1)&(ME_MAP_SIZE-1)];
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
419 const int tr= score_map[(index-(1<<ME_MAP_SHIFT)+1)&(ME_MAP_SIZE-1)];
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
420 const int br= score_map[(index+(1<<ME_MAP_SHIFT)+1)&(ME_MAP_SIZE-1)];
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
421 // if(l < r && l < t && l < b && l < tl && l < bl && l < tr && l < br && bl < tl){
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
422 if(tl<br){
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
423
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
424 // nx= FFMAX(4*mx - bx, bx - 4*mx);
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
425 // ny= FFMAX(4*my - by, by - 4*my);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
426
954
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
427 static int stats[7][7], count;
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
428 count++;
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
429 stats[4*mx - bx + 3][4*my - by + 3]++;
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
430 if(256*256*256*64 % count ==0){
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
431 for(i=0; i<49; i++){
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
432 if((i%7)==0) printf("\n");
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
433 printf("%6d ", stats[0][i]);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
434 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
435 printf("\n");
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
436 }
954
13aec7e50c52 qpel in mmx2/3dnow
michaelni
parents: 952
diff changeset
437 }
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
438 #endif
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
439 #else
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
440
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
441 CHECK_QUARTER_MV(2, 2, mx-1, my-1)
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
442 CHECK_QUARTER_MV(0, 2, mx , my-1)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
443 CHECK_QUARTER_MV(2, 2, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
444 CHECK_QUARTER_MV(2, 0, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
445 CHECK_QUARTER_MV(2, 2, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
446 CHECK_QUARTER_MV(0, 2, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
447 CHECK_QUARTER_MV(2, 2, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
448 CHECK_QUARTER_MV(2, 0, mx-1, my )
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
449
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
450 nx= bx;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
451 ny= by;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
452
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
453 for(i=0; i<8; i++){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
454 int ox[8]= {0, 1, 1, 1, 0,-1,-1,-1};
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
455 int oy[8]= {1, 1, 0,-1,-1,-1, 0, 1};
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
456 CHECK_QUARTER_MV((nx + ox[i])&3, (ny + oy[i])&3, (nx + ox[i])>>2, (ny + oy[i])>>2)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
457 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
458 #endif
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
459 #if 0
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
460 //outer ring
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
461 CHECK_QUARTER_MV(1, 3, mx-1, my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
462 CHECK_QUARTER_MV(1, 2, mx-1, my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
463 CHECK_QUARTER_MV(1, 1, mx-1, my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
464 CHECK_QUARTER_MV(2, 1, mx-1, my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
465 CHECK_QUARTER_MV(3, 1, mx-1, my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
466 CHECK_QUARTER_MV(0, 1, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
467 CHECK_QUARTER_MV(1, 1, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
468 CHECK_QUARTER_MV(2, 1, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
469 CHECK_QUARTER_MV(3, 1, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
470 CHECK_QUARTER_MV(3, 2, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
471 CHECK_QUARTER_MV(3, 3, mx , my-1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
472 CHECK_QUARTER_MV(3, 0, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
473 CHECK_QUARTER_MV(3, 1, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
474 CHECK_QUARTER_MV(3, 2, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
475 CHECK_QUARTER_MV(3, 3, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
476 CHECK_QUARTER_MV(2, 3, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
477 CHECK_QUARTER_MV(1, 3, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
478 CHECK_QUARTER_MV(0, 3, mx , my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
479 CHECK_QUARTER_MV(3, 3, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
480 CHECK_QUARTER_MV(2, 3, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
481 CHECK_QUARTER_MV(1, 3, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
482 CHECK_QUARTER_MV(1, 2, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
483 CHECK_QUARTER_MV(1, 1, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
484 CHECK_QUARTER_MV(1, 0, mx-1, my )
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
485 #endif
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
486 assert(bx >= xmin*4 && bx <= xmax*4 && by >= ymin*4 && by <= ymax*4);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
487
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
488 *mx_ptr = bx;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
489 *my_ptr = by;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
490 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
491 *mx_ptr =4*mx;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
492 *my_ptr =4*my;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
493 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
494
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
495 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
496 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
497
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
498
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
499 #define CHECK_MV(x,y)\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
500 {\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
501 const int key= ((y)<<ME_MAP_MV_BITS) + (x) + map_generation;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
502 const int index= (((y)<<ME_MAP_SHIFT) + (x))&(ME_MAP_SIZE-1);\
2072
c200977cdf78 clip input motion vectors
michael
parents: 2015
diff changeset
503 assert((x) >= xmin);\
c200977cdf78 clip input motion vectors
michael
parents: 2015
diff changeset
504 assert((x) <= xmax);\
c200977cdf78 clip input motion vectors
michael
parents: 2015
diff changeset
505 assert((y) >= ymin);\
c200977cdf78 clip input motion vectors
michael
parents: 2015
diff changeset
506 assert((y) <= ymax);\
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
507 /*printf("check_mv %d %d\n", x, y);*/\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
508 if(map[index]!=key){\
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
509 d= cmp(s, x, y, 0, 0, size, h, ref_index, src_index, cmpf, chroma_cmpf, flags);\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
510 map[index]= key;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
511 score_map[index]= d;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
512 d += (mv_penalty[((x)<<shift)-pred_x] + mv_penalty[((y)<<shift)-pred_y])*penalty_factor;\
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
513 /*printf("score:%d\n", d);*/\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
514 COPY3_IF_LT(dmin, d, best[0], x, best[1], y)\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
515 }\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
516 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
517
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
518 #define CHECK_CLIPED_MV(ax,ay)\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
519 {\
4248
4be0e20c0eeb cosmetic (prevent name clashes of variables in a macro with surrounding code)
michael
parents: 4246
diff changeset
520 const int Lx= ax;\
4be0e20c0eeb cosmetic (prevent name clashes of variables in a macro with surrounding code)
michael
parents: 4246
diff changeset
521 const int Ly= ay;\
4be0e20c0eeb cosmetic (prevent name clashes of variables in a macro with surrounding code)
michael
parents: 4246
diff changeset
522 const int Lx2= FFMAX(xmin, FFMIN(Lx, xmax));\
4be0e20c0eeb cosmetic (prevent name clashes of variables in a macro with surrounding code)
michael
parents: 4246
diff changeset
523 const int Ly2= FFMAX(ymin, FFMIN(Ly, ymax));\
4be0e20c0eeb cosmetic (prevent name clashes of variables in a macro with surrounding code)
michael
parents: 4246
diff changeset
524 CHECK_MV(Lx2, Ly2)\
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
525 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
526
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
527 #define CHECK_MV_DIR(x,y,new_dir)\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
528 {\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
529 const int key= ((y)<<ME_MAP_MV_BITS) + (x) + map_generation;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
530 const int index= (((y)<<ME_MAP_SHIFT) + (x))&(ME_MAP_SIZE-1);\
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
531 /*printf("check_mv_dir %d %d %d\n", x, y, new_dir);*/\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
532 if(map[index]!=key){\
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
533 d= cmp(s, x, y, 0, 0, size, h, ref_index, src_index, cmpf, chroma_cmpf, flags);\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
534 map[index]= key;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
535 score_map[index]= d;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
536 d += (mv_penalty[((x)<<shift)-pred_x] + mv_penalty[((y)<<shift)-pred_y])*penalty_factor;\
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
537 /*printf("score:%d\n", d);*/\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
538 if(d<dmin){\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
539 best[0]=x;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
540 best[1]=y;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
541 dmin=d;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
542 next_dir= new_dir;\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
543 }\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
544 }\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
545 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
546
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
547 #define check(x,y,S,v)\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
548 if( (x)<(xmin<<(S)) ) printf("%d %d %d %d %d xmin" #v, xmin, (x), (y), s->mb_x, s->mb_y);\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
549 if( (x)>(xmax<<(S)) ) printf("%d %d %d %d %d xmax" #v, xmax, (x), (y), s->mb_x, s->mb_y);\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
550 if( (y)<(ymin<<(S)) ) printf("%d %d %d %d %d ymin" #v, ymin, (x), (y), s->mb_x, s->mb_y);\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
551 if( (y)>(ymax<<(S)) ) printf("%d %d %d %d %d ymax" #v, ymax, (x), (y), s->mb_x, s->mb_y);\
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
552
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
553 #define LOAD_COMMON2\
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
554 uint32_t *map= c->map;\
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
555 const int qpel= flags&FLAG_QPEL;\
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
556 const int shift= 1+qpel;\
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
557
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
558 static always_inline int small_diamond_search(MpegEncContext * s, int *best, int dmin,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
559 int src_index, int ref_index, int const penalty_factor,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
560 int size, int h, int flags)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
561 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
562 MotionEstContext * const c= &s->me;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
563 me_cmp_func cmpf, chroma_cmpf;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
564 int next_dir=-1;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
565 LOAD_COMMON
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
566 LOAD_COMMON2
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
567 int map_generation= c->map_generation;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
568
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
569 cmpf= s->dsp.me_cmp[size];
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
570 chroma_cmpf= s->dsp.me_cmp[size+1];
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
571
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
572 { /* ensure that the best point is in the MAP as h/qpel refinement needs it */
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
573 const int key= (best[1]<<ME_MAP_MV_BITS) + best[0] + map_generation;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
574 const int index= ((best[1]<<ME_MAP_SHIFT) + best[0])&(ME_MAP_SIZE-1);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
575 if(map[index]!=key){ //this will be executed only very rarey
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
576 score_map[index]= cmp(s, best[0], best[1], 0, 0, size, h, ref_index, src_index, cmpf, chroma_cmpf, flags);
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
577 map[index]= key;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
578 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
579 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
580
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
581 for(;;){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
582 int d;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
583 const int dir= next_dir;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
584 const int x= best[0];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
585 const int y= best[1];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
586 next_dir=-1;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
587
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
588 //printf("%d", dir);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
589 if(dir!=2 && x>xmin) CHECK_MV_DIR(x-1, y , 0)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
590 if(dir!=3 && y>ymin) CHECK_MV_DIR(x , y-1, 1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
591 if(dir!=0 && x<xmax) CHECK_MV_DIR(x+1, y , 2)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
592 if(dir!=1 && y<ymax) CHECK_MV_DIR(x , y+1, 3)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
593
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
594 if(next_dir==-1){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
595 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
596 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
597 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
598 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
599
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
600 static int funny_diamond_search(MpegEncContext * s, int *best, int dmin,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
601 int src_index, int ref_index, int const penalty_factor,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
602 int size, int h, int flags)
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
603 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
604 MotionEstContext * const c= &s->me;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
605 me_cmp_func cmpf, chroma_cmpf;
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
606 int dia_size;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
607 LOAD_COMMON
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
608 LOAD_COMMON2
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
609 int map_generation= c->map_generation;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
610
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
611 cmpf= s->dsp.me_cmp[size];
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
612 chroma_cmpf= s->dsp.me_cmp[size+1];
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
613
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
614 for(dia_size=1; dia_size<=4; dia_size++){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
615 int dir;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
616 const int x= best[0];
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
617 const int y= best[1];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
618
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
619 if(dia_size&(dia_size-1)) continue;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
620
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
621 if( x + dia_size > xmax
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
622 || x - dia_size < xmin
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
623 || y + dia_size > ymax
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
624 || y - dia_size < ymin)
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
625 continue;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
626
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
627 for(dir= 0; dir<dia_size; dir+=2){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
628 int d;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
629
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
630 CHECK_MV(x + dir , y + dia_size - dir);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
631 CHECK_MV(x + dia_size - dir, y - dir );
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
632 CHECK_MV(x - dir , y - dia_size + dir);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
633 CHECK_MV(x - dia_size + dir, y + dir );
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
634 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
635
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
636 if(x!=best[0] || y!=best[1])
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
637 dia_size=0;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
638 #if 0
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
639 {
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
640 int dx, dy, i;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
641 static int stats[8*8];
4001
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3959
diff changeset
642 dx= FFABS(x-best[0]);
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3959
diff changeset
643 dy= FFABS(y-best[1]);
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
644 if(dy>dx){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
645 dx^=dy; dy^=dx; dx^=dy;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
646 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
647 stats[dy*8 + dx] ++;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
648 if(256*256*256*64 % (stats[0]+1)==0){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
649 for(i=0; i<64; i++){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
650 if((i&7)==0) printf("\n");
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
651 printf("%8d ", stats[i]);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
652 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
653 printf("\n");
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
654 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
655 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
656 #endif
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
657 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
658 return dmin;
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
659 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
660
4249
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
661 static int umh_search(MpegEncContext * s, int *best, int dmin,
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
662 int src_index, int ref_index, int const penalty_factor,
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
663 int size, int h, int flags)
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
664 {
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
665 MotionEstContext * const c= &s->me;
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
666 me_cmp_func cmpf, chroma_cmpf;
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
667 LOAD_COMMON
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
668 LOAD_COMMON2
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
669 int map_generation= c->map_generation;
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
670 int x,y,x2,y2, i, j, d;
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
671 static const int hex[16][2]={{-4,-2}, {-4,-1}, {-4, 0}, {-4, 1}, {-4, 2},
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
672 { 4,-2}, { 4,-1}, { 4, 0}, { 4, 1}, { 4, 2},
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
673 {-2, 3}, { 0, 4}, { 2, 3},
4250
507ba4483ffc typo in umh
michael
parents: 4249
diff changeset
674 {-2,-3}, { 0,-4}, { 2,-3},};
4249
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
675 static const int hex2[6][2]={{-2, 0}, { 2,0}, {-1,-2}, {1,-2}, {-1,2},{1,2}};
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
676
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
677 cmpf= s->dsp.me_cmp[size];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
678 chroma_cmpf= s->dsp.me_cmp[size+1];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
679
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
680 x= best[0];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
681 y= best[1];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
682 for(x2=FFMAX(x-15, xmin); x2<=FFMIN(x+15,xmax); x2+=2){
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
683 CHECK_MV(x2, y);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
684 }
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
685 for(y2=FFMAX(y- 7, ymin); y2<=FFMIN(y+ 7,ymax); y2+=2){
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
686 CHECK_MV(x, y2);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
687 }
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
688
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
689 x= best[0];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
690 y= best[1];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
691 for(y2=FFMAX(y-2, ymin); y2<=FFMIN(y+2,ymax); y2++){
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
692 for(x2=FFMAX(x-2, xmin); x2<=FFMIN(x+2,xmax); x2++){
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
693 CHECK_MV(x2, y2);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
694 }
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
695 }
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
696
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
697 //FIXME prevent the CLIP stuff
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
698
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
699 for(j=1; j<=4; j++){
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
700 for(i=0; i<16; i++){
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
701 CHECK_CLIPED_MV(x+hex[i][0]*j, y+hex[i][1]*j);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
702 }
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
703 }
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
704
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
705 do{
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
706 x= best[0];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
707 y= best[1];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
708 for(i=0; i<6; i++){
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
709 CHECK_CLIPED_MV(x+hex2[i][0], y+hex2[i][1]);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
710 }
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
711 }while(best[0] != x || best[1] != y);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
712
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
713 do{
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
714 x= best[0];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
715 y= best[1];
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
716 CHECK_CLIPED_MV(x+1, y);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
717 CHECK_CLIPED_MV(x, y+1);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
718 CHECK_CLIPED_MV(x-1, y);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
719 CHECK_CLIPED_MV(x, y-1);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
720 }while(best[0] != x || best[1] != y);
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
721
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
722 return dmin;
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
723 }
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
724
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
725 #define SAB_CHECK_MV(ax,ay)\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
726 {\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
727 const int key= ((ay)<<ME_MAP_MV_BITS) + (ax) + map_generation;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
728 const int index= (((ay)<<ME_MAP_SHIFT) + (ax))&(ME_MAP_SIZE-1);\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
729 /*printf("sab check %d %d\n", ax, ay);*/\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
730 if(map[index]!=key){\
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
731 d= cmp(s, ax, ay, 0, 0, size, h, ref_index, src_index, cmpf, chroma_cmpf, flags);\
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
732 map[index]= key;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
733 score_map[index]= d;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
734 d += (mv_penalty[((ax)<<shift)-pred_x] + mv_penalty[((ay)<<shift)-pred_y])*penalty_factor;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
735 /*printf("score: %d\n", d);*/\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
736 if(d < minima[minima_count-1].height){\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
737 int j=0;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
738 \
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
739 while(d >= minima[j].height) j++;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
740 \
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
741 memmove(&minima [j+1], &minima [j], (minima_count - j - 1)*sizeof(Minima));\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
742 \
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
743 minima[j].checked= 0;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
744 minima[j].height= d;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
745 minima[j].x= ax;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
746 minima[j].y= ay;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
747 \
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
748 i=-1;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
749 continue;\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
750 }\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
751 }\
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
752 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
753
2113
07a663d46be2 shape-adaptive diamond + prediction crash patch by (Loren Merritt <lorenm at u dot washington dot edu>)
michael
parents: 2079
diff changeset
754 #define MAX_SAB_SIZE ME_MAP_SIZE
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
755 static int sab_diamond_search(MpegEncContext * s, int *best, int dmin,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
756 int src_index, int ref_index, int const penalty_factor,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
757 int size, int h, int flags)
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
758 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
759 MotionEstContext * const c= &s->me;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
760 me_cmp_func cmpf, chroma_cmpf;
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
761 Minima minima[MAX_SAB_SIZE];
4001
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3959
diff changeset
762 const int minima_count= FFABS(c->dia_size);
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
763 int i, j;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
764 LOAD_COMMON
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
765 LOAD_COMMON2
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
766 int map_generation= c->map_generation;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
767
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
768 cmpf= s->dsp.me_cmp[size];
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
769 chroma_cmpf= s->dsp.me_cmp[size+1];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
770
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
771 for(j=i=0; i<ME_MAP_SIZE; i++){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
772 uint32_t key= map[i];
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
773
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
774 key += (1<<(ME_MAP_MV_BITS-1)) + (1<<(2*ME_MAP_MV_BITS-1));
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
775
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
776 if((key&((-1)<<(2*ME_MAP_MV_BITS))) != map_generation) continue;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
777
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
778 assert(j<MAX_SAB_SIZE); //max j = number of predictors
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
779
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
780 minima[j].height= score_map[i];
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
781 minima[j].x= key & ((1<<ME_MAP_MV_BITS)-1); key>>=ME_MAP_MV_BITS;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
782 minima[j].y= key & ((1<<ME_MAP_MV_BITS)-1);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
783 minima[j].x-= (1<<(ME_MAP_MV_BITS-1));
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
784 minima[j].y-= (1<<(ME_MAP_MV_BITS-1));
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
785 minima[j].checked=0;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
786 if(minima[j].x || minima[j].y)
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
787 minima[j].height+= (mv_penalty[((minima[j].x)<<shift)-pred_x] + mv_penalty[((minima[j].y)<<shift)-pred_y])*penalty_factor;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
788
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
789 j++;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
790 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
791
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
792 qsort(minima, j, sizeof(Minima), minima_cmp);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
793
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
794 for(; j<minima_count; j++){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
795 minima[j].height=256*256*256*64;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
796 minima[j].checked=0;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
797 minima[j].x= minima[j].y=0;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
798 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
799
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
800 for(i=0; i<minima_count; i++){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
801 const int x= minima[i].x;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
802 const int y= minima[i].y;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
803 int d;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
804
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
805 if(minima[i].checked) continue;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
806
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
807 if( x >= xmax || x <= xmin
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
808 || y >= ymax || y <= ymin)
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
809 continue;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
810
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
811 SAB_CHECK_MV(x-1, y)
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
812 SAB_CHECK_MV(x+1, y)
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
813 SAB_CHECK_MV(x , y-1)
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
814 SAB_CHECK_MV(x , y+1)
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
815
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
816 minima[i].checked= 1;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
817 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
818
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
819 best[0]= minima[0].x;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
820 best[1]= minima[0].y;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
821 dmin= minima[0].height;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
822
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
823 if( best[0] < xmax && best[0] > xmin
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
824 && best[1] < ymax && best[1] > ymin){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
825 int d;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
826 //ensure that the refernece samples for hpel refinement are in the map
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
827 CHECK_MV(best[0]-1, best[1])
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
828 CHECK_MV(best[0]+1, best[1])
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
829 CHECK_MV(best[0], best[1]-1)
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
830 CHECK_MV(best[0], best[1]+1)
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
831 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
832 return dmin;
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
833 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
834
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
835 static int var_diamond_search(MpegEncContext * s, int *best, int dmin,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
836 int src_index, int ref_index, int const penalty_factor,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
837 int size, int h, int flags)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
838 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
839 MotionEstContext * const c= &s->me;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
840 me_cmp_func cmpf, chroma_cmpf;
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
841 int dia_size;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
842 LOAD_COMMON
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
843 LOAD_COMMON2
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
844 int map_generation= c->map_generation;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
845
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
846 cmpf= s->dsp.me_cmp[size];
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
847 chroma_cmpf= s->dsp.me_cmp[size+1];
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
848
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
849 for(dia_size=1; dia_size<=c->dia_size; dia_size++){
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
850 int dir, start, end;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
851 const int x= best[0];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
852 const int y= best[1];
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
853
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
854 start= FFMAX(0, y + dia_size - ymax);
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
855 end = FFMIN(dia_size, xmax - x + 1);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
856 for(dir= start; dir<end; dir++){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
857 int d;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
858
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
859 //check(x + dir,y + dia_size - dir,0, a0)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
860 CHECK_MV(x + dir , y + dia_size - dir);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
861 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
862
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
863 start= FFMAX(0, x + dia_size - xmax);
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
864 end = FFMIN(dia_size, y - ymin + 1);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
865 for(dir= start; dir<end; dir++){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
866 int d;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
867
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
868 //check(x + dia_size - dir, y - dir,0, a1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
869 CHECK_MV(x + dia_size - dir, y - dir );
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
870 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
871
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
872 start= FFMAX(0, -y + dia_size + ymin );
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
873 end = FFMIN(dia_size, x - xmin + 1);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
874 for(dir= start; dir<end; dir++){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
875 int d;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
876
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
877 //check(x - dir,y - dia_size + dir,0, a2)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
878 CHECK_MV(x - dir , y - dia_size + dir);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
879 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
880
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
881 start= FFMAX(0, -x + dia_size + xmin );
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
882 end = FFMIN(dia_size, ymax - y + 1);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
883 for(dir= start; dir<end; dir++){
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
884 int d;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
885
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
886 //check(x - dia_size + dir, y + dir,0, a3)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
887 CHECK_MV(x - dia_size + dir, y + dir );
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
888 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
889
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
890 if(x!=best[0] || y!=best[1])
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
891 dia_size=0;
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
892 #if 0
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
893 {
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
894 int dx, dy, i;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
895 static int stats[8*8];
4001
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3959
diff changeset
896 dx= FFABS(x-best[0]);
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3959
diff changeset
897 dy= FFABS(y-best[1]);
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
898 stats[dy*8 + dx] ++;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
899 if(256*256*256*64 % (stats[0]+1)==0){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
900 for(i=0; i<64; i++){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
901 if((i&7)==0) printf("\n");
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
902 printf("%6d ", stats[i]);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
903 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
904 printf("\n");
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
905 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
906 }
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
907 #endif
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
908 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
909 return dmin;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
910 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
911
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
912 static always_inline int diamond_search(MpegEncContext * s, int *best, int dmin,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
913 int src_index, int ref_index, int const penalty_factor,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
914 int size, int h, int flags){
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
915 MotionEstContext * const c= &s->me;
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
916 if(c->dia_size==-1)
4249
17e04bdf8d69 UMH support (dia_size=-1 activates it)
michael
parents: 4248
diff changeset
917 return umh_search(s, best, dmin, src_index, ref_index, penalty_factor, size, h, flags);
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
918 else if(c->dia_size<-1)
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
919 return sab_diamond_search(s, best, dmin, src_index, ref_index, penalty_factor, size, h, flags);
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
920 else if(c->dia_size<2)
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
921 return small_diamond_search(s, best, dmin, src_index, ref_index, penalty_factor, size, h, flags);
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
922 else
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
923 return var_diamond_search(s, best, dmin, src_index, ref_index, penalty_factor, size, h, flags);
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
924 }
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
925
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
926 static always_inline int epzs_motion_search_internal(MpegEncContext * s, int *mx_ptr, int *my_ptr,
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
927 int P[10][2], int src_index, int ref_index, int16_t (*last_mv)[2],
2184
3378d0677903 4x4 SSE compare function
michael
parents: 2113
diff changeset
928 int ref_mv_scale, int flags, int size, int h)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
929 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
930 MotionEstContext * const c= &s->me;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
931 int best[2]={0, 0};
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
932 int d, dmin;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
933 int map_generation;
2226
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
934 int penalty_factor;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
935 const int ref_mv_stride= s->mb_stride; //pass as arg FIXME
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
936 const int ref_mv_xy= s->mb_x + s->mb_y*ref_mv_stride; //add to last_mv beforepassing FIXME
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
937 me_cmp_func cmpf, chroma_cmpf;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
938
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
939 LOAD_COMMON
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
940 LOAD_COMMON2
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
941
2226
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
942 if(c->pre_pass){
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
943 penalty_factor= c->pre_penalty_factor;
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
944 cmpf= s->dsp.me_pre_cmp[size];
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
945 chroma_cmpf= s->dsp.me_pre_cmp[size+1];
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
946 }else{
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
947 penalty_factor= c->penalty_factor;
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
948 cmpf= s->dsp.me_cmp[size];
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
949 chroma_cmpf= s->dsp.me_cmp[size+1];
4306aa6f8f16 pre_cmp fix
michael
parents: 2189
diff changeset
950 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
951
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
952 map_generation= update_map_generation(c);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
953
2184
3378d0677903 4x4 SSE compare function
michael
parents: 2113
diff changeset
954 assert(cmpf);
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
955 dmin= cmp(s, 0, 0, 0, 0, size, h, ref_index, src_index, cmpf, chroma_cmpf, flags);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
956 map[0]= map_generation;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
957 score_map[0]= dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
958
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
959 /* first line */
1799
95612d423fde multithreaded/SMP motion estimation
michael
parents: 1739
diff changeset
960 if (s->first_slice_line) {
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
961 CHECK_MV(P_LEFT[0]>>shift, P_LEFT[1]>>shift)
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
962 CHECK_CLIPED_MV((last_mv[ref_mv_xy][0]*ref_mv_scale + (1<<15))>>16,
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
963 (last_mv[ref_mv_xy][1]*ref_mv_scale + (1<<15))>>16)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
964 }else{
3293
f1bcb9ae510b make zero motion vector threshold user setable
michael
parents: 3036
diff changeset
965 if(dmin<((h*h*s->avctx->mv0_threshold)>>8)
f1bcb9ae510b make zero motion vector threshold user setable
michael
parents: 3036
diff changeset
966 && ( P_LEFT[0] |P_LEFT[1]
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
967 |P_TOP[0] |P_TOP[1]
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
968 |P_TOPRIGHT[0]|P_TOPRIGHT[1])==0){
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
969 *mx_ptr= 0;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
970 *my_ptr= 0;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
971 c->skip=1;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
972 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
973 }
4246
a579dae361e2 1 step diamond search around the median MV predictor
michael
parents: 4001
diff changeset
974 CHECK_MV( P_MEDIAN[0] >>shift , P_MEDIAN[1] >>shift)
a579dae361e2 1 step diamond search around the median MV predictor
michael
parents: 4001
diff changeset
975 CHECK_CLIPED_MV((P_MEDIAN[0]>>shift) , (P_MEDIAN[1]>>shift)-1)
a579dae361e2 1 step diamond search around the median MV predictor
michael
parents: 4001
diff changeset
976 CHECK_CLIPED_MV((P_MEDIAN[0]>>shift) , (P_MEDIAN[1]>>shift)+1)
a579dae361e2 1 step diamond search around the median MV predictor
michael
parents: 4001
diff changeset
977 CHECK_CLIPED_MV((P_MEDIAN[0]>>shift)-1, (P_MEDIAN[1]>>shift) )
a579dae361e2 1 step diamond search around the median MV predictor
michael
parents: 4001
diff changeset
978 CHECK_CLIPED_MV((P_MEDIAN[0]>>shift)+1, (P_MEDIAN[1]>>shift) )
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
979 CHECK_CLIPED_MV((last_mv[ref_mv_xy][0]*ref_mv_scale + (1<<15))>>16,
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
980 (last_mv[ref_mv_xy][1]*ref_mv_scale + (1<<15))>>16)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
981 CHECK_MV(P_LEFT[0] >>shift, P_LEFT[1] >>shift)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
982 CHECK_MV(P_TOP[0] >>shift, P_TOP[1] >>shift)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
983 CHECK_MV(P_TOPRIGHT[0]>>shift, P_TOPRIGHT[1]>>shift)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
984 }
2184
3378d0677903 4x4 SSE compare function
michael
parents: 2113
diff changeset
985 if(dmin>h*h*4){
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
986 if(c->pre_pass){
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
987 CHECK_CLIPED_MV((last_mv[ref_mv_xy-1][0]*ref_mv_scale + (1<<15))>>16,
952
f348d302a51e pre motion estimation cleanup/bugfix
michaelni
parents: 948
diff changeset
988 (last_mv[ref_mv_xy-1][1]*ref_mv_scale + (1<<15))>>16)
1799
95612d423fde multithreaded/SMP motion estimation
michael
parents: 1739
diff changeset
989 if(!s->first_slice_line)
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
990 CHECK_CLIPED_MV((last_mv[ref_mv_xy-ref_mv_stride][0]*ref_mv_scale + (1<<15))>>16,
1799
95612d423fde multithreaded/SMP motion estimation
michael
parents: 1739
diff changeset
991 (last_mv[ref_mv_xy-ref_mv_stride][1]*ref_mv_scale + (1<<15))>>16)
952
f348d302a51e pre motion estimation cleanup/bugfix
michaelni
parents: 948
diff changeset
992 }else{
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
993 CHECK_CLIPED_MV((last_mv[ref_mv_xy+1][0]*ref_mv_scale + (1<<15))>>16,
952
f348d302a51e pre motion estimation cleanup/bugfix
michaelni
parents: 948
diff changeset
994 (last_mv[ref_mv_xy+1][1]*ref_mv_scale + (1<<15))>>16)
1800
e039d79185c2 simplify MV availability check / dont use below last row MV which is always zero
michael
parents: 1799
diff changeset
995 if(s->mb_y+1<s->end_mb_y) //FIXME replace at least with last_slice_line
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
996 CHECK_CLIPED_MV((last_mv[ref_mv_xy+ref_mv_stride][0]*ref_mv_scale + (1<<15))>>16,
1799
95612d423fde multithreaded/SMP motion estimation
michael
parents: 1739
diff changeset
997 (last_mv[ref_mv_xy+ref_mv_stride][1]*ref_mv_scale + (1<<15))>>16)
952
f348d302a51e pre motion estimation cleanup/bugfix
michaelni
parents: 948
diff changeset
998 }
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
999 }
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1000
2015
3ab8f3e2ae6a moving motion estimation specific variables from MpegEncContext -> MotionEstContext
michael
parents: 2014
diff changeset
1001 if(c->avctx->last_predictor_count){
3ab8f3e2ae6a moving motion estimation specific variables from MpegEncContext -> MotionEstContext
michael
parents: 2014
diff changeset
1002 const int count= c->avctx->last_predictor_count;
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1003 const int xstart= FFMAX(0, s->mb_x - count);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1004 const int ystart= FFMAX(0, s->mb_y - count);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1005 const int xend= FFMIN(s->mb_width , s->mb_x + count + 1);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1006 const int yend= FFMIN(s->mb_height, s->mb_y + count + 1);
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1007 int mb_y;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1008
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1009 for(mb_y=ystart; mb_y<yend; mb_y++){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1010 int mb_x;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1011 for(mb_x=xstart; mb_x<xend; mb_x++){
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1012 const int xy= mb_x + 1 + (mb_y + 1)*ref_mv_stride;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1013 int mx= (last_mv[xy][0]*ref_mv_scale + (1<<15))>>16;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1014 int my= (last_mv[xy][1]*ref_mv_scale + (1<<15))>>16;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1015
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1016 if(mx>xmax || mx<xmin || my>ymax || my<ymin) continue;
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1017 CHECK_MV(mx,my)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1018 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1019 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1020 }
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1021
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1022 //check(best[0],best[1],0, b0)
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1023 dmin= diamond_search(s, best, dmin, src_index, ref_index, penalty_factor, size, h, flags);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1024
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1025 //check(best[0],best[1],0, b1)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1026 *mx_ptr= best[0];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1027 *my_ptr= best[1];
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1028
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1029 // printf("%d %d %d \n", best[0], best[1], dmin);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1030 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1031 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1032
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1033 //this function is dedicated to the braindamaged gcc
2184
3378d0677903 4x4 SSE compare function
michael
parents: 2113
diff changeset
1034 inline int ff_epzs_motion_search(MpegEncContext * s, int *mx_ptr, int *my_ptr,
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1035 int P[10][2], int src_index, int ref_index, int16_t (*last_mv)[2],
2184
3378d0677903 4x4 SSE compare function
michael
parents: 2113
diff changeset
1036 int ref_mv_scale, int size, int h)
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1037 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
1038 MotionEstContext * const c= &s->me;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1039 //FIXME convert other functions in the same way if faster
2184
3378d0677903 4x4 SSE compare function
michael
parents: 2113
diff changeset
1040 if(c->flags==0 && h==16 && size==0){
3378d0677903 4x4 SSE compare function
michael
parents: 2113
diff changeset
1041 return epzs_motion_search_internal(s, mx_ptr, my_ptr, P, src_index, ref_index, last_mv, ref_mv_scale, 0, 0, 16);
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1042 // case FLAG_QPEL:
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1043 // return epzs_motion_search_internal(s, mx_ptr, my_ptr, P, src_index, ref_index, last_mv, ref_mv_scale, FLAG_QPEL);
2184
3378d0677903 4x4 SSE compare function
michael
parents: 2113
diff changeset
1044 }else{
3378d0677903 4x4 SSE compare function
michael
parents: 2113
diff changeset
1045 return epzs_motion_search_internal(s, mx_ptr, my_ptr, P, src_index, ref_index, last_mv, ref_mv_scale, c->flags, size, h);
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1046 }
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1047 }
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1048
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1049 static int epzs_motion_search4(MpegEncContext * s,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1050 int *mx_ptr, int *my_ptr, int P[10][2],
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1051 int src_index, int ref_index, int16_t (*last_mv)[2],
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1052 int ref_mv_scale)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1053 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
1054 MotionEstContext * const c= &s->me;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1055 int best[2]={0, 0};
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1056 int d, dmin;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1057 int map_generation;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
1058 const int penalty_factor= c->penalty_factor;
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1059 const int size=1;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1060 const int h=8;
1177
fea03d2c4946 simplified adressing of most mb based arrays (mb_x + mb_y*s->mb_stride) now instead of mb_x + mb_y*mb_width and 1+mb_x + (1+mb_y)*(mb_width+2) and ... mixture
michaelni
parents: 1162
diff changeset
1061 const int ref_mv_stride= s->mb_stride;
fea03d2c4946 simplified adressing of most mb based arrays (mb_x + mb_y*s->mb_stride) now instead of mb_x + mb_y*mb_width and 1+mb_x + (1+mb_y)*(mb_width+2) and ... mixture
michaelni
parents: 1162
diff changeset
1062 const int ref_mv_xy= s->mb_x + s->mb_y *ref_mv_stride;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1063 me_cmp_func cmpf, chroma_cmpf;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1064 LOAD_COMMON
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
1065 int flags= c->flags;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1066 LOAD_COMMON2
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1067
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1068 cmpf= s->dsp.me_cmp[size];
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1069 chroma_cmpf= s->dsp.me_cmp[size+1];
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1070
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
1071 map_generation= update_map_generation(c);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1072
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1073 dmin = 1000000;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1074 //printf("%d %d %d %d //",xmin, ymin, xmax, ymax);
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1075 /* first line */
1799
95612d423fde multithreaded/SMP motion estimation
michael
parents: 1739
diff changeset
1076 if (s->first_slice_line) {
2979
bfabfdf9ce55 COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 2967
diff changeset
1077 CHECK_MV(P_LEFT[0]>>shift, P_LEFT[1]>>shift)
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1078 CHECK_CLIPED_MV((last_mv[ref_mv_xy][0]*ref_mv_scale + (1<<15))>>16,
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1079 (last_mv[ref_mv_xy][1]*ref_mv_scale + (1<<15))>>16)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1080 CHECK_MV(P_MV1[0]>>shift, P_MV1[1]>>shift)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1081 }else{
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1082 CHECK_MV(P_MV1[0]>>shift, P_MV1[1]>>shift)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1083 //FIXME try some early stop
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1084 CHECK_MV(P_MEDIAN[0]>>shift, P_MEDIAN[1]>>shift)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1085 CHECK_MV(P_LEFT[0]>>shift, P_LEFT[1]>>shift)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1086 CHECK_MV(P_TOP[0]>>shift, P_TOP[1]>>shift)
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1087 CHECK_MV(P_TOPRIGHT[0]>>shift, P_TOPRIGHT[1]>>shift)
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1088 CHECK_CLIPED_MV((last_mv[ref_mv_xy][0]*ref_mv_scale + (1<<15))>>16,
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1089 (last_mv[ref_mv_xy][1]*ref_mv_scale + (1<<15))>>16)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1090 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1091 if(dmin>64*4){
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1092 CHECK_CLIPED_MV((last_mv[ref_mv_xy+1][0]*ref_mv_scale + (1<<15))>>16,
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1093 (last_mv[ref_mv_xy+1][1]*ref_mv_scale + (1<<15))>>16)
1800
e039d79185c2 simplify MV availability check / dont use below last row MV which is always zero
michael
parents: 1799
diff changeset
1094 if(s->mb_y+1<s->end_mb_y) //FIXME replace at least with last_slice_line
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1095 CHECK_CLIPED_MV((last_mv[ref_mv_xy+ref_mv_stride][0]*ref_mv_scale + (1<<15))>>16,
1799
95612d423fde multithreaded/SMP motion estimation
michael
parents: 1739
diff changeset
1096 (last_mv[ref_mv_xy+ref_mv_stride][1]*ref_mv_scale + (1<<15))>>16)
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1097 }
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1098
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1099 dmin= diamond_search(s, best, dmin, src_index, ref_index, penalty_factor, size, h, flags);
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1100
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1101 *mx_ptr= best[0];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1102 *my_ptr= best[1];
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1103
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1104 // printf("%d %d %d \n", best[0], best[1], dmin);
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1105 return dmin;
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1106 }
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1107
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1108 //try to merge with above FIXME (needs PSNR test)
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1109 static int epzs_motion_search2(MpegEncContext * s,
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1110 int *mx_ptr, int *my_ptr, int P[10][2],
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1111 int src_index, int ref_index, int16_t (*last_mv)[2],
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1112 int ref_mv_scale)
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1113 {
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
1114 MotionEstContext * const c= &s->me;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1115 int best[2]={0, 0};
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1116 int d, dmin;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1117 int map_generation;
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
1118 const int penalty_factor= c->penalty_factor;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1119 const int size=0; //FIXME pass as arg
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1120 const int h=8;
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1121 const int ref_mv_stride= s->mb_stride;
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1122 const int ref_mv_xy= s->mb_x + s->mb_y *ref_mv_stride;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1123 me_cmp_func cmpf, chroma_cmpf;
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1124 LOAD_COMMON
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
1125 int flags= c->flags;
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1126 LOAD_COMMON2
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1127
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1128 cmpf= s->dsp.me_cmp[size];
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1129 chroma_cmpf= s->dsp.me_cmp[size+1];
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1130
2014
15c885db82a8 reduce dependancy between motion estimation and MpegEncContext
michael
parents: 1950
diff changeset
1131 map_generation= update_map_generation(c);
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1132
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1133 dmin = 1000000;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1134 //printf("%d %d %d %d //",xmin, ymin, xmax, ymax);
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1135 /* first line */
1799
95612d423fde multithreaded/SMP motion estimation
michael
parents: 1739
diff changeset
1136 if (s->first_slice_line) {
2979
bfabfdf9ce55 COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 2967
diff changeset
1137 CHECK_MV(P_LEFT[0]>>shift, P_LEFT[1]>>shift)
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1138 CHECK_CLIPED_MV((last_mv[ref_mv_xy][0]*ref_mv_scale + (1<<15))>>16,
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1139 (last_mv[ref_mv_xy][1]*ref_mv_scale + (1<<15))>>16)
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1140 CHECK_MV(P_MV1[0]>>shift, P_MV1[1]>>shift)
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1141 }else{
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1142 CHECK_MV(P_MV1[0]>>shift, P_MV1[1]>>shift)
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1143 //FIXME try some early stop
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1144 CHECK_MV(P_MEDIAN[0]>>shift, P_MEDIAN[1]>>shift)
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1145 CHECK_MV(P_LEFT[0]>>shift, P_LEFT[1]>>shift)
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1146 CHECK_MV(P_TOP[0]>>shift, P_TOP[1]>>shift)
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1147 CHECK_MV(P_TOPRIGHT[0]>>shift, P_TOPRIGHT[1]>>shift)
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1148 CHECK_CLIPED_MV((last_mv[ref_mv_xy][0]*ref_mv_scale + (1<<15))>>16,
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1149 (last_mv[ref_mv_xy][1]*ref_mv_scale + (1<<15))>>16)
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1150 }
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1151 if(dmin>64*4){
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1152 CHECK_CLIPED_MV((last_mv[ref_mv_xy+1][0]*ref_mv_scale + (1<<15))>>16,
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1153 (last_mv[ref_mv_xy+1][1]*ref_mv_scale + (1<<15))>>16)
1800
e039d79185c2 simplify MV availability check / dont use below last row MV which is always zero
michael
parents: 1799
diff changeset
1154 if(s->mb_y+1<s->end_mb_y) //FIXME replace at least with last_slice_line
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1155 CHECK_CLIPED_MV((last_mv[ref_mv_xy+ref_mv_stride][0]*ref_mv_scale + (1<<15))>>16,
1799
95612d423fde multithreaded/SMP motion estimation
michael
parents: 1739
diff changeset
1156 (last_mv[ref_mv_xy+ref_mv_stride][1]*ref_mv_scale + (1<<15))>>16)
1708
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1157 }
dea5b2946999 interlaced motion estimation
michael
parents: 1419
diff changeset
1158
1950
a3c60fa850dc motion estimation cleanup
michael
parents: 1800
diff changeset
1159 dmin= diamond_search(s, best, dmin, src_index, ref_index, penalty_factor, size, h, flags);
948
371bc36a9c5c shape adaptive diamonds for EPZS
michaelni
parents: 936
diff changeset
1160
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1161 *mx_ptr= best[0];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2522
diff changeset
1162 *my_ptr= best[1];
936
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1163
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1164 // printf("%d %d %d \n", best[0], best[1], dmin);
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1165 return dmin;
caa77cd960c0 qpel encoding
michaelni
parents:
diff changeset
1166 }