lib/psytune.c

   1 /********************************************************************
   2  *                                                                  *
   3  * THIS FILE IS PART OF THE Ogg Vorbis SOFTWARE CODEC SOURCE CODE.  *
   4  * USE, DISTRIBUTION AND REPRODUCTION OF THIS SOURCE IS GOVERNED BY *
   5  * THE GNU PUBLIC LICENSE 2, WHICH IS INCLUDED WITH THIS SOURCE.    *
   6  * PLEASE READ THESE TERMS DISTRIBUTING.                            *
   7  *                                                                  *
   8  * THE OggSQUISH SOURCE CODE IS (C) COPYRIGHT 1994-2000             *
   9  * by Monty <monty@xiph.org> and The XIPHOPHORUS Company            *
  10  * http://www.xiph.org/                                             *
  11  *                                                                  *
  12  ********************************************************************
  13
  14  function: simple utility that runs audio through the psychoacoustics
  15            without encoding
  16  last mod: $Id: psytune.c,v 1.2 2000/04/03 08:30:49 xiphmont Exp $
  17
  18  ********************************************************************/
  19
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include <math.h>
  24
  25 #include "vorbis/codec.h"
  26 #include "os.h"
  27 #include "psy.h"
  28 #include "mdct.h"
  29 #include "window.h"
  30
  31 static vorbis_info_psy _psy_set0={
  32   {-20, -20, -14, -14, -14, -14, -14, -14, -14, -14,
  33    -14, -14, -16, -16, -16, -16, -18, -18, -16, -16,
  34    -12, -10, -6, -3, -1, -1, -0}, 0., (.6/1024), 10,4
  35 };
  36
  37 int main(int argc,char *argv[]){
  38   int eos=0;
  39   double acc=0.;
  40   double tot=0.;
  41   int framesize=argv[1]?atoi(argv[1]):2048;
  42   double *pcm[2],*out[2],*window,*mask,*decay[2];
  43   signed char *buffer,*buffer2;
  44   mdct_lookup m_look;
  45   vorbis_look_psy p_look;
  46
  47   pcm[0]=malloc(framesize*sizeof(double));
  48   pcm[1]=malloc(framesize*sizeof(double));
  49   out[0]=calloc(framesize/2,sizeof(double));
  50   out[1]=calloc(framesize/2,sizeof(double));
  51   decay[0]=calloc(framesize/2,sizeof(double));
  52   decay[1]=calloc(framesize/2,sizeof(double));
  53   mask=malloc(framesize/2*sizeof(double));
  54   buffer=malloc(framesize*4);
  55   buffer2=buffer+framesize*2;
  56   window=_vorbis_window(0,framesize,framesize/2,framesize/2);
  57   mdct_init(&m_look,framesize);
  58   _vp_psy_init(&p_look,&_psy_set0,framesize/2,44100);
  59
  60   /* we cheat on the WAV header; we just bypass 44 bytes and never
  61      verify that it matches 16bit/stereo/44.1kHz. */
  62
  63   fread(buffer,1,44,stdin);
  64   fwrite(buffer,1,44,stdout);
  65   memset(buffer,0,framesize*2);
  66
  67   fprintf(stderr,"Processing for frame size %d...\n",framesize);
  68
  69   while(!eos){
  70     long i,j,k;
  71     long bytes=fread(buffer2,1,framesize*2,stdin);
  72     if(bytes<framesize*2)
  73       memset(buffer2+bytes,0,framesize*2-bytes);
  74
  75     if(bytes!=0){
  76
  77       /* uninterleave samples */
  78       for(i=0;i<framesize;i++){
  79         pcm[0][i]=((buffer[i*4+1]<<8)|
  80                       (0x00ff&(int)buffer[i*4]))/32768.;
  81         pcm[1][i]=((buffer[i*4+3]<<8)|
  82                    (0x00ff&(int)buffer[i*4+2]))/32768.;
  83       }
  84
  85       for(i=0;i<2;i++){
  86         for(j=0;j<framesize;j++)
  87           pcm[i][j]*=window[j];
  88
  89         mdct_forward(&m_look,pcm[i],pcm[i]);
  90
  91         /* do the psychacoustics */
  92
  93         memset(mask,0,sizeof(double)*framesize/2);
  94         _vp_mask_floor(&p_look,pcm[i],mask,decay[i],1);
  95
  96         /* quantize according to masking */
  97         for(j=0;j<framesize/2;j++){
  98           double val;
  99           if(mask[j]==0)
 100             val=0;
 101           else
 102             val=rint(pcm[i][j]/mask[j]);
 103           acc+=log(fabs(val)*2.+1.)/log(2);
 104           tot++;
 105           pcm[i][j]=val*mask[j];
 106         }
 107
 108         /* take it back to time */
 109         mdct_backward(&m_look,pcm[i],pcm[i]);
 110         for(j=0;j<framesize/2;j++)
 111           out[i][j]+=pcm[i][j]*window[j];
 112       }
 113
 114       /* write data.  Use the part of buffer we're about to shift out */
 115       for(i=0;i<2;i++){
 116         char *ptr=buffer+i*2;
 117         double  *mono=out[i];
 118         for(j=0;j<framesize/2;j++){
 119           int val=mono[j]*32767.;
 120           /* might as well guard against clipping */
 121           if(val>32767)val=32767;
 122           if(val<-32768)val=-32768;
 123           ptr[0]=val&0xff;
 124           ptr[1]=(val>>8)&0xff;
 125           ptr+=4;
 126         }
 127       }
 128
 129       fwrite(buffer,1,framesize*2,stdout);
 130       memmove(buffer,buffer2,framesize*2);
 131
 132       for(i=0;i<2;i++){
 133         for(j=0,k=framesize/2;j<framesize/2;j++,k++)
 134           out[i][j]=pcm[i][k]*window[k];
 135       }
 136     }else
 137       eos=1;
 138   }
 139   fprintf(stderr,"average raw bits of entropy: %.03g/sample\n",acc/tot);
 140   fprintf(stderr,"Done\n\n");
 141   return 0;
 142 }