mirror of
https://github.com/ultravideo/uvg266.git
synced 2024-11-27 19:24:06 +00:00
Refactor intra sad calculation functions.
- Move SATD (hadamart) code to it's own functions. - Generate functions for fixed block sizes with macros, as if using templates. - Define new interface with function pointer cost_16bit_nxn_func that different cost functions can use. - Hide fixed size block cost functions with getters.
This commit is contained in:
Ari Koivula
parent
122806bcf7
commit
d74d4e4ef1
10
src/intra.c
10
src/intra.c
|
|
@ -216,12 +216,12 @@ void intra_filter(int16_t *ref, int32_t stride,int32_t width, int8_t mode)
|
|||
int16_t intra_prediction(pixel *orig, int32_t origstride, int16_t *rec, int32_t recstride, uint32_t xpos,
|
||||
uint32_t ypos, uint32_t width, int16_t *dst, int32_t dststride, uint32_t *sad_out)
|
||||
{
|
||||
typedef uint32_t (*sad_function)(int16_t *block,uint32_t stride1,int16_t *block2, uint32_t stride2);
|
||||
uint32_t best_sad = 0xffffffff;
|
||||
uint32_t sad = 0;
|
||||
int16_t best_mode = 1;
|
||||
int32_t x,y,i;
|
||||
sad_function calc_sad;
|
||||
|
||||
cost_16bit_nxn_func cost_func = get_satd_16bit_nxn_func(width);
|
||||
|
||||
// Temporary block arrays
|
||||
// TODO: alloc with alignment
|
||||
|
|
@ -233,11 +233,10 @@ int16_t intra_prediction(pixel *orig, int32_t origstride, int16_t *rec, int32_t
|
|||
pixel *orig_shift = &orig[xpos + ypos*origstride]; //!< pointer to orig with offset of (1,1)
|
||||
int8_t filter = (width<32); // TODO: chroma support
|
||||
|
||||
sad_function sad_array[5] = {&sad4x4,&sad8x8,&sad16x16,&sad32x32,&sad64x64}; //TODO: get SAD functions from parameters
|
||||
uint8_t threshold = intra_hor_ver_dist_thres[g_to_bits[width]]; //!< Intra filtering threshold
|
||||
|
||||
#define COPY_PRED_TO_DST() for (y = 0; y < (int32_t)width; y++) { for (x = 0; x < (int32_t)width; x++) { dst[x + y*dststride] = pred[x + y*width]; } }
|
||||
#define CHECK_FOR_BEST(mode, additional_sad) sad = calc_sad(pred,width,orig_block,width); \
|
||||
#define CHECK_FOR_BEST(mode, additional_sad) sad = cost_func(pred, orig_block); \
|
||||
sad += additional_sad;\
|
||||
if(sad < best_sad)\
|
||||
{\
|
||||
|
|
@ -246,9 +245,6 @@ int16_t intra_prediction(pixel *orig, int32_t origstride, int16_t *rec, int32_t
|
|||
COPY_PRED_TO_DST();\
|
||||
}
|
||||
|
||||
// Choose SAD function according to width
|
||||
calc_sad = sad_array[g_to_bits[width]];
|
||||
|
||||
// Store original block for SAD computation
|
||||
i = 0;
|
||||
for(y = 0; y < (int32_t)width; y++) {
|
||||
|
|
|
|||
313
src/picture.c
313
src/picture.c
|
|
@ -300,12 +300,13 @@ double image_psnr(pixel *frame1, pixel *frame2, int32_t x, int32_t y)
|
|||
}
|
||||
|
||||
/**
|
||||
* \brief
|
||||
* \brief Calculate SATD between two 8x8 blocks inside bigger arrays.
|
||||
*/
|
||||
uint32_t Hadamard8x8(int16_t *piOrg, int32_t iStrideOrg, int16_t *piCur, int32_t iStrideCur)
|
||||
unsigned satd_16bit_8x8_general(int16_t *piOrg, int32_t iStrideOrg, int16_t *piCur, int32_t iStrideCur)
|
||||
{
|
||||
int32_t k, i, j, jj, sad=0;
|
||||
int32_t diff[64], m1[8][8], m2[8][8], m3[8][8];
|
||||
|
||||
for (k = 0; k < 64; k += 8) {
|
||||
diff[k+0] = piOrg[0] - piCur[0];
|
||||
diff[k+1] = piOrg[1] - piCur[1];
|
||||
|
|
@ -392,194 +393,144 @@ uint32_t Hadamard8x8(int16_t *piOrg, int32_t iStrideOrg, int16_t *piCur, int32_t
|
|||
return sad;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief
|
||||
*/
|
||||
uint32_t sad64x64(int16_t *block1, uint32_t stride1,
|
||||
int16_t *block2, uint32_t stride2)
|
||||
{
|
||||
int32_t y, x;
|
||||
uint32_t sum = 0;
|
||||
/*
|
||||
for (y=0; y<64; y++) {
|
||||
i = y * stride1;
|
||||
ii = y * stride2;
|
||||
for (x = 0; x < 64; x++) {
|
||||
sum += abs((int16_t)block1[i + x] - (int16_t)block2[ii + x]);
|
||||
}
|
||||
}
|
||||
}*/
|
||||
int32_t iOffsetOrg = stride1 << 3;
|
||||
int32_t iOffsetCur = stride2 << 3;
|
||||
for (y = 0; y < 64; y += 8) {
|
||||
for (x = 0; x < 64; x += 8) {
|
||||
sum += Hadamard8x8(&block1[x], stride1, &block2[x], stride2);
|
||||
}
|
||||
block1 += iOffsetOrg;
|
||||
block2 += iOffsetCur;
|
||||
// Function macro for defining hadamart calculating functions
|
||||
// for fixed size blocks. They calculate hadamart for integer
|
||||
// multiples of 8x8 with the 8x8 hadamart function.
|
||||
#define SATD_NXN(n, pixel_type, suffix) \
|
||||
unsigned satd_ ## suffix ## _ ## n ## x ## n ## ( \
|
||||
pixel_type *block1, pixel_type *block2) \
|
||||
{ \
|
||||
unsigned y, x; \
|
||||
unsigned sum = 0; \
|
||||
for (y = 0; y < (n); y += 8) { \
|
||||
unsigned row = y * (n); \
|
||||
for (x = 0; x < (n); x += 8) { \
|
||||
sum += satd_16bit_8x8_general(&block1[row + x], (n), &block2[row + x], (n)); \
|
||||
} \
|
||||
} \
|
||||
return sum; \
|
||||
}
|
||||
|
||||
return sum;
|
||||
// These macros define sadt_16bit_NxN for N = 8, 16, 32, 64
|
||||
SATD_NXN(8, int16_t, 16bit)
|
||||
SATD_NXN(16, int16_t, 16bit)
|
||||
SATD_NXN(32, int16_t, 16bit)
|
||||
SATD_NXN(64, int16_t, 16bit)
|
||||
|
||||
|
||||
// Function macro for defining SAD calculating functions
|
||||
// for fixed size blocks.
|
||||
#define SAD_NXN(n, pixel_type, suffix) \
|
||||
unsigned sad_ ## suffix ## _ ## n ## x ## n ## ( \
|
||||
pixel_type *block1, pixel_type *block2) \
|
||||
{ \
|
||||
unsigned x, y, row; \
|
||||
unsigned sum = 0; \
|
||||
for(y = 0; y < (n); y++) { \
|
||||
row = y * (n); \
|
||||
for (x = 0; x < (n); ++x) { \
|
||||
sum += abs(block1[row + x] - block2[row + x]); \
|
||||
} \
|
||||
} \
|
||||
return sum; \
|
||||
}
|
||||
|
||||
// These macros define sad_16bit_nxn functions for n = 4, 8, 16, 32, 64
|
||||
// with function signatures of cost_16bit_nxn_func.
|
||||
// They are used through get_sad_16bit_nxn_func.
|
||||
SAD_NXN(4, int16_t, 16bit)
|
||||
SAD_NXN(8, int16_t, 16bit)
|
||||
SAD_NXN(16, int16_t, 16bit)
|
||||
SAD_NXN(32, int16_t, 16bit)
|
||||
SAD_NXN(64, int16_t, 16bit)
|
||||
|
||||
/**
|
||||
* \brief Get a function that calculates SATD for NxN block.
|
||||
*
|
||||
* \param n Width of the region for which SATD is calculated.
|
||||
*
|
||||
* \returns Pointer to cost_16bit_nxn_func.
|
||||
*/
|
||||
cost_16bit_nxn_func get_satd_16bit_nxn_func(unsigned n)
|
||||
{
|
||||
switch (n) {
|
||||
case 8:
|
||||
return &satd_16bit_8x8;
|
||||
case 16:
|
||||
return &satd_16bit_16x16;
|
||||
case 32:
|
||||
return &satd_16bit_32x32;
|
||||
case 64:
|
||||
return &satd_16bit_64x64;
|
||||
default:
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief
|
||||
* \brief Get a function that calculates SAD for NxN block.
|
||||
*
|
||||
* \param n Width of the region for which SAD is calculated.
|
||||
*
|
||||
* \returns Pointer to cost_16bit_nxn_func.
|
||||
*/
|
||||
uint32_t sad32x32(int16_t *block1, uint32_t stride1,
|
||||
int16_t *block2, uint32_t stride2)
|
||||
cost_16bit_nxn_func get_sad_16bit_nxn_func(unsigned n)
|
||||
{
|
||||
int32_t x, y;
|
||||
int32_t sum = 0;
|
||||
int32_t iOffsetOrg = stride1 << 3;
|
||||
int32_t iOffsetCur = stride2 << 3;
|
||||
|
||||
for (y = 0; y < 32; y += 8) {
|
||||
for ( x = 0; x < 32; x += 8 ) {
|
||||
sum += Hadamard8x8(&block1[x], stride1, &block2[x], stride2);
|
||||
switch (n) {
|
||||
case 4:
|
||||
return &sad_16bit_4x4;
|
||||
case 8:
|
||||
return &sad_16bit_8x8;
|
||||
case 16:
|
||||
return &sad_16bit_16x16;
|
||||
case 32:
|
||||
return &sad_16bit_32x32;
|
||||
case 64:
|
||||
return &sad_16bit_64x64;
|
||||
default:
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Calculate SATD for NxN block of size N.
|
||||
*
|
||||
* \param block1 Start of the first block.
|
||||
* \param block2 Start of the second block.
|
||||
* \param n Width of the region for which SAD is calculated.
|
||||
*
|
||||
* \returns Sum of Absolute Transformed Differences (SATD)
|
||||
*/
|
||||
unsigned satd_nxn_16bit(int16_t *block1, int16_t *block2, unsigned n)
|
||||
{
|
||||
cost_16bit_nxn_func sad_func = get_satd_16bit_nxn_func(n);
|
||||
return sad_func(block1, block2);
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Calculate SAD for NxN block of size N.
|
||||
*
|
||||
* \param block1 Start of the first block.
|
||||
* \param block2 Start of the second block.
|
||||
* \param n Width of the region for which SAD is calculated.
|
||||
*
|
||||
* \returns Sum of Absolute Differences
|
||||
*/
|
||||
unsigned sad_nxn_16bit(int16_t *block1, int16_t *block2, unsigned n)
|
||||
{
|
||||
cost_16bit_nxn_func sad_func = get_sad_16bit_nxn_func(n);
|
||||
if (sad_func) {
|
||||
return sad_func(block1, block2);
|
||||
} else {
|
||||
unsigned row, x;
|
||||
unsigned sum = 0;
|
||||
for (row = 0; row < n; row += n) {
|
||||
for (x = 0; x < n; ++x) {
|
||||
sum += abs(block1[row + x] - block2[row + x]);
|
||||
}
|
||||
}
|
||||
block1 += iOffsetOrg;
|
||||
block2 += iOffsetCur;
|
||||
return sum;
|
||||
}
|
||||
|
||||
/*
|
||||
uint32_t sum=0;
|
||||
int32_t i,ii;
|
||||
for(y=0;y<32;y++)
|
||||
{
|
||||
i = y*stride1;
|
||||
ii = y*stride2;
|
||||
sum+=abs((int32_t)block[i]-(int32_t)block2[ii]);
|
||||
sum+=abs((int32_t)block[i+1]-(int32_t)block2[ii+1]);
|
||||
sum+=abs((int32_t)block[i+2]-(int32_t)block2[ii+2]);
|
||||
sum+=abs((int32_t)block[i+3]-(int32_t)block2[ii+3]);
|
||||
sum+=abs((int32_t)block[i+4]-(int32_t)block2[ii+4]);
|
||||
sum+=abs((int32_t)block[i+5]-(int32_t)block2[ii+5]);
|
||||
sum+=abs((int32_t)block[i+6]-(int32_t)block2[ii+6]);
|
||||
sum+=abs((int32_t)block[i+7]-(int32_t)block2[ii+7]);
|
||||
sum+=abs((int32_t)block[i+8]-(int32_t)block2[ii+8]);
|
||||
sum+=abs((int32_t)block[i+9]-(int32_t)block2[ii+9]);
|
||||
sum+=abs((int32_t)block[i+10]-(int32_t)block2[ii+10]);
|
||||
sum+=abs((int32_t)block[i+11]-(int32_t)block2[ii+11]);
|
||||
sum+=abs((int32_t)block[i+12]-(int32_t)block2[ii+12]);
|
||||
sum+=abs((int32_t)block[i+13]-(int32_t)block2[ii+13]);
|
||||
sum+=abs((int32_t)block[i+14]-(int32_t)block2[ii+14]);
|
||||
sum+=abs((int32_t)block[i+15]-(int32_t)block2[ii+15]);
|
||||
sum+=abs((int32_t)block[i+16]-(int32_t)block2[ii+16]);
|
||||
sum+=abs((int32_t)block[i+17]-(int32_t)block2[ii+17]);
|
||||
sum+=abs((int32_t)block[i+18]-(int32_t)block2[ii+18]);
|
||||
sum+=abs((int32_t)block[i+19]-(int32_t)block2[ii+19]);
|
||||
sum+=abs((int32_t)block[i+20]-(int32_t)block2[ii+20]);
|
||||
sum+=abs((int32_t)block[i+21]-(int32_t)block2[ii+21]);
|
||||
sum+=abs((int32_t)block[i+22]-(int32_t)block2[ii+22]);
|
||||
sum+=abs((int32_t)block[i+23]-(int32_t)block2[ii+23]);
|
||||
sum+=abs((int32_t)block[i+24]-(int32_t)block2[ii+24]);
|
||||
sum+=abs((int32_t)block[i+25]-(int32_t)block2[ii+25]);
|
||||
sum+=abs((int32_t)block[i+26]-(int32_t)block2[ii+26]);
|
||||
sum+=abs((int32_t)block[i+27]-(int32_t)block2[ii+27]);
|
||||
sum+=abs((int32_t)block[i+28]-(int32_t)block2[ii+28]);
|
||||
sum+=abs((int32_t)block[i+29]-(int32_t)block2[ii+29]);
|
||||
sum+=abs((int32_t)block[i+30]-(int32_t)block2[ii+30]);
|
||||
sum+=abs((int32_t)block[i+31]-(int32_t)block2[ii+31]);
|
||||
}
|
||||
*/
|
||||
return sum;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief
|
||||
*/
|
||||
uint32_t sad16x16(int16_t *block1, uint32_t stride1,
|
||||
int16_t* block2, uint32_t stride2)
|
||||
{
|
||||
int32_t x, y;
|
||||
|
||||
int32_t sum = 0;
|
||||
int32_t iOffsetOrg = stride1 << 3;
|
||||
int32_t iOffsetCur = stride2 << 3;
|
||||
|
||||
for (y = 0; y < 16; y += 8) {
|
||||
for (x = 0; x < 16; x += 8) {
|
||||
sum += Hadamard8x8(&block1[x], stride1, &block2[x], stride2);
|
||||
}
|
||||
block1 += iOffsetOrg;
|
||||
block2 += iOffsetCur;
|
||||
}
|
||||
|
||||
/*
|
||||
uint32_t sum=0;
|
||||
int32_t i,ii;
|
||||
for(y=0;y<16;y++)
|
||||
{
|
||||
i = y*stride1;
|
||||
ii = y*stride2;
|
||||
sum+=abs((int32_t)block[i]-(int32_t)block2[ii]);
|
||||
sum+=abs((int32_t)block[i+1]-(int32_t)block2[ii+1]);
|
||||
sum+=abs((int32_t)block[i+2]-(int32_t)block2[ii+2]);
|
||||
sum+=abs((int32_t)block[i+3]-(int32_t)block2[ii+3]);
|
||||
sum+=abs((int32_t)block[i+4]-(int32_t)block2[ii+4]);
|
||||
sum+=abs((int32_t)block[i+5]-(int32_t)block2[ii+5]);
|
||||
sum+=abs((int32_t)block[i+6]-(int32_t)block2[ii+6]);
|
||||
sum+=abs((int32_t)block[i+7]-(int32_t)block2[ii+7]);
|
||||
sum+=abs((int32_t)block[i+8]-(int32_t)block2[ii+8]);
|
||||
sum+=abs((int32_t)block[i+9]-(int32_t)block2[ii+9]);
|
||||
sum+=abs((int32_t)block[i+10]-(int32_t)block2[ii+10]);
|
||||
sum+=abs((int32_t)block[i+11]-(int32_t)block2[ii+11]);
|
||||
sum+=abs((int32_t)block[i+12]-(int32_t)block2[ii+12]);
|
||||
sum+=abs((int32_t)block[i+13]-(int32_t)block2[ii+13]);
|
||||
sum+=abs((int32_t)block[i+14]-(int32_t)block2[ii+14]);
|
||||
sum+=abs((int32_t)block[i+15]-(int32_t)block2[ii+15]);
|
||||
}
|
||||
*/
|
||||
return sum;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief
|
||||
*/
|
||||
uint32_t sad8x8(int16_t *block1, uint32_t stride1,
|
||||
int16_t* block2, uint32_t stride2)
|
||||
{
|
||||
uint32_t sum = 0;
|
||||
sum = Hadamard8x8(block1, stride1, block2, stride2);
|
||||
/*
|
||||
|
||||
for(y=0;y<8;y++)
|
||||
{
|
||||
i = y*stride1;
|
||||
ii = y*stride2;
|
||||
sum+=abs((int32_t)block[i]-(int32_t)block2[ii]);
|
||||
sum+=abs((int32_t)block[i+1]-(int32_t)block2[ii+1]);
|
||||
sum+=abs((int32_t)block[i+2]-(int32_t)block2[ii+2]);
|
||||
sum+=abs((int32_t)block[i+3]-(int32_t)block2[ii+3]);
|
||||
sum+=abs((int32_t)block[i+4]-(int32_t)block2[ii+4]);
|
||||
sum+=abs((int32_t)block[i+5]-(int32_t)block2[ii+5]);
|
||||
sum+=abs((int32_t)block[i+6]-(int32_t)block2[ii+6]);
|
||||
sum+=abs((int32_t)block[i+7]-(int32_t)block2[ii+7]);
|
||||
}
|
||||
*/
|
||||
|
||||
return sum;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief
|
||||
*/
|
||||
uint32_t sad4x4(int16_t *block1, uint32_t stride1,
|
||||
int16_t *block2, uint32_t stride2)
|
||||
{
|
||||
int32_t i, ii, y;
|
||||
uint32_t sum = 0;
|
||||
|
||||
for (y = 0; y < 4; y++) {
|
||||
i = y * stride1;
|
||||
ii = y * stride2;
|
||||
sum += abs((int32_t)block1[i] - (int32_t)block2[ii]);
|
||||
sum += abs((int32_t)block1[i+1] - (int32_t)block2[ii+1]);
|
||||
sum += abs((int32_t)block1[i+2] - (int32_t)block2[ii+2]);
|
||||
sum += abs((int32_t)block1[i+3] - (int32_t)block2[ii+3]);
|
||||
}
|
||||
|
||||
return sum;
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
|||
|
|
@ -113,16 +113,13 @@ int picture_list_destroy(picture_list *list);
|
|||
int picture_list_add(picture_list *list, picture *pic);
|
||||
int picture_list_rem(picture_list *list, int n, int8_t destroy);
|
||||
|
||||
uint32_t sad64x64(int16_t *block1, uint32_t stride1,
|
||||
int16_t* block2, uint32_t stride2);
|
||||
uint32_t sad32x32(int16_t *block1, uint32_t stride1,
|
||||
int16_t* block2, uint32_t stride2);
|
||||
uint32_t sad16x16(int16_t *block1, uint32_t stride1,
|
||||
int16_t* block2, uint32_t stride2);
|
||||
uint32_t sad8x8(int16_t *block1, uint32_t stride1,
|
||||
int16_t* block2, uint32_t stride2);
|
||||
uint32_t sad4x4(int16_t *block1, uint32_t stride1,
|
||||
int16_t* block2, uint32_t stride2);
|
||||
typedef unsigned (*cost_16bit_nxn_func)(int16_t *block1, int16_t *block2);
|
||||
|
||||
cost_16bit_nxn_func get_satd_16bit_nxn_func(unsigned n);
|
||||
cost_16bit_nxn_func get_sad_16bit_nxn_func(unsigned n);
|
||||
|
||||
unsigned satd_16bit_nxn(int16_t *block1, int16_t *block2, unsigned n);
|
||||
unsigned sad_16bit_nxn(int16_t *block1, int16_t *block2, unsigned n);
|
||||
|
||||
unsigned calc_sad(picture *pic, picture *ref,
|
||||
int pic_x, int pic_y, int ref_x, int ref_y,
|
||||
|
|
|
|||
Loading…
Reference in a new issue