2004-07-26 01:36:44 +02:00
/*
* Copyright ( C ) 2004 Michael Niedermayer < michaelni @ gmx . at >
*
* This library is free software ; you can redistribute it and / or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation ; either
* version 2 of the License , or ( at your option ) any later version .
*
* This library is distributed in the hope that it will be useful ,
* but WITHOUT ANY WARRANTY ; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the GNU
* Lesser General Public License for more details .
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library ; if not , write to the Free Software
2006-01-12 23:43:26 +01:00
* Foundation , Inc . , 51 Franklin Street , Fifth Floor , Boston , MA 02110 - 1301 USA
2004-07-26 01:36:44 +02:00
*/
# include "avcodec.h"
# include "common.h"
# include "dsputil.h"
2004-10-30 04:16:27 +02:00
# include "rangecoder.h"
# define MID_STATE 128
2004-07-26 01:36:44 +02:00
# include "mpegvideo.h"
# undef NDEBUG
# include <assert.h>
# define MAX_DECOMPOSITIONS 8
# define MAX_PLANES 4
# define DWTELEM int
2005-04-10 00:15:48 +02:00
# define QSHIFT 5
2005-04-09 21:48:59 +02:00
# define QROOT (1<<QSHIFT)
2004-08-02 21:09:28 +02:00
# define LOSSLESS_QLOG -128
2004-09-22 01:10:10 +02:00
# define FRAC_BITS 8
2004-07-26 01:36:44 +02:00
static const int8_t quant3 [ 256 ] = {
0 , 0 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , 0 ,
} ;
static const int8_t quant3b [ 256 ] = {
0 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
- 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 , - 1 ,
} ;
2005-04-05 19:59:05 +02:00
static const int8_t quant3bA [ 256 ] = {
0 , 0 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 , 1 , - 1 ,
} ;
2004-07-26 01:36:44 +02:00
static const int8_t quant5 [ 256 ] = {
0 , 1 , 1 , 1 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 1 , - 1 , - 1 ,
} ;
static const int8_t quant7 [ 256 ] = {
0 , 1 , 1 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 ,
2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 ,
3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 ,
3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 ,
3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 ,
3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 ,
3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 ,
- 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 ,
- 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 ,
- 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 ,
- 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 ,
- 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 ,
- 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 ,
- 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 2 , - 1 , - 1 ,
} ;
static const int8_t quant9 [ 256 ] = {
0 , 1 , 1 , 2 , 2 , 2 , 2 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 3 ,
3 , 3 , 3 , 3 , 3 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 ,
4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 ,
4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 ,
4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 ,
4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 ,
4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 ,
4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 3 , - 3 , - 3 , - 3 ,
- 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 2 , - 2 , - 2 , - 2 , - 1 , - 1 ,
} ;
static const int8_t quant11 [ 256 ] = {
0 , 1 , 2 , 2 , 2 , 3 , 3 , 3 , 3 , 3 , 3 , 3 , 4 , 4 , 4 , 4 ,
4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 ,
4 , 4 , 4 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 ,
5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 ,
5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 ,
5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 ,
5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 ,
5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 ,
- 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 ,
- 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 ,
- 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 ,
- 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 ,
- 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 ,
- 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 4 , - 4 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 3 , - 2 , - 2 , - 2 , - 1 ,
} ;
static const int8_t quant13 [ 256 ] = {
0 , 1 , 2 , 2 , 3 , 3 , 3 , 3 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 ,
4 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 ,
5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 , 5 ,
5 , 5 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 ,
6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 ,
6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 ,
6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 ,
6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 ,
- 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 ,
- 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 ,
- 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 ,
- 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 ,
- 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 6 , - 5 ,
- 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 ,
- 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 , - 5 ,
- 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 4 , - 3 , - 3 , - 3 , - 3 , - 2 , - 2 , - 1 ,
} ;
2004-09-23 14:20:41 +02:00
# define LOG2_OBMC_MAX 6
# define OBMC_MAX (1<<(LOG2_OBMC_MAX))
2004-07-26 01:36:44 +02:00
#if 0 //64*cubic
static const uint8_t obmc32 [ 1024 ] = {
0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 ,
0 , 0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 2 , 2 , 2 , 2 , 2 , 2 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 , 0 ,
0 , 0 , 0 , 1 , 1 , 1 , 1 , 2 , 2 , 3 , 3 , 3 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 3 , 3 , 3 , 2 , 2 , 1 , 1 , 1 , 1 , 0 , 0 , 0 ,
0 , 0 , 1 , 1 , 2 , 2 , 3 , 4 , 4 , 5 , 6 , 6 , 7 , 7 , 8 , 8 , 8 , 8 , 7 , 7 , 6 , 6 , 5 , 4 , 4 , 3 , 2 , 2 , 1 , 1 , 0 , 0 ,
0 , 0 , 1 , 2 , 2 , 3 , 4 , 6 , 7 , 8 , 9 , 10 , 11 , 12 , 12 , 12 , 12 , 12 , 12 , 11 , 10 , 9 , 8 , 7 , 6 , 4 , 3 , 2 , 2 , 1 , 0 , 0 ,
0 , 1 , 1 , 2 , 3 , 5 , 6 , 8 , 10 , 11 , 13 , 14 , 15 , 16 , 17 , 18 , 18 , 17 , 16 , 15 , 14 , 13 , 11 , 10 , 8 , 6 , 5 , 3 , 2 , 1 , 1 , 0 ,
0 , 1 , 1 , 3 , 4 , 6 , 8 , 10 , 13 , 15 , 17 , 19 , 20 , 22 , 22 , 23 , 23 , 22 , 22 , 20 , 19 , 17 , 15 , 13 , 10 , 8 , 6 , 4 , 3 , 1 , 1 , 0 ,
0 , 1 , 2 , 4 , 6 , 8 , 10 , 13 , 16 , 19 , 21 , 23 , 25 , 27 , 28 , 29 , 29 , 28 , 27 , 25 , 23 , 21 , 19 , 16 , 13 , 10 , 8 , 6 , 4 , 2 , 1 , 0 ,
0 , 1 , 2 , 4 , 7 , 10 , 13 , 16 , 19 , 22 , 25 , 28 , 31 , 33 , 34 , 35 , 35 , 34 , 33 , 31 , 28 , 25 , 22 , 19 , 16 , 13 , 10 , 7 , 4 , 2 , 1 , 0 ,
0 , 1 , 3 , 5 , 8 , 11 , 15 , 19 , 22 , 26 , 30 , 33 , 36 , 38 , 40 , 41 , 41 , 40 , 38 , 36 , 33 , 30 , 26 , 22 , 19 , 15 , 11 , 8 , 5 , 3 , 1 , 0 ,
0 , 1 , 3 , 6 , 9 , 12 , 17 , 21 , 25 , 30 , 34 , 38 , 41 , 44 , 45 , 46 , 46 , 45 , 44 , 41 , 38 , 34 , 30 , 25 , 21 , 17 , 12 , 9 , 6 , 3 , 1 , 0 ,
0 , 1 , 3 , 6 , 10 , 14 , 19 , 23 , 28 , 33 , 38 , 42 , 45 , 48 , 51 , 52 , 52 , 51 , 48 , 45 , 42 , 38 , 33 , 28 , 23 , 19 , 14 , 10 , 6 , 3 , 1 , 0 ,
0 , 1 , 4 , 7 , 11 , 15 , 20 , 25 , 31 , 36 , 41 , 45 , 49 , 52 , 55 , 56 , 56 , 55 , 52 , 49 , 45 , 41 , 36 , 31 , 25 , 20 , 15 , 11 , 7 , 4 , 1 , 0 ,
0 , 2 , 4 , 7 , 12 , 16 , 22 , 27 , 33 , 38 , 44 , 48 , 52 , 56 , 58 , 60 , 60 , 58 , 56 , 52 , 48 , 44 , 38 , 33 , 27 , 22 , 16 , 12 , 7 , 4 , 2 , 0 ,
0 , 1 , 4 , 8 , 12 , 17 , 22 , 28 , 34 , 40 , 45 , 51 , 55 , 58 , 61 , 62 , 62 , 61 , 58 , 55 , 51 , 45 , 40 , 34 , 28 , 22 , 17 , 12 , 8 , 4 , 1 , 0 ,
0 , 2 , 4 , 8 , 12 , 18 , 23 , 29 , 35 , 41 , 46 , 52 , 56 , 60 , 62 , 64 , 64 , 62 , 60 , 56 , 52 , 46 , 41 , 35 , 29 , 23 , 18 , 12 , 8 , 4 , 2 , 0 ,
0 , 2 , 4 , 8 , 12 , 18 , 23 , 29 , 35 , 41 , 46 , 52 , 56 , 60 , 62 , 64 , 64 , 62 , 60 , 56 , 52 , 46 , 41 , 35 , 29 , 23 , 18 , 12 , 8 , 4 , 2 , 0 ,
0 , 1 , 4 , 8 , 12 , 17 , 22 , 28 , 34 , 40 , 45 , 51 , 55 , 58 , 61 , 62 , 62 , 61 , 58 , 55 , 51 , 45 , 40 , 34 , 28 , 22 , 17 , 12 , 8 , 4 , 1 , 0 ,
0 , 2 , 4 , 7 , 12 , 16 , 22 , 27 , 33 , 38 , 44 , 48 , 52 , 56 , 58 , 60 , 60 , 58 , 56 , 52 , 48 , 44 , 38 , 33 , 27 , 22 , 16 , 12 , 7 , 4 , 2 , 0 ,
0 , 1 , 4 , 7 , 11 , 15 , 20 , 25 , 31 , 36 , 41 , 45 , 49 , 52 , 55 , 56 , 56 , 55 , 52 , 49 , 45 , 41 , 36 , 31 , 25 , 20 , 15 , 11 , 7 , 4 , 1 , 0 ,
0 , 1 , 3 , 6 , 10 , 14 , 19 , 23 , 28 , 33 , 38 , 42 , 45 , 48 , 51 , 52 , 52 , 51 , 48 , 45 , 42 , 38 , 33 , 28 , 23 , 19 , 14 , 10 , 6 , 3 , 1 , 0 ,
0 , 1 , 3 , 6 , 9 , 12 , 17 , 21 , 25 , 30 , 34 , 38 , 41 , 44 , 45 , 46 , 46 , 45 , 44 , 41 , 38 , 34 , 30 , 25 , 21 , 17 , 12 , 9 , 6 , 3 , 1 , 0 ,
0 , 1 , 3 , 5 , 8 , 11 , 15 , 19 , 22 , 26 , 30 , 33 , 36 , 38 , 40 , 41 , 41 , 40 , 38 , 36 , 33 , 30 , 26 , 22 , 19 , 15 , 11 , 8 , 5 , 3 , 1 , 0 ,
0 , 1 , 2 , 4 , 7 , 10 , 13 , 16 , 19 , 22 , 25 , 28 , 31 , 33 , 34 , 35 , 35 , 34 , 33 , 31 , 28 , 25 , 22 , 19 , 16 , 13 , 10 , 7 , 4 , 2 , 1 , 0 ,
0 , 1 , 2 , 4 , 6 , 8 , 10 , 13 , 16 , 19 , 21 , 23 , 25 , 27 , 28 , 29 , 29 , 28 , 27 , 25 , 23 , 21 , 19 , 16 , 13 , 10 , 8 , 6 , 4 , 2 , 1 , 0 ,
0 , 1 , 1 , 3 , 4 , 6 , 8 , 10 , 13 , 15 , 17 , 19 , 20 , 22 , 22 , 23 , 23 , 22 , 22 , 20 , 19 , 17 , 15 , 13 , 10 , 8 , 6 , 4 , 3 , 1 , 1 , 0 ,
0 , 1 , 1 , 2 , 3 , 5 , 6 , 8 , 10 , 11 , 13 , 14 , 15 , 16 , 17 , 18 , 18 , 17 , 16 , 15 , 14 , 13 , 11 , 10 , 8 , 6 , 5 , 3 , 2 , 1 , 1 , 0 ,
0 , 0 , 1 , 2 , 2 , 3 , 4 , 6 , 7 , 8 , 9 , 10 , 11 , 12 , 12 , 12 , 12 , 12 , 12 , 11 , 10 , 9 , 8 , 7 , 6 , 4 , 3 , 2 , 2 , 1 , 0 , 0 ,
0 , 0 , 1 , 1 , 2 , 2 , 3 , 4 , 4 , 5 , 6 , 6 , 7 , 7 , 8 , 8 , 8 , 8 , 7 , 7 , 6 , 6 , 5 , 4 , 4 , 3 , 2 , 2 , 1 , 1 , 0 , 0 ,
0 , 0 , 0 , 1 , 1 , 1 , 1 , 2 , 2 , 3 , 3 , 3 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 4 , 3 , 3 , 3 , 2 , 2 , 1 , 1 , 1 , 1 , 0 , 0 , 0 ,
0 , 0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 2 , 2 , 2 , 2 , 2 , 2 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 , 0 ,
0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 ,
//error:0.000022
} ;
static const uint8_t obmc16 [ 256 ] = {
0 , 0 , 0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 , 0 , 0 ,
0 , 1 , 1 , 2 , 4 , 5 , 5 , 6 , 6 , 5 , 5 , 4 , 2 , 1 , 1 , 0 ,
0 , 1 , 4 , 6 , 9 , 11 , 13 , 15 , 15 , 13 , 11 , 9 , 6 , 4 , 1 , 0 ,
0 , 2 , 6 , 11 , 15 , 20 , 24 , 26 , 26 , 24 , 20 , 15 , 11 , 6 , 2 , 0 ,
0 , 4 , 9 , 15 , 23 , 29 , 34 , 38 , 38 , 34 , 29 , 23 , 15 , 9 , 4 , 0 ,
0 , 5 , 11 , 20 , 29 , 38 , 45 , 49 , 49 , 45 , 38 , 29 , 20 , 11 , 5 , 0 ,
1 , 5 , 13 , 24 , 34 , 45 , 53 , 57 , 57 , 53 , 45 , 34 , 24 , 13 , 5 , 1 ,
1 , 6 , 15 , 26 , 38 , 49 , 57 , 62 , 62 , 57 , 49 , 38 , 26 , 15 , 6 , 1 ,
1 , 6 , 15 , 26 , 38 , 49 , 57 , 62 , 62 , 57 , 49 , 38 , 26 , 15 , 6 , 1 ,
1 , 5 , 13 , 24 , 34 , 45 , 53 , 57 , 57 , 53 , 45 , 34 , 24 , 13 , 5 , 1 ,
0 , 5 , 11 , 20 , 29 , 38 , 45 , 49 , 49 , 45 , 38 , 29 , 20 , 11 , 5 , 0 ,
0 , 4 , 9 , 15 , 23 , 29 , 34 , 38 , 38 , 34 , 29 , 23 , 15 , 9 , 4 , 0 ,
0 , 2 , 6 , 11 , 15 , 20 , 24 , 26 , 26 , 24 , 20 , 15 , 11 , 6 , 2 , 0 ,
0 , 1 , 4 , 6 , 9 , 11 , 13 , 15 , 15 , 13 , 11 , 9 , 6 , 4 , 1 , 0 ,
0 , 1 , 1 , 2 , 4 , 5 , 5 , 6 , 6 , 5 , 5 , 4 , 2 , 1 , 1 , 0 ,
0 , 0 , 0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 , 0 , 0 ,
//error:0.000033
} ;
# elif 1 // 64*linear
static const uint8_t obmc32 [ 1024 ] = {
0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 ,
0 , 1 , 1 , 1 , 2 , 2 , 2 , 3 , 3 , 4 , 4 , 4 , 5 , 5 , 5 , 6 , 6 , 5 , 5 , 5 , 4 , 4 , 4 , 3 , 3 , 2 , 2 , 2 , 1 , 1 , 1 , 0 ,
0 , 1 , 2 , 2 , 3 , 3 , 4 , 5 , 5 , 6 , 7 , 7 , 8 , 8 , 9 , 10 , 10 , 9 , 8 , 8 , 7 , 7 , 6 , 5 , 5 , 4 , 3 , 3 , 2 , 2 , 1 , 0 ,
0 , 1 , 2 , 3 , 4 , 5 , 6 , 7 , 7 , 8 , 9 , 10 , 11 , 12 , 13 , 14 , 14 , 13 , 12 , 11 , 10 , 9 , 8 , 7 , 7 , 6 , 5 , 4 , 3 , 2 , 1 , 0 ,
1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 10 , 11 , 12 , 13 , 14 , 15 , 16 , 17 , 17 , 16 , 15 , 14 , 13 , 12 , 11 , 10 , 8 , 7 , 6 , 5 , 4 , 3 , 2 , 1 ,
1 , 2 , 3 , 5 , 6 , 8 , 9 , 10 , 12 , 13 , 14 , 16 , 17 , 19 , 20 , 21 , 21 , 20 , 19 , 17 , 16 , 14 , 13 , 12 , 10 , 9 , 8 , 6 , 5 , 3 , 2 , 1 ,
1 , 2 , 4 , 6 , 7 , 9 , 11 , 12 , 14 , 15 , 17 , 19 , 20 , 22 , 24 , 25 , 25 , 24 , 22 , 20 , 19 , 17 , 15 , 14 , 12 , 11 , 9 , 7 , 6 , 4 , 2 , 1 ,
1 , 3 , 5 , 7 , 8 , 10 , 12 , 14 , 16 , 18 , 20 , 22 , 23 , 25 , 27 , 29 , 29 , 27 , 25 , 23 , 22 , 20 , 18 , 16 , 14 , 12 , 10 , 8 , 7 , 5 , 3 , 1 ,
1 , 3 , 5 , 7 , 10 , 12 , 14 , 16 , 18 , 20 , 22 , 24 , 27 , 29 , 31 , 33 , 33 , 31 , 29 , 27 , 24 , 22 , 20 , 18 , 16 , 14 , 12 , 10 , 7 , 5 , 3 , 1 ,
1 , 4 , 6 , 8 , 11 , 13 , 15 , 18 , 20 , 23 , 25 , 27 , 30 , 32 , 34 , 37 , 37 , 34 , 32 , 30 , 27 , 25 , 23 , 20 , 18 , 15 , 13 , 11 , 8 , 6 , 4 , 1 ,
1 , 4 , 7 , 9 , 12 , 14 , 17 , 20 , 22 , 25 , 28 , 30 , 33 , 35 , 38 , 41 , 41 , 38 , 35 , 33 , 30 , 28 , 25 , 22 , 20 , 17 , 14 , 12 , 9 , 7 , 4 , 1 ,
1 , 4 , 7 , 10 , 13 , 16 , 19 , 22 , 24 , 27 , 30 , 33 , 36 , 39 , 42 , 45 , 45 , 42 , 39 , 36 , 33 , 30 , 27 , 24 , 22 , 19 , 16 , 13 , 10 , 7 , 4 , 1 ,
2 , 5 , 8 , 11 , 14 , 17 , 20 , 23 , 27 , 30 , 33 , 36 , 39 , 42 , 45 , 48 , 48 , 45 , 42 , 39 , 36 , 33 , 30 , 27 , 23 , 20 , 17 , 14 , 11 , 8 , 5 , 2 ,
2 , 5 , 8 , 12 , 15 , 19 , 22 , 25 , 29 , 32 , 35 , 39 , 42 , 46 , 49 , 52 , 52 , 49 , 46 , 42 , 39 , 35 , 32 , 29 , 25 , 22 , 19 , 15 , 12 , 8 , 5 , 2 ,
2 , 5 , 9 , 13 , 16 , 20 , 24 , 27 , 31 , 34 , 38 , 42 , 45 , 49 , 53 , 56 , 56 , 53 , 49 , 45 , 42 , 38 , 34 , 31 , 27 , 24 , 20 , 16 , 13 , 9 , 5 , 2 ,
2 , 6 , 10 , 14 , 17 , 21 , 25 , 29 , 33 , 37 , 41 , 45 , 48 , 52 , 56 , 60 , 60 , 56 , 52 , 48 , 45 , 41 , 37 , 33 , 29 , 25 , 21 , 17 , 14 , 10 , 6 , 2 ,
2 , 6 , 10 , 14 , 17 , 21 , 25 , 29 , 33 , 37 , 41 , 45 , 48 , 52 , 56 , 60 , 60 , 56 , 52 , 48 , 45 , 41 , 37 , 33 , 29 , 25 , 21 , 17 , 14 , 10 , 6 , 2 ,
2 , 5 , 9 , 13 , 16 , 20 , 24 , 27 , 31 , 34 , 38 , 42 , 45 , 49 , 53 , 56 , 56 , 53 , 49 , 45 , 42 , 38 , 34 , 31 , 27 , 24 , 20 , 16 , 13 , 9 , 5 , 2 ,
2 , 5 , 8 , 12 , 15 , 19 , 22 , 25 , 29 , 32 , 35 , 39 , 42 , 46 , 49 , 52 , 52 , 49 , 46 , 42 , 39 , 35 , 32 , 29 , 25 , 22 , 19 , 15 , 12 , 8 , 5 , 2 ,
2 , 5 , 8 , 11 , 14 , 17 , 20 , 23 , 27 , 30 , 33 , 36 , 39 , 42 , 45 , 48 , 48 , 45 , 42 , 39 , 36 , 33 , 30 , 27 , 23 , 20 , 17 , 14 , 11 , 8 , 5 , 2 ,
1 , 4 , 7 , 10 , 13 , 16 , 19 , 22 , 24 , 27 , 30 , 33 , 36 , 39 , 42 , 45 , 45 , 42 , 39 , 36 , 33 , 30 , 27 , 24 , 22 , 19 , 16 , 13 , 10 , 7 , 4 , 1 ,
1 , 4 , 7 , 9 , 12 , 14 , 17 , 20 , 22 , 25 , 28 , 30 , 33 , 35 , 38 , 41 , 41 , 38 , 35 , 33 , 30 , 28 , 25 , 22 , 20 , 17 , 14 , 12 , 9 , 7 , 4 , 1 ,
1 , 4 , 6 , 8 , 11 , 13 , 15 , 18 , 20 , 23 , 25 , 27 , 30 , 32 , 34 , 37 , 37 , 34 , 32 , 30 , 27 , 25 , 23 , 20 , 18 , 15 , 13 , 11 , 8 , 6 , 4 , 1 ,
1 , 3 , 5 , 7 , 10 , 12 , 14 , 16 , 18 , 20 , 22 , 24 , 27 , 29 , 31 , 33 , 33 , 31 , 29 , 27 , 24 , 22 , 20 , 18 , 16 , 14 , 12 , 10 , 7 , 5 , 3 , 1 ,
1 , 3 , 5 , 7 , 8 , 10 , 12 , 14 , 16 , 18 , 20 , 22 , 23 , 25 , 27 , 29 , 29 , 27 , 25 , 23 , 22 , 20 , 18 , 16 , 14 , 12 , 10 , 8 , 7 , 5 , 3 , 1 ,
1 , 2 , 4 , 6 , 7 , 9 , 11 , 12 , 14 , 15 , 17 , 19 , 20 , 22 , 24 , 25 , 25 , 24 , 22 , 20 , 19 , 17 , 15 , 14 , 12 , 11 , 9 , 7 , 6 , 4 , 2 , 1 ,
1 , 2 , 3 , 5 , 6 , 8 , 9 , 10 , 12 , 13 , 14 , 16 , 17 , 19 , 20 , 21 , 21 , 20 , 19 , 17 , 16 , 14 , 13 , 12 , 10 , 9 , 8 , 6 , 5 , 3 , 2 , 1 ,
1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 10 , 11 , 12 , 13 , 14 , 15 , 16 , 17 , 17 , 16 , 15 , 14 , 13 , 12 , 11 , 10 , 8 , 7 , 6 , 5 , 4 , 3 , 2 , 1 ,
0 , 1 , 2 , 3 , 4 , 5 , 6 , 7 , 7 , 8 , 9 , 10 , 11 , 12 , 13 , 14 , 14 , 13 , 12 , 11 , 10 , 9 , 8 , 7 , 7 , 6 , 5 , 4 , 3 , 2 , 1 , 0 ,
0 , 1 , 2 , 2 , 3 , 3 , 4 , 5 , 5 , 6 , 7 , 7 , 8 , 8 , 9 , 10 , 10 , 9 , 8 , 8 , 7 , 7 , 6 , 5 , 5 , 4 , 3 , 3 , 2 , 2 , 1 , 0 ,
0 , 1 , 1 , 1 , 2 , 2 , 2 , 3 , 3 , 4 , 4 , 4 , 5 , 5 , 5 , 6 , 6 , 5 , 5 , 5 , 4 , 4 , 4 , 3 , 3 , 2 , 2 , 2 , 1 , 1 , 1 , 0 ,
0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 2 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 ,
//error:0.000020
} ;
static const uint8_t obmc16 [ 256 ] = {
0 , 1 , 1 , 2 , 2 , 3 , 3 , 4 , 4 , 3 , 3 , 2 , 2 , 1 , 1 , 0 ,
1 , 2 , 4 , 5 , 7 , 8 , 10 , 11 , 11 , 10 , 8 , 7 , 5 , 4 , 2 , 1 ,
1 , 4 , 6 , 9 , 11 , 14 , 16 , 19 , 19 , 16 , 14 , 11 , 9 , 6 , 4 , 1 ,
2 , 5 , 9 , 12 , 16 , 19 , 23 , 26 , 26 , 23 , 19 , 16 , 12 , 9 , 5 , 2 ,
2 , 7 , 11 , 16 , 20 , 25 , 29 , 34 , 34 , 29 , 25 , 20 , 16 , 11 , 7 , 2 ,
3 , 8 , 14 , 19 , 25 , 30 , 36 , 41 , 41 , 36 , 30 , 25 , 19 , 14 , 8 , 3 ,
3 , 10 , 16 , 23 , 29 , 36 , 42 , 49 , 49 , 42 , 36 , 29 , 23 , 16 , 10 , 3 ,
4 , 11 , 19 , 26 , 34 , 41 , 49 , 56 , 56 , 49 , 41 , 34 , 26 , 19 , 11 , 4 ,
4 , 11 , 19 , 26 , 34 , 41 , 49 , 56 , 56 , 49 , 41 , 34 , 26 , 19 , 11 , 4 ,
3 , 10 , 16 , 23 , 29 , 36 , 42 , 49 , 49 , 42 , 36 , 29 , 23 , 16 , 10 , 3 ,
3 , 8 , 14 , 19 , 25 , 30 , 36 , 41 , 41 , 36 , 30 , 25 , 19 , 14 , 8 , 3 ,
2 , 7 , 11 , 16 , 20 , 25 , 29 , 34 , 34 , 29 , 25 , 20 , 16 , 11 , 7 , 2 ,
2 , 5 , 9 , 12 , 16 , 19 , 23 , 26 , 26 , 23 , 19 , 16 , 12 , 9 , 5 , 2 ,
1 , 4 , 6 , 9 , 11 , 14 , 16 , 19 , 19 , 16 , 14 , 11 , 9 , 6 , 4 , 1 ,
1 , 2 , 4 , 5 , 7 , 8 , 10 , 11 , 11 , 10 , 8 , 7 , 5 , 4 , 2 , 1 ,
0 , 1 , 1 , 2 , 2 , 3 , 3 , 4 , 4 , 3 , 3 , 2 , 2 , 1 , 1 , 0 ,
//error:0.000015
} ;
# else //64*cos
static const uint8_t obmc32 [ 1024 ] = {
0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 ,
0 , 0 , 0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 2 , 1 , 1 , 2 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 , 0 , 0 ,
0 , 0 , 0 , 1 , 1 , 1 , 1 , 2 , 2 , 3 , 3 , 3 , 3 , 4 , 4 , 4 , 4 , 4 , 4 , 3 , 3 , 3 , 3 , 2 , 2 , 1 , 1 , 1 , 1 , 0 , 0 , 0 ,
0 , 0 , 1 , 1 , 1 , 2 , 2 , 3 , 4 , 5 , 5 , 6 , 7 , 7 , 7 , 7 , 7 , 7 , 7 , 7 , 6 , 5 , 5 , 4 , 3 , 2 , 2 , 1 , 1 , 1 , 0 , 0 ,
0 , 0 , 1 , 1 , 2 , 3 , 4 , 5 , 6 , 7 , 9 , 10 , 11 , 11 , 12 , 12 , 12 , 12 , 11 , 11 , 10 , 9 , 7 , 6 , 5 , 4 , 3 , 2 , 1 , 1 , 0 , 0 ,
0 , 0 , 1 , 2 , 3 , 5 , 6 , 8 , 9 , 11 , 12 , 14 , 15 , 16 , 17 , 17 , 17 , 17 , 16 , 15 , 14 , 12 , 11 , 9 , 8 , 6 , 5 , 3 , 2 , 1 , 0 , 0 ,
0 , 1 , 1 , 2 , 4 , 6 , 8 , 10 , 12 , 15 , 17 , 19 , 20 , 21 , 22 , 23 , 23 , 22 , 21 , 20 , 19 , 17 , 15 , 12 , 10 , 8 , 6 , 4 , 2 , 1 , 1 , 0 ,
0 , 1 , 2 , 3 , 5 , 8 , 10 , 13 , 16 , 19 , 21 , 24 , 26 , 27 , 28 , 29 , 29 , 28 , 27 , 26 , 24 , 21 , 19 , 16 , 13 , 10 , 8 , 5 , 3 , 2 , 1 , 0 ,
0 , 1 , 2 , 4 , 6 , 9 , 12 , 16 , 19 , 23 , 26 , 29 , 31 , 33 , 34 , 35 , 35 , 34 , 33 , 31 , 29 , 26 , 23 , 19 , 16 , 12 , 9 , 6 , 4 , 2 , 1 , 0 ,
0 , 1 , 3 , 5 , 7 , 11 , 15 , 19 , 23 , 26 , 30 , 34 , 37 , 39 , 40 , 41 , 41 , 40 , 39 , 37 , 34 , 30 , 26 , 23 , 19 , 15 , 11 , 7 , 5 , 3 , 1 , 0 ,
0 , 1 , 3 , 5 , 9 , 12 , 17 , 21 , 26 , 30 , 35 , 38 , 42 , 44 , 46 , 47 , 47 , 46 , 44 , 42 , 38 , 35 , 30 , 26 , 21 , 17 , 12 , 9 , 5 , 3 , 1 , 0 ,
0 , 1 , 3 , 6 , 9 , 14 , 19 , 24 , 29 , 34 , 38 , 43 , 46 , 49 , 51 , 52 , 52 , 51 , 49 , 46 , 43 , 38 , 34 , 29 , 24 , 19 , 14 , 9 , 6 , 3 , 1 , 0 ,
0 , 1 , 3 , 6 , 11 , 15 , 20 , 26 , 31 , 37 , 42 , 46 , 50 , 53 , 56 , 57 , 57 , 56 , 53 , 50 , 46 , 42 , 37 , 31 , 26 , 20 , 15 , 11 , 6 , 3 , 1 , 0 ,
0 , 1 , 3 , 7 , 11 , 16 , 21 , 27 , 33 , 39 , 44 , 49 , 53 , 57 , 59 , 60 , 60 , 59 , 57 , 53 , 49 , 44 , 39 , 33 , 27 , 21 , 16 , 11 , 7 , 3 , 1 , 0 ,
0 , 1 , 4 , 7 , 12 , 17 , 22 , 28 , 34 , 40 , 46 , 51 , 56 , 59 , 61 , 63 , 63 , 61 , 59 , 56 , 51 , 46 , 40 , 34 , 28 , 22 , 17 , 12 , 7 , 4 , 1 , 0 ,
0 , 1 , 4 , 7 , 12 , 17 , 23 , 29 , 35 , 41 , 47 , 52 , 57 , 60 , 63 , 64 , 64 , 63 , 60 , 57 , 52 , 47 , 41 , 35 , 29 , 23 , 17 , 12 , 7 , 4 , 1 , 0 ,
0 , 1 , 4 , 7 , 12 , 17 , 23 , 29 , 35 , 41 , 47 , 52 , 57 , 60 , 63 , 64 , 64 , 63 , 60 , 57 , 52 , 47 , 41 , 35 , 29 , 23 , 17 , 12 , 7 , 4 , 1 , 0 ,
0 , 1 , 4 , 7 , 12 , 17 , 22 , 28 , 34 , 40 , 46 , 51 , 56 , 59 , 61 , 63 , 63 , 61 , 59 , 56 , 51 , 46 , 40 , 34 , 28 , 22 , 17 , 12 , 7 , 4 , 1 , 0 ,
0 , 1 , 3 , 7 , 11 , 16 , 21 , 27 , 33 , 39 , 44 , 49 , 53 , 57 , 59 , 60 , 60 , 59 , 57 , 53 , 49 , 44 , 39 , 33 , 27 , 21 , 16 , 11 , 7 , 3 , 1 , 0 ,
0 , 1 , 3 , 6 , 11 , 15 , 20 , 26 , 31 , 37 , 42 , 46 , 50 , 53 , 56 , 57 , 57 , 56 , 53 , 50 , 46 , 42 , 37 , 31 , 26 , 20 , 15 , 11 , 6 , 3 , 1 , 0 ,
0 , 1 , 3 , 6 , 9 , 14 , 19 , 24 , 29 , 34 , 38 , 43 , 46 , 49 , 51 , 52 , 52 , 51 , 49 , 46 , 43 , 38 , 34 , 29 , 24 , 19 , 14 , 9 , 6 , 3 , 1 , 0 ,
0 , 1 , 3 , 5 , 9 , 12 , 17 , 21 , 26 , 30 , 35 , 38 , 42 , 44 , 46 , 47 , 47 , 46 , 44 , 42 , 38 , 35 , 30 , 26 , 21 , 17 , 12 , 9 , 5 , 3 , 1 , 0 ,
0 , 1 , 3 , 5 , 7 , 11 , 15 , 19 , 23 , 26 , 30 , 34 , 37 , 39 , 40 , 41 , 41 , 40 , 39 , 37 , 34 , 30 , 26 , 23 , 19 , 15 , 11 , 7 , 5 , 3 , 1 , 0 ,
0 , 1 , 2 , 4 , 6 , 9 , 12 , 16 , 19 , 23 , 26 , 29 , 31 , 33 , 34 , 35 , 35 , 34 , 33 , 31 , 29 , 26 , 23 , 19 , 16 , 12 , 9 , 6 , 4 , 2 , 1 , 0 ,
0 , 1 , 2 , 3 , 5 , 8 , 10 , 13 , 16 , 19 , 21 , 24 , 26 , 27 , 28 , 29 , 29 , 28 , 27 , 26 , 24 , 21 , 19 , 16 , 13 , 10 , 8 , 5 , 3 , 2 , 1 , 0 ,
0 , 1 , 1 , 2 , 4 , 6 , 8 , 10 , 12 , 15 , 17 , 19 , 20 , 21 , 22 , 23 , 23 , 22 , 21 , 20 , 19 , 17 , 15 , 12 , 10 , 8 , 6 , 4 , 2 , 1 , 1 , 0 ,
0 , 0 , 1 , 2 , 3 , 5 , 6 , 8 , 9 , 11 , 12 , 14 , 15 , 16 , 17 , 17 , 17 , 17 , 16 , 15 , 14 , 12 , 11 , 9 , 8 , 6 , 5 , 3 , 2 , 1 , 0 , 0 ,
0 , 0 , 1 , 1 , 2 , 3 , 4 , 5 , 6 , 7 , 9 , 10 , 11 , 11 , 12 , 12 , 12 , 12 , 11 , 11 , 10 , 9 , 7 , 6 , 5 , 4 , 3 , 2 , 1 , 1 , 0 , 0 ,
0 , 0 , 1 , 1 , 1 , 2 , 2 , 3 , 4 , 5 , 5 , 6 , 7 , 7 , 7 , 7 , 7 , 7 , 7 , 7 , 6 , 5 , 5 , 4 , 3 , 2 , 2 , 1 , 1 , 1 , 0 , 0 ,
0 , 0 , 0 , 1 , 1 , 1 , 1 , 2 , 2 , 3 , 3 , 3 , 3 , 4 , 4 , 4 , 4 , 4 , 4 , 3 , 3 , 3 , 3 , 2 , 2 , 1 , 1 , 1 , 1 , 0 , 0 , 0 ,
0 , 0 , 0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 2 , 1 , 1 , 2 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 , 0 , 0 ,
0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 ,
//error:0.000022
} ;
static const uint8_t obmc16 [ 256 ] = {
0 , 0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 , 0 ,
0 , 0 , 1 , 2 , 3 , 4 , 5 , 5 , 5 , 5 , 4 , 3 , 2 , 1 , 0 , 0 ,
0 , 1 , 3 , 6 , 8 , 11 , 13 , 14 , 14 , 13 , 11 , 8 , 6 , 3 , 1 , 0 ,
0 , 2 , 6 , 10 , 15 , 20 , 24 , 26 , 26 , 24 , 20 , 15 , 10 , 6 , 2 , 0 ,
0 , 3 , 8 , 16 , 23 , 30 , 35 , 38 , 38 , 35 , 30 , 23 , 16 , 8 , 3 , 0 ,
1 , 4 , 11 , 20 , 30 , 39 , 46 , 49 , 49 , 46 , 39 , 30 , 20 , 11 , 4 , 1 ,
1 , 5 , 13 , 24 , 35 , 46 , 54 , 58 , 58 , 54 , 46 , 35 , 24 , 13 , 5 , 1 ,
0 , 5 , 14 , 26 , 38 , 49 , 58 , 63 , 63 , 58 , 49 , 38 , 26 , 14 , 5 , 0 ,
0 , 5 , 14 , 26 , 38 , 49 , 58 , 63 , 63 , 58 , 49 , 38 , 26 , 14 , 5 , 0 ,
1 , 5 , 13 , 24 , 35 , 46 , 54 , 58 , 58 , 54 , 46 , 35 , 24 , 13 , 5 , 1 ,
1 , 4 , 11 , 20 , 30 , 39 , 46 , 49 , 49 , 46 , 39 , 30 , 20 , 11 , 4 , 1 ,
0 , 3 , 8 , 16 , 23 , 30 , 35 , 38 , 38 , 35 , 30 , 23 , 16 , 8 , 3 , 0 ,
0 , 2 , 6 , 10 , 15 , 20 , 24 , 26 , 26 , 24 , 20 , 15 , 10 , 6 , 2 , 0 ,
0 , 1 , 3 , 6 , 8 , 11 , 13 , 14 , 14 , 13 , 11 , 8 , 6 , 3 , 1 , 0 ,
0 , 0 , 1 , 2 , 3 , 4 , 5 , 5 , 5 , 5 , 4 , 3 , 2 , 1 , 0 , 0 ,
0 , 0 , 0 , 0 , 0 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 , 0 , 0 , 0 ,
//error:0.000022
} ;
# endif
2004-08-26 22:04:54 +02:00
//linear *64
static const uint8_t obmc8 [ 64 ] = {
1 , 3 , 5 , 7 , 7 , 5 , 3 , 1 ,
3 , 9 , 15 , 21 , 21 , 15 , 9 , 3 ,
5 , 15 , 25 , 35 , 35 , 25 , 15 , 5 ,
7 , 21 , 35 , 49 , 49 , 35 , 21 , 7 ,
7 , 21 , 35 , 49 , 49 , 35 , 21 , 7 ,
5 , 15 , 25 , 35 , 35 , 25 , 15 , 5 ,
3 , 9 , 15 , 21 , 21 , 15 , 9 , 3 ,
1 , 3 , 5 , 7 , 7 , 5 , 3 , 1 ,
//error:0.000000
} ;
//linear *64
static const uint8_t obmc4 [ 16 ] = {
4 , 12 , 12 , 4 ,
12 , 36 , 36 , 12 ,
12 , 36 , 36 , 12 ,
4 , 12 , 12 , 4 ,
//error:0.000000
} ;
static const uint8_t * obmc_tab [ 4 ] = {
obmc32 , obmc16 , obmc8 , obmc4
} ;
typedef struct BlockNode {
int16_t mx ;
int16_t my ;
uint8_t color [ 3 ] ;
uint8_t type ;
//#define TYPE_SPLIT 1
# define BLOCK_INTRA 1
2005-12-28 16:43:53 +01:00
# define BLOCK_OPT 2
2004-08-26 22:04:54 +02:00
//#define TYPE_NOCOLOR 4
uint8_t level ; //FIXME merge into type?
} BlockNode ;
2005-12-28 16:43:53 +01:00
static const BlockNode null_block = { //FIXME add border maybe
. color = { 128 , 128 , 128 } ,
. mx = 0 ,
. my = 0 ,
. type = 0 ,
. level = 0 ,
} ;
2004-08-26 22:04:54 +02:00
# define LOG2_MB_SIZE 4
# define MB_SIZE (1<<LOG2_MB_SIZE)
2005-04-03 17:43:57 +02:00
typedef struct x_and_coeff {
int16_t x ;
2005-04-05 19:59:05 +02:00
uint16_t coeff ;
2005-04-03 17:43:57 +02:00
} x_and_coeff ;
2004-07-26 01:36:44 +02:00
typedef struct SubBand {
int level ;
int stride ;
int width ;
int height ;
int qlog ; ///< log(qscale)/log[2^(1/6)]
DWTELEM * buf ;
2005-04-03 17:43:57 +02:00
int buf_x_offset ;
int buf_y_offset ;
int stride_line ; ///< Stride measured in lines, not pixels.
x_and_coeff * x_coeff ;
2004-07-26 01:36:44 +02:00
struct SubBand * parent ;
uint8_t state [ /*7*2*/ 7 + 512 ] [ 32 ] ;
} SubBand ;
typedef struct Plane {
int width ;
int height ;
SubBand band [ MAX_DECOMPOSITIONS ] [ 4 ] ;
} Plane ;
2005-04-03 17:43:57 +02:00
/** Used to minimize the amount of memory used in order to optimize cache performance. **/
typedef struct {
DWTELEM * * line ; ///< For use by idwt and predict_slices.
DWTELEM * * data_stack ; ///< Used for internal purposes.
int data_stack_top ;
int line_count ;
int line_width ;
int data_count ;
DWTELEM * base_buffer ; ///< Buffer that this structure is caching.
} slice_buffer ;
2004-07-26 01:36:44 +02:00
typedef struct SnowContext {
// MpegEncContext m; // needed for motion estimation, should not be used for anything else, the idea is to make the motion estimation eventually independant of MpegEncContext, so this will be removed then (FIXME/XXX)
AVCodecContext * avctx ;
2004-10-30 04:16:27 +02:00
RangeCoder c ;
2004-07-26 01:36:44 +02:00
DSPContext dsp ;
2005-12-28 16:43:53 +01:00
AVFrame new_picture ;
AVFrame input_picture ; ///< new_picture with the internal linesizes
2004-07-26 01:36:44 +02:00
AVFrame current_picture ;
AVFrame last_picture ;
AVFrame mconly_picture ;
// uint8_t q_context[16];
uint8_t header_state [ 32 ] ;
2004-08-26 22:04:54 +02:00
uint8_t block_state [ 128 + 32 * 128 ] ;
2004-07-26 01:36:44 +02:00
int keyframe ;
2004-08-31 18:28:49 +02:00
int always_reset ;
2004-07-26 01:36:44 +02:00
int version ;
int spatial_decomposition_type ;
int temporal_decomposition_type ;
int spatial_decomposition_count ;
int temporal_decomposition_count ;
DWTELEM * spatial_dwt_buffer ;
int colorspace_type ;
int chroma_h_shift ;
int chroma_v_shift ;
int spatial_scalability ;
int qlog ;
2004-08-26 22:04:54 +02:00
int lambda ;
int lambda2 ;
2004-07-26 01:36:44 +02:00
int mv_scale ;
int qbias ;
# define QBIAS_SHIFT 3
2004-08-26 22:04:54 +02:00
int b_width ;
int b_height ;
int block_max_depth ;
2004-07-26 01:36:44 +02:00
Plane plane [ MAX_PLANES ] ;
2004-08-26 22:04:54 +02:00
BlockNode * block ;
2005-12-28 16:43:53 +01:00
# define ME_CACHE_SIZE 1024
int me_cache [ ME_CACHE_SIZE ] ;
int me_cache_generation ;
2005-04-03 17:43:57 +02:00
slice_buffer sb ;
2004-08-26 22:04:54 +02:00
2004-07-26 01:36:44 +02:00
MpegEncContext m ; // needed for motion estimation, should not be used for anything else, the idea is to make the motion estimation eventually independant of MpegEncContext, so this will be removed then (FIXME/XXX)
} SnowContext ;
2005-03-15 22:05:34 +01:00
typedef struct {
DWTELEM * b0 ;
DWTELEM * b1 ;
DWTELEM * b2 ;
DWTELEM * b3 ;
int y ;
} dwt_compose_t ;
2005-04-03 17:43:57 +02:00
# define slice_buffer_get_line(slice_buf, line_num) ((slice_buf)->line[line_num] ? (slice_buf)->line[line_num] : slice_buffer_load_line((slice_buf), (line_num)))
//#define slice_buffer_get_line(slice_buf, line_num) (slice_buffer_load_line((slice_buf), (line_num)))
2005-12-28 16:43:53 +01:00
static void iterative_me ( SnowContext * s ) ;
2005-04-03 17:43:57 +02:00
static void slice_buffer_init ( slice_buffer * buf , int line_count , int max_allocated_lines , int line_width , DWTELEM * base_buffer )
{
int i ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
buf - > base_buffer = base_buffer ;
buf - > line_count = line_count ;
buf - > line_width = line_width ;
buf - > data_count = max_allocated_lines ;
buf - > line = ( DWTELEM * * ) av_mallocz ( sizeof ( DWTELEM * ) * line_count ) ;
buf - > data_stack = ( DWTELEM * * ) av_malloc ( sizeof ( DWTELEM * ) * max_allocated_lines ) ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
for ( i = 0 ; i < max_allocated_lines ; i + + )
{
buf - > data_stack [ i ] = ( DWTELEM * ) av_malloc ( sizeof ( DWTELEM ) * line_width ) ;
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
buf - > data_stack_top = max_allocated_lines - 1 ;
}
static DWTELEM * slice_buffer_load_line ( slice_buffer * buf , int line )
{
int offset ;
DWTELEM * buffer ;
2005-12-17 19:14:38 +01:00
// av_log(NULL, AV_LOG_DEBUG, "Cache hit: %d\n", line);
2005-04-03 17:43:57 +02:00
assert ( buf - > data_stack_top > = 0 ) ;
// assert(!buf->line[line]);
if ( buf - > line [ line ] )
return buf - > line [ line ] ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
offset = buf - > line_width * line ;
buffer = buf - > data_stack [ buf - > data_stack_top ] ;
buf - > data_stack_top - - ;
buf - > line [ line ] = buffer ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
// av_log(NULL, AV_LOG_DEBUG, "slice_buffer_load_line: line: %d remaining: %d\n", line, buf->data_stack_top + 1);
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
return buffer ;
}
static void slice_buffer_release ( slice_buffer * buf , int line )
{
int offset ;
DWTELEM * buffer ;
assert ( line > = 0 & & line < buf - > line_count ) ;
assert ( buf - > line [ line ] ) ;
offset = buf - > line_width * line ;
buffer = buf - > line [ line ] ;
buf - > data_stack_top + + ;
buf - > data_stack [ buf - > data_stack_top ] = buffer ;
buf - > line [ line ] = NULL ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
// av_log(NULL, AV_LOG_DEBUG, "slice_buffer_release: line: %d remaining: %d\n", line, buf->data_stack_top + 1);
}
static void slice_buffer_flush ( slice_buffer * buf )
{
int i ;
for ( i = 0 ; i < buf - > line_count ; i + + )
{
if ( buf - > line [ i ] )
{
// av_log(NULL, AV_LOG_DEBUG, "slice_buffer_flush: line: %d \n", i);
slice_buffer_release ( buf , i ) ;
}
}
}
static void slice_buffer_destroy ( slice_buffer * buf )
{
int i ;
slice_buffer_flush ( buf ) ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
for ( i = buf - > data_count - 1 ; i > = 0 ; i - - )
{
assert ( buf - > data_stack [ i ] ) ;
av_free ( buf - > data_stack [ i ] ) ;
}
assert ( buf - > data_stack ) ;
av_free ( buf - > data_stack ) ;
assert ( buf - > line ) ;
av_free ( buf - > line ) ;
}
2005-12-22 02:10:11 +01:00
# ifdef __sgi
2004-11-25 20:26:46 +01:00
// Avoid a name clash on SGI IRIX
2005-12-22 02:10:11 +01:00
# undef qexp
2004-11-25 20:26:46 +01:00
# endif
2004-09-22 01:10:10 +02:00
# define QEXPSHIFT (7-FRAC_BITS+8) //FIXME try to change this to 0
2005-04-09 21:48:59 +02:00
static uint8_t qexp [ QROOT ] ;
2004-07-26 01:36:44 +02:00
static inline int mirror ( int v , int m ) {
2005-12-30 16:12:02 +01:00
while ( ( unsigned ) v > ( unsigned ) m ) {
v = - v ;
if ( v < 0 ) v + = 2 * m ;
}
return v ;
2004-07-26 01:36:44 +02:00
}
2004-10-30 04:16:27 +02:00
static inline void put_symbol ( RangeCoder * c , uint8_t * state , int v , int is_signed ) {
2004-07-26 01:36:44 +02:00
int i ;
if ( v ) {
const int a = ABS ( v ) ;
const int e = av_log2 ( a ) ;
# if 1
2005-12-17 19:14:38 +01:00
const int el = FFMIN ( e , 10 ) ;
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 0 , 0 ) ;
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < el ; i + + ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 1 + i , 1 ) ; //1..10
2004-07-26 01:36:44 +02:00
}
for ( ; i < e ; i + + ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 1 + 9 , 1 ) ; //1..10
2004-07-26 01:36:44 +02:00
}
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 1 + FFMIN ( i , 9 ) , 0 ) ;
2004-07-26 01:36:44 +02:00
for ( i = e - 1 ; i > = el ; i - - ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 22 + 9 , ( a > > i ) & 1 ) ; //22..31
2004-07-26 01:36:44 +02:00
}
for ( ; i > = 0 ; i - - ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 22 + i , ( a > > i ) & 1 ) ; //22..31
2004-07-26 01:36:44 +02:00
}
if ( is_signed )
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 11 + el , v < 0 ) ; //11..21
2004-07-26 01:36:44 +02:00
# else
2005-12-17 19:14:38 +01:00
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 0 , 0 ) ;
2004-07-26 01:36:44 +02:00
if ( e < = 9 ) {
for ( i = 0 ; i < e ; i + + ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 1 + i , 1 ) ; //1..10
2004-07-26 01:36:44 +02:00
}
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 1 + i , 0 ) ;
2004-07-26 01:36:44 +02:00
for ( i = e - 1 ; i > = 0 ; i - - ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 22 + i , ( a > > i ) & 1 ) ; //22..31
2004-07-26 01:36:44 +02:00
}
if ( is_signed )
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 11 + e , v < 0 ) ; //11..21
2004-07-26 01:36:44 +02:00
} else {
for ( i = 0 ; i < e ; i + + ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 1 + FFMIN ( i , 9 ) , 1 ) ; //1..10
2004-07-26 01:36:44 +02:00
}
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 1 + FFMIN ( i , 9 ) , 0 ) ;
2004-07-26 01:36:44 +02:00
for ( i = e - 1 ; i > = 0 ; i - - ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 22 + FFMIN ( i , 9 ) , ( a > > i ) & 1 ) ; //22..31
2004-07-26 01:36:44 +02:00
}
if ( is_signed )
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 11 + FFMIN ( e , 10 ) , v < 0 ) ; //11..21
2004-07-26 01:36:44 +02:00
}
# endif
} else {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 0 , 1 ) ;
2004-07-26 01:36:44 +02:00
}
}
2004-10-30 04:16:27 +02:00
static inline int get_symbol ( RangeCoder * c , uint8_t * state , int is_signed ) {
if ( get_rac ( c , state + 0 ) )
2004-07-26 01:36:44 +02:00
return 0 ;
else {
2004-09-18 01:51:36 +02:00
int i , e , a ;
e = 0 ;
2004-10-30 04:16:27 +02:00
while ( get_rac ( c , state + 1 + FFMIN ( e , 9 ) ) ) { //1..10
2004-09-18 01:51:36 +02:00
e + + ;
2004-07-26 01:36:44 +02:00
}
2004-09-18 01:51:36 +02:00
2004-07-26 01:36:44 +02:00
a = 1 ;
2004-09-18 01:51:36 +02:00
for ( i = e - 1 ; i > = 0 ; i - - ) {
2004-10-30 04:16:27 +02:00
a + = a + get_rac ( c , state + 22 + FFMIN ( i , 9 ) ) ; //22..31
2004-07-26 01:36:44 +02:00
}
2004-10-30 04:16:27 +02:00
if ( is_signed & & get_rac ( c , state + 11 + FFMIN ( e , 10 ) ) ) //11..21
2004-07-26 01:36:44 +02:00
return - a ;
else
return a ;
}
}
2004-10-30 04:16:27 +02:00
static inline void put_symbol2 ( RangeCoder * c , uint8_t * state , int v , int log2 ) {
2004-08-02 03:03:52 +02:00
int i ;
2004-08-02 19:53:57 +02:00
int r = log2 > = 0 ? 1 < < log2 : 1 ;
2004-08-02 03:03:52 +02:00
assert ( v > = 0 ) ;
2004-08-02 19:53:57 +02:00
assert ( log2 > = - 4 ) ;
while ( v > = r ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 4 + log2 , 1 ) ;
2004-08-02 19:53:57 +02:00
v - = r ;
2004-08-02 03:03:52 +02:00
log2 + + ;
2004-08-02 19:53:57 +02:00
if ( log2 > 0 ) r + = r ;
2004-08-02 03:03:52 +02:00
}
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 4 + log2 , 0 ) ;
2005-12-17 19:14:38 +01:00
2004-08-02 03:03:52 +02:00
for ( i = log2 - 1 ; i > = 0 ; i - - ) {
2004-10-30 04:16:27 +02:00
put_rac ( c , state + 31 - i , ( v > > i ) & 1 ) ;
2004-08-02 03:03:52 +02:00
}
}
2004-10-30 04:16:27 +02:00
static inline int get_symbol2 ( RangeCoder * c , uint8_t * state , int log2 ) {
2004-08-02 03:03:52 +02:00
int i ;
2004-08-02 19:53:57 +02:00
int r = log2 > = 0 ? 1 < < log2 : 1 ;
2004-08-02 03:03:52 +02:00
int v = 0 ;
2004-08-02 19:53:57 +02:00
assert ( log2 > = - 4 ) ;
2004-10-30 04:16:27 +02:00
while ( get_rac ( c , state + 4 + log2 ) ) {
2004-08-02 19:53:57 +02:00
v + = r ;
2004-08-02 03:03:52 +02:00
log2 + + ;
2004-08-02 19:53:57 +02:00
if ( log2 > 0 ) r + = r ;
2004-08-02 03:03:52 +02:00
}
2005-12-17 19:14:38 +01:00
2004-08-02 03:03:52 +02:00
for ( i = log2 - 1 ; i > = 0 ; i - - ) {
2004-10-30 04:16:27 +02:00
v + = get_rac ( c , state + 31 - i ) < < i ;
2004-08-02 03:03:52 +02:00
}
return v ;
}
2004-07-26 01:36:44 +02:00
static always_inline void lift ( DWTELEM * dst , DWTELEM * src , DWTELEM * ref , int dst_step , int src_step , int ref_step , int width , int mul , int add , int shift , int highpass , int inverse ) {
const int mirror_left = ! highpass ;
const int mirror_right = ( width & 1 ) ^ highpass ;
const int w = ( width > > 1 ) - 1 + ( highpass & width ) ;
int i ;
# define LIFT(src, ref, inv) ((src) + ((inv) ? - (ref) : + (ref)))
if ( mirror_left ) {
dst [ 0 ] = LIFT ( src [ 0 ] , ( ( mul * 2 * ref [ 0 ] + add ) > > shift ) , inverse ) ;
dst + = dst_step ;
src + = src_step ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < w ; i + + ) {
dst [ i * dst_step ] = LIFT ( src [ i * src_step ] , ( ( mul * ( ref [ i * ref_step ] + ref [ ( i + 1 ) * ref_step ] ) + add ) > > shift ) , inverse ) ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
if ( mirror_right ) {
dst [ w * dst_step ] = LIFT ( src [ w * src_step ] , ( ( mul * 2 * ref [ w * ref_step ] + add ) > > shift ) , inverse ) ;
}
}
static always_inline void lift5 ( DWTELEM * dst , DWTELEM * src , DWTELEM * ref , int dst_step , int src_step , int ref_step , int width , int mul , int add , int shift , int highpass , int inverse ) {
const int mirror_left = ! highpass ;
const int mirror_right = ( width & 1 ) ^ highpass ;
const int w = ( width > > 1 ) - 1 + ( highpass & width ) ;
int i ;
if ( mirror_left ) {
int r = 3 * 2 * ref [ 0 ] ;
r + = r > > 4 ;
r + = r > > 8 ;
dst [ 0 ] = LIFT ( src [ 0 ] , ( ( r + add ) > > shift ) , inverse ) ;
dst + = dst_step ;
src + = src_step ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < w ; i + + ) {
int r = 3 * ( ref [ i * ref_step ] + ref [ ( i + 1 ) * ref_step ] ) ;
r + = r > > 4 ;
r + = r > > 8 ;
dst [ i * dst_step ] = LIFT ( src [ i * src_step ] , ( ( r + add ) > > shift ) , inverse ) ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
if ( mirror_right ) {
int r = 3 * 2 * ref [ w * ref_step ] ;
r + = r > > 4 ;
r + = r > > 8 ;
dst [ w * dst_step ] = LIFT ( src [ w * src_step ] , ( ( r + add ) > > shift ) , inverse ) ;
}
}
2005-04-10 16:21:53 +02:00
static always_inline void liftS ( DWTELEM * dst , DWTELEM * src , DWTELEM * ref , int dst_step , int src_step , int ref_step , int width , int mul , int add , int shift , int highpass , int inverse ) {
const int mirror_left = ! highpass ;
const int mirror_right = ( width & 1 ) ^ highpass ;
const int w = ( width > > 1 ) - 1 + ( highpass & width ) ;
int i ;
assert ( shift = = 4 ) ;
# define LIFTS(src, ref, inv) ((inv) ? (src) - (((ref) - 4*(src))>>shift): (16*4*(src) + 4*(ref) + 8 + (5<<27)) / (5*16) - (1<<23))
if ( mirror_left ) {
dst [ 0 ] = LIFTS ( src [ 0 ] , mul * 2 * ref [ 0 ] + add , inverse ) ;
dst + = dst_step ;
src + = src_step ;
}
2005-12-17 19:14:38 +01:00
2005-04-10 16:21:53 +02:00
for ( i = 0 ; i < w ; i + + ) {
dst [ i * dst_step ] = LIFTS ( src [ i * src_step ] , mul * ( ref [ i * ref_step ] + ref [ ( i + 1 ) * ref_step ] ) + add , inverse ) ;
}
2005-12-17 19:14:38 +01:00
2005-04-10 16:21:53 +02:00
if ( mirror_right ) {
dst [ w * dst_step ] = LIFTS ( src [ w * src_step ] , mul * 2 * ref [ w * ref_step ] + add , inverse ) ;
}
}
2004-07-26 01:36:44 +02:00
2004-09-19 16:43:09 +02:00
static void inplace_lift ( DWTELEM * dst , int width , int * coeffs , int n , int shift , int start , int inverse ) {
2004-07-26 01:36:44 +02:00
int x , i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( x = start ; x < width ; x + = 2 ) {
int64_t sum = 0 ;
for ( i = 0 ; i < n ; i + + ) {
int x2 = x + 2 * i - n + 1 ;
if ( x2 < 0 ) x2 = - x2 ;
else if ( x2 > = width ) x2 = 2 * width - x2 - 2 ;
sum + = coeffs [ i ] * ( int64_t ) dst [ x2 ] ;
}
if ( inverse ) dst [ x ] - = ( sum + ( 1 < < shift ) / 2 ) > > shift ;
else dst [ x ] + = ( sum + ( 1 < < shift ) / 2 ) > > shift ;
}
}
2004-09-19 16:43:09 +02:00
static void inplace_liftV ( DWTELEM * dst , int width , int height , int stride , int * coeffs , int n , int shift , int start , int inverse ) {
2004-07-26 01:36:44 +02:00
int x , y , i ;
for ( y = start ; y < height ; y + = 2 ) {
for ( x = 0 ; x < width ; x + + ) {
int64_t sum = 0 ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < n ; i + + ) {
int y2 = y + 2 * i - n + 1 ;
if ( y2 < 0 ) y2 = - y2 ;
else if ( y2 > = height ) y2 = 2 * height - y2 - 2 ;
sum + = coeffs [ i ] * ( int64_t ) dst [ x + y2 * stride ] ;
}
if ( inverse ) dst [ x + y * stride ] - = ( sum + ( 1 < < shift ) / 2 ) > > shift ;
else dst [ x + y * stride ] + = ( sum + ( 1 < < shift ) / 2 ) > > shift ;
}
}
}
# define SCALEX 1
# define LX0 0
# define LX1 1
2004-07-26 15:31:48 +02:00
#if 0 // more accurate 9/7
2004-07-26 01:36:44 +02:00
# define N1 2
# define SHIFT1 14
# define COEFFS1 (int[]){-25987,-25987}
# define N2 2
# define SHIFT2 19
# define COEFFS2 (int[]){-27777,-27777}
# define N3 2
# define SHIFT3 15
# define COEFFS3 (int[]){28931,28931}
# define N4 2
# define SHIFT4 15
# define COEFFS4 (int[]){14533,14533}
# elif 1 // 13/7 CRF
# define N1 4
# define SHIFT1 4
# define COEFFS1 (int[]){1,-9,-9,1}
2004-07-26 15:31:48 +02:00
# define N2 4
2004-07-26 01:36:44 +02:00
# define SHIFT2 4
# define COEFFS2 (int[]){-1,5,5,-1}
# define N3 0
# define SHIFT3 1
# define COEFFS3 NULL
# define N4 0
# define SHIFT4 1
# define COEFFS4 NULL
# elif 1 // 3/5
# define LX0 1
# define LX1 0
# define SCALEX 0.5
# define N1 2
# define SHIFT1 1
# define COEFFS1 (int[]){1,1}
# define N2 2
# define SHIFT2 2
# define COEFFS2 (int[]){-1,-1}
# define N3 0
# define SHIFT3 0
# define COEFFS3 NULL
# define N4 0
# define SHIFT4 0
# define COEFFS4 NULL
2005-12-17 19:14:38 +01:00
# elif 1 // 11/5
2004-07-26 01:36:44 +02:00
# define N1 0
# define SHIFT1 1
# define COEFFS1 NULL
# define N2 2
# define SHIFT2 2
# define COEFFS2 (int[]){-1,-1}
# define N3 2
# define SHIFT3 0
# define COEFFS3 (int[]){-1,-1}
# define N4 4
# define SHIFT4 7
# define COEFFS4 (int[]){-5,29,29,-5}
# define SCALEX 4
# elif 1 // 9/7 CDF
# define N1 2
# define SHIFT1 7
# define COEFFS1 (int[]){-203,-203}
# define N2 2
# define SHIFT2 12
# define COEFFS2 (int[]){-217,-217}
# define N3 2
# define SHIFT3 7
# define COEFFS3 (int[]){113,113}
# define N4 2
# define SHIFT4 9
# define COEFFS4 (int[]){227,227}
# define SCALEX 1
# elif 1 // 7/5 CDF
# define N1 0
# define SHIFT1 1
# define COEFFS1 NULL
# define N2 2
# define SHIFT2 2
# define COEFFS2 (int[]){-1,-1}
# define N3 2
# define SHIFT3 0
# define COEFFS3 (int[]){-1,-1}
# define N4 2
# define SHIFT4 4
# define COEFFS4 (int[]){3,3}
# elif 1 // 9/7 MN
# define N1 4
# define SHIFT1 4
# define COEFFS1 (int[]){1,-9,-9,1}
# define N2 2
# define SHIFT2 2
# define COEFFS2 (int[]){1,1}
# define N3 0
# define SHIFT3 1
# define COEFFS3 NULL
# define N4 0
# define SHIFT4 1
# define COEFFS4 NULL
# else // 13/7 CRF
# define N1 4
# define SHIFT1 4
# define COEFFS1 (int[]){1,-9,-9,1}
# define N2 4
# define SHIFT2 4
# define COEFFS2 (int[]){-1,5,5,-1}
# define N3 0
# define SHIFT3 1
# define COEFFS3 NULL
# define N4 0
# define SHIFT4 1
# define COEFFS4 NULL
# endif
2004-09-19 16:43:09 +02:00
static void horizontal_decomposeX ( DWTELEM * b , int width ) {
DWTELEM temp [ width ] ;
2004-07-26 01:36:44 +02:00
const int width2 = width > > 1 ;
const int w2 = ( width + 1 ) > > 1 ;
2005-09-22 01:09:16 +02:00
int x ;
2004-07-26 01:36:44 +02:00
inplace_lift ( b , width , COEFFS1 , N1 , SHIFT1 , LX1 , 0 ) ;
inplace_lift ( b , width , COEFFS2 , N2 , SHIFT2 , LX0 , 0 ) ;
inplace_lift ( b , width , COEFFS3 , N3 , SHIFT3 , LX1 , 0 ) ;
inplace_lift ( b , width , COEFFS4 , N4 , SHIFT4 , LX0 , 0 ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( x = 0 ; x < width2 ; x + + ) {
temp [ x ] = b [ 2 * x ] ;
temp [ x + w2 ] = b [ 2 * x + 1 ] ;
}
if ( width & 1 )
temp [ x ] = b [ 2 * x ] ;
memcpy ( b , temp , width * sizeof ( int ) ) ;
}
2004-09-19 16:43:09 +02:00
static void horizontal_composeX ( DWTELEM * b , int width ) {
DWTELEM temp [ width ] ;
2004-07-26 01:36:44 +02:00
const int width2 = width > > 1 ;
2005-09-22 01:09:16 +02:00
int x ;
2004-07-26 01:36:44 +02:00
const int w2 = ( width + 1 ) > > 1 ;
memcpy ( temp , b , width * sizeof ( int ) ) ;
for ( x = 0 ; x < width2 ; x + + ) {
b [ 2 * x ] = temp [ x ] ;
b [ 2 * x + 1 ] = temp [ x + w2 ] ;
}
if ( width & 1 )
b [ 2 * x ] = temp [ x ] ;
inplace_lift ( b , width , COEFFS4 , N4 , SHIFT4 , LX0 , 1 ) ;
inplace_lift ( b , width , COEFFS3 , N3 , SHIFT3 , LX1 , 1 ) ;
inplace_lift ( b , width , COEFFS2 , N2 , SHIFT2 , LX0 , 1 ) ;
inplace_lift ( b , width , COEFFS1 , N1 , SHIFT1 , LX1 , 1 ) ;
}
2004-09-19 16:43:09 +02:00
static void spatial_decomposeX ( DWTELEM * buffer , int width , int height , int stride ) {
2004-07-26 01:36:44 +02:00
int x , y ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < height ; y + + ) {
for ( x = 0 ; x < width ; x + + ) {
buffer [ y * stride + x ] * = SCALEX ;
}
}
for ( y = 0 ; y < height ; y + + ) {
horizontal_decomposeX ( buffer + y * stride , width ) ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
inplace_liftV ( buffer , width , height , stride , COEFFS1 , N1 , SHIFT1 , LX1 , 0 ) ;
inplace_liftV ( buffer , width , height , stride , COEFFS2 , N2 , SHIFT2 , LX0 , 0 ) ;
inplace_liftV ( buffer , width , height , stride , COEFFS3 , N3 , SHIFT3 , LX1 , 0 ) ;
2005-12-17 19:14:38 +01:00
inplace_liftV ( buffer , width , height , stride , COEFFS4 , N4 , SHIFT4 , LX0 , 0 ) ;
2004-07-26 01:36:44 +02:00
}
2004-09-19 16:43:09 +02:00
static void spatial_composeX ( DWTELEM * buffer , int width , int height , int stride ) {
2004-07-26 01:36:44 +02:00
int x , y ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
inplace_liftV ( buffer , width , height , stride , COEFFS4 , N4 , SHIFT4 , LX0 , 1 ) ;
inplace_liftV ( buffer , width , height , stride , COEFFS3 , N3 , SHIFT3 , LX1 , 1 ) ;
inplace_liftV ( buffer , width , height , stride , COEFFS2 , N2 , SHIFT2 , LX0 , 1 ) ;
inplace_liftV ( buffer , width , height , stride , COEFFS1 , N1 , SHIFT1 , LX1 , 1 ) ;
for ( y = 0 ; y < height ; y + + ) {
horizontal_composeX ( buffer + y * stride , width ) ;
}
for ( y = 0 ; y < height ; y + + ) {
for ( x = 0 ; x < width ; x + + ) {
buffer [ y * stride + x ] / = SCALEX ;
}
}
}
2004-09-19 16:43:09 +02:00
static void horizontal_decompose53i ( DWTELEM * b , int width ) {
DWTELEM temp [ width ] ;
2004-07-26 01:36:44 +02:00
const int width2 = width > > 1 ;
2005-09-22 01:09:16 +02:00
int x ;
2004-07-26 01:36:44 +02:00
const int w2 = ( width + 1 ) > > 1 ;
for ( x = 0 ; x < width2 ; x + + ) {
temp [ x ] = b [ 2 * x ] ;
temp [ x + w2 ] = b [ 2 * x + 1 ] ;
}
if ( width & 1 )
temp [ x ] = b [ 2 * x ] ;
#if 0
2005-09-22 01:09:16 +02:00
{
int A1 , A2 , A3 , A4 ;
2004-07-26 01:36:44 +02:00
A2 = temp [ 1 ] ;
A4 = temp [ 0 ] ;
A1 = temp [ 0 + width2 ] ;
A1 - = ( A2 + A4 ) > > 1 ;
A4 + = ( A1 + 1 ) > > 1 ;
b [ 0 + width2 ] = A1 ;
b [ 0 ] = A4 ;
for ( x = 1 ; x + 1 < width2 ; x + = 2 ) {
A3 = temp [ x + width2 ] ;
A4 = temp [ x + 1 ] ;
A3 - = ( A2 + A4 ) > > 1 ;
A2 + = ( A1 + A3 + 2 ) > > 2 ;
b [ x + width2 ] = A3 ;
b [ x ] = A2 ;
A1 = temp [ x + 1 + width2 ] ;
A2 = temp [ x + 2 ] ;
A1 - = ( A2 + A4 ) > > 1 ;
A4 + = ( A1 + A3 + 2 ) > > 2 ;
b [ x + 1 + width2 ] = A1 ;
b [ x + 1 ] = A4 ;
}
A3 = temp [ width - 1 ] ;
A3 - = A2 ;
A2 + = ( A1 + A3 + 2 ) > > 2 ;
b [ width - 1 ] = A3 ;
b [ width2 - 1 ] = A2 ;
2005-09-22 01:09:16 +02:00
}
2005-12-17 19:14:38 +01:00
# else
2004-07-26 01:36:44 +02:00
lift ( b + w2 , temp + w2 , temp , 1 , 1 , 1 , width , - 1 , 0 , 1 , 1 , 0 ) ;
lift ( b , temp , b + w2 , 1 , 1 , 1 , width , 1 , 2 , 2 , 0 , 0 ) ;
# endif
}
2004-09-19 16:43:09 +02:00
static void vertical_decompose53iH0 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
b1 [ i ] - = ( b0 [ i ] + b2 [ i ] ) > > 1 ;
}
}
2004-09-19 16:43:09 +02:00
static void vertical_decompose53iL0 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
b1 [ i ] + = ( b0 [ i ] + b2 [ i ] + 2 ) > > 2 ;
}
}
2004-09-19 16:43:09 +02:00
static void spatial_decompose53i ( DWTELEM * buffer , int width , int height , int stride ) {
2004-08-28 16:24:48 +02:00
int y ;
2004-07-26 01:36:44 +02:00
DWTELEM * b0 = buffer + mirror ( - 2 - 1 , height - 1 ) * stride ;
DWTELEM * b1 = buffer + mirror ( - 2 , height - 1 ) * stride ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( y = - 2 ; y < height ; y + = 2 ) {
DWTELEM * b2 = buffer + mirror ( y + 1 , height - 1 ) * stride ;
DWTELEM * b3 = buffer + mirror ( y + 2 , height - 1 ) * stride ;
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y + 1 < ( unsigned ) height ) horizontal_decompose53i ( b2 , width ) ;
if ( y + 2 < ( unsigned ) height ) horizontal_decompose53i ( b3 , width ) ;
2004-07-26 01:36:44 +02:00
STOP_TIMER ( " horizontal_decompose53i " ) }
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y + 1 < ( unsigned ) height ) vertical_decompose53iH0 ( b1 , b2 , b3 , width ) ;
if ( y + 0 < ( unsigned ) height ) vertical_decompose53iL0 ( b0 , b1 , b2 , width ) ;
2004-07-26 01:36:44 +02:00
STOP_TIMER ( " vertical_decompose53i* " ) }
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
b0 = b2 ;
b1 = b3 ;
}
}
2005-04-10 16:21:53 +02:00
# define liftS lift
2004-07-26 01:36:44 +02:00
# define lift5 lift
# if 1
# define W_AM 3
# define W_AO 0
# define W_AS 1
2005-04-10 16:21:53 +02:00
# undef liftS
2004-07-26 01:36:44 +02:00
# define W_BM 1
# define W_BO 8
# define W_BS 4
2005-04-10 16:21:53 +02:00
# define W_CM 1
# define W_CO 0
# define W_CS 0
2004-07-26 01:36:44 +02:00
2005-04-10 16:21:53 +02:00
# define W_DM 3
# define W_DO 4
# define W_DS 3
2004-07-26 01:36:44 +02:00
# elif 0
# define W_AM 55
# define W_AO 16
# define W_AS 5
# define W_BM 3
# define W_BO 32
# define W_BS 6
# define W_CM 127
# define W_CO 64
# define W_CS 7
# define W_DM 7
# define W_DO 8
# define W_DS 4
# elif 0
# define W_AM 97
# define W_AO 32
# define W_AS 6
# define W_BM 63
# define W_BO 512
# define W_BS 10
# define W_CM 13
# define W_CO 8
# define W_CS 4
# define W_DM 15
# define W_DO 16
# define W_DS 5
# else
# define W_AM 203
# define W_AO 64
# define W_AS 7
# define W_BM 217
# define W_BO 2048
# define W_BS 12
# define W_CM 113
# define W_CO 64
# define W_CS 7
# define W_DM 227
# define W_DO 128
# define W_DS 9
# endif
2004-09-19 16:43:09 +02:00
static void horizontal_decompose97i ( DWTELEM * b , int width ) {
DWTELEM temp [ width ] ;
2004-07-26 01:36:44 +02:00
const int w2 = ( width + 1 ) > > 1 ;
lift ( temp + w2 , b + 1 , b , 1 , 2 , 2 , width , - W_AM , W_AO , W_AS , 1 , 0 ) ;
2005-04-10 16:21:53 +02:00
liftS ( temp , b , temp + w2 , 1 , 2 , 1 , width , - W_BM , W_BO , W_BS , 0 , 0 ) ;
2004-07-26 01:36:44 +02:00
lift5 ( b + w2 , temp + w2 , temp , 1 , 1 , 1 , width , W_CM , W_CO , W_CS , 1 , 0 ) ;
lift ( b , temp , b + w2 , 1 , 1 , 1 , width , W_DM , W_DO , W_DS , 0 , 0 ) ;
}
2004-09-19 16:43:09 +02:00
static void vertical_decompose97iH0 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
b1 [ i ] - = ( W_AM * ( b0 [ i ] + b2 [ i ] ) + W_AO ) > > W_AS ;
}
}
2004-09-19 16:43:09 +02:00
static void vertical_decompose97iH1 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
# ifdef lift5
b1 [ i ] + = ( W_CM * ( b0 [ i ] + b2 [ i ] ) + W_CO ) > > W_CS ;
# else
int r = 3 * ( b0 [ i ] + b2 [ i ] ) ;
r + = r > > 4 ;
r + = r > > 8 ;
b1 [ i ] + = ( r + W_CO ) > > W_CS ;
# endif
}
}
2004-09-19 16:43:09 +02:00
static void vertical_decompose97iL0 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
2005-04-10 16:21:53 +02:00
# ifdef liftS
2004-07-26 01:36:44 +02:00
b1 [ i ] - = ( W_BM * ( b0 [ i ] + b2 [ i ] ) + W_BO ) > > W_BS ;
2005-04-10 16:21:53 +02:00
# else
b1 [ i ] = ( 16 * 4 * b1 [ i ] - 4 * ( b0 [ i ] + b2 [ i ] ) + 8 * 5 + ( 5 < < 27 ) ) / ( 5 * 16 ) - ( 1 < < 23 ) ;
# endif
2004-07-26 01:36:44 +02:00
}
}
2004-09-19 16:43:09 +02:00
static void vertical_decompose97iL1 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
b1 [ i ] + = ( W_DM * ( b0 [ i ] + b2 [ i ] ) + W_DO ) > > W_DS ;
}
}
2004-09-19 16:43:09 +02:00
static void spatial_decompose97i ( DWTELEM * buffer , int width , int height , int stride ) {
2004-08-28 16:24:48 +02:00
int y ;
2004-07-26 01:36:44 +02:00
DWTELEM * b0 = buffer + mirror ( - 4 - 1 , height - 1 ) * stride ;
DWTELEM * b1 = buffer + mirror ( - 4 , height - 1 ) * stride ;
DWTELEM * b2 = buffer + mirror ( - 4 + 1 , height - 1 ) * stride ;
DWTELEM * b3 = buffer + mirror ( - 4 + 2 , height - 1 ) * stride ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( y = - 4 ; y < height ; y + = 2 ) {
DWTELEM * b4 = buffer + mirror ( y + 3 , height - 1 ) * stride ;
DWTELEM * b5 = buffer + mirror ( y + 4 , height - 1 ) * stride ;
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y + 3 < ( unsigned ) height ) horizontal_decompose97i ( b4 , width ) ;
if ( y + 4 < ( unsigned ) height ) horizontal_decompose97i ( b5 , width ) ;
2004-07-26 01:36:44 +02:00
if ( width > 400 ) {
STOP_TIMER ( " horizontal_decompose97i " )
} }
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y + 3 < ( unsigned ) height ) vertical_decompose97iH0 ( b3 , b4 , b5 , width ) ;
if ( y + 2 < ( unsigned ) height ) vertical_decompose97iL0 ( b2 , b3 , b4 , width ) ;
if ( y + 1 < ( unsigned ) height ) vertical_decompose97iH1 ( b1 , b2 , b3 , width ) ;
if ( y + 0 < ( unsigned ) height ) vertical_decompose97iL1 ( b0 , b1 , b2 , width ) ;
2004-07-26 01:36:44 +02:00
if ( width > 400 ) {
STOP_TIMER ( " vertical_decompose97i " )
} }
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
b0 = b2 ;
b1 = b3 ;
b2 = b4 ;
b3 = b5 ;
}
}
2004-09-19 16:43:09 +02:00
void ff_spatial_dwt ( DWTELEM * buffer , int width , int height , int stride , int type , int decomposition_count ) {
2004-07-26 01:36:44 +02:00
int level ;
2005-12-17 19:14:38 +01:00
2004-08-03 04:34:50 +02:00
for ( level = 0 ; level < decomposition_count ; level + + ) {
switch ( type ) {
2004-07-26 01:36:44 +02:00
case 0 : spatial_decompose97i ( buffer , width > > level , height > > level , stride < < level ) ; break ;
case 1 : spatial_decompose53i ( buffer , width > > level , height > > level , stride < < level ) ; break ;
case 2 : spatial_decomposeX ( buffer , width > > level , height > > level , stride < < level ) ; break ;
}
}
}
2004-09-19 16:43:09 +02:00
static void horizontal_compose53i ( DWTELEM * b , int width ) {
DWTELEM temp [ width ] ;
2004-07-26 01:36:44 +02:00
const int width2 = width > > 1 ;
const int w2 = ( width + 1 ) > > 1 ;
2005-09-22 01:09:16 +02:00
int x ;
2004-07-26 01:36:44 +02:00
#if 0
2005-09-22 01:09:16 +02:00
int A1 , A2 , A3 , A4 ;
2004-07-26 01:36:44 +02:00
A2 = temp [ 1 ] ;
A4 = temp [ 0 ] ;
A1 = temp [ 0 + width2 ] ;
A1 - = ( A2 + A4 ) > > 1 ;
A4 + = ( A1 + 1 ) > > 1 ;
b [ 0 + width2 ] = A1 ;
b [ 0 ] = A4 ;
for ( x = 1 ; x + 1 < width2 ; x + = 2 ) {
A3 = temp [ x + width2 ] ;
A4 = temp [ x + 1 ] ;
A3 - = ( A2 + A4 ) > > 1 ;
A2 + = ( A1 + A3 + 2 ) > > 2 ;
b [ x + width2 ] = A3 ;
b [ x ] = A2 ;
A1 = temp [ x + 1 + width2 ] ;
A2 = temp [ x + 2 ] ;
A1 - = ( A2 + A4 ) > > 1 ;
A4 + = ( A1 + A3 + 2 ) > > 2 ;
b [ x + 1 + width2 ] = A1 ;
b [ x + 1 ] = A4 ;
}
A3 = temp [ width - 1 ] ;
A3 - = A2 ;
A2 + = ( A1 + A3 + 2 ) > > 2 ;
b [ width - 1 ] = A3 ;
b [ width2 - 1 ] = A2 ;
2005-12-17 19:14:38 +01:00
# else
2004-07-26 01:36:44 +02:00
lift ( temp , b , b + w2 , 1 , 1 , 1 , width , 1 , 2 , 2 , 0 , 1 ) ;
lift ( temp + w2 , b + w2 , temp , 1 , 1 , 1 , width , - 1 , 0 , 1 , 1 , 1 ) ;
# endif
for ( x = 0 ; x < width2 ; x + + ) {
b [ 2 * x ] = temp [ x ] ;
b [ 2 * x + 1 ] = temp [ x + w2 ] ;
}
if ( width & 1 )
b [ 2 * x ] = temp [ x ] ;
}
2004-09-19 16:43:09 +02:00
static void vertical_compose53iH0 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
b1 [ i ] + = ( b0 [ i ] + b2 [ i ] ) > > 1 ;
}
}
2004-09-19 16:43:09 +02:00
static void vertical_compose53iL0 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
b1 [ i ] - = ( b0 [ i ] + b2 [ i ] + 2 ) > > 2 ;
}
}
2005-04-03 17:43:57 +02:00
static void spatial_compose53i_buffered_init ( dwt_compose_t * cs , slice_buffer * sb , int height , int stride_line ) {
cs - > b0 = slice_buffer_get_line ( sb , mirror ( - 1 - 1 , height - 1 ) * stride_line ) ;
cs - > b1 = slice_buffer_get_line ( sb , mirror ( - 1 , height - 1 ) * stride_line ) ;
cs - > y = - 1 ;
}
2005-03-15 22:05:34 +01:00
static void spatial_compose53i_init ( dwt_compose_t * cs , DWTELEM * buffer , int height , int stride ) {
cs - > b0 = buffer + mirror ( - 1 - 1 , height - 1 ) * stride ;
cs - > b1 = buffer + mirror ( - 1 , height - 1 ) * stride ;
cs - > y = - 1 ;
}
2005-04-03 17:43:57 +02:00
static void spatial_compose53i_dy_buffered ( dwt_compose_t * cs , slice_buffer * sb , int width , int height , int stride_line ) {
int y = cs - > y ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
DWTELEM * b0 = cs - > b0 ;
DWTELEM * b1 = cs - > b1 ;
2006-01-04 23:34:11 +01:00
DWTELEM * b2 = slice_buffer_get_line ( sb , mirror ( y + 1 , height - 1 ) * stride_line ) ;
DWTELEM * b3 = slice_buffer_get_line ( sb , mirror ( y + 2 , height - 1 ) * stride_line ) ;
2005-04-03 17:43:57 +02:00
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y + 1 < ( unsigned ) height ) vertical_compose53iL0 ( b1 , b2 , b3 , width ) ;
if ( y + 0 < ( unsigned ) height ) vertical_compose53iH0 ( b0 , b1 , b2 , width ) ;
2005-04-03 17:43:57 +02:00
STOP_TIMER ( " vertical_compose53i* " ) }
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y - 1 < ( unsigned ) height ) horizontal_compose53i ( b0 , width ) ;
if ( y + 0 < ( unsigned ) height ) horizontal_compose53i ( b1 , width ) ;
2005-04-03 17:43:57 +02:00
STOP_TIMER ( " horizontal_compose53i " ) }
cs - > b0 = b2 ;
cs - > b1 = b3 ;
cs - > y + = 2 ;
}
2005-03-15 22:05:34 +01:00
static void spatial_compose53i_dy ( dwt_compose_t * cs , DWTELEM * buffer , int width , int height , int stride ) {
int y = cs - > y ;
DWTELEM * b0 = cs - > b0 ;
DWTELEM * b1 = cs - > b1 ;
DWTELEM * b2 = buffer + mirror ( y + 1 , height - 1 ) * stride ;
DWTELEM * b3 = buffer + mirror ( y + 2 , height - 1 ) * stride ;
2004-07-26 01:36:44 +02:00
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y + 1 < ( unsigned ) height ) vertical_compose53iL0 ( b1 , b2 , b3 , width ) ;
if ( y + 0 < ( unsigned ) height ) vertical_compose53iH0 ( b0 , b1 , b2 , width ) ;
2004-07-26 01:36:44 +02:00
STOP_TIMER ( " vertical_compose53i* " ) }
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y - 1 < ( unsigned ) height ) horizontal_compose53i ( b0 , width ) ;
if ( y + 0 < ( unsigned ) height ) horizontal_compose53i ( b1 , width ) ;
2004-07-26 01:36:44 +02:00
STOP_TIMER ( " horizontal_compose53i " ) }
2005-03-15 22:05:34 +01:00
cs - > b0 = b2 ;
cs - > b1 = b3 ;
cs - > y + = 2 ;
}
static void spatial_compose53i ( DWTELEM * buffer , int width , int height , int stride ) {
dwt_compose_t cs ;
spatial_compose53i_init ( & cs , buffer , height , stride ) ;
while ( cs . y < = height )
spatial_compose53i_dy ( & cs , buffer , width , height , stride ) ;
2005-12-17 19:14:38 +01:00
}
2004-07-26 01:36:44 +02:00
2004-09-19 16:43:09 +02:00
static void horizontal_compose97i ( DWTELEM * b , int width ) {
DWTELEM temp [ width ] ;
2004-07-26 01:36:44 +02:00
const int w2 = ( width + 1 ) > > 1 ;
lift ( temp , b , b + w2 , 1 , 1 , 1 , width , W_DM , W_DO , W_DS , 0 , 1 ) ;
lift5 ( temp + w2 , b + w2 , temp , 1 , 1 , 1 , width , W_CM , W_CO , W_CS , 1 , 1 ) ;
2005-04-10 16:21:53 +02:00
liftS ( b , temp , temp + w2 , 2 , 1 , 1 , width , - W_BM , W_BO , W_BS , 0 , 1 ) ;
2004-07-26 01:36:44 +02:00
lift ( b + 1 , temp + w2 , b , 2 , 1 , 2 , width , - W_AM , W_AO , W_AS , 1 , 1 ) ;
}
2004-09-19 16:43:09 +02:00
static void vertical_compose97iH0 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
b1 [ i ] + = ( W_AM * ( b0 [ i ] + b2 [ i ] ) + W_AO ) > > W_AS ;
}
}
2004-09-19 16:43:09 +02:00
static void vertical_compose97iH1 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
# ifdef lift5
b1 [ i ] - = ( W_CM * ( b0 [ i ] + b2 [ i ] ) + W_CO ) > > W_CS ;
# else
int r = 3 * ( b0 [ i ] + b2 [ i ] ) ;
r + = r > > 4 ;
r + = r > > 8 ;
b1 [ i ] - = ( r + W_CO ) > > W_CS ;
# endif
}
}
2004-09-19 16:43:09 +02:00
static void vertical_compose97iL0 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
2005-04-10 16:21:53 +02:00
# ifdef liftS
2004-07-26 01:36:44 +02:00
b1 [ i ] + = ( W_BM * ( b0 [ i ] + b2 [ i ] ) + W_BO ) > > W_BS ;
2005-04-10 16:21:53 +02:00
# else
b1 [ i ] + = ( W_BM * ( b0 [ i ] + b2 [ i ] ) + 4 * b1 [ i ] + W_BO ) > > W_BS ;
# endif
2004-07-26 01:36:44 +02:00
}
}
2004-09-19 16:43:09 +02:00
static void vertical_compose97iL1 ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , int width ) {
2004-07-26 01:36:44 +02:00
int i ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width ; i + + ) {
b1 [ i ] - = ( W_DM * ( b0 [ i ] + b2 [ i ] ) + W_DO ) > > W_DS ;
}
}
2005-04-04 23:35:18 +02:00
static void vertical_compose97i ( DWTELEM * b0 , DWTELEM * b1 , DWTELEM * b2 , DWTELEM * b3 , DWTELEM * b4 , DWTELEM * b5 , int width ) {
int i ;
2005-12-17 19:14:38 +01:00
2005-04-04 23:35:18 +02:00
for ( i = 0 ; i < width ; i + + ) {
2005-09-22 01:09:16 +02:00
# ifndef lift5
2005-04-04 23:35:18 +02:00
int r ;
2005-09-22 01:09:16 +02:00
# endif
2005-04-04 23:35:18 +02:00
b4 [ i ] - = ( W_DM * ( b3 [ i ] + b5 [ i ] ) + W_DO ) > > W_DS ;
# ifdef lift5
b3 [ i ] - = ( W_CM * ( b2 [ i ] + b4 [ i ] ) + W_CO ) > > W_CS ;
# else
r = 3 * ( b2 [ i ] + b4 [ i ] ) ;
r + = r > > 4 ;
r + = r > > 8 ;
b3 [ i ] - = ( r + W_CO ) > > W_CS ;
# endif
2005-04-10 16:21:53 +02:00
# ifdef liftS
2005-04-04 23:35:18 +02:00
b2 [ i ] + = ( W_BM * ( b1 [ i ] + b3 [ i ] ) + W_BO ) > > W_BS ;
2005-04-10 16:21:53 +02:00
# else
b2 [ i ] + = ( W_BM * ( b1 [ i ] + b3 [ i ] ) + 4 * b2 [ i ] + W_BO ) > > W_BS ;
# endif
2005-04-04 23:35:18 +02:00
b1 [ i ] + = ( W_AM * ( b0 [ i ] + b2 [ i ] ) + W_AO ) > > W_AS ;
}
}
2005-04-03 17:43:57 +02:00
static void spatial_compose97i_buffered_init ( dwt_compose_t * cs , slice_buffer * sb , int height , int stride_line ) {
cs - > b0 = slice_buffer_get_line ( sb , mirror ( - 3 - 1 , height - 1 ) * stride_line ) ;
cs - > b1 = slice_buffer_get_line ( sb , mirror ( - 3 , height - 1 ) * stride_line ) ;
cs - > b2 = slice_buffer_get_line ( sb , mirror ( - 3 + 1 , height - 1 ) * stride_line ) ;
cs - > b3 = slice_buffer_get_line ( sb , mirror ( - 3 + 2 , height - 1 ) * stride_line ) ;
cs - > y = - 3 ;
}
2005-03-15 22:05:34 +01:00
static void spatial_compose97i_init ( dwt_compose_t * cs , DWTELEM * buffer , int height , int stride ) {
cs - > b0 = buffer + mirror ( - 3 - 1 , height - 1 ) * stride ;
cs - > b1 = buffer + mirror ( - 3 , height - 1 ) * stride ;
cs - > b2 = buffer + mirror ( - 3 + 1 , height - 1 ) * stride ;
cs - > b3 = buffer + mirror ( - 3 + 2 , height - 1 ) * stride ;
cs - > y = - 3 ;
}
2004-07-26 01:36:44 +02:00
2005-04-03 17:43:57 +02:00
static void spatial_compose97i_dy_buffered ( dwt_compose_t * cs , slice_buffer * sb , int width , int height , int stride_line ) {
int y = cs - > y ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
DWTELEM * b0 = cs - > b0 ;
DWTELEM * b1 = cs - > b1 ;
DWTELEM * b2 = cs - > b2 ;
DWTELEM * b3 = cs - > b3 ;
2006-01-04 23:34:11 +01:00
DWTELEM * b4 = slice_buffer_get_line ( sb , mirror ( y + 3 , height - 1 ) * stride_line ) ;
DWTELEM * b5 = slice_buffer_get_line ( sb , mirror ( y + 4 , height - 1 ) * stride_line ) ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
{ START_TIMER
2005-04-04 23:35:18 +02:00
if ( y > 0 & & y + 4 < height ) {
vertical_compose97i ( b0 , b1 , b2 , b3 , b4 , b5 , width ) ;
} else {
2005-12-30 16:12:02 +01:00
if ( y + 3 < ( unsigned ) height ) vertical_compose97iL1 ( b3 , b4 , b5 , width ) ;
if ( y + 2 < ( unsigned ) height ) vertical_compose97iH1 ( b2 , b3 , b4 , width ) ;
if ( y + 1 < ( unsigned ) height ) vertical_compose97iL0 ( b1 , b2 , b3 , width ) ;
if ( y + 0 < ( unsigned ) height ) vertical_compose97iH0 ( b0 , b1 , b2 , width ) ;
2005-04-04 23:35:18 +02:00
}
2005-04-03 17:43:57 +02:00
if ( width > 400 ) {
STOP_TIMER ( " vertical_compose97i " ) } }
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y - 1 < ( unsigned ) height ) horizontal_compose97i ( b0 , width ) ;
if ( y + 0 < ( unsigned ) height ) horizontal_compose97i ( b1 , width ) ;
2006-01-04 23:34:11 +01:00
if ( width > 400 & & y + 0 < ( unsigned ) height ) {
2005-04-03 17:43:57 +02:00
STOP_TIMER ( " horizontal_compose97i " ) } }
cs - > b0 = b2 ;
cs - > b1 = b3 ;
cs - > b2 = b4 ;
cs - > b3 = b5 ;
cs - > y + = 2 ;
}
2005-03-15 22:05:34 +01:00
static void spatial_compose97i_dy ( dwt_compose_t * cs , DWTELEM * buffer , int width , int height , int stride ) {
int y = cs - > y ;
DWTELEM * b0 = cs - > b0 ;
DWTELEM * b1 = cs - > b1 ;
DWTELEM * b2 = cs - > b2 ;
DWTELEM * b3 = cs - > b3 ;
DWTELEM * b4 = buffer + mirror ( y + 3 , height - 1 ) * stride ;
DWTELEM * b5 = buffer + mirror ( y + 4 , height - 1 ) * stride ;
2004-07-26 01:36:44 +02:00
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y + 3 < ( unsigned ) height ) vertical_compose97iL1 ( b3 , b4 , b5 , width ) ;
if ( y + 2 < ( unsigned ) height ) vertical_compose97iH1 ( b2 , b3 , b4 , width ) ;
if ( y + 1 < ( unsigned ) height ) vertical_compose97iL0 ( b1 , b2 , b3 , width ) ;
if ( y + 0 < ( unsigned ) height ) vertical_compose97iH0 ( b0 , b1 , b2 , width ) ;
2004-07-26 01:36:44 +02:00
if ( width > 400 ) {
STOP_TIMER ( " vertical_compose97i " ) } }
{ START_TIMER
2005-12-30 16:12:02 +01:00
if ( y - 1 < ( unsigned ) height ) horizontal_compose97i ( b0 , width ) ;
if ( y + 0 < ( unsigned ) height ) horizontal_compose97i ( b1 , width ) ;
2004-07-26 01:36:44 +02:00
if ( width > 400 & & b0 < = b2 ) {
STOP_TIMER ( " horizontal_compose97i " ) } }
2005-03-15 22:05:34 +01:00
cs - > b0 = b2 ;
cs - > b1 = b3 ;
cs - > b2 = b4 ;
cs - > b3 = b5 ;
cs - > y + = 2 ;
}
static void spatial_compose97i ( DWTELEM * buffer , int width , int height , int stride ) {
dwt_compose_t cs ;
spatial_compose97i_init ( & cs , buffer , height , stride ) ;
while ( cs . y < = height )
spatial_compose97i_dy ( & cs , buffer , width , height , stride ) ;
}
2006-01-31 00:33:19 +01:00
static void ff_spatial_idwt_buffered_init ( dwt_compose_t * cs , slice_buffer * sb , int width , int height , int stride_line , int type , int decomposition_count ) {
2005-04-03 17:43:57 +02:00
int level ;
for ( level = decomposition_count - 1 ; level > = 0 ; level - - ) {
switch ( type ) {
case 0 : spatial_compose97i_buffered_init ( cs + level , sb , height > > level , stride_line < < level ) ; break ;
case 1 : spatial_compose53i_buffered_init ( cs + level , sb , height > > level , stride_line < < level ) ; break ;
/* not slicified yet */
case 2 : /*spatial_composeX(buffer, width>>level, height>>level, stride<<level); break;*/
av_log ( NULL , AV_LOG_ERROR , " spatial_composeX neither buffered nor slicified yet. \n " ) ; break ;
}
}
}
2006-01-31 00:33:19 +01:00
static void ff_spatial_idwt_init ( dwt_compose_t * cs , DWTELEM * buffer , int width , int height , int stride , int type , int decomposition_count ) {
2005-03-15 22:05:34 +01:00
int level ;
for ( level = decomposition_count - 1 ; level > = 0 ; level - - ) {
switch ( type ) {
case 0 : spatial_compose97i_init ( cs + level , buffer , height > > level , stride < < level ) ; break ;
case 1 : spatial_compose53i_init ( cs + level , buffer , height > > level , stride < < level ) ; break ;
/* not slicified yet */
case 2 : spatial_composeX ( buffer , width > > level , height > > level , stride < < level ) ; break ;
}
2004-07-26 01:36:44 +02:00
}
}
2006-01-31 00:33:19 +01:00
static void ff_spatial_idwt_slice ( dwt_compose_t * cs , DWTELEM * buffer , int width , int height , int stride , int type , int decomposition_count , int y ) {
2005-03-15 22:05:34 +01:00
const int support = type = = 1 ? 3 : 5 ;
2004-07-26 01:36:44 +02:00
int level ;
2005-03-15 22:05:34 +01:00
if ( type = = 2 ) return ;
2004-07-26 01:36:44 +02:00
2004-08-03 04:34:50 +02:00
for ( level = decomposition_count - 1 ; level > = 0 ; level - - ) {
2005-03-15 22:05:34 +01:00
while ( cs [ level ] . y < = FFMIN ( ( y > > level ) + support , height > > level ) ) {
switch ( type ) {
case 0 : spatial_compose97i_dy ( cs + level , buffer , width > > level , height > > level , stride < < level ) ;
break ;
case 1 : spatial_compose53i_dy ( cs + level , buffer , width > > level , height > > level , stride < < level ) ;
break ;
case 2 : break ;
}
2004-07-26 01:36:44 +02:00
}
}
}
2006-01-31 00:33:19 +01:00
static void ff_spatial_idwt_buffered_slice ( dwt_compose_t * cs , slice_buffer * slice_buf , int width , int height , int stride_line , int type , int decomposition_count , int y ) {
2005-04-03 17:43:57 +02:00
const int support = type = = 1 ? 3 : 5 ;
int level ;
if ( type = = 2 ) return ;
for ( level = decomposition_count - 1 ; level > = 0 ; level - - ) {
while ( cs [ level ] . y < = FFMIN ( ( y > > level ) + support , height > > level ) ) {
switch ( type ) {
case 0 : spatial_compose97i_dy_buffered ( cs + level , slice_buf , width > > level , height > > level , stride_line < < level ) ;
break ;
case 1 : spatial_compose53i_dy_buffered ( cs + level , slice_buf , width > > level , height > > level , stride_line < < level ) ;
break ;
case 2 : break ;
}
}
}
}
2006-01-31 00:33:19 +01:00
static void ff_spatial_idwt ( DWTELEM * buffer , int width , int height , int stride , int type , int decomposition_count ) {
2005-03-15 22:05:34 +01:00
if ( type = = 2 ) {
int level ;
for ( level = decomposition_count - 1 ; level > = 0 ; level - - )
spatial_composeX ( buffer , width > > level , height > > level , stride < < level ) ;
} else {
dwt_compose_t cs [ MAX_DECOMPOSITIONS ] ;
int y ;
ff_spatial_idwt_init ( cs , buffer , width , height , stride , type , decomposition_count ) ;
for ( y = 0 ; y < height ; y + = 4 )
ff_spatial_idwt_slice ( cs , buffer , width , height , stride , type , decomposition_count , y ) ;
}
}
2005-01-12 01:16:25 +01:00
static int encode_subband_c0run ( SnowContext * s , SubBand * b , DWTELEM * src , DWTELEM * parent , int stride , int orientation ) {
2004-08-02 03:03:52 +02:00
const int w = b - > width ;
const int h = b - > height ;
int x , y ;
2004-07-26 01:36:44 +02:00
if ( 1 ) {
int run = 0 ;
2004-07-27 19:02:31 +02:00
int runs [ w * h ] ;
2004-07-26 01:36:44 +02:00
int run_index = 0 ;
2005-04-15 15:24:30 +02:00
int max_index ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
2004-07-27 15:40:16 +02:00
int v , p = 0 ;
2004-07-27 03:30:44 +02:00
int /*ll=0, */ l = 0 , lt = 0 , t = 0 , rt = 0 ;
2004-07-27 19:02:31 +02:00
v = src [ x + y * stride ] ;
2004-07-26 01:36:44 +02:00
if ( y ) {
2004-07-27 19:02:31 +02:00
t = src [ x + ( y - 1 ) * stride ] ;
2004-07-26 01:36:44 +02:00
if ( x ) {
2004-07-27 19:02:31 +02:00
lt = src [ x - 1 + ( y - 1 ) * stride ] ;
2004-07-26 01:36:44 +02:00
}
if ( x + 1 < w ) {
2004-07-27 19:02:31 +02:00
rt = src [ x + 1 + ( y - 1 ) * stride ] ;
2004-07-26 01:36:44 +02:00
}
}
if ( x ) {
2004-07-27 19:02:31 +02:00
l = src [ x - 1 + y * stride ] ;
2004-07-27 03:30:44 +02:00
/*if(x > 1){
if ( orientation = = 1 ) ll = src [ y + ( x - 2 ) * stride ] ;
else ll = src [ x - 2 + y * stride ] ;
2004-07-26 01:36:44 +02:00
} */
}
2004-07-27 15:40:16 +02:00
if ( parent ) {
2004-07-27 19:02:31 +02:00
int px = x > > 1 ;
int py = y > > 1 ;
2005-12-17 19:14:38 +01:00
if ( px < b - > parent - > width & & py < b - > parent - > height )
2004-07-27 15:40:16 +02:00
p = parent [ px + py * 2 * stride ] ;
}
if ( ! ( /*ll|*/ l | lt | t | rt | p ) ) {
2004-07-26 01:36:44 +02:00
if ( v ) {
runs [ run_index + + ] = run ;
run = 0 ;
} else {
run + + ;
}
}
}
}
2005-04-15 15:24:30 +02:00
max_index = run_index ;
2004-07-26 01:36:44 +02:00
runs [ run_index + + ] = run ;
run_index = 0 ;
run = runs [ run_index + + ] ;
2005-04-15 15:24:30 +02:00
put_symbol2 ( & s - > c , b - > state [ 30 ] , max_index , 0 ) ;
if ( run_index < = max_index )
put_symbol2 ( & s - > c , b - > state [ 1 ] , run , 3 ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < h ; y + + ) {
2005-01-16 23:11:47 +01:00
if ( s - > c . bytestream_end - s - > c . bytestream < w * 40 ) {
2005-01-12 01:16:25 +01:00
av_log ( s - > avctx , AV_LOG_ERROR , " encoded frame too large \n " ) ;
return - 1 ;
}
2004-07-26 01:36:44 +02:00
for ( x = 0 ; x < w ; x + + ) {
2004-07-27 15:40:16 +02:00
int v , p = 0 ;
2004-07-27 03:30:44 +02:00
int /*ll=0, */ l = 0 , lt = 0 , t = 0 , rt = 0 ;
2004-07-27 19:02:31 +02:00
v = src [ x + y * stride ] ;
2004-07-26 01:36:44 +02:00
if ( y ) {
2004-07-27 19:02:31 +02:00
t = src [ x + ( y - 1 ) * stride ] ;
2004-07-26 01:36:44 +02:00
if ( x ) {
2004-07-27 19:02:31 +02:00
lt = src [ x - 1 + ( y - 1 ) * stride ] ;
2004-07-26 01:36:44 +02:00
}
if ( x + 1 < w ) {
2004-07-27 19:02:31 +02:00
rt = src [ x + 1 + ( y - 1 ) * stride ] ;
2004-07-26 01:36:44 +02:00
}
}
if ( x ) {
2004-07-27 19:02:31 +02:00
l = src [ x - 1 + y * stride ] ;
2004-07-27 03:30:44 +02:00
/*if(x > 1){
if ( orientation = = 1 ) ll = src [ y + ( x - 2 ) * stride ] ;
else ll = src [ x - 2 + y * stride ] ;
2004-07-26 01:36:44 +02:00
} */
}
2004-07-27 15:40:16 +02:00
if ( parent ) {
2004-07-27 19:02:31 +02:00
int px = x > > 1 ;
int py = y > > 1 ;
2005-12-17 19:14:38 +01:00
if ( px < b - > parent - > width & & py < b - > parent - > height )
2004-07-27 15:40:16 +02:00
p = parent [ px + py * 2 * stride ] ;
}
if ( /*ll|*/ l | lt | t | rt | p ) {
int context = av_log2 ( /*ABS(ll) + */ 3 * ABS ( l ) + ABS ( lt ) + 2 * ABS ( t ) + ABS ( rt ) + ABS ( p ) ) ;
2004-07-27 03:30:44 +02:00
2004-10-30 04:16:27 +02:00
put_rac ( & s - > c , & b - > state [ 0 ] [ context ] , ! ! v ) ;
2004-07-26 01:36:44 +02:00
} else {
if ( ! run ) {
run = runs [ run_index + + ] ;
2004-08-02 03:03:52 +02:00
2005-04-15 15:24:30 +02:00
if ( run_index < = max_index )
put_symbol2 ( & s - > c , b - > state [ 1 ] , run , 3 ) ;
2004-07-26 01:36:44 +02:00
assert ( v ) ;
} else {
run - - ;
assert ( ! v ) ;
}
}
if ( v ) {
2004-07-27 15:40:16 +02:00
int context = av_log2 ( /*ABS(ll) + */ 3 * ABS ( l ) + ABS ( lt ) + 2 * ABS ( t ) + ABS ( rt ) + ABS ( p ) ) ;
2005-04-05 19:59:05 +02:00
int l2 = 2 * ABS ( l ) + ( l < 0 ) ;
int t2 = 2 * ABS ( t ) + ( t < 0 ) ;
2004-07-27 03:30:44 +02:00
2004-08-02 19:53:57 +02:00
put_symbol2 ( & s - > c , b - > state [ context + 2 ] , ABS ( v ) - 1 , context - 4 ) ;
2005-04-05 19:59:05 +02:00
put_rac ( & s - > c , & b - > state [ 0 ] [ 16 + 1 + 3 + quant3bA [ l2 & 0xFF ] + 3 * quant3bA [ t2 & 0xFF ] ] , v < 0 ) ;
2004-07-26 01:36:44 +02:00
}
}
}
}
2005-01-12 01:16:25 +01:00
return 0 ;
2004-07-26 01:36:44 +02:00
}
2005-12-17 19:14:38 +01:00
static int encode_subband ( SnowContext * s , SubBand * b , DWTELEM * src , DWTELEM * parent , int stride , int orientation ) {
2004-08-02 03:03:52 +02:00
// encode_subband_qtree(s, b, src, parent, stride, orientation);
// encode_subband_z0run(s, b, src, parent, stride, orientation);
2005-01-12 01:16:25 +01:00
return encode_subband_c0run ( s , b , src , parent , stride , orientation ) ;
2004-08-02 03:03:52 +02:00
// encode_subband_dzr(s, b, src, parent, stride, orientation);
}
2005-04-03 17:43:57 +02:00
static inline void unpack_coeffs ( SnowContext * s , SubBand * b , SubBand * parent , int orientation ) {
2004-07-26 01:36:44 +02:00
const int w = b - > width ;
const int h = b - > height ;
int x , y ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
if ( 1 ) {
2005-04-15 15:24:30 +02:00
int run , runs ;
2005-04-14 22:54:03 +02:00
x_and_coeff * xc = b - > x_coeff ;
x_and_coeff * prev_xc = NULL ;
x_and_coeff * prev2_xc = xc ;
x_and_coeff * parent_xc = parent ? parent - > x_coeff : NULL ;
x_and_coeff * prev_parent_xc = parent_xc ;
2004-07-26 01:36:44 +02:00
2005-04-15 15:24:30 +02:00
runs = get_symbol2 ( & s - > c , b - > state [ 30 ] , 0 ) ;
if ( runs - - > 0 ) run = get_symbol2 ( & s - > c , b - > state [ 1 ] , 3 ) ;
else run = INT_MAX ;
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < h ; y + + ) {
2004-08-27 22:15:32 +02:00
int v = 0 ;
int lt = 0 , t = 0 , rt = 0 ;
2005-04-14 22:54:03 +02:00
if ( y & & prev_xc - > x = = 0 ) {
rt = prev_xc - > coeff ;
2004-08-27 22:15:32 +02:00
}
2004-07-26 01:36:44 +02:00
for ( x = 0 ; x < w ; x + + ) {
2004-08-27 22:15:32 +02:00
int p = 0 ;
const int l = v ;
2005-12-17 19:14:38 +01:00
2004-08-27 22:15:32 +02:00
lt = t ; t = rt ;
2004-07-26 01:36:44 +02:00
2004-08-27 22:33:16 +02:00
if ( y ) {
2005-04-14 22:54:03 +02:00
if ( prev_xc - > x < = x )
prev_xc + + ;
if ( prev_xc - > x = = x + 1 )
rt = prev_xc - > coeff ;
2004-08-27 22:33:16 +02:00
else
rt = 0 ;
}
2005-04-14 22:54:03 +02:00
if ( parent_xc ) {
if ( x > > 1 > parent_xc - > x ) {
parent_xc + + ;
2004-08-27 21:52:32 +02:00
}
2005-04-14 22:54:03 +02:00
if ( x > > 1 = = parent_xc - > x ) {
p = parent_xc - > coeff ;
2004-08-27 22:33:16 +02:00
}
2004-07-27 15:40:16 +02:00
}
if ( /*ll|*/ l | lt | t | rt | p ) {
2005-04-05 19:59:05 +02:00
int context = av_log2 ( /*ABS(ll) + */ 3 * ( l > > 1 ) + ( lt > > 1 ) + ( t & ~ 1 ) + ( rt > > 1 ) + ( p > > 1 ) ) ;
2004-07-27 03:30:44 +02:00
2004-10-30 04:16:27 +02:00
v = get_rac ( & s - > c , & b - > state [ 0 ] [ context ] ) ;
2005-04-14 20:33:26 +02:00
if ( v ) {
v = 2 * ( get_symbol2 ( & s - > c , b - > state [ context + 2 ] , context - 4 ) + 1 ) ;
v + = get_rac ( & s - > c , & b - > state [ 0 ] [ 16 + 1 + 3 + quant3bA [ l & 0xFF ] + 3 * quant3bA [ t & 0xFF ] ] ) ;
2005-12-17 19:14:38 +01:00
2005-04-14 22:54:03 +02:00
xc - > x = x ;
( xc + + ) - > coeff = v ;
2005-04-14 20:33:26 +02:00
}
2004-07-26 01:36:44 +02:00
} else {
if ( ! run ) {
2005-04-15 15:24:30 +02:00
if ( runs - - > 0 ) run = get_symbol2 ( & s - > c , b - > state [ 1 ] , 3 ) ;
else run = INT_MAX ;
2005-04-14 20:33:26 +02:00
v = 2 * ( get_symbol2 ( & s - > c , b - > state [ 0 + 2 ] , 0 - 4 ) + 1 ) ;
v + = get_rac ( & s - > c , & b - > state [ 0 ] [ 16 + 1 + 3 ] ) ;
2005-12-17 19:14:38 +01:00
2005-04-14 22:54:03 +02:00
xc - > x = x ;
( xc + + ) - > coeff = v ;
2004-07-26 01:36:44 +02:00
} else {
2005-04-14 22:30:08 +02:00
int max_run ;
2004-07-26 01:36:44 +02:00
run - - ;
v = 0 ;
2004-08-27 03:06:58 +02:00
2005-04-14 22:54:03 +02:00
if ( y ) max_run = FFMIN ( run , prev_xc - > x - x - 2 ) ;
2005-04-14 22:30:08 +02:00
else max_run = FFMIN ( run , w - x - 1 ) ;
2005-04-14 22:54:03 +02:00
if ( parent_xc )
max_run = FFMIN ( max_run , 2 * parent_xc - > x - x - 1 ) ;
2005-04-14 22:30:08 +02:00
x + = max_run ;
run - = max_run ;
2004-07-26 01:36:44 +02:00
}
}
2004-08-27 21:52:32 +02:00
}
2005-04-14 22:54:03 +02:00
( xc + + ) - > x = w + 1 ; //end marker
prev_xc = prev2_xc ;
prev2_xc = xc ;
2005-12-17 19:14:38 +01:00
2005-04-14 22:54:03 +02:00
if ( parent_xc ) {
2004-08-27 21:52:32 +02:00
if ( y & 1 ) {
2005-04-14 22:54:03 +02:00
while ( parent_xc - > x ! = parent - > width + 1 )
parent_xc + + ;
parent_xc + + ;
prev_parent_xc = parent_xc ;
2004-08-27 21:52:32 +02:00
} else {
2005-04-14 22:54:03 +02:00
parent_xc = prev_parent_xc ;
2004-07-26 01:36:44 +02:00
}
}
}
2005-04-03 17:43:57 +02:00
2005-04-14 22:54:03 +02:00
( xc + + ) - > x = w + 1 ; //end marker
2005-04-03 17:43:57 +02:00
}
}
static inline void decode_subband_slice_buffered ( SnowContext * s , SubBand * b , slice_buffer * sb , int start_y , int h , int save_state [ 1 ] ) {
const int w = b - > width ;
2005-09-22 01:09:16 +02:00
int y ;
2005-04-09 21:48:59 +02:00
const int qlog = clip ( s - > qlog + b - > qlog , 0 , QROOT * 16 ) ;
int qmul = qexp [ qlog & ( QROOT - 1 ) ] < < ( qlog > > QSHIFT ) ;
2005-04-03 17:43:57 +02:00
int qadd = ( s - > qbias * qmul ) > > QBIAS_SHIFT ;
int new_index = 0 ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
START_TIMER
if ( b - > buf = = s - > spatial_dwt_buffer | | s - > qlog = = LOSSLESS_QLOG ) {
qadd = 0 ;
qmul = 1 < < QEXPSHIFT ;
}
/* If we are on the second or later slice, restore our index. */
if ( start_y ! = 0 )
new_index = save_state [ 0 ] ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
for ( y = start_y ; y < h ; y + + ) {
int x = 0 ;
int v ;
DWTELEM * line = slice_buffer_get_line ( sb , y * b - > stride_line + b - > buf_y_offset ) + b - > buf_x_offset ;
memset ( line , 0 , b - > width * sizeof ( DWTELEM ) ) ;
v = b - > x_coeff [ new_index ] . coeff ;
x = b - > x_coeff [ new_index + + ] . x ;
while ( x < w )
{
2005-04-05 19:59:05 +02:00
register int t = ( ( v > > 1 ) * qmul + qadd ) > > QEXPSHIFT ;
register int u = - ( v & 1 ) ;
line [ x ] = ( t ^ u ) - u ;
2005-04-03 17:43:57 +02:00
v = b - > x_coeff [ new_index ] . coeff ;
x = b - > x_coeff [ new_index + + ] . x ;
}
2004-07-26 01:36:44 +02:00
}
2005-04-03 17:43:57 +02:00
if ( w > 200 & & start_y ! = 0 /*level+1 == s->spatial_decomposition_count*/ ) {
STOP_TIMER ( " decode_subband " )
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
/* Save our variables for the next slice. */
save_state [ 0 ] = new_index ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
return ;
2004-07-26 01:36:44 +02:00
}
static void reset_contexts ( SnowContext * s ) {
int plane_index , level , orientation ;
2004-08-31 18:28:49 +02:00
for ( plane_index = 0 ; plane_index < 3 ; plane_index + + ) {
2004-07-26 01:36:44 +02:00
for ( level = 0 ; level < s - > spatial_decomposition_count ; level + + ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
2004-10-30 04:16:27 +02:00
memset ( s - > plane [ plane_index ] . band [ level ] [ orientation ] . state , MID_STATE , sizeof ( s - > plane [ plane_index ] . band [ level ] [ orientation ] . state ) ) ;
2004-07-26 01:36:44 +02:00
}
}
}
2004-10-30 04:16:27 +02:00
memset ( s - > header_state , MID_STATE , sizeof ( s - > header_state ) ) ;
memset ( s - > block_state , MID_STATE , sizeof ( s - > block_state ) ) ;
2004-08-26 22:04:54 +02:00
}
static int alloc_blocks ( SnowContext * s ) {
int w = - ( ( - s - > avctx - > width ) > > LOG2_MB_SIZE ) ;
int h = - ( ( - s - > avctx - > height ) > > LOG2_MB_SIZE ) ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
s - > b_width = w ;
s - > b_height = h ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
s - > block = av_mallocz ( w * h * sizeof ( BlockNode ) < < ( s - > block_max_depth * 2 ) ) ;
return 0 ;
}
2004-10-30 04:16:27 +02:00
static inline void copy_rac_state ( RangeCoder * d , RangeCoder * s ) {
uint8_t * bytestream = d - > bytestream ;
uint8_t * bytestream_start = d - > bytestream_start ;
2004-08-26 22:04:54 +02:00
* d = * s ;
2004-10-30 04:16:27 +02:00
d - > bytestream = bytestream ;
d - > bytestream_start = bytestream_start ;
2004-08-26 22:04:54 +02:00
}
//near copy & paste from dsputil, FIXME
static int pix_sum ( uint8_t * pix , int line_size , int w )
{
int s , i , j ;
s = 0 ;
for ( i = 0 ; i < w ; i + + ) {
for ( j = 0 ; j < w ; j + + ) {
s + = pix [ 0 ] ;
pix + + ;
}
pix + = line_size - w ;
}
return s ;
}
//near copy & paste from dsputil, FIXME
static int pix_norm1 ( uint8_t * pix , int line_size , int w )
{
int s , i , j ;
uint32_t * sq = squareTbl + 256 ;
s = 0 ;
for ( i = 0 ; i < w ; i + + ) {
for ( j = 0 ; j < w ; j + + ) {
s + = sq [ pix [ 0 ] ] ;
pix + + ;
}
pix + = line_size - w ;
}
return s ;
}
static inline void set_blocks ( SnowContext * s , int level , int x , int y , int l , int cb , int cr , int mx , int my , int type ) {
const int w = s - > b_width < < s - > block_max_depth ;
const int rem_depth = s - > block_max_depth - level ;
const int index = ( x + y * w ) < < rem_depth ;
const int block_w = 1 < < rem_depth ;
BlockNode block ;
int i , j ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
block . color [ 0 ] = l ;
block . color [ 1 ] = cb ;
block . color [ 2 ] = cr ;
block . mx = mx ;
block . my = my ;
block . type = type ;
block . level = level ;
for ( j = 0 ; j < block_w ; j + + ) {
for ( i = 0 ; i < block_w ; i + + ) {
s - > block [ index + i + j * w ] = block ;
}
}
}
static inline void init_ref ( MotionEstContext * c , uint8_t * src [ 3 ] , uint8_t * ref [ 3 ] , uint8_t * ref2 [ 3 ] , int x , int y , int ref_index ) {
const int offset [ 3 ] = {
y * c - > stride + x ,
( ( y * c - > uvstride + x ) > > 1 ) ,
( ( y * c - > uvstride + x ) > > 1 ) ,
} ;
int i ;
for ( i = 0 ; i < 3 ; i + + ) {
c - > src [ 0 ] [ i ] = src [ i ] ;
c - > ref [ 0 ] [ i ] = ref [ i ] + offset [ i ] ;
}
assert ( ! ref_index ) ;
}
//FIXME copy&paste
# define P_LEFT P[1]
# define P_TOP P[2]
# define P_TOPRIGHT P[3]
# define P_MEDIAN P[4]
# define P_MV1 P[9]
# define FLAG_QPEL 1 //must be 1
static int encode_q_branch ( SnowContext * s , int level , int x , int y ) {
uint8_t p_buffer [ 1024 ] ;
uint8_t i_buffer [ 1024 ] ;
uint8_t p_state [ sizeof ( s - > block_state ) ] ;
uint8_t i_state [ sizeof ( s - > block_state ) ] ;
2004-10-30 04:16:27 +02:00
RangeCoder pc , ic ;
uint8_t * pbbak = s - > c . bytestream ;
uint8_t * pbbak_start = s - > c . bytestream_start ;
2004-08-26 22:04:54 +02:00
int score , score2 , iscore , i_len , p_len , block_s , sum ;
const int w = s - > b_width < < s - > block_max_depth ;
const int h = s - > b_height < < s - > block_max_depth ;
const int rem_depth = s - > block_max_depth - level ;
const int index = ( x + y * w ) < < rem_depth ;
const int block_w = 1 < < ( LOG2_MB_SIZE - level ) ;
int trx = ( x + 1 ) < < rem_depth ;
int try = ( y + 1 ) < < rem_depth ;
BlockNode * left = x ? & s - > block [ index - 1 ] : & null_block ;
BlockNode * top = y ? & s - > block [ index - w ] : & null_block ;
BlockNode * right = trx < w ? & s - > block [ index + 1 ] : & null_block ;
BlockNode * bottom = try < h ? & s - > block [ index + w ] : & null_block ;
BlockNode * tl = y & & x ? & s - > block [ index - w - 1 ] : left ;
BlockNode * tr = y & & trx < w & & ( ( x & 1 ) = = 0 | | level = = 0 ) ? & s - > block [ index - w + ( 1 < < rem_depth ) ] : tl ; //FIXME use lt
int pl = left - > color [ 0 ] ;
int pcb = left - > color [ 1 ] ;
int pcr = left - > color [ 2 ] ;
int pmx = mid_pred ( left - > mx , top - > mx , tr - > mx ) ;
int pmy = mid_pred ( left - > my , top - > my , tr - > my ) ;
int mx = 0 , my = 0 ;
2005-12-28 16:43:53 +01:00
int l , cr , cb ;
2004-08-26 22:04:54 +02:00
const int stride = s - > current_picture . linesize [ 0 ] ;
const int uvstride = s - > current_picture . linesize [ 1 ] ;
2005-12-28 16:43:53 +01:00
uint8_t * current_data [ 3 ] = { s - > input_picture . data [ 0 ] + ( x + y * stride ) * block_w ,
s - > input_picture . data [ 1 ] + ( x + y * uvstride ) * block_w / 2 ,
s - > input_picture . data [ 2 ] + ( x + y * uvstride ) * block_w / 2 } ;
2004-08-26 22:04:54 +02:00
int P [ 10 ] [ 2 ] ;
int16_t last_mv [ 3 ] [ 2 ] ;
int qpel = ! ! ( s - > avctx - > flags & CODEC_FLAG_QPEL ) ; //unused
const int shift = 1 + qpel ;
MotionEstContext * c = & s - > m . me ;
int mx_context = av_log2 ( 2 * ABS ( left - > mx - top - > mx ) ) ;
int my_context = av_log2 ( 2 * ABS ( left - > my - top - > my ) ) ;
int s_context = 2 * left - > level + 2 * top - > level + tl - > level + tr - > level ;
assert ( sizeof ( s - > block_state ) > = 256 ) ;
if ( s - > keyframe ) {
set_blocks ( s , level , x , y , pl , pcb , pcr , pmx , pmy , BLOCK_INTRA ) ;
return 0 ;
}
// clip predictors / edge ?
P_LEFT [ 0 ] = left - > mx ;
P_LEFT [ 1 ] = left - > my ;
P_TOP [ 0 ] = top - > mx ;
P_TOP [ 1 ] = top - > my ;
P_TOPRIGHT [ 0 ] = tr - > mx ;
P_TOPRIGHT [ 1 ] = tr - > my ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
last_mv [ 0 ] [ 0 ] = s - > block [ index ] . mx ;
last_mv [ 0 ] [ 1 ] = s - > block [ index ] . my ;
last_mv [ 1 ] [ 0 ] = right - > mx ;
last_mv [ 1 ] [ 1 ] = right - > my ;
last_mv [ 2 ] [ 0 ] = bottom - > mx ;
last_mv [ 2 ] [ 1 ] = bottom - > my ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
s - > m . mb_stride = 2 ;
2005-12-17 19:14:38 +01:00
s - > m . mb_x =
2004-08-26 22:04:54 +02:00
s - > m . mb_y = 0 ;
s - > m . me . skip = 0 ;
init_ref ( c , current_data , s - > last_picture . data , NULL , block_w * x , block_w * y , 0 ) ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
assert ( s - > m . me . stride = = stride ) ;
assert ( s - > m . me . uvstride = = uvstride ) ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
c - > penalty_factor = get_penalty_factor ( s - > lambda , s - > lambda2 , c - > avctx - > me_cmp ) ;
c - > sub_penalty_factor = get_penalty_factor ( s - > lambda , s - > lambda2 , c - > avctx - > me_sub_cmp ) ;
c - > mb_penalty_factor = get_penalty_factor ( s - > lambda , s - > lambda2 , c - > avctx - > mb_cmp ) ;
c - > current_mv_penalty = c - > mv_penalty [ s - > m . f_code = 1 ] + MAX_MV ;
2005-12-17 19:14:38 +01:00
2004-09-05 23:46:16 +02:00
c - > xmin = - x * block_w - 16 + 2 ;
c - > ymin = - y * block_w - 16 + 2 ;
c - > xmax = - ( x + 1 ) * block_w + ( w < < ( LOG2_MB_SIZE - s - > block_max_depth ) ) + 16 - 2 ;
c - > ymax = - ( y + 1 ) * block_w + ( h < < ( LOG2_MB_SIZE - s - > block_max_depth ) ) + 16 - 2 ;
2004-08-26 22:04:54 +02:00
if ( P_LEFT [ 0 ] > ( c - > xmax < < shift ) ) P_LEFT [ 0 ] = ( c - > xmax < < shift ) ;
2005-12-17 19:14:38 +01:00
if ( P_LEFT [ 1 ] > ( c - > ymax < < shift ) ) P_LEFT [ 1 ] = ( c - > ymax < < shift ) ;
2004-08-26 22:04:54 +02:00
if ( P_TOP [ 0 ] > ( c - > xmax < < shift ) ) P_TOP [ 0 ] = ( c - > xmax < < shift ) ;
if ( P_TOP [ 1 ] > ( c - > ymax < < shift ) ) P_TOP [ 1 ] = ( c - > ymax < < shift ) ;
if ( P_TOPRIGHT [ 0 ] < ( c - > xmin < < shift ) ) P_TOPRIGHT [ 0 ] = ( c - > xmin < < shift ) ;
if ( P_TOPRIGHT [ 0 ] > ( c - > xmax < < shift ) ) P_TOPRIGHT [ 0 ] = ( c - > xmax < < shift ) ; //due to pmx no clip
if ( P_TOPRIGHT [ 1 ] > ( c - > ymax < < shift ) ) P_TOPRIGHT [ 1 ] = ( c - > ymax < < shift ) ;
P_MEDIAN [ 0 ] = mid_pred ( P_LEFT [ 0 ] , P_TOP [ 0 ] , P_TOPRIGHT [ 0 ] ) ;
P_MEDIAN [ 1 ] = mid_pred ( P_LEFT [ 1 ] , P_TOP [ 1 ] , P_TOPRIGHT [ 1 ] ) ;
if ( ! y ) {
c - > pred_x = P_LEFT [ 0 ] ;
c - > pred_y = P_LEFT [ 1 ] ;
} else {
c - > pred_x = P_MEDIAN [ 0 ] ;
c - > pred_y = P_MEDIAN [ 1 ] ;
}
2005-12-17 19:14:38 +01:00
score = ff_epzs_motion_search ( & s - > m , & mx , & my , P , 0 , /*ref_index*/ 0 , last_mv ,
2004-08-26 22:04:54 +02:00
( 1 < < 16 ) > > shift , level - LOG2_MB_SIZE + 4 , block_w ) ;
assert ( mx > = c - > xmin ) ;
assert ( mx < = c - > xmax ) ;
assert ( my > = c - > ymin ) ;
assert ( my < = c - > ymax ) ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
score = s - > m . me . sub_motion_search ( & s - > m , & mx , & my , score , 0 , 0 , level - LOG2_MB_SIZE + 4 , block_w ) ;
score = ff_get_mb_score ( & s - > m , mx , my , 0 , 0 , level - LOG2_MB_SIZE + 4 , block_w , 0 ) ;
//FIXME if mb_cmp != SSE then intra cant be compared currently and mb_penalty vs. lambda2
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
// subpel search
pc = s - > c ;
2004-10-30 04:16:27 +02:00
pc . bytestream_start =
pc . bytestream = p_buffer ; //FIXME end/start? and at the other stoo
2004-08-26 22:04:54 +02:00
memcpy ( p_state , s - > block_state , sizeof ( s - > block_state ) ) ;
if ( level ! = s - > block_max_depth )
2004-10-30 04:16:27 +02:00
put_rac ( & pc , & p_state [ 4 + s_context ] , 1 ) ;
put_rac ( & pc , & p_state [ 1 + left - > type + top - > type ] , 0 ) ;
2004-08-26 22:04:54 +02:00
put_symbol ( & pc , & p_state [ 128 + 32 * mx_context ] , mx - pmx , 1 ) ;
put_symbol ( & pc , & p_state [ 128 + 32 * my_context ] , my - pmy , 1 ) ;
2004-10-30 04:16:27 +02:00
p_len = pc . bytestream - pc . bytestream_start ;
score + = ( s - > lambda2 * ( p_len * 8
+ ( pc . outstanding_count - s - > c . outstanding_count ) * 8
+ ( - av_log2 ( pc . range ) + av_log2 ( s - > c . range ) )
) ) > > FF_LAMBDA_SHIFT ;
2004-08-26 22:04:54 +02:00
block_s = block_w * block_w ;
2005-12-28 16:43:53 +01:00
sum = pix_sum ( current_data [ 0 ] , stride , block_w ) ;
2004-08-26 22:04:54 +02:00
l = ( sum + block_s / 2 ) / block_s ;
2005-12-28 16:43:53 +01:00
iscore = pix_norm1 ( current_data [ 0 ] , stride , block_w ) - 2 * l * sum + l * l * block_s ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
block_s = block_w * block_w > > 2 ;
2005-12-28 16:43:53 +01:00
sum = pix_sum ( current_data [ 1 ] , uvstride , block_w > > 1 ) ;
2004-08-26 22:04:54 +02:00
cb = ( sum + block_s / 2 ) / block_s ;
// iscore += pix_norm1(¤t_mb[1][0], uvstride, block_w>>1) - 2*cb*sum + cb*cb*block_s;
2005-12-28 16:43:53 +01:00
sum = pix_sum ( current_data [ 2 ] , uvstride , block_w > > 1 ) ;
2004-08-26 22:04:54 +02:00
cr = ( sum + block_s / 2 ) / block_s ;
// iscore += pix_norm1(¤t_mb[2][0], uvstride, block_w>>1) - 2*cr*sum + cr*cr*block_s;
ic = s - > c ;
2004-10-30 04:16:27 +02:00
ic . bytestream_start =
ic . bytestream = i_buffer ; //FIXME end/start? and at the other stoo
2004-08-26 22:04:54 +02:00
memcpy ( i_state , s - > block_state , sizeof ( s - > block_state ) ) ;
if ( level ! = s - > block_max_depth )
2004-10-30 04:16:27 +02:00
put_rac ( & ic , & i_state [ 4 + s_context ] , 1 ) ;
put_rac ( & ic , & i_state [ 1 + left - > type + top - > type ] , 1 ) ;
2004-08-26 22:04:54 +02:00
put_symbol ( & ic , & i_state [ 32 ] , l - pl , 1 ) ;
put_symbol ( & ic , & i_state [ 64 ] , cb - pcb , 1 ) ;
put_symbol ( & ic , & i_state [ 96 ] , cr - pcr , 1 ) ;
2004-10-30 04:16:27 +02:00
i_len = ic . bytestream - ic . bytestream_start ;
iscore + = ( s - > lambda2 * ( i_len * 8
+ ( ic . outstanding_count - s - > c . outstanding_count ) * 8
+ ( - av_log2 ( ic . range ) + av_log2 ( s - > c . range ) )
) ) > > FF_LAMBDA_SHIFT ;
2004-08-26 22:04:54 +02:00
// assert(score==256*256*256*64-1);
assert ( iscore < 255 * 255 * 256 + s - > lambda2 * 10 ) ;
assert ( iscore > = 0 ) ;
assert ( l > = 0 & & l < = 255 ) ;
assert ( pl > = 0 & & pl < = 255 ) ;
if ( level = = 0 ) {
int varc = iscore > > 8 ;
int vard = score > > 8 ;
if ( vard < = 64 | | vard < varc )
c - > scene_change_score + = ff_sqrt ( vard ) - ff_sqrt ( varc ) ;
else
c - > scene_change_score + = s - > m . qscale ;
}
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
if ( level ! = s - > block_max_depth ) {
2004-10-30 04:16:27 +02:00
put_rac ( & s - > c , & s - > block_state [ 4 + s_context ] , 0 ) ;
2004-08-26 22:04:54 +02:00
score2 = encode_q_branch ( s , level + 1 , 2 * x + 0 , 2 * y + 0 ) ;
score2 + = encode_q_branch ( s , level + 1 , 2 * x + 1 , 2 * y + 0 ) ;
score2 + = encode_q_branch ( s , level + 1 , 2 * x + 0 , 2 * y + 1 ) ;
score2 + = encode_q_branch ( s , level + 1 , 2 * x + 1 , 2 * y + 1 ) ;
score2 + = s - > lambda2 > > FF_LAMBDA_SHIFT ; //FIXME exact split overhead
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
if ( score2 < score & & score2 < iscore )
return score2 ;
}
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
if ( iscore < score ) {
2004-10-30 04:16:27 +02:00
memcpy ( pbbak , i_buffer , i_len ) ;
2004-08-26 22:04:54 +02:00
s - > c = ic ;
2004-10-30 04:16:27 +02:00
s - > c . bytestream_start = pbbak_start ;
s - > c . bytestream = pbbak + i_len ;
2004-08-26 22:04:54 +02:00
set_blocks ( s , level , x , y , l , cb , cr , pmx , pmy , BLOCK_INTRA ) ;
memcpy ( s - > block_state , i_state , sizeof ( s - > block_state ) ) ;
return iscore ;
} else {
2004-10-30 04:16:27 +02:00
memcpy ( pbbak , p_buffer , p_len ) ;
2004-08-26 22:04:54 +02:00
s - > c = pc ;
2004-10-30 04:16:27 +02:00
s - > c . bytestream_start = pbbak_start ;
s - > c . bytestream = pbbak + p_len ;
2004-08-26 22:04:54 +02:00
set_blocks ( s , level , x , y , pl , pcb , pcr , mx , my , 0 ) ;
memcpy ( s - > block_state , p_state , sizeof ( s - > block_state ) ) ;
return score ;
}
}
2005-12-28 16:43:53 +01:00
static always_inline int same_block ( BlockNode * a , BlockNode * b ) {
if ( ( a - > type & BLOCK_INTRA ) & & ( b - > type & BLOCK_INTRA ) ) {
return ! ( ( a - > color [ 0 ] - b - > color [ 0 ] ) | ( a - > color [ 1 ] - b - > color [ 1 ] ) | ( a - > color [ 2 ] - b - > color [ 2 ] ) ) ;
} else {
return ! ( ( a - > mx - b - > mx ) | ( a - > my - b - > my ) | ( ( a - > type ^ b - > type ) & BLOCK_INTRA ) ) ;
}
}
static void encode_q_branch2 ( SnowContext * s , int level , int x , int y ) {
const int w = s - > b_width < < s - > block_max_depth ;
const int rem_depth = s - > block_max_depth - level ;
const int index = ( x + y * w ) < < rem_depth ;
int trx = ( x + 1 ) < < rem_depth ;
BlockNode * b = & s - > block [ index ] ;
BlockNode * left = x ? & s - > block [ index - 1 ] : & null_block ;
BlockNode * top = y ? & s - > block [ index - w ] : & null_block ;
BlockNode * tl = y & & x ? & s - > block [ index - w - 1 ] : left ;
BlockNode * tr = y & & trx < w & & ( ( x & 1 ) = = 0 | | level = = 0 ) ? & s - > block [ index - w + ( 1 < < rem_depth ) ] : tl ; //FIXME use lt
int pl = left - > color [ 0 ] ;
int pcb = left - > color [ 1 ] ;
int pcr = left - > color [ 2 ] ;
int pmx = mid_pred ( left - > mx , top - > mx , tr - > mx ) ;
int pmy = mid_pred ( left - > my , top - > my , tr - > my ) ;
int mx_context = av_log2 ( 2 * ABS ( left - > mx - top - > mx ) ) ;
int my_context = av_log2 ( 2 * ABS ( left - > my - top - > my ) ) ;
int s_context = 2 * left - > level + 2 * top - > level + tl - > level + tr - > level ;
if ( s - > keyframe ) {
set_blocks ( s , level , x , y , pl , pcb , pcr , pmx , pmy , BLOCK_INTRA ) ;
return ;
}
if ( level ! = s - > block_max_depth ) {
if ( same_block ( b , b + 1 ) & & same_block ( b , b + w ) & & same_block ( b , b + w + 1 ) ) {
2005-12-29 03:17:51 +01:00
put_rac ( & s - > c , & s - > block_state [ 4 + s_context ] , 1 ) ;
} else {
2005-12-28 16:43:53 +01:00
put_rac ( & s - > c , & s - > block_state [ 4 + s_context ] , 0 ) ;
encode_q_branch2 ( s , level + 1 , 2 * x + 0 , 2 * y + 0 ) ;
encode_q_branch2 ( s , level + 1 , 2 * x + 1 , 2 * y + 0 ) ;
encode_q_branch2 ( s , level + 1 , 2 * x + 0 , 2 * y + 1 ) ;
encode_q_branch2 ( s , level + 1 , 2 * x + 1 , 2 * y + 1 ) ;
return ;
}
}
if ( b - > type & BLOCK_INTRA ) {
put_rac ( & s - > c , & s - > block_state [ 1 + ( left - > type & 1 ) + ( top - > type & 1 ) ] , 1 ) ;
put_symbol ( & s - > c , & s - > block_state [ 32 ] , b - > color [ 0 ] - pl , 1 ) ;
put_symbol ( & s - > c , & s - > block_state [ 64 ] , b - > color [ 1 ] - pcb , 1 ) ;
put_symbol ( & s - > c , & s - > block_state [ 96 ] , b - > color [ 2 ] - pcr , 1 ) ;
set_blocks ( s , level , x , y , b - > color [ 0 ] , b - > color [ 1 ] , b - > color [ 2 ] , pmx , pmy , BLOCK_INTRA ) ;
} else {
put_rac ( & s - > c , & s - > block_state [ 1 + ( left - > type & 1 ) + ( top - > type & 1 ) ] , 0 ) ;
put_symbol ( & s - > c , & s - > block_state [ 128 + 32 * mx_context ] , b - > mx - pmx , 1 ) ;
put_symbol ( & s - > c , & s - > block_state [ 128 + 32 * my_context ] , b - > my - pmy , 1 ) ;
set_blocks ( s , level , x , y , pl , pcb , pcr , b - > mx , b - > my , 0 ) ;
}
}
2004-08-26 22:04:54 +02:00
static void decode_q_branch ( SnowContext * s , int level , int x , int y ) {
const int w = s - > b_width < < s - > block_max_depth ;
const int rem_depth = s - > block_max_depth - level ;
const int index = ( x + y * w ) < < rem_depth ;
int trx = ( x + 1 ) < < rem_depth ;
BlockNode * left = x ? & s - > block [ index - 1 ] : & null_block ;
BlockNode * top = y ? & s - > block [ index - w ] : & null_block ;
BlockNode * tl = y & & x ? & s - > block [ index - w - 1 ] : left ;
BlockNode * tr = y & & trx < w & & ( ( x & 1 ) = = 0 | | level = = 0 ) ? & s - > block [ index - w + ( 1 < < rem_depth ) ] : tl ; //FIXME use lt
int s_context = 2 * left - > level + 2 * top - > level + tl - > level + tr - > level ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
if ( s - > keyframe ) {
set_blocks ( s , level , x , y , null_block . color [ 0 ] , null_block . color [ 1 ] , null_block . color [ 2 ] , null_block . mx , null_block . my , BLOCK_INTRA ) ;
return ;
}
2004-10-30 04:16:27 +02:00
if ( level = = s - > block_max_depth | | get_rac ( & s - > c , & s - > block_state [ 4 + s_context ] ) ) {
2004-08-26 22:04:54 +02:00
int type ;
int l = left - > color [ 0 ] ;
int cb = left - > color [ 1 ] ;
int cr = left - > color [ 2 ] ;
int mx = mid_pred ( left - > mx , top - > mx , tr - > mx ) ;
int my = mid_pred ( left - > my , top - > my , tr - > my ) ;
int mx_context = av_log2 ( 2 * ABS ( left - > mx - top - > mx ) ) + 0 * av_log2 ( 2 * ABS ( tr - > mx - top - > mx ) ) ;
int my_context = av_log2 ( 2 * ABS ( left - > my - top - > my ) ) + 0 * av_log2 ( 2 * ABS ( tr - > my - top - > my ) ) ;
2005-12-17 19:14:38 +01:00
2004-10-30 04:16:27 +02:00
type = get_rac ( & s - > c , & s - > block_state [ 1 + left - > type + top - > type ] ) ? BLOCK_INTRA : 0 ;
2004-08-26 22:04:54 +02:00
if ( type ) {
l + = get_symbol ( & s - > c , & s - > block_state [ 32 ] , 1 ) ;
cb + = get_symbol ( & s - > c , & s - > block_state [ 64 ] , 1 ) ;
cr + = get_symbol ( & s - > c , & s - > block_state [ 96 ] , 1 ) ;
} else {
mx + = get_symbol ( & s - > c , & s - > block_state [ 128 + 32 * mx_context ] , 1 ) ;
my + = get_symbol ( & s - > c , & s - > block_state [ 128 + 32 * my_context ] , 1 ) ;
}
set_blocks ( s , level , x , y , l , cb , cr , mx , my , type ) ;
} else {
decode_q_branch ( s , level + 1 , 2 * x + 0 , 2 * y + 0 ) ;
decode_q_branch ( s , level + 1 , 2 * x + 1 , 2 * y + 0 ) ;
decode_q_branch ( s , level + 1 , 2 * x + 0 , 2 * y + 1 ) ;
decode_q_branch ( s , level + 1 , 2 * x + 1 , 2 * y + 1 ) ;
}
}
static void encode_blocks ( SnowContext * s ) {
int x , y ;
int w = s - > b_width ;
int h = s - > b_height ;
2005-12-28 16:43:53 +01:00
if ( s - > avctx - > me_method = = ME_ITER & & ! s - > keyframe )
iterative_me ( s ) ;
2004-08-26 22:04:54 +02:00
for ( y = 0 ; y < h ; y + + ) {
2005-01-16 23:11:47 +01:00
if ( s - > c . bytestream_end - s - > c . bytestream < w * MB_SIZE * MB_SIZE * 3 ) { //FIXME nicer limit
2005-01-12 01:16:25 +01:00
av_log ( s - > avctx , AV_LOG_ERROR , " encoded frame too large \n " ) ;
return ;
}
2004-08-26 22:04:54 +02:00
for ( x = 0 ; x < w ; x + + ) {
2005-12-28 16:43:53 +01:00
if ( s - > avctx - > me_method = = ME_ITER )
encode_q_branch2 ( s , 0 , x , y ) ;
else
encode_q_branch ( s , 0 , x , y ) ;
2004-08-26 22:04:54 +02:00
}
}
}
static void decode_blocks ( SnowContext * s ) {
int x , y ;
int w = s - > b_width ;
int h = s - > b_height ;
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
decode_q_branch ( s , 0 , x , y ) ;
}
}
2004-07-26 01:36:44 +02:00
}
static void mc_block ( uint8_t * dst , uint8_t * src , uint8_t * tmp , int stride , int b_w , int b_h , int dx , int dy ) {
int x , y ;
2004-09-12 12:51:02 +02:00
START_TIMER
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < b_h + 5 ; y + + ) {
for ( x = 0 ; x < b_w ; x + + ) {
2004-09-12 12:51:02 +02:00
int a0 = src [ x ] ;
int a1 = src [ x + 1 ] ;
int a2 = src [ x + 2 ] ;
int a3 = src [ x + 3 ] ;
int a4 = src [ x + 4 ] ;
int a5 = src [ x + 5 ] ;
2004-07-26 01:36:44 +02:00
// int am= 9*(a1+a2) - (a0+a3);
int am = 20 * ( a2 + a3 ) - 5 * ( a1 + a4 ) + ( a0 + a5 ) ;
// int am= 18*(a2+a3) - 2*(a1+a4);
// int aL= (-7*a0 + 105*a1 + 35*a2 - 5*a3)>>3;
// int aR= (-7*a3 + 105*a2 + 35*a1 - 5*a0)>>3;
// if(b_w==16) am= 8*(a1+a2);
2005-04-17 11:27:53 +02:00
if ( dx < 8 ) am = ( 32 * a2 * ( 8 - dx ) + am * dx + 128 ) > > 8 ;
else am = ( am * ( 16 - dx ) + 32 * a3 * ( dx - 8 ) + 128 ) > > 8 ;
2005-12-17 19:14:38 +01:00
2005-04-17 11:27:53 +02:00
/* FIXME Try increasing tmp buffer to 16 bits and not clipping here. Should give marginally better results. - Robert*/
if ( am & ( ~ 255 ) ) am = ~ ( am > > 31 ) ;
2005-12-17 19:14:38 +01:00
2005-04-17 11:27:53 +02:00
tmp [ x ] = am ;
2004-07-26 01:36:44 +02:00
/* if (dx< 4) tmp[x + y*stride]= (16*a1*( 4-dx) + aL* dx + 32)>>6;
else if ( dx < 8 ) tmp [ x + y * stride ] = ( aL * ( 8 - dx ) + am * ( dx - 4 ) + 32 ) > > 6 ;
else if ( dx < 12 ) tmp [ x + y * stride ] = ( am * ( 12 - dx ) + aR * ( dx - 8 ) + 32 ) > > 6 ;
else tmp [ x + y * stride ] = ( aR * ( 16 - dx ) + 16 * a2 * ( dx - 12 ) + 32 ) > > 6 ; */
}
2004-09-12 12:51:02 +02:00
tmp + = stride ;
src + = stride ;
2004-07-26 01:36:44 +02:00
}
2004-09-12 12:51:02 +02:00
tmp - = ( b_h + 5 ) * stride ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < b_h ; y + + ) {
for ( x = 0 ; x < b_w ; x + + ) {
2004-09-12 12:51:02 +02:00
int a0 = tmp [ x + 0 * stride ] ;
int a1 = tmp [ x + 1 * stride ] ;
int a2 = tmp [ x + 2 * stride ] ;
int a3 = tmp [ x + 3 * stride ] ;
int a4 = tmp [ x + 4 * stride ] ;
int a5 = tmp [ x + 5 * stride ] ;
2004-07-26 01:36:44 +02:00
int am = 20 * ( a2 + a3 ) - 5 * ( a1 + a4 ) + ( a0 + a5 ) ;
// int am= 18*(a2+a3) - 2*(a1+a4);
/* int aL= (-7*a0 + 105*a1 + 35*a2 - 5*a3)>>3;
int aR = ( - 7 * a3 + 105 * a2 + 35 * a1 - 5 * a0 ) > > 3 ; */
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
// if(b_w==16) am= 8*(a1+a2);
2005-04-17 11:27:53 +02:00
if ( dy < 8 ) am = ( 32 * a2 * ( 8 - dy ) + am * dy + 128 ) > > 8 ;
else am = ( am * ( 16 - dy ) + 32 * a3 * ( dy - 8 ) + 128 ) > > 8 ;
2004-07-26 01:36:44 +02:00
2005-04-17 11:27:53 +02:00
if ( am & ( ~ 255 ) ) am = ~ ( am > > 31 ) ;
2005-12-17 19:14:38 +01:00
2005-04-17 11:27:53 +02:00
dst [ x ] = am ;
2004-07-26 01:36:44 +02:00
/* if (dy< 4) tmp[x + y*stride]= (16*a1*( 4-dy) + aL* dy + 32)>>6;
else if ( dy < 8 ) tmp [ x + y * stride ] = ( aL * ( 8 - dy ) + am * ( dy - 4 ) + 32 ) > > 6 ;
else if ( dy < 12 ) tmp [ x + y * stride ] = ( am * ( 12 - dy ) + aR * ( dy - 8 ) + 32 ) > > 6 ;
else tmp [ x + y * stride ] = ( aR * ( 16 - dy ) + 16 * a2 * ( dy - 12 ) + 32 ) > > 6 ; */
}
2004-09-12 12:51:02 +02:00
dst + = stride ;
tmp + = stride ;
2004-07-26 01:36:44 +02:00
}
2004-09-12 12:51:02 +02:00
STOP_TIMER ( " mc_block " )
2004-07-26 01:36:44 +02:00
}
# define mca(dx,dy,b_w)\
2004-09-13 01:09:54 +02:00
static void mc_block_hpel # # dx # # dy # # b_w ( uint8_t * dst , uint8_t * src , int stride , int h ) { \
2004-07-26 01:36:44 +02:00
uint8_t tmp [ stride * ( b_w + 5 ) ] ; \
assert ( h = = b_w ) ; \
mc_block ( dst , src - 2 - 2 * stride , tmp , stride , b_w , b_w , dx , dy ) ; \
}
mca ( 0 , 0 , 16 )
mca ( 8 , 0 , 16 )
mca ( 0 , 8 , 16 )
mca ( 8 , 8 , 16 )
2004-09-13 01:09:54 +02:00
mca ( 0 , 0 , 8 )
mca ( 8 , 0 , 8 )
mca ( 0 , 8 , 8 )
mca ( 8 , 8 , 8 )
2004-07-26 01:36:44 +02:00
2004-09-05 23:46:16 +02:00
static void pred_block ( SnowContext * s , uint8_t * dst , uint8_t * src , uint8_t * tmp , int stride , int sx , int sy , int b_w , int b_h , BlockNode * block , int plane_index , int w , int h ) {
2005-12-28 16:43:53 +01:00
if ( block - > type & BLOCK_INTRA ) {
2004-09-05 23:46:16 +02:00
int x , y ;
2006-01-07 19:34:19 +01:00
const int color = block - > color [ plane_index ] ;
const int color4 = color * 0x01010101 ;
2006-01-12 06:47:52 +01:00
if ( b_w = = 32 ) {
for ( y = 0 ; y < b_h ; y + + ) {
* ( uint32_t * ) & dst [ 0 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 4 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 8 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 12 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 16 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 20 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 24 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 28 + y * stride ] = color4 ;
}
} else if ( b_w = = 16 ) {
2006-01-07 19:34:19 +01:00
for ( y = 0 ; y < b_h ; y + + ) {
* ( uint32_t * ) & dst [ 0 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 4 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 8 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 12 + y * stride ] = color4 ;
}
} else if ( b_w = = 8 ) {
for ( y = 0 ; y < b_h ; y + + ) {
* ( uint32_t * ) & dst [ 0 + y * stride ] = color4 ;
* ( uint32_t * ) & dst [ 4 + y * stride ] = color4 ;
}
} else if ( b_w = = 4 ) {
for ( y = 0 ; y < b_h ; y + + ) {
* ( uint32_t * ) & dst [ 0 + y * stride ] = color4 ;
}
} else {
for ( y = 0 ; y < b_h ; y + + ) {
for ( x = 0 ; x < b_w ; x + + ) {
dst [ x + y * stride ] = color ;
}
2004-09-05 23:46:16 +02:00
}
}
} else {
const int scale = plane_index ? s - > mv_scale : 2 * s - > mv_scale ;
int mx = block - > mx * scale ;
int my = block - > my * scale ;
2004-09-12 23:32:36 +02:00
const int dx = mx & 15 ;
const int dy = my & 15 ;
2006-01-08 02:50:34 +01:00
const int tab_index = 3 - ( b_w > > 2 ) + ( b_w > > 4 ) ;
2004-09-05 23:46:16 +02:00
sx + = ( mx > > 4 ) - 2 ;
sy + = ( my > > 4 ) - 2 ;
src + = sx + sy * stride ;
if ( ( unsigned ) sx > = w - b_w - 4
| | ( unsigned ) sy > = h - b_h - 4 ) {
ff_emulated_edge_mc ( tmp + MB_SIZE , src , stride , b_w + 5 , b_h + 5 , sx , sy , w , h ) ;
src = tmp + MB_SIZE ;
}
2006-03-11 01:18:43 +01:00
// assert(b_w == b_h || 2*b_w == b_h || b_w == 2*b_h);
// assert(!(b_w&(b_w-1)));
2006-01-07 19:34:19 +01:00
assert ( b_w > 1 & & b_h > 1 ) ;
2006-01-12 06:47:52 +01:00
assert ( tab_index > = 0 & & tab_index < 4 | | b_w = = 32 ) ;
2006-03-11 01:18:43 +01:00
if ( ( dx & 3 ) | | ( dy & 3 ) | | ! ( b_w = = b_h | | 2 * b_w = = b_h | | b_w = = 2 * b_h ) | | ( b_w & ( b_w - 1 ) ) )
2004-09-12 23:32:36 +02:00
mc_block ( dst , src , tmp , stride , b_w , b_h , dx , dy ) ;
2006-01-12 06:47:52 +01:00
else if ( b_w = = 32 ) {
int y ;
for ( y = 0 ; y < b_h ; y + = 16 ) {
s - > dsp . put_h264_qpel_pixels_tab [ 0 ] [ dy + ( dx > > 2 ) ] ( dst + y * stride , src + 2 + ( y + 2 ) * stride , stride ) ;
s - > dsp . put_h264_qpel_pixels_tab [ 0 ] [ dy + ( dx > > 2 ) ] ( dst + 16 + y * stride , src + 18 + ( y + 2 ) * stride , stride ) ;
}
} else if ( b_w = = b_h )
2006-01-08 02:50:34 +01:00
s - > dsp . put_h264_qpel_pixels_tab [ tab_index ] [ dy + ( dx > > 2 ) ] ( dst , src + 2 + 2 * stride , stride ) ;
2006-01-07 19:34:19 +01:00
else if ( b_w = = 2 * b_h ) {
2006-01-08 02:50:34 +01:00
s - > dsp . put_h264_qpel_pixels_tab [ tab_index + 1 ] [ dy + ( dx > > 2 ) ] ( dst , src + 2 + 2 * stride , stride ) ;
s - > dsp . put_h264_qpel_pixels_tab [ tab_index + 1 ] [ dy + ( dx > > 2 ) ] ( dst + b_h , src + 2 + b_h + 2 * stride , stride ) ;
2006-01-07 19:34:19 +01:00
} else {
assert ( 2 * b_w = = b_h ) ;
2006-01-08 02:50:34 +01:00
s - > dsp . put_h264_qpel_pixels_tab [ tab_index ] [ dy + ( dx > > 2 ) ] ( dst , src + 2 + 2 * stride , stride ) ;
s - > dsp . put_h264_qpel_pixels_tab [ tab_index ] [ dy + ( dx > > 2 ) ] ( dst + b_w * stride , src + 2 + 2 * stride + b_w * stride , stride ) ;
2006-01-07 19:34:19 +01:00
}
2004-09-05 23:46:16 +02:00
}
}
2005-04-03 17:43:57 +02:00
//FIXME name clenup (b_w, block_w, b_width stuff)
static always_inline void add_yblock_buffered ( SnowContext * s , slice_buffer * sb , DWTELEM * old_dst , uint8_t * dst8 , uint8_t * src , uint8_t * obmc , int src_x , int src_y , int b_w , int b_h , int w , int h , int dst_stride , int src_stride , int obmc_stride , int b_x , int b_y , int add , int plane_index ) {
DWTELEM * dst = NULL ;
const int b_width = s - > b_width < < s - > block_max_depth ;
const int b_height = s - > b_height < < s - > block_max_depth ;
const int b_stride = b_width ;
BlockNode * lt = & s - > block [ b_x + b_y * b_stride ] ;
BlockNode * rt = lt + 1 ;
BlockNode * lb = lt + b_stride ;
BlockNode * rb = lb + 1 ;
2005-12-17 19:14:38 +01:00
uint8_t * block [ 4 ] ;
2005-08-22 00:17:41 +02:00
int tmp_step = src_stride > = 7 * MB_SIZE ? MB_SIZE : MB_SIZE * src_stride ;
uint8_t tmp [ src_stride * 7 * MB_SIZE ] ; //FIXME align
uint8_t * ptmp ;
2005-04-03 17:43:57 +02:00
int x , y ;
if ( b_x < 0 ) {
lt = rt ;
lb = rb ;
} else if ( b_x + 1 > = b_width ) {
rt = lt ;
rb = lb ;
}
if ( b_y < 0 ) {
lt = lb ;
rt = rb ;
} else if ( b_y + 1 > = b_height ) {
lb = lt ;
rb = rt ;
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
if ( src_x < 0 ) { //FIXME merge with prev & always round internal width upto *16
obmc - = src_x ;
b_w + = src_x ;
src_x = 0 ;
} else if ( src_x + b_w > w ) {
b_w = w - src_x ;
}
if ( src_y < 0 ) {
obmc - = src_y * obmc_stride ;
b_h + = src_y ;
src_y = 0 ;
} else if ( src_y + b_h > h ) {
b_h = h - src_y ;
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
if ( b_w < = 0 | | b_h < = 0 ) return ;
2005-08-22 00:17:41 +02:00
assert ( src_stride > 2 * MB_SIZE + 5 ) ;
2005-04-03 17:43:57 +02:00
// old_dst += src_x + src_y*dst_stride;
dst8 + = src_x + src_y * src_stride ;
// src += src_x + src_y*src_stride;
2005-08-22 00:17:41 +02:00
ptmp = tmp + 3 * tmp_step ;
block [ 0 ] = ptmp ;
ptmp + = tmp_step ;
2005-12-17 19:14:38 +01:00
pred_block ( s , block [ 0 ] , src , tmp , src_stride , src_x , src_y , b_w , b_h , lt , plane_index , w , h ) ;
2005-04-03 17:43:57 +02:00
if ( same_block ( lt , rt ) ) {
block [ 1 ] = block [ 0 ] ;
} else {
2005-08-22 00:17:41 +02:00
block [ 1 ] = ptmp ;
ptmp + = tmp_step ;
2005-04-03 17:43:57 +02:00
pred_block ( s , block [ 1 ] , src , tmp , src_stride , src_x , src_y , b_w , b_h , rt , plane_index , w , h ) ;
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
if ( same_block ( lt , lb ) ) {
block [ 2 ] = block [ 0 ] ;
} else if ( same_block ( rt , lb ) ) {
block [ 2 ] = block [ 1 ] ;
} else {
2005-08-22 00:17:41 +02:00
block [ 2 ] = ptmp ;
ptmp + = tmp_step ;
2005-04-03 17:43:57 +02:00
pred_block ( s , block [ 2 ] , src , tmp , src_stride , src_x , src_y , b_w , b_h , lb , plane_index , w , h ) ;
}
if ( same_block ( lt , rb ) ) {
block [ 3 ] = block [ 0 ] ;
} else if ( same_block ( rt , rb ) ) {
block [ 3 ] = block [ 1 ] ;
} else if ( same_block ( lb , rb ) ) {
block [ 3 ] = block [ 2 ] ;
} else {
2005-08-22 00:17:41 +02:00
block [ 3 ] = ptmp ;
2005-04-03 17:43:57 +02:00
pred_block ( s , block [ 3 ] , src , tmp , src_stride , src_x , src_y , b_w , b_h , rb , plane_index , w , h ) ;
}
#if 0
for ( y = 0 ; y < b_h ; y + + ) {
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc [ x + y * obmc_stride ] * block [ 3 ] [ x + y * src_stride ] * ( 256 / OBMC_MAX ) ;
if ( add ) dst [ x + y * dst_stride ] + = v ;
else dst [ x + y * dst_stride ] - = v ;
}
}
for ( y = 0 ; y < b_h ; y + + ) {
uint8_t * obmc2 = obmc + ( obmc_stride > > 1 ) ;
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc2 [ x + y * obmc_stride ] * block [ 2 ] [ x + y * src_stride ] * ( 256 / OBMC_MAX ) ;
if ( add ) dst [ x + y * dst_stride ] + = v ;
else dst [ x + y * dst_stride ] - = v ;
}
}
for ( y = 0 ; y < b_h ; y + + ) {
uint8_t * obmc3 = obmc + obmc_stride * ( obmc_stride > > 1 ) ;
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc3 [ x + y * obmc_stride ] * block [ 1 ] [ x + y * src_stride ] * ( 256 / OBMC_MAX ) ;
if ( add ) dst [ x + y * dst_stride ] + = v ;
else dst [ x + y * dst_stride ] - = v ;
}
}
for ( y = 0 ; y < b_h ; y + + ) {
uint8_t * obmc3 = obmc + obmc_stride * ( obmc_stride > > 1 ) ;
uint8_t * obmc4 = obmc3 + ( obmc_stride > > 1 ) ;
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc4 [ x + y * obmc_stride ] * block [ 0 ] [ x + y * src_stride ] * ( 256 / OBMC_MAX ) ;
if ( add ) dst [ x + y * dst_stride ] + = v ;
else dst [ x + y * dst_stride ] - = v ;
}
}
# else
{
START_TIMER
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
for ( y = 0 ; y < b_h ; y + + ) {
//FIXME ugly missue of obmc_stride
uint8_t * obmc1 = obmc + y * obmc_stride ;
uint8_t * obmc2 = obmc1 + ( obmc_stride > > 1 ) ;
uint8_t * obmc3 = obmc1 + obmc_stride * ( obmc_stride > > 1 ) ;
uint8_t * obmc4 = obmc3 + ( obmc_stride > > 1 ) ;
dst = slice_buffer_get_line ( sb , src_y + y ) ;
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc1 [ x ] * block [ 3 ] [ x + y * src_stride ]
+ obmc2 [ x ] * block [ 2 ] [ x + y * src_stride ]
+ obmc3 [ x ] * block [ 1 ] [ x + y * src_stride ]
+ obmc4 [ x ] * block [ 0 ] [ x + y * src_stride ] ;
v < < = 8 - LOG2_OBMC_MAX ;
if ( FRAC_BITS ! = 8 ) {
v + = 1 < < ( 7 - FRAC_BITS ) ;
v > > = 8 - FRAC_BITS ;
}
if ( add ) {
// v += old_dst[x + y*dst_stride];
v + = dst [ x + src_x ] ;
v = ( v + ( 1 < < ( FRAC_BITS - 1 ) ) ) > > FRAC_BITS ;
if ( v & ( ~ 255 ) ) v = ~ ( v > > 31 ) ;
dst8 [ x + y * src_stride ] = v ;
} else {
// old_dst[x + y*dst_stride] -= v;
dst [ x + src_x ] - = v ;
}
}
}
STOP_TIMER ( " Inner add y block " )
}
# endif
}
2004-09-05 23:46:16 +02:00
//FIXME name clenup (b_w, block_w, b_width stuff)
2006-01-12 06:47:52 +01:00
static always_inline void add_yblock ( SnowContext * s , DWTELEM * dst , uint8_t * dst8 , uint8_t * src , uint8_t * obmc , int src_x , int src_y , int b_w , int b_h , int w , int h , int dst_stride , int src_stride , int obmc_stride , int b_x , int b_y , int add , int offset_dst , int plane_index ) {
2004-09-05 23:46:16 +02:00
const int b_width = s - > b_width < < s - > block_max_depth ;
const int b_height = s - > b_height < < s - > block_max_depth ;
const int b_stride = b_width ;
BlockNode * lt = & s - > block [ b_x + b_y * b_stride ] ;
BlockNode * rt = lt + 1 ;
BlockNode * lb = lt + b_stride ;
BlockNode * rb = lb + 1 ;
2005-12-17 19:14:38 +01:00
uint8_t * block [ 4 ] ;
2005-08-22 00:17:41 +02:00
int tmp_step = src_stride > = 7 * MB_SIZE ? MB_SIZE : MB_SIZE * src_stride ;
uint8_t tmp [ src_stride * 7 * MB_SIZE ] ; //FIXME align
uint8_t * ptmp ;
2004-07-26 01:36:44 +02:00
int x , y ;
2004-09-05 23:46:16 +02:00
if ( b_x < 0 ) {
lt = rt ;
lb = rb ;
} else if ( b_x + 1 > = b_width ) {
rt = lt ;
rb = lb ;
2004-07-26 01:36:44 +02:00
}
2004-09-05 23:46:16 +02:00
if ( b_y < 0 ) {
lt = lb ;
rt = rb ;
} else if ( b_y + 1 > = b_height ) {
lb = lt ;
rb = rt ;
}
2005-12-17 19:14:38 +01:00
2004-09-05 23:46:16 +02:00
if ( src_x < 0 ) { //FIXME merge with prev & always round internal width upto *16
obmc - = src_x ;
b_w + = src_x ;
2006-01-12 06:47:52 +01:00
if ( ! offset_dst )
dst - = src_x ;
2004-09-05 23:46:16 +02:00
src_x = 0 ;
} else if ( src_x + b_w > w ) {
b_w = w - src_x ;
}
if ( src_y < 0 ) {
obmc - = src_y * obmc_stride ;
b_h + = src_y ;
2006-01-12 06:47:52 +01:00
if ( ! offset_dst )
dst - = src_y * dst_stride ;
2004-09-05 23:46:16 +02:00
src_y = 0 ;
} else if ( src_y + b_h > h ) {
b_h = h - src_y ;
2004-07-26 01:36:44 +02:00
}
2005-12-17 19:14:38 +01:00
2004-09-05 23:46:16 +02:00
if ( b_w < = 0 | | b_h < = 0 ) return ;
2004-08-26 22:04:54 +02:00
2005-08-22 00:17:41 +02:00
assert ( src_stride > 2 * MB_SIZE + 5 ) ;
2006-01-12 06:47:52 +01:00
if ( offset_dst )
dst + = src_x + src_y * dst_stride ;
2004-09-23 14:20:41 +02:00
dst8 + = src_x + src_y * src_stride ;
2004-09-05 23:46:16 +02:00
// src += src_x + src_y*src_stride;
2005-08-22 00:17:41 +02:00
ptmp = tmp + 3 * tmp_step ;
block [ 0 ] = ptmp ;
ptmp + = tmp_step ;
2005-12-17 19:14:38 +01:00
pred_block ( s , block [ 0 ] , src , tmp , src_stride , src_x , src_y , b_w , b_h , lt , plane_index , w , h ) ;
2004-09-05 23:46:16 +02:00
if ( same_block ( lt , rt ) ) {
block [ 1 ] = block [ 0 ] ;
2004-07-26 01:36:44 +02:00
} else {
2005-08-22 00:17:41 +02:00
block [ 1 ] = ptmp ;
ptmp + = tmp_step ;
2004-09-05 23:46:16 +02:00
pred_block ( s , block [ 1 ] , src , tmp , src_stride , src_x , src_y , b_w , b_h , rt , plane_index , w , h ) ;
}
2005-12-17 19:14:38 +01:00
2004-09-05 23:46:16 +02:00
if ( same_block ( lt , lb ) ) {
block [ 2 ] = block [ 0 ] ;
} else if ( same_block ( rt , lb ) ) {
block [ 2 ] = block [ 1 ] ;
} else {
2005-08-22 00:17:41 +02:00
block [ 2 ] = ptmp ;
ptmp + = tmp_step ;
2004-09-05 23:46:16 +02:00
pred_block ( s , block [ 2 ] , src , tmp , src_stride , src_x , src_y , b_w , b_h , lb , plane_index , w , h ) ;
}
2004-07-26 01:36:44 +02:00
2004-09-05 23:46:16 +02:00
if ( same_block ( lt , rb ) ) {
block [ 3 ] = block [ 0 ] ;
} else if ( same_block ( rt , rb ) ) {
block [ 3 ] = block [ 1 ] ;
} else if ( same_block ( lb , rb ) ) {
block [ 3 ] = block [ 2 ] ;
} else {
2005-08-22 00:17:41 +02:00
block [ 3 ] = ptmp ;
2004-09-05 23:46:16 +02:00
pred_block ( s , block [ 3 ] , src , tmp , src_stride , src_x , src_y , b_w , b_h , rb , plane_index , w , h ) ;
}
#if 0
for ( y = 0 ; y < b_h ; y + + ) {
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc [ x + y * obmc_stride ] * block [ 3 ] [ x + y * src_stride ] * ( 256 / OBMC_MAX ) ;
if ( add ) dst [ x + y * dst_stride ] + = v ;
else dst [ x + y * dst_stride ] - = v ;
}
}
for ( y = 0 ; y < b_h ; y + + ) {
uint8_t * obmc2 = obmc + ( obmc_stride > > 1 ) ;
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc2 [ x + y * obmc_stride ] * block [ 2 ] [ x + y * src_stride ] * ( 256 / OBMC_MAX ) ;
if ( add ) dst [ x + y * dst_stride ] + = v ;
else dst [ x + y * dst_stride ] - = v ;
}
}
for ( y = 0 ; y < b_h ; y + + ) {
uint8_t * obmc3 = obmc + obmc_stride * ( obmc_stride > > 1 ) ;
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc3 [ x + y * obmc_stride ] * block [ 1 ] [ x + y * src_stride ] * ( 256 / OBMC_MAX ) ;
if ( add ) dst [ x + y * dst_stride ] + = v ;
else dst [ x + y * dst_stride ] - = v ;
}
}
for ( y = 0 ; y < b_h ; y + + ) {
uint8_t * obmc3 = obmc + obmc_stride * ( obmc_stride > > 1 ) ;
uint8_t * obmc4 = obmc3 + ( obmc_stride > > 1 ) ;
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc4 [ x + y * obmc_stride ] * block [ 0 ] [ x + y * src_stride ] * ( 256 / OBMC_MAX ) ;
if ( add ) dst [ x + y * dst_stride ] + = v ;
else dst [ x + y * dst_stride ] - = v ;
}
}
# else
for ( y = 0 ; y < b_h ; y + + ) {
//FIXME ugly missue of obmc_stride
uint8_t * obmc1 = obmc + y * obmc_stride ;
uint8_t * obmc2 = obmc1 + ( obmc_stride > > 1 ) ;
uint8_t * obmc3 = obmc1 + obmc_stride * ( obmc_stride > > 1 ) ;
uint8_t * obmc4 = obmc3 + ( obmc_stride > > 1 ) ;
for ( x = 0 ; x < b_w ; x + + ) {
int v = obmc1 [ x ] * block [ 3 ] [ x + y * src_stride ]
+ obmc2 [ x ] * block [ 2 ] [ x + y * src_stride ]
+ obmc3 [ x ] * block [ 1 ] [ x + y * src_stride ]
+ obmc4 [ x ] * block [ 0 ] [ x + y * src_stride ] ;
2005-12-17 19:14:38 +01:00
2004-09-23 14:20:41 +02:00
v < < = 8 - LOG2_OBMC_MAX ;
2004-09-22 01:10:10 +02:00
if ( FRAC_BITS ! = 8 ) {
v + = 1 < < ( 7 - FRAC_BITS ) ;
v > > = 8 - FRAC_BITS ;
}
2004-09-23 14:20:41 +02:00
if ( add ) {
v + = dst [ x + y * dst_stride ] ;
v = ( v + ( 1 < < ( FRAC_BITS - 1 ) ) ) > > FRAC_BITS ;
if ( v & ( ~ 255 ) ) v = ~ ( v > > 31 ) ;
dst8 [ x + y * src_stride ] = v ;
} else {
dst [ x + y * dst_stride ] - = v ;
}
2004-07-26 01:36:44 +02:00
}
}
2004-09-05 23:46:16 +02:00
# endif
2004-07-26 01:36:44 +02:00
}
2005-04-03 17:43:57 +02:00
static always_inline void predict_slice_buffered ( SnowContext * s , slice_buffer * sb , DWTELEM * old_buffer , int plane_index , int add , int mb_y ) {
Plane * p = & s - > plane [ plane_index ] ;
const int mb_w = s - > b_width < < s - > block_max_depth ;
const int mb_h = s - > b_height < < s - > block_max_depth ;
int x , y , mb_x ;
int block_size = MB_SIZE > > s - > block_max_depth ;
int block_w = plane_index ? block_size / 2 : block_size ;
const uint8_t * obmc = plane_index ? obmc_tab [ s - > block_max_depth + 1 ] : obmc_tab [ s - > block_max_depth ] ;
int obmc_stride = plane_index ? block_size : 2 * block_size ;
int ref_stride = s - > current_picture . linesize [ plane_index ] ;
uint8_t * ref = s - > last_picture . data [ plane_index ] ;
uint8_t * dst8 = s - > current_picture . data [ plane_index ] ;
int w = p - > width ;
int h = p - > height ;
START_TIMER
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
if ( s - > keyframe | | ( s - > avctx - > debug & 512 ) ) {
if ( mb_y = = mb_h )
return ;
if ( add ) {
2005-04-13 03:57:19 +02:00
for ( y = block_w * mb_y ; y < FFMIN ( h , block_w * ( mb_y + 1 ) ) ; y + + )
2005-04-03 17:43:57 +02:00
{
// DWTELEM * line = slice_buffer_get_line(sb, y);
DWTELEM * line = sb - > line [ y ] ;
for ( x = 0 ; x < w ; x + + )
{
// int v= buf[x + y*w] + (128<<FRAC_BITS) + (1<<(FRAC_BITS-1));
int v = line [ x ] + ( 128 < < FRAC_BITS ) + ( 1 < < ( FRAC_BITS - 1 ) ) ;
v > > = FRAC_BITS ;
if ( v & ( ~ 255 ) ) v = ~ ( v > > 31 ) ;
dst8 [ x + y * ref_stride ] = v ;
}
}
} else {
2005-04-13 03:57:19 +02:00
for ( y = block_w * mb_y ; y < FFMIN ( h , block_w * ( mb_y + 1 ) ) ; y + + )
2005-04-03 17:43:57 +02:00
{
// DWTELEM * line = slice_buffer_get_line(sb, y);
DWTELEM * line = sb - > line [ y ] ;
for ( x = 0 ; x < w ; x + + )
{
line [ x ] - = 128 < < FRAC_BITS ;
// buf[x + y*w]-= 128<<FRAC_BITS;
}
}
}
return ;
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
for ( mb_x = 0 ; mb_x < = mb_w ; mb_x + + ) {
START_TIMER
2005-12-17 19:14:38 +01:00
add_yblock_buffered ( s , sb , old_buffer , dst8 , ref , obmc ,
2005-04-03 17:43:57 +02:00
block_w * mb_x - block_w / 2 ,
block_w * mb_y - block_w / 2 ,
block_w , block_w ,
w , h ,
w , ref_stride , obmc_stride ,
mb_x - 1 , mb_y - 1 ,
add , plane_index ) ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
STOP_TIMER ( " add_yblock " )
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
STOP_TIMER ( " predict_slice " )
}
2005-03-15 22:05:34 +01:00
static always_inline void predict_slice ( SnowContext * s , DWTELEM * buf , int plane_index , int add , int mb_y ) {
2004-07-26 01:36:44 +02:00
Plane * p = & s - > plane [ plane_index ] ;
2004-08-26 22:04:54 +02:00
const int mb_w = s - > b_width < < s - > block_max_depth ;
const int mb_h = s - > b_height < < s - > block_max_depth ;
2005-03-15 22:05:34 +01:00
int x , y , mb_x ;
2004-08-26 22:04:54 +02:00
int block_size = MB_SIZE > > s - > block_max_depth ;
int block_w = plane_index ? block_size / 2 : block_size ;
2004-09-05 23:46:16 +02:00
const uint8_t * obmc = plane_index ? obmc_tab [ s - > block_max_depth + 1 ] : obmc_tab [ s - > block_max_depth ] ;
2005-12-28 16:43:53 +01:00
const int obmc_stride = plane_index ? block_size : 2 * block_size ;
2004-09-23 14:20:41 +02:00
int ref_stride = s - > current_picture . linesize [ plane_index ] ;
2004-07-26 01:36:44 +02:00
uint8_t * ref = s - > last_picture . data [ plane_index ] ;
2004-09-23 14:20:41 +02:00
uint8_t * dst8 = s - > current_picture . data [ plane_index ] ;
2004-07-26 01:36:44 +02:00
int w = p - > width ;
int h = p - > height ;
2004-08-28 16:08:50 +02:00
START_TIMER
2005-12-17 19:14:38 +01:00
2004-09-05 23:46:16 +02:00
if ( s - > keyframe | | ( s - > avctx - > debug & 512 ) ) {
2005-03-15 22:05:34 +01:00
if ( mb_y = = mb_h )
return ;
2004-09-23 14:20:41 +02:00
if ( add ) {
2005-04-13 03:57:19 +02:00
for ( y = block_w * mb_y ; y < FFMIN ( h , block_w * ( mb_y + 1 ) ) ; y + + ) {
2004-09-23 14:20:41 +02:00
for ( x = 0 ; x < w ; x + + ) {
int v = buf [ x + y * w ] + ( 128 < < FRAC_BITS ) + ( 1 < < ( FRAC_BITS - 1 ) ) ;
v > > = FRAC_BITS ;
if ( v & ( ~ 255 ) ) v = ~ ( v > > 31 ) ;
dst8 [ x + y * ref_stride ] = v ;
}
}
} else {
2005-04-13 03:57:19 +02:00
for ( y = block_w * mb_y ; y < FFMIN ( h , block_w * ( mb_y + 1 ) ) ; y + + ) {
2004-09-23 14:20:41 +02:00
for ( x = 0 ; x < w ; x + + ) {
buf [ x + y * w ] - = 128 < < FRAC_BITS ;
}
2004-09-05 23:46:16 +02:00
}
2004-07-26 01:36:44 +02:00
}
2004-09-05 23:46:16 +02:00
return ;
2004-07-26 01:36:44 +02:00
}
2005-12-17 19:14:38 +01:00
2004-09-05 23:46:16 +02:00
for ( mb_x = 0 ; mb_x < = mb_w ; mb_x + + ) {
2004-08-28 16:08:50 +02:00
START_TIMER
2004-09-05 23:46:16 +02:00
2005-12-17 19:14:38 +01:00
add_yblock ( s , buf , dst8 , ref , obmc ,
2004-09-05 23:46:16 +02:00
block_w * mb_x - block_w / 2 ,
2004-07-26 01:36:44 +02:00
block_w * mb_y - block_w / 2 ,
2004-09-05 23:46:16 +02:00
block_w , block_w ,
2004-07-26 01:36:44 +02:00
w , h ,
2004-09-05 23:46:16 +02:00
w , ref_stride , obmc_stride ,
mb_x - 1 , mb_y - 1 ,
2006-01-12 06:47:52 +01:00
add , 1 , plane_index ) ;
2005-12-17 19:14:38 +01:00
2004-09-05 23:46:16 +02:00
STOP_TIMER ( " add_yblock " )
2004-07-26 01:36:44 +02:00
}
2005-12-17 19:14:38 +01:00
2005-03-15 22:05:34 +01:00
STOP_TIMER ( " predict_slice " )
}
static always_inline void predict_plane ( SnowContext * s , DWTELEM * buf , int plane_index , int add ) {
const int mb_h = s - > b_height < < s - > block_max_depth ;
int mb_y ;
for ( mb_y = 0 ; mb_y < = mb_h ; mb_y + + )
predict_slice ( s , buf , plane_index , add , mb_y ) ;
2004-07-26 01:36:44 +02:00
}
2005-12-28 16:43:53 +01:00
static int get_dc ( SnowContext * s , int mb_x , int mb_y , int plane_index ) {
int i , x2 , y2 ;
Plane * p = & s - > plane [ plane_index ] ;
const int block_size = MB_SIZE > > s - > block_max_depth ;
const int block_w = plane_index ? block_size / 2 : block_size ;
const uint8_t * obmc = plane_index ? obmc_tab [ s - > block_max_depth + 1 ] : obmc_tab [ s - > block_max_depth ] ;
const int obmc_stride = plane_index ? block_size : 2 * block_size ;
const int ref_stride = s - > current_picture . linesize [ plane_index ] ;
uint8_t * ref = s - > last_picture . data [ plane_index ] ;
uint8_t * src = s - > input_picture . data [ plane_index ] ;
2006-01-12 06:47:52 +01:00
DWTELEM * dst = ( DWTELEM * ) s - > m . obmc_scratchpad + plane_index * block_size * block_size * 4 ;
2005-12-28 16:43:53 +01:00
const int b_stride = s - > b_width < < s - > block_max_depth ;
const int w = p - > width ;
const int h = p - > height ;
int index = mb_x + mb_y * b_stride ;
BlockNode * b = & s - > block [ index ] ;
BlockNode backup = * b ;
int ab = 0 ;
int aa = 0 ;
b - > type | = BLOCK_INTRA ;
b - > color [ plane_index ] = 0 ;
2006-01-12 06:47:52 +01:00
memset ( dst , 0 , obmc_stride * obmc_stride * sizeof ( DWTELEM ) ) ;
2005-12-28 16:43:53 +01:00
for ( i = 0 ; i < 4 ; i + + ) {
int mb_x2 = mb_x + ( i & 1 ) - 1 ;
int mb_y2 = mb_y + ( i > > 1 ) - 1 ;
int x = block_w * mb_x2 + block_w / 2 ;
int y = block_w * mb_y2 + block_w / 2 ;
2006-01-24 11:33:14 +01:00
add_yblock ( s , dst + ( ( i & 1 ) + ( i > > 1 ) * obmc_stride ) * block_w , NULL , ref , obmc ,
2006-01-12 06:47:52 +01:00
x , y , block_w , block_w , w , h , obmc_stride , ref_stride , obmc_stride , mb_x2 , mb_y2 , 0 , 0 , plane_index ) ;
2005-12-28 16:43:53 +01:00
for ( y2 = FFMAX ( y , 0 ) ; y2 < FFMIN ( h , y + block_w ) ; y2 + + ) {
for ( x2 = FFMAX ( x , 0 ) ; x2 < FFMIN ( w , x + block_w ) ; x2 + + ) {
int index = x2 - ( block_w * mb_x - block_w / 2 ) + ( y2 - ( block_w * mb_y - block_w / 2 ) ) * obmc_stride ;
int obmc_v = obmc [ index ] ;
2006-01-12 06:47:52 +01:00
int d ;
2005-12-28 16:43:53 +01:00
if ( y < 0 ) obmc_v + = obmc [ index + block_w * obmc_stride ] ;
if ( x < 0 ) obmc_v + = obmc [ index + block_w ] ;
if ( y + block_w > h ) obmc_v + = obmc [ index - block_w * obmc_stride ] ;
if ( x + block_w > w ) obmc_v + = obmc [ index - block_w ] ;
//FIXME precalc this or simplify it somehow else
2006-01-12 06:47:52 +01:00
d = - dst [ index ] + ( 1 < < ( FRAC_BITS - 1 ) ) ;
dst [ index ] = d ;
ab + = ( src [ x2 + y2 * ref_stride ] - ( d > > FRAC_BITS ) ) * obmc_v ;
2005-12-28 16:43:53 +01:00
aa + = obmc_v * obmc_v ; //FIXME precalclate this
}
}
}
* b = backup ;
return clip ( ( ( ab < < 6 ) + aa / 2 ) / aa , 0 , 255 ) ; //FIXME we shouldnt need cliping
}
2006-01-17 09:27:39 +01:00
static inline int get_block_bits ( SnowContext * s , int x , int y , int w ) {
const int b_stride = s - > b_width < < s - > block_max_depth ;
const int b_height = s - > b_height < < s - > block_max_depth ;
int index = x + y * b_stride ;
BlockNode * b = & s - > block [ index ] ;
BlockNode * left = x ? & s - > block [ index - 1 ] : & null_block ;
BlockNode * top = y ? & s - > block [ index - b_stride ] : & null_block ;
BlockNode * tl = y & & x ? & s - > block [ index - b_stride - 1 ] : left ;
BlockNode * tr = y & & x + w < b_stride ? & s - > block [ index - b_stride + w ] : tl ;
int dmx , dmy ;
// int mx_context= av_log2(2*ABS(left->mx - top->mx));
// int my_context= av_log2(2*ABS(left->my - top->my));
if ( x < 0 | | x > = b_stride | | y > = b_height )
return 0 ;
dmx = b - > mx - mid_pred ( left - > mx , top - > mx , tr - > mx ) ;
dmy = b - > my - mid_pred ( left - > my , top - > my , tr - > my ) ;
/*
1 0 0
01 X 1 - 2 1
001 XX 3 - 6 2 - 3
0001 XXX 7 - 14 4 - 7
00001 XXXX 15 - 30 8 - 15
*/
//FIXME try accurate rate
//FIXME intra and inter predictors if surrounding blocks arent the same type
if ( b - > type & BLOCK_INTRA ) {
return 3 + 2 * ( av_log2 ( 2 * ABS ( left - > color [ 0 ] - b - > color [ 0 ] ) )
+ av_log2 ( 2 * ABS ( left - > color [ 1 ] - b - > color [ 1 ] ) )
+ av_log2 ( 2 * ABS ( left - > color [ 2 ] - b - > color [ 2 ] ) ) ) ;
} else
return 2 * ( 1 + av_log2 ( 2 * ABS ( dmx ) )
+ av_log2 ( 2 * ABS ( dmy ) ) ) ; //FIXME kill the 2* can be merged in lambda
}
2006-01-12 06:47:52 +01:00
static int get_block_rd ( SnowContext * s , int mb_x , int mb_y , int plane_index , const uint8_t * obmc_edged ) {
2005-12-28 16:43:53 +01:00
Plane * p = & s - > plane [ plane_index ] ;
const int block_size = MB_SIZE > > s - > block_max_depth ;
const int block_w = plane_index ? block_size / 2 : block_size ;
const uint8_t * obmc = plane_index ? obmc_tab [ s - > block_max_depth + 1 ] : obmc_tab [ s - > block_max_depth ] ;
const int obmc_stride = plane_index ? block_size : 2 * block_size ;
const int ref_stride = s - > current_picture . linesize [ plane_index ] ;
uint8_t * ref = s - > last_picture . data [ plane_index ] ;
uint8_t * dst = s - > current_picture . data [ plane_index ] ;
2006-01-12 06:47:52 +01:00
uint8_t * src = s - > input_picture . data [ plane_index ] ;
DWTELEM * pred = ( DWTELEM * ) s - > m . obmc_scratchpad + plane_index * block_size * block_size * 4 ;
uint8_t cur [ ref_stride * 2 * MB_SIZE ] ; //FIXME alignment
uint8_t tmp [ ref_stride * ( 2 * MB_SIZE + 5 ) ] ;
2005-12-28 16:43:53 +01:00
const int b_stride = s - > b_width < < s - > block_max_depth ;
const int b_height = s - > b_height < < s - > block_max_depth ;
const int w = p - > width ;
const int h = p - > height ;
2006-01-12 06:47:52 +01:00
int distortion ;
2005-12-28 16:43:53 +01:00
int rate = 0 ;
const int penalty_factor = get_penalty_factor ( s - > lambda , s - > lambda2 , s - > avctx - > me_cmp ) ;
2006-01-12 06:47:52 +01:00
int sx = block_w * mb_x - block_w / 2 ;
int sy = block_w * mb_y - block_w / 2 ;
const int x0 = FFMAX ( 0 , - sx ) ;
const int y0 = FFMAX ( 0 , - sy ) ;
const int x1 = FFMIN ( block_w * 2 , w - sx ) ;
const int y1 = FFMIN ( block_w * 2 , h - sy ) ;
int i , x , y ;
pred_block ( s , cur , ref , tmp , ref_stride , sx , sy , block_w * 2 , block_w * 2 , & s - > block [ mb_x + mb_y * b_stride ] , plane_index , w , h ) ;
for ( y = y0 ; y < y1 ; y + + ) {
const uint8_t * obmc1 = obmc_edged + y * obmc_stride ;
const DWTELEM * pred1 = pred + y * obmc_stride ;
uint8_t * cur1 = cur + y * ref_stride ;
uint8_t * dst1 = dst + sx + ( sy + y ) * ref_stride ;
for ( x = x0 ; x < x1 ; x + + ) {
int v = ( cur1 [ x ] * obmc1 [ x ] ) < < ( FRAC_BITS - LOG2_OBMC_MAX ) ;
v = ( v + pred1 [ x ] ) > > FRAC_BITS ;
if ( v & ( ~ 255 ) ) v = ~ ( v > > 31 ) ;
dst1 [ x ] = v ;
2005-12-28 16:43:53 +01:00
}
2006-01-12 06:47:52 +01:00
}
2005-12-28 16:43:53 +01:00
2006-01-12 06:47:52 +01:00
//FIXME sad/ssd can be broken up, but wavelet cmp should be one 32x32 block
if ( block_w = = 16 ) {
distortion = 0 ;
for ( i = 0 ; i < 4 ; i + + ) {
int off = sx + 16 * ( i & 1 ) + ( sy + 16 * ( i > > 1 ) ) * ref_stride ;
distortion + = s - > dsp . me_cmp [ 0 ] ( & s - > m , src + off , dst + off , ref_stride , 16 ) ;
}
} else {
assert ( block_w = = 8 ) ;
distortion = s - > dsp . me_cmp [ 0 ] ( & s - > m , src + sx + sy * ref_stride , dst + sx + sy * ref_stride , ref_stride , block_w * 2 ) ;
2005-12-28 16:43:53 +01:00
}
if ( plane_index = = 0 ) {
for ( i = 0 ; i < 4 ; i + + ) {
/* ..RRr
* . RXx .
* rxx . .
*/
2006-01-17 09:27:39 +01:00
rate + = get_block_bits ( s , mb_x + ( i & 1 ) - ( i > > 1 ) , mb_y + ( i > > 1 ) , 1 ) ;
}
2006-01-21 03:35:03 +01:00
if ( mb_x = = b_stride - 2 )
rate + = get_block_bits ( s , mb_x + 1 , mb_y + 1 , 1 ) ;
2006-01-17 09:27:39 +01:00
}
return distortion + rate * penalty_factor ;
}
static int get_4block_rd ( SnowContext * s , int mb_x , int mb_y , int plane_index ) {
int i , y2 ;
Plane * p = & s - > plane [ plane_index ] ;
const int block_size = MB_SIZE > > s - > block_max_depth ;
const int block_w = plane_index ? block_size / 2 : block_size ;
const uint8_t * obmc = plane_index ? obmc_tab [ s - > block_max_depth + 1 ] : obmc_tab [ s - > block_max_depth ] ;
const int obmc_stride = plane_index ? block_size : 2 * block_size ;
const int ref_stride = s - > current_picture . linesize [ plane_index ] ;
uint8_t * ref = s - > last_picture . data [ plane_index ] ;
uint8_t * dst = s - > current_picture . data [ plane_index ] ;
uint8_t * src = s - > input_picture . data [ plane_index ] ;
const static DWTELEM zero_dst [ 4096 ] ; //FIXME
const int b_stride = s - > b_width < < s - > block_max_depth ;
const int b_height = s - > b_height < < s - > block_max_depth ;
const int w = p - > width ;
const int h = p - > height ;
int distortion = 0 ;
int rate = 0 ;
const int penalty_factor = get_penalty_factor ( s - > lambda , s - > lambda2 , s - > avctx - > me_cmp ) ;
for ( i = 0 ; i < 9 ; i + + ) {
int mb_x2 = mb_x + ( i % 3 ) - 1 ;
int mb_y2 = mb_y + ( i / 3 ) - 1 ;
int x = block_w * mb_x2 + block_w / 2 ;
int y = block_w * mb_y2 + block_w / 2 ;
2006-01-24 11:33:14 +01:00
add_yblock ( s , zero_dst , dst , ref , obmc ,
2006-01-17 09:27:39 +01:00
x , y , block_w , block_w , w , h , /*dst_stride*/ 0 , ref_stride , obmc_stride , mb_x2 , mb_y2 , 1 , 1 , plane_index ) ;
//FIXME find a cleaner/simpler way to skip the outside stuff
for ( y2 = y ; y2 < 0 ; y2 + + )
memcpy ( dst + x + y2 * ref_stride , src + x + y2 * ref_stride , block_w ) ;
for ( y2 = h ; y2 < y + block_w ; y2 + + )
memcpy ( dst + x + y2 * ref_stride , src + x + y2 * ref_stride , block_w ) ;
if ( x < 0 ) {
for ( y2 = y ; y2 < y + block_w ; y2 + + )
memcpy ( dst + x + y2 * ref_stride , src + x + y2 * ref_stride , - x ) ;
2005-12-28 16:43:53 +01:00
}
2006-01-17 09:27:39 +01:00
if ( x + block_w > w ) {
for ( y2 = y ; y2 < y + block_w ; y2 + + )
memcpy ( dst + w + y2 * ref_stride , src + w + y2 * ref_stride , x + block_w - w ) ;
}
assert ( block_w = = 8 | | block_w = = 16 ) ;
distortion + = s - > dsp . me_cmp [ block_w = = 8 ] ( & s - > m , src + x + y * ref_stride , dst + x + y * ref_stride , ref_stride , block_w ) ;
2005-12-28 16:43:53 +01:00
}
2006-01-17 09:27:39 +01:00
if ( plane_index = = 0 ) {
BlockNode * b = & s - > block [ mb_x + mb_y * b_stride ] ;
int merged = same_block ( b , b + 1 ) & & same_block ( b , b + b_stride ) & & same_block ( b , b + b_stride + 1 ) ;
/* ..RRRr
* . RXXx .
* . RXXx .
* rxxx .
*/
if ( merged )
rate = get_block_bits ( s , mb_x , mb_y , 2 ) ;
for ( i = merged ? 4 : 0 ; i < 9 ; i + + ) {
static const int dxy [ 9 ] [ 2 ] = { { 0 , 0 } , { 1 , 0 } , { 0 , 1 } , { 1 , 1 } , { 2 , 0 } , { 2 , 1 } , { - 1 , 2 } , { 0 , 2 } , { 1 , 2 } } ;
rate + = get_block_bits ( s , mb_x + dxy [ i ] [ 0 ] , mb_y + dxy [ i ] [ 1 ] , 1 ) ;
}
}
2005-12-28 16:43:53 +01:00
return distortion + rate * penalty_factor ;
}
2006-01-12 06:47:52 +01:00
static always_inline int check_block ( SnowContext * s , int mb_x , int mb_y , int p [ 3 ] , int intra , const uint8_t * obmc_edged , int * best_rd ) {
2005-12-28 16:43:53 +01:00
const int b_stride = s - > b_width < < s - > block_max_depth ;
BlockNode * block = & s - > block [ mb_x + mb_y * b_stride ] ;
BlockNode backup = * block ;
int rd , index , value ;
assert ( mb_x > = 0 & & mb_y > = 0 ) ;
2005-12-29 00:45:16 +01:00
assert ( mb_x < b_stride ) ;
2005-12-28 16:43:53 +01:00
if ( intra ) {
block - > color [ 0 ] = p [ 0 ] ;
block - > color [ 1 ] = p [ 1 ] ;
block - > color [ 2 ] = p [ 2 ] ;
block - > type | = BLOCK_INTRA ;
} else {
index = ( p [ 0 ] + 31 * p [ 1 ] ) & ( ME_CACHE_SIZE - 1 ) ;
value = s - > me_cache_generation + ( p [ 0 ] > > 10 ) + ( p [ 1 ] < < 6 ) ;
if ( s - > me_cache [ index ] = = value )
return 0 ;
s - > me_cache [ index ] = value ;
block - > mx = p [ 0 ] ;
block - > my = p [ 1 ] ;
block - > type & = ~ BLOCK_INTRA ;
}
2006-01-12 06:47:52 +01:00
rd = get_block_rd ( s , mb_x , mb_y , 0 , obmc_edged ) ;
2005-12-28 16:43:53 +01:00
//FIXME chroma
if ( rd < * best_rd ) {
* best_rd = rd ;
return 1 ;
} else {
* block = backup ;
return 0 ;
}
}
2005-12-30 22:05:44 +01:00
/* special case for int[2] args we discard afterward, fixes compilation prob with gcc 2.95 */
2006-01-12 06:47:52 +01:00
static always_inline int check_block_inter ( SnowContext * s , int mb_x , int mb_y , int p0 , int p1 , int intra , const uint8_t * obmc_edged , int * best_rd ) {
2005-12-30 22:05:44 +01:00
int p [ 2 ] = { p0 , p1 } ;
2006-01-12 06:47:52 +01:00
return check_block ( s , mb_x , mb_y , p , intra , obmc_edged , best_rd ) ;
2005-12-30 22:05:44 +01:00
}
2006-01-17 09:27:39 +01:00
static always_inline int check_4block_inter ( SnowContext * s , int mb_x , int mb_y , int p0 , int p1 , int * best_rd ) {
const int b_stride = s - > b_width < < s - > block_max_depth ;
BlockNode * block = & s - > block [ mb_x + mb_y * b_stride ] ;
BlockNode backup [ 4 ] = { block [ 0 ] , block [ 1 ] , block [ b_stride ] , block [ b_stride + 1 ] } ;
int rd , index , value ;
assert ( mb_x > = 0 & & mb_y > = 0 ) ;
assert ( mb_x < b_stride ) ;
assert ( ( ( mb_x | mb_y ) & 1 ) = = 0 ) ;
index = ( p0 + 31 * p1 ) & ( ME_CACHE_SIZE - 1 ) ;
value = s - > me_cache_generation + ( p0 > > 10 ) + ( p1 < < 6 ) ;
if ( s - > me_cache [ index ] = = value )
return 0 ;
s - > me_cache [ index ] = value ;
block - > mx = p0 ;
block - > my = p1 ;
block - > type & = ~ BLOCK_INTRA ;
block [ 1 ] = block [ b_stride ] = block [ b_stride + 1 ] = * block ;
rd = get_4block_rd ( s , mb_x , mb_y , 0 ) ;
//FIXME chroma
if ( rd < * best_rd ) {
* best_rd = rd ;
return 1 ;
} else {
block [ 0 ] = backup [ 0 ] ;
block [ 1 ] = backup [ 1 ] ;
block [ b_stride ] = backup [ 2 ] ;
block [ b_stride + 1 ] = backup [ 3 ] ;
return 0 ;
}
}
2005-12-28 16:43:53 +01:00
static void iterative_me ( SnowContext * s ) {
int pass , mb_x , mb_y ;
const int b_width = s - > b_width < < s - > block_max_depth ;
const int b_height = s - > b_height < < s - > block_max_depth ;
const int b_stride = b_width ;
int color [ 3 ] ;
for ( pass = 0 ; pass < 50 ; pass + + ) {
int change = 0 ;
for ( mb_y = 0 ; mb_y < b_height ; mb_y + + ) {
for ( mb_x = 0 ; mb_x < b_width ; mb_x + + ) {
int dia_change , i , j ;
int best_rd = INT_MAX ;
BlockNode backup ;
const int index = mb_x + mb_y * b_stride ;
BlockNode * block = & s - > block [ index ] ;
2006-01-12 22:37:57 +01:00
BlockNode * tb = mb_y ? & s - > block [ index - b_stride ] : & null_block ;
BlockNode * lb = mb_x ? & s - > block [ index - 1 ] : & null_block ;
BlockNode * rb = mb_x + 1 < b_width ? & s - > block [ index + 1 ] : & null_block ;
BlockNode * bb = mb_y + 1 < b_height ? & s - > block [ index + b_stride ] : & null_block ;
BlockNode * tlb = mb_x & & mb_y ? & s - > block [ index - b_stride - 1 ] : & null_block ;
BlockNode * trb = mb_x + 1 < b_width & & mb_y ? & s - > block [ index - b_stride + 1 ] : & null_block ;
BlockNode * blb = mb_x & & mb_y + 1 < b_height ? & s - > block [ index + b_stride - 1 ] : & null_block ;
BlockNode * brb = mb_x + 1 < b_width & & mb_y + 1 < b_height ? & s - > block [ index + b_stride + 1 ] : & null_block ;
2006-01-12 06:47:52 +01:00
const int b_w = ( MB_SIZE > > s - > block_max_depth ) ;
uint8_t obmc_edged [ b_w * 2 ] [ b_w * 2 ] ;
2005-12-28 16:43:53 +01:00
if ( pass & & ( block - > type & BLOCK_OPT ) )
continue ;
block - > type | = BLOCK_OPT ;
backup = * block ;
if ( ! s - > me_cache_generation )
memset ( s - > me_cache , 0 , sizeof ( s - > me_cache ) ) ;
s - > me_cache_generation + = 1 < < 22 ;
2006-01-12 06:47:52 +01:00
//FIXME precalc
{
int x , y ;
memcpy ( obmc_edged , obmc_tab [ s - > block_max_depth ] , b_w * b_w * 4 ) ;
if ( mb_x = = 0 )
for ( y = 0 ; y < b_w * 2 ; y + + )
memset ( obmc_edged [ y ] , obmc_edged [ y ] [ 0 ] + obmc_edged [ y ] [ b_w - 1 ] , b_w ) ;
if ( mb_x = = b_stride - 1 )
for ( y = 0 ; y < b_w * 2 ; y + + )
memset ( obmc_edged [ y ] + b_w , obmc_edged [ y ] [ b_w ] + obmc_edged [ y ] [ b_w * 2 - 1 ] , b_w ) ;
if ( mb_y = = 0 ) {
for ( x = 0 ; x < b_w * 2 ; x + + )
obmc_edged [ 0 ] [ x ] + = obmc_edged [ b_w - 1 ] [ x ] ;
for ( y = 1 ; y < b_w ; y + + )
memcpy ( obmc_edged [ y ] , obmc_edged [ 0 ] , b_w * 2 ) ;
}
if ( mb_y = = b_height - 1 ) {
for ( x = 0 ; x < b_w * 2 ; x + + )
obmc_edged [ b_w * 2 - 1 ] [ x ] + = obmc_edged [ b_w ] [ x ] ;
for ( y = b_w ; y < b_w * 2 - 1 ; y + + )
memcpy ( obmc_edged [ y ] , obmc_edged [ b_w * 2 - 1 ] , b_w * 2 ) ;
}
}
//skip stuff outside the picture
if ( mb_x = = 0 | | mb_y = = 0 | | mb_x = = b_width - 1 | | mb_y = = b_height - 1 )
{
uint8_t * src = s - > input_picture . data [ 0 ] ;
uint8_t * dst = s - > current_picture . data [ 0 ] ;
const int stride = s - > current_picture . linesize [ 0 ] ;
const int block_w = MB_SIZE > > s - > block_max_depth ;
const int sx = block_w * mb_x - block_w / 2 ;
const int sy = block_w * mb_y - block_w / 2 ;
const int w = s - > plane [ 0 ] . width ;
const int h = s - > plane [ 0 ] . height ;
int y ;
for ( y = sy ; y < 0 ; y + + )
memcpy ( dst + sx + y * stride , src + sx + y * stride , block_w * 2 ) ;
for ( y = h ; y < sy + block_w * 2 ; y + + )
memcpy ( dst + sx + y * stride , src + sx + y * stride , block_w * 2 ) ;
if ( sx < 0 ) {
for ( y = sy ; y < sy + block_w * 2 ; y + + )
memcpy ( dst + sx + y * stride , src + sx + y * stride , - sx ) ;
}
if ( sx + block_w * 2 > w ) {
for ( y = sy ; y < sy + block_w * 2 ; y + + )
memcpy ( dst + w + y * stride , src + w + y * stride , sx + block_w * 2 - w ) ;
}
}
// intra(black) = neighbors' contribution to the current block
for ( i = 0 ; i < 3 ; i + + )
color [ i ] = get_dc ( s , mb_x , mb_y , i ) ;
2005-12-28 16:43:53 +01:00
// get previous score (cant be cached due to OBMC)
2006-01-21 03:35:03 +01:00
if ( pass > 0 & & ( block - > type & BLOCK_INTRA ) ) {
int color0 [ 3 ] = { block - > color [ 0 ] , block - > color [ 1 ] , block - > color [ 2 ] } ;
check_block ( s , mb_x , mb_y , color0 , 1 , * obmc_edged , & best_rd ) ;
} else
check_block_inter ( s , mb_x , mb_y , block - > mx , block - > my , 0 , * obmc_edged , & best_rd ) ;
2006-01-12 06:47:52 +01:00
check_block_inter ( s , mb_x , mb_y , 0 , 0 , 0 , * obmc_edged , & best_rd ) ;
check_block_inter ( s , mb_x , mb_y , tb - > mx , tb - > my , 0 , * obmc_edged , & best_rd ) ;
check_block_inter ( s , mb_x , mb_y , lb - > mx , lb - > my , 0 , * obmc_edged , & best_rd ) ;
check_block_inter ( s , mb_x , mb_y , rb - > mx , rb - > my , 0 , * obmc_edged , & best_rd ) ;
check_block_inter ( s , mb_x , mb_y , bb - > mx , bb - > my , 0 , * obmc_edged , & best_rd ) ;
2005-12-28 16:43:53 +01:00
/* fullpel ME */
//FIXME avoid subpel interpol / round to nearest integer
do {
dia_change = 0 ;
for ( i = 0 ; i < FFMAX ( s - > avctx - > dia_size , 1 ) ; i + + ) {
for ( j = 0 ; j < i ; j + + ) {
2006-01-12 06:47:52 +01:00
dia_change | = check_block_inter ( s , mb_x , mb_y , block - > mx + 4 * ( i - j ) , block - > my + ( 4 * j ) , 0 , * obmc_edged , & best_rd ) ;
dia_change | = check_block_inter ( s , mb_x , mb_y , block - > mx - 4 * ( i - j ) , block - > my - ( 4 * j ) , 0 , * obmc_edged , & best_rd ) ;
dia_change | = check_block_inter ( s , mb_x , mb_y , block - > mx + 4 * ( i - j ) , block - > my - ( 4 * j ) , 0 , * obmc_edged , & best_rd ) ;
dia_change | = check_block_inter ( s , mb_x , mb_y , block - > mx - 4 * ( i - j ) , block - > my + ( 4 * j ) , 0 , * obmc_edged , & best_rd ) ;
2005-12-28 16:43:53 +01:00
}
}
} while ( dia_change ) ;
/* subpel ME */
do {
static const int square [ 8 ] [ 2 ] = { { + 1 , 0 } , { - 1 , 0 } , { 0 , + 1 } , { 0 , - 1 } , { + 1 , + 1 } , { - 1 , - 1 } , { + 1 , - 1 } , { - 1 , + 1 } , } ;
dia_change = 0 ;
for ( i = 0 ; i < 8 ; i + + )
2006-01-12 06:47:52 +01:00
dia_change | = check_block_inter ( s , mb_x , mb_y , block - > mx + square [ i ] [ 0 ] , block - > my + square [ i ] [ 1 ] , 0 , * obmc_edged , & best_rd ) ;
2005-12-28 16:43:53 +01:00
} while ( dia_change ) ;
//FIXME or try the standard 2 pass qpel or similar
2005-12-30 16:12:02 +01:00
# if 1
2006-01-12 06:47:52 +01:00
check_block ( s , mb_x , mb_y , color , 1 , * obmc_edged , & best_rd ) ;
2005-12-28 16:43:53 +01:00
//FIXME RD style color selection
2005-12-30 16:12:02 +01:00
# endif
2005-12-28 16:43:53 +01:00
if ( ! same_block ( block , & backup ) ) {
if ( tb ! = & null_block ) tb - > type & = ~ BLOCK_OPT ;
if ( lb ! = & null_block ) lb - > type & = ~ BLOCK_OPT ;
if ( rb ! = & null_block ) rb - > type & = ~ BLOCK_OPT ;
if ( bb ! = & null_block ) bb - > type & = ~ BLOCK_OPT ;
if ( tlb ! = & null_block ) tlb - > type & = ~ BLOCK_OPT ;
if ( trb ! = & null_block ) trb - > type & = ~ BLOCK_OPT ;
if ( blb ! = & null_block ) blb - > type & = ~ BLOCK_OPT ;
if ( brb ! = & null_block ) brb - > type & = ~ BLOCK_OPT ;
change + + ;
}
}
}
av_log ( NULL , AV_LOG_ERROR , " pass:%d changed:%d \n " , pass , change ) ;
if ( ! change )
break ;
}
2006-01-17 09:27:39 +01:00
if ( s - > block_max_depth = = 1 ) {
int change = 0 ;
for ( mb_y = 0 ; mb_y < b_height ; mb_y + = 2 ) {
for ( mb_x = 0 ; mb_x < b_width ; mb_x + = 2 ) {
int dia_change , i , j ;
int best_rd , init_rd ;
const int index = mb_x + mb_y * b_stride ;
BlockNode * b [ 4 ] ;
b [ 0 ] = & s - > block [ index ] ;
b [ 1 ] = b [ 0 ] + 1 ;
b [ 2 ] = b [ 0 ] + b_stride ;
b [ 3 ] = b [ 2 ] + 1 ;
if ( same_block ( b [ 0 ] , b [ 1 ] ) & &
same_block ( b [ 0 ] , b [ 2 ] ) & &
same_block ( b [ 0 ] , b [ 3 ] ) )
continue ;
if ( ! s - > me_cache_generation )
memset ( s - > me_cache , 0 , sizeof ( s - > me_cache ) ) ;
s - > me_cache_generation + = 1 < < 22 ;
init_rd = best_rd = get_4block_rd ( s , mb_x , mb_y , 0 ) ;
check_4block_inter ( s , mb_x , mb_y ,
( b [ 0 ] - > mx + b [ 1 ] - > mx + b [ 2 ] - > mx + b [ 3 ] - > mx + 2 ) > > 2 ,
( b [ 0 ] - > my + b [ 1 ] - > my + b [ 2 ] - > my + b [ 3 ] - > my + 2 ) > > 2 , & best_rd ) ;
for ( i = 0 ; i < 4 ; i + + )
if ( ! ( b [ i ] - > type & BLOCK_INTRA ) )
check_4block_inter ( s , mb_x , mb_y , b [ i ] - > mx , b [ i ] - > my , & best_rd ) ;
if ( init_rd ! = best_rd )
change + + ;
}
}
av_log ( NULL , AV_LOG_ERROR , " pass:4mv changed:%d \n " , change * 4 ) ;
}
2005-12-28 16:43:53 +01:00
}
2004-07-26 01:36:44 +02:00
static void quantize ( SnowContext * s , SubBand * b , DWTELEM * src , int stride , int bias ) {
const int level = b - > level ;
const int w = b - > width ;
const int h = b - > height ;
2005-04-09 21:48:59 +02:00
const int qlog = clip ( s - > qlog + b - > qlog , 0 , QROOT * 16 ) ;
const int qmul = qexp [ qlog & ( QROOT - 1 ) ] < < ( qlog > > QSHIFT ) ;
2004-07-27 19:38:53 +02:00
int x , y , thres1 , thres2 ;
2005-09-22 01:09:16 +02:00
// START_TIMER
2004-07-26 01:36:44 +02:00
2004-08-02 21:09:28 +02:00
if ( s - > qlog = = LOSSLESS_QLOG ) return ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
bias = bias ? 0 : ( 3 * qmul ) > > 3 ;
2004-07-27 19:38:53 +02:00
thres1 = ( ( qmul - bias ) > > QEXPSHIFT ) - 1 ;
thres2 = 2 * thres1 ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
if ( ! bias ) {
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
2004-07-27 19:38:53 +02:00
int i = src [ x + y * stride ] ;
2005-12-17 19:14:38 +01:00
2004-07-27 19:38:53 +02:00
if ( ( unsigned ) ( i + thres1 ) > thres2 ) {
if ( i > = 0 ) {
i < < = QEXPSHIFT ;
i / = qmul ; //FIXME optimize
src [ x + y * stride ] = i ;
} else {
i = - i ;
i < < = QEXPSHIFT ;
i / = qmul ; //FIXME optimize
src [ x + y * stride ] = - i ;
}
} else
src [ x + y * stride ] = 0 ;
2004-07-26 01:36:44 +02:00
}
}
} else {
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
2005-12-17 19:14:38 +01:00
int i = src [ x + y * stride ] ;
2004-07-27 19:38:53 +02:00
if ( ( unsigned ) ( i + thres1 ) > thres2 ) {
if ( i > = 0 ) {
i < < = QEXPSHIFT ;
i = ( i + bias ) / qmul ; //FIXME optimize
src [ x + y * stride ] = i ;
} else {
i = - i ;
i < < = QEXPSHIFT ;
i = ( i + bias ) / qmul ; //FIXME optimize
src [ x + y * stride ] = - i ;
}
} else
src [ x + y * stride ] = 0 ;
2004-07-26 01:36:44 +02:00
}
}
}
2004-07-27 19:38:53 +02:00
if ( level + 1 = = s - > spatial_decomposition_count ) {
// STOP_TIMER("quantize")
}
2004-07-26 01:36:44 +02:00
}
2005-04-25 12:03:01 +02:00
static void dequantize_slice_buffered ( SnowContext * s , slice_buffer * sb , SubBand * b , DWTELEM * src , int stride , int start_y , int end_y ) {
2005-04-03 17:43:57 +02:00
const int w = b - > width ;
2005-04-09 21:48:59 +02:00
const int qlog = clip ( s - > qlog + b - > qlog , 0 , QROOT * 16 ) ;
const int qmul = qexp [ qlog & ( QROOT - 1 ) ] < < ( qlog > > QSHIFT ) ;
2005-04-03 17:43:57 +02:00
const int qadd = ( s - > qbias * qmul ) > > QBIAS_SHIFT ;
int x , y ;
START_TIMER
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
if ( s - > qlog = = LOSSLESS_QLOG ) return ;
2005-12-17 19:14:38 +01:00
2005-04-25 12:03:01 +02:00
for ( y = start_y ; y < end_y ; y + + ) {
2005-04-03 17:43:57 +02:00
// DWTELEM * line = slice_buffer_get_line_from_address(sb, src + (y * stride));
DWTELEM * line = slice_buffer_get_line ( sb , ( y * b - > stride_line ) + b - > buf_y_offset ) + b - > buf_x_offset ;
for ( x = 0 ; x < w ; x + + ) {
int i = line [ x ] ;
if ( i < 0 ) {
line [ x ] = - ( ( - i * qmul + qadd ) > > ( QEXPSHIFT ) ) ; //FIXME try different bias
} else if ( i > 0 ) {
line [ x ] = ( ( i * qmul + qadd ) > > ( QEXPSHIFT ) ) ;
}
}
}
if ( w > 200 /*level+1 == s->spatial_decomposition_count*/ ) {
STOP_TIMER ( " dquant " )
}
}
2004-07-26 01:36:44 +02:00
static void dequantize ( SnowContext * s , SubBand * b , DWTELEM * src , int stride ) {
const int w = b - > width ;
const int h = b - > height ;
2005-04-09 21:48:59 +02:00
const int qlog = clip ( s - > qlog + b - > qlog , 0 , QROOT * 16 ) ;
const int qmul = qexp [ qlog & ( QROOT - 1 ) ] < < ( qlog > > QSHIFT ) ;
2004-07-26 01:36:44 +02:00
const int qadd = ( s - > qbias * qmul ) > > QBIAS_SHIFT ;
int x , y ;
2004-08-27 23:16:51 +02:00
START_TIMER
2005-12-17 19:14:38 +01:00
2004-08-02 21:09:28 +02:00
if ( s - > qlog = = LOSSLESS_QLOG ) return ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
int i = src [ x + y * stride ] ;
if ( i < 0 ) {
src [ x + y * stride ] = - ( ( - i * qmul + qadd ) > > ( QEXPSHIFT ) ) ; //FIXME try different bias
} else if ( i > 0 ) {
src [ x + y * stride ] = ( ( i * qmul + qadd ) > > ( QEXPSHIFT ) ) ;
}
}
}
2004-08-27 23:16:51 +02:00
if ( w > 200 /*level+1 == s->spatial_decomposition_count*/ ) {
STOP_TIMER ( " dquant " )
}
2004-07-26 01:36:44 +02:00
}
static void decorrelate ( SnowContext * s , SubBand * b , DWTELEM * src , int stride , int inverse , int use_median ) {
const int w = b - > width ;
const int h = b - > height ;
int x , y ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( y = h - 1 ; y > = 0 ; y - - ) {
for ( x = w - 1 ; x > = 0 ; x - - ) {
int i = x + y * stride ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
if ( x ) {
if ( use_median ) {
if ( y & & x + 1 < w ) src [ i ] - = mid_pred ( src [ i - 1 ] , src [ i - stride ] , src [ i - stride + 1 ] ) ;
else src [ i ] - = src [ i - 1 ] ;
} else {
if ( y ) src [ i ] - = mid_pred ( src [ i - 1 ] , src [ i - stride ] , src [ i - 1 ] + src [ i - stride ] - src [ i - 1 - stride ] ) ;
else src [ i ] - = src [ i - 1 ] ;
}
} else {
if ( y ) src [ i ] - = src [ i - stride ] ;
}
}
}
}
2005-04-25 12:03:01 +02:00
static void correlate_slice_buffered ( SnowContext * s , slice_buffer * sb , SubBand * b , DWTELEM * src , int stride , int inverse , int use_median , int start_y , int end_y ) {
2005-04-03 17:43:57 +02:00
const int w = b - > width ;
int x , y ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
// START_TIMER
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
DWTELEM * line ;
DWTELEM * prev ;
2005-12-17 19:14:38 +01:00
2005-04-25 12:03:01 +02:00
if ( start_y ! = 0 )
line = slice_buffer_get_line ( sb , ( ( start_y - 1 ) * b - > stride_line ) + b - > buf_y_offset ) + b - > buf_x_offset ;
2005-12-17 19:14:38 +01:00
2005-04-25 12:03:01 +02:00
for ( y = start_y ; y < end_y ; y + + ) {
2005-04-03 17:43:57 +02:00
prev = line ;
// line = slice_buffer_get_line_from_address(sb, src + (y * stride));
line = slice_buffer_get_line ( sb , ( y * b - > stride_line ) + b - > buf_y_offset ) + b - > buf_x_offset ;
for ( x = 0 ; x < w ; x + + ) {
if ( x ) {
if ( use_median ) {
if ( y & & x + 1 < w ) line [ x ] + = mid_pred ( line [ x - 1 ] , prev [ x ] , prev [ x + 1 ] ) ;
else line [ x ] + = line [ x - 1 ] ;
} else {
if ( y ) line [ x ] + = mid_pred ( line [ x - 1 ] , prev [ x ] , line [ x - 1 ] + prev [ x ] - prev [ x - 1 ] ) ;
else line [ x ] + = line [ x - 1 ] ;
}
} else {
if ( y ) line [ x ] + = prev [ x ] ;
}
}
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
// STOP_TIMER("correlate")
}
2004-07-26 01:36:44 +02:00
static void correlate ( SnowContext * s , SubBand * b , DWTELEM * src , int stride , int inverse , int use_median ) {
const int w = b - > width ;
const int h = b - > height ;
int x , y ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
int i = x + y * stride ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
if ( x ) {
if ( use_median ) {
if ( y & & x + 1 < w ) src [ i ] + = mid_pred ( src [ i - 1 ] , src [ i - stride ] , src [ i - stride + 1 ] ) ;
else src [ i ] + = src [ i - 1 ] ;
} else {
if ( y ) src [ i ] + = mid_pred ( src [ i - 1 ] , src [ i - stride ] , src [ i - 1 ] + src [ i - stride ] - src [ i - 1 - stride ] ) ;
else src [ i ] + = src [ i - 1 ] ;
}
} else {
if ( y ) src [ i ] + = src [ i - stride ] ;
}
}
}
}
static void encode_header ( SnowContext * s ) {
int plane_index , level , orientation ;
2005-12-17 19:14:38 +01:00
uint8_t kstate [ 32 ] ;
memset ( kstate , MID_STATE , sizeof ( kstate ) ) ;
2004-07-26 01:36:44 +02:00
2004-10-30 04:16:27 +02:00
put_rac ( & s - > c , kstate , s - > keyframe ) ;
2004-08-31 18:28:49 +02:00
if ( s - > keyframe | | s - > always_reset )
reset_contexts ( s ) ;
2004-07-26 01:36:44 +02:00
if ( s - > keyframe ) {
put_symbol ( & s - > c , s - > header_state , s - > version , 0 ) ;
2004-10-30 04:16:27 +02:00
put_rac ( & s - > c , s - > header_state , s - > always_reset ) ;
2004-07-26 01:36:44 +02:00
put_symbol ( & s - > c , s - > header_state , s - > temporal_decomposition_type , 0 ) ;
put_symbol ( & s - > c , s - > header_state , s - > temporal_decomposition_count , 0 ) ;
put_symbol ( & s - > c , s - > header_state , s - > spatial_decomposition_count , 0 ) ;
put_symbol ( & s - > c , s - > header_state , s - > colorspace_type , 0 ) ;
put_symbol ( & s - > c , s - > header_state , s - > chroma_h_shift , 0 ) ;
put_symbol ( & s - > c , s - > header_state , s - > chroma_v_shift , 0 ) ;
2004-10-30 04:16:27 +02:00
put_rac ( & s - > c , s - > header_state , s - > spatial_scalability ) ;
// put_rac(&s->c, s->header_state, s->rate_scalability);
2004-07-26 01:36:44 +02:00
for ( plane_index = 0 ; plane_index < 2 ; plane_index + + ) {
for ( level = 0 ; level < s - > spatial_decomposition_count ; level + + ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
if ( orientation = = 2 ) continue ;
put_symbol ( & s - > c , s - > header_state , s - > plane [ plane_index ] . band [ level ] [ orientation ] . qlog , 1 ) ;
}
}
}
}
put_symbol ( & s - > c , s - > header_state , s - > spatial_decomposition_type , 0 ) ;
2005-12-17 19:14:38 +01:00
put_symbol ( & s - > c , s - > header_state , s - > qlog , 1 ) ;
put_symbol ( & s - > c , s - > header_state , s - > mv_scale , 0 ) ;
2004-07-26 01:36:44 +02:00
put_symbol ( & s - > c , s - > header_state , s - > qbias , 1 ) ;
2004-08-26 22:04:54 +02:00
put_symbol ( & s - > c , s - > header_state , s - > block_max_depth , 0 ) ;
2004-07-26 01:36:44 +02:00
}
static int decode_header ( SnowContext * s ) {
int plane_index , level , orientation ;
2004-10-30 04:16:27 +02:00
uint8_t kstate [ 32 ] ;
2005-12-17 19:14:38 +01:00
memset ( kstate , MID_STATE , sizeof ( kstate ) ) ;
2004-07-26 01:36:44 +02:00
2004-10-30 04:16:27 +02:00
s - > keyframe = get_rac ( & s - > c , kstate ) ;
2004-08-31 18:28:49 +02:00
if ( s - > keyframe | | s - > always_reset )
reset_contexts ( s ) ;
2004-07-26 01:36:44 +02:00
if ( s - > keyframe ) {
s - > version = get_symbol ( & s - > c , s - > header_state , 0 ) ;
if ( s - > version > 0 ) {
av_log ( s - > avctx , AV_LOG_ERROR , " version %d not supported " , s - > version ) ;
return - 1 ;
}
2004-10-30 04:16:27 +02:00
s - > always_reset = get_rac ( & s - > c , s - > header_state ) ;
2004-07-26 01:36:44 +02:00
s - > temporal_decomposition_type = get_symbol ( & s - > c , s - > header_state , 0 ) ;
s - > temporal_decomposition_count = get_symbol ( & s - > c , s - > header_state , 0 ) ;
s - > spatial_decomposition_count = get_symbol ( & s - > c , s - > header_state , 0 ) ;
s - > colorspace_type = get_symbol ( & s - > c , s - > header_state , 0 ) ;
s - > chroma_h_shift = get_symbol ( & s - > c , s - > header_state , 0 ) ;
s - > chroma_v_shift = get_symbol ( & s - > c , s - > header_state , 0 ) ;
2004-10-30 04:16:27 +02:00
s - > spatial_scalability = get_rac ( & s - > c , s - > header_state ) ;
// s->rate_scalability= get_rac(&s->c, s->header_state);
2004-07-26 01:36:44 +02:00
for ( plane_index = 0 ; plane_index < 3 ; plane_index + + ) {
for ( level = 0 ; level < s - > spatial_decomposition_count ; level + + ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
int q ;
if ( plane_index = = 2 ) q = s - > plane [ 1 ] . band [ level ] [ orientation ] . qlog ;
else if ( orientation = = 2 ) q = s - > plane [ plane_index ] . band [ level ] [ 1 ] . qlog ;
else q = get_symbol ( & s - > c , s - > header_state , 1 ) ;
s - > plane [ plane_index ] . band [ level ] [ orientation ] . qlog = q ;
}
}
}
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
s - > spatial_decomposition_type = get_symbol ( & s - > c , s - > header_state , 0 ) ;
if ( s - > spatial_decomposition_type > 2 ) {
av_log ( s - > avctx , AV_LOG_ERROR , " spatial_decomposition_type %d not supported " , s - > spatial_decomposition_type ) ;
return - 1 ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
s - > qlog = get_symbol ( & s - > c , s - > header_state , 1 ) ;
s - > mv_scale = get_symbol ( & s - > c , s - > header_state , 0 ) ;
s - > qbias = get_symbol ( & s - > c , s - > header_state , 1 ) ;
2004-08-26 22:04:54 +02:00
s - > block_max_depth = get_symbol ( & s - > c , s - > header_state , 0 ) ;
2005-12-03 15:20:03 +01:00
if ( s - > block_max_depth > 1 ) {
av_log ( s - > avctx , AV_LOG_ERROR , " block_max_depth= %d is too large " , s - > block_max_depth ) ;
s - > block_max_depth = 0 ;
return - 1 ;
}
2004-07-26 01:36:44 +02:00
return 0 ;
}
2006-01-31 00:33:19 +01:00
static void init_qexp ( void ) {
2005-04-09 21:48:59 +02:00
int i ;
double v = 128 ;
for ( i = 0 ; i < QROOT ; i + + ) {
qexp [ i ] = lrintf ( v ) ;
2005-12-17 19:14:38 +01:00
v * = pow ( 2 , 1.0 / QROOT ) ;
2005-04-09 21:48:59 +02:00
}
}
2004-07-26 01:36:44 +02:00
static int common_init ( AVCodecContext * avctx ) {
SnowContext * s = avctx - > priv_data ;
int width , height ;
int level , orientation , plane_index , dec ;
s - > avctx = avctx ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
dsputil_init ( & s - > dsp , avctx ) ;
# define mcf(dx,dy)\
s - > dsp . put_qpel_pixels_tab [ 0 ] [ dy + dx / 4 ] = \
s - > dsp . put_no_rnd_qpel_pixels_tab [ 0 ] [ dy + dx / 4 ] = \
2004-09-13 01:09:54 +02:00
s - > dsp . put_h264_qpel_pixels_tab [ 0 ] [ dy + dx / 4 ] ; \
s - > dsp . put_qpel_pixels_tab [ 1 ] [ dy + dx / 4 ] = \
s - > dsp . put_no_rnd_qpel_pixels_tab [ 1 ] [ dy + dx / 4 ] = \
s - > dsp . put_h264_qpel_pixels_tab [ 1 ] [ dy + dx / 4 ] ;
2004-07-26 01:36:44 +02:00
mcf ( 0 , 0 )
mcf ( 4 , 0 )
mcf ( 8 , 0 )
mcf ( 12 , 0 )
mcf ( 0 , 4 )
mcf ( 4 , 4 )
mcf ( 8 , 4 )
mcf ( 12 , 4 )
mcf ( 0 , 8 )
mcf ( 4 , 8 )
mcf ( 8 , 8 )
mcf ( 12 , 8 )
mcf ( 0 , 12 )
mcf ( 4 , 12 )
mcf ( 8 , 12 )
mcf ( 12 , 12 )
# define mcfh(dx,dy)\
s - > dsp . put_pixels_tab [ 0 ] [ dy / 4 + dx / 8 ] = \
s - > dsp . put_no_rnd_pixels_tab [ 0 ] [ dy / 4 + dx / 8 ] = \
2004-09-13 01:09:54 +02:00
mc_block_hpel # # dx # # dy # # 16 ; \
s - > dsp . put_pixels_tab [ 1 ] [ dy / 4 + dx / 8 ] = \
s - > dsp . put_no_rnd_pixels_tab [ 1 ] [ dy / 4 + dx / 8 ] = \
mc_block_hpel # # dx # # dy # # 8 ;
2004-07-26 01:36:44 +02:00
mcfh ( 0 , 0 )
mcfh ( 8 , 0 )
mcfh ( 0 , 8 )
mcfh ( 8 , 8 )
2005-04-09 21:48:59 +02:00
if ( ! qexp [ 0 ] )
init_qexp ( ) ;
2004-07-26 01:36:44 +02:00
dec = s - > spatial_decomposition_count = 5 ;
s - > spatial_decomposition_type = avctx - > prediction_method ; //FIXME add decorrelator type r transform_type
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
s - > chroma_h_shift = 1 ; //FIXME XXX
s - > chroma_v_shift = 1 ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
// dec += FFMAX(s->chroma_h_shift, s->chroma_v_shift);
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
width = s - > avctx - > width ;
height = s - > avctx - > height ;
s - > spatial_dwt_buffer = av_mallocz ( width * height * sizeof ( DWTELEM ) ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
s - > mv_scale = ( s - > avctx - > flags & CODEC_FLAG_QPEL ) ? 2 : 4 ;
2004-08-26 22:04:54 +02:00
s - > block_max_depth = ( s - > avctx - > flags & CODEC_FLAG_4MV ) ? 1 : 0 ;
2005-12-17 19:14:38 +01:00
for ( plane_index = 0 ; plane_index < 3 ; plane_index + + ) {
2004-07-26 01:36:44 +02:00
int w = s - > avctx - > width ;
int h = s - > avctx - > height ;
if ( plane_index ) {
w > > = s - > chroma_h_shift ;
h > > = s - > chroma_v_shift ;
}
s - > plane [ plane_index ] . width = w ;
s - > plane [ plane_index ] . height = h ;
2004-08-02 20:01:03 +02:00
//av_log(NULL, AV_LOG_DEBUG, "%d %d\n", w, h);
2004-07-26 01:36:44 +02:00
for ( level = s - > spatial_decomposition_count - 1 ; level > = 0 ; level - - ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
SubBand * b = & s - > plane [ plane_index ] . band [ level ] [ orientation ] ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
b - > buf = s - > spatial_dwt_buffer ;
b - > level = level ;
b - > stride = s - > plane [ plane_index ] . width < < ( s - > spatial_decomposition_count - level ) ;
b - > width = ( w + ! ( orientation & 1 ) ) > > 1 ;
b - > height = ( h + ! ( orientation > 1 ) ) > > 1 ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
b - > stride_line = 1 < < ( s - > spatial_decomposition_count - level ) ;
b - > buf_x_offset = 0 ;
b - > buf_y_offset = 0 ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
if ( orientation & 1 ) {
b - > buf + = ( w + 1 ) > > 1 ;
b - > buf_x_offset = ( w + 1 ) > > 1 ;
}
if ( orientation > 1 ) {
b - > buf + = b - > stride > > 1 ;
b - > buf_y_offset = b - > stride_line > > 1 ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
if ( level )
b - > parent = & s - > plane [ plane_index ] . band [ level - 1 ] [ orientation ] ;
2005-04-03 17:43:57 +02:00
b - > x_coeff = av_mallocz ( ( ( b - > width + 1 ) * b - > height + 1 ) * sizeof ( x_and_coeff ) ) ;
2004-07-26 01:36:44 +02:00
}
w = ( w + 1 ) > > 1 ;
h = ( h + 1 ) > > 1 ;
}
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
reset_contexts ( s ) ;
2005-12-17 19:14:38 +01:00
/*
2004-07-26 01:36:44 +02:00
width = s - > width = avctx - > width ;
height = s - > height = avctx - > height ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
assert ( width & & height ) ;
*/
s - > avctx - > get_buffer ( s - > avctx , & s - > mconly_picture ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
return 0 ;
}
static void calculate_vissual_weight ( SnowContext * s , Plane * p ) {
int width = p - > width ;
int height = p - > height ;
2004-08-28 16:24:48 +02:00
int level , orientation , x , y ;
2004-07-26 01:36:44 +02:00
for ( level = 0 ; level < s - > spatial_decomposition_count ; level + + ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
SubBand * b = & p - > band [ level ] [ orientation ] ;
DWTELEM * buf = b - > buf ;
int64_t error = 0 ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
memset ( s - > spatial_dwt_buffer , 0 , sizeof ( int ) * width * height ) ;
buf [ b - > width / 2 + b - > height / 2 * b - > stride ] = 256 * 256 ;
2004-08-03 04:34:50 +02:00
ff_spatial_idwt ( s - > spatial_dwt_buffer , width , height , width , s - > spatial_decomposition_type , s - > spatial_decomposition_count ) ;
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < height ; y + + ) {
for ( x = 0 ; x < width ; x + + ) {
int64_t d = s - > spatial_dwt_buffer [ x + y * width ] ;
error + = d * d ;
}
}
b - > qlog = ( int ) ( log ( 352256.0 / sqrt ( error ) ) / log ( pow ( 2.0 , 1.0 / QROOT ) ) + 0.5 ) ;
2004-08-03 04:34:50 +02:00
// av_log(NULL, AV_LOG_DEBUG, "%d %d %d\n", level, orientation, b->qlog/*, sqrt(error)*/);
2004-07-26 01:36:44 +02:00
}
}
}
static int encode_init ( AVCodecContext * avctx )
{
SnowContext * s = avctx - > priv_data ;
2004-08-28 16:24:48 +02:00
int plane_index ;
2004-07-26 01:36:44 +02:00
2005-05-08 22:15:42 +02:00
if ( avctx - > strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL ) {
2005-04-24 19:21:11 +02:00
av_log ( avctx , AV_LOG_ERROR , " this codec is under development, files encoded with it may not be decodable with future versions!!! \n "
2005-05-08 22:15:42 +02:00
" use vstrict=-2 / -strict -2 to use it anyway \n " ) ;
2004-07-28 02:09:03 +02:00
return - 1 ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
common_init ( avctx ) ;
2004-08-26 22:04:54 +02:00
alloc_blocks ( s ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
s - > version = 0 ;
2005-12-17 19:14:38 +01:00
2005-04-15 07:54:47 +02:00
s - > m . avctx = avctx ;
s - > m . flags = avctx - > flags ;
s - > m . bit_rate = avctx - > bit_rate ;
2004-07-26 01:36:44 +02:00
s - > m . me . scratchpad = av_mallocz ( ( avctx - > width + 64 ) * 2 * 16 * 2 * sizeof ( uint8_t ) ) ;
s - > m . me . map = av_mallocz ( ME_MAP_SIZE * sizeof ( uint32_t ) ) ;
s - > m . me . score_map = av_mallocz ( ME_MAP_SIZE * sizeof ( uint32_t ) ) ;
2006-01-12 06:47:52 +01:00
s - > m . obmc_scratchpad = av_mallocz ( MB_SIZE * MB_SIZE * 12 * sizeof ( uint32_t ) ) ;
2004-07-26 01:36:44 +02:00
h263_encode_init ( & s - > m ) ; //mv_penalty
2005-04-15 07:54:47 +02:00
if ( avctx - > flags & CODEC_FLAG_PASS1 ) {
if ( ! avctx - > stats_out )
avctx - > stats_out = av_mallocz ( 256 ) ;
}
if ( avctx - > flags & CODEC_FLAG_PASS2 ) {
if ( ff_rate_control_init ( & s - > m ) < 0 )
return - 1 ;
}
2004-07-26 01:36:44 +02:00
for ( plane_index = 0 ; plane_index < 3 ; plane_index + + ) {
calculate_vissual_weight ( s , & s - > plane [ plane_index ] ) ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
avctx - > coded_frame = & s - > current_picture ;
switch ( avctx - > pix_fmt ) {
// case PIX_FMT_YUV444P:
// case PIX_FMT_YUV422P:
case PIX_FMT_YUV420P :
case PIX_FMT_GRAY8 :
// case PIX_FMT_YUV411P:
// case PIX_FMT_YUV410P:
s - > colorspace_type = 0 ;
break ;
/* case PIX_FMT_RGBA32:
s - > colorspace = 1 ;
break ; */
default :
av_log ( avctx , AV_LOG_ERROR , " format not supported \n " ) ;
return - 1 ;
}
// avcodec_get_chroma_sub_sample(avctx->pix_fmt, &s->chroma_h_shift, &s->chroma_v_shift);
s - > chroma_h_shift = 1 ;
s - > chroma_v_shift = 1 ;
2005-12-28 16:43:53 +01:00
ff_set_cmp ( & s - > dsp , s - > dsp . me_cmp , s - > avctx - > me_cmp ) ;
ff_set_cmp ( & s - > dsp , s - > dsp . me_sub_cmp , s - > avctx - > me_sub_cmp ) ;
s - > avctx - > get_buffer ( s - > avctx , & s - > input_picture ) ;
2004-07-26 01:36:44 +02:00
return 0 ;
}
static int frame_start ( SnowContext * s ) {
AVFrame tmp ;
2004-08-25 02:26:01 +02:00
int w = s - > avctx - > width ; //FIXME round up to x16 ?
int h = s - > avctx - > height ;
2004-07-26 01:36:44 +02:00
2004-08-25 02:26:01 +02:00
if ( s - > current_picture . data [ 0 ] ) {
draw_edges ( s - > current_picture . data [ 0 ] , s - > current_picture . linesize [ 0 ] , w , h , EDGE_WIDTH ) ;
draw_edges ( s - > current_picture . data [ 1 ] , s - > current_picture . linesize [ 1 ] , w > > 1 , h > > 1 , EDGE_WIDTH / 2 ) ;
draw_edges ( s - > current_picture . data [ 2 ] , s - > current_picture . linesize [ 2 ] , w > > 1 , h > > 1 , EDGE_WIDTH / 2 ) ;
}
2004-07-26 01:36:44 +02:00
tmp = s - > last_picture ;
s - > last_picture = s - > current_picture ;
s - > current_picture = tmp ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
s - > current_picture . reference = 1 ;
if ( s - > avctx - > get_buffer ( s - > avctx , & s - > current_picture ) < 0 ) {
av_log ( s - > avctx , AV_LOG_ERROR , " get_buffer() failed \n " ) ;
return - 1 ;
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
return 0 ;
}
static int encode_frame ( AVCodecContext * avctx , unsigned char * buf , int buf_size , void * data ) {
SnowContext * s = avctx - > priv_data ;
2004-10-30 04:16:27 +02:00
RangeCoder * const c = & s - > c ;
2004-07-26 01:36:44 +02:00
AVFrame * pict = data ;
const int width = s - > avctx - > width ;
const int height = s - > avctx - > height ;
2005-12-28 16:43:53 +01:00
int level , orientation , plane_index , i , y ;
2004-07-26 01:36:44 +02:00
2004-10-30 04:16:27 +02:00
ff_init_range_encoder ( c , buf , buf_size ) ;
ff_build_rac_states ( c , 0.05 * ( 1LL < < 32 ) , 256 - 8 ) ;
2005-12-17 19:14:38 +01:00
2005-12-28 16:43:53 +01:00
for ( i = 0 ; i < 3 ; i + + ) {
int shift = ! ! i ;
for ( y = 0 ; y < ( height > > shift ) ; y + + )
memcpy ( & s - > input_picture . data [ i ] [ y * s - > input_picture . linesize [ i ] ] ,
& pict - > data [ i ] [ y * pict - > linesize [ i ] ] ,
width > > shift ) ;
}
s - > new_picture = * pict ;
2004-07-26 01:36:44 +02:00
2005-04-15 07:54:47 +02:00
if ( avctx - > flags & CODEC_FLAG_PASS2 ) {
s - > m . pict_type =
pict - > pict_type = s - > m . rc_context . entry [ avctx - > frame_number ] . new_pict_type ;
s - > keyframe = pict - > pict_type = = FF_I_TYPE ;
s - > m . picture_number = avctx - > frame_number ;
2005-12-19 19:26:17 +01:00
pict - > quality = ff_rate_estimate_qscale ( & s - > m , 0 ) ;
2005-04-15 07:54:47 +02:00
} else {
s - > keyframe = avctx - > gop_size = = 0 | | avctx - > frame_number % avctx - > gop_size = = 0 ;
pict - > pict_type = s - > keyframe ? FF_I_TYPE : FF_P_TYPE ;
}
2005-12-17 19:14:38 +01:00
2004-08-02 21:09:28 +02:00
if ( pict - > quality ) {
s - > qlog = rint ( QROOT * log ( pict - > quality / ( float ) FF_QP2LAMBDA ) / log ( 2 ) ) ;
//<64 >60
2005-04-09 21:48:59 +02:00
s - > qlog + = 61 * QROOT / 8 ;
2004-08-02 21:09:28 +02:00
} else {
s - > qlog = LOSSLESS_QLOG ;
}
2004-07-26 01:36:44 +02:00
frame_start ( s ) ;
2004-08-31 18:28:49 +02:00
s - > current_picture . key_frame = s - > keyframe ;
2004-07-26 01:36:44 +02:00
2005-04-15 07:54:47 +02:00
s - > m . current_picture_ptr = & s - > m . current_picture ;
2004-07-26 01:36:44 +02:00
if ( pict - > pict_type = = P_TYPE ) {
int block_width = ( width + 15 ) > > 4 ;
int block_height = ( height + 15 ) > > 4 ;
int stride = s - > current_picture . linesize [ 0 ] ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
assert ( s - > current_picture . data [ 0 ] ) ;
assert ( s - > last_picture . data [ 0 ] ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
s - > m . avctx = s - > avctx ;
s - > m . current_picture . data [ 0 ] = s - > current_picture . data [ 0 ] ;
s - > m . last_picture . data [ 0 ] = s - > last_picture . data [ 0 ] ;
s - > m . new_picture . data [ 0 ] = s - > input_picture . data [ 0 ] ;
s - > m . last_picture_ptr = & s - > m . last_picture ;
s - > m . linesize =
s - > m . last_picture . linesize [ 0 ] =
s - > m . new_picture . linesize [ 0 ] =
s - > m . current_picture . linesize [ 0 ] = stride ;
2004-08-26 22:04:54 +02:00
s - > m . uvlinesize = s - > current_picture . linesize [ 1 ] ;
2004-07-26 01:36:44 +02:00
s - > m . width = width ;
s - > m . height = height ;
s - > m . mb_width = block_width ;
s - > m . mb_height = block_height ;
s - > m . mb_stride = s - > m . mb_width + 1 ;
s - > m . b8_stride = 2 * s - > m . mb_width + 1 ;
s - > m . f_code = 1 ;
s - > m . pict_type = pict - > pict_type ;
s - > m . me_method = s - > avctx - > me_method ;
s - > m . me . scene_change_score = 0 ;
s - > m . flags = s - > avctx - > flags ;
s - > m . quarter_sample = ( s - > avctx - > flags & CODEC_FLAG_QPEL ) ! = 0 ;
s - > m . out_format = FMT_H263 ;
s - > m . unrestricted_mv = 1 ;
2004-08-26 22:04:54 +02:00
s - > lambda = s - > m . lambda = pict - > quality * 3 / 2 ; //FIXME bug somewhere else
2004-07-26 01:36:44 +02:00
s - > m . qscale = ( s - > m . lambda * 139 + FF_LAMBDA_SCALE * 64 ) > > ( FF_LAMBDA_SHIFT + 7 ) ;
2004-08-26 22:04:54 +02:00
s - > lambda2 = s - > m . lambda2 = ( s - > m . lambda * s - > m . lambda + FF_LAMBDA_SCALE / 2 ) > > FF_LAMBDA_SHIFT ;
2004-07-26 01:36:44 +02:00
s - > m . dsp = s - > dsp ; //move
ff_init_me ( & s - > m ) ;
2005-12-28 16:43:53 +01:00
s - > dsp = s - > m . dsp ;
2004-07-26 01:36:44 +02:00
}
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
redo_frame :
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
s - > qbias = pict - > pict_type = = P_TYPE ? 2 : 0 ;
encode_header ( s ) ;
2005-04-15 07:54:47 +02:00
s - > m . misc_bits = 8 * ( s - > c . bytestream - s - > c . bytestream_start ) ;
2004-08-26 22:04:54 +02:00
encode_blocks ( s ) ;
2005-04-15 07:54:47 +02:00
s - > m . mv_bits = 8 * ( s - > c . bytestream - s - > c . bytestream_start ) - s - > m . misc_bits ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( plane_index = 0 ; plane_index < 3 ; plane_index + + ) {
Plane * p = & s - > plane [ plane_index ] ;
int w = p - > width ;
int h = p - > height ;
int x , y ;
2004-08-28 16:24:48 +02:00
// int bits= put_bits_count(&s->c.pb);
2004-07-26 01:36:44 +02:00
//FIXME optimize
if ( pict - > data [ plane_index ] ) //FIXME gray hack
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
2004-09-22 01:10:10 +02:00
s - > spatial_dwt_buffer [ y * w + x ] = pict - > data [ plane_index ] [ y * pict - > linesize [ plane_index ] + x ] < < FRAC_BITS ;
2004-07-26 01:36:44 +02:00
}
}
predict_plane ( s , s - > spatial_dwt_buffer , plane_index , 0 ) ;
2005-12-17 19:14:38 +01:00
if ( plane_index = = 0
& & pict - > pict_type = = P_TYPE
2004-08-26 22:04:54 +02:00
& & s - > m . me . scene_change_score > s - > avctx - > scenechange_threshold ) {
2004-10-30 04:16:27 +02:00
ff_init_range_encoder ( c , buf , buf_size ) ;
ff_build_rac_states ( c , 0.05 * ( 1LL < < 32 ) , 256 - 8 ) ;
2004-08-26 22:04:54 +02:00
pict - > pict_type = FF_I_TYPE ;
s - > keyframe = 1 ;
reset_contexts ( s ) ;
goto redo_frame ;
}
2005-12-17 19:14:38 +01:00
2004-08-02 21:09:28 +02:00
if ( s - > qlog = = LOSSLESS_QLOG ) {
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
2005-04-05 11:37:43 +02:00
s - > spatial_dwt_buffer [ y * w + x ] = ( s - > spatial_dwt_buffer [ y * w + x ] + ( 1 < < ( FRAC_BITS - 1 ) ) - 1 ) > > FRAC_BITS ;
2004-08-02 21:09:28 +02:00
}
}
}
2005-12-17 19:14:38 +01:00
2004-08-03 04:34:50 +02:00
ff_spatial_dwt ( s - > spatial_dwt_buffer , w , h , w , s - > spatial_decomposition_type , s - > spatial_decomposition_count ) ;
2004-08-02 21:09:28 +02:00
2004-07-26 01:36:44 +02:00
for ( level = 0 ; level < s - > spatial_decomposition_count ; level + + ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
SubBand * b = & p - > band [ level ] [ orientation ] ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
quantize ( s , b , b - > buf , b - > stride , s - > qbias ) ;
if ( orientation = = 0 )
decorrelate ( s , b , b - > buf , b - > stride , pict - > pict_type = = P_TYPE , 0 ) ;
encode_subband ( s , b , b - > buf , b - > parent ? b - > parent - > buf : NULL , b - > stride , orientation ) ;
assert ( b - > parent = = NULL | | b - > parent - > stride = = b - > stride * 2 ) ;
if ( orientation = = 0 )
correlate ( s , b , b - > buf , b - > stride , 1 , 0 ) ;
}
}
// av_log(NULL, AV_LOG_DEBUG, "plane:%d bits:%d\n", plane_index, put_bits_count(&s->c.pb) - bits);
for ( level = 0 ; level < s - > spatial_decomposition_count ; level + + ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
SubBand * b = & p - > band [ level ] [ orientation ] ;
dequantize ( s , b , b - > buf , b - > stride ) ;
}
}
2004-08-02 21:09:28 +02:00
2004-08-03 04:34:50 +02:00
ff_spatial_idwt ( s - > spatial_dwt_buffer , w , h , w , s - > spatial_decomposition_type , s - > spatial_decomposition_count ) ;
2004-08-02 21:09:28 +02:00
if ( s - > qlog = = LOSSLESS_QLOG ) {
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
2004-09-22 01:10:10 +02:00
s - > spatial_dwt_buffer [ y * w + x ] < < = FRAC_BITS ;
2004-08-02 21:09:28 +02:00
}
}
}
2004-09-23 14:20:41 +02:00
{ START_TIMER
2004-07-26 01:36:44 +02:00
predict_plane ( s , s - > spatial_dwt_buffer , plane_index , 1 ) ;
2004-09-23 14:20:41 +02:00
STOP_TIMER ( " pred-conv " ) }
2004-07-26 01:36:44 +02:00
if ( s - > avctx - > flags & CODEC_FLAG_PSNR ) {
int64_t error = 0 ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
if ( pict - > data [ plane_index ] ) //FIXME gray hack
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
2004-08-02 21:09:28 +02:00
int d = s - > current_picture . data [ plane_index ] [ y * s - > current_picture . linesize [ plane_index ] + x ] - pict - > data [ plane_index ] [ y * pict - > linesize [ plane_index ] + x ] ;
2004-07-26 01:36:44 +02:00
error + = d * d ;
}
}
s - > avctx - > error [ plane_index ] + = error ;
2004-09-15 03:55:04 +02:00
s - > current_picture . error [ plane_index ] = error ;
2004-07-26 01:36:44 +02:00
}
}
if ( s - > last_picture . data [ 0 ] )
avctx - > release_buffer ( avctx , & s - > last_picture ) ;
2005-04-15 07:54:47 +02:00
s - > current_picture . coded_picture_number = avctx - > frame_number ;
s - > current_picture . pict_type = pict - > pict_type ;
s - > current_picture . quality = pict - > quality ;
if ( avctx - > flags & CODEC_FLAG_PASS1 ) {
s - > m . p_tex_bits = 8 * ( s - > c . bytestream - s - > c . bytestream_start ) - s - > m . misc_bits - s - > m . mv_bits ;
s - > m . current_picture . display_picture_number =
s - > m . current_picture . coded_picture_number = avctx - > frame_number ;
s - > m . pict_type = pict - > pict_type ;
s - > m . current_picture . quality = pict - > quality ;
ff_write_pass1_stats ( & s - > m ) ;
}
if ( avctx - > flags & CODEC_FLAG_PASS2 ) {
s - > m . total_bits + = 8 * ( s - > c . bytestream - s - > c . bytestream_start ) ;
}
2004-07-26 01:36:44 +02:00
emms_c ( ) ;
2005-12-17 19:14:38 +01:00
2004-10-30 04:16:27 +02:00
return ff_rac_terminate ( c ) ;
2004-07-26 01:36:44 +02:00
}
static void common_end ( SnowContext * s ) {
2004-08-27 21:52:32 +02:00
int plane_index , level , orientation ;
2004-07-26 01:36:44 +02:00
av_freep ( & s - > spatial_dwt_buffer ) ;
2005-12-17 19:14:38 +01:00
av_freep ( & s - > m . me . scratchpad ) ;
2004-07-26 01:36:44 +02:00
av_freep ( & s - > m . me . map ) ;
av_freep ( & s - > m . me . score_map ) ;
2006-01-12 06:47:52 +01:00
av_freep ( & s - > m . obmc_scratchpad ) ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
av_freep ( & s - > block ) ;
2004-08-27 21:52:32 +02:00
2005-12-17 19:14:38 +01:00
for ( plane_index = 0 ; plane_index < 3 ; plane_index + + ) {
2004-08-27 21:52:32 +02:00
for ( level = s - > spatial_decomposition_count - 1 ; level > = 0 ; level - - ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
SubBand * b = & s - > plane [ plane_index ] . band [ level ] [ orientation ] ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
av_freep ( & b - > x_coeff ) ;
2004-08-27 21:52:32 +02:00
}
}
}
2004-07-26 01:36:44 +02:00
}
static int encode_end ( AVCodecContext * avctx )
{
SnowContext * s = avctx - > priv_data ;
common_end ( s ) ;
2005-04-15 07:54:47 +02:00
av_free ( avctx - > stats_out ) ;
2004-07-26 01:36:44 +02:00
return 0 ;
}
static int decode_init ( AVCodecContext * avctx )
{
2005-04-03 17:43:57 +02:00
SnowContext * s = avctx - > priv_data ;
int block_size ;
2005-12-17 19:14:38 +01:00
2005-04-25 20:29:06 +02:00
avctx - > pix_fmt = PIX_FMT_YUV420P ;
2004-07-26 01:36:44 +02:00
common_init ( avctx ) ;
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
block_size = MB_SIZE > > s - > block_max_depth ;
2006-01-07 04:39:10 +01:00
slice_buffer_init ( & s - > sb , s - > plane [ 0 ] . height , ( block_size ) + ( s - > spatial_decomposition_count * ( s - > spatial_decomposition_count + 3 ) ) + 1 , s - > plane [ 0 ] . width , s - > spatial_dwt_buffer ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
return 0 ;
}
static int decode_frame ( AVCodecContext * avctx , void * data , int * data_size , uint8_t * buf , int buf_size ) {
SnowContext * s = avctx - > priv_data ;
2004-10-30 04:16:27 +02:00
RangeCoder * const c = & s - > c ;
2004-07-26 01:36:44 +02:00
int bytes_read ;
AVFrame * picture = data ;
2004-08-28 16:24:48 +02:00
int level , orientation , plane_index ;
2004-07-26 01:36:44 +02:00
2004-10-30 04:16:27 +02:00
ff_init_range_decoder ( c , buf , buf_size ) ;
ff_build_rac_states ( c , 0.05 * ( 1LL < < 32 ) , 256 - 8 ) ;
2004-07-26 01:36:44 +02:00
s - > current_picture . pict_type = FF_I_TYPE ; //FIXME I vs. P
decode_header ( s ) ;
2004-08-26 22:04:54 +02:00
if ( ! s - > block ) alloc_blocks ( s ) ;
2004-07-26 01:36:44 +02:00
frame_start ( s ) ;
//keyframe flag dupliaction mess FIXME
if ( avctx - > debug & FF_DEBUG_PICT_INFO )
av_log ( avctx , AV_LOG_ERROR , " keyframe:%d qlog:%d \n " , s - > keyframe , s - > qlog ) ;
2005-12-17 19:14:38 +01:00
2004-08-26 22:04:54 +02:00
decode_blocks ( s ) ;
2004-07-26 01:36:44 +02:00
for ( plane_index = 0 ; plane_index < 3 ; plane_index + + ) {
Plane * p = & s - > plane [ plane_index ] ;
int w = p - > width ;
int h = p - > height ;
int x , y ;
2005-04-03 17:43:57 +02:00
int decode_state [ MAX_DECOMPOSITIONS ] [ 4 ] [ 1 ] ; /* Stored state info for unpack_coeffs. 1 variable per instance. */
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
if ( s - > avctx - > debug & 2048 ) {
memset ( s - > spatial_dwt_buffer , 0 , sizeof ( DWTELEM ) * w * h ) ;
predict_plane ( s , s - > spatial_dwt_buffer , plane_index , 1 ) ;
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
2004-09-23 14:20:41 +02:00
int v = s - > current_picture . data [ plane_index ] [ y * s - > current_picture . linesize [ plane_index ] + x ] ;
2004-07-26 01:36:44 +02:00
s - > mconly_picture . data [ plane_index ] [ y * s - > mconly_picture . linesize [ plane_index ] + x ] = v ;
}
}
}
2005-04-03 17:43:57 +02:00
{ START_TIMER
for ( level = 0 ; level < s - > spatial_decomposition_count ; level + + ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
SubBand * b = & p - > band [ level ] [ orientation ] ;
unpack_coeffs ( s , b , b - > parent , orientation ) ;
2004-07-26 01:36:44 +02:00
}
2005-04-03 17:43:57 +02:00
}
STOP_TIMER ( " unpack coeffs " ) ;
}
2004-07-26 01:36:44 +02:00
2005-03-15 22:05:34 +01:00
{ START_TIMER
const int mb_h = s - > b_height < < s - > block_max_depth ;
const int block_size = MB_SIZE > > s - > block_max_depth ;
const int block_w = plane_index ? block_size / 2 : block_size ;
int mb_y ;
dwt_compose_t cs [ MAX_DECOMPOSITIONS ] ;
int yd = 0 , yq = 0 ;
2005-04-03 17:43:57 +02:00
int y ;
int end_y ;
2005-03-15 22:05:34 +01:00
2005-04-03 17:43:57 +02:00
ff_spatial_idwt_buffered_init ( cs , & s - > sb , w , h , 1 , s - > spatial_decomposition_type , s - > spatial_decomposition_count ) ;
2005-03-15 22:05:34 +01:00
for ( mb_y = 0 ; mb_y < = mb_h ; mb_y + + ) {
2005-12-17 19:14:38 +01:00
2005-04-25 12:03:01 +02:00
int slice_starty = block_w * mb_y ;
int slice_h = block_w * ( mb_y + 1 ) ;
if ( ! ( s - > keyframe | | s - > avctx - > debug & 512 ) ) {
slice_starty = FFMAX ( 0 , slice_starty - ( block_w > > 1 ) ) ;
slice_h - = ( block_w > > 1 ) ;
}
2005-03-15 22:05:34 +01:00
2005-12-17 19:14:38 +01:00
{
2005-04-03 17:43:57 +02:00
START_TIMER
for ( level = 0 ; level < s - > spatial_decomposition_count ; level + + ) {
2005-04-25 12:03:01 +02:00
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
2005-04-03 17:43:57 +02:00
SubBand * b = & p - > band [ level ] [ orientation ] ;
int start_y ;
int end_y ;
int our_mb_start = mb_y ;
int our_mb_end = ( mb_y + 1 ) ;
2006-01-04 23:34:11 +01:00
const int extra = 3 ;
start_y = ( mb_y ? ( ( block_w * our_mb_start ) > > ( s - > spatial_decomposition_count - level ) ) + s - > spatial_decomposition_count - level + extra : 0 ) ;
end_y = ( ( ( block_w * our_mb_end ) > > ( s - > spatial_decomposition_count - level ) ) + s - > spatial_decomposition_count - level + extra ) ;
2005-04-25 12:03:01 +02:00
if ( ! ( s - > keyframe | | s - > avctx - > debug & 512 ) ) {
start_y = FFMAX ( 0 , start_y - ( block_w > > ( 1 + s - > spatial_decomposition_count - level ) ) ) ;
end_y = FFMAX ( 0 , end_y - ( block_w > > ( 1 + s - > spatial_decomposition_count - level ) ) ) ;
}
start_y = FFMIN ( b - > height , start_y ) ;
end_y = FFMIN ( b - > height , end_y ) ;
2005-12-17 19:14:38 +01:00
2005-04-25 12:03:01 +02:00
if ( start_y ! = end_y ) {
if ( orientation = = 0 ) {
SubBand * correlate_band = & p - > band [ 0 ] [ 0 ] ;
int correlate_end_y = FFMIN ( b - > height , end_y + 1 ) ;
int correlate_start_y = FFMIN ( b - > height , ( start_y ? start_y + 1 : 0 ) ) ;
decode_subband_slice_buffered ( s , correlate_band , & s - > sb , correlate_start_y , correlate_end_y , decode_state [ 0 ] [ 0 ] ) ;
correlate_slice_buffered ( s , & s - > sb , correlate_band , correlate_band - > buf , correlate_band - > stride , 1 , 0 , correlate_start_y , correlate_end_y ) ;
dequantize_slice_buffered ( s , & s - > sb , correlate_band , correlate_band - > buf , correlate_band - > stride , start_y , end_y ) ;
}
else
decode_subband_slice_buffered ( s , b , & s - > sb , start_y , end_y , decode_state [ level ] [ orientation ] ) ;
}
2005-04-03 17:43:57 +02:00
}
}
STOP_TIMER ( " decode_subband_slice " ) ;
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
{ START_TIMER
for ( ; yd < slice_h ; yd + = 4 ) {
ff_spatial_idwt_buffered_slice ( cs , & s - > sb , w , h , 1 , s - > spatial_decomposition_type , s - > spatial_decomposition_count , yd ) ;
}
STOP_TIMER ( " idwt slice " ) ; }
2005-12-17 19:14:38 +01:00
2004-08-02 21:09:28 +02:00
if ( s - > qlog = = LOSSLESS_QLOG ) {
2005-03-15 22:05:34 +01:00
for ( ; yq < slice_h & & yq < h ; yq + + ) {
2005-04-03 17:43:57 +02:00
DWTELEM * line = slice_buffer_get_line ( & s - > sb , yq ) ;
2004-08-02 21:09:28 +02:00
for ( x = 0 ; x < w ; x + + ) {
2005-04-03 17:43:57 +02:00
line [ x ] < < = FRAC_BITS ;
2004-08-02 21:09:28 +02:00
}
}
}
2005-03-15 22:05:34 +01:00
2005-04-03 17:43:57 +02:00
predict_slice_buffered ( s , & s - > sb , s - > spatial_dwt_buffer , plane_index , 1 , mb_y ) ;
2005-12-17 19:14:38 +01:00
2005-04-25 12:03:01 +02:00
y = FFMIN ( p - > height , slice_starty ) ;
end_y = FFMIN ( p - > height , slice_h ) ;
2005-04-03 17:43:57 +02:00
while ( y < end_y )
slice_buffer_release ( & s - > sb , y + + ) ;
2005-03-15 22:05:34 +01:00
}
2005-12-17 19:14:38 +01:00
2005-04-03 17:43:57 +02:00
slice_buffer_flush ( & s - > sb ) ;
2005-12-17 19:14:38 +01:00
2005-03-15 22:05:34 +01:00
STOP_TIMER ( " idwt + predict_slices " ) }
2004-07-26 01:36:44 +02:00
}
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
emms_c ( ) ;
if ( s - > last_picture . data [ 0 ] )
avctx - > release_buffer ( avctx , & s - > last_picture ) ;
2005-12-17 19:14:38 +01:00
if ( ! ( s - > avctx - > debug & 2048 ) )
2004-07-26 01:36:44 +02:00
* picture = s - > current_picture ;
else
* picture = s - > mconly_picture ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
* data_size = sizeof ( AVFrame ) ;
2005-12-17 19:14:38 +01:00
2004-10-30 04:16:27 +02:00
bytes_read = c - > bytestream - c - > bytestream_start ;
if ( bytes_read = = 0 ) av_log ( s - > avctx , AV_LOG_ERROR , " error at end of frame \n " ) ; //FIXME
2004-07-26 01:36:44 +02:00
return bytes_read ;
}
static int decode_end ( AVCodecContext * avctx )
{
SnowContext * s = avctx - > priv_data ;
2005-04-03 17:43:57 +02:00
slice_buffer_destroy ( & s - > sb ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
common_end ( s ) ;
return 0 ;
}
AVCodec snow_decoder = {
" snow " ,
CODEC_TYPE_VIDEO ,
CODEC_ID_SNOW ,
sizeof ( SnowContext ) ,
decode_init ,
NULL ,
decode_end ,
decode_frame ,
0 /*CODEC_CAP_DR1*/ /*| CODEC_CAP_DRAW_HORIZ_BAND*/ ,
NULL
} ;
2005-01-03 17:13:52 +01:00
# ifdef CONFIG_ENCODERS
2004-07-26 01:36:44 +02:00
AVCodec snow_encoder = {
" snow " ,
CODEC_TYPE_VIDEO ,
CODEC_ID_SNOW ,
sizeof ( SnowContext ) ,
encode_init ,
encode_frame ,
encode_end ,
} ;
2005-01-03 17:13:52 +01:00
# endif
2004-07-26 01:36:44 +02:00
#if 0
# undef malloc
# undef free
# undef printf
int main ( ) {
int width = 256 ;
int height = 256 ;
int buffer [ 2 ] [ width * height ] ;
SnowContext s ;
int i ;
s . spatial_decomposition_count = 6 ;
s . spatial_decomposition_type = 1 ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
printf ( " testing 5/3 DWT \n " ) ;
for ( i = 0 ; i < width * height ; i + + )
buffer [ 0 ] [ i ] = buffer [ 1 ] [ i ] = random ( ) % 54321 - 12345 ;
2005-12-17 19:14:38 +01:00
2005-12-03 04:07:54 +01:00
ff_spatial_dwt ( buffer [ 0 ] , width , height , width , s . spatial_decomposition_type , s . spatial_decomposition_count ) ;
ff_spatial_idwt ( buffer [ 0 ] , width , height , width , s . spatial_decomposition_type , s . spatial_decomposition_count ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width * height ; i + + )
if ( buffer [ 0 ] [ i ] ! = buffer [ 1 ] [ i ] ) printf ( " fsck: %d %d %d \n " , i , buffer [ 0 ] [ i ] , buffer [ 1 ] [ i ] ) ;
printf ( " testing 9/7 DWT \n " ) ;
s . spatial_decomposition_type = 0 ;
for ( i = 0 ; i < width * height ; i + + )
buffer [ 0 ] [ i ] = buffer [ 1 ] [ i ] = random ( ) % 54321 - 12345 ;
2005-12-17 19:14:38 +01:00
2005-12-03 04:07:54 +01:00
ff_spatial_dwt ( buffer [ 0 ] , width , height , width , s . spatial_decomposition_type , s . spatial_decomposition_count ) ;
ff_spatial_idwt ( buffer [ 0 ] , width , height , width , s . spatial_decomposition_type , s . spatial_decomposition_count ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = 0 ; i < width * height ; i + + )
2005-12-03 04:07:54 +01:00
if ( ABS ( buffer [ 0 ] [ i ] - buffer [ 1 ] [ i ] ) > 20 ) printf ( " fsck: %d %d %d \n " , i , buffer [ 0 ] [ i ] , buffer [ 1 ] [ i ] ) ;
2005-12-17 19:14:38 +01:00
2005-12-03 04:07:54 +01:00
#if 0
2004-07-26 01:36:44 +02:00
printf ( " testing AC coder \n " ) ;
memset ( s . header_state , 0 , sizeof ( s . header_state ) ) ;
2004-10-30 04:16:27 +02:00
ff_init_range_encoder ( & s . c , buffer [ 0 ] , 256 * 256 ) ;
2004-07-26 01:36:44 +02:00
ff_init_cabac_states ( & s . c , ff_h264_lps_range , ff_h264_mps_state , ff_h264_lps_state , 64 ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = - 256 ; i < 256 ; i + + ) {
START_TIMER
put_symbol ( & s . c , s . header_state , i * i * i / 3 * ABS ( i ) , 1 ) ;
STOP_TIMER ( " put_symbol " )
}
2004-10-30 04:16:27 +02:00
ff_rac_terminate ( & s . c ) ;
2004-07-26 01:36:44 +02:00
memset ( s . header_state , 0 , sizeof ( s . header_state ) ) ;
2004-10-30 04:16:27 +02:00
ff_init_range_decoder ( & s . c , buffer [ 0 ] , 256 * 256 ) ;
2004-07-26 01:36:44 +02:00
ff_init_cabac_states ( & s . c , ff_h264_lps_range , ff_h264_mps_state , ff_h264_lps_state , 64 ) ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
for ( i = - 256 ; i < 256 ; i + + ) {
int j ;
START_TIMER
j = get_symbol ( & s . c , s . header_state , 1 ) ;
STOP_TIMER ( " get_symbol " )
if ( j ! = i * i * i / 3 * ABS ( i ) ) printf ( " fsck: %d != %d \n " , i , j ) ;
}
2005-12-03 04:07:54 +01:00
# endif
2004-07-26 01:36:44 +02:00
{
int level , orientation , x , y ;
int64_t errors [ 8 ] [ 4 ] ;
int64_t g = 0 ;
memset ( errors , 0 , sizeof ( errors ) ) ;
s . spatial_decomposition_count = 3 ;
s . spatial_decomposition_type = 0 ;
for ( level = 0 ; level < s . spatial_decomposition_count ; level + + ) {
for ( orientation = level ? 1 : 0 ; orientation < 4 ; orientation + + ) {
int w = width > > ( s . spatial_decomposition_count - level ) ;
int h = height > > ( s . spatial_decomposition_count - level ) ;
int stride = width < < ( s . spatial_decomposition_count - level ) ;
DWTELEM * buf = buffer [ 0 ] ;
int64_t error = 0 ;
if ( orientation & 1 ) buf + = w ;
if ( orientation > 1 ) buf + = stride > > 1 ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
memset ( buffer [ 0 ] , 0 , sizeof ( int ) * width * height ) ;
buf [ w / 2 + h / 2 * stride ] = 256 * 256 ;
2005-12-03 04:07:54 +01:00
ff_spatial_idwt ( buffer [ 0 ] , width , height , width , s . spatial_decomposition_type , s . spatial_decomposition_count ) ;
2004-07-26 01:36:44 +02:00
for ( y = 0 ; y < height ; y + + ) {
for ( x = 0 ; x < width ; x + + ) {
int64_t d = buffer [ 0 ] [ x + y * width ] ;
error + = d * d ;
if ( ABS ( width / 2 - x ) < 9 & & ABS ( height / 2 - y ) < 9 & & level = = 2 ) printf ( " %8lld " , d ) ;
}
if ( ABS ( height / 2 - y ) < 9 & & level = = 2 ) printf ( " \n " ) ;
}
error = ( int ) ( sqrt ( error ) + 0.5 ) ;
errors [ level ] [ orientation ] = error ;
if ( g ) g = ff_gcd ( g , error ) ;
else g = error ;
}
}
printf ( " static int const visual_weight[][4]={ \n " ) ;
for ( level = 0 ; level < s . spatial_decomposition_count ; level + + ) {
printf ( " { " ) ;
for ( orientation = 0 ; orientation < 4 ; orientation + + ) {
printf ( " %8lld, " , errors [ level ] [ orientation ] / g ) ;
}
printf ( " }, \n " ) ;
}
printf ( " }; \n " ) ;
{
int level = 2 ;
int orientation = 3 ;
int w = width > > ( s . spatial_decomposition_count - level ) ;
int h = height > > ( s . spatial_decomposition_count - level ) ;
int stride = width < < ( s . spatial_decomposition_count - level ) ;
DWTELEM * buf = buffer [ 0 ] ;
int64_t error = 0 ;
buf + = w ;
buf + = stride > > 1 ;
2005-12-17 19:14:38 +01:00
2004-07-26 01:36:44 +02:00
memset ( buffer [ 0 ] , 0 , sizeof ( int ) * width * height ) ;
# if 1
for ( y = 0 ; y < height ; y + + ) {
for ( x = 0 ; x < width ; x + + ) {
int tab [ 4 ] = { 0 , 2 , 3 , 1 } ;
buffer [ 0 ] [ x + width * y ] = 256 * 256 * tab [ ( x & 1 ) + 2 * ( y & 1 ) ] ;
}
}
2005-12-03 04:07:54 +01:00
ff_spatial_dwt ( buffer [ 0 ] , width , height , width , s . spatial_decomposition_type , s . spatial_decomposition_count ) ;
2004-07-26 01:36:44 +02:00
# else
for ( y = 0 ; y < h ; y + + ) {
for ( x = 0 ; x < w ; x + + ) {
buf [ x + y * stride ] = 169 ;
buf [ x + y * stride - w ] = 64 ;
}
}
2005-12-03 04:07:54 +01:00
ff_spatial_idwt ( buffer [ 0 ] , width , height , width , s . spatial_decomposition_type , s . spatial_decomposition_count ) ;
2004-07-26 01:36:44 +02:00
# endif
for ( y = 0 ; y < height ; y + + ) {
for ( x = 0 ; x < width ; x + + ) {
int64_t d = buffer [ 0 ] [ x + y * width ] ;
error + = d * d ;
if ( ABS ( width / 2 - x ) < 9 & & ABS ( height / 2 - y ) < 9 ) printf ( " %8lld " , d ) ;
}
if ( ABS ( height / 2 - y ) < 9 ) printf ( " \n " ) ;
}
}
}
return 0 ;
}
# endif