Commit ebc854e4 authored by Michael Niedermayer's avatar Michael Niedermayer

swr/auto_matrix: move layouts to local variables

Signed-off-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parent 9ab538e3
...@@ -123,35 +123,37 @@ static int auto_matrix(SwrContext *s) ...@@ -123,35 +123,37 @@ static int auto_matrix(SwrContext *s)
{ {
int i, j, out_i; int i, j, out_i;
double matrix[64][64]={{0}}; double matrix[64][64]={{0}};
int64_t unaccounted= s->in_ch_layout & ~s->out_ch_layout; int64_t unaccounted, in_ch_layout = s->in_ch_layout, out_ch_layout = s->out_ch_layout;
double maxcoef=0; double maxcoef=0;
char buf[128]; char buf[128];
const int matrix_encoding = s->matrix_encoding; const int matrix_encoding = s->matrix_encoding;
memset(s->matrix, 0, sizeof(s->matrix)); if(!sane_layout(in_ch_layout)){
for(i=0; i<64; i++){
if(s->in_ch_layout & s->out_ch_layout & (1LL<<i))
matrix[i][i]= 1.0;
}
if(!sane_layout(s->in_ch_layout)){
av_get_channel_layout_string(buf, sizeof(buf), -1, s->in_ch_layout); av_get_channel_layout_string(buf, sizeof(buf), -1, s->in_ch_layout);
av_log(s, AV_LOG_ERROR, "Input channel layout '%s' is not supported\n", buf); av_log(s, AV_LOG_ERROR, "Input channel layout '%s' is not supported\n", buf);
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
if(!sane_layout(s->out_ch_layout)){ if(!sane_layout(out_ch_layout)){
av_get_channel_layout_string(buf, sizeof(buf), -1, s->out_ch_layout); av_get_channel_layout_string(buf, sizeof(buf), -1, s->out_ch_layout);
av_log(s, AV_LOG_ERROR, "Output channel layout '%s' is not supported\n", buf); av_log(s, AV_LOG_ERROR, "Output channel layout '%s' is not supported\n", buf);
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
memset(s->matrix, 0, sizeof(s->matrix));
for(i=0; i<64; i++){
if(in_ch_layout & out_ch_layout & (1LL<<i))
matrix[i][i]= 1.0;
}
unaccounted= in_ch_layout & ~out_ch_layout;
//FIXME implement dolby surround //FIXME implement dolby surround
//FIXME implement full ac3 //FIXME implement full ac3
if(unaccounted & AV_CH_FRONT_CENTER){ if(unaccounted & AV_CH_FRONT_CENTER){
if((s->out_ch_layout & AV_CH_LAYOUT_STEREO) == AV_CH_LAYOUT_STEREO){ if((out_ch_layout & AV_CH_LAYOUT_STEREO) == AV_CH_LAYOUT_STEREO){
if(s->in_ch_layout & AV_CH_LAYOUT_STEREO) { if(in_ch_layout & AV_CH_LAYOUT_STEREO) {
matrix[ FRONT_LEFT][FRONT_CENTER]+= s->clev; matrix[ FRONT_LEFT][FRONT_CENTER]+= s->clev;
matrix[FRONT_RIGHT][FRONT_CENTER]+= s->clev; matrix[FRONT_RIGHT][FRONT_CENTER]+= s->clev;
} else { } else {
...@@ -162,23 +164,23 @@ static int auto_matrix(SwrContext *s) ...@@ -162,23 +164,23 @@ static int auto_matrix(SwrContext *s)
av_assert0(0); av_assert0(0);
} }
if(unaccounted & AV_CH_LAYOUT_STEREO){ if(unaccounted & AV_CH_LAYOUT_STEREO){
if(s->out_ch_layout & AV_CH_FRONT_CENTER){ if(out_ch_layout & AV_CH_FRONT_CENTER){
matrix[FRONT_CENTER][ FRONT_LEFT]+= M_SQRT1_2; matrix[FRONT_CENTER][ FRONT_LEFT]+= M_SQRT1_2;
matrix[FRONT_CENTER][FRONT_RIGHT]+= M_SQRT1_2; matrix[FRONT_CENTER][FRONT_RIGHT]+= M_SQRT1_2;
if(s->in_ch_layout & AV_CH_FRONT_CENTER) if(in_ch_layout & AV_CH_FRONT_CENTER)
matrix[FRONT_CENTER][ FRONT_CENTER] = s->clev*sqrt(2); matrix[FRONT_CENTER][ FRONT_CENTER] = s->clev*sqrt(2);
}else }else
av_assert0(0); av_assert0(0);
} }
if(unaccounted & AV_CH_BACK_CENTER){ if(unaccounted & AV_CH_BACK_CENTER){
if(s->out_ch_layout & AV_CH_BACK_LEFT){ if(out_ch_layout & AV_CH_BACK_LEFT){
matrix[ BACK_LEFT][BACK_CENTER]+= M_SQRT1_2; matrix[ BACK_LEFT][BACK_CENTER]+= M_SQRT1_2;
matrix[BACK_RIGHT][BACK_CENTER]+= M_SQRT1_2; matrix[BACK_RIGHT][BACK_CENTER]+= M_SQRT1_2;
}else if(s->out_ch_layout & AV_CH_SIDE_LEFT){ }else if(out_ch_layout & AV_CH_SIDE_LEFT){
matrix[ SIDE_LEFT][BACK_CENTER]+= M_SQRT1_2; matrix[ SIDE_LEFT][BACK_CENTER]+= M_SQRT1_2;
matrix[SIDE_RIGHT][BACK_CENTER]+= M_SQRT1_2; matrix[SIDE_RIGHT][BACK_CENTER]+= M_SQRT1_2;
}else if(s->out_ch_layout & AV_CH_FRONT_LEFT){ }else if(out_ch_layout & AV_CH_FRONT_LEFT){
if (matrix_encoding == AV_MATRIX_ENCODING_DOLBY || if (matrix_encoding == AV_MATRIX_ENCODING_DOLBY ||
matrix_encoding == AV_MATRIX_ENCODING_DPLII) { matrix_encoding == AV_MATRIX_ENCODING_DPLII) {
if (unaccounted & (AV_CH_BACK_LEFT | AV_CH_SIDE_LEFT)) { if (unaccounted & (AV_CH_BACK_LEFT | AV_CH_SIDE_LEFT)) {
...@@ -192,24 +194,24 @@ static int auto_matrix(SwrContext *s) ...@@ -192,24 +194,24 @@ static int auto_matrix(SwrContext *s)
matrix[ FRONT_LEFT][BACK_CENTER]+= s->slev*M_SQRT1_2; matrix[ FRONT_LEFT][BACK_CENTER]+= s->slev*M_SQRT1_2;
matrix[FRONT_RIGHT][BACK_CENTER]+= s->slev*M_SQRT1_2; matrix[FRONT_RIGHT][BACK_CENTER]+= s->slev*M_SQRT1_2;
} }
}else if(s->out_ch_layout & AV_CH_FRONT_CENTER){ }else if(out_ch_layout & AV_CH_FRONT_CENTER){
matrix[ FRONT_CENTER][BACK_CENTER]+= s->slev*M_SQRT1_2; matrix[ FRONT_CENTER][BACK_CENTER]+= s->slev*M_SQRT1_2;
}else }else
av_assert0(0); av_assert0(0);
} }
if(unaccounted & AV_CH_BACK_LEFT){ if(unaccounted & AV_CH_BACK_LEFT){
if(s->out_ch_layout & AV_CH_BACK_CENTER){ if(out_ch_layout & AV_CH_BACK_CENTER){
matrix[BACK_CENTER][ BACK_LEFT]+= M_SQRT1_2; matrix[BACK_CENTER][ BACK_LEFT]+= M_SQRT1_2;
matrix[BACK_CENTER][BACK_RIGHT]+= M_SQRT1_2; matrix[BACK_CENTER][BACK_RIGHT]+= M_SQRT1_2;
}else if(s->out_ch_layout & AV_CH_SIDE_LEFT){ }else if(out_ch_layout & AV_CH_SIDE_LEFT){
if(s->in_ch_layout & AV_CH_SIDE_LEFT){ if(in_ch_layout & AV_CH_SIDE_LEFT){
matrix[ SIDE_LEFT][ BACK_LEFT]+= M_SQRT1_2; matrix[ SIDE_LEFT][ BACK_LEFT]+= M_SQRT1_2;
matrix[SIDE_RIGHT][BACK_RIGHT]+= M_SQRT1_2; matrix[SIDE_RIGHT][BACK_RIGHT]+= M_SQRT1_2;
}else{ }else{
matrix[ SIDE_LEFT][ BACK_LEFT]+= 1.0; matrix[ SIDE_LEFT][ BACK_LEFT]+= 1.0;
matrix[SIDE_RIGHT][BACK_RIGHT]+= 1.0; matrix[SIDE_RIGHT][BACK_RIGHT]+= 1.0;
} }
}else if(s->out_ch_layout & AV_CH_FRONT_LEFT){ }else if(out_ch_layout & AV_CH_FRONT_LEFT){
if (matrix_encoding == AV_MATRIX_ENCODING_DOLBY) { if (matrix_encoding == AV_MATRIX_ENCODING_DOLBY) {
matrix[FRONT_LEFT ][BACK_LEFT ] -= s->slev * M_SQRT1_2; matrix[FRONT_LEFT ][BACK_LEFT ] -= s->slev * M_SQRT1_2;
matrix[FRONT_LEFT ][BACK_RIGHT] -= s->slev * M_SQRT1_2; matrix[FRONT_LEFT ][BACK_RIGHT] -= s->slev * M_SQRT1_2;
...@@ -224,7 +226,7 @@ static int auto_matrix(SwrContext *s) ...@@ -224,7 +226,7 @@ static int auto_matrix(SwrContext *s)
matrix[ FRONT_LEFT][ BACK_LEFT] += s->slev; matrix[ FRONT_LEFT][ BACK_LEFT] += s->slev;
matrix[FRONT_RIGHT][BACK_RIGHT] += s->slev; matrix[FRONT_RIGHT][BACK_RIGHT] += s->slev;
} }
}else if(s->out_ch_layout & AV_CH_FRONT_CENTER){ }else if(out_ch_layout & AV_CH_FRONT_CENTER){
matrix[ FRONT_CENTER][BACK_LEFT ]+= s->slev*M_SQRT1_2; matrix[ FRONT_CENTER][BACK_LEFT ]+= s->slev*M_SQRT1_2;
matrix[ FRONT_CENTER][BACK_RIGHT]+= s->slev*M_SQRT1_2; matrix[ FRONT_CENTER][BACK_RIGHT]+= s->slev*M_SQRT1_2;
}else }else
...@@ -232,20 +234,20 @@ static int auto_matrix(SwrContext *s) ...@@ -232,20 +234,20 @@ static int auto_matrix(SwrContext *s)
} }
if(unaccounted & AV_CH_SIDE_LEFT){ if(unaccounted & AV_CH_SIDE_LEFT){
if(s->out_ch_layout & AV_CH_BACK_LEFT){ if(out_ch_layout & AV_CH_BACK_LEFT){
/* if back channels do not exist in the input, just copy side /* if back channels do not exist in the input, just copy side
channels to back channels, otherwise mix side into back */ channels to back channels, otherwise mix side into back */
if (s->in_ch_layout & AV_CH_BACK_LEFT) { if (in_ch_layout & AV_CH_BACK_LEFT) {
matrix[BACK_LEFT ][SIDE_LEFT ] += M_SQRT1_2; matrix[BACK_LEFT ][SIDE_LEFT ] += M_SQRT1_2;
matrix[BACK_RIGHT][SIDE_RIGHT] += M_SQRT1_2; matrix[BACK_RIGHT][SIDE_RIGHT] += M_SQRT1_2;
} else { } else {
matrix[BACK_LEFT ][SIDE_LEFT ] += 1.0; matrix[BACK_LEFT ][SIDE_LEFT ] += 1.0;
matrix[BACK_RIGHT][SIDE_RIGHT] += 1.0; matrix[BACK_RIGHT][SIDE_RIGHT] += 1.0;
} }
}else if(s->out_ch_layout & AV_CH_BACK_CENTER){ }else if(out_ch_layout & AV_CH_BACK_CENTER){
matrix[BACK_CENTER][ SIDE_LEFT]+= M_SQRT1_2; matrix[BACK_CENTER][ SIDE_LEFT]+= M_SQRT1_2;
matrix[BACK_CENTER][SIDE_RIGHT]+= M_SQRT1_2; matrix[BACK_CENTER][SIDE_RIGHT]+= M_SQRT1_2;
}else if(s->out_ch_layout & AV_CH_FRONT_LEFT){ }else if(out_ch_layout & AV_CH_FRONT_LEFT){
if (matrix_encoding == AV_MATRIX_ENCODING_DOLBY) { if (matrix_encoding == AV_MATRIX_ENCODING_DOLBY) {
matrix[FRONT_LEFT ][SIDE_LEFT ] -= s->slev * M_SQRT1_2; matrix[FRONT_LEFT ][SIDE_LEFT ] -= s->slev * M_SQRT1_2;
matrix[FRONT_LEFT ][SIDE_RIGHT] -= s->slev * M_SQRT1_2; matrix[FRONT_LEFT ][SIDE_RIGHT] -= s->slev * M_SQRT1_2;
...@@ -260,7 +262,7 @@ static int auto_matrix(SwrContext *s) ...@@ -260,7 +262,7 @@ static int auto_matrix(SwrContext *s)
matrix[ FRONT_LEFT][ SIDE_LEFT] += s->slev; matrix[ FRONT_LEFT][ SIDE_LEFT] += s->slev;
matrix[FRONT_RIGHT][SIDE_RIGHT] += s->slev; matrix[FRONT_RIGHT][SIDE_RIGHT] += s->slev;
} }
}else if(s->out_ch_layout & AV_CH_FRONT_CENTER){ }else if(out_ch_layout & AV_CH_FRONT_CENTER){
matrix[ FRONT_CENTER][SIDE_LEFT ]+= s->slev*M_SQRT1_2; matrix[ FRONT_CENTER][SIDE_LEFT ]+= s->slev*M_SQRT1_2;
matrix[ FRONT_CENTER][SIDE_RIGHT]+= s->slev*M_SQRT1_2; matrix[ FRONT_CENTER][SIDE_RIGHT]+= s->slev*M_SQRT1_2;
}else }else
...@@ -268,10 +270,10 @@ static int auto_matrix(SwrContext *s) ...@@ -268,10 +270,10 @@ static int auto_matrix(SwrContext *s)
} }
if(unaccounted & AV_CH_FRONT_LEFT_OF_CENTER){ if(unaccounted & AV_CH_FRONT_LEFT_OF_CENTER){
if(s->out_ch_layout & AV_CH_FRONT_LEFT){ if(out_ch_layout & AV_CH_FRONT_LEFT){
matrix[ FRONT_LEFT][ FRONT_LEFT_OF_CENTER]+= 1.0; matrix[ FRONT_LEFT][ FRONT_LEFT_OF_CENTER]+= 1.0;
matrix[FRONT_RIGHT][FRONT_RIGHT_OF_CENTER]+= 1.0; matrix[FRONT_RIGHT][FRONT_RIGHT_OF_CENTER]+= 1.0;
}else if(s->out_ch_layout & AV_CH_FRONT_CENTER){ }else if(out_ch_layout & AV_CH_FRONT_CENTER){
matrix[ FRONT_CENTER][ FRONT_LEFT_OF_CENTER]+= M_SQRT1_2; matrix[ FRONT_CENTER][ FRONT_LEFT_OF_CENTER]+= M_SQRT1_2;
matrix[ FRONT_CENTER][FRONT_RIGHT_OF_CENTER]+= M_SQRT1_2; matrix[ FRONT_CENTER][FRONT_RIGHT_OF_CENTER]+= M_SQRT1_2;
}else }else
...@@ -279,9 +281,9 @@ static int auto_matrix(SwrContext *s) ...@@ -279,9 +281,9 @@ static int auto_matrix(SwrContext *s)
} }
/* mix LFE into front left/right or center */ /* mix LFE into front left/right or center */
if (unaccounted & AV_CH_LOW_FREQUENCY) { if (unaccounted & AV_CH_LOW_FREQUENCY) {
if (s->out_ch_layout & AV_CH_FRONT_CENTER) { if (out_ch_layout & AV_CH_FRONT_CENTER) {
matrix[FRONT_CENTER][LOW_FREQUENCY] += s->lfe_mix_level; matrix[FRONT_CENTER][LOW_FREQUENCY] += s->lfe_mix_level;
} else if (s->out_ch_layout & AV_CH_FRONT_LEFT) { } else if (out_ch_layout & AV_CH_FRONT_LEFT) {
matrix[FRONT_LEFT ][LOW_FREQUENCY] += s->lfe_mix_level * M_SQRT1_2; matrix[FRONT_LEFT ][LOW_FREQUENCY] += s->lfe_mix_level * M_SQRT1_2;
matrix[FRONT_RIGHT][LOW_FREQUENCY] += s->lfe_mix_level * M_SQRT1_2; matrix[FRONT_RIGHT][LOW_FREQUENCY] += s->lfe_mix_level * M_SQRT1_2;
} else } else
...@@ -296,11 +298,11 @@ static int auto_matrix(SwrContext *s) ...@@ -296,11 +298,11 @@ static int auto_matrix(SwrContext *s)
if(matrix[i][j]){ if(matrix[i][j]){
sum += fabs(matrix[i][j]); sum += fabs(matrix[i][j]);
} }
if(s->in_ch_layout & (1ULL<<j)) if(in_ch_layout & (1ULL<<j))
in_i++; in_i++;
} }
maxcoef= FFMAX(maxcoef, sum); maxcoef= FFMAX(maxcoef, sum);
if(s->out_ch_layout & (1ULL<<i)) if(out_ch_layout & (1ULL<<i))
out_i++; out_i++;
} }
if(s->rematrix_volume < 0) if(s->rematrix_volume < 0)
...@@ -321,8 +323,8 @@ static int auto_matrix(SwrContext *s) ...@@ -321,8 +323,8 @@ static int auto_matrix(SwrContext *s)
} }
} }
for(i=0; i<av_get_channel_layout_nb_channels(s->out_ch_layout); i++){ for(i=0; i<av_get_channel_layout_nb_channels(out_ch_layout); i++){
for(j=0; j<av_get_channel_layout_nb_channels(s->in_ch_layout); j++){ for(j=0; j<av_get_channel_layout_nb_channels(in_ch_layout); j++){
av_log(NULL, AV_LOG_DEBUG, "%f ", s->matrix[i][j]); av_log(NULL, AV_LOG_DEBUG, "%f ", s->matrix[i][j]);
} }
av_log(NULL, AV_LOG_DEBUG, "\n"); av_log(NULL, AV_LOG_DEBUG, "\n");
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment