ref: 98c796a020ba9bed691702b524f41bfbd609cdde
dir: /libfaac/psychkni.c/
/*
* FAAC - Freeware Advanced Audio Coder
* Copyright (C) 2002 Krzysztof Nikiel
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
* $Id: psychkni.c,v 1.1 2002/08/07 18:16:21 knik Exp $
*/
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include "psych.h"
#include "coder.h"
#include "fft.h"
#include "util.h"
#include "frame.h"
typedef struct
{
// bandwidth
int band;
int bandS;
/* FFT data */
/* energy */
double *fftEnrg;
double *fftEnrgNext;
double *fftEnrgNext2;
double *fftEnrgS[8];
double *fftEnrgNextS[8];
double *fftEnrgNext2S[8];
}
psydata_t;
static void Hann(GlobalPsyInfo * gpsyInfo, double *inSamples, int size)
{
int i;
/* Applying Hann window */
if (size == BLOCK_LEN_LONG * 2)
{
for (i = 0; i < size; i++)
inSamples[i] *= gpsyInfo->hannWindow[i];
}
else
{
for (i = 0; i < size; i++)
inSamples[i] *= gpsyInfo->hannWindowS[i];
}
}
static void PsyThreshold(GlobalPsyInfo * gpsyInfo, PsyInfo * psyInfo, int *cb_width_long,
int num_cb_long, int *cb_width_short, int num_cb_short)
{
int b, j;
double enb[8][NSFB_SHORT]; // total energy in each band in each window
double enchg[NSFB_SHORT]; // energy change
double totchg;
int lastband;
psydata_t *psydata = psyInfo->data;
/* Long window */
for (b = 0; b < num_cb_long; b++)
{
psyInfo->maskThr[b] = 0.8;
psyInfo->maskEn[b] = 1.0;
// extremely simple constant masking threshold
// but it works better than those elaborate but buggy models >:)
}
/* Short windows */
for (j = 0; j < 8; j++)
{
for (b = 0; b < num_cb_short; b++)
{
psyInfo->maskThrS[j][b] = 0.5;
psyInfo->maskEnS[j][b] = 1.0;
}
}
/* long/short block switch */
// compute energy
for (j = 0; j < 8; j++)
{
int l = 0;
for (b = 2; b < num_cb_short; b++)
{
int last = l + cb_width_short[b];
double e = 0;
for (; l < last; l++)
e += psydata->fftEnrgS[j][l];
if (l >= psydata->bandS)
break;
enb[j][b] = e;
//printf("en[%d][%d] = %g\n", j, b, e);
}
}
lastband = b;
// compare levels of energy in each band of short widows
totchg = 0.0;
for (b = 2; b < lastband; b++)
{
double maxdif = 0;
enchg[b] = 0.0;
for (j = 1; j < 7; j++)
{
double slowen = (1.0 / 7.0) * ((7 - j) * enb[0][b] + j * enb[7][b]);
double endif = enb[j][b] - slowen;
endif /= enb[j][b];
if (endif < 0)
endif = -endif;
if (endif > maxdif)
maxdif = endif;
}
enchg[b] += maxdif;
totchg += enchg[b];
//printf("enchg[%d] = %g\n", b, enchg[b]);
}
totchg = totchg / lastband;
psyInfo->block_type = (totchg > 7.0) ? ONLY_SHORT_WINDOW : ONLY_LONG_WINDOW;
#if 0
printf("totchg: %s %g\n", (psyInfo->block_type == ONLY_SHORT_WINDOW)
? "****" : " ", totchg);
#endif
}
static void PsyThresholdMS(ChannelInfo * channelInfoL, GlobalPsyInfo * gpsyInfo,
PsyInfo * psyInfoL, PsyInfo * psyInfoR, int
*cb_width_long, int num_cb_long, int *cb_width_short,
int num_cb_short)
{
int b, j;
// do nothing
// long
for (b = 0; b < num_cb_long; b++)
{
psyInfoL->maskThrMS[b] = 0.7;
psyInfoR->maskThrMS[b] = 0.7;
psyInfoL->maskEnMS[b] = 1.0;
psyInfoR->maskEnMS[b] = 1.0;
//channelInfoL->msInfo.ms_used[b] = 1;
channelInfoL->msInfo.ms_used[b] = 0;
}
/* Short windows */
for (j = 0; j < 8; j++)
{
for (b = 0; b < num_cb_short; b++)
{
psyInfoL->maskThrSMS[j][b] = 0.7;
psyInfoR->maskThrSMS[j][b] = 0.7;
psyInfoL->maskEnSMS[j][b] = 1.0;
psyInfoR->maskEnSMS[j][b] = 1.0;
}
//channelInfoL->msInfo.ms_usedS[j][b] = 1;
channelInfoL->msInfo.ms_usedS[j][b] = 0;
}
}
static void PsyInit(GlobalPsyInfo * gpsyInfo, PsyInfo * psyInfo, unsigned int numChannels,
unsigned int sampleRate, int *cb_width_long, int num_cb_long,
int *cb_width_short, int num_cb_short)
{
unsigned int channel;
int i, j, size;
gpsyInfo->hannWindow = (double *) AllocMemory(2 * BLOCK_LEN_LONG * sizeof(double));
gpsyInfo->hannWindowS = (double *) AllocMemory(2 * BLOCK_LEN_SHORT *
sizeof(double));
for (i = 0; i < BLOCK_LEN_LONG * 2; i++)
gpsyInfo->hannWindow[i] = 0.5 * (1 - cos(2.0 * M_PI * (i + 0.5) /
(BLOCK_LEN_LONG * 2)));
for (i = 0; i < BLOCK_LEN_SHORT * 2; i++)
gpsyInfo->hannWindowS[i] = 0.5 * (1 - cos(2.0 * M_PI * (i + 0.5) /
(BLOCK_LEN_SHORT * 2)));
gpsyInfo->sampleRate = (double) sampleRate;
for (channel = 0; channel < numChannels; channel++)
{
psydata_t *psydata = AllocMemory(sizeof(psydata_t));
psyInfo[channel].data = psydata;
}
size = BLOCK_LEN_LONG;
for (channel = 0; channel < numChannels; channel++)
{
psydata_t *psydata = psyInfo[channel].data;
psyInfo[channel].size = size;
psyInfo[channel].maskThr = (double *) AllocMemory(MAX_SCFAC_BANDS *
sizeof(double));
psyInfo[channel].maskEn = (double *) AllocMemory(MAX_SCFAC_BANDS *
sizeof(double));
psyInfo[channel].maskThrMS = (double *) AllocMemory(MAX_SCFAC_BANDS *
sizeof(double));
psyInfo[channel].maskEnMS = (double *) AllocMemory(MAX_SCFAC_BANDS *
sizeof(double));
psyInfo[channel].prevSamples = (double *) AllocMemory(size * sizeof(double));
memset(psyInfo[channel].prevSamples, 0, size * sizeof(double));
psydata->fftEnrg = (double *) AllocMemory(size * sizeof(double));
memset(psydata->fftEnrg, 0, size * sizeof(double));
psydata->fftEnrgNext = (double *) AllocMemory(size * sizeof(double));
memset(psydata->fftEnrgNext, 0, size * sizeof(double));
psydata->fftEnrgNext2 = (double *) AllocMemory(size * sizeof(double));
memset(psydata->fftEnrgNext2, 0, size * sizeof(double));
}
size = BLOCK_LEN_SHORT;
for (channel = 0; channel < numChannels; channel++)
{
psydata_t *psydata = psyInfo[channel].data;
psyInfo[channel].sizeS = size;
psyInfo[channel].prevSamplesS = (double *) AllocMemory(size * sizeof(double));
memset(psyInfo[channel].prevSamplesS, 0, size * sizeof(double));
for (j = 0; j < 8; j++)
{
psyInfo[channel].maskThrS[j] = (double *) AllocMemory(MAX_SCFAC_BANDS
* sizeof(double));
psyInfo[channel].maskEnS[j] = (double *) AllocMemory(MAX_SCFAC_BANDS *
sizeof(double));
psyInfo[channel].maskThrSMS[j] = (double *)
AllocMemory(MAX_SCFAC_BANDS
* sizeof(double));
psyInfo[channel].maskEnSMS[j] = (double *) AllocMemory(MAX_SCFAC_BANDS
* sizeof(double));
psydata->fftEnrgS[j] = (double *) AllocMemory(size * sizeof(double));
memset(psydata->fftEnrgS[j], 0, size * sizeof(double));
psydata->fftEnrgNextS[j] = (double *) AllocMemory(size * sizeof(double));
memset(psydata->fftEnrgNextS[j], 0, size * sizeof(double));
psydata->fftEnrgNext2S[j] = (double *) AllocMemory(size * sizeof(double));
memset(psydata->fftEnrgNext2S[j], 0, size * sizeof(double));
}
}
}
static void PsyEnd(GlobalPsyInfo * gpsyInfo, PsyInfo * psyInfo, unsigned int numChannels)
{
unsigned int channel;
int j;
if (gpsyInfo->hannWindow)
FreeMemory(gpsyInfo->hannWindow);
if (gpsyInfo->hannWindowS)
FreeMemory(gpsyInfo->hannWindowS);
for (channel = 0; channel < numChannels; channel++)
{
psydata_t *psydata = psyInfo[channel].data;
if (psyInfo[channel].prevSamples)
FreeMemory(psyInfo[channel].prevSamples);
if (psyInfo[channel].maskThr)
FreeMemory(psyInfo[channel].maskThr);
if (psyInfo[channel].maskEn)
FreeMemory(psyInfo[channel].maskEn);
if (psyInfo[channel].maskThrMS)
FreeMemory(psyInfo[channel].maskThrMS);
if (psyInfo[channel].maskEnMS)
FreeMemory(psyInfo[channel].maskEnMS);
if (psydata->fftEnrg)
FreeMemory(psydata->fftEnrg);
if (psydata->fftEnrgNext)
FreeMemory(psydata->fftEnrgNext);
if (psydata->fftEnrgNext2)
FreeMemory(psydata->fftEnrgNext2);
}
for (channel = 0; channel < numChannels; channel++)
{
psydata_t *psydata = psyInfo[channel].data;
if (psyInfo[channel].prevSamplesS)
FreeMemory(psyInfo[channel].prevSamplesS);
for (j = 0; j < 8; j++)
{
if (psyInfo[channel].maskThrS[j])
FreeMemory(psyInfo[channel].maskThrS[j]);
if (psyInfo[channel].maskEnS[j])
FreeMemory(psyInfo[channel].maskEnS[j]);
if (psyInfo[channel].maskThrSMS[j])
FreeMemory(psyInfo[channel].maskThrSMS[j]);
if (psyInfo[channel].maskEnSMS[j])
FreeMemory(psyInfo[channel].maskEnSMS[j]);
if (psydata->fftEnrgS[j])
FreeMemory(psydata->fftEnrgS[j]);
if (psydata->fftEnrgNextS[j])
FreeMemory(psydata->fftEnrgNextS[j]);
if (psydata->fftEnrgNext2S[j])
FreeMemory(psydata->fftEnrgNext2S[j]);
}
}
for (channel = 0; channel < numChannels; channel++)
{
if (psyInfo[channel].data)
FreeMemory(psyInfo[channel].data);
}
}
/* Do psychoacoustical analysis */
static void PsyCalculate(ChannelInfo * channelInfo, GlobalPsyInfo * gpsyInfo,
PsyInfo * psyInfo, int *cb_width_long, int
num_cb_long, int *cb_width_short,
int num_cb_short, unsigned int numChannels)
{
unsigned int channel;
for (channel = 0; channel < numChannels; channel++)
{
if (channelInfo[channel].present)
{
if (channelInfo[channel].cpe &&
channelInfo[channel].ch_is_left)
{ /* CPE */
int leftChan = channel;
int rightChan = channelInfo[channel].paired_ch;
/* Calculate the threshold */
PsyThreshold(gpsyInfo, &psyInfo[leftChan], cb_width_long, num_cb_long,
cb_width_short, num_cb_short);
PsyThreshold(gpsyInfo, &psyInfo[rightChan], cb_width_long, num_cb_long,
cb_width_short, num_cb_short);
/* And for MS */
PsyThresholdMS(&channelInfo[leftChan], gpsyInfo, &psyInfo[leftChan],
&psyInfo[rightChan], cb_width_long, num_cb_long, cb_width_short,
num_cb_short);
}
else if (!channelInfo[channel].cpe &&
channelInfo[channel].lfe)
{ /* LFE */
/* NOT FINISHED */
}
else if (!channelInfo[channel].cpe)
{ /* SCE */
/* Calculate the threshold */
PsyThreshold(gpsyInfo, &psyInfo[channel], cb_width_long, num_cb_long,
cb_width_short, num_cb_short);
}
}
}
}
static void PsyBufferUpdate(GlobalPsyInfo * gpsyInfo, PsyInfo * psyInfo, double
*newSamples, unsigned int bandwidth)
{
int i, j;
double a, b;
double *transBuff, *transBuffS;
psydata_t *psydata = psyInfo->data;
double *tmp;
psydata->band = psyInfo->size * bandwidth * 2 / gpsyInfo->sampleRate;
psydata->bandS = psyInfo->sizeS * bandwidth * 2 / gpsyInfo->sampleRate;
transBuff = (double *) AllocMemory(2 * psyInfo->size * sizeof(double));
memcpy(transBuff, psyInfo->prevSamples, psyInfo->size * sizeof(double));
memcpy(transBuff + psyInfo->size, newSamples, psyInfo->size * sizeof(double));
Hann(gpsyInfo, transBuff, 2 * psyInfo->size);
rsfft(transBuff, 11);
// shift bufs
tmp = psydata->fftEnrg;
psydata->fftEnrg = psydata->fftEnrgNext;
psydata->fftEnrgNext = psydata->fftEnrgNext2;
psydata->fftEnrgNext2 = tmp;
for (i = 0; i < psydata->band; i++)
{
a = transBuff[i];
b = transBuff[i + psyInfo->size];
// spectral line energy
psydata->fftEnrgNext2[i] = a * a + b * b;
//printf("psyInfo->fftEnrg[%d]: %g\n", i, psyInfo->fftEnrg[i]);
}
for (; i < psyInfo->size; i++)
{
psydata->fftEnrgNext2[i] = 0;
//printf("psyInfo->fftEnrg[%d]: %g\n", i, psyInfo->fftEnrg[i]);
}
transBuffS = (double *) AllocMemory(2 * psyInfo->sizeS * sizeof(double));
memcpy(transBuff, psyInfo->prevSamples, psyInfo->size * sizeof(double));
memcpy(transBuff + psyInfo->size, newSamples, psyInfo->size * sizeof(double));
for (j = 0; j < 8; j++)
{
memcpy(transBuffS, transBuff + (j * 128) + (1024 - 128),
2 * psyInfo->sizeS * sizeof(double));
Hann(gpsyInfo, transBuffS, 2 * psyInfo->sizeS);
rsfft(transBuffS, 8);
// shift bufs
tmp = psydata->fftEnrgS[j];
psydata->fftEnrgS[j] = psydata->fftEnrgNextS[j];
psydata->fftEnrgNextS[j] = psydata->fftEnrgNext2S[j];
psydata->fftEnrgNext2S[j] = tmp;
for (i = 0; i < psydata->bandS; i++)
{
a = transBuffS[i];
b = transBuffS[i + psyInfo->sizeS];
// spectral line energy
psydata->fftEnrgNext2S[j][i] = a * a + b * b;
//printf("psyInfo->fftEnrgS[%d][%d]: %g\n", j, i, psyInfo->fftEnrgS[j][i]);
}
for (; i < psyInfo->sizeS; i++)
{
psydata->fftEnrgNext2S[j][i] = 0;
//printf("psyInfo->fftEnrgS[%d][%d]: %g\n", j, i, psyInfo->fftEnrgS[j][i]);
}
}
memcpy(psyInfo->prevSamples, newSamples, psyInfo->size * sizeof(double));
if (transBuff)
FreeMemory(transBuff);
if (transBuffS)
FreeMemory(transBuffS);
}
static void BlockSwitch(CoderInfo * coderInfo, PsyInfo * psyInfo, unsigned int numChannels)
{
unsigned int channel;
int desire = ONLY_LONG_WINDOW;
/* Use the same block type for all channels
If there is 1 channel that wants a short block,
use a short block on all channels.
*/
for (channel = 0; channel < numChannels; channel++)
{
if (psyInfo[channel].block_type == ONLY_SHORT_WINDOW)
desire = ONLY_SHORT_WINDOW;
}
for (channel = 0; channel < numChannels; channel++)
{
int lasttype = coderInfo[channel].block_type;
if (desire == ONLY_SHORT_WINDOW
|| coderInfo[channel].desired_block_type == ONLY_SHORT_WINDOW)
{
if (lasttype == ONLY_LONG_WINDOW || lasttype == SHORT_LONG_WINDOW)
coderInfo[channel].block_type = LONG_SHORT_WINDOW;
else
coderInfo[channel].block_type = ONLY_SHORT_WINDOW;
}
else
{
if (lasttype == ONLY_SHORT_WINDOW || lasttype == LONG_SHORT_WINDOW)
coderInfo[channel].block_type = SHORT_LONG_WINDOW;
else
coderInfo[channel].block_type = ONLY_LONG_WINDOW;
}
coderInfo[channel].desired_block_type = desire;
}
}
psymodel_t psymodel2 =
{
PsyInit,
PsyEnd,
PsyCalculate,
PsyBufferUpdate,
BlockSwitch
};